Robust/src/Analysis/TaskStateAnalysis/TaskQueueIterator.java

   1 package Analysis.TaskStateAnalysis;
   2 import IR.*;
   3 import IR.Tree.*;
   4 import IR.Flat.*;
   5 import java.util.*;
   6
   7 public class TaskQueueIterator {
   8     TaskQueue tq;
   9     int index;
  10     FlagTagState fts;
  11     FlagTagState ftsarray[];
  12     Hashtable<TempDescriptor, TagState> tsarray;
  13     Hashtable<TempDescriptor, Integer> tsindexarray;
  14     Iterator<FlagTagState> itarray[];
  15     boolean needit;
  16     boolean needinit;
  17
  18     public TaskQueueIterator(TaskQueue tq, int index, FlagTagState fts) {
  19         this.tq=tq;
  20         this.index=index;
  21         this.fts=fts;
  22         this.ftsarray=new FlagTagState[tq.numParameters()];
  23         this.ftsarray[index]=fts;
  24         this.tsarray=new Hashtable<TempDescriptor, TagState>();
  25         this.tsindexarray=new Hashtable<TempDescriptor, Integer>();
  26         this.itarray=(Iterator<FlagTagState>[]) new Iterator[tq.numParameters()];
  27         needit=false;
  28         needinit=true;
  29         init();
  30     }
  31
  32     private void init() {
  33         for(int i=ftsarray.length-1;i>=0;i--) {
  34             if (i!=index)
  35                 itarray[i]=tq.parameterset[i].iterator();
  36             VarDescriptor vd=tq.task.getParameter(i);
  37             TagExpressionList tel=tq.task.getTag(vd);
  38             if (tel!=null)
  39                 for(int j=0;j<tel.numTags();j++) {
  40                     TempDescriptor tmp=tel.getTemp(j);
  41                     if (!tsindexarray.containsKey(tmp)) {
  42                         tsindexarray.put(tmp, new Integer(i));
  43                     }
  44                 }
  45         }
  46     }
  47
  48     public boolean hasNext() {
  49         TaskDescriptor td=tq.task;
  50         int i;
  51         if (needinit) {
  52             i=ftsarray.length-1;
  53             if (i!=index)
  54                 ftsarray[i]=itarray[i].next();
  55         } else {
  56             i=0;
  57         }
  58
  59         if (i==0&&index==0&&ftsarray[0]!=null&&!needit) {
  60             needinit=false;
  61             return true;
  62         }
  63
  64         objloop:
  65         for(;i<ftsarray.length;i++) {
  66             FlagState currfs=ftsarray[i].fs;
  67             VarDescriptor vd=td.getParameter(i);
  68             TagExpressionList tel=td.getTag(vd);
  69             int j;
  70             if (needinit) {
  71                 j=(tel!=null)&&tel.numTags()>0?tel.numTags()-1:0;
  72                 needinit=false;
  73             } else
  74                 j=0;
  75             tagloop:
  76             for(;tel!=null&&j<tel.numTags();j++) {
  77                 TempDescriptor tmp=tel.getTemp(j);
  78                 TagState currtag=tsarray.get(tmp);
  79                 String type=tel.getType(j);
  80
  81                 if (tsindexarray.get(tmp).intValue()==i) {
  82                     //doing the assignment right here!!!
  83                     Vector<FlagTagState> possts=tq.map.get(currfs);
  84                     int index=0;
  85                     if (currtag!=null)
  86                         index=possts.indexOf(new FlagTagState(currtag,currfs));
  87                     if (needit) {
  88                         index++;
  89                         needit=false;
  90                     }
  91                     for(int k=index;k<possts.size();k++) {
  92                         FlagTagState posstag=possts.get(k);
  93                         if (posstag.ts.getTag().getSymbol().equals(type)) {
  94                             tsarray.put(tmp, posstag.ts);
  95                             if (j==0) {
  96                                 if (i==0) {
  97                                     //We are done!
  98                                     return true;
  99                                 } else {
 100                                     //Backtrack on objects
 101                                     i-=2;
 102                                     continue objloop;
 103                                 }
 104                             } else {
 105                                 //Backtrack on tags
 106                                 j-=2;
 107                                 continue tagloop;
 108                             }
 109                         }
 110                     }
 111                     //couldn't find a tag
 112                     tsarray.put(tmp, null);
 113                     needit=true;
 114                 } else {
 115                     //check tag compatibility
 116                     if (!currtag.containsFS(currfs)) {
 117                         //incompatible tag set by previous level
 118                         //need to increment object state
 119                         needit=true;
 120                         break;
 121                     }
 122                 }
 123             }
 124             if (index==i) {
 125                 continue;
 126             }
 127             if (needit) {
 128                 if (itarray[i].hasNext()) {
 129                     ftsarray[i]=itarray[i].next();
 130                     needit=false;
 131                     i-=1;
 132                     continue objloop; //backtrack and fix up everything
 133                 } else {
 134                     itarray[i]=tq.parameterset[i].iterator();//keep going backwards
 135                 }
 136             } else {
 137                 throw new Error();
 138             }
 139         }
 140         return false;
 141     }
 142
 143     public FlagTagState getFTS(int index) {
 144         return ftsarray[index];
 145     }
 146
 147     public TagState getTS(TempDescriptor tmp) {
 148         return tsarray.get(tmp);
 149     }
 150
 151     public void next() {
 152         needit=true;
 153     }
 154 }