semantic.hpp

#pragma once
#include <string>
#include <vector>
#include <unordered_map>
#include "ast.hpp"

struct SemanticResult {
    bool ok;
    std::vector<std::string> errors;
    std::vector<std::string> warnings;
    std::vector<std::string> optimizations;

    SemanticResult() : ok(true) {}

    void addError(const std::string& s) {
        errors.push_back(s);
        ok = false;
    }
};

class SemanticAnalyzer {
public:
    SemanticAnalyzer();
    SemanticResult analyze(AST::Program* p);

private:
    std::unordered_map<std::string, AST::ClassDecl*> classes;
    AST::ClassDecl* curClass;
    AST::MethodDecl* curMethod;

    std::vector<std::unordered_map<std::string, std::string>> localsStack;
    SemanticResult result;
    std::vector<std::string> declaredLocals;
    std::vector<std::string> usedLocals;

    void indexClasses(AST::Program* p);
    void analyzeClass(AST::ClassDecl* c);
    void analyzeFields(AST::ClassDecl* c);
    void analyzeCtor(AST::CtorDecl* c);
    void analyzeMethod(AST::MethodDecl* m);
    void analyzeBlock(AST::Block* b);
    void analyzeStmt(AST::Stmt*& s);
    void analyzeExpr(AST::Expr*& e);

    std::string typeOfExpr(AST::Expr* e);

    void pushScope();
    void popScope();
    void declareLocal(const std::string& name, const std::string& type);
    void markUsed(const std::string& name);

    bool foldConstantsInExpr(AST::Expr*& e);
    void simplifyIf(AST::Stmt*& s);
    void removeUnreachableInBlock(AST::Block* b);
};