llvmcodegen.cpp

#include "llvmcodegen.hpp"
#include "ast.hpp"
#include <llvm/IR/Verifier.h>
#include <llvm/Support/raw_ostream.h>
#include <llvm/Support/FileSystem.h>

void CodeGenContext::generateCode(ASTNode* root) {
    // Create main function
    llvm::FunctionType *ftype = llvm::FunctionType::get(builder.getInt32Ty(), false);
    llvm::Function *mainFunc = llvm::Function::Create(ftype, llvm::GlobalValue::ExternalLinkage, "main", module.get());
    llvm::BasicBlock *bblock = llvm::BasicBlock::Create(context, "entry", mainFunc);
    builder.SetInsertPoint(bblock);

    // Generate code from AST
    root->codegen(*this);

    // Ensure main returns 0
    builder.CreateRet(llvm::ConstantInt::get(builder.getInt32Ty(), 0));

    // Validate generated code
    llvm::verifyFunction(*mainFunc);
    llvm::verifyModule(*module, &llvm::errs());
}

void CodeGenContext::writeIR(const std::string &filename) {
    std::error_code EC;
    llvm::raw_fd_ostream out(filename, EC, llvm::sys::fs::OpenFlags::OF_Text);
    if (EC) {
        fprintf(stderr, "Could not open file: %s\n", EC.message().c_str());
        return;
    }
    module->print(out, nullptr);
}