Merged changes from feature/scl to master.

[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / elaboration / chr / CHRLiteral.java

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/CHRLiteral.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/CHRLiteral.java
new file mode 100644 (file)
index 0000000..597ee9d
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/CHRLiteral.java
@@ -0,0 +1,138 @@
+package org.simantics.scl.compiler.elaboration.chr;\r
+\r
+import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;\r
+import org.simantics.scl.compiler.elaboration.chr.relations.CHRConstraint;\r
+import org.simantics.scl.compiler.elaboration.chr.relations.ExternalCHRRelation;\r
+import org.simantics.scl.compiler.elaboration.chr.relations.SpecialCHRRelation;\r
+import org.simantics.scl.compiler.elaboration.chr.relations.UnresolvedCHRRelation;\r
+import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.TypingContext;\r
+import org.simantics.scl.compiler.elaboration.expressions.Expression;\r
+import org.simantics.scl.compiler.elaboration.expressions.Variable;\r
+import org.simantics.scl.compiler.elaboration.expressions.VariableProcedure;\r
+import org.simantics.scl.compiler.elaboration.expressions.printing.ExpressionToStringVisitor;\r
+import org.simantics.scl.compiler.elaboration.relations.SCLRelation;\r
+import org.simantics.scl.compiler.errors.Locations;\r
+import org.simantics.scl.compiler.internal.parsing.Symbol;\r
+import org.simantics.scl.compiler.types.TVar;\r
+import org.simantics.scl.compiler.types.Type;\r
+import org.simantics.scl.compiler.types.Types;\r
+import org.simantics.scl.compiler.types.kinds.Kinds;\r
+\r
+import gnu.trove.map.hash.TObjectIntHashMap;\r
+import gnu.trove.set.hash.THashSet;\r
+import gnu.trove.set.hash.TIntHashSet;\r
+\r
+public class CHRLiteral extends Symbol {\r
+    \r
+    public CHRRelation relation;\r
+    public Type[] typeParameters;\r
+    public Expression[] parameters;\r
+    public boolean killAfterMatch;\r
+    public boolean negated;\r
+    public boolean passive = true;\r
+    \r
+    public CHRLiteral(long location, CHRRelation relation, Expression[] parameters, boolean remove, boolean negated) {\r
+        this.location = location;\r
+        this.relation = relation;\r
+        this.parameters = parameters;\r
+        this.killAfterMatch = remove;\r
+        this.negated = negated;\r
+    }\r
+\r
+    public void resolve(TranslationContext context) {\r
+        if(relation instanceof UnresolvedCHRRelation) {\r
+            UnresolvedCHRRelation unresolved = (UnresolvedCHRRelation)relation;\r
+            CHRConstraint constraint = context.resolveCHRConstraint(unresolved.name);\r
+            if(constraint != null) {\r
+                relation = constraint;\r
+                passive = false;\r
+            }\r
+            else {\r
+                SCLRelation sclRelation = context.resolveRelation(unresolved.location, unresolved.name);\r
+                if(sclRelation != null)\r
+                    relation = new ExternalCHRRelation(sclRelation);\r
+                else {\r
+                    Type[] parameterTypes = new Type[parameters.length];\r
+                    for(int i=0;i<parameterTypes.length;++i)\r
+                        parameterTypes[i] = Types.metaVar(Kinds.STAR);\r
+                    constraint = new CHRConstraint(location, unresolved.name, parameterTypes);\r
+                    constraint.implicitlyDeclared = true;\r
+                    context.newCHRConstraint(constraint.name, constraint);\r
+                    relation = constraint;\r
+                    passive = false;\r
+                    //context.getErrorLog().log(unresolved.location, "Couldn't resolve constraint " + unresolved.name + ".");\r
+                }\r
+            }\r
+        }\r
+        for(int i=0;i<parameters.length;++i)\r
+            parameters[i] = parameters[i].resolve(context);\r
+    }\r
+\r
+    public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {\r
+        for(Expression parameter : parameters)\r
+            parameter.collectRefs(allRefs, refs);\r
+    }\r
+\r
+    public void checkType(TypingContext context) {\r
+        if(relation == SpecialCHRRelation.EXECUTE) {\r
+            if(parameters.length != 1)\r
+                throw new InternalCompilerError("Wrong number of parameters for EXECUTE constraint.");\r
+            parameters[0] = parameters[0].checkIgnoredType(context);\r
+        }\r
+        else {\r
+            TVar[] typeVariables = relation.getTypeVariables();\r
+            typeParameters = typeVariables.length == 0 ? Type.EMPTY_ARRAY : new Type[typeVariables.length];\r
+            for(int i=0;i<typeVariables.length;++i)\r
+                typeParameters[i] = Types.metaVar(typeVariables[i].getKind());\r
+            Type[] parameterTypes = Types.replace(relation.getParameterTypes(), typeVariables, typeParameters);\r
+            if(parameterTypes.length != parameters.length)\r
+                context.getErrorLog().log(location, "Constraint is applied with wrong number of parameters");\r
+            else\r
+                for(int i=0;i<parameters.length;++i)\r
+                    parameters[i] = parameters[i].checkType(context, parameterTypes[i]);\r
+        }\r
+    }\r
+\r
+    public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {\r
+        for(Expression parameter : parameters)\r
+            parameter.collectVars(allVars, vars);\r
+    }\r
+\r
+    public void forVariables(VariableProcedure procedure) {\r
+        for(Expression parameter : parameters)\r
+            parameter.forVariables(procedure);\r
+    }\r
+\r
+    public void collectFreeVariables(THashSet<Variable> vars) {\r
+        for(Expression parameter : parameters)\r
+            parameter.collectFreeVariables(vars);\r
+    }\r
+\r
+    public void setLocationDeep(long loc) {\r
+        if(location == Locations.NO_LOCATION) {\r
+            this.location = loc;\r
+            for(Expression parameter : parameters)\r
+                parameter.setLocationDeep(loc);\r
+        }\r
+    }\r
+    \r
+    public void simplify(SimplificationContext context) {\r
+        for(int i=0;i<parameters.length;++i)\r
+            parameters[i] = parameters[i].simplify(context);\r
+    }\r
+    \r
+    public String toString() {\r
+        StringBuilder b = new StringBuilder();\r
+        ExpressionToStringVisitor visitor = new ExpressionToStringVisitor(b);\r
+        visitor.visit(this);\r
+        return b.toString();\r
+    }\r
+\r
+    public void collectQueryEffects(THashSet<Type> effects) {\r
+    }\r
+\r
+    public void collectEnforceEffects(THashSet<Type> effects) {\r
+    }\r
+}\r