net.nowina.cadmelia.script.ScriptContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cadmelia-openscad Show documentation
The newest version!
/*
 * Copyright 2017 David Naramski.
 *
 * Licensed under the EUPL, Version 1.1 or - as soon they will be approved by the
 * European Commission - subsequent versions of the EUPL (the "Licence");
 * You may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 *      http://joinup.ec.europa.eu/software/page/eupl/licence-eupl
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package net.nowina.cadmelia.script;

import java.util.HashMap;
import java.util.Map;

public class ScriptContext {

    private Map variables = new HashMap<>();

    private Map modules = new HashMap<>();

    private Map functions = new HashMap<>();

    private final ScriptContext parent;

    public ScriptContext() {
        this.parent = null;
    }

    public ScriptContext(ScriptContext parent) {
        this.parent = parent;
    }

    public Object getVariableValue(String variable) {
        if("$children".equals(variable)) {
            return ExpressionBuilder.element(0d);
        }
        if("$t".equals(variable)) {
            return ExpressionBuilder.element(0d);
        }
        Object local = variables.get(variable);
        if (local == null && parent != null) {
            return parent.getVariableValue(variable);
        }
        return local;
    }

    public void defineVariableValue(String variable, Object valueOrExpr) {
        if(variables.containsKey(variable)) {
            throw new IllegalStateException("Variable '" + variable + "' already defined");
        }
        variables.put(variable, valueOrExpr);
    }

    public void registerModule(ModuleExec module) {

        /* It is possible to override the existing modules */
        modules.put(module.getName(), module);

    }

    public void registerFunction(Function fun) {

        /* It is possible to override the existing modules */
        functions.put(fun.getName(), fun);

    }

    public ModuleExec getModule(String moduleName) {
        ModuleExec exec = modules.get(moduleName);
        if (exec == null && parent != null) {
            return parent.getModule(moduleName);
        }
        return exec;
    }

    public Function getFunction(String functionName) {
        Function exec = functions.get(functionName);
        if (exec == null && parent != null) {
            return parent.getFunction(functionName);
        }
        return exec;
    }

}