64c336d651cc94b35f24e912f91ccb3de8733488
[repair.git] / Repair / RepairCompiler / MCC / IR / SumExpr.java
1 package MCC.IR;
2
3 import java.util.*;
4
5 public class SumExpr extends Expr {
6
7     SetDescriptor sd;
8     RelationDescriptor rd;
9
10
11     public SumExpr(SetDescriptor sd, RelationDescriptor rd) {
12         if (sd == null||rd==null) {
13             throw new NullPointerException();
14         }
15         this.sd=sd;
16         this.rd=rd;
17     }
18
19     public String name() {
20         return "sum("+sd.toString()+"."+rd.toString()+")";
21     }
22
23     public boolean equals(Map remap, Expr e) {
24         if (e==null||!(e instanceof SumExpr))
25             return false;
26         SumExpr se=(SumExpr)e;
27         return (se.sd==sd)&&(se.rd==rd);
28     }
29
30     public boolean usesDescriptor(Descriptor d) {
31         return (sd==d)||(rd==d);
32     }
33
34     public Set useDescriptor(Descriptor d) {
35         HashSet newset=new HashSet();
36         if ((d==sd)||(d==rd))
37             newset.add(this);
38         return newset;
39     }
40
41     public Descriptor getDescriptor() {
42         throw new Error("Sum shouldn't appear on left hand side!");
43     }
44
45     public boolean inverted() {
46         return false;
47     }
48
49     public Set getRequiredDescriptors() {
50         HashSet v=new HashSet();
51         v.add(sd);
52         v.add(rd);
53         return v;
54     }
55
56     public void generate(CodeWriter writer, VarDescriptor dest) {
57         writer.addDeclaration("int",dest.getSafeSymbol());
58         writer.outputline(dest.getSafeSymbol()+"=0;");
59
60
61         VarDescriptor itvd=VarDescriptor.makeNew("iterator");
62         writer.addDeclaration("struct SimpleIterator",itvd.getSafeSymbol());
63         writer.outputline("SimpleHashiterator("+sd.getSafeSymbol()+"_hash , &"+itvd.getSafeSymbol()+");");
64         writer.outputline("while (hasNext(&"+itvd.getSafeSymbol()+"))");
65         writer.startblock();
66         {
67             VarDescriptor keyvd=VarDescriptor.makeNew("key");
68             writer.addDeclaration("int",keyvd.getSafeSymbol());
69             writer.outputline(keyvd.getSafeSymbol()+"=next(&"+itvd.getSafeSymbol()+");");
70             VarDescriptor tmpvar=VarDescriptor.makeNew("tmp");
71             writer.addDeclaration("int",tmpvar.getSafeSymbol());
72
73             VarDescriptor newset=VarDescriptor.makeNew("newset");
74             writer.addDeclaration("struct SimpleHash *",newset.getSafeSymbol());
75             writer.outputline(newset.getSafeSymbol()+"=SimpleHashimageSet("+rd.getSafeSymbol()+"_hash, "+keyvd.getSafeSymbol()+");");
76
77             VarDescriptor itvd2=VarDescriptor.makeNew("iterator");
78             writer.addDeclaration("struct SimpleIterator",itvd2.getSafeSymbol());
79             writer.outputline("SimpleHashiterator("+newset.getSafeSymbol()+", &"+itvd2.getSafeSymbol()+");");
80
81             writer.outputline("while (hasNext(&"+itvd2.getSafeSymbol()+"))");
82             writer.startblock();
83             {
84                 VarDescriptor keyvd2=VarDescriptor.makeNew("keyinner");
85                 writer.outputline(dest.getSafeSymbol()+"+=next(&"+itvd2.getSafeSymbol()+");");
86                 writer.endblock();
87             }
88             writer.outputline("freeSimpleHash("+newset.getSafeSymbol()+");");
89             writer.endblock();
90         }
91     }
92
93     public void prettyPrint(PrettyPrinter pp) {
94         pp.output("sum(");
95         pp.output(sd.toString());
96         pp.output(".");
97         pp.output(rd.toString());
98         pp.output(")");
99     }
100
101     public TypeDescriptor typecheck(SemanticAnalyzer sa) {
102         this.td = ReservedTypeDescriptor.INT;
103         return this.td;
104     }
105
106     public Set getInversedRelations() {
107         return new HashSet();
108     }
109
110 }