1 package org.simantics.scl.compiler.elaboration.chr;
3 import java.util.ArrayList;
5 import org.cojen.classfile.TypeDesc;
6 import org.simantics.scl.compiler.compilation.CompilationContext;
7 import org.simantics.scl.compiler.constants.BooleanConstant;
8 import org.simantics.scl.compiler.constants.Constant;
9 import org.simantics.scl.compiler.constants.IntegerConstant;
10 import org.simantics.scl.compiler.constants.JavaConstructor;
11 import org.simantics.scl.compiler.constants.JavaMethod;
12 import org.simantics.scl.compiler.constants.generic.CallJava;
13 import org.simantics.scl.compiler.constants.generic.MethodRef.FieldRef;
14 import org.simantics.scl.compiler.constants.generic.MethodRef.SetFieldRef;
15 import org.simantics.scl.compiler.elaboration.chr.analysis.UsageAnalysis;
16 import org.simantics.scl.compiler.elaboration.chr.plan.CHRSearchPlan;
17 import org.simantics.scl.compiler.elaboration.chr.plan.PlanRealizer;
18 import org.simantics.scl.compiler.elaboration.chr.relations.CHRConstraint;
19 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
20 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
21 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
22 import org.simantics.scl.compiler.elaboration.expressions.Variable;
23 import org.simantics.scl.compiler.elaboration.expressions.VariableProcedure;
24 import org.simantics.scl.compiler.errors.Locations;
25 import org.simantics.scl.compiler.internal.codegen.chr.CHRCodeGenerationConstants;
26 import org.simantics.scl.compiler.internal.codegen.chr.CHRRuntimeRulesetCodeGenerator;
27 import org.simantics.scl.compiler.internal.codegen.references.BoundVar;
28 import org.simantics.scl.compiler.internal.codegen.references.IVal;
29 import org.simantics.scl.compiler.internal.codegen.types.StandardTypeConstructor;
30 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;
31 import org.simantics.scl.compiler.internal.parsing.Symbol;
32 import org.simantics.scl.compiler.types.TCon;
33 import org.simantics.scl.compiler.types.TVar;
34 import org.simantics.scl.compiler.types.Type;
35 import org.simantics.scl.compiler.types.Types;
37 import gnu.trove.map.hash.TObjectIntHashMap;
38 import gnu.trove.set.hash.THashSet;
39 import gnu.trove.set.hash.TIntHashSet;
41 public class CHRRuleset extends Symbol {
43 public static final String INIT_CONSTRAINT = "__INIT__";
45 public ArrayList<CHRConstraint> constraints = new ArrayList<CHRConstraint>();
46 public ArrayList<CHRRule> rules = new ArrayList<CHRRule>();
48 public CHRConstraint initConstraint;
49 public int priorityCount;
51 public String runtimeRulesetName;
52 public TCon runtimeRulesetType;
53 public BoundVar runtimeRulesetVariable;
54 public TypeDesc runtimeRulesetTypeDesc;
55 public Constant readCurrentId;
56 public Constant writeCurrentId;
58 // FIXME remove and change the parameter of Expression.toVal
59 private CompilationContext cachedContext;
61 // For code generation
62 public BoundVar this_;
63 public BoundVar[] parameters;
64 public TypeDesc[] parameterTypeDescs;
67 initConstraint = new CHRConstraint(Locations.NO_LOCATION, INIT_CONSTRAINT, Type.EMPTY_ARRAY);
68 constraints.add(initConstraint);
71 public void resolve(TranslationContext context) {
72 for(CHRConstraint constraint : constraints)
73 context.newCHRConstraint(constraint.name, constraint);
75 for(CHRRule rule : rules) {
76 rule.resolve(context);
77 rule.priority = priorityCount++;
79 /*for(CHRConstraint constraint : constraints) {
80 Variable newVariable = context.newVariable("claim" + constraint.factClassName);
84 public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
85 for(CHRRule rule : rules)
86 rule.collectRefs(allRefs, refs);
89 public void checkType(TypingContext context) {
90 for(CHRRule rule : rules)
91 rule.checkType(context);
94 public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
95 for(CHRRule rule : rules)
96 rule.collectVars(allVars, vars);
99 public void forVariables(VariableProcedure procedure) {
100 for(CHRRule rule : rules)
101 rule.forVariables(procedure);
104 public void collectFreeVariables(THashSet<Variable> vars) {
105 for(CHRRule rule : rules)
106 rule.collectFreeVariables(vars);
109 public void setLocationDeep(long loc) {
110 if(location == Locations.NO_LOCATION) {
112 for(CHRRule rule : rules)
113 rule.setLocationDeep(loc);
117 public void compile(SimplificationContext context) {
118 initializeCodeGeneration(context.getCompilationContext());
119 UsageAnalysis.analyzeUsage(this);
120 for(CHRRule rule : rules)
121 rule.compile(context.getCompilationContext(), initConstraint);
122 // remove init constraint if it is not useful
123 if(initConstraint.minimumPriority == Integer.MAX_VALUE) {
124 constraints.remove(0);
125 initConstraint = null;
129 public void simplify(SimplificationContext context) {
130 for(CHRRule rule : rules)
131 rule.simplify(context);
134 public void initializeCodeGeneration(CompilationContext context) {
135 cachedContext = context; // FIXME remove
137 String suffix = context.namingPolicy.getFreshClosureClassNameSuffix();
138 runtimeRulesetType = Types.con(context.namingPolicy.getModuleName(), "CHR" + suffix);
139 runtimeRulesetName = context.namingPolicy.getModuleClassName() + suffix;
140 runtimeRulesetTypeDesc = TypeDesc.forClass(runtimeRulesetName);
141 runtimeRulesetVariable = new BoundVar(runtimeRulesetType);
142 for(CHRConstraint constraint : constraints)
143 constraint.initializeCodeGeneration(context, this);
144 readCurrentId = new CallJava(TVar.EMPTY_ARRAY, Types.PROC, Types.INTEGER, new Type[] {Types.CHRContext},
145 null, new FieldRef(CHRCodeGenerationConstants.CHRContext_name, "currentId", CHRRuntimeRulesetCodeGenerator.FACT_ID_TYPE), null);
146 writeCurrentId = new CallJava(TVar.EMPTY_ARRAY, Types.PROC, Types.UNIT, new Type[] {Types.CHRContext, Types.INTEGER},
147 null, new SetFieldRef(CHRCodeGenerationConstants.CHRContext_name, "currentId", CHRRuntimeRulesetCodeGenerator.FACT_ID_TYPE), null);
148 if(context.module != null) // for unit testing
149 context.module.addTypeDescriptor(runtimeRulesetType.name, new StandardTypeConstructor(runtimeRulesetType, TVar.EMPTY_ARRAY, runtimeRulesetTypeDesc));
152 public static final Constant ACTIVATE = new JavaMethod(true, CHRCodeGenerationConstants.CHRContext_name, "activate", Types.PROC, Types.UNIT, Types.CHRContext, Types.INTEGER);
153 private static final Constant CREATE_CHR_CONTEXT = new JavaConstructor("org/simantics/scl/runtime/chr/CHRContext", Types.PROC, Types.CHRContext);
155 public void generateCode(CodeWriter w) {
156 CHRRulesetObject object = new CHRRulesetObject(runtimeRulesetVariable, this);
157 w.defineObject(object);
158 for(CHRRule rule : rules) {
159 for(CHRSearchPlan plan : rule.plans) {
160 /*System.out.println(" plan " + plan.priority);
161 for(PlanOp planOp : plan.ops)
162 System.out.println(" " + planOp);*/
163 CodeWriter methodWriter = object.createMethod(w.getModuleWriter(), TVar.EMPTY_ARRAY, Types.PROC, Types.BOOLEAN, new Type[] {Types.CHRContext, plan.constraint.factType});
164 plan.implementation = methodWriter.getFunction();
165 IVal[] implementationParameters = methodWriter.getParameters();
166 plan.activeFact.setVal(implementationParameters[1]);
167 PlanRealizer realizer = new PlanRealizer(cachedContext, this, runtimeRulesetVariable, implementationParameters[0], plan.ops);
168 realizer.nextOp(methodWriter);
169 if(methodWriter.isUnfinished())
170 methodWriter.return_(BooleanConstant.TRUE);
173 if(initConstraint != null) {
174 IVal chrContext = w.apply(location, CREATE_CHR_CONTEXT);
175 IVal initFact = w.apply(location, initConstraint.constructor, IntegerConstant.ZERO);
176 w.apply(location, initConstraint.addProcedure, runtimeRulesetVariable, chrContext, initFact);
177 w.apply(location, ACTIVATE, chrContext, new IntegerConstant(Integer.MAX_VALUE));
181 public void collectEffects(THashSet<Type> effects) {
182 for(CHRRule rule : rules) {
183 for(CHRLiteral literal : rule.head.literals)
184 literal.collectQueryEffects(effects);
185 for(CHRLiteral literal : rule.head.literals)
186 literal.collectEnforceEffects(effects);