001: /*
002: * Copyright (c) 1998-2008 Caucho Technology -- all rights reserved
003: *
004: * This file is part of Resin(R) Open Source
005: *
006: * Each copy or derived work must preserve the copyright notice and this
007: * notice unmodified.
008: *
009: * Resin Open Source is free software; you can redistribute it and/or modify
010: * it under the terms of the GNU General Public License as published by
011: * the Free Software Foundation; either version 2 of the License, or
012: * (at your option) any later version.
013: *
014: * Resin Open Source is distributed in the hope that it will be useful,
015: * but WITHOUT ANY WARRANTY; without even the implied warranty of
016: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
017: * of NON-INFRINGEMENT. See the GNU General Public License for more
018: * details.
019: *
020: * You should have received a copy of the GNU General Public License
021: * along with Resin Open Source; if not, write to the
022: * Free Software Foundation, Inc.
023: * 59 Temple Place, Suite 330
024: * Boston, MA 02111-1307 USA
025: *
026: * @author Scott Ferguson
027: */
028:
029: package com.caucho.amber.expr.fun;
030:
031: import com.caucho.amber.expr.AmberExpr;
032: import com.caucho.amber.query.QueryParser;
033: import com.caucho.amber.type.DoubleType;
034: import com.caucho.amber.type.Type;
035: import com.caucho.util.CharBuffer;
036: import com.caucho.util.L10N;
037:
038: import java.util.ArrayList;
039:
040: /**
041: * SQRT function expression
042: */
043: public class SqrtFunExpr extends FunExpr {
044: private static final L10N L = new L10N(SqrtFunExpr.class);
045:
046: /**
047: * Creates a new expression
048: */
049: protected SqrtFunExpr(QueryParser parser, ArrayList<AmberExpr> args) {
050: super (parser, "sqrt", args, false);
051: }
052:
053: public static FunExpr create(QueryParser parser,
054: ArrayList<AmberExpr> args) {
055: return new SqrtFunExpr(parser, args);
056: }
057:
058: /**
059: * Generates the where expression.
060: */
061: public void generateWhere(CharBuffer cb) {
062: generateInternalWhere(cb, true);
063: }
064:
065: /**
066: * Generates the (update) where expression.
067: */
068: public void generateUpdateWhere(CharBuffer cb) {
069: generateInternalWhere(cb, false);
070: }
071:
072: /**
073: * Returns the expr type.
074: */
075: public Type getType() {
076: // jpa/1197
077: return DoubleType.create();
078: }
079:
080: //
081: // protected
082:
083: void generateInternalWhere(CharBuffer cb, boolean select) {
084: ArrayList<AmberExpr> args = getArgs();
085:
086: int n = args.size();
087:
088: // XXX: this validation should be moved to QueryParser
089: // if (n != 1)
090: // throw new QueryParseException(L.l("expected 1 argument for SQRT"));
091:
092: cb.append("sqrt(");
093:
094: if (select)
095: args.get(0).generateWhere(cb);
096: else
097: args.get(0).generateUpdateWhere(cb);
098:
099: cb.append(")");
100: }
101: }
|