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

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

	public Object calculate(Context ctx) {
		Object val = param.getLeafExpression().calculate(ctx);
		if (val instanceof Number) {
			int days = ((Number)val).intValue();
			if (option == null || option.indexOf('w') == -1) {
				days = DateFactory.toDay(days);
			} else {
				Date date = DateFactory.toDate(days);
				days = DateFactory.get().week(date);
			}
			
			return ObjectCache.getInteger(days);
		} else if (val == null) {
			return null;
		}
		
		if (val instanceof String) {
			val = Variant.parseDate((String)val);
		}

		if (val instanceof Date) {
			if (option == null || option.indexOf('w') == -1) {
				int day = DateFactory.get().day((Date)val);
				return ObjectCache.getInteger(day);
			} else {
				int week = DateFactory.get().week((Date)val);
				return ObjectCache.getInteger(week);
			}
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("day" + mm.getMessage("function.paramTypeError"));
		}
	}
	
	/**
	 * ??????????еĽ??
	 * @param ctx ??????????
	 * @return IArray
	 */
	public IArray calculateAll(Context ctx) {
		IArray array = param.getLeafExpression().calculateAll(ctx);
		int size = array.size();
		boolean isMonth = option == null || option.indexOf('w') == -1;
		Calendar calendar = DateFactory.get().calendar();
		
		if (array instanceof ConstArray) {
			Object obj = array.get(1);
			if (obj != null) {
				int day = day(obj, calendar, isMonth);
				Integer value = ObjectCache.getInteger(day);
				return new ConstArray(value, size);
			} else {
				return new ConstArray(null, 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) {
					result.pushInt(day(date, calendar, isMonth));
				} 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);
					if (isMonth) {
						days = DateFactory.toDay(days);
					} else {
						Date date = DateFactory.toDate(days);
						days = DateFactory.get().week(date);
					}

					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) {
						result.pushInt(day((Date)obj, calendar, isMonth));
					} else {
						MessageManager mm = EngineMessage.get();
						throw new RQException("day" + mm.getMessage("function.paramTypeError"));
					}
				} else {
					result.pushNull();
				}
			}
		} else {
			for (int i = 1; i <= size; ++i) {
				Object obj = array.get(i);
				if (obj != null) {
					int day = day(obj, calendar, isMonth);
					result.pushInt(day);
				} 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();
		boolean isMonth = option == null || option.indexOf('w') == -1;
		Calendar calendar = DateFactory.get().calendar();
		
		boolean[] signDatas;
		if (sign) {
			signDatas = signArray.isTrue().getDatas();
		} else {
			signDatas = signArray.isFalse().getDatas();
		}
		
		
		if (array instanceof ConstArray) {
			Object obj = array.get(1);
			if (obj != null) {
				int day = day(obj, calendar, isMonth);
				Integer value = ObjectCache.getInteger(day);
				return new ConstArray(value, size);
			} else {
				return new ConstArray(null, size);
			}
		}
		
		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) {
						result.pushInt(day(date, calendar, isMonth));
					} 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);
					if (isMonth) {
						days = DateFactory.toDay(days);
					} else {
						Date date = DateFactory.toDate(days);
						days = DateFactory.get().week(date);
					}

					result.pushInt(days);
				}
			}
		} else if (array instanceof StringArray) {
			StringArray stringArray = (StringArray)array;
			for (int i = 1; i <= size; ++i) {
				if (!signDatas[i]) {
					result.pushInt(0);
				} else if (stringArray.isNull(i)) {
					result.pushNull();
				} else {
					Object obj = Variant.parseDate(stringArray.getString(i));
					if (obj instanceof Date) {
						result.pushInt(day((Date)obj, calendar, isMonth));
					} else {
						MessageManager mm = EngineMessage.get();
						throw new RQException("day" + mm.getMessage("function.paramTypeError"));
					}
				}
			}
		} else {
			for (int i = 1; i <= size; ++i) {
				if (!signDatas[i]) {
					result.pushInt(0);
				} else if (array.isNull(i)) {
					result.pushNull();
				} else {
					int day = day(array.get(i), calendar, isMonth);
					result.pushInt(day);
				}
			}
		}
		
		return result;
	}
	
	private static int day(Date date, Calendar calendar, boolean isMonth) {
		calendar.setTime(date);
		if (isMonth) {
			return calendar.get(Calendar.DAY_OF_MONTH);
		} else {
			return calendar.get(Calendar.DAY_OF_WEEK);
		}
	}
	
	private static int day(Object obj, Calendar calendar, boolean isMonth) {
		if (obj instanceof Number) {
			int days = ((Number)obj).intValue();
			if (isMonth) {
				return DateFactory.toDay(days);
			} else {
				Date date = DateFactory.toDate(days);
				return DateFactory.get().week(date);
			}
		}

		if (obj instanceof String) {
			obj = Variant.parseDate((String)obj);
		}
		
		if (obj instanceof Date) {
			calendar.setTime((Date)obj);
			if (isMonth) {
				return calendar.get(Calendar.DAY_OF_MONTH);
			} else {
				return calendar.get(Calendar.DAY_OF_WEEK);
			}
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("day" + mm.getMessage("function.paramTypeError"));
		}
	}
}