1 package org.simantics.scl.compiler.elaboration.query;
3 import java.util.ArrayList;
6 import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
7 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
8 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
9 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
10 import org.simantics.scl.compiler.elaboration.expressions.EError;
11 import org.simantics.scl.compiler.elaboration.expressions.Expression;
12 import org.simantics.scl.compiler.elaboration.expressions.QueryTransformer;
13 import org.simantics.scl.compiler.elaboration.expressions.StandardExpressionVisitor;
14 import org.simantics.scl.compiler.elaboration.expressions.Variable;
15 import org.simantics.scl.compiler.elaboration.expressions.VariableProcedure;
16 import org.simantics.scl.compiler.elaboration.expressions.printing.ExpressionToStringVisitor;
17 import org.simantics.scl.compiler.elaboration.expressions.visitors.CollectRefsVisitor;
18 import org.simantics.scl.compiler.elaboration.expressions.visitors.ForVariablesUsesVisitor;
19 import org.simantics.scl.compiler.elaboration.query.compilation.ConstraintCollectionContext;
20 import org.simantics.scl.compiler.elaboration.query.compilation.DerivateException;
21 import org.simantics.scl.compiler.elaboration.query.compilation.DynamicProgrammingOrdering;
22 import org.simantics.scl.compiler.elaboration.query.compilation.EnforcingContext;
23 import org.simantics.scl.compiler.elaboration.query.compilation.QueryCompilationContext;
24 import org.simantics.scl.compiler.elaboration.query.compilation.QueryConstraint;
25 import org.simantics.scl.compiler.elaboration.query.compilation.UnsolvableQueryException;
26 import org.simantics.scl.compiler.elaboration.relations.CompositeRelation;
27 import org.simantics.scl.compiler.elaboration.relations.LocalRelation;
28 import org.simantics.scl.compiler.elaboration.relations.SCLRelation;
29 import org.simantics.scl.compiler.internal.parsing.Symbol;
31 import gnu.trove.map.hash.THashMap;
32 import gnu.trove.map.hash.TIntObjectHashMap;
33 import gnu.trove.map.hash.TObjectIntHashMap;
34 import gnu.trove.set.hash.THashSet;
35 import gnu.trove.set.hash.TIntHashSet;
37 public abstract class Query extends Symbol {
38 public static final Query[] EMPTY_ARRAY = new Query[0];
40 public abstract void collectFreeVariables(THashSet<Variable> vars);
41 public abstract void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars);
42 public abstract void checkType(TypingContext context);
44 public Query resolve(TranslationContext context) {
45 throw new InternalCompilerError(location, getClass().getSimpleName() + " does not support resolve.");
48 public Expression generateEnforce(EnforcingContext context) {
49 context.getErrorLog().log(location, getClass().getSimpleName() + " does not support enforcing.");
50 return new EError(location);
53 public abstract void collectConstraints(ConstraintCollectionContext context) throws UnsolvableQueryException;
55 protected QueryConstraint[] getOrderedConstraints(QueryCompilationContext context) throws UnsolvableQueryException {
56 ConstraintCollectionContext collectionContext = new ConstraintCollectionContext(context);
57 collectConstraints(collectionContext);
58 QueryConstraint[] constraints = collectionContext.getConstraints();
59 DynamicProgrammingOrdering.order(collectionContext, constraints, 0L);
63 protected void applyConstraints(QueryCompilationContext context, QueryConstraint[] constraints) {
64 for(int i=constraints.length-1;i>=0;--i)
65 constraints[i].generateAndUpdateCost(context);
68 public void generate(QueryCompilationContext context) throws UnsolvableQueryException {
69 applyConstraints(context, getOrderedConstraints(context));
72 protected static final TIntObjectHashMap<ArrayList<Query>> NO_DERIVATE = new TIntObjectHashMap<ArrayList<Query>>();
74 public static class Diffable {
77 public Variable[] parameters;
78 public Diffable(int id, SCLRelation relation, Variable[] parameters) {
80 this.relation = relation;
81 this.parameters = parameters;
85 public static class Diff {
89 public Diff(int id, Query query) {
95 public static final Diff[] NO_DIFF = new Diff[0];
97 public Diff[] derivate(THashMap<LocalRelation, Diffable> diffables) throws DerivateException {
98 throw new DerivateException(location);
101 public static final QDisjunction EMPTY_QUERY = new QDisjunction();
103 public Query removeRelations(Set<SCLRelation> relations) {
104 throw new UnsupportedOperationException();
107 public Query copy() {
108 return replace(new ReplaceContext(null));
111 public Query copy(TypingContext context) {
112 return replace(new ReplaceContext(context));
115 public abstract Query replace(ReplaceContext context);
117 public abstract void setLocationDeep(long loc);
119 public abstract void accept(QueryVisitor visitor);
121 public void collectRelationRefs(
122 final TObjectIntHashMap<SCLRelation> allRefs, final TIntHashSet refs) {
123 accept(new StandardExpressionVisitor() {
125 public void visit(QAtom query) {
126 visit(query.relation);
128 private void visit(SCLRelation relation) {
129 int id = allRefs.get(relation);
132 else if(relation instanceof CompositeRelation)
133 for(SCLRelation subrelation : ((CompositeRelation)relation).getSubrelations())
139 public TIntObjectHashMap<ArrayList<Query>> splitToPhases() {
140 TIntObjectHashMap<ArrayList<Query>> result = new TIntObjectHashMap<ArrayList<Query>>(2);
141 splitToPhases(result);
145 public void splitToPhases(TIntObjectHashMap<ArrayList<Query>> result) {
146 throw new UnsupportedOperationException(getClass().getSimpleName() + " does not support splitToPhases.");
150 public String toString() {
151 StringBuilder b = new StringBuilder();
152 ExpressionToStringVisitor visitor = new ExpressionToStringVisitor(b);
157 public abstract Query accept(QueryTransformer transformer);
159 public void forVariables(VariableProcedure procedure) {
160 accept(new ForVariablesUsesVisitor(procedure));
162 public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
163 accept(new CollectRefsVisitor(allRefs, refs));