--- /dev/null
+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