com.graphbuilder.math.FuncNode Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of curvesapi Show documentation

Implementation of various mathematical curves that define themselves over a set of control points. The API is written in Java. The curves supported are: Bezier, B-Spline, Cardinal Spline, Catmull-Rom Spline, Lagrange, Natural Cubic Spline, and NURBS.

There is a newer version: 1.08

Show newest version

package com.graphbuilder.math;

import com.graphbuilder.struc.Bag;

/**
A node of an expression tree that represents a function.  A FuncNode can have 0 or more children.
*/
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);
	}

	/**
	Adds the expression as a child.
	*/
	public void add(Expression x) {
		insert(x, bag.size());
	}

	/**
	Adds the expression as a child at the specified index.
	*/
	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;
	}

	/**
	Removes the specified expression as a child.  Does nothing if the expression was not a child.
	*/
	public void remove(Expression x) {
		int size = bag.size();
		bag.remove(x);
		if (size != bag.size())
			x.parent = null;
	}

	/**
	Returns the number of child expressions.
	*/
	public int numChildren() {
		return bag.size();
	}

	/**
	Returns the child expression at the specified index.
	*/
	public Expression child(int i) {
		return (Expression) bag.get(i);
	}

	/**
	Evaluates each of the children, storing the result in an internal double array.  The FuncMap is
	used to obtain a Function object based on the name of this FuncNode.  The function is passed
	the double array and returns a result.  If negate is true, the result is negated.  The result
	is then returned.  The numParam passed to the function is the number of children of this FuncNode.
	*/
	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;
	}
}