]> gerrit.simantics Code Review - simantics/platform.git/blob
04f6791ac32b55aa238bfd109e7803a8f5b67b0e
[simantics/platform.git] /
1 package org.simantics.scl.compiler.elaboration.expressions.visitors;
2
3 import org.simantics.scl.compiler.elaboration.chr.CHRLiteral;
4 import org.simantics.scl.compiler.elaboration.chr.CHRQuery;
5 import org.simantics.scl.compiler.elaboration.chr.CHRRule;
6 import org.simantics.scl.compiler.elaboration.chr.relations.SpecialCHRRelation;
7 import org.simantics.scl.compiler.elaboration.expressions.Assignment;
8 import org.simantics.scl.compiler.elaboration.expressions.Case;
9 import org.simantics.scl.compiler.elaboration.expressions.EAsPattern;
10 import org.simantics.scl.compiler.elaboration.expressions.EBind;
11 import org.simantics.scl.compiler.elaboration.expressions.ECHRSelect;
12 import org.simantics.scl.compiler.elaboration.expressions.ELet;
13 import org.simantics.scl.compiler.elaboration.expressions.ERuleset.DatalogRule;
14 import org.simantics.scl.compiler.elaboration.expressions.ESelect;
15 import org.simantics.scl.compiler.elaboration.expressions.ESimpleLambda;
16 import org.simantics.scl.compiler.elaboration.expressions.ESimpleLet;
17 import org.simantics.scl.compiler.elaboration.expressions.EVariable;
18 import org.simantics.scl.compiler.elaboration.expressions.EViewPattern;
19 import org.simantics.scl.compiler.elaboration.expressions.EWhen;
20 import org.simantics.scl.compiler.elaboration.expressions.Expression;
21 import org.simantics.scl.compiler.elaboration.expressions.Variable;
22 import org.simantics.scl.compiler.elaboration.expressions.list.ListAssignment;
23 import org.simantics.scl.compiler.elaboration.expressions.list.ListGenerator;
24 import org.simantics.scl.compiler.elaboration.query.QExists;
25
26 import gnu.trove.set.hash.THashSet;
27
28 public class CollectFreeVariablesVisitor extends StandardExpressionVisitor {
29     private THashSet<Variable> freeVariables = new THashSet<Variable>();
30     
31     private StandardExpressionVisitor collectBoundVariablesVisitor = new StandardExpressionVisitor() {
32         @Override
33         public void visit(EVariable expression) {
34             if(expression.variable != null)
35                 freeVariables.remove(expression.variable);
36         }
37         
38         
39         @Override
40         public void visit(EAsPattern expression) {
41             freeVariables.remove(expression.var);
42             expression.pattern.accept(this);
43         }
44         
45         @Override
46         public void visit(EViewPattern expression) {
47             expression.expression.accept(CollectFreeVariablesVisitor.this);
48             expression.pattern.accept(this);
49         }
50     };
51     
52     @Override
53     public void visit(EVariable expression) {
54         if(expression.variable != null)
55             freeVariables.add(expression.variable);
56     }
57     
58     @Override
59     public void visit(EBind expression) {
60         if(expression.monadEvidence != null)
61             visit(expression.monadEvidence);
62         expression.in.accept(this);
63         expression.value.accept(this);
64         expression.pattern.accept(collectBoundVariablesVisitor);
65     }
66     
67     @Override
68     public void visit(Assignment assignment) {
69         assignment.value.accept(this);
70         assignment.pattern.accept(collectBoundVariablesVisitor);
71     }
72     
73     @Override
74     public void visit(ELet expression) {
75         expression.in.accept(this);
76         for(int i=expression.assignments.length-1;i>=0;--i)
77             visit(expression.assignments[i]);
78     }
79     
80     @Override
81     public void visit(Case case_) {
82         case_.value.accept(this);
83         for(Expression pattern : case_.patterns)
84             pattern.accept(collectBoundVariablesVisitor);
85     }
86     
87     @Override
88     public void visit(ESimpleLambda expression) {
89         expression.value.accept(this);
90         freeVariables.remove(expression.parameter);
91     }
92     
93     @Override
94     public void visit(ESimpleLet expression) {
95         expression.in.accept(this);
96         expression.value.accept(this);
97         freeVariables.remove(expression.variable);
98     }
99     
100     @Override
101     public void visit(ECHRSelect expression) {
102         expression.expression.accept(this);
103         visit(expression.query);
104         for(Variable variable : expression.existentialVariables)
105             freeVariables.remove(variable);
106     }
107     
108     @Override
109     public void visit(ESelect expression) {
110         expression.expression.accept(this);
111         expression.query.accept(this);
112         for(Variable variable : expression.variables)
113             freeVariables.remove(variable);
114     }
115     
116     @Override
117     public void visit(EWhen expression) {
118         expression.action.accept(this);
119         expression.query.accept(this);
120         for(Variable variable : expression.variables)
121             freeVariables.remove(variable);
122     }
123
124     @Override
125     public void visit(DatalogRule rule) {
126         for(Expression parameter : rule.headParameters)
127             parameter.accept(this);
128         rule.body.accept(this);
129         for(Variable variable : rule.variables)
130             freeVariables.remove(variable);
131     }
132     
133     @Override
134     public void visit(ListGenerator qualifier) {
135         qualifier.pattern.accept(collectBoundVariablesVisitor);
136         qualifier.value.accept(this);
137     }
138     
139     @Override
140     public void visit(ListAssignment qualifier) {
141         qualifier.pattern.accept(collectBoundVariablesVisitor);
142         qualifier.value.accept(this);
143     }
144     
145     @Override
146     public void visit(CHRLiteral literal) {
147         if(literal.relation == SpecialCHRRelation.ASSIGN) {
148             literal.parameters[0].accept(collectBoundVariablesVisitor);
149             literal.parameters[1].accept(this);
150         }
151         else {
152             for(Expression parameter : literal.parameters)
153                 parameter.accept(this);
154             if(literal.typeConstraintEvidenceParameters != null)
155                 for(Expression parameter : literal.typeConstraintEvidenceParameters)
156                     parameter.accept(this);
157         }
158     }
159     
160     @Override
161     public void visit(CHRQuery query) {
162         for(int i=query.literals.length-1;i>=0;--i) {
163             visit(query.literals[i]);
164         }
165     }
166     
167     @Override
168     public void visit(CHRRule rule) {
169         super.visit(rule);
170         for(Variable variable : rule.existentialVariables)
171             freeVariables.remove(variable);
172     }
173     
174     @Override
175     public void visit(QExists query) {
176         query.query.accept(this);
177         for(Variable variable : query.variables)
178             freeVariables.remove(variable);
179     }
180     
181     public THashSet<Variable> getFreeVariables() {
182         return freeVariables;
183     }
184 }