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));
  17
  18         return functions;
  19     }
  20
  21     public SetDescriptor getSet() {
  22         if (inverse)
  23             return relation.domain;
  24         else
  25             return relation.range;
  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 RelationDescriptor getRelation() {
  71         return relation;
  72     }
  73
  74     public Descriptor getDescriptor() {
  75         return relation;
  76     }
  77
  78     public boolean inverted() {
  79         return inverse;
  80     }
  81
  82     public Set getInversedRelations() {
  83         Set set = expr.getInversedRelations();
  84         if (inverse) {
  85             set.add(relation);
  86         }
  87         return set;
  88     }
  89
  90     public Set getRequiredDescriptors() {
  91         Set v = expr.getRequiredDescriptors();
  92         v.add(relation);
  93         return v;
  94     }
  95
  96     public void generate(CodeWriter writer, VarDescriptor dest) {
  97         VarDescriptor domain = VarDescriptor.makeNew("domain");
  98         String strinverse = inverse ? "inv" : "";
  99         String found = (VarDescriptor.makeNew("found")).getSafeSymbol();
 100         expr.generate(writer, domain);
 101         writer.outputline(relation.getRange().getType().getGenerateType().getSafeSymbol() + " " + dest.getSafeSymbol() + ";");
 102         writer.outputline("int " + found + " = " + relation.getSafeSymbol() + "_hash" + strinverse + "->get(" + domain.getSafeSymbol() + ", " + dest.getSafeSymbol() + ");");
 103         writer.outputline("if (!" + found + ") { maybe = 1; }");
 104     }
 105
 106     // #TBD#: don't think this method is needed (or even called/referenced)
 107     /*
 108       public void generate_set(CodeWriter writer, VarDescriptor dest) {
 109       VarDescriptor domain = VarDescriptor.makeNew("domain");
 110       expr.generate(writer, domain);
 111       writer.outputline(relation.getRange().getType().getGenerateType().getSafeSymbol() + " " + dest.getSafeSymbol() + " = " + relation.getSafeSymbol() + "_hash->get(" + domain.getSafeSymbol() + ");");
 112       }
 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 }