Robust/src/IR/Tree/BlockNode.java

   1 package IR.Tree;
   2 import java.util.Vector;
   3 import IR.*;
   4
   5 public class BlockNode extends TreeNode {
   6     Vector blockstatements;
   7     int printStyle=0;
   8     protected SymbolTable table;
   9
  10     public final static int NORMAL=0;
  11     public final static int NOBRACES=1;
  12     public final static int EXPRLIST=2;
  13
  14     public BlockNode() {
  15         blockstatements=new Vector();
  16         table=new SymbolTable();
  17     }
  18
  19     public SymbolTable getVarTable() {
  20         return table;
  21     }
  22
  23     public void addBlockStatement(BlockStatementNode bsn) {
  24         blockstatements.add(bsn);
  25     }
  26
  27     public void addFirstBlockStatement(BlockStatementNode bsn) {
  28         blockstatements.insertElementAt(bsn,0);
  29     }
  30
  31     public void setStyle(int style) {
  32         printStyle=style;
  33     }
  34
  35     public int size() {
  36         return blockstatements.size();
  37     }
  38
  39     public BlockStatementNode get(int i) {
  40         return (BlockStatementNode) blockstatements.get(i);
  41     }
  42
  43     public String printNode(int indent) {
  44         if (printStyle==NORMAL) {
  45             String st="{\n";
  46             for(int i=0;i<blockstatements.size();i++) {
  47                 BlockStatementNode bsn=(BlockStatementNode)blockstatements.get(i);
  48                 st+=printSpace(indent+INDENT)+bsn.printNode(indent+INDENT);
  49                 if (!((bsn instanceof SubBlockNode)||
  50                       (bsn instanceof LoopNode)||
  51                       (bsn instanceof IfStatementNode)))
  52                     st+=";\n";
  53                 if (bsn instanceof IfStatementNode)
  54                     st+="\n";
  55             }
  56             st+=printSpace(indent)+"}";
  57             return st;
  58         } else if (printStyle==NOBRACES) {
  59             String st="";
  60             for(int i=0;i<blockstatements.size();i++) {
  61                 BlockStatementNode bsn=(BlockStatementNode)blockstatements.get(i);
  62                 st+=printSpace(indent)+bsn.printNode(indent);
  63                 if (!((bsn instanceof SubBlockNode)||
  64                       (bsn instanceof LoopNode)||
  65                       (bsn instanceof IfStatementNode)))
  66                     st+=";";
  67             }
  68             return st;
  69         } else if (printStyle==EXPRLIST) {
  70             String st="";
  71             for(int i=0;i<blockstatements.size();i++) {
  72                 BlockStatementNode bsn=(BlockStatementNode)blockstatements.get(i);
  73                 st+=bsn.printNode(0);
  74                 if ((i+1)!=blockstatements.size())
  75                     st+=", ";
  76             }
  77             return st;
  78         } else throw new Error();
  79     }
  80
  81     public int kind() {
  82         return Kind.BlockNode;
  83     }
  84 }