com.scudata.expression.fn.Range Maven / Gradle / Ivy

Go to download
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: 20241126
package com.scudata.expression.fn;

import java.sql.Date;
import java.sql.Timestamp;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.dm.Sequence;
import com.scudata.expression.Function;
import com.scudata.expression.IParam;
import com.scudata.resources.EngineMessage;
import com.scudata.util.Variant;

/**
 * ???ָ??????εĶ?ͷ????һ??ͷ????ʱ??/????
 * range(s,e,k:n)
 * ??s,e֮??ȷֳ?n?ݣ????ص?k?κ͵?k+1?εĶ?ͷ?????ؽ??Ϊ?????У?????s,e?????????;??????????ݾ?ȷ?ȣ?
 * date?;?ȷ???죬datetime?;?ȷ???룬s,e????????????????kʡ?Ե?ʱ?򷵻????еķֶ?ͷ?Լ????ݶε?ͷβ???ݣ?
 * ?????ؽ??Ϊn+1???С?
 * @author runqian
 *
 */
public class Range extends Function {
	/**
	 * ??????ʽ????Ч?ԣ???Ч???׳??쳣
	 */
	public void checkValidity() {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("range" + mm.getMessage("function.missingParam"));
		} else if (param.getSubSize() != 3) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("range" + mm.getMessage("function.invalidParam"));
		}
	}

	public Object calculate(Context ctx) {
		IParam startParam = param.getSub(0);
		IParam endParam = param.getSub(1);
		IParam segParam = param.getSub(2);
		if (startParam == null || endParam == null || segParam == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("range" + mm.getMessage("function.invalidParam"));
		}
		
		Object startVal = startParam.getLeafExpression().calculate(ctx);
		Object endVal = endParam.getLeafExpression().calculate(ctx);
		if (Variant.compare(startVal, endVal) >= 0) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("range" + mm.getMessage("function.invalidParam"));
		}
		
		int k = -1;
		int n = 1;
		
		if (segParam.isLeaf()) {
			Object obj = segParam.getLeafExpression().calculate(ctx);
			if (obj instanceof Number) {
				n = ((Number)obj).intValue();
			} else {
				MessageManager mm = EngineMessage.get();
				throw new RQException("range" + mm.getMessage("function.paramTypeError"));
			}
		} else {
			if (segParam.getSubSize() != 2) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("range" + mm.getMessage("function.invalidParam"));
			}
			
			IParam sub0 = segParam.getSub(0);
			IParam sub1 = segParam.getSub(1);
			if (sub0 == null || sub1 == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("range" + mm.getMessage("function.invalidParam"));
			}
			
			Object obj0 = sub0.getLeafExpression().calculate(ctx);
			if (!(obj0 instanceof Number)) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("range" + mm.getMessage("function.paramTypeError"));
			}
			
			k = ((Number)obj0).intValue();
			Object obj1 = sub1.getLeafExpression().calculate(ctx);
			if (!(obj1 instanceof Number)) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("range" + mm.getMessage("function.paramTypeError"));
			}
			
			n = ((Number)obj1).intValue();
			if (k < 1 || k > n) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("range" + mm.getMessage("function.invalidParam"));
			}
		}

		if (k == -1) {
			return range(startVal, endVal, n);
		} else {
			return range(startVal, endVal, k, n);
		}
	}
	
	private Sequence range(Object startVal, Object endVal, int n) {
		if (n <= 1) {
			Sequence seq = new Sequence(2);
			seq.add(startVal);
			seq.add(endVal);
			return seq;
		}
		
		if (startVal instanceof Number && endVal instanceof Number) {
			long len = ((Number)endVal).longValue() - ((Number)startVal).longValue();
			if (len < n) {
				n = (int)len;
				Sequence seq = new Sequence(n + 1);
				seq.add(startVal);
				
				for (int i = 1; i < n; ++i) {
					Object val = Variant.add(startVal, i);
					seq.add(val);
				}
				
				seq.add(endVal);
				return seq;
			} else {
				long avg = len / n;
				long mod = len % n;
				Sequence seq = new Sequence(n + 1);
				seq.add(startVal);
				
				for (int i = 1; i < n; ++i) {
					if (i <= mod) {
						Object val = Variant.add(startVal, avg * i + i);
						seq.add(val);
					} else {
						Object val = Variant.add(startVal, avg * i + mod);
						seq.add(val);
					}
				}
				
				seq.add(endVal);
				return seq;
			}
		} else if (startVal instanceof Date && endVal instanceof Date) {
			Date startDate = (Date)startVal;
			Date endDate = (Date)endVal;
			int len = (int)Variant.dayInterval(startDate, endDate);
			
			if (len < n) {
				n = (int)len;
				Sequence seq = new Sequence(n + 1);
				seq.add(startVal);
				
				for (int i = 1; i < n; ++i) {
					Object val = Variant.elapse(startDate, i, null);
					seq.add(val);
				}
				
				seq.add(endVal);
				return seq;
			} else {
				int avg = len / n;
				int mod = len % n;
				Sequence seq = new Sequence(n + 1);
				seq.add(startVal);
				
				for (int i = 1; i < n; ++i) {
					if (i <= mod) {
						Object val = Variant.elapse(startDate, avg * i + i, null);
						seq.add(val);
					} else {
						Object val = Variant.elapse(startDate, avg * i + mod, null);
						seq.add(val);
					}
				}
				
				seq.add(endVal);
				return seq;
			}
		} else if (startVal instanceof Timestamp && endVal instanceof Timestamp) {
			Timestamp startDate = (Timestamp)startVal;
			Timestamp endDate = (Timestamp)endVal;
			int len = (int)Variant.secondInterval(startDate, endDate);
			
			if (len < n) {
				n = (int)len;
				Sequence seq = new Sequence(n + 1);
				seq.add(startVal);
				
				for (int i = 1; i < n; ++i) {
					Object val = Variant.elapse(startDate, i, "s");
					seq.add(val);
				}
				
				seq.add(endVal);
				return seq;
			} else {
				int avg = len / n;
				int mod = len % n;
				Sequence seq = new Sequence(n + 1);
				seq.add(startVal);
				
				for (int i = 1; i < n; ++i) {
					if (i <= mod) {
						Object val = Variant.elapse(startDate, avg * i + i, "s");
						seq.add(val);
					} else {
						Object val = Variant.elapse(startDate, avg * i + mod, "s");
						seq.add(val);
					}
				}
				
				seq.add(endVal);
				return seq;
			}
		}
		
		MessageManager mm = EngineMessage.get();
		throw new RQException("range" + mm.getMessage("function.paramTypeError"));
	}
	
	private Sequence range(Object startVal, Object endVal, int k, int n) {
		Sequence total = range(startVal, endVal, n);
		int len = total.length();
		
		if (k < len) {
			Sequence seq = new Sequence(2);
			seq.add(total.getMem(k));
			seq.add(total.getMem(k + 1));
			return seq;
		} else {
			return null;
		}
	}
}