Application: gvSIG desktop

package org.gvsig.expresionevaluator.impl;

import static junit.framework.Assert.assertEquals;

import org.gvsig.expressionevaluator.impl.SQLLexicalAnalyzer;

import org.gvsig.expressionevaluator.impl.DefaultCompiler;

import junit.framework.TestCase;

import org.cresques.cts.IProjection;

import org.gvsig.expressionevaluator.LexicalAnalyzer;

import org.gvsig.expressionevaluator.Compiler;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.ExpressionBuilder;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.MutableSymbolTable;

import org.gvsig.expressionevaluator.Optimizer;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.expressionevaluator.impl.DefaultExpressionBuilder;

import org.gvsig.expressionevaluator.impl.DefaultInterpreter;

import org.gvsig.expressionevaluator.impl.DefaultOptimizer;

import org.gvsig.expressionevaluator.impl.DefaultSymbolTable;

import org.gvsig.fmap.crs.CRSFactory;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.exception.CreateGeometryException;

import org.gvsig.fmap.geom.primitive.Point;

import org.gvsig.fmap.geom.primitive.Polygon;

import org.gvsig.tools.library.impl.DefaultLibrariesInitializer;

public class TestOptimizer extends TestCase {

    public TestOptimizer(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

    // TODO add test methods here. The name must begin with 'test'. For example:

    // public void testHello() {}

    protected LexicalAnalyzer createLexicalAnalyzer() {

        SQLLexicalAnalyzer lexer = new SQLLexicalAnalyzer();

        return lexer;

    }

    protected Compiler createCompiler() {

        Compiler compiler = new DefaultCompiler();

        compiler.setLexicalAnalyzer(createLexicalAnalyzer());

        return compiler;

    }

    protected Interpreter createInterpreter(SymbolTable symbolTable) {

        Interpreter interpreter = new DefaultInterpreter();

        interpreter.setSymbolTable(symbolTable);

        return interpreter;

    }

    protected Optimizer createOptimizer(SymbolTable symbolTable) {

        Optimizer optimizer = new DefaultOptimizer(symbolTable);

        return optimizer;

    }

    protected MutableSymbolTable createSymbolTable() {

        DefaultSymbolTable symbolTable = new DefaultSymbolTable();

        symbolTable.setVar("precio", 200);

        symbolTable.setVar("1990", 0.168873933773767);

        return symbolTable;

    }

    private String createExpr1() {

        ExpressionBuilder builder = new DefaultExpressionBuilder();

        ExpressionBuilder.Value expr1 = builder.gt(

                builder.column("campo1"),

                builder.constant(10)

        );

        System.out.println(expr1.toString());

        assertEquals(

                "( (\"campo1\") > (10) )",

                expr1.toString()

        );

        return expr1.toString();

    }

    private Polygon createPolygon() throws CreateGeometryException {

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        IProjection proj = CRSFactory.getCRS("EPSG:4326");

        Polygon polygon = geometryManager.createPolygon(Geometry.SUBTYPES.GEOM2D);

        polygon.addVertex(0, 0);

        polygon.addVertex(0, 100);

        polygon.addVertex(100, 100);

        polygon.addVertex(100, 0);

        polygon.addVertex(0, 0);

        polygon.setProjection(proj);

        return polygon;

    }

    private Geometry createPoint(double x, double y) throws CreateGeometryException {

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        IProjection proj = CRSFactory.getCRS("EPSG:4326");

        Point point = geometryManager.createPoint(x, y, Geometry.SUBTYPES.GEOM2D);

        point.setProjection(proj);

        return point;

    }

    public void testIdentifier1() {

        String source = "precio";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("\"precio\"", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("\"precio\"", code2.toString());

    }

    public void testIdentifier2() {

        String source = "[precio]";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("\"precio\"", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("\"precio\"", code2.toString());

    }

    public void testIdentifier3() {

        String source = "\"precio\"";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("\"precio\"", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("\"precio\"", code2.toString());

    }

    public void testTrue() {

        String source = "true";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("true", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("true", code2.toString());

    }

    public void testFalse() {

        String source = "false";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("false", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("false", code2.toString());

    }

    public void testNull() {

        String source = "null";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("null", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("null", code2.toString());

    }

    public void testNotTrue() {

        String source = "not true";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("not(true)", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("false", code2.toString());

    }

    public void testInteger() {

        String source = "23";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("23", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("23", code2.toString());

    }

    public void operator(String operatorName) {

        String source = "precio " + operatorName + " 23";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals(operatorName + "(\"precio\", 23)", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals(operatorName + "(\"precio\", 23)", code2.toString());

    }

    public void testOperators() {

        operator("=");

        operator("<>");

        operator(">");

        operator(">=");

        operator("<");

        operator("<=");

        operator("like");

        operator("ilike");

        operator("+");

        operator("-");

        operator("*");

//        operator("^");

        operator("or");

        operator("and");

        operator("%");

        operator("is");

    }

    public void testAddMul() {

        String source = "precio + 10 * 2 + 20 + 30";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("+(+(+(\"precio\", *(10, 2)), 20), 30)", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("+(+(+(\"precio\", 20), 20), 30)", code2.toString());

    }

    public void testAddMulPar() {

        String source = "(precio + 10) * 2 + 20 + 30";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("+(+(*(+(\"precio\", 10), 2), 20), 30)", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("+(+(*(+(\"precio\", 10), 2), 20), 30)", code2.toString());

    }

    public void testAbs() {

        String source = "precio + abs(10)";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("+(\"precio\", abs(10))", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("+(\"precio\", 10)", code2.toString());

    }

    public void testAbs2() {

        String source = "precio + abs(-10)";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("+(\"precio\", abs(-10))", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("+(\"precio\", 10)", code2.toString());

    }

    public void testPI() {

        String source = "precio + PI()";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("+(\"precio\", PI())", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("+(\"precio\", 3.141592653589793)", code2.toString());

    }

    public void testCeil() {

        String source = "precio + CEIL(PI())";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("+(\"precio\", CEIL(PI()))", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("+(\"precio\", 4.0)", code2.toString());

    }

    public void testConcat() {

        String source = "CONCAT(precio,' euros')";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("CONCAT(\"precio\", ' euros')", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("CONCAT(\"precio\", ' euros')", code2.toString());

    }

    public void testInvokeFunction1() {

        String source = "ST_Area(GEOMETRY)";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("ST_Area(\"GEOMETRY\")", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("ST_Area(\"GEOMETRY\")", code2.toString());

    }

    public void testInvokeMethod1() {

        String source = "'hola'.length()";

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("'hola'->length()", code.toString());

        SymbolTable symbolTable = createSymbolTable();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

        assertEquals("'hola'->length()", code2.toString());

    }

    public void testSpatialExpression1() throws Exception {

        IProjection proj = CRSFactory.getCRS("EPSG:4326");

        ExpressionBuilder builder = new DefaultExpressionBuilder();

        builder.getConfig().set(ExpressionBuilder.Config.geometry_type_support, ExpressionBuilder.GeometrySupportType.WKB);

        String source = builder.and(

                builder.custom(this.createExpr1()),

                builder.ST_Intersects(

                        builder.geometry(createPolygon()),

                        builder.column("GEOMETRY")

                )

        ).toString();

        MutableSymbolTable symbolTable = createSymbolTable();

        symbolTable.setVar("campo1", 11);

        symbolTable.setVar("GEOMETRY", createPoint(50, 50));

        Interpreter interpreter = createInterpreter(symbolTable);

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        System.out.println(code.toString());

        assertEquals(

                "and(>(\"campo1\", 10), ST_Intersects(ST_GeomFromWKB(DECODE('000000000300000001000000050000000000000000000000000000000000000000000000004059000000000000405900000000000040590000000000004059000000000000000000000000000000000000000000000000000000000000', 'hex'), 4326), \"GEOMETRY\"))",

                code.toString()

        );

//        long t1 = System.currentTimeMillis();

//        for (int i = 0; i < 1000; i++) {

//            interpreter.run(code);

//        }

//        System.out.println("Sin optimizar: " + (System.currentTimeMillis() - t1));

//        t1 = System.currentTimeMillis();

        Optimizer optimizer = createOptimizer(symbolTable);

        Code code2 = optimizer.optimize(code);

//        System.out.println("Tiempo de optimizacion: " + (System.currentTimeMillis() - t1));

        System.out.println(code2.toString());

        assertEquals(

                "and(>(\"campo1\", 10), ST_Intersects('POLYGON ((0 0, 0 100, 100 100, 100 0, 0 0))'::geometry, \"GEOMETRY\"))",

                code2.toString()

        );

//        t1 = System.currentTimeMillis();

//        for (int i = 0; i < 1000; i++) {

//            interpreter.run(code2);

//        }

//        System.out.println("Optimizado: " + (System.currentTimeMillis() - t1));

    }

}