com.scudata.expression.fn.datetime.Year 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.array.ConstArray;
import com.scudata.array.DateArray;
import com.scudata.array.IArray;
import com.scudata.array.IntArray;
import com.scudata.array.NumberArray;
import com.scudata.array.StringArray;
import com.scudata.common.DateFactory;
import com.scudata.common.MessageManager;
import com.scudata.common.ObjectCache;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.expression.Function;
import com.scudata.resources.EngineMessage;
import com.scudata.util.Variant;

/**
 * year(dateExp) ????????????dateExp?л??????Ϣ
 * @author runqian
 *
 */
public class Year extends Function {
	/**
	 * ??????ʽ????Ч?ԣ???Ч???׳??쳣
	 */
	public void checkValidity() {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("year" + mm.getMessage("function.missingParam"));
		} else if (!param.isLeaf()) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("year" + mm.getMessage("function.invalidParam"));
		}
	}

	public Object calculate(Context ctx) {
		Object result = param.getLeafExpression().calculate(ctx);
		if (result instanceof Number) {
			int days = ((Number)result).intValue();
			days = DateFactory.toYear(days);
			return ObjectCache.getInteger(days);
		} else if (result == null) {
			return null;
		}

		if (result instanceof String) {
			result = Variant.parseDate((String)result);
		}
		
		if (result instanceof Date) {
			return DateFactory.get().year((Date)result);
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("year" + mm.getMessage("function.paramTypeError"));
		}
	}
	
	/**
	 * ??????????еĽ??
	 * @param ctx ??????????
	 * @return IArray
	 */
	public IArray calculateAll(Context ctx) {
		IArray array = param.getLeafExpression().calculateAll(ctx);
		int size = array.size();
		Calendar calendar = DateFactory.get().calendar();
		
		if (array instanceof ConstArray) {
			int year = year(array.get(1), calendar);
			Integer value = year != -1 ? ObjectCache.getInteger(year) : null;
			return new ConstArray(value, size);
		}
		
		IntArray result = new IntArray(size);
		result.setTemporary(true);
		
		if (array instanceof DateArray) {
			DateArray dateArray = (DateArray)array;
			for (int i = 1; i <= size; ++i) {
				Date date = dateArray.getDate(i);
				if (date != null) {
					calendar.setTime(date);
					result.pushInt(calendar.get(Calendar.YEAR));
				} else {
					result.pushNull();
				}
			}
		} else if (array instanceof NumberArray) {
			for (int i = 1; i <= size; ++i) {
				if (array.isNull(i)) {
					result.pushNull();
				} else {
					int days = array.getInt(i);
					days = DateFactory.toYear(days);
					result.pushInt(days);
				}
			}
		} else if (array instanceof StringArray) {
			StringArray stringArray = (StringArray)array;
			for (int i = 1; i <= size; ++i) {
				String str = stringArray.getString(i);
				if (str != null) {
					Object obj = Variant.parseDate(str);
					if (obj instanceof Date) {
						calendar.setTime((Date)obj);
						result.pushInt(calendar.get(Calendar.YEAR));
					} else {
						MessageManager mm = EngineMessage.get();
						throw new RQException("year" + mm.getMessage("function.paramTypeError"));
					}
				} else {
					result.pushNull();
				}
			}
		} else {
			for (int i = 1; i <= size; ++i) {
				int year = year(array.get(i), calendar);
				if (year != -1) {
					result.pushInt(year);
				} else {
					result.pushNull();
				}
			}
		}
		
		return result;
	}
	
	/**
	 * ????signArray??ȡֵΪsign????
	 * @param ctx
	 * @param signArray ?б?ʶ????
	 * @param sign ??ʶ
	 * @return IArray
	 */
	public IArray calculateAll(Context ctx, IArray signArray, boolean sign) {
		IArray array = param.getLeafExpression().calculateAll(ctx);
		int size = array.size();
		Calendar calendar = DateFactory.get().calendar();
		
		if (array instanceof ConstArray) {
			int year = year(array.get(1), calendar);
			Integer value = year != -1 ? ObjectCache.getInteger(year) : null;
			return new ConstArray(value, size);
		}
		
		boolean[] signDatas;
		if (sign) {
			signDatas = signArray.isTrue().getDatas();
		} else {
			signDatas = signArray.isFalse().getDatas();
		}
		
		IntArray result = new IntArray(size);
		result.setTemporary(true);
		
		if (array instanceof DateArray) {
			DateArray dateArray = (DateArray)array;
			for (int i = 1; i <= size; ++i) {
				if (signDatas[i]) {
					Date date = dateArray.getDate(i);
					if (date != null) {
						calendar.setTime(date);
						result.pushInt(calendar.get(Calendar.YEAR));
					} else {
						result.pushNull();
					}
				} else {
					result.pushInt(0);
				}
			}
		} else if (array instanceof NumberArray) {
			for (int i = 1; i <= size; ++i) {
				if (!signDatas[i]) {
					result.pushInt(0);
				} else if (array.isNull(i)) {
					result.pushNull();
				} else {
					int days = array.getInt(i);
					days = DateFactory.toYear(days);
					result.pushInt(days);
				}
			}
		} else if (array instanceof StringArray) {
			StringArray stringArray = (StringArray)array;
			for (int i = 1; i <= size; ++i) {
				if (signDatas[i]) {
					String str = stringArray.getString(i);
					if (str != null) {
						Object obj = Variant.parseDate(str);
						if (obj instanceof Date) {
							calendar.setTime((Date)obj);
							result.pushInt(calendar.get(Calendar.YEAR));
						} else {
							MessageManager mm = EngineMessage.get();
							throw new RQException("year" + mm.getMessage("function.paramTypeError"));
						}
					} else {
						result.pushNull();
					}
				} else {
					result.pushInt(0);
				}
			}
		} else {
			for (int i = 1; i <= size; ++i) {
				if (signDatas[i]) {
					int year = year(array.get(i), calendar);
					if (year != -1) {
						result.pushInt(year);
					} else {
						result.pushNull();
					}
				} else {
					result.pushInt(0);
				}
			}
		}
		
		return result;
	}
	
	private static int year(Object obj, Calendar calendar) {
		if (obj instanceof Number) {
			int days = ((Number)obj).intValue();
			return DateFactory.toYear(days);
		} else if (obj == null) {
			return -1;
		}

		if (obj instanceof String) {
			obj = Variant.parseDate((String)obj);
		}
		
		if (obj instanceof Date) {
			calendar.setTime((Date)obj);
			return calendar.get(Calendar.YEAR);
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("year" + mm.getMessage("function.paramTypeError"));
		}
	}
	
	/**
	 * ???ؽڵ??Ƿ񵥵???????
	 * @return true???ǵ????????ģ?false??????
	 */
	public boolean isMonotone() {
		return param.getLeafExpression().isMonotone();
	}
}