FuncNode.java

Go to the documentation of this file.

package com.graphbuilder.math;

import com.graphbuilder.struc.Bag;

public class FuncNode extends TermNode {

    private Bag bag = new Bag(1);
    private double[] of = new double[1];

    public FuncNode(String name, boolean negate) {
        super(name, negate);
    }

    public void add(Expression x) {
        insert(x, bag.size());
    }

    public void insert(Expression x, int i) {
        checkBeforeAccept(x);
        int oldCap = bag.getCapacity();
        bag.insert(x, i);
        int newCap = bag.getCapacity();

        if (oldCap != newCap)
            of = new double[newCap];

        x.parent = this;
    }

    public void remove(Expression x) {
        int size = bag.size();
        bag.remove(x);
        if (size != bag.size())
            x.parent = null;
    }

    public int numChildren() {
        return bag.size();
    }

    public Expression child(int i) {
        return (Expression) bag.get(i);
    }

    public double eval(VarMap v, FuncMap f) {
        int numParam = bag.size();

        for (int i = 0; i < numParam; i++)
            of[i] = child(i).eval(v, f);

        double result = f.getFunction(name, numParam).of(of, numParam);

        if (negate) result = -result;

        return result;
    }
}