VarMap.java

Go to the documentation of this file.

package com.graphbuilder.math;

public class VarMap {

    private boolean caseSensitive = true;
    private String[] name = new String[2];
    private double[] value = new double[2];
    private int numVars = 0;

    public VarMap() {
        this(true);
    }

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

    public double getValue(String varName) {
        for (int i = 0; i < numVars; i++)
            if (caseSensitive && name[i].equals(varName) || !caseSensitive && name[i].equalsIgnoreCase(varName))
                return value[i];

        throw new RuntimeException("variable value has not been set: " + varName);
    }

    public void setValue(String varName, double val) {
        if (varName == null)
            throw new IllegalArgumentException("varName cannot be null");

        for (int i = 0; i < numVars; i++) {
            if (caseSensitive && name[i].equals(varName) || !caseSensitive && name[i].equalsIgnoreCase(varName)) {
                value[i] = val;
                return;
            }
        }

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

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

            name = tmp1;
            value = tmp2;
        }

        name[numVars] = varName;
        value[numVars] = val;
        numVars++;
    }

    public boolean isCaseSensitive() {
        return caseSensitive;
    }

    public String[] getVariableNames() {
        String[] arr = new String[numVars];

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

        return arr;
    }

    public double[] getValues() {
        double[] arr = new double[numVars];

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

        return arr;
    }

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