com.scudata.expression.mfn.sequence.Pivot Maven / Gradle / Ivy

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

SPL(Structured Process Language) A programming language specially for structured data computing.

There is a newer version: 20240823

package com.scudata.expression.mfn.sequence;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.expression.Expression;
import com.scudata.expression.IParam;
import com.scudata.expression.ParamInfo2;
import com.scudata.expression.SequenceFunction;
import com.scudata.resources.EngineMessage;

/**
 * ????????????ת??
 * A.pivot(g:G,??;F,V;Ni:N'i,??)
 * @author RunQian
 *
 */
public class Pivot extends SequenceFunction {
	/**
	 * ??????ʽ????Ч?ԣ???Ч???׳??쳣
	 */
	public void checkValidity() {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("pivot" + mm.getMessage("function.missingParam"));
		} else if (param.getType() != IParam.Semicolon) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("pivot" + mm.getMessage("function.invalidParam"));
		}
	}

	public Object calculate(Context ctx) {
		int size = param.getSubSize();
		if (size > 3) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("pivot" + mm.getMessage("function.invalidParam"));
		}

		Expression []gexps = null;
		String []gnames = null;
		Expression fexp;
		Expression vexp; 
		Expression []nexps = null;
		Object []nameObjects = null;
		
		IParam sub0 = param.getSub(0);
		if (sub0 != null) {
			ParamInfo2 pi = ParamInfo2.parse(sub0, "pivot", true, false);
			gexps = pi.getExpressions1();
			gnames = pi.getExpressionStrs2();
		}
				
		IParam sub1 = param.getSub(1);
		if (sub1 == null || sub1.getSubSize() != 2) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("pivot" + mm.getMessage("function.invalidParam"));
		}
		
		IParam sub = sub1.getSub(0);
		if (sub == null || !sub.isLeaf()) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("pivot" + mm.getMessage("function.invalidParam"));
		}
		
		fexp = sub.getLeafExpression();
		sub = sub1.getSub(1);
		if (sub == null || !sub.isLeaf()) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("pivot" + mm.getMessage("function.invalidParam"));
		}
		
		vexp = sub.getLeafExpression();
		if (size == 3 && param.getSub(2) != null) {
			IParam sub2 = param.getSub(2);
			ParamInfo2 pi = ParamInfo2.parse(sub2, "pivot", false, false);
			nexps = pi.getExpressions1();
			nameObjects = pi.getValues2(ctx);
		}
		
		if (option == null || option.indexOf('r') == -1) {
			return srcSequence.pivot(gexps, gnames, fexp, vexp, nexps, nameObjects, option, ctx);
		} else {
			String fname = fexp.getIdentifierName();
			String vname = vexp.getIdentifierName();
			return srcSequence.unpivot(gexps, gnames, fname, vname, nexps, nameObjects, ctx);
		}
	}
}