Application: gvSIG desktop

package org.gvsig.expresionevaluator.impl;

import java.util.ArrayList;

import java.util.List;

import junit.framework.TestCase;

import org.apache.commons.lang3.ArrayUtils;

import org.cresques.cts.IProjection;

import org.gvsig.expressionevaluator.ExpressionBuilder;

import org.gvsig.expressionevaluator.ExpressionBuilder.Config;

import org.gvsig.expressionevaluator.ExpressionBuilder.GeometrySupportType;

import org.gvsig.expressionevaluator.ExpressionBuilder.Parameter;

import org.gvsig.expressionevaluator.ExpressionBuilder.Variable;

import org.gvsig.expressionevaluator.impl.DefaultExpressionBuilder;

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.tools.library.impl.DefaultLibrariesInitializer;

public class TestExpressionBuilder extends TestCase {

    public TestExpressionBuilder(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

//        new ExpressionEvaluatorImplLibrary().initialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

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

    // public void testHello() {}

    List<String> getVariableNames(ExpressionBuilder builder) {

        List<String> vars = new ArrayList<>();

        for (Variable var : builder.getVariables()) {

            vars.add(var.getName());

        }

        return vars;

    }

    List<String> getParameterNames(ExpressionBuilder builder) {

        List<String> params = new ArrayList<>();

        for (Parameter param : builder.getParameters()) {

            String s;

            switch(param.getType()) {

                case Constant:

                    Object value = param.getValue();

                    if( value==null ) {

                        s = "null";

                    } else if( value instanceof String ) {

                        s = "'" + (String)value + "'";

                    } else {

                        s = value.toString();

                    }    

                    break;

                case Geometry:

                case Variable:

                default:

                    s = "\"" + param.getName() + "\"";

            }

            params.add(s);

        }

        return params;

    }

    public void test2() {

        ExpressionBuilder builder = new DefaultExpressionBuilder();

        builder.and(

          builder.eq(

            builder.lcase(builder.variable("colum_name_c")),

            builder.parameter("colum_name_p")

          )

        );

        builder.and(

            builder.group(

                builder.or(

                    builder.like(

                        builder.lcase( builder.variable("uno")),

                        builder.constant("%10")

                    ),

                    builder.lt(

                        builder.variable("dos"),

                        builder.constant(-3.5)

                    )

                )

            )

        );

        builder.and(

                builder.ST_Intersects(

                    builder.variable("geom1"), 

                    builder.ST_Envelope(

                        builder.ST_GeomFromWKB(

                                builder.parameter("geom2"), 

                                builder.parameter().value(4326).as_constant()

                        )

                    )

                )

        );

        builder.and(

            builder.gt(

                builder.variable("tres"),

                builder.constant(123456789)

            )

        );

        assertEquals(

                "( (LCASE(\"colum_name_c\")) = (?) ) AND ( ( (LCASE(\"uno\")) LIKE ('%10') ) OR ( (\"dos\") < (-3.5) ) ) AND ST_Intersects((\"geom1\"), (ST_Envelope(ST_GeomFromWKB((?), (?))))) AND ( (\"tres\") > (123456789) )",

                builder.toString()

        );

        assertEquals(

                "[colum_name_c, dos, geom1, tres, uno]",

                ArrayUtils.toString(getVariableNames(builder))

        );

        assertEquals(

                "[\"colum_name_p\", \"geom2\", 4326]",

                ArrayUtils.toString(getParameterNames(builder))

        );

    }

    public void test3() throws CreateGeometryException {

        ExpressionBuilder builder = new DefaultExpressionBuilder();

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

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

        Point point = geometryManager.createPoint(10, 20, Geometry.SUBTYPES.GEOM2D);

        builder.set(        

            builder.ST_Intersects(

              builder.geometry(point,proj),

              builder.variable("the_geom")

            )

        );

        builder.getConfig().set(Config.geometry_type_support, GeometrySupportType.WKT);

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

        assertEquals(

                "ST_Intersects((ST_GeomFromText('POINT (10 20)', (4326))), (\"the_geom\"))",

                builder.toString()

        );

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

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

        assertEquals(

                "ST_Intersects((ST_GeomFromWKB((DECODE('000000000140240000000000004034000000000000','hex')), (4326))), (\"the_geom\"))",

                builder.toString()

        );

        assertEquals(

                "[the_geom]",

                ArrayUtils.toString(getVariableNames(builder))

        );

        assertEquals(

                "[]",

                ArrayUtils.toString(getParameterNames(builder))

        );

    }

    public void test4() throws CreateGeometryException {

        ExpressionBuilder builder = new DefaultExpressionBuilder();

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

        builder.and(

                builder.ST_Intersects(

                    builder.variable("geom1"), 

                    builder.ST_Envelope(

                        builder.ST_GeomFromWKB(

                                builder.parameter("geom2"), 

                                builder.parameter().value(proj).as_constant()

                        )

                    )

                )

        );

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

        assertEquals(

                "ST_Intersects((\"geom1\"), (ST_Envelope(ST_GeomFromWKB((?), (?)))))",

                builder.toString()

        );

        assertEquals(

                "[geom1]",

                ArrayUtils.toString(getVariableNames(builder))

        );

        assertEquals(

                "[\"geom2\", 4326]",

                ArrayUtils.toString(getParameterNames(builder))

        );

    }

    public void test5() throws CreateGeometryException {

        ExpressionBuilder builder = new DefaultExpressionBuilder();

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

        builder.and(

                builder.eq(

                        builder.ST_SRID( builder.variable("geom") ),

                        builder.srs(proj)

                )

        );

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

        assertEquals(

                "( (ST_SRID(\"geom\")) = (4326) )",

                builder.toString()

        );

        assertEquals(

                "[geom]",

                ArrayUtils.toString(getVariableNames(builder))

        );

        assertEquals(

                "[]",

                ArrayUtils.toString(getParameterNames(builder))

        );

    }

}