com.scudata.expression.mfn.sequence.MJoin 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.SequenceFunction;
import com.scudata.resources.EngineMessage;

/**
 * ????????????ʽ????
 * A.mjoin(Ki,??,K;B:F, Ki:??,K:a:b)
 * A.mjoin(K;B,K:a:b,x:F,??)
 * @author RunQian
 *
 */
public class MJoin extends SequenceFunction {
	public Object calculate(Context ctx) {
		if (param == null || param.getSubSize() != 2) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mjoin" + mm.getMessage("function.missingParam"));
		} else if (param.getType() != IParam.Semicolon) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
		}
		
		IParam keyParam = param.getSub(0);
		IParam tableParam = param.getSub(1);
		if (keyParam == null || tableParam == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
		}
		
		//int keyCount;
		Expression []srcKeyExps;
		
		if (keyParam.isLeaf()) {
			//keyCount = 1;
			srcKeyExps = new Expression[] {keyParam.getLeafExpression()};
		} else {
//			keyCount = keyParam.getSubSize();
//			srcKeyExps = new Expression[keyCount];
//			
//			for (int i = 0; i < keyCount; ++i) {
//				IParam sub = keyParam.getSub(i);
//				if (sub == null) {
//					MessageManager mm = EngineMessage.get();
//					throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
//				}
//				
//				srcKeyExps[i] = sub.getLeafExpression();
//			}
			MessageManager mm = EngineMessage.get();
			throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
		}
		
		//B:F, Ki:??,K:a:b
		Sequence table = null;
		Expression keyExp;
		Object from, to = null;
		if (tableParam == null || tableParam.getType() != IParam.Comma || tableParam.getSubSize() < 3) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
		}
		IParam sub = tableParam.getSub(0);
		if (sub == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
		} else if (sub.isLeaf()) {
//			IParam sub0 = sub.getSub(0);
//			IParam sub1 = sub.getSub(1);
//			if (sub0 == null || sub1 == null) {
//				MessageManager mm = EngineMessage.get();
//				throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
//			}
			
			Object obj = sub.getLeafExpression().calculate(ctx);
			if (obj instanceof Sequence) {
				table = (Sequence)obj;
			} else if (obj != null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("pjoin" + mm.getMessage("function.paramTypeError"));
			}
			//name = sub1.getLeafExpression().toString();
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
		}
		
		keyParam = tableParam.getSub(1);
		if (keyParam == null || keyParam.isLeaf() || keyParam.getSubSize() < 3) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
		} else {
			int size = keyParam.getSubSize();
			if (size != 2 && size != 3) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("mjoin" + mm.getMessage("function.paramCountNotMatch"));
			}
			
			sub = keyParam.getSub(0);
			if (sub == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
			}
			
			keyExp = sub.getLeafExpression();
			
			sub = keyParam.getSub(1);
			if (sub == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
			}
			from = sub.getLeafExpression().calculate(ctx);
			sub = keyParam.getSub(2);
			if (sub != null) {
				to = sub.getLeafExpression().calculate(ctx);
			}
		}
		
		int size = tableParam.getSubSize() - 2;
		Expression[] newExps = new Expression[size];
		String[] newNames = new String[size];
		for (int i = 0; i < size; i++) {
			IParam param = tableParam.getSub(2 + i);
			if (param == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
			} else if (param.isLeaf()) {
				newExps[i] = param.getLeafExpression();
				newNames[i] = newExps[i].getIdentifierName();
			} else if (param.getType() == IParam.Colon && param.getSubSize() == 2) {
				IParam sub0 = param.getSub(0);
				if (sub0 == null) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
				}
				newExps[i] = sub0.getLeafExpression();
				
				IParam sub1 = param.getSub(1);
				if (sub1 == null) {
					newNames[i] = newExps[i].getIdentifierName();
				} else {
					newNames[i] = sub1.getLeafExpression().getIdentifierName();
				}
			} else {
				MessageManager mm = EngineMessage.get();
				throw new RQException("mjoin" + mm.getMessage("function.invalidParam"));
			}
		}
		
		
		return srcSequence.mjoin(srcKeyExps[0], table, keyExp, from, to, newExps, newNames, option, ctx);
	}
}