com.scudata.expression.fn.Func 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.
The newest version!
package com.scudata.expression.fn;

import com.scudata.cellset.ICellSet;
import com.scudata.cellset.INormalCell;
import com.scudata.cellset.datamodel.Command;
import com.scudata.cellset.datamodel.PgmCellSet;
import com.scudata.cellset.datamodel.PgmNormalCell;
import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.expression.CSVariable;
import com.scudata.expression.Expression;
import com.scudata.expression.Function;
import com.scudata.expression.IParam;
import com.scudata.expression.Node;
import com.scudata.resources.EngineMessage;

/**
 * ?????ӳ???
 * func(a,arg)?ڲ?ѯ?ж???????ӳ???󣬾Ϳ????????ⵥԪ???н????ӳ???ĵ??á?  
 * @author runqian
 *
 */
public class Func extends Function {
	// һ???????Ż?mѡ??
	private Expression resultExp; // ?˱???ʽΪfunc???????ڵı???ʽ
	private Expression []paramExps; // ????????ʽ????
	private INormalCell funcCell; // func???ڵĵ?Ԫ??
	//private String []argNames = null; // ??????????
	
	public class CallInfo {
		private INormalCell cell; // ???庯??ʱû???????֣??ú??????ڵĵ?Ԫ??????
		private String fnName; // ???庯??ʱ????????
		private Object []args;
		
		public CallInfo(INormalCell cell) {
			this.cell = cell;
		}
		
		public CallInfo(String fnName) {
			this.fnName = fnName;
		}
		
		public PgmCellSet getPgmCellSet() {
			return (PgmCellSet)cs;
		}
		
		public INormalCell getCell() {
			return cell;
		}

		public int getRow(){
			return cell.getRow();
		}
		
		public int getCol() {
			return cell.getCol();
		}
		
		public Object[] getArgs() {
			return args;
		}

		public void setArgs(Object[] args) {
			this.args = args;
		}

		public String getFnName() {
			return fnName;
		}
	}

	public void setParameter(ICellSet cs, Context ctx, String strParam) {
		super.setParameter(cs, ctx, strParam);
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("func" + mm.getMessage("function.missingParam"));
		}
		
		if (option != null && option.indexOf('m') != -1) {
			PgmCellSet pcs = (PgmCellSet)cs;
			Expression fnExp = null;
			
			if (param.isLeaf()) {
				fnExp = param.getLeafExpression();
			} else {
				IParam sub0 = param.getSub(0);
				if (sub0 == null) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("func" + mm.getMessage("function.invalidParam"));
				}

				fnExp = sub0.getLeafExpression();
				int size = param.getSubSize();
				paramExps = new Expression[size - 1];
				
				for (int i = 1; i < size; ++i) {
					IParam sub = param.getSub(i);
					if (sub != null) {
						paramExps[i - 1] = sub.getLeafExpression();
					}
				}
			}

			if(fnExp.getHome() instanceof CSVariable) {
				funcCell = fnExp.calculateCell(ctx);
			} else {
				return;
				//String fnName = fnExp.toString();
				//PgmCellSet.FuncInfo fi = pcs.getFuncInfo(fnName);
				//funcCell = fi.getCell();
				//argNames = fi.getArgNames();
			}
			
			int row = funcCell.getRow();
			int col = funcCell.getCol();
			int endRow = pcs.getCodeBlockEndRow(row, col);
			int colCount = pcs.getColCount();
			
			// ֻ?Ż???????????һ??????ʽ?????
			
			for (int r = row; r <= endRow; ++r) {
				for (int c = col + 1; c <= colCount; ++c) {
					PgmNormalCell cell = pcs.getPgmNormalCell(r, c);
					if (cell.isBlankCell()) {
						continue;
					} else if (resultExp != null) {
						resultExp = null;
						return;
					} else if (cell.isCommandCell()) {
						Command cmd = cell.getCommand();
						if (cmd.getType() == Command.RETURN) {
							IParam resultParam = cmd.getParam(pcs, ctx);
							if (resultParam != null && resultParam.isLeaf()) {
								resultExp = resultParam.getLeafExpression();
							} else {
								resultExp = null;
								return;
							}
						} else {
							resultExp = null;
							return;
						}
					} else if (cell.isCalculableCell()) {
						resultExp = cell.getExpression();
					} else {
						resultExp = null;
						return;
					}
				}
			}
		}
	}
	
	public Node optimize(Context ctx) {
		if (param != null) param.optimize(ctx);
		return this;
	}
	
	public Object calculate(Context ctx) {
		// ??mѡ??????Ż?
		if (resultExp != null) {
			// ???ò???ֵ
			if (paramExps != null) {
				int paramRow = funcCell.getRow();
				int paramCol = funcCell.getCol();
				int colCount = cs.getColCount();
				for (int i = 0, pcount = paramExps.length; i < pcount; ++i) {
					Object value = paramExps[i].calculate(ctx);
					cs.getCell(paramRow, paramCol).setValue(value);
					if (paramCol < colCount) {
						paramCol++;
					} else {
						break;
					}
				}
			}
			
			return resultExp.calculate(ctx);
		}
		
		CallInfo callInfo = getCallInfo(ctx);
		PgmCellSet pcs = (PgmCellSet)cs;
		INormalCell cell = callInfo.getCell();
		Object []args = callInfo.getArgs();
		
		if (cell != null) {
			return pcs.executeFunc(cell.getRow(), cell.getCol(), args, option);
		} else {
			return pcs.executeFunc(callInfo.getFnName(), args, option);
		}
	}
	
	private CallInfo getCallInfo(Expression exp, Context ctx) {
		if(exp.getHome() instanceof CSVariable) {
			INormalCell cell = exp.calculateCell(ctx);
			return new CallInfo(cell);
		} else {
			return new CallInfo(exp.toString());
		}
	}
	
	// ide????ȡ??????Ϣ???е???????
	public CallInfo getCallInfo(Context ctx) {
		CallInfo callInfo;
		if (param.isLeaf()) {
			callInfo = getCallInfo(param.getLeafExpression(), ctx);
		} else {
			IParam sub0 = param.getSub(0);
			if (sub0 == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("func" + mm.getMessage("function.invalidParam"));
			}

			int size = param.getSubSize();
			Object []args = new Object[size - 1];
			callInfo = getCallInfo(sub0.getLeafExpression(), ctx);
			callInfo.setArgs(args);
			
			for (int i = 1; i < size; ++i) {
				IParam sub = param.getSub(i);
				if (sub != null) {
					args[i - 1] = sub.getLeafExpression().calculate(ctx);
				}
			}
		}
		
		return callInfo;
	}
}