Updates

[repair.git] / Repair / RepairCompiler / MCC / IR / InclusionPredicate.java

diff --git a/Repair/RepairCompiler/MCC/IR/InclusionPredicate.java b/Repair/RepairCompiler/MCC/IR/InclusionPredicate.java
index fd4d8ff9e6d869f987bbb5391c9a964f832dfb97..26b9f4a2f8b412c7fbd930bf951476b14b606123 100755 (executable)
--- a/Repair/RepairCompiler/MCC/IR/InclusionPredicate.java
+++ b/Repair/RepairCompiler/MCC/IR/InclusionPredicate.java
@@ -40,6 +40,19 @@ public class InclusionPredicate extends Predicate {
         //set.addAll(expr.getInversedRelations());
         //return set;
     }
-            
+
+    public int[] getRepairs(boolean negated) {
+       if (setexpr instanceof ImageSetExpr) {
+           if (negated)
+               return new int[] {AbstractRepair.REMOVEFROMRELATION};
+           else
+               return new int[] {AbstractRepair.ADDTORELATION};
+       } else {
+           if (negated)
+               return new int[] {AbstractRepair.REMOVEFROMSET};
+           else
+               return new int[] {AbstractRepair.ADDTOSET};
+       }
+    }
 }