com.scudata.expression.fn.datetime.Days 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.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;
/**
* days(dateExp) ???ָ??????dateExp?????ꡢ???Ȼ????·ݵ?????
* @author runqian
*
*/
public class Days extends Function {
/**
* ??????ʽ????Ч?ԣ???Ч???׳??쳣
*/
public void checkValidity() {
if (param == null) {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.missingParam"));
} else if (!param.isLeaf()) {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.invalidParam"));
}
}
public Object calculate(Context ctx) {
Object value = param.getLeafExpression().calculate(ctx);
if (value instanceof String) {
value = Variant.parseDate((String)value);
} else if (value == null) {
return null;
}
if (option != null) {
if (option.indexOf('y') != -1) {
if (value instanceof Date) {
return ObjectCache.getInteger(DateFactory.get().daysInYear((Date)value));
} else if (value instanceof Number) {
return ObjectCache.getInteger(DateFactory.get().daysInYear(((Number)value).intValue()));
} else {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.paramTypeError"));
}
} else if (option.indexOf('q') != -1) {
if (!(value instanceof Date)) {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.paramTypeError"));
}
Date date = (Date)value;
Calendar calendar = Calendar.getInstance();
calendar.setTime(date);
calendar.set(Calendar.DATE, 1);
int month = calendar.get(Calendar.MONTH);
if (month < 3) {
month = 0;
} else if (month < 6) {
month = 3;
} else if (month < 9) {
month = 6;
} else {
month = 9;
}
int count = 0;
for (int i = 0; i < 3; ++i) {
calendar.set(Calendar.MONTH, month + i);
count += calendar.getActualMaximum(Calendar.DAY_OF_MONTH);
}
return ObjectCache.getInteger(count);
} else if (option.indexOf('o') != -1) {
if (!(value instanceof Date)) {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.paramTypeError"));
}
return ObjectCache.getInteger(DateFactory.toDays((Date)value));
}
}
if (value instanceof Date) {
return ObjectCache.getInteger(DateFactory.get().daysInMonth((Date)value));
} else {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + 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();
char opt = 'm';
if (option != null) {
if (option.indexOf('y') != -1) {
opt = 'y';
} else if (option.indexOf('q') != -1) {
opt = 'q';
} else if (option.indexOf('o') != -1) {
opt = 'o';
}
}
if (array instanceof ConstArray) {
Object obj = array.get(1);
if (obj != null) {
int days = days(obj, calendar, opt);
return new ConstArray(ObjectCache.getInteger(days), 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(days(date, calendar, opt));
} else {
result.pushNull();
}
}
} 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(days((Date)obj, calendar, opt));
} else {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.paramTypeError"));
}
} else {
result.pushNull();
}
}
} else if (array instanceof NumberArray) {
if (opt != 'y') {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.paramTypeError"));
}
calendar.set(Calendar.DAY_OF_YEAR, 1);
for (int i = 1; i <= size; ++i) {
if (array.isNull(i)) {
result.pushNull();
} else {
calendar.set(Calendar.YEAR, array.getInt(i));
result.pushInt(calendar.getActualMaximum(Calendar.DAY_OF_YEAR));
}
}
} else {
for (int i = 1; i <= size; ++i) {
Object obj = array.get(i);
if (obj != null) {
int days = days(obj, calendar, opt);
result.pushInt(days);
} 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();
char opt = 'm';
if (option != null) {
if (option.indexOf('y') != -1) {
opt = 'y';
} else if (option.indexOf('q') != -1) {
opt = 'q';
} else if (option.indexOf('o') != -1) {
opt = 'o';
}
}
if (array instanceof ConstArray) {
Object obj = array.get(1);
if (obj != null) {
int days = days(obj, calendar, opt);
return new ConstArray(ObjectCache.getInteger(days), size);
} else {
return new ConstArray(null, 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) {
result.pushInt(days(date, calendar, opt));
} else {
result.pushNull();
}
} else {
result.pushInt(0);
}
}
} 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) {
result.pushInt(days((Date)obj, calendar, opt));
} else {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.paramTypeError"));
}
} else {
result.pushNull();
}
} else {
result.pushInt(0);
}
}
} else if (array instanceof NumberArray) {
if (opt != 'y') {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.paramTypeError"));
}
calendar.set(Calendar.DAY_OF_YEAR, 1);
for (int i = 1; i <= size; ++i) {
if (!signDatas[i]) {
result.pushInt(0);
} else if (array.isNull(i)) {
result.pushNull();
} else {
calendar.set(Calendar.YEAR, array.getInt(i));
result.pushInt(calendar.getActualMaximum(Calendar.DAY_OF_YEAR));
}
}
} else {
for (int i = 1; i <= size; ++i) {
if (signDatas[i]) {
Object obj = array.get(i);
if (obj != null) {
int days = days(obj, calendar, opt);
result.pushInt(days);
} else {
result.pushNull();
}
} else {
result.pushInt(0);
}
}
}
return result;
}
private static int days(Date date, Calendar calendar, char opt) {
calendar.setTime(date);
if (opt == 'm') {
return calendar.getActualMaximum(Calendar.DAY_OF_MONTH);
} else if (opt == 'q') {
calendar.set(Calendar.DATE, 1);
int month = calendar.get(Calendar.MONTH);
if (month < 3) {
month = 0;
} else if (month < 6) {
month = 3;
} else if (month < 9) {
month = 6;
} else {
month = 9;
}
int count = 0;
for (int i = 0; i < 3; ++i) {
calendar.set(Calendar.MONTH, month + i);
count += calendar.getActualMaximum(Calendar.DAY_OF_MONTH);
}
return count;
} else if (opt == 'o') {
return DateFactory.toDays(date);
} else {
return calendar.getActualMaximum(Calendar.DAY_OF_YEAR);
}
}
private static int days(Object obj, Calendar calendar, char opt) {
if (obj instanceof String) {
obj = Variant.parseDate((String)obj);
}
if (obj instanceof Date) {
calendar.setTime((Date)obj);
if (opt == 'm') {
return calendar.getActualMaximum(Calendar.DAY_OF_MONTH);
} else if (opt == 'q') {
calendar.set(Calendar.DATE, 1);
int month = calendar.get(Calendar.MONTH);
if (month < 3) {
month = 0;
} else if (month < 6) {
month = 3;
} else if (month < 9) {
month = 6;
} else {
month = 9;
}
int count = 0;
for (int i = 0; i < 3; ++i) {
calendar.set(Calendar.MONTH, month + i);
count += calendar.getActualMaximum(Calendar.DAY_OF_MONTH);
}
return count;
} else if (opt == 'o') {
return DateFactory.toDays((Date)obj);
} else {
return calendar.getActualMaximum(Calendar.DAY_OF_YEAR);
}
} else if (obj instanceof Number) {
if (opt == 'y') {
calendar.set(Calendar.DAY_OF_YEAR, 1);
calendar.set(Calendar.YEAR, ((Number)obj).intValue());
return calendar.getActualMaximum(Calendar.DAY_OF_YEAR);
}
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.paramTypeError"));
} else {
MessageManager mm = EngineMessage.get();
throw new RQException("days" + mm.getMessage("function.paramTypeError"));
}
}
}