com.scudata.expression.fn.datetime.Period 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.datetime;

import java.util.Calendar;
import java.util.Date;

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.expression.Node;
import com.scudata.resources.EngineMessage;
import com.scudata.util.Variant;

/**
 * periods(s,e,i) ??ȡ??s??e???????˵㣩ÿ???i??????ʱ??ֵ???ɵ????С?
 * @author runqian
 *
 */
public class Period extends Function {
	public Node optimize(Context ctx) {
		param.optimize(ctx);
		return this;
	}

	/**
	 * ??????ʽ????Ч?ԣ???Ч???׳??쳣
	 */
	public void checkValidity() {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("period" + mm.getMessage("function.missingParam"));
		}
	}

	public Object calculate(Context ctx) {
		int size = param.getSubSize();
		if (size < 2 || size > 3) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("period" + mm.getMessage("function.invalidParam"));
		}
		
		IParam sub1 = param.getSub(0);
		IParam sub2 = param.getSub(1);
		if (sub1 == null || sub2 == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("period" + mm.getMessage("function.invalidParam"));
		}

		Object start = sub1.getLeafExpression().calculate(ctx);
		Object end = sub2.getLeafExpression().calculate(ctx);

		if (start instanceof String) {
			start = Variant.parseDate((String)start);
		}

		if (end instanceof String) {
			end = Variant.parseDate((String)end);
		}

		if (!(start instanceof Date) || !(end instanceof Date)) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("period" + mm.getMessage("function.paramTypeError"));
		}
		
		int dist = 1;
		if (size > 2) {
			IParam sub3 = param.getSub(2);
			if (sub3 != null) {
				Object distance = sub3.getLeafExpression().calculate(ctx);
				if (!(distance instanceof Number)) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("period" + mm.getMessage("function.paramTypeError"));
				}
				
				dist = ((Number)distance).intValue();
				if (dist == 0) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("period" + mm.getMessage("function.invalidParam"));
				}
			}
		}

		if (option == null || (option.indexOf('o') == -1 && option.indexOf('t') == -1)) {
			return periodA((Date) start, (Date) end, dist, option); // ????
		} else {
			return periodO((Date) start, (Date) end, dist, option); // ????????Ѯ????
		}
	}

	private Sequence periodO(Date start, Date end, int distance, String opt) {
		int field;
		if (opt == null) { // ??
			field = Calendar.DATE;
		} else if (opt.indexOf('y') != -1) { // ??
			field = Calendar.YEAR;
		} else if (opt.indexOf('q') != -1) { // ??
			field = Calendar.MONTH;
			distance *= 3;
		} else if (opt.indexOf('m') != -1) { // ??
			field = Calendar.MONTH;
		} else if (opt.indexOf('t') != -1) { // Ѯ
			field = -1;
		} else if (opt.indexOf('s') != -1) { // ??
			field = Calendar.SECOND;
		} else { // ??
			field = Calendar.DATE;
		}

		Sequence series = new Sequence();
		series.add(start);

		Calendar gc = Calendar.getInstance();
		if (distance > 0) {
			if (end.compareTo(start) < 0) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("period" + mm.getMessage("function.invalidParam"));
			}

			if (field == -1) { // Ѯ
				Date prev = start;
				gc.setTime(start);
				int day = gc.get(Calendar.DATE);

				// ?޸?dayΪѮ?ĵ?һ??
				if (day < 11) {
					day = 1;
				} else if (day < 21) {
					day = 11;
				} else {
					day = 21;
				}

				while (true) {
					gc.setTime(prev);
					for (int i = 0; i < distance; ++i) {
						day += 10;
						if (day == 31) {
							day = 1;
							gc.set(Calendar.DATE, 1);
							gc.add(Calendar.MONTH, 1);
						}
					}

					gc.set(Calendar.DATE, day);
					Date tmp = (Date)prev.clone();
					tmp.setTime(gc.getTimeInMillis());
					gc.clear();

					int cmp = tmp.compareTo(end);
					if (cmp < 0) {
						series.add(tmp);
						prev = tmp;
					} else {
						if (opt == null || opt.indexOf('x') == -1) {
							series.add(end);
						}
						
						break;
					}
				}
			} else {
				int times = 1;
				while (true) {
					gc.setTime(start);
					gc.add(field, distance * times);
					Date tmp = (Date) start.clone();
					tmp.setTime(gc.getTimeInMillis());
					gc.clear();

					int cmp = tmp.compareTo(end);
					if (cmp < 0) {
						series.add(tmp);
						times++;
					} else {
						if (opt == null || opt.indexOf('x') == -1) {
							series.add(end);
						}
						
						break;
					}
				}
			}
		} else { // distance < 0
			if (start.compareTo(end) < 0) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("period" + mm.getMessage("function.invalidParam"));
			}

			if (field == -1) { // Ѯ
				Date prev = start;
				gc.setTime(start);
				int day = gc.get(Calendar.DATE);

				// ?޸?dayΪѮ?ĵ?һ??
				if (day < 11) {
					day = 1;
				} else if (day < 21) {
					day = 11;
				} else {
					day = 21;
				}

				while (true) {
					gc.setTime(prev);
					for (int i = distance; i < 0; ++i) {
						day -= 10;
						if (day < 0) {
							day = 21;
							gc.set(Calendar.DATE, 21);
							gc.add(Calendar.MONTH, -1);
						}
					}

					gc.set(Calendar.DATE, day);
					Date tmp = (Date)prev.clone();
					tmp.setTime(gc.getTimeInMillis());
					gc.clear();

					int cmp = tmp.compareTo(end);
					if (cmp > 0) {
						series.add(tmp);
						prev = tmp;
					} else {
						if (opt == null || opt.indexOf('x') == -1) {
							series.add(end);
						}
						
						break;
					}
				}
			} else {
				int times = 1;
				while (true) {
					gc.setTime(start);
					gc.add(field, distance * times);
					Date tmp = (Date) start.clone();
					tmp.setTime(gc.getTimeInMillis());
					gc.clear();

					int cmp = tmp.compareTo(end);
					if (cmp > 0) {
						series.add(tmp);
						times++;
					} else {
						if (opt == null || opt.indexOf('x') == -1) {
							series.add(end);
						}
						
						break;
					}
				}
			}
		}
		
		return series;
	}

	private Sequence periodA(Date start, Date end, int distance, String opt) {
		Calendar gc = Calendar.getInstance();
		gc.setTime(start);

		int field;
		if (opt == null) { // ??
			field = Calendar.DATE;

			gc.set(Calendar.HOUR_OF_DAY, 0);
			gc.set(Calendar.MINUTE, 0);
			gc.set(Calendar.SECOND, 0);
			gc.set(Calendar.MILLISECOND, 0);
		} else if (opt.indexOf('y') != -1) { // ??
			field = Calendar.YEAR;

			gc.set(Calendar.DATE, 1);
			gc.set(Calendar.MONTH, Calendar.JANUARY);
			gc.set(Calendar.HOUR_OF_DAY, 0);
			gc.set(Calendar.MINUTE, 0);
			gc.set(Calendar.SECOND, 0);
			gc.set(Calendar.MILLISECOND, 0);
		} else if (opt.indexOf('q') != -1) { // ??
			field = Calendar.MONTH;
			distance *= 3;

			int month = gc.get(Calendar.MONTH);
			if (month < Calendar.APRIL) {
				month = Calendar.JANUARY;
			} else if (month < Calendar.JULY) {
				month = Calendar.APRIL;
			} else if (month < Calendar.OCTOBER) {
				month = Calendar.JULY;
			} else {
				month = Calendar.OCTOBER;
			}

			gc.set(Calendar.DATE, 1);
			gc.set(Calendar.MONTH, month);
			gc.set(Calendar.HOUR_OF_DAY, 0);
			gc.set(Calendar.MINUTE, 0);
			gc.set(Calendar.SECOND, 0);
			gc.set(Calendar.MILLISECOND, 0);
		} else if (opt.indexOf('m') != -1) { // ??
			field = Calendar.MONTH;

			gc.set(Calendar.DATE, 1);
			gc.set(Calendar.HOUR_OF_DAY, 0);
			gc.set(Calendar.MINUTE, 0);
			gc.set(Calendar.SECOND, 0);
			gc.set(Calendar.MILLISECOND, 0);
		} else if (opt.indexOf('s') != -1) { // ??
			field = Calendar.SECOND;
			gc.set(Calendar.MILLISECOND, 0);
		} else { // ??
			field = Calendar.DATE;

			gc.set(Calendar.HOUR_OF_DAY, 0);
			gc.set(Calendar.MINUTE, 0);
			gc.set(Calendar.SECOND, 0);
			gc.set(Calendar.MILLISECOND, 0);
		}

		long startTime = gc.getTimeInMillis();
		gc.clear();

		Sequence series = new Sequence();
		series.add(start);
		int times = 1;

		if (distance > 0) {
			if (end.compareTo(start) < 0) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("period" + mm.getMessage("function.invalidParam"));
			}

			while (true) {
				gc.setTimeInMillis(startTime);
				gc.add(field, distance * times);
				Date tmp = (Date)start.clone();
				tmp.setTime(gc.getTimeInMillis());
				gc.clear();

				int cmp = tmp.compareTo(end);
				if (cmp < 0) {
					series.add(tmp);
					times++;
				} else {
					if (opt == null || opt.indexOf('x') == -1) {
						series.add(end);
					}

					break;
				}
			}
		} else {
			if (start.compareTo(end) < 0) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("period" + mm.getMessage("function.invalidParam"));
			}

			while (true) {
				gc.setTimeInMillis(startTime);
				gc.add(field, distance * times);
				Date tmp = (Date)start.clone();
				tmp.setTime(gc.getTimeInMillis());
				gc.clear();

				int cmp = tmp.compareTo(end);
				if (cmp > 0) {
					series.add(tmp);
					times++;
				} else {
					if (opt == null || opt.indexOf('x') == -1) {
						series.add(end);
					}

					break;
				}
			}
		}

		return series;
	}
}