com.scudata.expression.fn.gather.Maxp 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.BaseRecord;
import com.scudata.dm.ComputeStack;
import com.scudata.dm.Context;
import com.scudata.dm.DataStruct;
import com.scudata.dm.Env;
import com.scudata.dm.Sequence;
import com.scudata.expression.Expression;
import com.scudata.expression.Gather;
import com.scudata.resources.EngineMessage;
import com.scudata.util.Variant;
/**
* ȡʹ????ʽȡֵ??????????¼
* maxp(x)
* @author RunQian
*
*/
public class Maxp extends Gather {
private Expression exp;
private int fieldIndex = Integer.MIN_VALUE; // ????ʽ????Ӧ???ֶ?
private boolean isOne;
private boolean needFinish = true;
public Object calculate(Context ctx) {
MessageManager mm = EngineMessage.get();
throw new RQException(mm.getMessage("Expression.unknownFunction") + "minp");
}
public void prepare(Context ctx) {
if (param == null) {
needFinish = false;
} else if (param.isLeaf()) {
exp = param.getLeafExpression();
} else {
MessageManager mm = EngineMessage.get();
throw new RQException("maxp" + mm.getMessage("function.invalidParam"));
}
isOne = option == null || option.indexOf('a') == -1;
}
public Expression getRegatherExpression(int q) {
String f = "#" + q;
if (exp == null) {
String str = "top@1(-1, 1," + f + ")";
return new Expression(str);
}
//maxp(x):f
//top@1(-1,~.x,f)
if (isOne) {
String str = "top@1(-1," + "~.(" + exp.toString() + ")," + f + ")";
return new Expression(str);
} else {
String str = "top@2(-1," + "~.(" + exp.toString() + ")," + f + ")";
return new Expression(str);
}
}
public boolean needFinish() {
return needFinish;
}
public boolean needFinish1() {
return needFinish;
}
public Object finish1(Object val) {
if (val != null) {
Object []array = (Object[])val;
return array[1];
} else {
return null;
}
}
public Object finish(Object val) {
if (val != null) {
Object []array = (Object[])val;
return array[1];
} else {
return null;
}
}
public Object gather(Context ctx) {
if (exp == null) {
return ctx.getComputeStack().getTopObject().getCurrent();
}
Object val = exp.calculate(ctx);
Object r = ctx.getComputeStack().getTopObject().getCurrent();
if (isOne) {
return new Object[] {val, r};
} else {
Sequence seq = new Sequence();
seq.add(r);
return new Object[] {val, seq};
}
}
public Object gather(Object oldValue, Context ctx) {
if (exp == null) {
return ctx.getComputeStack().getTopObject().getCurrent();
}
Object val = exp.calculate(ctx);
if (val == null) {
return oldValue;
}
Object []array = (Object[])oldValue;
int cmp = Variant.compare(val, array[0], true);
if (cmp > 0) {
array[0] = val;
Object r = ctx.getComputeStack().getTopObject().getCurrent();
if (isOne) {
array[1] = r;
} else {
Sequence seq = (Sequence)array[1];
seq.clear();
seq.add(r);
}
} else if (cmp == 0 && !isOne) {
Object r = ctx.getComputeStack().getTopObject().getCurrent();
((Sequence)array[1]).add(r);
}
return oldValue;
}
/**
* ???????м?¼??ֵ?????ܵ??????????
* @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);
}
ComputeStack computeStack = ctx.getComputeStack();
Sequence src = computeStack.getTopSequence();
IArray mems = src.getMems();
needFinish = false;
if (exp == null) {
for (int i = 1, len = mems.size(); i <= len; ++i) {
if (result.size() < resultSeqs[i]) {
result.add(mems.get(i));
} else {
result.set(resultSeqs[i], mems.get(i));
}
}
return result;
}
int fieldIndex = this.fieldIndex;
if (fieldIndex == Integer.MIN_VALUE) {
DataStruct ds = src.dataStruct();
if (ds != null) {
this.fieldIndex = fieldIndex = exp.getFieldIndex(ds);
}
}
boolean isOne = this.isOne;
IArray array = exp.calculateAll(ctx);
if (fieldIndex > -1) {
for (int i = 1, len = array.size(); i <= len; ++i) {
if (result.size() < resultSeqs[i]) {
if (array.isNull(i)) {
result.add(null);
} else if (isOne) {
result.add(mems.get(i));
} else {
Sequence seq = new Sequence();
seq.add(mems.get(i));
result.add(seq);
}
} else if (!array.isNull(i)) {
if (isOne) {
BaseRecord r = (BaseRecord)result.get(resultSeqs[i]);
if (r == null || r.compare(fieldIndex, array, i) < 0) {
result.set(resultSeqs[i], mems.get(i));
}
} else {
Sequence seq = (Sequence)result.get(resultSeqs[i]);
if (seq == null) {
seq = new Sequence();
seq.add(mems.get(i));
result.set(resultSeqs[i], seq);
} else {
BaseRecord r = (BaseRecord)seq.get(1);
int cmp = r.compare(fieldIndex, array, i);
if(cmp < 0) {
seq.clear();
seq.add(mems.get(i));
} else if (cmp == 0) {
seq.add(mems.get(i));
}
}
}
}
}
} else {
Expression exp = this.exp;
for (int i = 1, len = array.size(); i <= len; ++i) {
if (result.size() < resultSeqs[i]) {
if (array.isNull(i)) {
result.add(null);
} else if (isOne) {
result.add(mems.get(i));
} else {
Sequence seq = new Sequence();
seq.add(mems.get(i));
result.add(seq);
}
} else if (!array.isNull(i)) {
if (isOne) {
BaseRecord r = (BaseRecord)result.get(resultSeqs[i]);
if (r == null || array.compareTo(i, r.calc(exp, ctx)) > 0) {
result.set(resultSeqs[i], mems.get(i));
}
} else {
Sequence seq = (Sequence)result.get(resultSeqs[i]);
if (seq == null) {
seq = new Sequence();
seq.add(mems.get(i));
result.set(resultSeqs[i], seq);
} else {
BaseRecord r = (BaseRecord)seq.get(1);
int cmp = array.compareTo(i, r.calc(exp, ctx));
if(cmp > 0) {
seq.clear();
seq.add(mems.get(i));
} else if (cmp == 0) {
seq.add(mems.get(i));
}
}
}
}
}
}
return result;
}
/**
* ??̷̳???Ķ??λ???????
* @param result һ???̵߳ķ?????
* @param result2 ??һ???̵߳ķ?????
* @param seqs ??һ???̵߳ķ??????һ???̷߳???Ķ?Ӧ??ϵ
* @param ctx ??????????
* @return
*/
public void gather2(IArray result, IArray result2, int []seqs, Context ctx) {
int fieldIndex = this.fieldIndex;
if (exp == null) {
for (int i = 1, len = result2.size(); i <= len; ++i) {
if (seqs[i] != 0) {
result.set(seqs[i], result2.get(i));
}
}
} else if (fieldIndex > -1) {
if (isOne) {
for (int i = 1, len = result2.size(); i <= len; ++i) {
if (seqs[i] != 0) {
BaseRecord r1 = (BaseRecord)result.get(seqs[i]);
BaseRecord r2 = (BaseRecord)result2.get(i);
if (r1 == null) {
result.set(seqs[i], r2);
} else if (r2 != null && r1.compare(r2, fieldIndex) < 0) {
result.set(seqs[i], r2);
}
}
}
} else {
for (int i = 1, len = result2.size(); i <= len; ++i) {
if (seqs[i] != 0) {
Sequence sequence1 = (Sequence)result.get(seqs[i]);
Sequence sequence2 = (Sequence)result2.get(i);
if (sequence1 == null) {
result.set(seqs[i], sequence2);
} else if (sequence2 != null) {
BaseRecord r1 = (BaseRecord)sequence1.getMem(1);
BaseRecord r2 = (BaseRecord)sequence2.getMem(1);
int cmp = r1.compare(r2, fieldIndex);
if (cmp < 0) {
result.set(seqs[i], sequence2);
} else if (cmp == 0) {
sequence1.addAll(sequence2);
}
}
}
}
}
} else {
Expression exp = this.exp;
if (isOne) {
for (int i = 1, len = result2.size(); i <= len; ++i) {
if (seqs[i] != 0) {
BaseRecord r1 = (BaseRecord)result.get(seqs[i]);
BaseRecord r2 = (BaseRecord)result2.get(i);
if (r1 == null) {
result.set(seqs[i], r2);
} else if (r2 != null && Variant.compare(r1.calc(exp, ctx), r2.calc(exp, ctx), true) < 0) {
result.set(seqs[i], r2);
}
}
}
} else {
for (int i = 1, len = result2.size(); i <= len; ++i) {
if (seqs[i] != 0) {
Sequence sequence1 = (Sequence)result.get(seqs[i]);
Sequence sequence2 = (Sequence)result2.get(i);
if (sequence1 == null) {
result.set(seqs[i], sequence2);
} else if (sequence2 != null) {
BaseRecord r1 = (BaseRecord)sequence1.getMem(1);
BaseRecord r2 = (BaseRecord)sequence2.getMem(1);
int cmp = Variant.compare(r1.calc(exp, ctx), r2.calc(exp, ctx), true);
if (cmp < 0) {
result.set(seqs[i], sequence2);
} else if (cmp == 0) {
sequence1.addAll(sequence2);
}
}
}
}
}
}
}
}