Application: gvSIG desktop

package org.gvsig.expresionevaluator.impl;

import org.gvsig.expressionevaluator.impl.SQLLexicalAnalyzer;

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.ExpressionUtils;

import org.gvsig.expressionevaluator.GeometryExpressionBuilder;

import org.gvsig.expressionevaluator.GeometryExpressionBuilderHelper.GeometrySupportType;

import org.gvsig.expressionevaluator.GeometryExpressionUtils;

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.DefaultInterpreter;

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 = ExpressionUtils.createCompiler();

        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 = ExpressionUtils.createOptimizer();

        optimizer.setSymbolTable(symbolTable);

        return optimizer;

    }

    protected MutableSymbolTable createSymbolTable() {

        DefaultSymbolTable symbolTable = new DefaultSymbolTable();

        symbolTable.setVar("precio", 200);

        symbolTable.setVar("1990", 0.168873933773767);

        return symbolTable;

    }

    private String createExpr1() {

        GeometryExpressionBuilder builder = GeometryExpressionUtils.createExpressionBuilder();

        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 testSpatialExpression1() throws Exception {

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

        GeometryExpressionBuilder builder = GeometryExpressionUtils.createExpressionBuilder();

        builder.geometry_support_type(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(

                "((\"campo1\" > 10) AND 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(

                "((\"campo1\" > 10) AND 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));

    }

}