diff options
| -rw-r--r-- | src/main/java/com/healthmarketscience/jackcess/impl/expr/BuiltinOperators.java | 1 | ||||
| -rw-r--r-- | src/main/java/com/healthmarketscience/jackcess/impl/expr/DefaultFunctions.java | 241 |
2 files changed, 238 insertions, 4 deletions
diff --git a/src/main/java/com/healthmarketscience/jackcess/impl/expr/BuiltinOperators.java b/src/main/java/com/healthmarketscience/jackcess/impl/expr/BuiltinOperators.java index 5a3d7b5..93b4bf2 100644 --- a/src/main/java/com/healthmarketscience/jackcess/impl/expr/BuiltinOperators.java +++ b/src/main/java/com/healthmarketscience/jackcess/impl/expr/BuiltinOperators.java @@ -51,6 +51,7 @@ public class BuiltinOperators public static final Value TRUE_VAL = new LongValue(-1L); public static final Value FALSE_VAL = new LongValue(0L); public static final Value EMPTY_STR_VAL = new StringValue(""); + public static final Value ZERO_VAL = FALSE_VAL; private enum CoercionType { SIMPLE(true, true), GENERAL(false, true), COMPARE(false, false); diff --git a/src/main/java/com/healthmarketscience/jackcess/impl/expr/DefaultFunctions.java b/src/main/java/com/healthmarketscience/jackcess/impl/expr/DefaultFunctions.java index fbcd683..49fecb7 100644 --- a/src/main/java/com/healthmarketscience/jackcess/impl/expr/DefaultFunctions.java +++ b/src/main/java/com/healthmarketscience/jackcess/impl/expr/DefaultFunctions.java @@ -16,13 +16,14 @@ limitations under the License. package com.healthmarketscience.jackcess.impl.expr; +import java.math.BigDecimal; +import java.math.RoundingMode; import java.util.HashMap; import java.util.Map; - -import com.healthmarketscience.jackcess.expr.Value; -import com.healthmarketscience.jackcess.expr.Function; import com.healthmarketscience.jackcess.expr.EvalContext; +import com.healthmarketscience.jackcess.expr.Function; +import com.healthmarketscience.jackcess.expr.Value; /** * @@ -33,6 +34,8 @@ public class DefaultFunctions private static final Map<String,Function> FUNCS = new HashMap<String,Function>(); + private static final char NON_VAR_SUFFIX = '$'; + private DefaultFunctions() {} public static Function getFunction(String name) { @@ -78,6 +81,20 @@ public class DefaultFunctions } } + public static abstract class Func0 extends BaseFunction + { + protected Func0(String name) { + super(name, 0, 9); + } + + public final Value eval(EvalContext ctx, Value... params) { + validateNumParams(params); + return eval0(ctx); + } + + protected abstract Value eval0(EvalContext ctx); + } + public static abstract class Func1 extends BaseFunction { protected Func1(String name) { @@ -130,15 +147,231 @@ public class DefaultFunctions } }); + public static final Function ASC = registerFunc(new Func1("Asc") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + String str = param1.getAsString(); + int len = str.length(); + if(len == 0) { + throw new IllegalStateException("No characters in string"); + } + long lv = str.charAt(0); + if((lv < 0) || (lv > 255)) { + throw new IllegalStateException("Character code '" + lv + + "' out of range "); + } + return BuiltinOperators.toValue(lv); + } + }); + + public static final Function ASCW = registerFunc(new Func1("AscW") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + String str = param1.getAsString(); + int len = str.length(); + if(len == 0) { + throw new IllegalStateException("No characters in string"); + } + long lv = str.charAt(0); + return BuiltinOperators.toValue(lv); + } + }); + + public static final Function CHR = registerStringFunc(new Func1("Chr") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + long lv = param1.getAsLong(); + if((lv < 0) || (lv > 255)) { + throw new IllegalStateException("Character code '" + lv + + "' out of range "); + } + char[] cs = Character.toChars((int)lv); + return BuiltinOperators.toValue(new String(cs)); + } + }); + + public static final Function CHRW = registerStringFunc(new Func1("ChrW") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + long lv = param1.getAsLong(); + char[] cs = Character.toChars((int)lv); + return BuiltinOperators.toValue(new String(cs)); + } + }); + + public static final Function HEX = registerStringFunc(new Func1("Hex") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + if(param1.isNull()) { + return param1; + } + if((param1.getType() == Value.Type.STRING) && + (param1.getAsString().length() == 0)) { + return BuiltinOperators.ZERO_VAL; + } + long lv = param1.getAsLong(); + return BuiltinOperators.toValue(Long.toHexString(lv)); + } + }); + + public static final Function NZ = registerFunc(new BaseFunction("Nz", 1, 2) { + public Value eval(EvalContext ctx, Value... params) { + validateNumParams(params); + Value param1 = params[0]; + if(!param1.isNull()) { + return param1; + } + if(params.length > 1) { + return params[1]; + } + Value.Type resultType = ctx.getResultType(); + return (((resultType == null) || + (resultType == Value.Type.STRING)) ? + BuiltinOperators.EMPTY_STR_VAL : BuiltinOperators.ZERO_VAL); + } + }); + + public static final Function OCT = registerStringFunc(new Func1("Oct") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + if(param1.isNull()) { + return param1; + } + if((param1.getType() == Value.Type.STRING) && + (param1.getAsString().length() == 0)) { + return BuiltinOperators.ZERO_VAL; + } + long lv = param1.getAsLong(); + return BuiltinOperators.toValue(Long.toOctalString(lv)); + } + }); + + public static final Function STR = registerStringFunc(new Func1("Str") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + BigDecimal bd = param1.getAsBigDecimal(); + String str = bd.toPlainString(); + if(bd.compareTo(BigDecimal.ZERO) >= 0) { + str = " " + str; + } + return BuiltinOperators.toValue(str); + } + }); + + public static final Function CBOOL = registerFunc(new Func1("CBool") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + boolean b = param1.getAsBoolean(); + return BuiltinOperators.toValue(b); + } + }); + + public static final Function CBYTE = registerFunc(new Func1("CByte") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + Long lv = param1.getAsLong(); + if((lv < 0) || (lv > 255)) { + throw new IllegalStateException("Byte code '" + lv + "' out of range "); + } + return BuiltinOperators.toValue(lv); + } + }); + + public static final Function CCUR = registerFunc(new Func1("CCur") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + BigDecimal bd = param1.getAsBigDecimal(); + bd.setScale(4, RoundingMode.HALF_EVEN); + return BuiltinOperators.toValue(bd); + } + }); + + // public static final Function CDATE = registerFunc(new Func1("CDate") { + // @Override + // protected Value eval1(EvalContext ctx, Value param1) { + // FIXME + // BigDecimal bd = param1.getAsBigDecimal(); + // bd.setScale(4, RoundingMode.HALF_EVEN); + // return BuiltinOperators.toValue(bd); + // } + // }); + + public static final Function CDBL = registerFunc(new Func1("CDbl") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + Double db = param1.getAsDouble(); + return BuiltinOperators.toValue(db); + } + }); + + public static final Function CDEC = registerFunc(new Func1("CDec") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + BigDecimal bd = param1.getAsBigDecimal(); + return BuiltinOperators.toValue(bd); + } + }); + + public static final Function CINT = registerFunc(new Func1("CInt") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + // FIXME, fix rounding for this and clng + Long lv = param1.getAsLong(); + if((lv < Short.MIN_VALUE) || (lv > Short.MAX_VALUE)) { + throw new IllegalStateException("Int value '" + lv + "' out of range "); + } + return BuiltinOperators.toValue(lv); + } + }); + + public static final Function CLNG = registerFunc(new Func1("CLng") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + Long lv = param1.getAsLong(); + if((lv < Integer.MIN_VALUE) || (lv > Integer.MAX_VALUE)) { + throw new IllegalStateException("Long value '" + lv + "' out of range "); + } + return BuiltinOperators.toValue(lv); + } + }); + + public static final Function CSNG = registerFunc(new Func1("CSng") { + @Override + protected Value eval1(EvalContext ctx, Value param1) { + Double db = param1.getAsDouble(); + if((db < Float.MIN_VALUE) || (db > Float.MAX_VALUE)) { + throw new IllegalStateException("Single value '" + db + "' out of range "); + } + return BuiltinOperators.toValue(db); + } + }); + + // FIXME, CSTR, CVAR // https://www.techonthenet.com/access/functions/ // https://support.office.com/en-us/article/Access-Functions-by-category-b8b136c3-2716-4d39-94a2-658ce330ed83 private static Function registerFunc(Function func) { - if(FUNCS.put(func.getName().toLowerCase(), func) != null) { + return registerFunc(false, func); + } + + private static Function registerStringFunc(Function func) { + return registerFunc(true, func); + } + + private static Function registerFunc(boolean includeNonVar, Function func) { + String fname = func.getName().toLowerCase(); + if(FUNCS.put(fname, func) != null) { throw new IllegalStateException("Duplicate function " + func); } + if(includeNonVar) { + // for our purposes the non-variant versions are the same function + fname += NON_VAR_SUFFIX; + if(FUNCS.put(fname, func) != null) { + throw new IllegalStateException("Duplicate function " + func); + } + } return func; } } |