001/**
002 *   GRANITE DATA SERVICES
003 *   Copyright (C) 2006-2013 GRANITE DATA SERVICES S.A.S.
004 *
005 *   This file is part of the Granite Data Services Platform.
006 *
007 *   Granite Data Services is free software; you can redistribute it and/or
008 *   modify it under the terms of the GNU Lesser General Public
009 *   License as published by the Free Software Foundation; either
010 *   version 2.1 of the License, or (at your option) any later version.
011 *
012 *   Granite Data Services is distributed in the hope that it will be useful,
013 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
014 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
015 *   General Public License for more details.
016 *
017 *   You should have received a copy of the GNU Lesser General Public
018 *   License along with this library; if not, write to the Free Software
019 *   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
020 *   USA, or see <http://www.gnu.org/licenses/>.
021 */
022package org.granite.gravity.selector;
023
024import javax.jms.JMSException;
025
026/**
027 * An expression which performs an operation on two expression values
028 *
029 * @version $Revision: 1.2 $
030 */
031public abstract class ArithmeticExpression extends BinaryExpression {
032
033    protected static final int INTEGER = 1;
034    protected static final int LONG = 2;
035    protected static final int DOUBLE = 3;
036
037    /**
038     * @param left
039     * @param right
040     */
041    public ArithmeticExpression(Expression left, Expression right) {
042        super(left, right);
043    }
044
045    public static Expression createPlus(Expression left, Expression right) {
046        return new ArithmeticExpression(left, right) {
047            @Override
048            protected Object evaluate(Object lvalue, Object rvalue) {
049                if (lvalue instanceof String) {
050                    String text = (String) lvalue;
051                    String answer = text + rvalue;
052                    return answer;
053                }
054                else if (lvalue instanceof Number) {
055                    return plus((Number) lvalue, asNumber(rvalue));
056                }
057                throw new RuntimeException("Cannot call plus operation on: " + lvalue + " and: " + rvalue);
058            }
059
060            @Override
061            public String getExpressionSymbol() {
062                return "+";
063            }
064        };
065    }
066
067    public static Expression createMinus(Expression left, Expression right) {
068        return new ArithmeticExpression(left, right) {
069            @Override
070            protected Object evaluate(Object lvalue, Object rvalue) {
071                if (lvalue instanceof Number) {
072                    return minus((Number) lvalue, asNumber(rvalue));
073                }
074                throw new RuntimeException("Cannot call minus operation on: " + lvalue + " and: " + rvalue);
075            }
076
077            @Override
078            public String getExpressionSymbol() {
079                return "-";
080            }
081        };
082    }
083
084    public static Expression createMultiply(Expression left, Expression right) {
085        return new ArithmeticExpression(left, right) {
086
087            @Override
088            protected Object evaluate(Object lvalue, Object rvalue) {
089                if (lvalue instanceof Number) {
090                    return multiply((Number) lvalue, asNumber(rvalue));
091                }
092                throw new RuntimeException("Cannot call multiply operation on: " + lvalue + " and: " + rvalue);
093            }
094
095            @Override
096            public String getExpressionSymbol() {
097                return "*";
098            }
099        };
100    }
101
102    public static Expression createDivide(Expression left, Expression right) {
103        return new ArithmeticExpression(left, right) {
104
105            @Override
106            protected Object evaluate(Object lvalue, Object rvalue) {
107                if (lvalue instanceof Number) {
108                    return divide((Number) lvalue, asNumber(rvalue));
109                }
110                throw new RuntimeException("Cannot call divide operation on: " + lvalue + " and: " + rvalue);
111            }
112
113            @Override
114            public String getExpressionSymbol() {
115                return "/";
116            }
117        };
118    }
119
120    public static Expression createMod(Expression left, Expression right) {
121        return new ArithmeticExpression(left, right) {
122
123            @Override
124            protected Object evaluate(Object lvalue, Object rvalue) {
125                if (lvalue instanceof Number) {
126                    return mod((Number) lvalue, asNumber(rvalue));
127                }
128                throw new RuntimeException("Cannot call mod operation on: " + lvalue + " and: " + rvalue);
129            }
130
131            @Override
132            public String getExpressionSymbol() {
133                return "%";
134            }
135        };
136    }
137
138    protected Number plus(Number left, Number right) {
139        switch (numberType(left, right)) {
140            case INTEGER:
141                return new Integer(left.intValue() + right.intValue());
142            case LONG:
143                return new Long(left.longValue() + right.longValue());
144            default:
145                return new Double(left.doubleValue() + right.doubleValue());
146        }
147    }
148
149    protected Number minus(Number left, Number right) {
150        switch (numberType(left, right)) {
151            case INTEGER:
152                return new Integer(left.intValue() - right.intValue());
153            case LONG:
154                return new Long(left.longValue() - right.longValue());
155            default:
156                return new Double(left.doubleValue() - right.doubleValue());
157        }
158    }
159
160    protected Number multiply(Number left, Number right) {
161        switch (numberType(left, right)) {
162            case INTEGER:
163                return new Integer(left.intValue() * right.intValue());
164            case LONG:
165                return new Long(left.longValue() * right.longValue());
166            default:
167                return new Double(left.doubleValue() * right.doubleValue());
168        }
169    }
170
171    protected Number divide(Number left, Number right) {
172        return new Double(left.doubleValue() / right.doubleValue());
173    }
174
175    protected Number mod(Number left, Number right) {
176        return new Double(left.doubleValue() % right.doubleValue());
177    }
178
179    private int numberType(Number left, Number right) {
180        if (isDouble(left) || isDouble(right)) {
181            return DOUBLE;
182        }
183        else if (left instanceof Long || right instanceof Long) {
184            return LONG;
185        }
186        else {
187            return INTEGER;
188        }
189    }
190
191    private boolean isDouble(Number n) {
192        return n instanceof Float || n instanceof Double;
193    }
194
195    protected Number asNumber(Object value) {
196        if (value instanceof Number) {
197            return (Number) value;
198        }
199        throw new RuntimeException("Cannot convert value: " + value + " into a number");
200    }
201
202    public Object evaluate(MessageEvaluationContext message) throws JMSException {
203        Object lvalue = left.evaluate(message);
204        if (lvalue == null) {
205            return null;
206        }
207        Object rvalue = right.evaluate(message);
208        if (rvalue == null) {
209            return null;
210        }
211        return evaluate(lvalue, rvalue);
212    }
213
214
215    /**
216     * @param lvalue
217     * @param rvalue
218     * @return the evaluated value
219     */
220    abstract protected Object evaluate(Object lvalue, Object rvalue);
221
222}