6 public class BuildFlat {
10 public BuildFlat(State st) {
12 temptovar=new Hashtable();
15 public Hashtable getMap() {
19 public void buildFlat() {
20 Iterator it=state.getClassSymbolTable().getDescriptorsIterator();
22 ClassDescriptor cn=(ClassDescriptor)it.next();
26 Iterator task_it=state.getTaskSymbolTable().getDescriptorsIterator();
27 while(task_it.hasNext()) {
28 TaskDescriptor td=(TaskDescriptor)task_it.next();
33 private void flattenTask(TaskDescriptor td) {
34 BlockNode bn=state.getMethodBody(td);
35 FlatNode fn=flattenBlockNode(bn).getBegin();
36 FlatFlagActionNode ffan=new FlatFlagActionNode();
38 FlatMethod fm=new FlatMethod(td, ffan);
40 for(int i=0;i<td.numParameters();i++) {
41 fm.addParameterTemp(getTempforVar(td.getParameter(i)));
44 /* Flatten Vector of Flag Effects */
45 Vector flags=td.getFlagEffects();
46 updateFlagActionNode(ffan,flags);
48 state.addFlatCode(td,fm);
52 /* This method transforms a vector of FlagEffects into the FlatFlagActionNode */
53 private void updateFlagActionNode(FlatFlagActionNode ffan, Vector flags) {
54 for(int i=0;i<flags.size();i++) {
55 FlagEffects fes=(FlagEffects)flags.get(i);
56 TempDescriptor flagtemp=getTempforVar(fes.getVar());
57 for(int j=0;j<fes.numEffects();j++) {
58 FlagEffect fe=fes.getEffect(j);
59 ffan.addFlagAction(flagtemp, fe.getFlag(), fe.getStatus());
64 private void flattenClass(ClassDescriptor cn) {
65 Iterator methodit=cn.getMethods();
66 while(methodit.hasNext()) {
67 MethodDescriptor md=(MethodDescriptor)methodit.next();
68 BlockNode bn=state.getMethodBody(md);
69 FlatNode fn=flattenBlockNode(bn).getBegin();
70 FlatMethod fm=new FlatMethod(md, fn);
72 fm.addParameterTemp(getTempforVar(md.getThis()));
73 for(int i=0;i<md.numParameters();i++) {
74 fm.addParameterTemp(getTempforVar(md.getParameter(i)));
76 System.out.println(fm.printMethod());
77 state.addFlatCode(md,fm);
81 private NodePair flattenBlockNode(BlockNode bn) {
84 for(int i=0;i<bn.size();i++) {
85 NodePair np=flattenBlockStatementNode(bn.get(i));
86 FlatNode np_begin=np.getBegin();
87 FlatNode np_end=np.getEnd();
94 end.addNext(np_begin);
99 end=begin=new FlatNop();
101 return new NodePair(begin,end);
104 private NodePair flattenBlockExpressionNode(BlockExpressionNode en) {
105 TempDescriptor tmp=TempDescriptor.tempFactory("neverused",en.getExpression().getType());
106 return flattenExpressionNode(en.getExpression(),tmp);
109 private NodePair flattenCastNode(CastNode cn,TempDescriptor out_temp) {
110 TempDescriptor tmp=TempDescriptor.tempFactory("tocast",cn.getExpression().getType());
111 NodePair np=flattenExpressionNode(cn.getExpression(), tmp);
112 FlatCastNode fcn=new FlatCastNode(cn.getType(), tmp, out_temp);
113 np.getEnd().addNext(fcn);
114 return new NodePair(np.getBegin(),fcn);
117 private NodePair flattenLiteralNode(LiteralNode ln,TempDescriptor out_temp) {
118 FlatLiteralNode fln=new FlatLiteralNode(ln.getType(), ln.getValue(), out_temp);
119 return new NodePair(fln,fln);
122 private NodePair flattenCreateObjectNode(CreateObjectNode con,TempDescriptor out_temp) {
123 TypeDescriptor td=con.getType();
125 FlatNew fn=new FlatNew(td, out_temp);
126 TempDescriptor[] temps=new TempDescriptor[con.numArgs()];
129 for(int i=0;i<con.numArgs();i++) {
130 ExpressionNode en=con.getArg(i);
131 TempDescriptor tmp=TempDescriptor.tempFactory("arg",en.getType());
133 NodePair np=flattenExpressionNode(en, tmp);
134 last.addNext(np.getBegin());
137 MethodDescriptor md=con.getConstructor();
138 //Call to constructor
139 FlatCall fc=new FlatCall(md, null, out_temp, temps);
141 return new NodePair(fn,fc);
145 TempDescriptor[] temps=new TempDescriptor[con.numArgs()];
146 for (int i=0;i<con.numArgs();i++) {
147 ExpressionNode en=con.getArg(i);
148 TempDescriptor tmp=TempDescriptor.tempFactory("arg",en.getType());
150 NodePair np=flattenExpressionNode(en, tmp);
154 last.addNext(np.getBegin());
157 TempDescriptor tmp2=(i==0)?
159 TempDescriptor.tempFactory("arg",en.getType());
161 FlatNew fn=new FlatNew(td, out_temp, temps[0]);
163 if (temps.length>1) {
164 NodePair np=generateNewArrayLoop(temps, td.dereference(), out_temp, 0);
165 fn.addNext(np.getBegin());
166 return new NodePair(first,np.getEnd());
168 return new NodePair(first, fn);
172 private NodePair generateNewArrayLoop(TempDescriptor[] temparray, TypeDescriptor td, TempDescriptor tmp, int i) {
173 TempDescriptor index=TempDescriptor.tempFactory("index",new TypeDescriptor(TypeDescriptor.INT));
174 TempDescriptor tmpone=TempDescriptor.tempFactory("index",new TypeDescriptor(TypeDescriptor.INT));
175 FlatNop fnop=new FlatNop();//last node
178 FlatLiteralNode fln=new FlatLiteralNode(index.getType(),new Integer(0),index);
180 FlatLiteralNode fln2=new FlatLiteralNode(tmpone.getType(),new Integer(1),tmpone);
182 TempDescriptor tmpbool=TempDescriptor.tempFactory("comp",new TypeDescriptor(TypeDescriptor.BOOLEAN));
184 FlatOpNode fcomp=new FlatOpNode(tmpbool,index,temparray[i],new Operation(Operation.LT));
185 FlatCondBranch fcb=new FlatCondBranch(tmpbool);
187 TempDescriptor new_tmp=TempDescriptor.tempFactory("tmp",td);
188 FlatNew fn=new FlatNew(td, new_tmp, temparray[i+1]);
189 FlatSetElementNode fsen=new FlatSetElementNode(tmp,index,new_tmp);
191 FlatOpNode fon=new FlatOpNode(index,index,tmpone,new Operation(Operation.ADD));
197 fcb.addFalseNext(fnop);
199 //Recursive call here
200 if ((i+2)<temparray.length) {
201 NodePair np2=generateNewArrayLoop(temparray, td.dereference(), new_tmp, i+1);
202 fsen.addNext(np2.getBegin());
203 np2.getEnd().addNext(fon);
208 return new NodePair(fln, fnop);
211 private NodePair flattenMethodInvokeNode(MethodInvokeNode min,TempDescriptor out_temp) {
212 TempDescriptor[] temps=new TempDescriptor[min.numArgs()];
215 TempDescriptor thisarg=null;
217 if (min.getExpression()!=null) {
218 thisarg=TempDescriptor.tempFactory("thisarg",min.getExpression().getType());
219 NodePair np=flattenExpressionNode(min.getExpression(),thisarg);
225 for(int i=0;i<min.numArgs();i++) {
226 ExpressionNode en=min.getArg(i);
227 TempDescriptor td=TempDescriptor.tempFactory("arg",en.getType());
229 NodePair np=flattenExpressionNode(en, td);
233 last.addNext(np.getBegin());
237 MethodDescriptor md=min.getMethod();
239 //Call to constructor
242 if(md.getReturnType()==null||md.getReturnType().isVoid())
243 fc=new FlatCall(md, null, thisarg, temps);
245 fc=new FlatCall(md, out_temp, thisarg, temps);
250 return new NodePair(first,fc);
253 private NodePair flattenFieldAccessNode(FieldAccessNode fan,TempDescriptor out_temp) {
254 TempDescriptor tmp=TempDescriptor.tempFactory("temp",fan.getExpression().getType());
255 NodePair npe=flattenExpressionNode(fan.getExpression(),tmp);
256 FlatFieldNode fn=new FlatFieldNode(fan.getField(),tmp,out_temp);
257 npe.getEnd().addNext(fn);
258 return new NodePair(npe.getBegin(),fn);
261 private NodePair flattenArrayAccessNode(ArrayAccessNode aan,TempDescriptor out_temp) {
262 TempDescriptor tmp=TempDescriptor.tempFactory("temp",aan.getExpression().getType());
263 TempDescriptor tmpindex=TempDescriptor.tempFactory("temp",aan.getIndex().getType());
264 NodePair npe=flattenExpressionNode(aan.getExpression(),tmp);
265 NodePair npi=flattenExpressionNode(aan.getIndex(),tmpindex);
266 FlatElementNode fn=new FlatElementNode(tmp,tmpindex,out_temp);
267 npe.getEnd().addNext(npi.getBegin());
268 npi.getEnd().addNext(fn);
269 return new NodePair(npe.getBegin(),fn);
272 private NodePair flattenAssignmentNode(AssignmentNode an,TempDescriptor out_temp) {
274 // left side is variable
275 // left side is field
277 Operation base=an.getOperation().getBaseOp();
278 TempDescriptor src_tmp=TempDescriptor.tempFactory("src",an.getSrc().getType());
279 NodePair np_src=flattenExpressionNode(an.getSrc(),src_tmp);
280 FlatNode last=np_src.getEnd();
282 TempDescriptor src_tmp2=TempDescriptor.tempFactory("tmp", an.getDest().getType());
283 NodePair np_dst_init=flattenExpressionNode(an.getDest(),src_tmp2);
284 last.addNext(np_dst_init.getBegin());
285 TempDescriptor dst_tmp=TempDescriptor.tempFactory("dst_tmp",an.getDest().getType());
286 FlatOpNode fon=new FlatOpNode(dst_tmp, src_tmp,src_tmp2, base);
287 np_dst_init.getEnd().addNext(fon);
292 if (an.getDest().kind()==Kind.FieldAccessNode) {
293 FieldAccessNode fan=(FieldAccessNode)an.getDest();
294 ExpressionNode en=fan.getExpression();
295 TempDescriptor dst_tmp=TempDescriptor.tempFactory("dst",en.getType());
296 NodePair np_baseexp=flattenExpressionNode(en, dst_tmp);
297 last.addNext(np_baseexp.getBegin());
298 FlatSetFieldNode fsfn=new FlatSetFieldNode(dst_tmp, fan.getField(), src_tmp);
299 np_baseexp.getEnd().addNext(fsfn);
300 return new NodePair(np_src.getBegin(), fsfn);
301 } else if (an.getDest().kind()==Kind.ArrayAccessNode) {
302 ArrayAccessNode aan=(ArrayAccessNode)an.getDest();
303 ExpressionNode en=aan.getExpression();
304 ExpressionNode enindex=aan.getIndex();
305 TempDescriptor dst_tmp=TempDescriptor.tempFactory("dst",en.getType());
306 TempDescriptor index_tmp=TempDescriptor.tempFactory("index",enindex.getType());
307 NodePair np_baseexp=flattenExpressionNode(en, dst_tmp);
308 NodePair np_indexexp=flattenExpressionNode(enindex, index_tmp);
309 last.addNext(np_baseexp.getBegin());
310 np_baseexp.getEnd().addNext(np_indexexp.getBegin());
311 FlatSetElementNode fsen=new FlatSetElementNode(dst_tmp, index_tmp, src_tmp);
312 np_indexexp.getEnd().addNext(fsen);
313 return new NodePair(np_src.getBegin(), fsen);
314 } else if (an.getDest().kind()==Kind.NameNode) {
315 NameNode nn=(NameNode)an.getDest();
316 if (nn.getExpression()!=null) {
317 FieldAccessNode fan=(FieldAccessNode)nn.getExpression();
318 ExpressionNode en=fan.getExpression();
319 TempDescriptor dst_tmp=TempDescriptor.tempFactory("dst",en.getType());
320 NodePair np_baseexp=flattenExpressionNode(en, dst_tmp);
321 last.addNext(np_baseexp.getBegin());
322 FlatSetFieldNode fsfn=new FlatSetFieldNode(dst_tmp, fan.getField(), src_tmp);
323 np_baseexp.getEnd().addNext(fsfn);
324 return new NodePair(np_src.getBegin(), fsfn);
326 if (nn.getField()!=null) {
327 FlatSetFieldNode fsfn=new FlatSetFieldNode(getTempforVar(nn.getVar()), nn.getField(), src_tmp);
329 return new NodePair(np_src.getBegin(), fsfn);
331 FlatOpNode fon=new FlatOpNode(getTempforVar(nn.getVar()), src_tmp, null, new Operation(Operation.ASSIGN));
333 return new NodePair(np_src.getBegin(),fon);
340 private NodePair flattenNameNode(NameNode nn,TempDescriptor out_temp) {
341 if (nn.getExpression()!=null) {
342 /* Hack - use subtree instead */
343 return flattenExpressionNode(nn.getExpression(),out_temp);
344 } else if (nn.getField()!=null) {
345 TempDescriptor tmp=getTempforVar(nn.getVar());
346 FlatFieldNode ffn=new FlatFieldNode(nn.getField(), tmp, out_temp);
347 return new NodePair(ffn,ffn);
349 TempDescriptor tmp=getTempforVar(nn.getVar());
350 FlatOpNode fon=new FlatOpNode(out_temp, tmp, null, new Operation(Operation.ASSIGN));
351 return new NodePair(fon,fon);
355 private NodePair flattenOpNode(OpNode on,TempDescriptor out_temp) {
356 TempDescriptor temp_left=TempDescriptor.tempFactory("leftop",on.getLeft().getType());
357 TempDescriptor temp_right=null;
359 Operation op=on.getOp();
360 if (op.getOp()==Operation.POSTINC||
361 op.getOp()==Operation.POSTDEC||
362 op.getOp()==Operation.PREINC||
363 op.getOp()==Operation.PREDEC) {
364 LiteralNode ln=new LiteralNode("int",new Integer(1));
365 ln.setType(new TypeDescriptor(TypeDescriptor.INT));
366 AssignmentNode an=new AssignmentNode(on.getLeft(),ln,
367 new AssignOperation((op.getOp()==Operation.POSTINC||op.getOp()==Operation.PREINC)?AssignOperation.PLUSEQ:AssignOperation.MINUSEQ));
368 if (op.getOp()==Operation.POSTINC||
369 op.getOp()==Operation.POSTDEC) {
370 NodePair left=flattenExpressionNode(on.getLeft(),out_temp);
371 NodePair assign=flattenAssignmentNode(an,temp_left);
372 left.getEnd().addNext(assign.getBegin());
373 return new NodePair(left.getBegin(),assign.getEnd());
375 NodePair assign=flattenAssignmentNode(an,out_temp);
380 NodePair left=flattenExpressionNode(on.getLeft(),temp_left);
382 if (on.getRight()!=null) {
383 temp_right=TempDescriptor.tempFactory("rightop",on.getRight().getType());
384 right=flattenExpressionNode(on.getRight(),temp_right);
386 FlatNop nop=new FlatNop();
387 right=new NodePair(nop,nop);
390 if (op.getOp()==Operation.LOGIC_OR) {
391 /* Need to do shortcircuiting */
392 FlatCondBranch fcb=new FlatCondBranch(temp_left);
393 FlatOpNode fon1=new FlatOpNode(out_temp,temp_left,null,new Operation(Operation.ASSIGN));
394 FlatOpNode fon2=new FlatOpNode(out_temp,temp_right,null,new Operation(Operation.ASSIGN));
395 FlatNop fnop=new FlatNop();
396 left.getEnd().addNext(fcb);
397 fcb.addFalseNext(right.getBegin());
398 right.getEnd().addNext(fon2);
400 fcb.addTrueNext(fon1);
402 return new NodePair(left.getBegin(), fnop);
403 } else if (op.getOp()==Operation.LOGIC_AND) {
404 /* Need to do shortcircuiting */
405 FlatCondBranch fcb=new FlatCondBranch(temp_left);
406 FlatOpNode fon1=new FlatOpNode(out_temp,temp_left,null,new Operation(Operation.ASSIGN));
407 FlatOpNode fon2=new FlatOpNode(out_temp,temp_right,null,new Operation(Operation.ASSIGN));
408 FlatNop fnop=new FlatNop();
409 left.getEnd().addNext(fcb);
410 fcb.addTrueNext(right.getBegin());
411 right.getEnd().addNext(fon2);
413 fcb.addFalseNext(fon1);
415 return new NodePair(left.getBegin(), fnop);
418 FlatOpNode fon=new FlatOpNode(out_temp,temp_left,temp_right,op);
419 left.getEnd().addNext(right.getBegin());
420 right.getEnd().addNext(fon);
421 return new NodePair(left.getBegin(),fon);
424 private NodePair flattenExpressionNode(ExpressionNode en, TempDescriptor out_temp) {
426 case Kind.AssignmentNode:
427 return flattenAssignmentNode((AssignmentNode)en,out_temp);
429 return flattenCastNode((CastNode)en,out_temp);
430 case Kind.CreateObjectNode:
431 return flattenCreateObjectNode((CreateObjectNode)en,out_temp);
432 case Kind.FieldAccessNode:
433 return flattenFieldAccessNode((FieldAccessNode)en,out_temp);
434 case Kind.ArrayAccessNode:
435 return flattenArrayAccessNode((ArrayAccessNode)en,out_temp);
436 case Kind.LiteralNode:
437 return flattenLiteralNode((LiteralNode)en,out_temp);
438 case Kind.MethodInvokeNode:
439 return flattenMethodInvokeNode((MethodInvokeNode)en,out_temp);
441 return flattenNameNode((NameNode)en,out_temp);
443 return flattenOpNode((OpNode)en,out_temp);
448 private NodePair flattenDeclarationNode(DeclarationNode dn) {
449 VarDescriptor vd=dn.getVarDescriptor();
450 TempDescriptor td=getTempforVar(vd);
451 if (dn.getExpression()!=null)
452 return flattenExpressionNode(dn.getExpression(),td);
454 FlatNop fn=new FlatNop();
455 return new NodePair(fn,fn);
459 private TempDescriptor getTempforVar(VarDescriptor vd) {
460 if (temptovar.containsKey(vd))
461 return (TempDescriptor)temptovar.get(vd);
463 TempDescriptor td=TempDescriptor.tempFactory(vd.getName(),vd.getType());
464 temptovar.put(vd,td);
469 private NodePair flattenIfStatementNode(IfStatementNode isn) {
470 TempDescriptor cond_temp=TempDescriptor.tempFactory("condition",new TypeDescriptor(TypeDescriptor.BOOLEAN));
471 NodePair cond=flattenExpressionNode(isn.getCondition(),cond_temp);
472 FlatCondBranch fcb=new FlatCondBranch(cond_temp);
473 NodePair true_np=flattenBlockNode(isn.getTrueBlock());
475 FlatNop nopend=new FlatNop();
477 if (isn.getFalseBlock()!=null)
478 false_np=flattenBlockNode(isn.getFalseBlock());
480 FlatNop nop=new FlatNop();
481 false_np=new NodePair(nop,nop);
484 cond.getEnd().addNext(fcb);
485 fcb.addTrueNext(true_np.getBegin());
486 fcb.addFalseNext(false_np.getBegin());
487 true_np.getEnd().addNext(nopend);
488 false_np.getEnd().addNext(nopend);
489 return new NodePair(cond.getBegin(), nopend);
492 private NodePair flattenLoopNode(LoopNode ln) {
493 if (ln.getType()==LoopNode.FORLOOP) {
494 NodePair initializer=flattenBlockNode(ln.getInitializer());
495 TempDescriptor cond_temp=TempDescriptor.tempFactory("condition", new TypeDescriptor(TypeDescriptor.BOOLEAN));
496 NodePair condition=flattenExpressionNode(ln.getCondition(),cond_temp);
497 NodePair update=flattenBlockNode(ln.getUpdate());
498 NodePair body=flattenBlockNode(ln.getBody());
499 FlatNode begin=initializer.getBegin();
500 FlatCondBranch fcb=new FlatCondBranch(cond_temp);
501 FlatNop nopend=new FlatNop();
503 initializer.getEnd().addNext(condition.getBegin());
504 body.getEnd().addNext(update.getBegin());
505 update.getEnd().addNext(condition.getBegin());
506 condition.getEnd().addNext(fcb);
507 fcb.addFalseNext(nopend);
508 fcb.addTrueNext(body.getBegin());
509 return new NodePair(begin,nopend);
510 } else if (ln.getType()==LoopNode.WHILELOOP) {
511 TempDescriptor cond_temp=TempDescriptor.tempFactory("condition", new TypeDescriptor(TypeDescriptor.BOOLEAN));
512 NodePair condition=flattenExpressionNode(ln.getCondition(),cond_temp);
513 NodePair body=flattenBlockNode(ln.getBody());
514 FlatNode begin=condition.getBegin();
515 FlatCondBranch fcb=new FlatCondBranch(cond_temp);
516 FlatNop nopend=new FlatNop();
518 body.getEnd().addNext(condition.getBegin());
519 condition.getEnd().addNext(fcb);
520 fcb.addFalseNext(nopend);
521 fcb.addTrueNext(body.getBegin());
522 return new NodePair(begin,nopend);
523 } else if (ln.getType()==LoopNode.DOWHILELOOP) {
524 TempDescriptor cond_temp=TempDescriptor.tempFactory("condition", new TypeDescriptor(TypeDescriptor.BOOLEAN));
525 NodePair condition=flattenExpressionNode(ln.getCondition(),cond_temp);
526 NodePair body=flattenBlockNode(ln.getBody());
527 FlatNode begin=body.getBegin();
528 FlatCondBranch fcb=new FlatCondBranch(cond_temp);
529 FlatNop nopend=new FlatNop();
531 body.getEnd().addNext(condition.getBegin());
532 condition.getEnd().addNext(fcb);
533 fcb.addFalseNext(nopend);
534 fcb.addTrueNext(body.getBegin());
535 return new NodePair(begin,nopend);
536 } else throw new Error();
539 private NodePair flattenReturnNode(ReturnNode rntree) {
540 TempDescriptor retval=null;
542 if (rntree.getReturnExpression()!=null) {
543 retval=TempDescriptor.tempFactory("ret_value", rntree.getReturnExpression().getType());
544 cond=flattenExpressionNode(rntree.getReturnExpression(),retval);
547 FlatReturnNode rnflat=new FlatReturnNode(retval);
550 cond.getEnd().addNext(rnflat);
551 return new NodePair(cond.getBegin(),rnflat);
553 return new NodePair(rnflat,rnflat);
556 private NodePair flattenTaskExitNode(TaskExitNode ten) {
557 FlatTaskExitNode tenflat=new FlatTaskExitNode();
558 return new NodePair(tenflat,tenflat);
561 private NodePair flattenSubBlockNode(SubBlockNode sbn) {
562 return flattenBlockNode(sbn.getBlockNode());
565 private NodePair flattenBlockStatementNode(BlockStatementNode bsn) {
567 case Kind.BlockExpressionNode:
568 return flattenBlockExpressionNode((BlockExpressionNode)bsn);
570 case Kind.DeclarationNode:
571 return flattenDeclarationNode((DeclarationNode)bsn);
573 case Kind.IfStatementNode:
574 return flattenIfStatementNode((IfStatementNode)bsn);
577 return flattenLoopNode((LoopNode)bsn);
579 case Kind.ReturnNode:
580 return flattenReturnNode((IR.Tree.ReturnNode)bsn);
582 case Kind.TaskExitNode:
583 return flattenTaskExitNode((IR.Tree.TaskExitNode)bsn);
585 case Kind.SubBlockNode:
586 return flattenSubBlockNode((SubBlockNode)bsn);