java-projects/Own-Programming-Language-Tutorial
1
0
Fork 0
mirror of https://github.com/aNNiMON/Own-Programming-Language-Tutorial.git synced 2024-09-20 08:44:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

Исправлен OptimizationVisitor на паттерн матчинге

Browse Source
This commit is contained in:
Victor 2016-07-22 19:03:55 +03:00
parent bc93f73eab
commit 0921d8cb17
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/com/annimon/ownlang/parser/optimization/OptimizationVisitor.java
View File

@ -1,6 +1,8 @@
package com.annimon.ownlang.parser.optimization;
import com.annimon.ownlang.lib.Value;
import com.annimon.ownlang.parser.ast.*;
import static com.annimon.ownlang.parser.visitors.VisitorUtils.isValue;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
@ -246,11 +248,37 @@ public abstract class OptimizationVisitor<T> implements ResultVisitor<Node, T> {
boolean changed = expression != s.expression;
final List<MatchExpression.Pattern> patterns = new ArrayList<>(s.patterns.size());
for (MatchExpression.Pattern pattern : s.patterns) {
if (pattern instanceof MatchExpression.VariablePattern) {
final String variable = ((MatchExpression.VariablePattern) pattern).variable;
final VariableExpression expr = new VariableExpression(variable);
final Node node = expr.accept(this, t);
if (node != expr) {
if (isValue(node)) {
changed = true;
final Value value = ((ValueExpression) node).value;
final Expression optCondition = pattern.optCondition;
final Statement result = pattern.result;
pattern = new MatchExpression.ConstantPattern(value);
pattern.optCondition = optCondition;
pattern.result = result;
}
}
}
final Node patternResult = pattern.result.accept(this, t);
if (patternResult != pattern.result) {
changed = true;
pattern.result = consumeStatement(patternResult);
}
if (pattern.optCondition != null) {
Node optCond = pattern.optCondition.accept(this, t);
if (optCond != pattern.optCondition) {
changed = true;
pattern.optCondition = (Expression) optCond;
}
}
patterns.add(pattern);
}
if (changed) {