Added INPUT, OUTPUT and DECLARE

1 files changed, 63 insertions, 0 deletions

diff --git a/include/cps/AST.h b/include/cps/AST.h
new file mode 100644
index 0000000..5b28e8a
--- /dev/null
+++ b/include/cps/AST.h
@@ -0,0 +1,63 @@
+#pragma once
+#include <string>
+#include <memory>
+#include <vector>
+
+namespace cps {
+
+class ExprAST {
+public:
+    virtual ~ExprAST() = default;
+};
+
+class NumberExprAST : public ExprAST {
+    int Val;
+public:
+    NumberExprAST(int Val) : Val(Val) {}
+    int getVal() const { return Val; }
+};
+
+class VariableExprAST : public ExprAST {
+    std::string Name;
+public:
+    VariableExprAST(const std::string &Name) : Name(Name) {}
+    const std::string &getName() const { return Name; }
+};
+
+class StmtAST {
+public:
+    virtual ~StmtAST() = default;
+};
+
+class DeclareStmtAST : public StmtAST {
+    std::string Name;
+public:
+    DeclareStmtAST(const std::string &Name) : Name(Name) {}
+    const std::string &getName() const { return Name; }
+};
+
+class AssignStmtAST : public StmtAST {
+    std::string Name;
+    std::unique_ptr<ExprAST> Expr;
+public:
+    AssignStmtAST(const std::string &Name, std::unique_ptr<ExprAST> Expr)
+        : Name(Name), Expr(std::move(Expr)) {}
+    const std::string &getName() const { return Name; }
+    ExprAST *getExpr() const { return Expr.get(); }
+};
+
+class InputStmtAST : public StmtAST {
+    std::string Name;
+public:
+    InputStmtAST(const std::string &Name) : Name(Name) {}
+    const std::string &getName() const { return Name; }
+};
+
+class OutputStmtAST : public StmtAST {
+    std::unique_ptr<ExprAST> Expr;
+public:
+    OutputStmtAST(std::unique_ptr<ExprAST> Expr) : Expr(std::move(Expr)) {}
+    ExprAST *getExpr() const { return Expr.get(); }
+};
+
+} // namespace cps