PP_Visitor.java

import syntaxtree.*;

/* 
 * this class uses the Visitor pattern 
 * to pretty print the miniC program
 * The idea is to have each node return its string representation!
*/



/* 
 * Video 1: https://brandeis.zoom.us/rec/share/1MhxpqLQM9OeNTOUnk6lNUReAgUGw0aMtWXXXoq9lN1VzqE1bYo8rIo-J2LGv4wZ.pbfOxx1qH-WXYhH8?startTime=1710985894000
 * Video 2: https://brandeis.zoom.us/rec/share/bqNp_Dlf_PztN0WFbvI-6XCPUBPZnNmgHY9fuF7JMicsf5PJru62dqz7yHY_fw-O.lXGsx5cORhzcv4Tz?startTime=1710971702000
 * Team: Xin Chen
*/

 public class PP_Visitor implements Visitor
 {
 
   private String indentString(int indent) {
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < 4*indent; ++i) {
        sb.append(' ');
        }
        return sb.toString();
   }

   public Object visit(Program node, Object data){

     String m = (String) node.m.accept(this,data);
     String c = (String) node.c.accept(this,data);


     return m + "\n" +  c;
   }

   public Object visit(MainClass node, Object data){
     int indent = (int) data;
     String i = (String) node.i.accept(this, data);
     String s = (String) node.s.accept(this, data);


     return "class " + i + "{\n" +
     indentString(indent+1) + "public static void main(String[] args){\n" +
     indentString(indent+2) + s + "\n" +
     indentString(indent+1) + "}\n" +
     indentString(indent) + "}\n";
     
   }


   public Object visit(ClassDecl node, Object data){
     int indent = (int) data;
     String i = (String) node.i.accept(this, data);
     String v = (String) node.v.accept(this, indent + 1);
     String m = (String) node.m.accept(this, indent + 1);

     


     return "class " + i + "{\n" +
      v + m + "}"
     ;
   }



   public Object visit(VarDecl node, Object data){
        int indent = (int) data; 
        String t = (String) node.t.accept(this,indent);
        String i = (String) node.i.accept(this,indent);
        return indentString(indent)+t + " "+i +";\n";
   }
   public Object visit(MethodDecl node, Object data){
        // 
        int indent = (int) data; 
        String type = (String) node.t.accept(this,indent);
        String id = (String) node.i.accept(this,indent);
        String formalList = "";
        if (node.f!=null){
            formalList = (String) node.f.accept(this,indent);
        }
        String varList = "";
        if (node.v!=null){
            varList = (String) node.v.accept(this,indent+1);
        }
        String statementList = "";
        if (node.s!=null){
            statementList = (String) node.s.accept(this,indent+1);
        }
        String expr = (String) node.e.accept(this,indent);
        return 
        indentString(indent)+ "public " +
        type+" "+id+"("+formalList+"){\n"+
        varList +
        statementList+
        indentString(indent+1)+"return "+expr +";\n"+
        indentString(indent)+"}\n";
   }
   public Object visit(Formal node, Object data){
        // 
        int indent = (int) data; 
        String type = (String) node.t.accept(this,indent);
        String id = (String) node.i.accept(this,indent);
        return type+" "+id;
   }

   public Object visit(IntArrayType node, Object data){
     return "int[]";
   }
   public Object visit(IntegerType node, Object data){
        // 
        int indent = (int) data; 
        return "int";
   }
   public Object visit(BooleanType node, Object data){
        // 
        int indent = (int) data; 
        return "boolean";
   }
   public Object visit(IdentifierType node, Object data){
     
     return node.s;
   }
   public Object visit(Block node, Object data){
        // 
        int indent = (int) data; 
        String result="";
        if (node.slist!=null){
            result = (String) node.slist.accept(this,indent+1);
        }
        
        return indentString(indent)+"{\n"+
               result+
               indentString(indent)+"}\n";
   }
   public Object visit(If node, Object data){
        // 
        int indent = (int) data; 
        String expr = (String)node.e.accept(this,indent+1);
        String s1 = (String) node.s1.accept(this,indent+1);
        String s2 = (String) node.s2.accept(this,indent+1);
        return indentString(indent)+
        "if ("+expr+")\n"+
        s1+
        indentString(indent)+"else\n"+
        s2;
   }
   public Object visit(While node, Object data){
     int indent = (int) data; 
     String e = (String) node.e.accept(this, data);
     String s = (String) node.s.accept(this, indent+1);

     return indentString(indent) + 
     "while(" + e + "){\n" 
     + s + indentString(indent) + "}\n"
     ;
   }
   public Object visit(Print node, Object data){
        // 
        int indent = (int) data; 
        String e = (String) node.e.accept(this,indent);
        //--indent;
        return indentString(indent) + "System.out.println("+e+");\n";
   }
   public Object visit(Assign node, Object data){
        // 
        int indent = (int) data; 
        String i = (String) node.i.accept(this,indent);
        String e = (String) node.e.accept(this,indent);
        //--indent;
        return indentString(indent)+i+" = "+e+";\n";
   }

   
   public Object visit(ArrayAssign node, Object data){ 
     int indent = (int) data;
     Identifier i = node.i;
     Exp e1 = node.e1;
     Exp e2 = node.e2;
     String id = (String) i.accept(this,indent);
     String exp1 = (String) e1.accept(this,indent);
     String exp2 = (String) e2.accept(this,indent);
     data = indentString(indent) + id+"["+exp1+"] = "+exp2+";\n";
     return data; 
   } 


   public Object visit(And node, Object data){
     int indent = (int) data; 
     String e1 = (String) node.e1.accept(this,indent);
     String e2 = (String) node.e2.accept(this,indent);
     return e1+" && "+e2;
   }

   public Object visit(LessThan node, Object data){
        // 
        int indent = (int) data; 
        String e1 = (String) node.e1.accept(this,indent);
        String e2 = (String) node.e2.accept(this,indent);
        return e1+" < "+e2;
   }
   public Object visit(Plus node, Object data){
        // 
        int indent = (int) data; 
        String e1 = (String) node.e1.accept(this,indent);
        String e2 = (String) node.e2.accept(this,indent);
        return e1+" + "+e2;
   }
   public Object visit(Minus node, Object data){
        // 
        int indent = (int) data; 
        String e1 = (String) node.e1.accept(this,indent);
        String e2 = (String) node.e2.accept(this,indent);
        return e1+" - "+e2;
   }
   public Object visit(Times node, Object data){
        // 
        int indent = (int) data; 
        String e1 = (String) node.e1.accept(this,indent);
        String e2 = (String) node.e2.accept(this,indent);
        return e1+" * "+e2;
   }
   public Object visit(ArrayLookup node, Object data){

     String e1 = (String) node.e1.accept(this,data);
     String e2 = (String) node.e2.accept(this,data);

     return e1 + "[" + e2 + "]";
   }
   public Object visit(ArrayLength node, Object data){
     String e = (String) node.e.accept(this,data);


     return e + ".length";
   }
   public Object visit(Call node, Object data){
        // 
        int indent = (int) data; 
        String e1 = (String) node.e1.accept(this,indent);
        String i = (String) node.i.accept(this,indent);
        if (node.e2 != null){
          String e2 = (String) node.e2.accept(this,indent);
          return e1 + "." + i +"("+e2+")";
        }
        

        return e1 + "." + i +"()";
   }
   public Object visit(IntegerLiteral node, Object data){
        int indent = (int) data; 
        int i = node.i;
        return " "+i+" ";
   }
   public Object visit(True node, Object data){
        // 
        int indent = (int) data; 
        //--indent;
        return "true";
   }
   public Object visit(False node, Object data){
        // 
        int indent = (int) data; 
        //--indent;
        return "false";
   }
   public Object visit(IdentifierExp node, Object data){

        int indent = (int) data; 
        String s = node.s;
        //--indent;
        return " "+s+" ";
   }
   public Object visit(This node, Object data){
     return "this";
   }
   public Object visit(NewArray node, Object data){

     String e = (String) node.e.accept(this,data);

     return "new int["+e+"]";
   }
   public Object visit(NewObject node, Object data){
     String i = (String) node.i.accept(this,data);

     return "new "+i+"()";
   }

   public Object visit(Not node, Object data){
     String e = (String) node.e.accept(this,data);

     return "! " + e;
   }
   public Object visit(Identifier node, Object data){

        int indent = (int) data; 
        String s = node.s;
        //--indent;
        return s;
   }

   public Object visit(ExpGroup node, Object data){
     String e="";
     if (node.e!=null){
         e = (String) node.e.accept(this,data);
     }
     return "("+e+")";
   }

   public Object visit(ClassDeclList node, Object data){
     // 
     
     String c = (String) node.c.accept(this,data);
     String clist = "";
     if(node.clist != null){
          clist = (String) node.clist.accept(this,data);
     }

     return c + "\n" + clist;
   }



   public Object visit(ExpList node, Object data){
        // 
        int indent = (int) data; 
        String e = (String) node.e.accept(this,indent);
        if (node.elist!=null){
            String e1 = (String) node.elist.accept(this,indent);
            e = e1 + ","+e;
        }

        return e;
   }
   public Object visit(FormalList node, Object data){
        // 
        int indent = (int) data; 
        String f = (String) node.f.accept(this,indent);
        if (node.flist!=null){
            String f1 = (String) node.flist.accept(this,indent);
            f = f1 +","+f;
        }
        //--indent;
        return f;
   }
   public Object visit(MethodDeclList node, Object data){
        // 
        int indent = (int) data; 
        String m = (String) node.m.accept(this,indent);
        if (node.mlist!=null){
            String m2 = (String) node.mlist.accept(this,indent);
            m = m2 +"\n"+m; 
        }
        return m;
   }
   public Object visit(StatementList node, Object data){
        // 
        int indent = (int) data; 
        String result="";
        
        if (node.slist!=null){
            String s = (String) node.slist.accept(this,indent);
            result = result+s;
        }
        String t = (String) node.s.accept(this,indent);
        //--indent;
        result = result+t;
        return result;
   }
   public Object visit(VarDeclList node, Object data){
        // 
        int indent = (int) data; 
        String v = (String) node.v.accept(this,indent);
        
        if (node.vlist!=null){
            String vlist = (String) node.vlist.accept(this,indent);
            v = vlist + v;
        }
        
        return v;
   }
}