llvmcodegen.hpp

#ifndef CODEGEN_HPP
#define CODEGEN_HPP

#include <map>
#include <string>
#include <vector>
#include <memory>
#include <llvm/IR/IRBuilder.h>
#include <llvm/IR/LLVMContext.h>
#include <llvm/IR/Module.h>

struct ArrayInfo {
    llvm::Value* alloc;
    std::vector<int> dims;
};

class ASTNode; // Forward declaration

class CodeGenContext {
public:
    llvm::LLVMContext context;
    llvm::IRBuilder<> builder;
    std::unique_ptr<llvm::Module> module;
    llvm::Value *currentValue;
    std::map<std::string, ArrayInfo> arrays;

    CodeGenContext() : builder(context) {
        module = std::make_unique<llvm::Module>("main", context);
        currentValue = nullptr;
    }

    void generateCode(ASTNode* root);
    void writeIR(const std::string &filename);
    void setCurrentValue(llvm::Value *v) { currentValue = v; }
    llvm::Value* getCurrentValue() { return currentValue; }
    llvm::Type* getDoublePtrTy() {
        return llvm::PointerType::get(builder.getDoubleTy(),0);
    }
};

#endif // CODEGEN_HPP