initcap.java

来自「mysql集群」· Java 代码 · 共 66 行

/*****************************************************************************
      SQLJEP - Java SQL Expression Parser 0.2
      November 1 2006
         (c) Copyright 2006, Alexey Gaidukov
      SQLJEP Author: Alexey Gaidukov

      SQLJEP is based on JEP 2.24 (http://www.singularsys.com/jep/)
           (c) Copyright 2002, Nathan Funk
 
      See LICENSE.txt for license information.
*****************************************************************************/

package com.meidusa.amoeba.sqljep.function;


import com.meidusa.amoeba.sqljep.function.PostfixCommand;
import com.meidusa.amoeba.sqljep.ASTFunNode;
import com.meidusa.amoeba.sqljep.JepRuntime;
import com.meidusa.amoeba.sqljep.ParseException;

public class Initcap extends PostfixCommand {
	final public int getNumberOfParameters() {
		return 1;
	}
	
	public Comparable<?>[] evaluate(ASTFunNode node, JepRuntime runtime) throws ParseException {
		node.childrenAccept(runtime.ev, null);
		Comparable<?>  param = runtime.stack.pop();
		return new Comparable<?>[]{param};
	}

	public static String initcap(Comparable<?>  param) throws ParseException {
		if (param == null) {
			return null;
		}
		String inputStr = param.toString();
		StringBuilder output = new StringBuilder(inputStr);
		boolean needUpper = true;
		final int len = inputStr.length();
		for (int i = 0; i < len; i++) {
			char c = inputStr.charAt(i);
			if (needUpper && Character.isLetter(c)) {
				if (Character.isLowerCase(c)) {
					output.setCharAt(i, Character.toUpperCase(c));
				}
				needUpper = false;
			}
			else if (!needUpper) {
				if (!Character.isLetterOrDigit(c)) {
					needUpper = true;
				}
				else if (Character.isUpperCase(c)) {
					output.setCharAt(i, Character.toLowerCase(c));
				}
			}
		}
		return output.toString();
	}

	public Comparable<?> getResult(Comparable<?>... comparables)
			throws ParseException {
		return initcap(comparables[0]);
	}
}