ast.h

Go to the documentation of this file.

#ifndef AST_H
#define AST_H
 
#include <cmath>
 
#include "../generator/generator.h"
#include "../shared/enums.h"
#include "../shared/globalStructs.h"
 
void printIndentationSpaces(int indent);
 
std::string generateIfCondition(Generator& generator);
 
class Node {
   public:
    virtual ~Node() = default;
 
    virtual void gen(Generator&) = 0;
 
    virtual void print(int indent = 0) = 0;
};
 
class StatementCode : public Node {
   private:
    std::shared_ptr<Node> stmt;  // The statement node
    std::shared_ptr<Node> code;  // The following code block
 
   public:
    StatementCode(std::shared_ptr<Node> stmt, std::shared_ptr<Node> code) : stmt(stmt), code(code) {
    }
 
    void gen(Generator&) override;
 
    void print(int) override;
};
 
class LambdaCode : public Node {
   public:
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class Id : public Node {
   private:
    std::string id;  // The identifier's name
 
   public:
    Id(std::string id) : id(id) {
    }
 
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class Insert : public Node {
   public:
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class Remove : public Node {
   public:
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class New : public Node {
   public:
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class Contains : public Node {
   public:
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class Loop : public Node {
   private:
    std::shared_ptr<Node> code;  // The code block to be executed in the loop
 
   public:
    Loop(std::shared_ptr<Node> code) : code(code) {
    }
 
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class Call : public Node {
   private:
    int id;                      // The ID of the function being called
    std::shared_ptr<Node> code;  // The code block of the function
 
   public:
    int conditionalCounts;  // Tracks the number of conditional statements in the call
 
    Call(int id, std::shared_ptr<Node> code) : id(id), code(code), conditionalCounts(0) {
    }
 
    Call() : conditionalCounts(0) {
    }
 
    void setId(int id) {
        this->id = id;
    }
 
    void setCode(std::shared_ptr<Node> code) {
        this->code = code;
    }
 
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class Seq : public Node {
   private:
    std::shared_ptr<Node> code;  // The code block for the sequence
 
   public:
    Seq(std::shared_ptr<Node> code) : code(code) {
    }
 
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class If : public Node {
   private:
    std::shared_ptr<Node> ifParam;  // The parameters for the if statement
 
   public:
    If(std::shared_ptr<Node> ifParam) : ifParam(ifParam) {
    }
 
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class IfParam : public Node {
   private:
    std::shared_ptr<Node> code;   // The code block for the if statement
    std::shared_ptr<Node> else_;  // The else clause
 
   public:
    IfParam(std::shared_ptr<Node> code, std::shared_ptr<Node> else_) : code(code), else_(else_) {
    }
 
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
class CodeElse : public Node {
   private:
    std::shared_ptr<Node> code;  // The code block for the else clause
 
   public:
    CodeElse(std::shared_ptr<Node> code) : code(code) {
    }
 
    void gen(Generator&) override;
 
    void print(int indent) override;
};
 
#endif