aboutsummaryrefslogtreecommitdiff
path: root/lib/CodeGen/ArithmeticHandler.cc
blob: 762f0a313dfb113b3257240ef674eeca0ac9fddd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

#include "cps/ArithmeticHandler.h"
#include <cstdio>

using namespace llvm;
using namespace cps;

Value *ArithmeticHandler::emitBinaryOp(int Op, Value *LHS, Value *RHS, int Line) {
    bool LIsInt = LHS->getType()->isIntegerTy(64);
    bool RIsInt = RHS->getType()->isIntegerTy(64);
    bool LIsDouble = LHS->getType()->isDoubleTy();
    bool RIsDouble = RHS->getType()->isDoubleTy();

    if (Op == '/') {
        Value *LVal = LHS;
        Value *RVal = RHS;
        if (LIsInt) LVal = Builder.CreateSIToFP(LHS, Type::getDoubleTy(Context));
        if (RIsInt) RVal = Builder.CreateSIToFP(RHS, Type::getDoubleTy(Context));
        
        // TODO: DIV divide by 0 runtime check
        return Builder.CreateFDiv(LVal, RVal, "divtmp");
    }

    // DIV / MOD : Only Integer.
    if (Op == tok_div || Op == tok_mod) {
        if (!LIsInt || !RIsInt) {
            fprintf(stderr, "Error: DIV and MOD operators require INTEGER operands.\n");
            return nullptr;
        }
        if (Op == tok_div) return Builder.CreateSDiv(LHS, RHS, "div_int_tmp");
        if (Op == tok_mod) return Builder.CreateSRem(LHS, RHS, "mod_tmp");
    }

    if (LIsDouble || RIsDouble) {
        Value *LVal = LHS;
        Value *RVal = RHS;
        if (LIsInt) LVal = Builder.CreateSIToFP(LHS, Type::getDoubleTy(Context));
        if (RIsInt) RVal = Builder.CreateSIToFP(RHS, Type::getDoubleTy(Context));
        
        switch (Op) {
            case '+': return Builder.CreateFAdd(LVal, RVal, "addtmp");
            case '-': return Builder.CreateFSub(LVal, RVal, "subtmp");
            case '*': return Builder.CreateFMul(LVal, RVal, "multmp");

            case tok_eq: return Builder.CreateFCmpOEQ(LVal, RVal, "eqtmp");
            case tok_ne: return Builder.CreateFCmpONE(LVal, RVal, "netmp");
            case '<':    return Builder.CreateFCmpOLT(LVal, RVal, "slttmp");
            case '>':    return Builder.CreateFCmpOGT(LVal, RVal, "sgttmp");
            case tok_le: return Builder.CreateFCmpOLE(LVal, RVal, "sletmp");
            case tok_ge: return Builder.CreateFCmpOGE(LVal, RVal, "sgetmp");
        }
    } else {
        switch (Op) {
            case '+': return Builder.CreateAdd(LHS, RHS, "addtmp");
            case '-': return Builder.CreateSub(LHS, RHS, "subtmp");
            case '*': return Builder.CreateMul(LHS, RHS, "multmp");
            
            case tok_eq: return Builder.CreateICmpEQ(LHS, RHS, "eqtmp");
            case tok_ne: return Builder.CreateICmpNE(LHS, RHS, "netmp");
            case '<':    return Builder.CreateICmpSLT(LHS, RHS, "slttmp");
            case '>':    return Builder.CreateICmpSGT(LHS, RHS, "sgttmp");
            case tok_le: return Builder.CreateICmpSLE(LHS, RHS, "sletmp");
            case tok_ge: return Builder.CreateICmpSGE(LHS, RHS, "sgetmp");
        }
    }
    
    return nullptr;
}