generator.h

Go to the documentation of this file.

#ifndef GENERATOR_H
#define GENERATOR_H
#include <filesystem>
#include <fstream>
#include <iostream>
 
#include "../shared/enums.h"
#include "../shared/globalStructs.h"
#include "../shared/varTypes.h"
#include "generatorFunction.h"
#include "generatorScope.h"
#include "generatorVariable.h"
 
class Generator {
   private:
    std::vector<std::string> includes;    // List of include statements for the generated code
    std::vector<std::string> globalVars;  // List of global variable declarations
 
    void generateIncludes();
 
    void generateGlobalVars();
 
    void generateMainFunction();
 
    void generateRandomNumberGenerator();
 
    std::string createParams();
 
    void genMakefile(std::string dir, std::string target);
 
   public:
    GeneratorFunction mainFunction;                  // Main function for the generated program
    std::list<GeneratorFunction> functions;          // List of all functions in the generated program
    std::stack<GeneratorFunction*> currentFunction;  // Stack of current functions being generated
    std::stack<int> ifCounter;                       // Counter for managing nested if statements
    int varCounter;                                  // Counter for variables
    int loopLevel;                                   // Current nesting level of loops
    int loopCounter;                                 // Counter for loop iterations
    std::string varType;                             // Type of variables to use in the generated code
    std::map<int, GeneratorVariable*> variables;     // Map of variables by their ID
    std::stack<GeneratorScope> currentScope;         // Stack of current scopes
 
    Generator(std::string variableType);
 
    ~Generator() {
        for (auto& vpair : variables) {
            delete vpair.second;
        }
    }
 
    void addLine(std::string, int = 0);
 
    void addLine(std::vector<std::string>, int = 0);
 
    void startScope();
 
    void startFunc(int, int);
 
    bool functionExists(int);
 
    void callFunc(int, int);
 
    int addVar(std::string);
 
    void freeVars(bool = false, int = 0);
 
    void returnFunc(int);
 
    void endScope();
 
    void endFunc();
 
    void generateFiles(std::string);
};
 
#endif