com.scudata.expression.fn.gather.Iterate Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of esproc Show documentation
Show all versions of esproc Show documentation
SPL(Structured Process Language) A programming language specially for structured data computing.
package com.scudata.expression.fn.gather;
import com.scudata.array.IArray;
import com.scudata.array.ObjectArray;
import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.dm.Current;
import com.scudata.dm.Env;
import com.scudata.dm.Param;
import com.scudata.expression.Expression;
import com.scudata.expression.Gather;
import com.scudata.expression.IParam;
import com.scudata.expression.Node;
import com.scudata.resources.EngineMessage;
import com.scudata.util.Variant;
/**
* ???????㺯??
* iterate(x,a) iterate(x,a;Gi,??)
* @author WangXiaoJun
*
*/
public class Iterate extends Gather {
private Expression exp;
private Expression initExp;
private Param valParam;
private Expression []gexps;
private Current prevCurrent;
private Object prevVal;
private Object []prevGroupVals;
public Node optimize(Context ctx) {
if (param != null) param.optimize(ctx);
return this;
}
public void prepare(Context ctx) {
if (param == null) {
MessageManager mm = EngineMessage.get();
throw new RQException("iterate" + mm.getMessage("function.missingParam"));
} else if (param.isLeaf()) {
exp = param.getLeafExpression();
} else {
if (param.getSubSize() != 2) {
MessageManager mm = EngineMessage.get();
throw new RQException("iterate" + mm.getMessage("function.invalidParam"));
}
IParam sub0 = param.getSub(0);
if (sub0 == null || !sub0.isLeaf()) {
MessageManager mm = EngineMessage.get();
throw new RQException("iterate" + mm.getMessage("function.invalidParam"));
}
exp = sub0.getLeafExpression();
IParam sub1 = param.getSub(1);
if (sub1 != null) {
if (!sub1.isLeaf()) {
MessageManager mm = EngineMessage.get();
throw new RQException("iterate" + mm.getMessage("function.invalidParam"));
}
initExp = sub1.getLeafExpression();
}
}
valParam = ctx.getIterateParam();
}
public Object gather(Context ctx) {
if (initExp == null) {
valParam.setValue(null);
} else {
valParam.setValue(initExp.calculate(ctx));
}
return exp.calculate(ctx);
}
public Object gather(Object oldValue, Context ctx) {
valParam.setValue(oldValue);
return exp.calculate(ctx);
}
public Expression getRegatherExpression(int q) {
MessageManager mm = EngineMessage.get();
throw new RQException(mm.getMessage("engine.invalidLoopsGroups"));
}
private void prepare(IParam param, Context ctx) {
if (param == null) {
MessageManager mm = EngineMessage.get();
throw new RQException("iterate" + mm.getMessage("function.missingParam"));
}
if (param.getType() == IParam.Semicolon) {
if (param.getSubSize() != 2) {
MessageManager mm = EngineMessage.get();
throw new RQException("iterate" + 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("iterate" + mm.getMessage("function.invalidParam"));
}
if (sub1.isLeaf()) {
gexps = new Expression[]{sub1.getLeafExpression()};
} else {
int size = sub1.getSubSize();
gexps = new Expression[size];
for (int i = 0; i < size; ++i) {
IParam sub = sub1.getSub(i);
if (sub == null) {
MessageManager mm = EngineMessage.get();
throw new RQException("iterate" + mm.getMessage("function.invalidParam"));
}
gexps[i] = sub.getLeafExpression();
}
}
param = sub0;
}
if (param.isLeaf()) {
exp = param.getLeafExpression();
} else {
if (param.getSubSize() != 2) {
MessageManager mm = EngineMessage.get();
throw new RQException("iterate" + mm.getMessage("function.invalidParam"));
}
IParam sub0 = param.getSub(0);
if (sub0 == null || !sub0.isLeaf()) {
MessageManager mm = EngineMessage.get();
throw new RQException("iterate" + mm.getMessage("function.invalidParam"));
}
exp = sub0.getLeafExpression();
IParam sub1 = param.getSub(1);
if (sub1 != null) {
if (!sub1.isLeaf()) {
MessageManager mm = EngineMessage.get();
throw new RQException("iterate" + mm.getMessage("function.invalidParam"));
}
initExp = sub1.getLeafExpression();
}
}
valParam = ctx.getIterateParam();
}
public Object calculate(Context ctx) {
Current current = ctx.getComputeStack().getTopCurrent();
if (valParam == null) {
// ????ִ??
prepare(param, ctx);
Object oldVal = valParam.getValue();
if (initExp == null) {
valParam.setValue(null);
} else {
valParam.setValue(initExp.calculate(ctx));
}
if (gexps != null) {
int gcount = gexps.length;
prevGroupVals = new Object[gcount];
for (int i = 0; i < gcount; ++i) {
prevGroupVals[i] = gexps[i].calculate(ctx);
}
}
prevVal = exp.calculate(ctx);
valParam.setValue(oldVal);
} else {
Object oldVal = valParam.getValue();
// ?ж?ջ???????еĵ?ǰ????Ƿ???1????????????ó?ʼֵ
if (current != null && current != prevCurrent) {
if (initExp == null) {
valParam.setValue(null);
} else {
valParam.setValue(initExp.calculate(ctx));
}
if (gexps != null) {
int gcount = gexps.length;
for (int i = 0; i < gcount; ++i) {
prevGroupVals[i] = gexps[i].calculate(ctx);
}
}
} else {
if (gexps == null) {
valParam.setValue(prevVal);
} else {
boolean isSame = true;
int gcount = gexps.length;
for (int i = 0; i < gcount; ++i) {
Object val = gexps[i].calculate(ctx);
if (!Variant.isEquals(prevGroupVals[i], val)) {
isSame = false;
prevGroupVals[i] = val;
for (++i; i < gcount; ++i) {
prevGroupVals[i] = gexps[i].calculate(ctx);
}
break;
}
}
if (isSame) {
valParam.setValue(prevVal);
} else if (initExp == null) {
valParam.setValue(null);
} else {
valParam.setValue(initExp.calculate(ctx));
}
}
}
prevVal = exp.calculate(ctx);
valParam.setValue(oldVal);
}
prevCurrent = current;
return prevVal;
}
/**
* ???????м?¼??ֵ?????ܵ??????????
* @param result ???????
* @param resultSeqs ÿ????¼??Ӧ?Ľ??????????
* @param ctx ??????????
* @return IArray ???????
*/
public IArray gather(IArray result, int []resultSeqs, Context ctx) {
if (result == null) {
result = new ObjectArray(Env.INITGROUPSIZE);
}
Expression exp = this.exp;
Expression initExp = this.initExp;
Param valParam = this.valParam;
Current current = ctx.getComputeStack().getTopCurrent();
int len = current.length();
for (int i = 1; i <= len; ++i) {
current.setCurrent(i);
if (result.size() < resultSeqs[i]) {
if (initExp == null) {
valParam.setValue(null);
} else {
valParam.setValue(initExp.calculate(ctx));
}
result.add(exp.calculate(ctx));
} else {
valParam.setValue(result.get(resultSeqs[i]));
result.set(resultSeqs[i], exp.calculate(ctx));
}
}
return result;
}
}