1) bug fix to guard generateOptionalArrays
[IRC.git] / Robust / src / IR / Flat / BuildCode.java
1 package IR.Flat;
2 import IR.Tree.FlagExpressionNode;
3 import IR.Tree.DNFFlag;
4 import IR.Tree.DNFFlagAtom;
5 import IR.Tree.TagExpressionList;
6 import IR.*;
7 import java.util.*;
8 import java.io.*;
9 import Util.Relation;
10
11 public class BuildCode {
12     State state;
13     Hashtable temptovar;
14     Hashtable paramstable;
15     Hashtable tempstable;
16     Hashtable fieldorder;
17     Hashtable flagorder;
18     int tag=0;
19     String localsprefix="___locals___";
20     String paramsprefix="___params___";
21     public static boolean GENERATEPRECISEGC=false;
22     public static String PREFIX="";
23     public static String arraytype="ArrayObject";
24     Virtual virtualcalls;
25     TypeUtil typeutil;
26     private int maxtaskparams=0;
27     ClassDescriptor[] cdarray;
28     TypeDescriptor[] arraytable;
29
30     public BuildCode(State st, Hashtable temptovar, TypeUtil typeutil) {
31         state=st;
32         this.temptovar=temptovar;
33         paramstable=new Hashtable();    
34         tempstable=new Hashtable();
35         fieldorder=new Hashtable();
36         flagorder=new Hashtable();
37         this.typeutil=typeutil;
38         virtualcalls=new Virtual(state);
39     }
40
41     /** The buildCode method outputs C code for all the methods.  The Flat
42      * versions of the methods must already be generated and stored in
43      * the State object. */
44
45     public void buildCode() {
46         /* Create output streams to write to */
47         PrintWriter outclassdefs=null;
48         PrintWriter outstructs=null;
49         PrintWriter outrepairstructs=null;
50         PrintWriter outmethodheader=null;
51         PrintWriter outmethod=null;
52         PrintWriter outvirtual=null;
53         PrintWriter outtask=null;
54         PrintWriter outtaskdefs=null;
55         PrintWriter outoptionalarrays=null;
56
57         try {
58             OutputStream str=new FileOutputStream(PREFIX+"structdefs.h");
59             outstructs=new java.io.PrintWriter(str, true);
60             str=new FileOutputStream(PREFIX+"methodheaders.h");
61             outmethodheader=new java.io.PrintWriter(str, true);
62             str=new FileOutputStream(PREFIX+"classdefs.h");
63             outclassdefs=new java.io.PrintWriter(str, true);
64             str=new FileOutputStream(PREFIX+"methods.c");
65             outmethod=new java.io.PrintWriter(str, true);
66             str=new FileOutputStream(PREFIX+"virtualtable.h");
67             outvirtual=new java.io.PrintWriter(str, true);
68             if (state.TASK) {
69                 str=new FileOutputStream(PREFIX+"task.h");
70                 outtask=new java.io.PrintWriter(str, true);
71                 str=new FileOutputStream(PREFIX+"taskdefs.c");
72                 outtaskdefs=new java.io.PrintWriter(str, true);
73                 if (state.OPTIONAL){
74                     str=new FileOutputStream(PREFIX+"optionnalarrays.c");
75                     outoptionalarrays=new java.io.PrintWriter(str, true);
76                 } 
77             }
78             if (state.structfile!=null) {
79                 str=new FileOutputStream(PREFIX+state.structfile+".struct");
80                 outrepairstructs=new java.io.PrintWriter(str, true);
81             }
82         } catch (Exception e) {
83             e.printStackTrace();
84             System.exit(-1);
85         }
86
87         /* Build the virtual dispatch tables */
88         buildVirtualTables(outvirtual);
89
90
91         /* Output includes */
92
93         outmethodheader.println("#ifndef METHODHEADERS_H");
94         outmethodheader.println("#define METHODHEADERS_H");
95         outmethodheader.println("#include \"structdefs.h\"");
96
97         outstructs.println("#ifndef STRUCTDEFS_H");
98         outstructs.println("#define STRUCTDEFS_H");
99         outstructs.println("#include \"classdefs.h\"");
100
101
102
103         /* Output types for short array and string */
104         outstructs.println("#define STRINGARRAYTYPE "+
105                            (state.getArrayNumber(
106                                                  (new TypeDescriptor(typeutil.getClass(TypeUtil.StringClass))).makeArray(state))+state.numClasses()));
107
108         outstructs.println("#define OBJECTARRAYTYPE "+
109                            (state.getArrayNumber(
110                                                  (new TypeDescriptor(typeutil.getClass(TypeUtil.ObjectClass))).makeArray(state))+state.numClasses()));
111
112
113         outstructs.println("#define STRINGTYPE "+typeutil.getClass(TypeUtil.StringClass).getId());
114         outstructs.println("#define CHARARRAYTYPE "+
115                            (state.getArrayNumber((new TypeDescriptor(TypeDescriptor.CHAR)).makeArray(state))+state.numClasses()));
116
117         outstructs.println("#define BYTEARRAYTYPE "+
118                            (state.getArrayNumber((new TypeDescriptor(TypeDescriptor.BYTE)).makeArray(state))+state.numClasses()));
119
120         outstructs.println("#define BYTEARRAYARRAYTYPE "+
121                            (state.getArrayNumber((new TypeDescriptor(TypeDescriptor.BYTE)).makeArray(state).makeArray(state))+state.numClasses()));
122         
123         outstructs.println("#define NUMCLASSES "+state.numClasses());
124         if (state.TASK) {
125             outstructs.println("#define STARTUPTYPE "+typeutil.getClass(TypeUtil.StartupClass).getId());
126             outstructs.println("#define TAGTYPE "+typeutil.getClass(TypeUtil.TagClass).getId());
127             outstructs.println("#define TAGARRAYTYPE "+
128                                (state.getArrayNumber(new TypeDescriptor(typeutil.getClass(TypeUtil.TagClass)).makeArray(state))+state.numClasses()));
129         }
130         
131         // Output the C class declarations
132         // These could mutually reference each other
133         if (state.THREAD)
134             outclassdefs.println("#include <pthread.h>");
135
136         outclassdefs.println("struct "+arraytype+";");
137
138         Iterator it=state.getClassSymbolTable().getDescriptorsIterator();
139         while(it.hasNext()) {
140             ClassDescriptor cn=(ClassDescriptor)it.next();
141             outclassdefs.println("struct "+cn.getSafeSymbol()+";");
142         }
143         outclassdefs.println("");
144         {
145             //Print out definition for array type
146             outclassdefs.println("struct "+arraytype+" {");
147             outclassdefs.println("  int type;");
148             if (state.THREAD) {
149                 outclassdefs.println("  pthread_t tid;");
150                 outclassdefs.println("  void * lockentry;");
151                 outclassdefs.println("  int lockcount;");
152             }
153                 
154             if (state.TASK) {
155                 outclassdefs.println("  int flag;");
156                 outclassdefs.println("  void * flagptr;");
157                 outclassdefs.println("  int failedstatus;");
158                 outclassdefs.println("  int * flagset;");
159             }
160             printClassStruct(typeutil.getClass(TypeUtil.ObjectClass), outclassdefs);
161             outclassdefs.println("  int ___length___;");
162             outclassdefs.println("};\n");
163
164             if (state.TASK) {
165             //Print out definitions for task types
166                 outtask.println("struct parameterdescriptor {");
167                 outtask.println("int type;");
168                 outtask.println("int numberterms;");
169                 outtask.println("int *intarray;");
170                 outtask.println("void * queue;");
171                 outtask.println("int numbertags;");
172                 outtask.println("int *tagarray;");
173                 outtask.println("};");
174
175                 outtask.println("struct taskdescriptor {");
176                 outtask.println("void * taskptr;");
177                 outtask.println("int numParameters;");
178                 outtask.println("int numTotal;");
179                 outtask.println("struct parameterdescriptor **descriptorarray;");
180                 outtask.println("char * name;");
181                 outtask.println("};");
182                 outtask.println("extern struct taskdescriptor * taskarray[];");
183                 outtask.println("extern numtasks;");
184             }
185         }
186
187         // Output function prototypes and structures for parameters
188         it=state.getClassSymbolTable().getDescriptorsIterator();
189         while(it.hasNext()) {
190             ClassDescriptor cn=(ClassDescriptor)it.next();
191             generateCallStructs(cn, outclassdefs, outstructs, outmethodheader);
192         }
193
194         if (state.TASK) {
195             /* Map flags to integers */
196             it=state.getClassSymbolTable().getDescriptorsIterator();
197             while(it.hasNext()) {
198                 ClassDescriptor cn=(ClassDescriptor)it.next();
199                 mapFlags(cn);
200             }
201             /* Generate Tasks */
202             generateTaskStructs(outstructs, outmethodheader);
203         }
204
205         outmethodheader.println("#endif");
206
207         outmethodheader.close();
208
209         /* Build the actual methods */
210         outmethod.println("#include \"methodheaders.h\"");
211         outmethod.println("#include \"virtualtable.h\"");
212         outmethod.println("#include <runtime.h>");
213         if (state.THREAD)
214             outmethod.println("#include <thread.h>");
215         if (state.main!=null) {
216             outmethod.println("#include <string.h>");       
217         }
218
219         if (state.CONSCHECK) {
220             outmethod.println("#include \"checkers.h\"");
221         }
222         outclassdefs.println("extern int classsize[];");
223         outclassdefs.println("extern int hasflags[];");
224         outclassdefs.println("extern int * pointerarray[];");
225         outclassdefs.println("extern int supertypes[];");
226
227         //Store the sizes of classes & array elements
228         generateSizeArray(outmethod);
229         
230         //Store table of supertypes
231         generateSuperTypeTable(outmethod);
232
233         //Store the layout of classes
234         generateLayoutStructs(outmethod);
235
236         /* Generate code for methods */
237         Iterator classit=state.getClassSymbolTable().getDescriptorsIterator();
238         while(classit.hasNext()) {
239             ClassDescriptor cn=(ClassDescriptor)classit.next();
240             Iterator methodit=cn.getMethods();
241             while(methodit.hasNext()) {
242                 /* Classify parameters */
243                 MethodDescriptor md=(MethodDescriptor)methodit.next();
244                 FlatMethod fm=state.getMethodFlat(md);
245                 if (!md.getModifiers().isNative())
246                     generateFlatMethod(fm,outmethod);
247             }
248         }
249
250         if (state.TASK) {
251             /* Compile task based program */
252             outtaskdefs.println("#include \"task.h\"");
253             outtaskdefs.println("#include \"methodheaders.h\"");
254             Iterator taskit=state.getTaskSymbolTable().getDescriptorsIterator();
255             while(taskit.hasNext()) {
256                 TaskDescriptor td=(TaskDescriptor)taskit.next();
257                 FlatMethod fm=state.getMethodFlat(td);
258                 generateFlatMethod(fm, outmethod);
259                 generateTaskDescriptor(outtaskdefs, fm, td);
260             }
261
262             {
263                 //Output task descriptors
264                 taskit=state.getTaskSymbolTable().getDescriptorsIterator();
265                 outtaskdefs.println("struct taskdescriptor * taskarray[]= {");
266                 boolean first=true;
267                 while(taskit.hasNext()) {
268                     TaskDescriptor td=(TaskDescriptor)taskit.next();
269                     if (first)
270                         first=false;
271                     else
272                         outtaskdefs.println(",");
273                     outtaskdefs.print("&task_"+td.getSafeSymbol());
274                 }
275                 outtaskdefs.println("};");
276             }
277
278             outtaskdefs.println("int numtasks="+state.getTaskSymbolTable().getValueSet().size()+";");
279
280         } else if (state.main!=null) {
281             /* Generate main method */
282             outmethod.println("int main(int argc, const char *argv[]) {");
283             outmethod.println("  int i;");
284             if (GENERATEPRECISEGC) {
285                 outmethod.println("  struct ArrayObject * stringarray=allocate_newarray(NULL, STRINGARRAYTYPE, argc-1);");
286             } else {
287                 outmethod.println("  struct ArrayObject * stringarray=allocate_newarray(STRINGARRAYTYPE, argc-1);");
288             }
289             if (state.THREAD) {
290                 outmethod.println("initializethreads();");
291             }
292             outmethod.println("  for(i=1;i<argc;i++) {");
293             outmethod.println("    int length=strlen(argv[i]);");
294             if (GENERATEPRECISEGC) {
295                 outmethod.println("    struct ___String___ *newstring=NewString(NULL, argv[i], length);");
296             } else {
297                 outmethod.println("    struct ___String___ *newstring=NewString(argv[i], length);");
298             }
299             outmethod.println("    ((void **)(((char *)& stringarray->___length___)+sizeof(int)))[i-1]=newstring;");
300             outmethod.println("  }");
301
302
303             MethodDescriptor md=typeutil.getMain();
304             ClassDescriptor cd=typeutil.getMainClass();
305
306             outmethod.println("   {");
307             if (GENERATEPRECISEGC) {
308                 outmethod.print("       struct "+cd.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"_params __parameterlist__={");
309                 outmethod.println("1, NULL,"+"stringarray};");
310                 outmethod.println("     "+cd.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"(& __parameterlist__);");
311             } else
312                 outmethod.println("     "+cd.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"(stringarray);");
313             outmethod.println("   }");
314             
315             if (state.THREAD) {
316                 outmethod.println("pthread_mutex_lock(&gclistlock);");
317                 outmethod.println("threadcount--;");
318                 outmethod.println("pthread_cond_signal(&gccond);");
319                 outmethod.println("pthread_mutex_unlock(&gclistlock);");
320                 outmethod.println("pthread_exit(NULL);");
321             }
322             outmethod.println("}");
323         }
324         if (state.TASK)
325             outstructs.println("#define MAXTASKPARAMS "+maxtaskparams);
326
327
328         /* Output structure definitions for repair tool */
329         if (state.structfile!=null) {
330             buildRepairStructs(outrepairstructs);
331             outrepairstructs.close();
332         }
333         outstructs.println("#endif");
334
335         outstructs.close();
336         outmethod.close();
337         
338         if (state.TASK&&state.OPTIONAL) {
339             generateOptionalArrays(outoptionalarrays);
340             outoptionalarrays.close();
341         }
342     }
343
344     private int maxcount=0;
345
346     private void buildRepairStructs(PrintWriter outrepairstructs) {
347         Iterator classit=state.getClassSymbolTable().getDescriptorsIterator();
348         while(classit.hasNext()) {
349             ClassDescriptor cn=(ClassDescriptor)classit.next();
350             outrepairstructs.println("structure "+cn.getSymbol()+" {");
351             outrepairstructs.println("  int __type__;");
352             if (state.TASK) {
353                 outrepairstructs.println("  int __flag__;");
354                 outrepairstructs.println("  int __flagptr__;");
355             }
356             printRepairStruct(cn, outrepairstructs);
357             outrepairstructs.println("}\n");
358         }
359         
360         for(int i=0;i<state.numArrays();i++) {
361             TypeDescriptor tdarray=arraytable[i];
362             TypeDescriptor tdelement=tdarray.dereference();
363             outrepairstructs.println("structure "+arraytype+"_"+state.getArrayNumber(tdarray)+" {");
364             outrepairstructs.println("  int __type__;");
365             printRepairStruct(typeutil.getClass(TypeUtil.ObjectClass), outrepairstructs);
366             outrepairstructs.println("  int length;");
367             /*
368               // Need to add support to repair tool for this
369               if (tdelement.isClass()||tdelement.isArray())
370                 outrepairstructs.println("  "+tdelement.getRepairSymbol()+" * elem[this.length];");
371             else
372                 outrepairstructs.println("  "+tdelement.getRepairSymbol()+" elem[this.length];");
373             */
374             outrepairstructs.println("}\n");
375         }
376
377     }
378
379     private void printRepairStruct(ClassDescriptor cn, PrintWriter output) {
380         ClassDescriptor sp=cn.getSuperDesc();
381         if (sp!=null)
382             printRepairStruct(sp, output);
383         
384         Vector fields=(Vector)fieldorder.get(cn);
385
386         for(int i=0;i<fields.size();i++) {
387             FieldDescriptor fd=(FieldDescriptor)fields.get(i);
388             if (fd.getType().isArray()) {
389                 output.println("  "+arraytype+"_"+ state.getArrayNumber(fd.getType()) +" * "+fd.getSymbol()+";");
390             } else if (fd.getType().isClass())
391                 output.println("  "+fd.getType().getRepairSymbol()+" * "+fd.getSymbol()+";");
392             else if (fd.getType().isFloat())
393                 output.println("  int "+fd.getSymbol()+"; /* really float */");
394             else 
395                 output.println("  "+fd.getType().getRepairSymbol()+" "+fd.getSymbol()+";");
396         }
397     }
398
399     /** This method outputs TaskDescriptor information */
400     void generateTaskDescriptor(PrintWriter output, FlatMethod fm, TaskDescriptor task) {
401         for (int i=0;i<task.numParameters();i++) {
402             VarDescriptor param_var=task.getParameter(i);
403             TypeDescriptor param_type=task.getParamType(i);
404             FlagExpressionNode param_flag=task.getFlag(param_var);
405             TagExpressionList param_tag=task.getTag(param_var);
406
407             int dnfterms;
408             if (param_flag==null) {
409                 output.println("int parameterdnf_"+i+"_"+task.getSafeSymbol()+"[]={");
410                 output.println("0x0, 0x0 };");
411                 dnfterms=1;
412             } else {
413
414                 DNFFlag dflag=param_flag.getDNF();
415                 dnfterms=dflag.size();
416                 
417                 Hashtable flags=(Hashtable)flagorder.get(param_type.getClassDesc());
418                 output.println("int parameterdnf_"+i+"_"+task.getSafeSymbol()+"[]={");
419                 for(int j=0;j<dflag.size();j++) {
420                     if (j!=0)
421                         output.println(",");
422                     Vector term=dflag.get(j);
423                     int andmask=0;
424                     int checkmask=0;
425                     for(int k=0;k<term.size();k++) {
426                         DNFFlagAtom dfa=(DNFFlagAtom)term.get(k);
427                         FlagDescriptor fd=dfa.getFlag();
428                         boolean negated=dfa.getNegated();
429                         int flagid=1<<((Integer)flags.get(fd)).intValue();
430                         andmask|=flagid;
431                         if (!negated)
432                             checkmask|=flagid;
433                     }
434                     output.print("0x"+Integer.toHexString(andmask)+", 0x"+Integer.toHexString(checkmask));
435                 }
436                 output.println("};");
437             }
438
439             output.println("int parametertag_"+i+"_"+task.getSafeSymbol()+"[]={");
440             //BUG...added next line to fix, test with any task program
441             if (param_tag!=null)
442                 for(int j=0;j<param_tag.numTags();j++) {
443                     if (j!=0)
444                         output.println(",");
445                     /* for each tag we need */
446                     /* which slot it is */
447                     /* what type it is */
448                     TagVarDescriptor tvd=(TagVarDescriptor)task.getParameterTable().get(param_tag.getName(j));
449                     TempDescriptor tmp=param_tag.getTemp(j);
450                     int slot=fm.getTagInt(tmp);
451                     output.println(slot+", "+state.getTagId(tvd.getTag()));
452                 }
453             output.println("};");
454
455             output.println("struct parameterdescriptor parameter_"+i+"_"+task.getSafeSymbol()+"={");
456             output.println("/* type */"+param_type.getClassDesc().getId()+",");
457             output.println("/* number of DNF terms */"+dnfterms+",");
458             output.println("parameterdnf_"+i+"_"+task.getSafeSymbol()+",");
459             output.println("0,");
460             //BUG, added next line to fix and else statement...test
461             //with any task program
462             if (param_tag!=null)
463                 output.println("/* number of tags */"+param_tag.numTags()+",");
464             else
465                 output.println("/* number of tags */ 0,");
466             output.println("parametertag_"+i+"_"+task.getSafeSymbol());
467             output.println("};");
468         }
469
470
471         output.println("struct parameterdescriptor * parameterdescriptors_"+task.getSafeSymbol()+"[] = {");
472         for (int i=0;i<task.numParameters();i++) {
473             if (i!=0)
474                 output.println(",");
475             output.print("&parameter_"+i+"_"+task.getSafeSymbol());
476         }
477         output.println("};");
478
479         output.println("struct taskdescriptor task_"+task.getSafeSymbol()+"={");
480         output.println("&"+task.getSafeSymbol()+",");
481         output.println("/* number of parameters */" +task.numParameters() + ",");
482         int numtotal=task.numParameters()+fm.numTags();
483         output.println("/* number total parameters */" +numtotal + ",");
484         output.println("parameterdescriptors_"+task.getSafeSymbol()+",");
485         output.println("\""+task.getSymbol()+"\"");
486         output.println("};");
487     }
488
489
490     /** The buildVirtualTables method outputs the virtual dispatch
491      * tables for methods. */
492
493     private void buildVirtualTables(PrintWriter outvirtual) {
494         Iterator classit=state.getClassSymbolTable().getDescriptorsIterator();
495         while(classit.hasNext()) {
496             ClassDescriptor cd=(ClassDescriptor)classit.next();
497             if (virtualcalls.getMethodCount(cd)>maxcount)
498                 maxcount=virtualcalls.getMethodCount(cd);
499         }
500         MethodDescriptor[][] virtualtable=new MethodDescriptor[state.numClasses()+state.numArrays()][maxcount];
501
502         /* Fill in virtual table */
503         classit=state.getClassSymbolTable().getDescriptorsIterator();
504         while(classit.hasNext()) {
505             ClassDescriptor cd=(ClassDescriptor)classit.next();
506             fillinRow(cd, virtualtable, cd.getId());
507         }
508
509         ClassDescriptor objectcd=typeutil.getClass(TypeUtil.ObjectClass);
510         Iterator arrayit=state.getArrayIterator();
511         while(arrayit.hasNext()) {
512             TypeDescriptor td=(TypeDescriptor)arrayit.next();
513             int id=state.getArrayNumber(td);
514             fillinRow(objectcd, virtualtable, id+state.numClasses());
515         }
516         
517         outvirtual.print("void * virtualtable[]={");
518         boolean needcomma=false;
519         for(int i=0;i<state.numClasses()+state.numArrays();i++) {
520             for(int j=0;j<maxcount;j++) {
521                 if (needcomma)
522                     outvirtual.print(", ");
523                 if (virtualtable[i][j]!=null) {
524                     MethodDescriptor md=virtualtable[i][j];
525                     outvirtual.print("& "+md.getClassDesc().getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor());
526                 } else {
527                     outvirtual.print("0");
528                 }
529                 needcomma=true;
530             }
531             outvirtual.println("");
532         }
533         outvirtual.println("};");
534         outvirtual.close();
535     }
536
537     private void fillinRow(ClassDescriptor cd, MethodDescriptor[][] virtualtable, int rownum) {
538         /* Get inherited methods */
539         if (cd.getSuperDesc()!=null)
540             fillinRow(cd.getSuperDesc(), virtualtable, rownum);
541         /* Override them with our methods */
542         for(Iterator it=cd.getMethods();it.hasNext();) {
543             MethodDescriptor md=(MethodDescriptor)it.next();
544             if (md.isStatic()||md.getReturnType()==null)
545                 continue;
546             int methodnum=virtualcalls.getMethodNumber(md);
547             virtualtable[rownum][methodnum]=md;
548         }
549     }
550
551     /** Generate array that contains the sizes of class objects.  The
552      * object allocation functions in the runtime use this
553      * information. */
554
555     private void generateSizeArray(PrintWriter outclassdefs) {
556         outclassdefs.print("int classsize[]={");
557         Iterator it=state.getClassSymbolTable().getDescriptorsIterator();
558         cdarray=new ClassDescriptor[state.numClasses()];
559         while(it.hasNext()) {
560             ClassDescriptor cd=(ClassDescriptor)it.next();
561             cdarray[cd.getId()]=cd;
562         }
563         boolean needcomma=false;
564         for(int i=0;i<state.numClasses();i++) {
565             if (needcomma)
566                 outclassdefs.print(", ");
567             outclassdefs.print("sizeof(struct "+cdarray[i].getSafeSymbol()+")");            
568             needcomma=true;
569         }
570
571         arraytable=new TypeDescriptor[state.numArrays()];
572
573         Iterator arrayit=state.getArrayIterator();
574         while(arrayit.hasNext()) {
575             TypeDescriptor td=(TypeDescriptor)arrayit.next();
576             int id=state.getArrayNumber(td);
577             arraytable[id]=td;
578         }
579         
580         for(int i=0;i<state.numArrays();i++) {
581             if (needcomma)
582                 outclassdefs.print(", ");
583             TypeDescriptor tdelement=arraytable[i].dereference();
584             if (tdelement.isArray()||tdelement.isClass())
585                 outclassdefs.print("sizeof(void *)");
586             else
587                 outclassdefs.print("sizeof("+tdelement.getSafeSymbol()+")");
588             needcomma=true;
589         }
590
591         outclassdefs.println("};");
592     }
593
594     /** Constructs params and temp objects for each method or task.
595      * These objects tell the compiler which temps need to be
596      * allocated.  */
597
598     private void generateTempStructs(FlatMethod fm) {
599         MethodDescriptor md=fm.getMethod();
600         TaskDescriptor task=fm.getTask();
601
602         ParamsObject objectparams=md!=null?new ParamsObject(md,tag++):new ParamsObject(task, tag++);
603         if (md!=null)
604             paramstable.put(md, objectparams);
605         else
606             paramstable.put(task, objectparams);
607
608         for(int i=0;i<fm.numParameters();i++) {
609             TempDescriptor temp=fm.getParameter(i);
610             TypeDescriptor type=temp.getType();
611             if ((type.isPtr()||type.isArray())&&GENERATEPRECISEGC)
612                 objectparams.addPtr(temp);
613             else
614                 objectparams.addPrim(temp);
615         }
616
617         for(int i=0;i<fm.numTags();i++) {
618             TempDescriptor temp=fm.getTag(i);
619             if (GENERATEPRECISEGC)
620                 objectparams.addPtr(temp);
621             else
622                 objectparams.addPrim(temp);
623         }
624
625         TempObject objecttemps=md!=null?new TempObject(objectparams,md,tag++):new TempObject(objectparams, task, tag++);
626         if (md!=null)
627             tempstable.put(md, objecttemps);
628         else
629             tempstable.put(task, objecttemps);
630
631         for(Iterator nodeit=fm.getNodeSet().iterator();nodeit.hasNext();) {
632             FlatNode fn=(FlatNode)nodeit.next();
633             TempDescriptor[] writes=fn.writesTemps();
634             for(int i=0;i<writes.length;i++) {
635                 TempDescriptor temp=writes[i];
636                 TypeDescriptor type=temp.getType();
637                 if ((type.isPtr()||type.isArray())&&GENERATEPRECISEGC)
638                     objecttemps.addPtr(temp);
639                 else
640                     objecttemps.addPrim(temp);
641             }
642         }
643     }
644     
645     private void generateLayoutStructs(PrintWriter output) {
646         Iterator it=state.getClassSymbolTable().getDescriptorsIterator();
647         while(it.hasNext()) {
648             ClassDescriptor cn=(ClassDescriptor)it.next();
649             output.println("int "+cn.getSafeSymbol()+"_pointers[]={");
650             Iterator allit=cn.getFieldTable().getAllDescriptorsIterator();
651             int count=0;
652             while(allit.hasNext()) {
653                 FieldDescriptor fd=(FieldDescriptor)allit.next();
654                 TypeDescriptor type=fd.getType();
655                 if (type.isPtr()||type.isArray())
656                     count++;
657             }
658             output.print(count);
659             allit=cn.getFieldTable().getAllDescriptorsIterator();
660             while(allit.hasNext()) {
661                 FieldDescriptor fd=(FieldDescriptor)allit.next();
662                 TypeDescriptor type=fd.getType();
663                 if (type.isPtr()||type.isArray()) {
664                     output.println(",");
665                     output.print("((int)&(((struct "+cn.getSafeSymbol() +" *)0)->"+fd.getSafeSymbol()+"))");
666                 }
667             }
668             output.println("};");
669         }
670         output.println("int * pointerarray[]={");
671         boolean needcomma=false;
672         for(int i=0;i<state.numClasses();i++) {
673             ClassDescriptor cn=cdarray[i];
674             if (needcomma)
675                 output.println(",");
676             needcomma=true;
677             output.print(cn.getSafeSymbol()+"_pointers");
678         }
679
680         for(int i=0;i<state.numArrays();i++) {
681             if (needcomma)
682                 output.println(", ");
683             TypeDescriptor tdelement=arraytable[i].dereference();
684             if (tdelement.isArray()||tdelement.isClass())
685                 output.print("((int *)1)");
686             else
687                 output.print("0");
688             needcomma=true;
689         }
690         
691         output.println("};");
692         needcomma=false;
693         output.println("int hasflags[]={");
694         for(int i=0;i<state.numClasses();i++) {
695             ClassDescriptor cn=cdarray[i];
696             if (needcomma)
697                 output.println(", ");
698             needcomma=true;
699             if (cn.hasFlags())
700                 output.print("1");
701             else
702                 output.print("0");
703         }
704         output.println("};");
705     }
706
707     /** Print out table to give us supertypes */
708     private void generateSuperTypeTable(PrintWriter output) {
709         output.println("int supertypes[]={");
710         boolean needcomma=false;
711         for(int i=0;i<state.numClasses();i++) {
712             ClassDescriptor cn=cdarray[i];
713             if (needcomma)
714                 output.println(",");
715             needcomma=true;
716             if (cn.getSuperDesc()!=null) {
717                 ClassDescriptor cdsuper=cn.getSuperDesc();
718                 output.print(cdsuper.getId());
719             } else
720                 output.print("-1");
721         }
722         output.println("};");
723     }
724
725     /** Force consistent field ordering between inherited classes. */
726
727     private void printClassStruct(ClassDescriptor cn, PrintWriter classdefout) {
728         ClassDescriptor sp=cn.getSuperDesc();
729         if (sp!=null)
730             printClassStruct(sp, classdefout);
731         
732         if (!fieldorder.containsKey(cn)) {
733             Vector fields=new Vector();
734             fieldorder.put(cn,fields);
735             Iterator fieldit=cn.getFields();
736             while(fieldit.hasNext()) {
737                 FieldDescriptor fd=(FieldDescriptor)fieldit.next();
738                 if (sp==null||!sp.getFieldTable().contains(fd.getSymbol()))
739                     fields.add(fd);
740             }
741         }
742         Vector fields=(Vector)fieldorder.get(cn);
743
744         for(int i=0;i<fields.size();i++) {
745             FieldDescriptor fd=(FieldDescriptor)fields.get(i);
746             if (fd.getType().isClass()||fd.getType().isArray())
747                 classdefout.println("  struct "+fd.getType().getSafeSymbol()+" * "+fd.getSafeSymbol()+";");
748             else 
749                 classdefout.println("  "+fd.getType().getSafeSymbol()+" "+fd.getSafeSymbol()+";");
750         }
751     }
752
753
754     /* Map flags to integers consistently between inherited
755      * classes. */
756
757     private void mapFlags(ClassDescriptor cn) {
758         ClassDescriptor sp=cn.getSuperDesc();
759         if (sp!=null)
760             mapFlags(sp);
761         int max=0;
762         if (!flagorder.containsKey(cn)) {
763             Hashtable flags=new Hashtable();
764             flagorder.put(cn,flags);
765             if (sp!=null) {
766                 Hashtable superflags=(Hashtable)flagorder.get(sp);
767                 Iterator superflagit=superflags.keySet().iterator();
768                 while(superflagit.hasNext()) {
769                     FlagDescriptor fd=(FlagDescriptor)superflagit.next();
770                     Integer number=(Integer)superflags.get(fd);
771                     flags.put(fd, number);
772                     if ((number.intValue()+1)>max)
773                         max=number.intValue()+1;
774                 }
775             }
776             
777             Iterator flagit=cn.getFlags();
778             while(flagit.hasNext()) {
779                 FlagDescriptor fd=(FlagDescriptor)flagit.next();
780                 if (sp==null||!sp.getFlagTable().contains(fd.getSymbol()))
781                     flags.put(fd, new Integer(max++));
782             }
783         }
784     }
785
786
787     /** This function outputs (1) structures that parameters are
788      * passed in (when PRECISE GC is enabled) and (2) function
789      * prototypes for the methods */
790
791     private void generateCallStructs(ClassDescriptor cn, PrintWriter classdefout, PrintWriter output, PrintWriter headersout) {
792         /* Output class structure */
793         classdefout.println("struct "+cn.getSafeSymbol()+" {");
794         classdefout.println("  int type;");
795         if (state.THREAD) {
796             classdefout.println("  pthread_t tid;");
797             classdefout.println("  void * lockentry;");
798             classdefout.println("  int lockcount;");
799         }
800
801         if (state.TASK) {
802             classdefout.println("  int flag;");
803             classdefout.println("  void * flagptr;");
804             classdefout.println("  int failedstatus;");
805             classdefout.println("  int * flagset;");
806         }
807         printClassStruct(cn, classdefout);
808         classdefout.println("};\n");
809
810         /* Cycle through methods */
811         Iterator methodit=cn.getMethods();
812         while(methodit.hasNext()) {
813             /* Classify parameters */
814             MethodDescriptor md=(MethodDescriptor)methodit.next();
815             FlatMethod fm=state.getMethodFlat(md);
816             generateTempStructs(fm);
817
818             ParamsObject objectparams=(ParamsObject) paramstable.get(md);
819             TempObject objecttemps=(TempObject) tempstable.get(md);
820
821             /* Output parameter structure */
822             if (GENERATEPRECISEGC) {
823                 output.println("struct "+cn.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"_params {");
824                 output.println("  int size;");
825                 output.println("  void * next;");
826                 for(int i=0;i<objectparams.numPointers();i++) {
827                     TempDescriptor temp=objectparams.getPointer(i);
828                     output.println("  struct "+temp.getType().getSafeSymbol()+" * "+temp.getSafeSymbol()+";");
829                 }
830                 output.println("};\n");
831             }
832
833             /* Output temp structure */
834             if (GENERATEPRECISEGC) {
835                 output.println("struct "+cn.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"_locals {");
836                 output.println("  int size;");
837                 output.println("  void * next;");
838                 for(int i=0;i<objecttemps.numPointers();i++) {
839                     TempDescriptor temp=objecttemps.getPointer(i);
840                     if (temp.getType().isNull())
841                         output.println("  void * "+temp.getSafeSymbol()+";");
842                     else
843                         output.println("  struct "+temp.getType().getSafeSymbol()+" * "+temp.getSafeSymbol()+";");
844                 }
845                 output.println("};\n");
846             }
847             
848             /* Output method declaration */
849             if (md.getReturnType()!=null) {
850                 if (md.getReturnType().isClass()||md.getReturnType().isArray())
851                     headersout.print("struct " + md.getReturnType().getSafeSymbol()+" * ");
852                 else
853                     headersout.print(md.getReturnType().getSafeSymbol()+" ");
854             } else 
855                 //catch the constructor case
856                 headersout.print("void ");
857             headersout.print(cn.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"(");
858             
859             boolean printcomma=false;
860             if (GENERATEPRECISEGC) {
861                 headersout.print("struct "+cn.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"_params * "+paramsprefix);
862                 printcomma=true;
863             }
864
865             //output parameter list
866             for(int i=0;i<objectparams.numPrimitives();i++) {
867                 TempDescriptor temp=objectparams.getPrimitive(i);
868                 if (printcomma)
869                     headersout.print(", ");
870                 printcomma=true;
871                 if (temp.getType().isClass()||temp.getType().isArray())
872                     headersout.print("struct " + temp.getType().getSafeSymbol()+" * "+temp.getSafeSymbol());
873                 else
874                     headersout.print(temp.getType().getSafeSymbol()+" "+temp.getSafeSymbol());
875             }
876             headersout.println(");\n");
877         }
878     }
879
880
881     /** This function outputs (1) structures that parameters are
882      * passed in (when PRECISE GC is enabled) and (2) function
883      * prototypes for the tasks */
884
885     private void generateTaskStructs(PrintWriter output, PrintWriter headersout) {
886         /* Cycle through tasks */
887         Iterator taskit=state.getTaskSymbolTable().getDescriptorsIterator();
888
889         while(taskit.hasNext()) {
890             /* Classify parameters */
891             TaskDescriptor task=(TaskDescriptor)taskit.next();
892             FlatMethod fm=state.getMethodFlat(task);
893             generateTempStructs(fm);
894
895             ParamsObject objectparams=(ParamsObject) paramstable.get(task);
896             TempObject objecttemps=(TempObject) tempstable.get(task);
897
898             /* Output parameter structure */
899             if (GENERATEPRECISEGC) {
900                 output.println("struct "+task.getSafeSymbol()+"_params {");
901
902                 output.println("  int size;");
903                 output.println("  void * next;");
904                 for(int i=0;i<objectparams.numPointers();i++) {
905                     TempDescriptor temp=objectparams.getPointer(i);
906                     output.println("  struct "+temp.getType().getSafeSymbol()+" * "+temp.getSafeSymbol()+";");
907                 }
908
909                 output.println("};\n");
910                 if ((objectparams.numPointers()+fm.numTags())>maxtaskparams) {
911                     maxtaskparams=objectparams.numPointers()+fm.numTags();
912                 }
913             }
914
915             /* Output temp structure */
916             if (GENERATEPRECISEGC) {
917                 output.println("struct "+task.getSafeSymbol()+"_locals {");
918                 output.println("  int size;");
919                 output.println("  void * next;");
920                 for(int i=0;i<objecttemps.numPointers();i++) {
921                     TempDescriptor temp=objecttemps.getPointer(i);
922                     if (temp.getType().isNull())
923                         output.println("  void * "+temp.getSafeSymbol()+";");
924                     else if(temp.getType().isTag())
925                         output.println("  struct "+
926                                        (new TypeDescriptor(typeutil.getClass(TypeUtil.TagClass))).getSafeSymbol()+" * "+temp.getSafeSymbol()+";");
927                     else
928                         output.println("  struct "+temp.getType().getSafeSymbol()+" * "+temp.getSafeSymbol()+";");
929                 }
930                 output.println("};\n");
931             }
932             
933             /* Output task declaration */
934             headersout.print("void " + task.getSafeSymbol()+"(");
935             
936             boolean printcomma=false;
937             if (GENERATEPRECISEGC) {
938                 headersout.print("struct "+task.getSafeSymbol()+"_params * "+paramsprefix);
939             } else
940                 headersout.print("void * parameterarray[]");
941             headersout.println(");\n");
942         }
943     }
944
945     /** Generate code for flatmethod fm. */
946
947     private void generateFlatMethod(FlatMethod fm, PrintWriter output) {
948         MethodDescriptor md=fm.getMethod();
949         TaskDescriptor task=fm.getTask();
950
951         ClassDescriptor cn=md!=null?md.getClassDesc():null;
952
953         ParamsObject objectparams=(ParamsObject)paramstable.get(md!=null?md:task);
954
955         generateHeader(fm, md!=null?md:task,output);
956
957         TempObject objecttemp=(TempObject) tempstable.get(md!=null?md:task);
958
959         /* Print code */
960         if (GENERATEPRECISEGC) {
961             if (md!=null)
962                 output.print("   struct "+cn.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"_locals "+localsprefix+"={");
963             else
964                 output.print("   struct "+task.getSafeSymbol()+"_locals "+localsprefix+"={");
965
966             output.print(objecttemp.numPointers()+",");
967             output.print(paramsprefix);
968             for(int j=0;j<objecttemp.numPointers();j++)
969                 output.print(", NULL");
970             output.println("};");
971         }
972
973         for(int i=0;i<objecttemp.numPrimitives();i++) {
974             TempDescriptor td=objecttemp.getPrimitive(i);
975             TypeDescriptor type=td.getType();
976             if (type.isNull())
977                 output.println("   void * "+td.getSafeSymbol()+";");
978             else if (type.isClass()||type.isArray())
979                 output.println("   struct "+type.getSafeSymbol()+" * "+td.getSafeSymbol()+";");
980             else
981                 output.println("   "+type.getSafeSymbol()+" "+td.getSafeSymbol()+";");
982         }
983
984         /* Generate labels first */
985         HashSet tovisit=new HashSet();
986         HashSet visited=new HashSet();
987         int labelindex=0;
988         Hashtable nodetolabel=new Hashtable();
989         tovisit.add(fm.getNext(0));
990         FlatNode current_node=null;
991
992         //Assign labels 1st
993         //Node needs a label if it is
994         while(!tovisit.isEmpty()) {
995             FlatNode fn=(FlatNode)tovisit.iterator().next();
996             tovisit.remove(fn);
997             visited.add(fn);
998             for(int i=0;i<fn.numNext();i++) {
999                 FlatNode nn=fn.getNext(i);
1000                 if(i>0) {
1001                     //1) Edge >1 of node
1002                     nodetolabel.put(nn,new Integer(labelindex++));
1003                 }
1004                 if (!visited.contains(nn)&&!tovisit.contains(nn)) {
1005                     tovisit.add(nn);
1006                 } else {
1007                     //2) Join point
1008                     nodetolabel.put(nn,new Integer(labelindex++));
1009                 }
1010             }
1011         }
1012
1013         if (state.THREAD&&GENERATEPRECISEGC) {
1014             output.println("checkcollect(&"+localsprefix+");");
1015         }
1016         
1017         //Do the actual code generation
1018         tovisit=new HashSet();
1019         visited=new HashSet();
1020         tovisit.add(fm.getNext(0));
1021         while(current_node!=null||!tovisit.isEmpty()) {
1022             if (current_node==null) {
1023                 current_node=(FlatNode)tovisit.iterator().next();
1024                 tovisit.remove(current_node);
1025             }
1026             visited.add(current_node);
1027             if (nodetolabel.containsKey(current_node))
1028                 output.println("L"+nodetolabel.get(current_node)+":");
1029             if (state.INSTRUCTIONFAILURE) {
1030                 if (state.THREAD) {
1031                     output.println("if ((++instructioncount)>failurecount) {instructioncount=0;injectinstructionfailure();}");
1032                 }
1033                 else
1034                     output.println("if ((--instructioncount)==0) injectinstructionfailure();");
1035             }
1036             if (current_node.numNext()==0) {
1037                 output.print("   ");
1038                 generateFlatNode(fm, current_node, output);
1039                 if (current_node.kind()!=FKind.FlatReturnNode) {
1040                     output.println("   return;");
1041                 }
1042                 current_node=null;
1043             } else if(current_node.numNext()==1) {
1044                 output.print("   ");
1045                 generateFlatNode(fm, current_node, output);
1046                 FlatNode nextnode=current_node.getNext(0);
1047                 if (visited.contains(nextnode)) {
1048                     output.println("goto L"+nodetolabel.get(nextnode)+";");
1049                     current_node=null;
1050                 } else
1051                     current_node=nextnode;
1052             } else if (current_node.numNext()==2) {
1053                 /* Branch */
1054                 output.print("   ");
1055                 generateFlatCondBranch(fm, (FlatCondBranch)current_node, "L"+nodetolabel.get(current_node.getNext(1)), output);
1056                 if (!visited.contains(current_node.getNext(1)))
1057                     tovisit.add(current_node.getNext(1));
1058                 if (visited.contains(current_node.getNext(0))) {
1059                     output.println("goto L"+nodetolabel.get(current_node.getNext(0))+";");
1060                     current_node=null;
1061                 } else
1062                     current_node=current_node.getNext(0);
1063             } else throw new Error();
1064         }
1065
1066
1067         output.println("}\n\n");
1068     }
1069
1070     /** Generate text string that corresponds to the Temp td. */
1071     private String generateTemp(FlatMethod fm, TempDescriptor td) {
1072         MethodDescriptor md=fm.getMethod();
1073         TaskDescriptor task=fm.getTask();
1074         TempObject objecttemps=(TempObject) tempstable.get(md!=null?md:task);
1075
1076         if (objecttemps.isLocalPrim(td)||objecttemps.isParamPrim(td)) {
1077             return td.getSafeSymbol();
1078         }
1079
1080         if (objecttemps.isLocalPtr(td)) {
1081             return localsprefix+"."+td.getSafeSymbol();
1082         }
1083
1084         if (objecttemps.isParamPtr(td)) {
1085             return paramsprefix+"->"+td.getSafeSymbol();
1086         }
1087         throw new Error();
1088     }
1089
1090     private void generateFlatNode(FlatMethod fm, FlatNode fn, PrintWriter output) {
1091         switch(fn.kind()) {
1092         case FKind.FlatTagDeclaration:
1093             generateFlatTagDeclaration(fm, (FlatTagDeclaration) fn,output);
1094             return;
1095         case FKind.FlatCall:
1096             generateFlatCall(fm, (FlatCall) fn,output);
1097             return;
1098         case FKind.FlatFieldNode:
1099             generateFlatFieldNode(fm, (FlatFieldNode) fn,output);
1100             return;
1101         case FKind.FlatElementNode:
1102             generateFlatElementNode(fm, (FlatElementNode) fn,output);
1103             return;
1104         case FKind.FlatSetElementNode:
1105             generateFlatSetElementNode(fm, (FlatSetElementNode) fn,output);
1106             return;
1107         case FKind.FlatSetFieldNode:
1108             generateFlatSetFieldNode(fm, (FlatSetFieldNode) fn,output);
1109             return;
1110         case FKind.FlatNew:
1111             generateFlatNew(fm, (FlatNew) fn,output);
1112             return;
1113         case FKind.FlatOpNode:
1114             generateFlatOpNode(fm, (FlatOpNode) fn,output);
1115             return;
1116         case FKind.FlatCastNode:
1117             generateFlatCastNode(fm, (FlatCastNode) fn,output);
1118             return;
1119         case FKind.FlatLiteralNode:
1120             generateFlatLiteralNode(fm, (FlatLiteralNode) fn,output);
1121             return;
1122         case FKind.FlatReturnNode:
1123             generateFlatReturnNode(fm, (FlatReturnNode) fn,output);
1124             return;
1125         case FKind.FlatNop:
1126             output.println("/* nop */");
1127             return;
1128         case FKind.FlatBackEdge:
1129             if (state.THREAD&&GENERATEPRECISEGC) {
1130                 output.println("checkcollect(&"+localsprefix+");");
1131             } else
1132                 output.println("/* nop */");
1133             return;
1134         case FKind.FlatCheckNode:
1135             generateFlatCheckNode(fm, (FlatCheckNode) fn, output);
1136             return;
1137         case FKind.FlatFlagActionNode:
1138             generateFlatFlagActionNode(fm, (FlatFlagActionNode) fn, output);
1139             return;
1140         }
1141         throw new Error();
1142
1143     }
1144
1145     private void generateFlatCheckNode(FlatMethod fm,  FlatCheckNode fcn, PrintWriter output) {
1146
1147         if (state.CONSCHECK) {
1148             String specname=fcn.getSpec();
1149             String varname="repairstate___";
1150             output.println("{");
1151             output.println("struct "+specname+"_state * "+varname+"=allocate"+specname+"_state();");
1152
1153             TempDescriptor[] temps=fcn.getTemps();
1154             String[] vars=fcn.getVars();
1155             for(int i=0;i<temps.length;i++) {
1156                 output.println(varname+"->"+vars[i]+"=(int)"+generateTemp(fm, temps[i])+";");
1157             }
1158
1159             output.println("if (doanalysis"+specname+"("+varname+")) {");
1160             output.println("free"+specname+"_state("+varname+");");
1161             output.println("} else {");
1162             output.println("/* Bad invariant */");
1163             output.println("free"+specname+"_state("+varname+");");
1164             output.println("abort_task();");
1165             output.println("}");
1166
1167             output.println("}");
1168         }
1169     }
1170
1171     private void generateFlatCall(FlatMethod fm, FlatCall fc, PrintWriter output) {
1172         MethodDescriptor md=fc.getMethod();
1173         ParamsObject objectparams=(ParamsObject) paramstable.get(md);
1174         ClassDescriptor cn=md.getClassDesc();
1175         output.println("{");
1176         if (GENERATEPRECISEGC) {
1177             output.print("       struct "+cn.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"_params __parameterlist__={");
1178             
1179             output.print(objectparams.numPointers());
1180             //      output.print(objectparams.getUID());
1181             output.print(", & "+localsprefix);
1182             if (fc.getThis()!=null) {
1183                 output.print(", ");
1184                 output.print("(struct "+md.getThis().getType().getSafeSymbol() +" *)"+ generateTemp(fm,fc.getThis()));
1185             }
1186             for(int i=0;i<fc.numArgs();i++) {
1187                 Descriptor var=md.getParameter(i);
1188                 TempDescriptor paramtemp=(TempDescriptor)temptovar.get(var);
1189                 if (objectparams.isParamPtr(paramtemp)) {
1190                     TempDescriptor targ=fc.getArg(i);
1191                     output.print(", ");
1192                     TypeDescriptor td=md.getParamType(i);
1193                     if (td.isTag())
1194                         output.print("(struct "+(new TypeDescriptor(typeutil.getClass(TypeUtil.TagClass))).getSafeSymbol()  +" *)"+generateTemp(fm, targ));
1195                     else
1196                         output.print("(struct "+md.getParamType(i).getSafeSymbol()  +" *)"+generateTemp(fm, targ));
1197                 }
1198             }
1199             output.println("};");
1200         }
1201         output.print("       ");
1202
1203
1204         if (fc.getReturnTemp()!=null)
1205             output.print(generateTemp(fm,fc.getReturnTemp())+"=");
1206
1207         /* Do we need to do virtual dispatch? */
1208         if (md.isStatic()||md.getReturnType()==null||singleCall(fc.getThis().getType().getClassDesc(),md)) {
1209             output.print(cn.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor());
1210         } else {
1211             
1212             output.print("((");
1213             if (md.getReturnType().isClass()||md.getReturnType().isArray())
1214                 output.print("struct " + md.getReturnType().getSafeSymbol()+" * ");
1215             else
1216                 output.print(md.getReturnType().getSafeSymbol()+" ");
1217             output.print("(*)(");
1218
1219             boolean printcomma=false;
1220             if (GENERATEPRECISEGC) {
1221                 output.print("struct "+cn.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"_params * ");
1222                 printcomma=true;
1223             } 
1224
1225
1226             for(int i=0;i<objectparams.numPrimitives();i++) {
1227                 TempDescriptor temp=objectparams.getPrimitive(i);
1228                 if (printcomma)
1229                     output.print(", ");
1230                 printcomma=true;
1231                 if (temp.getType().isClass()||temp.getType().isArray())
1232                     output.print("struct " + temp.getType().getSafeSymbol()+" * ");
1233                 else
1234                     output.print(temp.getType().getSafeSymbol());
1235             }
1236
1237             output.print("))virtualtable["+generateTemp(fm,fc.getThis())+"->type*"+maxcount+"+"+virtualcalls.getMethodNumber(md)+"])");
1238         }
1239
1240         output.print("(");
1241         boolean needcomma=false;
1242         if (GENERATEPRECISEGC) {
1243             output.print("&__parameterlist__");
1244             needcomma=true;
1245         } else {
1246             if (fc.getThis()!=null) {
1247                 TypeDescriptor ptd=md.getThis().getType();
1248                 if (ptd.isClass()&&!ptd.isArray())
1249                     output.print("(struct "+ptd.getSafeSymbol()+" *) ");
1250                 output.print(generateTemp(fm,fc.getThis()));
1251                 needcomma=true;
1252             }
1253         }
1254         for(int i=0;i<fc.numArgs();i++) {
1255             Descriptor var=md.getParameter(i);
1256             TempDescriptor paramtemp=(TempDescriptor)temptovar.get(var);
1257             if (objectparams.isParamPrim(paramtemp)) {
1258                 TempDescriptor targ=fc.getArg(i);
1259                 if (needcomma)
1260                     output.print(", ");
1261
1262                 TypeDescriptor ptd=md.getParamType(i);
1263                 if (ptd.isClass()&&!ptd.isArray())
1264                     output.print("(struct "+ptd.getSafeSymbol()+" *) ");
1265                 output.print(generateTemp(fm, targ));
1266                 needcomma=true;
1267             }
1268         }
1269         output.println(");");
1270         output.println("   }");
1271     }
1272
1273     private boolean singleCall(ClassDescriptor thiscd, MethodDescriptor md) {
1274         Set subclasses=typeutil.getSubClasses(thiscd);
1275         if (subclasses==null)
1276             return true;
1277         for(Iterator classit=subclasses.iterator();classit.hasNext();) {
1278             ClassDescriptor cd=(ClassDescriptor)classit.next();
1279             Set possiblematches=cd.getMethodTable().getSet(md.getSymbol());
1280             for(Iterator matchit=possiblematches.iterator();matchit.hasNext();) {
1281                 MethodDescriptor matchmd=(MethodDescriptor)matchit.next();
1282                 if (md.matches(matchmd))
1283                     return false;
1284             }
1285         }
1286         return true;
1287     }
1288
1289     private void generateFlatFieldNode(FlatMethod fm, FlatFieldNode ffn, PrintWriter output) {
1290         output.println(generateTemp(fm, ffn.getDst())+"="+ generateTemp(fm,ffn.getSrc())+"->"+ ffn.getField().getSafeSymbol()+";");
1291     }
1292
1293     private void generateFlatSetFieldNode(FlatMethod fm, FlatSetFieldNode fsfn, PrintWriter output) {
1294         if (fsfn.getField().getSymbol().equals("length")&&fsfn.getDst().getType().isArray())
1295             throw new Error("Can't set array length");
1296         output.println(generateTemp(fm, fsfn.getDst())+"->"+ fsfn.getField().getSafeSymbol()+"="+ generateTemp(fm,fsfn.getSrc())+";");
1297     }
1298
1299     private void generateFlatElementNode(FlatMethod fm, FlatElementNode fen, PrintWriter output) {
1300         TypeDescriptor elementtype=fen.getSrc().getType().dereference();
1301         String type="";
1302
1303         if (elementtype.isArray()||elementtype.isClass())
1304             type="void *";
1305         else 
1306             type=elementtype.getSafeSymbol()+" ";
1307
1308         if (fen.needsBoundsCheck()) {
1309             output.println("if ("+generateTemp(fm, fen.getIndex())+"< 0 || "+generateTemp(fm, fen.getIndex())+" >= "+generateTemp(fm,fen.getSrc()) + "->___length___)");
1310             output.println("failedboundschk();");
1311         }
1312
1313         output.println(generateTemp(fm, fen.getDst())+"=(("+ type+"*)(((char *) &("+ generateTemp(fm,fen.getSrc())+"->___length___))+sizeof(int)))["+generateTemp(fm, fen.getIndex())+"];");
1314     }
1315
1316     private void generateFlatSetElementNode(FlatMethod fm, FlatSetElementNode fsen, PrintWriter output) {
1317         //TODO: need dynamic check to make sure this assignment is actually legal
1318         //Because Object[] could actually be something more specific...ie. Integer[]
1319
1320         TypeDescriptor elementtype=fsen.getDst().getType().dereference();
1321         String type="";
1322
1323         if (elementtype.isArray()||elementtype.isClass())
1324             type="void *";
1325         else 
1326             type=elementtype.getSafeSymbol()+" ";
1327
1328         if (fsen.needsBoundsCheck()) {
1329             output.println("if ("+generateTemp(fm, fsen.getIndex())+"< 0 || "+generateTemp(fm, fsen.getIndex())+" >= "+generateTemp(fm,fsen.getDst()) + "->___length___)");
1330             output.println("failedboundschk();");
1331         }
1332
1333         output.println("(("+type +"*)(((char *) &("+ generateTemp(fm,fsen.getDst())+"->___length___))+sizeof(int)))["+generateTemp(fm, fsen.getIndex())+"]="+generateTemp(fm,fsen.getSrc())+";");
1334     }
1335
1336     private void generateFlatNew(FlatMethod fm, FlatNew fn, PrintWriter output) {
1337         if (fn.getType().isArray()) {
1338             int arrayid=state.getArrayNumber(fn.getType())+state.numClasses();
1339             if (GENERATEPRECISEGC) {
1340                 output.println(generateTemp(fm,fn.getDst())+"=allocate_newarray(&"+localsprefix+", "+arrayid+", "+generateTemp(fm, fn.getSize())+");");
1341             } else {
1342                 output.println(generateTemp(fm,fn.getDst())+"=allocate_newarray("+arrayid+", "+generateTemp(fm, fn.getSize())+");");
1343             }
1344         } else {
1345             if (GENERATEPRECISEGC) {
1346                 output.println(generateTemp(fm,fn.getDst())+"=allocate_new(&"+localsprefix+", "+fn.getType().getClassDesc().getId()+");");
1347             } else {
1348                 output.println(generateTemp(fm,fn.getDst())+"=allocate_new("+fn.getType().getClassDesc().getId()+");");
1349             }
1350         }
1351     }
1352
1353
1354     private void generateFlatTagDeclaration(FlatMethod fm, FlatTagDeclaration fn, PrintWriter output) {
1355         if (GENERATEPRECISEGC) {
1356             output.println(generateTemp(fm,fn.getDst())+"=allocate_tag(&"+localsprefix+", "+state.getTagId(fn.getType())+");");
1357         } else {
1358             output.println(generateTemp(fm,fn.getDst())+"=allocate_tag("+state.getTagId(fn.getType())+");");
1359         }
1360     }
1361
1362     private void generateFlatOpNode(FlatMethod fm, FlatOpNode fon, PrintWriter output) {
1363
1364         if (fon.getRight()!=null)
1365             output.println(generateTemp(fm, fon.getDest())+" = "+generateTemp(fm, fon.getLeft())+fon.getOp().toString()+generateTemp(fm,fon.getRight())+";");
1366         else if (fon.getOp().getOp()==Operation.ASSIGN)
1367             output.println(generateTemp(fm, fon.getDest())+" = "+generateTemp(fm, fon.getLeft())+";");
1368         else if (fon.getOp().getOp()==Operation.UNARYPLUS)
1369             output.println(generateTemp(fm, fon.getDest())+" = "+generateTemp(fm, fon.getLeft())+";");
1370         else if (fon.getOp().getOp()==Operation.UNARYMINUS)
1371             output.println(generateTemp(fm, fon.getDest())+" = -"+generateTemp(fm, fon.getLeft())+";");
1372         else if (fon.getOp().getOp()==Operation.LOGIC_NOT)
1373             output.println(generateTemp(fm, fon.getDest())+" = !"+generateTemp(fm, fon.getLeft())+";");
1374         else
1375             output.println(generateTemp(fm, fon.getDest())+fon.getOp().toString()+generateTemp(fm, fon.getLeft())+";");
1376     }
1377
1378     private void generateFlatCastNode(FlatMethod fm, FlatCastNode fcn, PrintWriter output) {
1379         /* TODO: Do type check here */
1380         if (fcn.getType().isArray()) {
1381             throw new Error();
1382         } else if (fcn.getType().isClass())
1383             output.println(generateTemp(fm,fcn.getDst())+"=(struct "+fcn.getType().getSafeSymbol()+" *)"+generateTemp(fm,fcn.getSrc())+";");
1384         else
1385             output.println(generateTemp(fm,fcn.getDst())+"=("+fcn.getType().getSafeSymbol()+")"+generateTemp(fm,fcn.getSrc())+";");
1386     }
1387
1388     private void generateFlatLiteralNode(FlatMethod fm, FlatLiteralNode fln, PrintWriter output) {
1389         if (fln.getValue()==null)
1390             output.println(generateTemp(fm, fln.getDst())+"=0;");
1391         else if (fln.getType().getSymbol().equals(TypeUtil.StringClass)) {
1392             if (GENERATEPRECISEGC) {
1393                 output.println(generateTemp(fm, fln.getDst())+"=NewString(&"+localsprefix+", \""+FlatLiteralNode.escapeString((String)fln.getValue())+"\","+((String)fln.getValue()).length()+");");
1394             } else {
1395                 output.println(generateTemp(fm, fln.getDst())+"=NewString(\""+FlatLiteralNode.escapeString((String)fln.getValue())+"\","+((String)fln.getValue()).length()+");");
1396             }
1397         } else if (fln.getType().isBoolean()) {
1398             if (((Boolean)fln.getValue()).booleanValue())
1399                 output.println(generateTemp(fm, fln.getDst())+"=1;");
1400             else
1401                 output.println(generateTemp(fm, fln.getDst())+"=0;");
1402         } else if (fln.getType().isChar()) {
1403             String st=FlatLiteralNode.escapeString(fln.getValue().toString());
1404             output.println(generateTemp(fm, fln.getDst())+"='"+st+"';");
1405         } else
1406             output.println(generateTemp(fm, fln.getDst())+"="+fln.getValue()+";");
1407     }
1408
1409     private void generateFlatReturnNode(FlatMethod fm, FlatReturnNode frn, PrintWriter output) {
1410         
1411         if (frn.getReturnTemp()!=null)
1412             output.println("return "+generateTemp(fm, frn.getReturnTemp())+";");
1413         else
1414             output.println("return;");
1415     }
1416
1417     private void generateFlatCondBranch(FlatMethod fm, FlatCondBranch fcb, String label, PrintWriter output) {
1418         output.println("if (!"+generateTemp(fm, fcb.getTest())+") goto "+label+";");
1419     }
1420
1421     private void generateHeader(FlatMethod fm, Descriptor des, PrintWriter output) {
1422         /* Print header */
1423         ParamsObject objectparams=(ParamsObject)paramstable.get(des);
1424         MethodDescriptor md=null;
1425         TaskDescriptor task=null;
1426         if (des instanceof MethodDescriptor)
1427             md=(MethodDescriptor) des;
1428         else
1429             task=(TaskDescriptor) des;
1430
1431         ClassDescriptor cn=md!=null?md.getClassDesc():null;
1432         
1433         if (md!=null&&md.getReturnType()!=null) {
1434             if (md.getReturnType().isClass()||md.getReturnType().isArray())
1435                 output.print("struct " + md.getReturnType().getSafeSymbol()+" * ");
1436             else
1437                 output.print(md.getReturnType().getSafeSymbol()+" ");
1438         } else 
1439             //catch the constructor case
1440             output.print("void ");
1441         if (md!=null)
1442             output.print(cn.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"(");
1443         else
1444             output.print(task.getSafeSymbol()+"(");
1445         
1446         boolean printcomma=false;
1447         if (GENERATEPRECISEGC) {
1448             if (md!=null)
1449                 output.print("struct "+cn.getSafeSymbol()+md.getSafeSymbol()+"_"+md.getSafeMethodDescriptor()+"_params * "+paramsprefix);
1450             else
1451                 output.print("struct "+task.getSafeSymbol()+"_params * "+paramsprefix);
1452             printcomma=true;
1453         }
1454
1455         if (md!=null) {
1456             /* Method */
1457             for(int i=0;i<objectparams.numPrimitives();i++) {
1458                 TempDescriptor temp=objectparams.getPrimitive(i);
1459                 if (printcomma)
1460                     output.print(", ");
1461                 printcomma=true;
1462                 if (temp.getType().isClass()||temp.getType().isArray())
1463                     output.print("struct "+temp.getType().getSafeSymbol()+" * "+temp.getSafeSymbol());
1464                 else
1465                     output.print(temp.getType().getSafeSymbol()+" "+temp.getSafeSymbol());
1466             }
1467             output.println(") {");
1468         } else if (!GENERATEPRECISEGC) {
1469             /* Imprecise Task */
1470             output.println("void * parameterarray[]) {");
1471             /* Unpack variables */
1472             for(int i=0;i<objectparams.numPrimitives();i++) {
1473                 TempDescriptor temp=objectparams.getPrimitive(i);
1474                 output.println("struct "+temp.getType().getSafeSymbol()+" * "+temp.getSafeSymbol()+"=parameterarray["+i+"];");
1475             }
1476             for(int i=0;i<fm.numTags();i++) {
1477                 TempDescriptor temp=fm.getTag(i);
1478                 int offset=i+objectparams.numPrimitives();
1479                 output.println("struct ___TagDescriptor___ * "+temp.getSafeSymbol()+"=parameterarray["+offset+"];");
1480             }
1481
1482             if ((objectparams.numPrimitives()+fm.numTags())>maxtaskparams)
1483                 maxtaskparams=objectparams.numPrimitives()+fm.numTags();
1484         } else output.println(") {");
1485     }
1486     
1487     public void generateFlatFlagActionNode(FlatMethod fm, FlatFlagActionNode ffan, PrintWriter output) {
1488         output.println("/* FlatFlagActionNode */");
1489
1490
1491         /* Process tag changes */
1492         Relation tagsettable=new Relation();
1493         Relation tagcleartable=new Relation();
1494
1495         Iterator tagsit=ffan.getTempTagPairs(); 
1496         while (tagsit.hasNext()) {
1497             TempTagPair ttp=(TempTagPair) tagsit.next();
1498             TempDescriptor objtmp=ttp.getTemp();
1499             TagDescriptor tag=ttp.getTag();
1500             TempDescriptor tagtmp=ttp.getTagTemp();
1501             boolean tagstatus=ffan.getTagChange(ttp);
1502             if (tagstatus) {
1503                 tagsettable.put(objtmp, tagtmp);
1504             } else {
1505                 tagcleartable.put(objtmp, tagtmp);
1506             }
1507         }
1508
1509
1510         Hashtable flagandtable=new Hashtable();
1511         Hashtable flagortable=new Hashtable();
1512
1513         /* Process flag changes */
1514         Iterator flagsit=ffan.getTempFlagPairs();
1515         while(flagsit.hasNext()) {
1516             TempFlagPair tfp=(TempFlagPair)flagsit.next();
1517             TempDescriptor temp=tfp.getTemp();
1518             Hashtable flagtable=(Hashtable)flagorder.get(temp.getType().getClassDesc());
1519             FlagDescriptor flag=tfp.getFlag();
1520             if (flag==null) {
1521                 //Newly allocate objects that don't set any flags case
1522                 if (flagortable.containsKey(temp)) {
1523                     throw new Error();
1524                 }
1525                 int mask=0;
1526                 flagortable.put(temp,new Integer(mask));
1527             } else {
1528                 int flagid=1<<((Integer)flagtable.get(flag)).intValue();
1529                 boolean flagstatus=ffan.getFlagChange(tfp);
1530                 if (flagstatus) {
1531                     int mask=0;
1532                     if (flagortable.containsKey(temp)) {
1533                         mask=((Integer)flagortable.get(temp)).intValue();
1534                     }
1535                     mask|=flagid;
1536                     flagortable.put(temp,new Integer(mask));
1537                 } else {
1538                     int mask=0xFFFFFFFF;
1539                     if (flagandtable.containsKey(temp)) {
1540                         mask=((Integer)flagandtable.get(temp)).intValue();
1541                     }
1542                     mask&=(0xFFFFFFFF^flagid);
1543                     flagandtable.put(temp,new Integer(mask));
1544                 }
1545             }
1546         }
1547
1548
1549         HashSet flagtagset=new HashSet();
1550         flagtagset.addAll(flagortable.keySet());
1551         flagtagset.addAll(flagandtable.keySet());
1552         flagtagset.addAll(tagsettable.keySet());
1553         flagtagset.addAll(tagcleartable.keySet());
1554
1555         Iterator ftit=flagtagset.iterator();
1556         while(ftit.hasNext()) {
1557             TempDescriptor temp=(TempDescriptor)ftit.next();
1558             
1559             
1560             Set tagtmps=tagcleartable.get(temp);
1561             if (tagtmps!=null) {
1562                 Iterator tagit=tagtmps.iterator();
1563                 while(tagit.hasNext()) {
1564                     TempDescriptor tagtmp=(TempDescriptor)tagit.next();
1565                     if (GENERATEPRECISEGC) 
1566                         output.println("tagclear(&"+localsprefix+", (struct ___Object___ *)"+generateTemp(fm, temp)+", "+generateTemp(fm,tagtmp)+");");
1567                     else
1568                         output.println("tagclear((struct ___Object___ *)"+generateTemp(fm, temp)+", "+generateTemp(fm,tagtmp)+");");
1569                 }
1570             }
1571
1572             tagtmps=tagsettable.get(temp);
1573             if (tagtmps!=null) {
1574                 Iterator tagit=tagtmps.iterator();
1575                 while(tagit.hasNext()) {
1576                     TempDescriptor tagtmp=(TempDescriptor)tagit.next();
1577                     if (GENERATEPRECISEGC)
1578                         output.println("tagset(&"+localsprefix+", (struct ___Object___ *)"+generateTemp(fm, temp)+", "+generateTemp(fm,tagtmp)+");");
1579                     else
1580                         output.println("tagset((struct ___Object___ *)"+generateTemp(fm, temp)+", "+generateTemp(fm,tagtmp)+");");
1581                 }
1582             }
1583
1584             int ormask=0;
1585             int andmask=0xFFFFFFF;
1586             
1587             if (flagortable.containsKey(temp))
1588                 ormask=((Integer)flagortable.get(temp)).intValue();
1589             if (flagandtable.containsKey(temp))
1590                 andmask=((Integer)flagandtable.get(temp)).intValue();
1591             if (ffan.getTaskType()==FlatFlagActionNode.NEWOBJECT) {
1592                 output.println("flagorandinit("+generateTemp(fm, temp)+", 0x"+Integer.toHexString(ormask)+", 0x"+Integer.toHexString(andmask)+");");
1593             } else {
1594                 output.println("flagorand("+generateTemp(fm, temp)+", 0x"+Integer.toHexString(ormask)+", 0x"+Integer.toHexString(andmask)+");");
1595             }
1596         }
1597     }
1598
1599      void generateOptionalArrays(PrintWriter output) {
1600
1601          SymbolTable classes = state.getClassSymbolTable();
1602          for( Iterator it = classes.getAllDescriptorsIterator(); it.hasNext();){
1603              ClassDescriptor cd = (ClassDescriptor)it.next();
1604              output.println("Class "+cd.getSymbol());
1605              Hashtable flags=(Hashtable)flagorder.get(cd);
1606              for (Iterator fit = cd.getFlags(); fit.hasNext();){
1607                  FlagDescriptor fd = (FlagDescriptor)fit.next();
1608                  int flagid=1<<((Integer)flags.get(fd)).intValue();
1609                  output.println("\tFlag associated with "+fd.getSymbol()+" : 0x"+Integer.toHexString(flagid)+" bx"+Integer.toBinaryString(flagid));
1610              }
1611          }
1612         
1613      }
1614
1615 }
1616          
1617
1618
1619
1620
1621