522 lines
20 KiB
Java
522 lines
20 KiB
Java
package com.hypixel.hytale.server.npc.util.expression;
|
|
|
|
import com.hypixel.hytale.common.util.ArrayUtil;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.Objects;
|
|
import java.util.function.BooleanSupplier;
|
|
import java.util.function.DoubleSupplier;
|
|
import java.util.function.Supplier;
|
|
import javax.annotation.Nonnull;
|
|
import javax.annotation.Nullable;
|
|
|
|
public class StdScope implements Scope {
|
|
protected static final StdScope.SymbolStringArray VAR_EMPTY_STRING_ARRAY = new StdScope.SymbolStringArray(false, () -> ArrayUtil.EMPTY_STRING_ARRAY);
|
|
protected static final StdScope.SymbolNumberArray VAR_EMPTY_NUMBER_ARRAY = new StdScope.SymbolNumberArray(false, () -> ArrayUtil.EMPTY_DOUBLE_ARRAY);
|
|
protected static final StdScope.SymbolBooleanArray VAR_EMPTY_BOOLEAN_ARRAY = new StdScope.SymbolBooleanArray(false, () -> ArrayUtil.EMPTY_BOOLEAN_ARRAY);
|
|
protected static final StdScope.SymbolStringArray VAR_NULL_STRING_ARRAY = new StdScope.SymbolStringArray(false, () -> null);
|
|
protected static final StdScope.SymbolNumberArray VAR_NULL_NUMBER_ARRAY = new StdScope.SymbolNumberArray(false, () -> null);
|
|
protected static final StdScope.SymbolBooleanArray VAR_NULL_BOOLEAN_ARRAY = new StdScope.SymbolBooleanArray(false, () -> null);
|
|
protected static final StdScope.SymbolString VAR_NULL_STRING = new StdScope.SymbolString(false, () -> null);
|
|
protected static final StdScope.SymbolString VAR_EMPTY_STRING = new StdScope.SymbolString(false, () -> "");
|
|
protected static final StdScope.SymbolBoolean VAR_BOOLEAN_TRUE = new StdScope.SymbolBoolean(false, () -> true);
|
|
protected static final StdScope.SymbolBoolean VAR_BOOLEAN_FALSE = new StdScope.SymbolBoolean(false, () -> false);
|
|
protected static final StdScope.SymbolStringArray CONST_EMPTY_STRING_ARRAY = new StdScope.SymbolStringArray(true, () -> ArrayUtil.EMPTY_STRING_ARRAY);
|
|
protected static final StdScope.SymbolNumberArray CONST_EMPTY_NUMBER_ARRAY = new StdScope.SymbolNumberArray(true, () -> ArrayUtil.EMPTY_DOUBLE_ARRAY);
|
|
protected static final StdScope.SymbolBooleanArray CONST_EMPTY_BOOLEAN_ARRAY = new StdScope.SymbolBooleanArray(true, () -> ArrayUtil.EMPTY_BOOLEAN_ARRAY);
|
|
protected static final StdScope.SymbolStringArray CONST_NULL_STRING_ARRAY = new StdScope.SymbolStringArray(true, () -> null);
|
|
protected static final StdScope.SymbolNumberArray CONST_NULL_NUMBER_ARRAY = new StdScope.SymbolNumberArray(true, () -> null);
|
|
protected static final StdScope.SymbolBooleanArray CONST_NULL_BOOLEAN_ARRAY = new StdScope.SymbolBooleanArray(true, () -> null);
|
|
protected static final StdScope.SymbolString CONST_NULL_STRING = new StdScope.SymbolString(true, () -> null);
|
|
protected static final StdScope.SymbolString CONST_EMPTY_STRING = new StdScope.SymbolString(true, () -> "");
|
|
protected static final StdScope.SymbolBoolean CONST_BOOLEAN_TRUE = new StdScope.SymbolBoolean(true, () -> true);
|
|
protected static final StdScope.SymbolBoolean CONST_BOOLEAN_FALSE = new StdScope.SymbolBoolean(true, () -> false);
|
|
protected Scope parent;
|
|
protected Map<String, StdScope.Symbol> symbolTable;
|
|
|
|
public StdScope(Scope parent) {
|
|
this.parent = parent;
|
|
this.symbolTable = new HashMap<>();
|
|
}
|
|
|
|
@Nonnull
|
|
public static StdScope copyOf(@Nonnull StdScope other) {
|
|
StdScope scope = new StdScope(other.parent);
|
|
scope.mergeSymbols(other);
|
|
return scope;
|
|
}
|
|
|
|
@Nonnull
|
|
public StdScope merge(@Nonnull StdScope other) {
|
|
this.mergeSymbols(other);
|
|
return this;
|
|
}
|
|
|
|
@Nonnull
|
|
public static StdScope mergeScopes(@Nonnull StdScope first, @Nonnull StdScope second) {
|
|
return copyOf(first).merge(second);
|
|
}
|
|
|
|
protected void mergeSymbols(@Nonnull StdScope other) {
|
|
other.symbolTable.forEach(this::add);
|
|
}
|
|
|
|
protected void add(String name, StdScope.Symbol symbol) {
|
|
if (this.symbolTable.containsKey(name)) {
|
|
throw new IllegalStateException("Trying to add symbol twice to scope " + name);
|
|
} else {
|
|
this.symbolTable.put(name, symbol);
|
|
}
|
|
}
|
|
|
|
public void addConst(String name, @Nullable String value) {
|
|
if (value == null) {
|
|
this.add(name, CONST_NULL_STRING);
|
|
} else if (value.isEmpty()) {
|
|
this.add(name, CONST_EMPTY_STRING);
|
|
} else {
|
|
this.add(name, new StdScope.SymbolString(true, () -> value));
|
|
}
|
|
}
|
|
|
|
public void addConst(String name, double value) {
|
|
this.add(name, new StdScope.SymbolNumber(true, () -> value));
|
|
}
|
|
|
|
public void addConst(String name, boolean value) {
|
|
this.add(name, value ? CONST_BOOLEAN_TRUE : CONST_BOOLEAN_FALSE);
|
|
}
|
|
|
|
public void addConst(String name, @Nullable String[] value) {
|
|
if (value == null) {
|
|
this.add(name, CONST_NULL_STRING_ARRAY);
|
|
} else if (value.length == 0) {
|
|
this.add(name, CONST_EMPTY_STRING_ARRAY);
|
|
} else {
|
|
this.add(name, new StdScope.SymbolStringArray(true, () -> value));
|
|
}
|
|
}
|
|
|
|
public void addConst(String name, @Nullable double[] value) {
|
|
if (value == null) {
|
|
this.add(name, CONST_NULL_NUMBER_ARRAY);
|
|
} else if (value.length == 0) {
|
|
this.add(name, CONST_EMPTY_NUMBER_ARRAY);
|
|
} else {
|
|
this.add(name, new StdScope.SymbolNumberArray(true, () -> value));
|
|
}
|
|
}
|
|
|
|
public void addConst(String name, @Nullable boolean[] value) {
|
|
if (value == null) {
|
|
this.add(name, CONST_NULL_BOOLEAN_ARRAY);
|
|
} else if (value.length == 0) {
|
|
this.add(name, CONST_EMPTY_BOOLEAN_ARRAY);
|
|
} else {
|
|
this.add(name, new StdScope.SymbolBooleanArray(true, () -> value));
|
|
}
|
|
}
|
|
|
|
public void addConstEmptyArray(String name) {
|
|
this.add(name, new StdScope.Symbol(true, ValueType.EMPTY_ARRAY));
|
|
}
|
|
|
|
public void addVar(String name, @Nullable String value) {
|
|
if (value == null) {
|
|
this.add(name, VAR_NULL_STRING);
|
|
} else if (value.isEmpty()) {
|
|
this.add(name, VAR_EMPTY_STRING);
|
|
} else {
|
|
this.add(name, new StdScope.SymbolString(false, () -> value));
|
|
}
|
|
}
|
|
|
|
public void addVar(String name, double value) {
|
|
this.add(name, new StdScope.SymbolNumber(false, () -> value));
|
|
}
|
|
|
|
public void addVar(String name, boolean value) {
|
|
this.add(name, value ? VAR_BOOLEAN_TRUE : VAR_BOOLEAN_FALSE);
|
|
}
|
|
|
|
public void addVar(String name, @Nullable String[] value) {
|
|
if (value == null) {
|
|
this.add(name, VAR_NULL_STRING_ARRAY);
|
|
} else if (value.length == 0) {
|
|
this.add(name, VAR_EMPTY_STRING_ARRAY);
|
|
} else {
|
|
this.add(name, new StdScope.SymbolStringArray(false, () -> value));
|
|
}
|
|
}
|
|
|
|
public void addVar(String name, @Nullable double[] value) {
|
|
if (value == null) {
|
|
this.add(name, VAR_NULL_NUMBER_ARRAY);
|
|
} else if (value.length == 0) {
|
|
this.add(name, VAR_EMPTY_NUMBER_ARRAY);
|
|
} else {
|
|
this.add(name, new StdScope.SymbolNumberArray(false, () -> value));
|
|
}
|
|
}
|
|
|
|
public void addVar(String name, @Nullable boolean[] value) {
|
|
if (value == null) {
|
|
this.add(name, VAR_NULL_BOOLEAN_ARRAY);
|
|
} else if (value.length == 0) {
|
|
this.add(name, VAR_EMPTY_BOOLEAN_ARRAY);
|
|
} else {
|
|
this.add(name, new StdScope.SymbolBooleanArray(false, () -> value));
|
|
}
|
|
}
|
|
|
|
public void addInvariant(@Nonnull String name, Scope.Function function, ValueType returnType, @Nonnull ValueType... argumentTypes) {
|
|
this.add(Scope.encodeFunctionName(name, argumentTypes), new StdScope.SymbolFunction(true, returnType, function));
|
|
this.add(name, new StdScope.SymbolFunction(false, returnType, null));
|
|
}
|
|
|
|
public void addVariant(@Nonnull String name, Scope.Function function, ValueType returnType, @Nonnull ValueType... argumentTypes) {
|
|
this.add(Scope.encodeFunctionName(name, argumentTypes), new StdScope.SymbolFunction(false, returnType, function));
|
|
this.add(name, new StdScope.SymbolFunction(false, returnType, null));
|
|
}
|
|
|
|
public void addSupplier(String name, Supplier<String> value) {
|
|
this.add(name, new StdScope.SymbolString(false, value));
|
|
}
|
|
|
|
public void addSupplier(String name, DoubleSupplier value) {
|
|
this.add(name, new StdScope.SymbolNumber(false, value));
|
|
}
|
|
|
|
public void addSupplier(String name, BooleanSupplier value) {
|
|
this.add(name, new StdScope.SymbolBoolean(false, value));
|
|
}
|
|
|
|
public void addStringArraySupplier(String name, Supplier<String[]> value) {
|
|
this.add(name, new StdScope.SymbolStringArray(false, value));
|
|
}
|
|
|
|
public void addDoubleArraySupplier(String name, Supplier<double[]> value) {
|
|
this.add(name, new StdScope.SymbolNumberArray(false, value));
|
|
}
|
|
|
|
public void addBooleanArraySupplier(String name, Supplier<boolean[]> value) {
|
|
this.add(name, new StdScope.SymbolBooleanArray(false, value));
|
|
}
|
|
|
|
protected StdScope.Symbol get(String name) {
|
|
return this.symbolTable.get(name);
|
|
}
|
|
|
|
@Nonnull
|
|
protected StdScope.Symbol get(String name, ValueType valueType) {
|
|
StdScope.Symbol symbol = this.symbolTable.get(name);
|
|
if (symbol == null) {
|
|
throw new IllegalStateException("Can't find symbol " + name + " in symbol table");
|
|
} else if (!ValueType.isAssignableType(valueType, symbol.valueType)) {
|
|
throw new IllegalStateException("Type mismatch with " + name + ". Got " + valueType + " but expected " + symbol.valueType);
|
|
} else {
|
|
return symbol;
|
|
}
|
|
}
|
|
|
|
protected void replace(String name, @Nonnull StdScope.Symbol symbol) {
|
|
StdScope.Symbol oldSymbol = this.get(name, symbol.valueType);
|
|
if (oldSymbol.isConstant) {
|
|
throw new IllegalStateException("Can't replace a constant in symbol table: " + name);
|
|
} else if (symbol.isConstant) {
|
|
throw new IllegalStateException("Can't replace a variable with a constant: " + name);
|
|
} else {
|
|
this.symbolTable.put(name, symbol);
|
|
}
|
|
}
|
|
|
|
public void changeValue(String name, @Nullable String value) {
|
|
if (value == null) {
|
|
this.replace(name, VAR_NULL_STRING);
|
|
} else if (value.isEmpty()) {
|
|
this.replace(name, VAR_EMPTY_STRING);
|
|
} else {
|
|
this.replace(name, new StdScope.SymbolString(false, () -> value));
|
|
}
|
|
}
|
|
|
|
public void changeValue(String name, double value) {
|
|
this.replace(name, new StdScope.SymbolNumber(false, () -> value));
|
|
}
|
|
|
|
public void changeValue(String name, boolean value) {
|
|
this.replace(name, value ? VAR_BOOLEAN_TRUE : VAR_BOOLEAN_FALSE);
|
|
}
|
|
|
|
public void changeValue(String name, @Nullable String[] value) {
|
|
if (value == null) {
|
|
this.replace(name, VAR_NULL_STRING_ARRAY);
|
|
} else if (value.length == 0) {
|
|
this.replace(name, VAR_EMPTY_STRING_ARRAY);
|
|
} else {
|
|
this.replace(name, new StdScope.SymbolStringArray(false, () -> value));
|
|
}
|
|
}
|
|
|
|
public void changeValue(String name, @Nullable double[] value) {
|
|
if (value == null) {
|
|
this.replace(name, VAR_NULL_NUMBER_ARRAY);
|
|
} else if (value.length == 0) {
|
|
this.replace(name, VAR_EMPTY_NUMBER_ARRAY);
|
|
} else {
|
|
this.replace(name, new StdScope.SymbolNumberArray(false, () -> value));
|
|
}
|
|
}
|
|
|
|
public void changeValue(String name, @Nullable boolean[] value) {
|
|
if (value == null) {
|
|
this.replace(name, VAR_NULL_BOOLEAN_ARRAY);
|
|
} else if (value.length == 0) {
|
|
this.replace(name, VAR_EMPTY_BOOLEAN_ARRAY);
|
|
} else {
|
|
this.replace(name, new StdScope.SymbolBooleanArray(false, () -> value));
|
|
}
|
|
}
|
|
|
|
public void changeValueToEmptyArray(String name) {
|
|
StdScope.Symbol symbol = this.get(name);
|
|
Objects.requireNonNull(symbol, "Can't find symbol in symbol table in changeValue()");
|
|
if (symbol.isConstant) {
|
|
throw new IllegalStateException("Can't replace a constant in symbol table: " + name);
|
|
} else {
|
|
switch (symbol.valueType) {
|
|
case VOID:
|
|
case NUMBER:
|
|
case STRING:
|
|
case BOOLEAN:
|
|
default:
|
|
throw new IllegalStateException("Can't assign an empty array to symbol " + name + " of type " + symbol.valueType);
|
|
case EMPTY_ARRAY:
|
|
return;
|
|
case NUMBER_ARRAY:
|
|
this.symbolTable.put(name, VAR_EMPTY_NUMBER_ARRAY);
|
|
break;
|
|
case STRING_ARRAY:
|
|
this.symbolTable.put(name, VAR_EMPTY_STRING_ARRAY);
|
|
break;
|
|
case BOOLEAN_ARRAY:
|
|
this.symbolTable.put(name, VAR_EMPTY_BOOLEAN_ARRAY);
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Supplier<String> getStringSupplier(String name) {
|
|
StdScope.Symbol symbol = this.get(name);
|
|
if (symbol == null) {
|
|
if (this.parent != null) {
|
|
return this.parent.getStringSupplier(name);
|
|
} else {
|
|
throw new IllegalStateException("Unable to find symbol: " + name);
|
|
}
|
|
} else if (symbol instanceof StdScope.SymbolString) {
|
|
return ((StdScope.SymbolString)symbol).value;
|
|
} else {
|
|
throw new IllegalStateException("Symbol is not a string: " + name);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public DoubleSupplier getNumberSupplier(String name) {
|
|
StdScope.Symbol symbol = this.get(name);
|
|
if (symbol == null) {
|
|
if (this.parent != null) {
|
|
return this.parent.getNumberSupplier(name);
|
|
} else {
|
|
throw new IllegalStateException("Unable to find symbol: " + name);
|
|
}
|
|
} else if (symbol instanceof StdScope.SymbolNumber) {
|
|
return ((StdScope.SymbolNumber)symbol).value;
|
|
} else {
|
|
throw new IllegalStateException("Symbol is not a number: " + name);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public BooleanSupplier getBooleanSupplier(String name) {
|
|
StdScope.Symbol symbol = this.get(name);
|
|
if (symbol == null) {
|
|
if (this.parent != null) {
|
|
return this.parent.getBooleanSupplier(name);
|
|
} else {
|
|
throw new IllegalStateException("Unable to find symbol: " + name);
|
|
}
|
|
} else if (symbol instanceof StdScope.SymbolBoolean) {
|
|
return ((StdScope.SymbolBoolean)symbol).value;
|
|
} else {
|
|
throw new IllegalStateException("Symbol is not a boolean: " + name);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Supplier<String[]> getStringArraySupplier(String name) {
|
|
StdScope.Symbol symbol = this.get(name);
|
|
if (symbol == null) {
|
|
if (this.parent != null) {
|
|
return this.parent.getStringArraySupplier(name);
|
|
} else {
|
|
throw new IllegalStateException("Unable to find symbol: " + name);
|
|
}
|
|
} else if (symbol.valueType == ValueType.EMPTY_ARRAY) {
|
|
return () -> ArrayUtil.EMPTY_STRING_ARRAY;
|
|
} else if (symbol instanceof StdScope.SymbolStringArray) {
|
|
return ((StdScope.SymbolStringArray)symbol).value;
|
|
} else {
|
|
throw new IllegalStateException("Symbol is not a string array: " + name);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Supplier<double[]> getNumberArraySupplier(String name) {
|
|
StdScope.Symbol symbol = this.get(name);
|
|
if (symbol == null) {
|
|
if (this.parent != null) {
|
|
return this.parent.getNumberArraySupplier(name);
|
|
} else {
|
|
throw new IllegalStateException("Unable to find symbol: " + name);
|
|
}
|
|
} else if (symbol.valueType == ValueType.EMPTY_ARRAY) {
|
|
return () -> ArrayUtil.EMPTY_DOUBLE_ARRAY;
|
|
} else if (symbol instanceof StdScope.SymbolNumberArray) {
|
|
return ((StdScope.SymbolNumberArray)symbol).value;
|
|
} else {
|
|
throw new IllegalStateException("Symbol is not a number array: " + name);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Supplier<boolean[]> getBooleanArraySupplier(String name) {
|
|
StdScope.Symbol symbol = this.get(name);
|
|
if (symbol == null) {
|
|
if (this.parent != null) {
|
|
return this.parent.getBooleanArraySupplier(name);
|
|
} else {
|
|
throw new IllegalStateException("Unable to find symbol: " + name);
|
|
}
|
|
} else if (symbol.valueType == ValueType.EMPTY_ARRAY) {
|
|
return () -> ArrayUtil.EMPTY_BOOLEAN_ARRAY;
|
|
} else if (symbol instanceof StdScope.SymbolBooleanArray) {
|
|
return ((StdScope.SymbolBooleanArray)symbol).value;
|
|
} else {
|
|
throw new IllegalStateException("Symbol is not a boolean array: " + name);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Scope.Function getFunction(String name) {
|
|
StdScope.Symbol symbol = this.get(name);
|
|
if (symbol == null) {
|
|
if (this.parent != null) {
|
|
return this.parent.getFunction(name);
|
|
} else {
|
|
throw new IllegalStateException("Unable to find function: " + name);
|
|
}
|
|
} else if (symbol instanceof StdScope.SymbolFunction) {
|
|
return ((StdScope.SymbolFunction)symbol).value;
|
|
} else {
|
|
throw new IllegalStateException("Symbol is not a function: " + name);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean isConstant(String name) {
|
|
StdScope.Symbol symbol = this.get(name);
|
|
if (symbol != null) {
|
|
return symbol.isConstant;
|
|
} else if (this.parent == null) {
|
|
throw new IllegalStateException("Unable to find symbol: " + name);
|
|
} else {
|
|
return this.parent.isConstant(name);
|
|
}
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
public ValueType getType(String name) {
|
|
StdScope.Symbol symbol = this.get(name);
|
|
if (symbol != null) {
|
|
return symbol.valueType;
|
|
} else {
|
|
return this.parent != null ? this.parent.getType(name) : null;
|
|
}
|
|
}
|
|
|
|
protected static class Symbol {
|
|
public final boolean isConstant;
|
|
public final ValueType valueType;
|
|
|
|
public Symbol(boolean isConstant, ValueType valueType) {
|
|
this.isConstant = isConstant;
|
|
this.valueType = valueType;
|
|
}
|
|
}
|
|
|
|
protected static class SymbolBoolean extends StdScope.Symbol {
|
|
public final BooleanSupplier value;
|
|
|
|
public SymbolBoolean(boolean isConstant, BooleanSupplier value) {
|
|
super(isConstant, ValueType.BOOLEAN);
|
|
this.value = value;
|
|
}
|
|
}
|
|
|
|
protected static class SymbolBooleanArray extends StdScope.Symbol {
|
|
public final Supplier<boolean[]> value;
|
|
|
|
public SymbolBooleanArray(boolean isConstant, Supplier<boolean[]> value) {
|
|
super(isConstant, ValueType.BOOLEAN_ARRAY);
|
|
this.value = value;
|
|
}
|
|
}
|
|
|
|
protected static class SymbolFunction extends StdScope.Symbol {
|
|
public final Scope.Function value;
|
|
|
|
public SymbolFunction(boolean isConstant, ValueType returnType, Scope.Function value) {
|
|
super(isConstant, returnType);
|
|
this.value = value;
|
|
}
|
|
}
|
|
|
|
protected static class SymbolNumber extends StdScope.Symbol {
|
|
public final DoubleSupplier value;
|
|
|
|
public SymbolNumber(boolean isConstant, DoubleSupplier value) {
|
|
super(isConstant, ValueType.NUMBER);
|
|
this.value = value;
|
|
}
|
|
}
|
|
|
|
protected static class SymbolNumberArray extends StdScope.Symbol {
|
|
public final Supplier<double[]> value;
|
|
|
|
public SymbolNumberArray(boolean isConstant, Supplier<double[]> value) {
|
|
super(isConstant, ValueType.NUMBER_ARRAY);
|
|
this.value = value;
|
|
}
|
|
}
|
|
|
|
protected static class SymbolString extends StdScope.Symbol {
|
|
public final Supplier<String> value;
|
|
|
|
public SymbolString(boolean isConstant, Supplier<String> value) {
|
|
super(isConstant, ValueType.STRING);
|
|
this.value = value;
|
|
}
|
|
}
|
|
|
|
protected static class SymbolStringArray extends StdScope.Symbol {
|
|
public final Supplier<String[]> value;
|
|
|
|
public SymbolStringArray(boolean isConstant, Supplier<String[]> value) {
|
|
super(isConstant, ValueType.STRING_ARRAY);
|
|
this.value = value;
|
|
}
|
|
}
|
|
}
|