X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.scl.compiler%2Fsrc%2Forg%2Fsimantics%2Fscl%2Fcompiler%2Felaboration%2Fjava%2FMemberRelation.java;fp=bundles%2Forg.simantics.scl.compiler%2Fsrc%2Forg%2Fsimantics%2Fscl%2Fcompiler%2Felaboration%2Fjava%2FMemberRelation.java;h=6c22babec4264fcd5b8cd290e65eae96c2fce294;hb=969bd23cab98a79ca9101af33334000879fb60c5;hp=0000000000000000000000000000000000000000;hpb=866dba5cd5a3929bbeae85991796acb212338a08;p=simantics%2Fplatform.git

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/java/MemberRelation.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/java/MemberRelation.java
new file mode 100644
index 000000000..6c22babec
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/java/MemberRelation.java
@@ -0,0 +1,84 @@
+package org.simantics.scl.compiler.elaboration.java;
+
+import org.simantics.scl.compiler.common.names.Name;
+import org.simantics.scl.compiler.elaboration.expressions.EApply;
+import org.simantics.scl.compiler.elaboration.expressions.EVariable;
+import org.simantics.scl.compiler.elaboration.expressions.Expression;
+import org.simantics.scl.compiler.elaboration.expressions.Variable;
+import org.simantics.scl.compiler.elaboration.query.compilation.QueryCompilationContext;
+import org.simantics.scl.compiler.elaboration.relations.AbstractRelation;
+import org.simantics.scl.compiler.types.TVar;
+import org.simantics.scl.compiler.types.Type;
+import org.simantics.scl.compiler.types.Types;
+import org.simantics.scl.compiler.types.kinds.Kinds;
+
+public class MemberRelation extends AbstractRelation {
+    public static final MemberRelation INSTANCE = new MemberRelation();
+    
+    private MemberRelation() {}
+    
+    private static final TVar A = Types.var(Kinds.STAR);
+    
+    private static final TVar[] TYPE_VARIABLES = new TVar[] {A};
+    
+    @Override
+    public TVar[] getTypeVariables() {
+        return TYPE_VARIABLES;
+    }
+    
+    private static final Type[] PARAMETER_TYPES = new Type[] {A, Types.list(A)};
+    
+    @Override
+    public Type[] getParameterTypes() {
+        return PARAMETER_TYPES;
+    }
+    
+    @Override
+    public int getPhase() {
+        return 0;
+    }
+
+    @Override
+    public double getSelectivity(int boundVariables) {
+        switch(boundVariables) {
+        case FF: 
+        case BF: return Double.POSITIVE_INFINITY;
+        case FB: return 10.0;
+        case BB: return 0.5;
+        default: throw new IllegalArgumentException();
+        }
+    }
+    
+    @Override
+    public int getRequiredVariablesMask() {
+        return FB;
+    }
+
+    @Override
+    public void generate(long location,
+            QueryCompilationContext context,
+            Type[] typeParameters, Variable[] parameters, int boundVariables) {
+        switch(boundVariables) {
+        case FB:
+            context.iterateList(parameters[0], new EVariable(parameters[1]));
+            break;
+        case BB: 
+            context.condition(
+                    new EApply(
+                            context.getConstant(Name.create("Prelude", "elem"), typeParameters),
+                            new Expression[] {
+                                context.getEvidence(location, Types.pred(Types.EQ, typeParameters[0])),
+                                new EVariable(parameters[0]),
+                                new EVariable(parameters[1])
+                            }
+                            ));
+            break;
+        default: throw new IllegalArgumentException();
+        }
+    }
+
+    @Override
+    public String toString() {
+        return "<-";
+    }
+}