Class BinaryOperationNodeType

libchigraph/src/LangModule.cpp:375–459

enum class BinOp { Add, Subtract, Multiply, Divide };

struct BinaryOperationNodeType : NodeType {
	BinaryOperationNodeType(LangModule& mod, DataType ty, BinOp binaryOperation)
	    : NodeType(mod), mBinOp(binaryOperation), mType{ty} {
		makePure();

		std::string opStr = [](BinOp b) {
			switch (b) {
			case BinOp::Add: return "+";
			case BinOp::Subtract: return "-";
			case BinOp::Multiply: return "*";
			case BinOp::Divide: return "/";
			default: return "";
			}
			return "";
		}(binaryOperation);

		setName(ty.unqualifiedName() + opStr + ty.unqualifiedName());

		std::string opVerb = [](BinOp b) {
			switch (b) {
			case BinOp::Add: return "Add";
			case BinOp::Subtract: return "Subtract";
			case BinOp::Multiply: return "Multiply";
			case BinOp::Divide: return "Divide";
			default: return "";
			}
			return "";
		}(binaryOperation);

		setDescription(opVerb + " two " + ty.unqualifiedName() + "s");

		setDataInputs({{"a", ty}, {"b", ty}});
		setDataOutputs({{"", ty}});
	}

	Result codegen(
	    size_t /*execInputID*/, const llvm::DebugLoc& nodeLocation,
	    const std::vector<llvm::Value*>& io, llvm::BasicBlock* codegenInto,
	    const std::vector<llvm::BasicBlock*>& outputBlocks,
	    std::unordered_map<std::string, std::shared_ptr<void>>& /*compileCache*/) override {
		assert(io.size() == 3 && codegenInto != nullptr && outputBlocks.size() == 1);

		llvm::IRBuilder<> builder(codegenInto);
		builder.SetCurrentDebugLocation(nodeLocation);

		llvm::Value* result = nullptr;
		if (mType.unqualifiedName() == "i32") {
			result = [&](BinOp b) {
				switch (b) {
				case BinOp::Add: return builder.CreateAdd(io[0], io[1]);
				case BinOp::Subtract: return builder.CreateSub(io[0], io[1]);
				case BinOp::Multiply: return builder.CreateMul(io[0], io[1]);
				case BinOp::Divide: return builder.CreateSDiv(io[0], io[1]);
				default: return (llvm::Value*)nullptr;
				}
				return (llvm::Value*)nullptr;
			}(mBinOp);