com.scudata.expression.fn.string.Mid 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.string;

import com.scudata.array.ConstArray;
import com.scudata.array.IArray;
import com.scudata.array.StringArray;
import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.expression.Function;
import com.scudata.expression.IParam;
import com.scudata.resources.EngineMessage;

/**
 * mid(s, start{, len}) ?????ַ???s?д?startλ???𳤶?Ϊlen???Ӵ???
 * @author runqian
 *
 */
public class Mid extends Function {
	/**
	 * ??????ʽ????Ч?ԣ???Ч???׳??쳣
	 */
	public void checkValidity() {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.missingParam"));
		}

		int size = param.getSubSize();
		if (size != 2 && size != 3) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.invalidParam"));
		}
	}

	private static String mid(String str, int begin, int count) {
		if (count < 1) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.invalidParam") + count + " < 1");
		}
		
		int len = str.length();
		if (begin > len) {
			return "";
		} else if (begin > 0) {
			begin--;
			int end = begin + count;
			if (end > len) {
				return str.substring(begin, len);
			} else {
				return str.substring(begin, end);
			}
		} else if (begin < 0) {
			begin += len;
			if (begin < 0) {
				int end = begin + count;
				if (end < 1) {
					return "";
				} else if (end < len) {
					return str.substring(0, end);
				} else {
					return str;
				}
			} else {
				int end = begin + count;
				if (end > len) {
					return str.substring(begin, len);
				} else {
					return str.substring(begin, end);
				}
			}
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.invalidParam"));
		}
	}
	
	private static String mid(String str, int begin) {
		int len = str.length();
		if (begin > len) {
			return "";
		} else if (begin > 0) {
			begin--;
			return str.substring(begin);
		} else if (begin < 0) {
			begin += len;
			if (begin <= 0) {
				return str;
			} else {
				return str.substring(begin);
			}
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.invalidParam"));
		}
	}
	
	public Object calculate(Context ctx) {
		IParam sub1 = param.getSub(0);
		IParam sub2 = param.getSub(1);
		if (sub1 == null || sub2 == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.invalidParam"));
		}

		Object result1 = sub1.getLeafExpression().calculate(ctx);
		if (result1 == null) {
			return null;
		} else if (!(result1 instanceof String)) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
		}

		Object result2 = sub2.getLeafExpression().calculate(ctx);
		if (!(result2 instanceof Number)) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
		}

		int begin = ((Number)result2).intValue();

		if (param.getSubSize() > 2) {
			IParam sub3 = param.getSub(2);
			if (sub3 != null) {
				Object result3 = sub3.getLeafExpression().calculate(ctx);
				if (!(result3 instanceof Number)) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
				}
				
				int count = ((Number)result3).intValue();
				return mid((String)result1, begin, count);
			}
		}

		return mid((String)result1, begin);
	}

	/**
	 * ??????????еĽ??
	 * @param ctx ??????????
	 * @return IArray
	 */
	public IArray calculateAll(Context ctx) {
		IParam sub1 = param.getSub(0);
		IParam sub2 = param.getSub(1);
		if (sub1 == null || sub2 == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.invalidParam"));
		}
		
		IArray array2 = sub2.getLeafExpression().calculateAll(ctx);
		if (!array2.isNumberArray()) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
		}
		
		IArray array1 = sub1.getLeafExpression().calculateAll(ctx);
		int size = array1.size();
				
		if (param.getSubSize() > 2) {
			IParam sub3 = param.getSub(2);
			if (sub3 == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("mid" + mm.getMessage("function.invalidParam"));
			}
			
			IArray array3 = sub3.getLeafExpression().calculateAll(ctx);
			if (!array3.isNumberArray()) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
			}

			if (array1 instanceof StringArray && array2 instanceof ConstArray && array3 instanceof ConstArray) {
				int begin = array2.getInt(1);
				int count = array3.getInt(1);
				StringArray stringArray = (StringArray)array1;
				StringArray result = new StringArray(size);
				result.setTemporary(true);
				
				for (int i = 1; i <= size; ++i) {
					String str = stringArray.getString(i);
					if (str != null) {
						str = mid(str, begin, count);
						result.push(str);
					} else {
						result.push(null);
					}
				}
				
				return result;
			} else if (array1 instanceof ConstArray && array2 instanceof ConstArray && array3 instanceof ConstArray) {
				int begin = array2.getInt(1);
				int count = array3.getInt(1);
				Object obj1 = array1.get(1);
				
				if (obj1 instanceof String) {
					String str = mid((String)obj1, begin, count);
					return new ConstArray(str, size);
				} else if (obj1 != null) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
				} else {
					return new ConstArray(null, size);
				}
			} else {
				StringArray result = new StringArray(size);
				result.setTemporary(true);
				
				for (int i = 1; i <= size; ++i) {
					Object obj1 = array1.get(i);
					if (obj1 instanceof String) {
						int begin = array2.getInt(i);
						int count = array3.getInt(i);
						String str = (String)obj1;
						str = mid(str, begin, count);
						result.push(str);
					} else if (obj1 != null) {
						MessageManager mm = EngineMessage.get();
						throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
					} else {
						result.push(null);
					}
				}
				
				return result;
			}
		} else {
			if (array1 instanceof StringArray && array2 instanceof ConstArray) {
				int begin = array2.getInt(1);
				StringArray stringArray = (StringArray)array1;
				StringArray result = new StringArray(size);
				result.setTemporary(true);
				
				for (int i = 1; i <= size; ++i) {
					String str = stringArray.getString(i);
					if (str != null) {
						str = mid(str, begin);
						result.push(str);
					} else {
						result.push(null);
					}
				}
				
				return result;
			} else if (array1 instanceof ConstArray && array2 instanceof ConstArray) {
				int begin = array2.getInt(1);
				Object obj1 = array1.get(1);
				
				if (obj1 instanceof String) {
					String str = mid((String)obj1, begin);
					return new ConstArray(str, size);
				} else if (obj1 != null) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
				} else {
					return new ConstArray(null, size);
				}
			} else {
				StringArray result = new StringArray(size);
				result.setTemporary(true);
				
				for (int i = 1; i <= size; ++i) {
					Object obj1 = array1.get(i);
					if (obj1 instanceof String) {
						int begin = array2.getInt(i);
						String str = mid((String)obj1, begin);
						result.push(str);
					} else if (obj1 != null) {
						MessageManager mm = EngineMessage.get();
						throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
					} else {
						result.push(null);
					}
				}
				
				return result;
			}
		}
	}
	
	/**
	 * ????signArray??ȡֵΪsign????
	 * @param ctx
	 * @param signArray ?б?ʶ????
	 * @param sign ??ʶ
	 * @return IArray
	 */
	public IArray calculateAll(Context ctx, IArray signArray, boolean sign) {
		boolean[] signDatas;
		if (sign) {
			signDatas = signArray.isTrue().getDatas();
		} else {
			signDatas = signArray.isFalse().getDatas();
		}		

		IParam sub1 = param.getSub(0);
		IParam sub2 = param.getSub(1);
		if (sub1 == null || sub2 == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.invalidParam"));
		}
		
		IArray array2 = sub2.getLeafExpression().calculateAll(ctx);
		if (!array2.isNumberArray()) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
		}
		
		IArray array1 = sub1.getLeafExpression().calculateAll(ctx);
		int size = array1.size();
				
		if (param.getSubSize() > 2) {
			IParam sub3 = param.getSub(2);
			if (sub3 == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("mid" + mm.getMessage("function.invalidParam"));
			}
			
			IArray array3 = sub3.getLeafExpression().calculateAll(ctx);
			if (!array3.isNumberArray()) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
			}

			if (array1 instanceof StringArray && array2 instanceof ConstArray && array3 instanceof ConstArray) {
				int begin = array2.getInt(1);
				int count = array3.getInt(1);
				StringArray stringArray = (StringArray)array1;
				StringArray result = new StringArray(size);
				result.setTemporary(true);
				
				for (int i = 1; i <= size; ++i) {
					if (signDatas[i] == false) {
						result.pushNull();
						continue;
					}
					
					String str = stringArray.getString(i);
					if (str != null) {
						str = mid(str, begin, count);
						result.push(str);
					} else {
						result.push(null);
					}
				}
				
				return result;
			} else if (array1 instanceof ConstArray && array2 instanceof ConstArray && array3 instanceof ConstArray) {
				int begin = array2.getInt(1);
				int count = array3.getInt(1);
				Object obj1 = array1.get(1);
				
				if (obj1 instanceof String) {
					String str = mid((String)obj1, begin, count);
					return new ConstArray(str, size);
				} else if (obj1 != null) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
				} else {
					return new ConstArray(null, size);
				}
			} else {
				StringArray result = new StringArray(size);
				result.setTemporary(true);
				
				for (int i = 1; i <= size; ++i) {
					if (signDatas[i] == false) {
						result.pushNull();
						continue;
					}
					
					Object obj1 = array1.get(i);
					if (obj1 instanceof String) {
						int begin = array2.getInt(i);
						int count = array3.getInt(i);
						String str = (String)obj1;
						str = mid(str, begin, count);
						result.push(str);
					} else if (obj1 != null) {
						MessageManager mm = EngineMessage.get();
						throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
					} else {
						result.push(null);
					}
				}
				
				return result;
			}
		} else {
			if (array1 instanceof StringArray && array2 instanceof ConstArray) {
				int begin = array2.getInt(1);
				StringArray stringArray = (StringArray)array1;
				StringArray result = new StringArray(size);
				result.setTemporary(true);
				
				for (int i = 1; i <= size; ++i) {
					if (signDatas[i] == false) {
						result.pushNull();
						continue;
					}
					
					String str = stringArray.getString(i);
					if (str != null) {
						str = mid(str, begin);
						result.push(str);
					} else {
						result.push(null);
					}
				}
				
				return result;
			} else if (array1 instanceof ConstArray && array2 instanceof ConstArray) {
				int begin = array2.getInt(1);
				Object obj1 = array1.get(1);
				
				if (obj1 instanceof String) {
					String str = mid((String)obj1, begin);
					return new ConstArray(str, size);
				} else if (obj1 != null) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
				} else {
					return new ConstArray(null, size);
				}
			} else {
				StringArray result = new StringArray(size);
				result.setTemporary(true);
				
				for (int i = 1; i <= size; ++i) {
					if (signDatas[i] == false) {
						result.pushNull();
						continue;
					}
					
					Object obj1 = array1.get(i);
					if (obj1 instanceof String) {
						int begin = array2.getInt(i);
						String str = mid((String)obj1, begin);
						result.push(str);
					} else if (obj1 != null) {
						MessageManager mm = EngineMessage.get();
						throw new RQException("mid" + mm.getMessage("function.paramTypeError"));
					} else {
						result.push(null);
					}
				}
				
				return result;
			}
		}
	}
}