com.scudata.expression.mfn.RowField 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;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.BaseRecord;
import com.scudata.dm.Context;
import com.scudata.dm.DataStruct;
import com.scudata.dm.Sequence;
import com.scudata.expression.IParam;
import com.scudata.expression.MemberFunction;
import com.scudata.resources.EngineMessage;


/**
 * ?????к?ȡ???еļ?¼???ֶ?
 * k.r(T), k.r(T, F)
 * @author RunQian
 *
 */
public class RowField extends MemberFunction {
	private int r;
	private String field;
	private DataStruct ds;
	private int col;
	
	/**
	 * ??????ʽ????Ч?ԣ???Ч???׳??쳣
	 */
	public void checkValidity() {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("r" + mm.getMessage("function.missingParam"));
		}
	}

	public boolean isLeftTypeMatch(Object obj) {
		return obj instanceof Number;
	}
	
	public void setDotLeftObject(Object obj) {
		r = ((Number)obj).intValue();
	}
	
	/**
	 * ?ͷŽڵ????õĵ?????????Ķ???
	 */
	public void releaseDotLeftObject() {
	}
	
	public Object calculate(Context ctx) {
		if (param.isLeaf()) {
			Object obj = param.getLeafExpression().calculate(ctx);
			if (obj instanceof Sequence) {
				return ((Sequence)obj).getMem(r);
			} else if (obj == null) {
				return null;
			} else {
				MessageManager mm = EngineMessage.get();
				throw new RQException("r" + mm.getMessage("function.paramTypeError"));
			}
		} else {
			if (param.getSubSize() != 2) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("r" + mm.getMessage("function.invalidParam"));
			}
			
			IParam sub0 = param.getSub(0);
			IParam sub1 = param.getSub(1);
			if (sub0 == null || sub1 == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("r" + mm.getMessage("function.invalidParam"));
			}
			
			Object obj = sub0.getLeafExpression().calculate(ctx);
			if (obj instanceof Sequence) {
				obj = ((Sequence)obj).getMem(r);
				if (obj instanceof BaseRecord) {
					BaseRecord r = (BaseRecord)obj;
					if (field == null) {
						field = sub1.getLeafExpression().getIdentifierName();
						ds = r.dataStruct();
						col = ds.getFieldIndex(field);
						if (col == -1) {
							MessageManager mm = EngineMessage.get();
							throw new RQException(field + mm.getMessage("ds.fieldNotExist"));
						}
						
						return r.getNormalFieldValue(col);
					} else {
						DataStruct curDs = r.dataStruct();
						if (curDs != ds) {
							ds = curDs;
							col = curDs.getFieldIndex(field);
							if (col == -1) {
								MessageManager mm = EngineMessage.get();
								throw new RQException(field + mm.getMessage("ds.fieldNotExist"));
							}
						}
						
						return r.getNormalFieldValue(col);
					}
				} else if (obj == null) {
					return null;
				} else {
					MessageManager mm = EngineMessage.get();
					throw new RQException("r" + mm.getMessage("function.paramTypeError"));
				}
			} else if (obj == null) {
				return null;
			} else {
				MessageManager mm = EngineMessage.get();
				throw new RQException("r" + mm.getMessage("function.paramTypeError"));
			}
		}
	}
}