Application: gvSIG desktop

package org.gvsig.expressionevaluator.impl;

import java.util.HashMap;

import java.util.Iterator;

import java.util.Map;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.expressionevaluator.impl.function.ScriptFunction;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.script.Script;

import org.gvsig.tools.script.ScriptManager;

/**

 *

 * @author jjdelcerro

 */

public abstract class AbstractSymbolTable implements SymbolTable {

    protected Map<String, Function> functions;

    protected SymbolTable delegatedSymbolTable;

    protected Script script = null;

    public AbstractSymbolTable() {

        this.functions = new HashMap<>();

    }

    public AbstractSymbolTable(SymbolTable delegatedSymbolTable) {

        this();

        this.delegatedSymbolTable = delegatedSymbolTable;

    }

    @Override

    public SymbolTable clone() throws CloneNotSupportedException {

        AbstractSymbolTable other = (AbstractSymbolTable) super.clone();

        if( this.delegatedSymbolTable!=null ) {

            other.delegatedSymbolTable = this.delegatedSymbolTable.clone();

        }

        other.functions = new HashMap<>(functions);

        return other;

    }

    @Override

    public Script getScript() {

        return this.script;

    }

    @Override

    public void setScript(String name) {

        ScriptManager scriptManager = ToolsLocator.getScriptManager();

        this.script = scriptManager.locateScript(name);

    }

    @Override

    public void setScript(Script script) {

        this.script = script;

    }

    public void addFunction(Function function) {

        this.functions.put(function.name(), function);

    }

    @Override

    public Function function(String name) {

        Function fn = this.functions.get(name);

        if( fn == null ) {

            if( this.delegatedSymbolTable != null ) {

                fn = this.delegatedSymbolTable.function(name);  

            }

            if( fn == null && this.script != null ) {

                fn = new ScriptFunction(this.script, name);

            }

        }

        return fn;

    }

    @Override

    public Iterator<Function> iterfuncs() {

        return this.functions.values().iterator();

    }

    @Override

    public Iterator<Function> iterator() {

        return iterfuncs();

    }

    public void setSymbolTable(SymbolTable symbolTable) {

        this.delegatedSymbolTable = symbolTable;

    }

    public SymbolTable getSymbolTable() {

        return this.delegatedSymbolTable;

    }

    public void removeFunction(String name) {

        this.functions.remove(name);

    }    

}