FuncMap.java

Go to the documentation of this file.

package com.graphbuilder.math;

import com.graphbuilder.math.func.*;

public class FuncMap {

    private String[] name = new String[50];
    private Function[] func = new Function[50];
    private int numFunc = 0;
    private boolean caseSensitive = false;

    public FuncMap() {}

    public FuncMap(boolean caseSensitive) {
        this.caseSensitive = caseSensitive;
    }

    public void loadDefaultFunctions() {
        // >= 0 parameters
        setFunction("min", new MinFunction());
        setFunction("max", new MaxFunction());

        // > 0 parameters
        setFunction("sum", new SumFunction());
        setFunction("avg", new AvgFunction());

        // 0 parameters
        //setFunction("pi", new PiFunction());
        setFunction("e", new EFunction());
//      setFunction("rand", new RandFunction());

        // 1 parameter
//      setFunction("sin", new SinFunction());
//      setFunction("cos", new CosFunction());
//      setFunction("tan", new TanFunction());
        setFunction("sqrt", new SqrtFunction());
        setFunction("abs", new AbsFunction());
        setFunction("ceil", new CeilFunction());
        setFunction("floor", new FloorFunction());
        setFunction("exp", new ExpFunction());
        setFunction("lg", new LgFunction());
        setFunction("ln", new LnFunction());
        setFunction("sign", new SignFunction());
        setFunction("round", new RoundFunction());
        setFunction("fact", new FactFunction());
//      setFunction("cosh", new CoshFunction());
//      setFunction("sinh", new SinhFunction());
//      setFunction("tanh", new TanhFunction());
//      setFunction("acos", new AcosFunction());
//      setFunction("asin", new AsinFunction());
//      setFunction("atan", new AtanFunction());
//      setFunction("acosh", new AcoshFunction());
//      setFunction("asinh", new AsinhFunction());
//      setFunction("atanh", new AtanhFunction());

        // 2 parameters
        setFunction("pow", new PowFunction());
        setFunction("mod", new ModFunction());
        setFunction("combin", new CombinFunction());

        // 1 or 2 parameters
        setFunction("log", new LogFunction());
        
        // 2 or 3 parameters
        setFunction("if", new IfFunction());
    }

    public Function getFunction(String funcName, int numParam) {
        for (int i = 0; i < numFunc; i++) {
            if (func[i].acceptNumParam(numParam) && (caseSensitive && name[i].equals(funcName) || !caseSensitive && name[i].equalsIgnoreCase(funcName)))
                return func[i];
        }

        throw new RuntimeException("function not found: " + funcName + " " + numParam);
    }

    public void setFunction(String funcName, Function f) {
        if (funcName == null)
            throw new IllegalArgumentException("function name cannot be null");

        if (f == null)
            throw new IllegalArgumentException("function cannot be null");

        for (int i = 0; i < numFunc; i++) {
            if (caseSensitive && name[i].equals(funcName) || !caseSensitive && name[i].equalsIgnoreCase(funcName)) {
                func[i] = f;
                return;
            }
        }

        if (numFunc == name.length) {
            String[] tmp1 = new String[2 * numFunc];
            Function[] tmp2 = new Function[tmp1.length];

            for (int i = 0; i < numFunc; i++) {
                tmp1[i] = name[i];
                tmp2[i] = func[i];
            }

            name = tmp1;
            func = tmp2;
        }

        name[numFunc] = funcName;
        func[numFunc] = f;
        numFunc++;
    }

    public boolean isCaseSensitive() {
        return caseSensitive;
    }

    public String[] getFunctionNames() {
        String[] arr = new String[numFunc];

        for (int i = 0; i < arr.length; i++)
            arr[i] = name[i];

        return arr;
    }

    public Function[] getFunctions() {
        Function[] arr = new Function[numFunc];

        for (int i = 0; i < arr.length; i++)
            arr[i] = func[i];

        return arr;
    }

    public void remove(String funcName) {
        for (int i = 0; i < numFunc; i++) {
            if (caseSensitive && name[i].equals(funcName) || !caseSensitive && name[i].equalsIgnoreCase(funcName)) {
                for (int j = i + 1; j < numFunc; j++) {
                    name[j - 1] = name[j];
                    func[j - 1] = func[j];
                }
                numFunc--;
                name[numFunc] = null;
                func[numFunc] = null;
                break;
            }
        }
    }
}