Class EntryNodeType

libchigraph/src/LangModule.cpp:50–107

};

struct EntryNodeType : NodeType {
	EntryNodeType(LangModule& mod, std::vector<NamedDataType> dataInputs,
	              std::vector<std::string> execInputs)
	    : NodeType(mod, "entry", "entry to a function") {
		setExecOutputs(std::move(execInputs));

		setDataOutputs(std::move(dataInputs));
	}

	Result codegen(
	    size_t /*inputExecID*/, 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() == dataOutputs().size() && codegenInto != nullptr &&
		       outputBlocks.size() == execOutputs().size());

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

		// store the arguments
		auto arg_iter = codegenInto->getParent()->arg_begin();
		++arg_iter;  // skip the first argument, which is the input exec ID
		for (const auto& iovalue : io) {
			builder.CreateStore(&*arg_iter, iovalue);

			++arg_iter;
		}

		auto inExecID   = &*codegenInto->getParent()->arg_begin();
		auto switchInst = builder.CreateSwitch(inExecID, outputBlocks[0], execOutputs().size());

		for (auto id = 0ull; id < execOutputs().size(); ++id) {
			switchInst->addCase(builder.getInt32(id), outputBlocks[id]);
		}
		return {};
	}

	std::unique_ptr<NodeType> clone() const override {
		return std::make_unique<EntryNodeType>(*this);
	}

	nlohmann::json toJSON() const override {
		nlohmann::json ret = nlohmann::json::object();

		auto& data = ret["data"];
		data       = nlohmann::json::array();
		for (auto& pair : dataOutputs()) {
			data.push_back({{pair.name, pair.type.qualifiedName()}});
		}

		auto& exec = ret["exec"];
		exec       = nlohmann::json::array();
		for (auto& name : execOutputs()) { exec.push_back(name); }

		return ret;
	}
};