Repair/RepairCompiler/MCC/IR/RelationExpr.java

   1 package MCC.IR;
   2
   3 import java.util.*;
   4
   5 public class RelationExpr extends Expr {
   6
   7     Expr expr;
   8     RelationDescriptor relation;
   9     boolean inverse;
  10
  11     public Set getfunctions() {
  12         HashSet functions=new HashSet();
  13         Set exprfunctions=expr.getfunctions();
  14         if (exprfunctions!=null)
  15             functions.addAll(exprfunctions);
  16         functions.add(new ConstraintDependence.Function(relation,expr.getSet(),inverse,this));
  17
  18         return functions;
  19     }
  20
  21     public SetDescriptor getSet() {
  22         if (inverse)
  23             return relation.getDomain();
  24         else
  25             return relation.getRange();
  26     }
  27
  28     public RelationExpr(Expr expr, RelationDescriptor relation, boolean inverse) {
  29         this.expr = expr;
  30         this.relation = relation;
  31         this.inverse = inverse;
  32     }
  33
  34     public Set freeVars() {
  35         return expr.freeVars();
  36     }
  37
  38     public String name() {
  39         String name=expr.name()+".";
  40         if (inverse)
  41             name+="~";
  42         name+=relation.toString();
  43         return name;
  44     }
  45
  46     public Expr getExpr() {
  47         return expr;
  48     }
  49
  50     public boolean equals(Map remap, Expr e) {
  51         if (e==null||!(e instanceof RelationExpr))
  52             return false;
  53         RelationExpr re=(RelationExpr)e;
  54         if (re.relation!=relation)
  55             return false;
  56         if (!expr.equals(remap,re.expr))
  57             return false;
  58         if (inverse!=re.inverse)
  59             return false;
  60         return true;
  61     }
  62
  63     public boolean usesDescriptor(Descriptor rd) {
  64         if (rd==relation)
  65             return true;
  66         else
  67             return expr.usesDescriptor(rd);
  68     }
  69
  70     public Set useDescriptor(Descriptor rd) {
  71         HashSet newset=new HashSet();
  72         if (rd==relation)
  73             newset.add(this);
  74         newset.addAll(expr.useDescriptor(rd));
  75         return newset;
  76     }
  77
  78     public RelationDescriptor getRelation() {
  79         return relation;
  80     }
  81
  82     public Descriptor getDescriptor() {
  83         return relation;
  84     }
  85
  86     public boolean inverted() {
  87         return inverse;
  88     }
  89
  90     public Set getInversedRelations() {
  91         Set set = expr.getInversedRelations();
  92         if (inverse) {
  93             set.add(relation);
  94         }
  95         return set;
  96     }
  97
  98     public Set getRequiredDescriptors() {
  99         Set v = expr.getRequiredDescriptors();
 100         v.add(relation);
 101         return v;
 102     }
 103
 104     public void generate(CodeWriter writer, VarDescriptor dest) {
 105         VarDescriptor domain = VarDescriptor.makeNew("domain");
 106         String strinverse = inverse ? "inv" : "";
 107         String found = (VarDescriptor.makeNew("found")).getSafeSymbol();
 108         expr.generate(writer, domain);
 109         writer.addDeclaration(relation.getRange().getType().getGenerateType().getSafeSymbol(), dest.getSafeSymbol());
 110         writer.addDeclaration("int",found);
 111         writer.outputline(found+" = SimpleHashget(" +relation.getSafeSymbol()+"_hash"+strinverse+", "+ domain.getSafeSymbol() + ", & " + dest.getSafeSymbol() + ");");
 112         writer.outputline("if (!" + found + ") { maybe = 1; }");
 113     }
 114
 115     public void prettyPrint(PrettyPrinter pp) {
 116         expr.prettyPrint(pp);
 117         pp.output(".");
 118
 119         if (inverse) {
 120             pp.output("~");
 121         }
 122
 123         pp.output(relation.getSafeSymbol());
 124     }
 125
 126     public TypeDescriptor typecheck(SemanticAnalyzer sa) {
 127
 128         TypeDescriptor type = expr.typecheck(sa);
 129
 130         if (type == null) {
 131             return null;
 132         }
 133
 134         /* check to make sure that the types of the relation match up */
 135         if (inverse) {
 136             TypeDescriptor rangetype = relation.getRange().getType();
 137
 138             if (rangetype != type) {
 139                 sa.getErrorReporter().report(null, "Type of left side of relation operator '.' is '" + type.getSymbol() +
 140                                     "' but must be '" + rangetype.getSymbol() +
 141                                     "', the type of the range of the relation '" + relation.getSymbol() + "'");
 142                 return null;
 143             }
 144
 145             this.td = relation.getDomain().getType();
 146             return this.td;
 147         } else { /* not inverse */
 148             TypeDescriptor domaintype = relation.getDomain().getType();
 149
 150             if (domaintype != type) {
 151                 sa.getErrorReporter().report(null, "Type of left side of relation operator '.' is '" + type.getSymbol() +
 152                                     "' but must be '" + domaintype.getSymbol() +
 153                                     "', the type of the domain of the relation '" + relation.getSymbol() + "'");
 154                 return null;
 155             }
 156
 157             this.td = relation.getRange().getType();
 158             return this.td;
 159         }
 160     }
 161
 162 }