com.scudata.expression.fn.string.Trim 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.resources.EngineMessage;

/**
 * ȥ???ַ??????˵Ŀմ?
 * @author runqian
 *
 */
public class Trim extends Function {
	/**
	 * ??????ʽ????Ч?ԣ???Ч???׳??쳣
	 */
	public void checkValidity() {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("trim" + mm.getMessage("function.missingParam"));
		} else if (!param.isLeaf()) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("trim" + mm.getMessage("function.invalidParam"));
		}
	}
	
	public Object calculate(Context ctx) {
		Object obj = param.getLeafExpression().calculate(ctx);
		if (obj instanceof String) {
			return trim((String)obj, option);
		} else if (obj == null) {
			return null;
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("trim" + mm.getMessage("function.paramTypeError"));
		}
	}
	
	/**
	 * ??????????еĽ??
	 * @param ctx ??????????
	 * @return IArray
	 */
	public IArray calculateAll(Context ctx) {
		IArray array = param.getLeafExpression().calculateAll(ctx);
		int size = array.size();
		
		if (array instanceof StringArray) {
			StringArray stringArray = (StringArray)array;
			StringArray result = new StringArray(size);
			result.setTemporary(true);
			
			for (int i = 1; i <= size; ++i) {
				String str = stringArray.getString(i);
				if (str != null) {
					str = trim(str, option);
				} 
				
				result.push(str);
			}
			
			return result;
		} else if (array instanceof ConstArray) {
			Object obj = array.get(1);
			if (obj instanceof String) {
				obj = trim((String)obj, option);
			} else if (obj != null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("trim" + mm.getMessage("function.paramTypeError"));
			}
			
			return new ConstArray(obj, size);
		} else {
			StringArray result = new StringArray(size);
			result.setTemporary(true);
			
			for (int i = 1; i <= size; ++i) {
				Object obj = array.get(i);
				String str = null;
				if (obj instanceof String) {
					str = trim((String)obj, option);
				} else if (obj != null) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("trim" + mm.getMessage("function.paramTypeError"));
				}
				
				result.push(str);
			}
			
			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();
		}
		
		IArray array = param.getLeafExpression().calculateAll(ctx);
		int size = array.size();
		
		if (array instanceof StringArray) {
			StringArray stringArray = (StringArray)array;
			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 = trim(str, option);
				} 
				
				result.push(str);
			}
			
			return result;
		} else if (array instanceof ConstArray) {
			Object obj = array.get(1);
			if (obj instanceof String) {
				obj = trim((String)obj, option);
			} else if (obj != null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("trim" + mm.getMessage("function.paramTypeError"));
			}
			
			return new ConstArray(obj, 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 obj = array.get(i);
				String str = null;
				if (obj instanceof String) {
					str = trim((String)obj, option);
				} else if (obj != null) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("trim" + mm.getMessage("function.paramTypeError"));
				}
				
				result.push(str);
			}
			
			return result;
		}
	}

	private static String trim(String str, String option) {
		if (option == null) {
			return str.trim();
		} else if (option.indexOf('l') != -1) {
			int i = 0, ilength = str.length();
			for (i = 0; i < ilength; i++) {
				if (!Character.isWhitespace(str.charAt(i))) {
					break;
				}
			}
			if (i >= str.length()) {
				return "";
			} else {
				return str.substring(i);
			}
		} else if (option.indexOf('r') != -1) {
			int i = 0;
			for (i = str.length() - 1; i >= 0; i--) {
				if (!Character.isWhitespace(str.charAt(i))) {
					break;
				}
			}
			if (i < 0) {
				return "";
			} else {
				return str.substring(0, i + 1);
			}
		} else if (option.indexOf('a') != -1) {
			char []chars = str.toCharArray();
			int len = chars.length;
			StringBuffer sb = new StringBuffer(len);
			boolean isPrevWhiteSpace = true;
			for (int i = 0; i < len; ++i) {
				char c = chars[i];
				if (Character.isWhitespace(c)) {
					if (!isPrevWhiteSpace) {
						isPrevWhiteSpace = true;
						sb.append(c);
					}
				} else if (c == '"' || c == '\'') {
					int match = scanString(chars, i);
					if (match > 0) {
						sb.append(chars, i, match - i + 1);
						i = match;
					} else {
						sb.append(c);
					}
					
					isPrevWhiteSpace = false;
				} else if (isSymbol(c)) {
					int last = sb.length() - 1;
					if (last >= 0 && Character.isWhitespace(sb.charAt(last))) {
						sb.deleteCharAt(last);
					}
					
					sb.append(c);
					isPrevWhiteSpace = true;
				} else {
					sb.append(c);
					isPrevWhiteSpace = false;
				}
			}
			
			int last = sb.length() - 1;
			if (last >= 0 && Character.isWhitespace(sb.charAt(last))) {
				sb.deleteCharAt(last);
			}
			
			return sb.toString();
		} else {
			return str.trim();
		}
	}
	
	private static int scanString(char []chars, int start) {
		char q = chars[start++];
		for (int len = chars.length; start < len; ++start) {
			if (chars[start] == q) {
				start++;
				if (start == len || chars[start] != q) {
					return start;
				}
			}
		}

		return -1;
	}
	
	private static boolean isSymbol(char c) {
		return c == '+' || c == '-' || c == '*' || c == '/' || c == '%' ||
			c == '=' || c == '&' || c == '|' || c == '!' ||
			c == ',' || c == '>' || c == '<' || c == '(' || c == ')' ||
			c == '[' || c == ']' || c == ':' || c == '{' || c == '}' ||
			c == '^' || c == '.';
	}
}