decodes.tsdb.algo.jep.RatingFunction Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of opendcs Show documentation
A collection of software for aggregatting and processing environmental data such as from NOAA GOES satellites.
The newest version!
package decodes.tsdb.algo.jep;

import hec.data.RatingException;
import hec.data.cwmsRating.RatingSet;
import hec.lang.Const;
import ilex.util.Logger;

import java.util.Date;
import java.util.Stack;

import org.nfunk.jep.ParseException;
import org.nfunk.jep.function.PostfixMathCommand;

import decodes.cwms.CwmsTimeSeriesDb;
import decodes.cwms.rating.CwmsRatingDao;

/**
 * This implements the rating function in the Expression Parser.
 * 
 * The function takes two arguments: location and variable name.
 */
public class RatingFunction
	extends PostfixMathCommand
{
	public static final String funcName = "rating";
	private JepContext ctx = null;
	private int numParms = 0;

	public RatingFunction(JepContext ctx)
	{
		super();
		this.ctx = ctx;
		this.numberOfParameters = -1;
	}
	
	@Override
	public boolean checkNumberOfParameters(int np)
	{
		numParms = np;
//System.out.println("rating checkNumParams, np=" + np);
		// Required arg table name followed by 1...9 indeps
		return np >= 2 && np <= 10;
	}
	
	private double getArgAsDouble(Object o, int pos)
		throws ParseException
	{
		if (o instanceof Number)
			return ((Number) o).doubleValue();
		throw new ParseException("invalid independent parameter in position " + pos + ": " + o);
	}
	
	@Override
	@SuppressWarnings({ "rawtypes", "unchecked" })
	public void run(Stack inStack)
		throws ParseException
	{
//System.out.println("rating-1");
		checkStack(inStack);
//System.out.println("rating-2");
		Date tsbt = ctx.getTimeSliceBaseTime();
		if (tsbt == null)
			throw new ParseException(funcName + " can only be called from within a time-slice script.");

//System.out.println("rating-2a -- calling getNumberOfParameters()");
//
//System.out.println("rating-3, np=" + numParms);
		double valueSet[] = new double[numParms-1];


		switch(numParms)
		{
		case 10: valueSet[8] = getArgAsDouble(inStack.pop(), 9);
		case  9: valueSet[7] = getArgAsDouble(inStack.pop(), 8);
		case  8: valueSet[6] = getArgAsDouble(inStack.pop(), 7);
		case  7: valueSet[5] = getArgAsDouble(inStack.pop(), 6);
		case  6: valueSet[4] = getArgAsDouble(inStack.pop(), 5);
		case  5: valueSet[3] = getArgAsDouble(inStack.pop(), 4);
		case  4: valueSet[2] = getArgAsDouble(inStack.pop(), 3);
		case  3: valueSet[1] = getArgAsDouble(inStack.pop(), 2);
		case  2: valueSet[0] = getArgAsDouble(inStack.pop(), 1);
		}
		
		String specId = inStack.pop().toString();
//System.out.println("rating-4, spec='" + specId + "'");
		
		if (ctx.getTsdb() == null)
		{
			double sum = 0.0;
			for(double v : valueSet)
				sum += v;
			Logger.instance().warning("TEST-MODE: No database, rating returning sum of all inputs=" + sum);
			inStack.push(Double.valueOf(sum));
			return;
		}

		String what = "reading rating";
		CwmsRatingDao crd = new CwmsRatingDao((CwmsTimeSeriesDb)ctx.getTsdb());
		try
		{
			RatingSet ratingSet = crd.getRatingSet(specId);
			what = "performing rating";
			double d = ratingSet.rateOne(valueSet, tsbt.getTime());
			if (d == Const.UNDEFINED_DOUBLE)
			{
				throw new RatingException("input value(s) outside rating bounds.");
			}
			inStack.push(Double.valueOf(d));
		}
		catch (RatingException ex)
		{
			throw new ParseException("Error " + what + " for '" + specId + "': " + ex);
		}
		finally
		{
			crd.close();
		}
	}
}