com.scudata.expression.mfn.sequence.Groupc 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.dm.Sequence;
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.groupc(g;v) A.groupc@r(g;v;k)
 * ????????????ת??P.groupc(g:G,??;F,??;N,??) P.groupc@r(g:G,??;F,??;N,??)
 * @author RunQian
 *
 */
public class Groupc extends SequenceFunction {
	/**
	 * ??????ʽ????Ч?ԣ???Ч???׳??쳣
	 */
	public void checkValidity() {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("groupc" + mm.getMessage("function.missingParam"));
		} else if (param.getType() != IParam.Semicolon) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("groupc" + mm.getMessage("function.invalidParam"));
		}
	}

	private Object groupcSequence(Context ctx) {
		if (option == null || option.indexOf('r') == -1) {
			if (param.getSubSize() != 2) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("groupc" + mm.getMessage("function.invalidParam"));
			}
			
			IParam sub = param.getSub(0);
			if (sub == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("groupc" + mm.getMessage("function.invalidParam"));
			}
			
			Expression gexp = sub.getLeafExpression();
			sub = param.getSub(1);
			if (sub == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("groupc" + mm.getMessage("function.invalidParam"));
			}
			
			Expression vexp = sub.getLeafExpression();
			return srcSequence.groupc(gexp, vexp, ctx);
		} else {
			if (param.getSubSize() != 3) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("groupc" + mm.getMessage("function.invalidParam"));
			}
			
			IParam sub = param.getSub(0);
			if (sub == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("groupc" + mm.getMessage("function.invalidParam"));
			}
			
			Expression gexp = sub.getLeafExpression();
			sub = param.getSub(1);
			if (sub == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("groupc" + mm.getMessage("function.invalidParam"));
			}
			
			Expression vexp = sub.getLeafExpression();
			sub = param.getSub(2);
			if (sub == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("groupc" + mm.getMessage("function.invalidParam"));
			}
			
			Object value = sub.getLeafExpression().calculate(ctx);
			if (!(value instanceof Number)) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("groupc" + mm.getMessage("function.paramTypeError"));
			}
			
			int k = ((Number)value).intValue();
			return srcSequence.ungroupc(gexp, vexp, k, ctx);
		}
	}
	
	private Object groupcTable(Context ctx) {
		Expression[] gexps;
		String []gnames;
		Expression[] vexps = null;
		String []newNames = null;
		
		if (param.getType() == IParam.Semicolon) {
			int size = param.getSubSize();
			if (size > 3) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("groupc" + mm.getMessage("function.invalidParam"));
			}
			
			IParam sub = param.getSub(0);
			if (sub == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("groupc" + mm.getMessage("function.invalidParam"));
			}
			
			ParamInfo2 pi = ParamInfo2.parse(sub, "groupc", true, false);
			gexps = pi.getExpressions1();
			gnames = pi.getExpressionStrs2();
			
			sub = param.getSub(1);
			if (sub != null) {
				vexps = sub.toArray("groupc", false);
			}
			
			if (size > 2 && param.getSub(2) != null) {
				sub = param.getSub(2);
				newNames = sub.toIdentifierNames("groupc");
			}
		} else {
			ParamInfo2 pi = ParamInfo2.parse(param, "groupc", true, false);
			gexps = pi.getExpressions1();
			gnames = pi.getExpressionStrs2();
		}
		
		if (option == null || option.indexOf('r') == -1) {
			return srcSequence.groupc(gexps, gnames, vexps, newNames, ctx);
		} else {
			return srcSequence.ungroupc(gexps, gnames, vexps, newNames, ctx);
		}
	}

	public Object calculate(Context ctx) {
		if (srcSequence.ifn() instanceof Sequence) {
			return groupcSequence(ctx);
		} else {
			return groupcTable(ctx);
		}
	}
}