Class CFuncNode

libchigraph/src/GraphModule.cpp:176–316

The NodeType for calling C functions

/// The NodeType for calling C functions
struct CFuncNode : NodeType {
	CFuncNode(GraphModule& mod, std::string cCode, std::string functionName,
	          std::vector<std::string> extraArgs, std::vector<NamedDataType> inputs,
	          DataType output)
	    : NodeType{mod, "c-call", "call C code"},
	      mFunctionName{std::move(functionName)},
	      mCCode(std::move(cCode)),
	      mExtraArguments{std::move(extraArgs)},
	      mInputs{std::move(inputs)},
	      mOutput{std::move(output)},
	      mGraphModule{&mod} {
		setExecInputs({""});
		setExecOutputs({""});

		setDataInputs(mInputs);
		if (mOutput.valid()) { setDataOutputs({{"", mOutput}}); }
	}

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

		Result res;

		// compile the c code if it hasn't already been compiled
		if (llcompiledmod == nullptr) {
			auto args = mExtraArguments;

			// add -I for the .c dir
			args.push_back("-I");
			args.push_back(mGraphModule->pathToCSources().string());

			llcompiledmod = compileCCode(llvm::sys::fs::getMainExecutable(nullptr, nullptr).c_str(),
			                             mCCode, args, context().llvmContext(), res);

			if (!res) { return res; }
		}

		// create a copy of the module
		auto copymod = llvm::CloneModule(llcompiledmod.get());

		// link it in

		auto parentModule = codegenInto
		                        ->
#if LLVM_VERSION_LESS_EQUAL(3, 6)
		                    getParent()
		                        ->getParent()
#else
		                    getModule()
#endif
		    ;

#if LLVM_VERSION_LESS_EQUAL(3, 7)
		llvm::Linker::LinkModules(parentModule, copymod