Set up visitor (#3)

Co-authored-by: geno <gabriele.genovese2@studio.unibo.it>

1 files changed, 64 insertions, 0 deletions

diff --git a/src/ast/nodes/RootNode.java b/src/ast/nodes/RootNode.java
new file mode 100644
index 0000000..12d2f22
--- /dev/null
+++ b/src/ast/nodes/RootNode.java
@@ -0,0 +1,64 @@
+package com.clp.project.ast.nodes;
+
+import java.util.ArrayList;
+
+import com.clp.project.semanticanalysis.SemanticError;
+import com.clp.project.semanticanalysis.SymbolTable;
+import com.clp.project.ast.types.*;
+
+/**
+ * Node for the `root` statement of the grammar.
+ */
+public class RootNode implements Node {
+    // stms and compundStmts are protected because they are reused for a
+    // BlockNode
+    protected ArrayList<Node> stmts;
+    protected ArrayList<Node> compoundStmts;
+
+    public RootNode(ArrayList<Node> stmts, ArrayList<Node> compoundStmts) {
+        this.stmts = stmts;
+        this.compoundStmts = compoundStmts;
+    }
+
+    @Override
+    public ArrayList<SemanticError> checkSemantics(SymbolTable ST, int _nesting) {
+        ArrayList<SemanticError> errors = new ArrayList<SemanticError>();
+
+        for (Node stmt : stmts) {
+            errors.addAll(stmt.checkSemantics(ST, _nesting));
+        }
+        for (Node stmt : compoundStmts) {
+            errors.addAll(stmt.checkSemantics(ST, _nesting));
+        }
+
+        return errors;
+    }
+
+    @Override
+    public Type typeCheck() {
+        return new VoidType();
+    }
+
+    // TODO: Code generation for RootNode
+    @Override
+    public String codeGeneration() {
+        return "";
+    }
+
+    @Override
+    public String toPrint(String prefix) {
+        String str = "Root\n";
+
+        prefix += "  ";
+
+        for (Node stmt : stmts) {
+            str += stmt.toPrint(prefix);
+        }
+        for (Node stmt : compoundStmts) {
+            str += stmt.toPrint(prefix);
+        }
+
+        return str;
+    }
+
+}