Robust/src/IR/Tree/BuildIR.java

   1 package IR.Tree;
   2 import IR.*;
   3 import java.util.Vector;
   4
   5 public class BuildIR {
   6     State state;
   7     public BuildIR(State state) {
   8         this.state=state;
   9     }
  10     public void buildtree() {
  11         ParseNode pn=state.parsetree;
  12         parseFile(pn);
  13     }
  14
  15     /** Parse the classes in this file */
  16     public void parseFile(ParseNode pn) {
  17         ParseNode tpn=pn.getChild("type_declaration_list");
  18         if (tpn!=null) {
  19             ParseNodeVector pnv=tpn.getChildren();
  20             for(int i=0;i<pnv.size();i++) {
  21                 ParseNode type_pn=pnv.elementAt(i);
  22                 if (isEmpty(type_pn)) /* Skip the semicolon */
  23                     continue;
  24                 ClassDescriptor cn=parseTypeDecl(type_pn);
  25                 state.addClass(cn);
  26             }
  27         }
  28     }
  29
  30     public ClassDescriptor parseTypeDecl(ParseNode pn) {
  31         if (isNode(pn, "class_declaration")) {
  32             ClassDescriptor cn=new ClassDescriptor(pn.getChild("name").getTerminal());
  33             if (!isEmpty(pn.getChild("super").getTerminal())) {
  34                 /* parse superclass name */
  35                 ParseNode snn=pn.getChild("super").getChild("type").getChild("class").getChild("name");
  36                 NameDescriptor nd=parseName(snn);
  37                 cn.setSuper(nd.toString());
  38             }
  39             cn.setModifiers(parseModifiersList(pn.getChild("modifiers")));
  40             parseClassBody(cn, pn.getChild("classbody"));
  41             return cn;
  42         } else throw new Error();
  43     }
  44
  45     private void parseClassBody(ClassDescriptor cn, ParseNode pn) {
  46         ParseNode decls=pn.getChild("class_body_declaration_list");
  47         if (decls!=null) {
  48             ParseNodeVector pnv=decls.getChildren();
  49             for(int i=0;i<pnv.size();i++) {
  50                 ParseNode decl=pnv.elementAt(i);
  51                 if (isNode(decl,"member")) {
  52                     parseClassMember(cn,decl);
  53                 } else if (isNode(decl,"constructor")) {
  54                 } else if (isNode(decl,"block")) {
  55                 } else throw new Error();
  56             }
  57         }
  58     }
  59
  60     private void parseClassMember(ClassDescriptor cn, ParseNode pn) {
  61         ParseNode fieldnode=pn.getChild("field");
  62
  63         if (fieldnode!=null) {
  64             parseFieldDecl(cn,fieldnode.getChild("field_declaration"));
  65             return;
  66         }
  67         ParseNode methodnode=pn.getChild("method");
  68         if (methodnode!=null) {
  69             parseMethodDecl(cn,methodnode.getChild("method_declaration"));
  70             return;
  71         }
  72         throw new Error();
  73     }
  74
  75     private TypeDescriptor parseTypeDescriptor(ParseNode pn) {
  76         ParseNode tn=pn.getChild("type");
  77         String type_st=tn.getTerminal();
  78         if(type_st.equals("byte")) {
  79             return state.getTypeDescriptor(TypeDescriptor.BYTE);
  80         } else if(type_st.equals("short")) {
  81             return state.getTypeDescriptor(TypeDescriptor.SHORT);
  82         } else if(type_st.equals("boolean")) {
  83             return state.getTypeDescriptor(TypeDescriptor.BOOLEAN);
  84         } else if(type_st.equals("int")) {
  85             return state.getTypeDescriptor(TypeDescriptor.INT);
  86         } else if(type_st.equals("long")) {
  87             return state.getTypeDescriptor(TypeDescriptor.LONG);
  88         } else if(type_st.equals("char")) {
  89             return state.getTypeDescriptor(TypeDescriptor.CHAR);
  90         } else if(type_st.equals("float")) {
  91             return state.getTypeDescriptor(TypeDescriptor.FLOAT);
  92         } else if(type_st.equals("double")) {
  93             return state.getTypeDescriptor(TypeDescriptor.DOUBLE);
  94         } else if(type_st.equals("class")) {
  95             ParseNode nn=tn.getChild("class");
  96             return state.getTypeDescriptor(parseName(nn.getChild("name")));
  97         } else {
  98             throw new Error();
  99         }
 100     }
 101
 102     private NameDescriptor parseName(ParseNode nn) {
 103         ParseNode base=nn.getChild("base");
 104         ParseNode id=nn.getChild("identifier");
 105         if (base==null)
 106             return new NameDescriptor(id.getTerminal());
 107         return new NameDescriptor(parseName(base.getChild("name")),id.getTerminal());
 108
 109     }
 110
 111     private void parseFieldDecl(ClassDescriptor cn,ParseNode pn) {
 112         ParseNode mn=pn.getChild("modifier");
 113         Modifiers m=parseModifiersList(mn);
 114
 115         ParseNode tn=pn.getChild("type");
 116         TypeDescriptor t=parseTypeDescriptor(tn);
 117         ParseNode vn=pn.getChild("variables").getChild("variable_declarators_list");
 118         ParseNodeVector pnv=vn.getChildren();
 119         for(int i=0;i<pnv.size();i++) {
 120             ParseNode vardecl=pnv.elementAt(i);
 121             String identifier=vardecl.getChild("single").getTerminal();
 122             ParseNode epn=vardecl.getChild("initializer");
 123
 124             ExpressionNode en=null;
 125             if (epn!=null)
 126                 en=parseExpression(epn.getFirstChild());
 127
 128             cn.addField(new FieldDescriptor(m,t,identifier, en));
 129         }
 130
 131     }
 132
 133     private ExpressionNode parseExpression(ParseNode pn) {
 134         if (isNode(pn,"assignment"))
 135             return parseAssignmentExpression(pn);
 136         else if (isNode(pn,"logical_or")||isNode(pn,"logical_and")||
 137                  isNode(pn,"bitwise_or")||isNode(pn,"bitwise_xor")||
 138                  isNode(pn,"bitwise_and")||isNode(pn,"equal")||
 139                  isNode(pn,"not_equal")||isNode(pn,"comp_lt")||
 140                  isNode(pn,"comp_lte")||isNode(pn,"comp_gt")||
 141                  isNode(pn,"comp_gte")||isNode(pn,"leftshift")||
 142                  isNode(pn,"rightshift")||isNode(pn,"sub")||
 143                  isNode(pn,"add")||isNode(pn,"mult")||
 144                  isNode(pn,"div")||isNode(pn,"mod")) {
 145             ParseNodeVector pnv=pn.getChildren();
 146             ParseNode left=pnv.elementAt(0);
 147             ParseNode right=pnv.elementAt(1);
 148             Operation op=new Operation(pn.getLabel());
 149             return new OpNode(parseExpression(left),parseExpression(right),op);
 150         } else if (isNode(pn,"unaryplus")||
 151                    isNode(pn,"unaryminus")||
 152                    isNode(pn,"postinc")||
 153                    isNode(pn,"postdec")||
 154                    isNode(pn,"preinc")||
 155                    isNode(pn,"predec")) {
 156             ParseNode left=pn.getFirstChild();
 157             Operation op=new Operation(pn.getLabel());
 158             return new OpNode(parseExpression(left),op);
 159         } else if (isNode(pn,"literal")) {
 160             String literaltype=pn.getTerminal();
 161             ParseNode literalnode=pn.getChild(literaltype);
 162             Object literal_obj=literalnode.getLiteral();
 163             return new LiteralNode(literaltype, literal_obj);
 164         } else if (isNode(pn,"createobject")) {
 165             TypeDescriptor td=parseTypeDescriptor(pn);
 166             Vector args=parseArgumentList(pn);
 167             CreateObjectNode con=new CreateObjectNode(td);
 168             for(int i=0;i<args.size();i++) {
 169                 con.addArgument((ExpressionNode)args.get(i));
 170             }
 171             return con;
 172         } else if (isNode(pn,"name")) {
 173             NameDescriptor nd=parseName(pn);
 174             return new NameNode(nd);
 175         } else if (isNode(pn,"this")) {
 176             NameDescriptor nd=new NameDescriptor("this");
 177             return new NameNode(nd);
 178         } else if (isNode(pn,"methodinvoke1")) {
 179             NameDescriptor nd=parseName(pn.getChild("name"));
 180             Vector args=parseArgumentList(pn);
 181             MethodInvokeNode min=new MethodInvokeNode(nd);
 182             for(int i=0;i<args.size();i++) {
 183                 min.addArgument((ExpressionNode)args.get(i));
 184             }
 185             return min;
 186         } else if (isNode(pn,"methodinvoke2")) {
 187             String methodid=pn.getChild("id").getTerminal();
 188             ExpressionNode exp=parseExpression(pn.getChild("base").getFirstChild());
 189             Vector args=parseArgumentList(pn);
 190             MethodInvokeNode min=new MethodInvokeNode(methodid,exp);
 191             for(int i=0;i<args.size();i++) {
 192                 min.addArgument((ExpressionNode)args.get(i));
 193             }
 194             return min;
 195         } else if (isNode(pn,"fieldaccess")) {
 196             ExpressionNode en=parseExpression(pn.getChild("base").getFirstChild());
 197             String fieldname=pn.getChild("field").getTerminal();
 198             return new FieldAccessNode(en,fieldname);
 199         } else if (isNode(pn,"cast1")) {
 200             return new CastNode(parseTypeDescriptor(pn.getChild("type")),parseExpression(pn.getChild("exp").getFirstChild()));
 201         } else if (isNode(pn,"cast2")) {
 202             return new CastNode(parseExpression(pn.getChild("type").getFirstChild()),parseExpression(pn.getChild("exp").getFirstChild()));
 203         } else {
 204             System.out.println("---------------------");
 205             System.out.println(pn.PPrint(3,true));
 206             throw new Error();
 207         }
 208     }
 209
 210     private Vector parseArgumentList(ParseNode pn) {
 211         Vector arglist=new Vector();
 212         ParseNode an=pn.getChild("argument_list");
 213         if (an==null)   /* No argument list */
 214             return arglist;
 215         ParseNodeVector anv=an.getChildren();
 216         for(int i=0;i<anv.size();i++) {
 217             arglist.add(parseExpression(anv.elementAt(i)));
 218         }
 219         return arglist;
 220     }
 221
 222     private ExpressionNode parseAssignmentExpression(ParseNode pn) {
 223         AssignOperation ao=new AssignOperation(pn.getChild("op").getTerminal());
 224         ParseNodeVector pnv=pn.getChild("args").getChildren();
 225
 226         AssignmentNode an=new AssignmentNode(parseExpression(pnv.elementAt(0)),parseExpression(pnv.elementAt(1)),ao);
 227         return an;
 228     }
 229
 230
 231     private void parseMethodDecl(ClassDescriptor cn, ParseNode pn) {
 232         ParseNode headern=pn.getChild("method_header");
 233         ParseNode bodyn=pn.getChild("body");
 234         MethodDescriptor md=parseMethodHeader(headern);
 235         BlockNode bn=parseBlock(bodyn);
 236         cn.addMethod(md);
 237         state.addTreeCode(md,bn);
 238     }
 239
 240     public BlockNode parseBlock(ParseNode pn) {
 241         if (isEmpty(pn.getTerminal()))
 242             return new BlockNode();
 243         ParseNode bsn=pn.getChild("block_statement_list");
 244         return parseBlockHelper(bsn);
 245     }
 246
 247     private BlockNode parseBlockHelper(ParseNode pn) {
 248         ParseNodeVector pnv=pn.getChildren();
 249         BlockNode bn=new BlockNode();
 250         for(int i=0;i<pnv.size();i++) {
 251             Vector bsv=parseBlockStatement(pnv.elementAt(i));
 252             for(int j=0;j<bsv.size();j++) {
 253                 bn.addBlockStatement((BlockStatementNode)bsv.get(j));
 254             }
 255         }
 256         return bn;
 257     }
 258
 259     public BlockNode parseSingleBlock(ParseNode pn) {
 260         BlockNode bn=new BlockNode();
 261         Vector bsv=parseBlockStatement(pn);
 262         for(int j=0;j<bsv.size();j++) {
 263             bn.addBlockStatement((BlockStatementNode)bsv.get(j));
 264         }
 265         bn.setStyle(BlockNode.NOBRACES);
 266         return bn;
 267     }
 268
 269     public Vector parseBlockStatement(ParseNode pn) {
 270         Vector blockstatements=new Vector();
 271         if (isNode(pn,"local_variable_declaration")) {
 272             TypeDescriptor t=parseTypeDescriptor(pn);
 273             ParseNode vn=pn.getChild("variable_declarators_list");
 274             ParseNodeVector pnv=vn.getChildren();
 275             for(int i=0;i<pnv.size();i++) {
 276                 ParseNode vardecl=pnv.elementAt(i);
 277                 String identifier=vardecl.getChild("single").getTerminal();
 278                 ParseNode epn=vardecl.getChild("initializer");
 279
 280                 ExpressionNode en=null;
 281                 if (epn!=null)
 282                     en=parseExpression(epn.getFirstChild());
 283
 284                 blockstatements.add(new DeclarationNode(new VarDescriptor(t,identifier),en));
 285             }
 286         } else if (isNode(pn,"nop")) {
 287             /* Do Nothing */
 288         } else if (isNode(pn,"expression")) {
 289             blockstatements.add(new BlockExpressionNode(parseExpression(pn.getFirstChild())));
 290         } else if (isNode(pn,"ifstatement")) {
 291             blockstatements.add(new IfStatementNode(parseExpression(pn.getChild("condition").getFirstChild()),
 292                                        parseSingleBlock(pn.getChild("statement").getFirstChild()),
 293                                        pn.getChild("else_statement")!=null?parseSingleBlock(pn.getChild("else_statement").getFirstChild()):null));
 294         } else if (isNode(pn,"return")) {
 295             if (isEmpty(pn.getTerminal()))
 296                 blockstatements.add(new ReturnNode());
 297             else {
 298                 ExpressionNode en=parseExpression(pn.getFirstChild());
 299                 blockstatements.add(new ReturnNode(en));
 300             }
 301         } else if (isNode(pn,"block_statement_list")) {
 302             BlockNode bn=parseBlockHelper(pn);
 303             blockstatements.add(new SubBlockNode(bn));
 304         } else if (isNode(pn,"empty")) {
 305             /* nop */
 306         } else if (isNode(pn,"statement_expression_list")) {
 307             ParseNodeVector pnv=pn.getChildren();
 308             BlockNode bn=new BlockNode();
 309             for(int i=0;i<pnv.size();i++) {
 310                 ExpressionNode en=parseExpression(pnv.elementAt(i));
 311                 blockstatements.add(new BlockExpressionNode(en));
 312             }
 313             bn.setStyle(BlockNode.EXPRLIST);
 314         } else if (isNode(pn,"forstatement")) {
 315             BlockNode init=parseSingleBlock(pn.getChild("initializer").getFirstChild());
 316             BlockNode update=parseSingleBlock(pn.getChild("update").getFirstChild());
 317             ExpressionNode condition=parseExpression(pn.getChild("condition").getFirstChild());
 318             BlockNode body=parseSingleBlock(pn.getChild("statement").getFirstChild());
 319             blockstatements.add(new LoopNode(init,condition,update,body));
 320         } else if (isNode(pn,"whilestatement")) {
 321             ExpressionNode condition=parseExpression(pn.getChild("condition").getFirstChild());
 322             BlockNode body=parseSingleBlock(pn.getChild("statement").getFirstChild());
 323             blockstatements.add(new LoopNode(condition,body,LoopNode.WHILELOOP));
 324         } else if (isNode(pn,"dowhilestatement")) {
 325             ExpressionNode condition=parseExpression(pn.getChild("condition").getFirstChild());
 326             BlockNode body=parseSingleBlock(pn.getChild("statement").getFirstChild());
 327             blockstatements.add(new LoopNode(condition,body,LoopNode.DOWHILELOOP));
 328         } else {
 329             System.out.println("---------------");
 330             System.out.println(pn.PPrint(3,true));
 331             throw new Error();
 332         }
 333         return blockstatements;
 334     }
 335
 336     public MethodDescriptor parseMethodHeader(ParseNode pn) {
 337         ParseNode mn=pn.getChild("modifiers");
 338         Modifiers m=parseModifiersList(mn);
 339
 340         ParseNode tn=pn.getChild("returntype");
 341         TypeDescriptor returntype;
 342         if (tn!=null)
 343             returntype=parseTypeDescriptor(tn);
 344         else
 345             returntype=new TypeDescriptor(TypeDescriptor.VOID);
 346
 347         ParseNode pmd=pn.getChild("method_declarator");
 348         String name=pmd.getChild("name").getTerminal();
 349         MethodDescriptor md=new MethodDescriptor(m, returntype, name);
 350
 351         ParseNode paramnode=pmd.getChild("parameters");
 352         parseParameterList(md,paramnode);
 353         return md;
 354     }
 355
 356     public void parseParameterList(MethodDescriptor md, ParseNode pn) {
 357         ParseNode paramlist=pn.getChild("formal_parameter_list");
 358         if (paramlist==null)
 359             return;
 360          ParseNodeVector pnv=paramlist.getChildren();
 361          for(int i=0;i<pnv.size();i++) {
 362              ParseNode paramn=pnv.elementAt(i);
 363              TypeDescriptor type=parseTypeDescriptor(paramn);
 364              String paramname=paramn.getChild("single").getTerminal();
 365              md.addParameter(type,paramname);
 366          }
 367     }
 368
 369     public Modifiers parseModifiersList(ParseNode pn) {
 370         Modifiers m=new Modifiers();
 371         ParseNode modlist=pn.getChild("modifier_list");
 372         if (modlist!=null) {
 373             ParseNodeVector pnv=modlist.getChildren();
 374             for(int i=0;i<pnv.size();i++) {
 375                 ParseNode modn=pnv.elementAt(i);
 376                 if (isNode(modn,"public"))
 377                     m.addModifier(Modifiers.PUBLIC);
 378                 if (isNode(modn,"protected"))
 379                     m.addModifier(Modifiers.PROTECTED);
 380                 if (isNode(modn,"private"))
 381                     m.addModifier(Modifiers.PRIVATE);
 382                 if (isNode(modn,"static"))
 383                     m.addModifier(Modifiers.STATIC);
 384                 if (isNode(modn,"final"))
 385                     m.addModifier(Modifiers.FINAL);
 386                 if (isNode(modn,"native"))
 387                     m.addModifier(Modifiers.NATIVE);
 388             }
 389         }
 390         return m;
 391     }
 392
 393     private boolean isNode(ParseNode pn, String label) {
 394         if (pn.getLabel().equals(label))
 395             return true;
 396         else return false;
 397     }
 398
 399     private static boolean isEmpty(ParseNode pn) {
 400         if (pn.getLabel().equals("empty"))
 401             return true;
 402         else
 403             return false;
 404     }
 405
 406     private static boolean isEmpty(String s) {
 407         if (s.equals("empty"))
 408             return true;
 409         else
 410             return false;
 411     }
 412
 413     /** Throw an exception if something is unexpected */
 414     private void check(ParseNode pn, String label) {
 415         if (pn == null) {
 416             throw new Error(pn+ "IE: Expected '" + label + "', got null");
 417         }
 418         if (! pn.getLabel().equals(label)) {
 419             throw new Error(pn+ "IE: Expected '" + label + "', got '"+pn.getLabel()+"'");
 420         }
 421     }
 422 }