Application: gvSIG desktop

package org.gvsig.expresionevaluator.impl;

import java.net.URL;

import junit.framework.TestCase;

import org.gvsig.expressionevaluator.LexicalAnalyzer;

import org.gvsig.expressionevaluator.Compiler;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.ExpressionEvaluatorLocator;

import org.gvsig.expressionevaluator.ExpressionEvaluatorManager;

import org.gvsig.expressionevaluator.MutableSymbolTable;

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

import org.gvsig.tools.resourcesstorage.AbstractResourcesStorage;

import org.gvsig.tools.resourcesstorage.URLResource;

public class TestCreateFunction extends TestCase {

    private static class Test1ResourcesStorage extends AbstractResourcesStorage {

        public Test1ResourcesStorage() {

        }

        @Override

        public Resource getResource(String resourceName) {

            URL url = this.getClass().getResource("/org/gvsig/expressionevaluator/impl/"+resourceName+".cosa");

            if( url == null ) {

                return null;

            }

            Resource res = new URLResource(url);

            return res;

        }        

    }

    public TestCreateFunction(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() {

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        LexicalAnalyzer lexer = manager.createLexicalAnalyzer();

        return lexer;

    }

    protected org.gvsig.expressionevaluator.Compiler createCompiler() {

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        Compiler compiler = manager.createCompiler();

        compiler.setLexicalAnalyzer(createLexicalAnalyzer());

        return compiler;

    }

    protected SymbolTable createSymbolTable() {

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        MutableSymbolTable symbolTable = manager.createSymbolTable();

        symbolTable.setVar("precio", 200);

        symbolTable.setVar("1990", 0.168873933773767);

        return symbolTable;

    }

    protected Interpreter createInterpreter(SymbolTable symbolTable) {

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        manager.setScriptsResourcesStorage(new Test1ResourcesStorage());

        Interpreter interpreter = manager.createInterpreter();

        interpreter.setSymbolTable(symbolTable);

        return interpreter;

    }

    public void testStringUtilsCapitalize() {

        SymbolTable symbolTable = createSymbolTable();

        Compiler compiler = createCompiler();

        Interpreter interpreter = createInterpreter(symbolTable);

        String source = "BEGIN "

                + "CREATE FUNCTION CAPITALIZE(s) AS "

                + "'org.apache.commons.lang3.StringUtils', 'capitalize' "

                + "LANGUAGE 'java';"

                + "CAPITALIZE('hola');"

                + "END";

        Code code = compiler.compileExpression(source);

        Object v = interpreter.run(code);

        assertEquals("Hola", v.toString());

    }

    public void testSQRTScriptEngine() {

        SymbolTable symbolTable = createSymbolTable();

        Compiler compiler = createCompiler();

        Interpreter interpreter = createInterpreter(symbolTable);

        String source = "BEGIN "

                + "CREATE FUNCTION SQUARE(num) AS 'testlib', 'SQUARE';"

                + "SQUARE(4);"

                + "END";

        Code code = compiler.compileExpression(source);

        Object v = interpreter.run(code);

        assertEquals(16, ((Number)v).intValue());

    }

}

        assertEquals("IFF((\"V1\" = 11), LET('V2', 22))", code.toString());

        assertEquals("IFF((\"V1\" = 11), BLOCK(LET('V2', 22), LET('V1', 10)))", code.toString());

        assertEquals("CASE(((\"Field_1\" >= 75) AND (\"Field_1\" <= 79)), ((\"Field_1\" >= 80) AND (\"Field_1\" <= 84)), 100, 110)", code.toString());

        assertEquals("CREATE_FUNCTION('test1', print('Hola'))", code.toString());

        assertEquals("CREATE_FUNCTION('test1', LIST('nombre'), print(('Hola ' + \"nombre\")))", code.toString());

        assertEquals("CREATE_FUNCTION('test1', LIST('nombre'), RETURN(('Hola ' || \"nombre\")))", code.toString());

        String source = "CREATE FUNCTION test1(nombre) AS 'addons/test/test', 'test1' LANGUAGE 'cosa'";

        assertEquals("CREATE_FUNCTION('test1', LIST('nombre'), 'addons/test/test', 'test1', 'cosa')", code.toString());

        InputStream is = this.getClass().getResource("/org/gvsig/expressionevaluator/impl/TestScript1_1."+CosaScriptEngineFactory.LANGUAGE_NAME).openStream();

        InputStream is = this.getClass().getResource("/org/gvsig/expressionevaluator/impl/TestScript1_2."+CosaScriptEngineFactory.LANGUAGE_NAME).openStream();

        InputStream is = this.getClass().getResource("/org/gvsig/expressionevaluator/impl/TestScript1_2."+CosaScriptEngineFactory.LANGUAGE_NAME).openStream();

-- encoding: utf-8; lang: cosa

create procedure form_onSetvalues form, values as

begin

  print("form_onSetvalues");

end 

create procedure form_onFieldEnter form, field as

begin

  print("form_onFieldEnter");

end 

create procedure form_onFieldExit form, field as

begin

  print("form_onFieldExit");

end 

create procedure form_onFieldChanged form, field as

begin

  print("form_onFieldChanged");

end 

create function form_validate form as

$$

  return true;

$$ 

create procedure form_onLoad form as

$$

  print("form_onLoad");

$$

create procedure form_clear form as

$$

  print("form_clear");

$$

create function form_isReadOnly form as

begin

    if not user().isAuthenticated() then

      return true;

    end if

    return false;

    -- set fecha_accidente = form.getField('FECHA_ACCIDENTE').getValue();

    -- if fecha_accidente = null then

    --   return false;

    -- end if

    -- set fecha_de_cierre = application().getProperty('CEGESEV.accidentes.fecha_de_cierre');

    -- if fecha_de_cierre = null then

    --   return true;

    -- end if

    -- return Date(fecha_de_cierre) > Date(fecha_accidente);

end 

-- encoding: utf-8; lang: cosa

CREATE FUNCTION test1 AS

BEGIN

    SET x  ''; 

    FOR n IN LIST('hola','adios','fin'):

        SET x = x || ' ' || n; 

    END 

    RETURN x;

END

CREATE FUNCTION test2 AS

$$

    SET s  ''; 

    SET x TO LIST('hola','adios','fin'); 

    FOR n IN RANGE(3):

        SET s = s || ' ' || x[n]; 

    END FOR;

    RETURN s;

$$

CREATE FUNCTION saluda AS

BEGIN

    IF LEN($ARGS) > 0 THEN

        SET s TO 'Hola ' || $ARGS[0];

    ELSE

        SET s = 'Hola.';

    END;

    PRINT(s);

    RETURN s;

END

CREATE FUNCTION test3 AS

BEGIN 

    RETURN saluda('amigos');

END

-- encoding: utf-8; lang: cosa

CREATE FUNCTION SQUARE(num) AS

BEGIN

    RETURN num*num;

END

    private static final Pattern RE_LANG = Pattern.compile(".*lang[:]\\s*([a-zA-Z_][a-zA-Z_0-9_]*).*");

    private static final Pattern RE_ENCODING = Pattern.compile(".*encoding[:]\\s*([a-zA-Z_][a-zA-Z0-9_-]*).*");

            head = StringUtils.split(head, "\n")[0];

            if( m!=null && m.matches() && m.groupCount()==1 ) {

            if( m!=null && m.matches() && m.groupCount()==1 ) {

import org.gvsig.tools.script.impl.DummyScriptManager;

        ToolsLocator.registerScriptManager(DefaultExpressionEvaluatorManager.class);

                        .addRuleOnTrue(stmt.require_literal_string("SCRIPT_FUNCTION"))

        // name, body

        // name, args, body

        // name, script_name, script_func, lang

        // name, args, script_name, script_func, lang

                script_path = (args.get(3)==null)? null:((String)((Code.Constant)args.get(3)).value());

                script_function = (args.get(4)==null)? null:((String)((Code.Constant)args.get(4)).value());

                language = (args.get(5)==null)? null:((String)((Code.Constant)args.get(5)).value()).toLowerCase();

        private final Script script;

            this.script = scriptManager.loadScript(resourcesStorage, script_path);

            if( this.script == null ) {

        }

        @Override

        public Object call(Interpreter interpreter, Object[] args) throws Exception {

            Object r = this.script.invokeFunction(this.script_function, args);

            String identifier = (String) token.getValue();

                String s = (String) token.getValue();

//                    if( code == null) {

//                        code = context.getCodeBuilder().constant(null);

//                    }

        if( arg==null ) {

            return null;

        }