ICmp.h

Go to the documentation of this file.

// SPDX-License-Identifier: MIT
 
#pragma once
 
#include <cstdlib>
#include <memory>
#include <string>
#include <utility>
 
#include "mini-llvm/ir/Constant.h"
#include "mini-llvm/ir/Instruction/BinaryIntegerRelationalOperator.h"
#include "mini-llvm/ir/InstructionVisitor.h"
#include "mini-llvm/ir/Value.h"
#include "mini-llvm/utils/Compiler.h"
 
namespace mini_llvm::ir {
 
class MINI_LLVM_EXPORT ICmp final : public BinaryIntegerRelationalOperator {
public:
    enum class Condition {
        kEQ,
        kNE,
        kSLT,
        kSGT,
        kSLE,
        kSGE,
        kULT,
        kUGT,
        kULE,
        kUGE,
    };
 
    ICmp(Condition cond, std::shared_ptr<Value> lhs, std::shared_ptr<Value> rhs)
        : BinaryIntegerRelationalOperator(std::move(lhs), std::move(rhs)), cond_(cond) {}
 
    Condition cond() const {
        return cond_;
    }
 
    void setCond(Condition cond) {
        cond_ = cond;
    }
 
    std::shared_ptr<Constant> fold() const override;
 
    void accept(InstructionVisitor &visitor) override {
        visitor.visitICmp(*this);
    }
 
    void accept(InstructionVisitor &visitor) const override {
        visitor.visitICmp(*this);
    }
 
    std::string format() const override;
    std::unique_ptr<Value> clone() const override;
 
private:
    Condition cond_;
};
 
inline constexpr const char *specifier(ICmp::Condition cond) {
    using enum ICmp::Condition;
    switch (cond) {
        case kEQ: return "eq";
        case kNE: return "ne";
        case kSLT: return "slt";
        case kSGT: return "sgt";
        case kSLE: return "sle";
        case kSGE: return "sge";
        case kULT: return "ult";
        case kUGT: return "ugt";
        case kULE: return "ule";
        case kUGE: return "uge";
        default: abort();
    }
}
 
} // namespace mini_llvm::ir