libscratchcpp/src/engine/internal/llvm/instructions/comparison.cpp at 0c65342044a6956864984122f73228b19bd46170 · scratchcpp/libscratchcpp · GitHub

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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
// SPDX-License-Identifier: Apache-2.0

#include "comparison.h"
#include "../llvminstruction.h"
#include "../llvmbuildutils.h"

using namespace libscratchcpp;
using namespace libscratchcpp::llvmins;

ProcessResult Comparison::process(LLVMInstruction *ins)
{
    ProcessResult ret(true, ins);

    switch (ins->type) {
        case LLVMInstruction::Type::CmpEQ:
            ret.next = buildCmpEQ(ins);
            break;

        case LLVMInstruction::Type::CmpGT:
            ret.next = buildCmpGT(ins);
            break;

        case LLVMInstruction::Type::CmpLT:
            ret.next = buildCmpLT(ins);
            break;

        case LLVMInstruction::Type::StrCmpEQCS:
            ret.next = buildStrCmpEQCS(ins);
            break;

        case LLVMInstruction::Type::StrCmpEQCI:
            ret.next = buildStrCmpEQCI(ins);
            break;

        default:
            ret.match = false;
            break;
    }

    return ret;
}

LLVMInstruction *Comparison::buildCmpEQ(LLVMInstruction *ins)
{
    assert(ins->args.size() == 2);
    const auto &arg1 = ins->args[0].second;
    const auto &arg2 = ins->args[1].second;
    ins->functionReturnReg->value = m_utils.createComparison(arg1, arg2, LLVMBuildUtils::Comparison::EQ);

    return ins->next;
}

LLVMInstruction *Comparison::buildCmpGT(LLVMInstruction *ins)
{
    assert(ins->args.size() == 2);
    const auto &arg1 = ins->args[0].second;
    const auto &arg2 = ins->args[1].second;
    ins->functionReturnReg->value = m_utils.createComparison(arg1, arg2, LLVMBuildUtils::Comparison::GT);

    return ins->next;
}

LLVMInstruction *Comparison::buildCmpLT(LLVMInstruction *ins)
{
    assert(ins->args.size() == 2);
    const auto &arg1 = ins->args[0].second;
    const auto &arg2 = ins->args[1].second;
    ins->functionReturnReg->value = m_utils.createComparison(arg1, arg2, LLVMBuildUtils::Comparison::LT);

    return ins->next;
}

LLVMInstruction *Comparison::buildStrCmpEQCS(LLVMInstruction *ins)
{
    assert(ins->args.size() == 2);
    const auto &arg1 = ins->args[0].second;
    const auto &arg2 = ins->args[1].second;
    ins->functionReturnReg->value = m_utils.createStringComparison(arg1, arg2, true);

    return ins->next;
}

LLVMInstruction *Comparison::buildStrCmpEQCI(LLVMInstruction *ins)
{
    assert(ins->args.size() == 2);
    const auto &arg1 = ins->args[0].second;
    const auto &arg2 = ins->args[1].second;
    ins->functionReturnReg->value = m_utils.createStringComparison(arg1, arg2, false);

    return ins->next;
}