org.lsmp.djep.xjep.function.XAssign Maven / Gradle / Ivy

Go to download

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

JEP is a Java library for parsing and evaluating mathematical expressions.

There is a newer version: 2.4.2

/* @author rich
 * Created on 18-Nov-2003
 */
package org.lsmp.djep.xjep.function;

import org.lsmp.djep.xjep.CommandVisitorI;
import org.lsmp.djep.xjep.TreeUtils;
import org.lsmp.djep.xjep.XJep;
import org.lsmp.djep.xjep.XVariable;
import org.nfunk.jep.*;
import org.nfunk.jep.function.*;

/**
 * An assignment operator so we can do
 * x=3+4.
 * This function implements the SpecialEvaluationI interface
 * so that it handles seting the value of a variable. 
 * @author Rich Morris
 * Created on 18-Nov-2003
 */
public class XAssign extends Assign implements CommandVisitorI {

	public XAssign() {}
	{
		numberOfParameters = 2;
	}

	/**
	 * In the pre-process stage, set the equation of the lhs variable to the rhs equation.
	 */
	public Node process(Node node,Node children[],XJep xjep) throws ParseException
	{
		if(node.jjtGetNumChildren()!=2)
			throw new ParseException("Assignment opperator must have 2 operators.");

		// evaluate the value of the righthand side. Left on top of stack

		// Set the value of the variable on the lhs. 
		Node lhsNode = children[0];
		if(lhsNode instanceof ASTVarNode)
		{
			ASTVarNode vn = (ASTVarNode) lhsNode;
			XVariable var = (XVariable) vn.getVar();
			var.setEquation(xjep.deepCopy(children[1]));
			TreeUtils.copyChildrenIfNeeded(node,children);
			return node;
		}
		else if(lhsNode instanceof ASTFunNode && ((ASTFunNode) lhsNode).getPFMC() instanceof LValueI)
		{
			//TODO 
			TreeUtils.copyChildrenIfNeeded(node,children);
			return node;
		}
		throw new ParseException("Assignment should have a variable for the lhs.");
	}
}