main/core/runtime/Scope.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

package com.jantuomi.interpreter.main.core.runtime;

import com.jantuomi.interpreter.main.core.parser.datatype.DataContainer;
import com.jantuomi.interpreter.main.exception.InterpreterException;

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

/**
 * Created by jan on 20.6.2016.
 */
public class Scope {
    private Scope parent = null;

    private Map<String, DataContainer> variables = new HashMap<>();
    private Map<String, Function> functions = new HashMap<>();

    public void addVariable(String symbol) {
        variables.put(symbol, null);
    }

    public void setVariableValue(String symbol, DataContainer value) {
        variables.replace(symbol, value);
    }

    public DataContainer resolveSymbol(String symbol, List<DataContainer> params) throws InterpreterException {
        if (functions.containsKey(symbol)) {
            return functions.get(symbol).evaluate(params);
        }
        if (variables.containsKey(symbol)) {
            return variables.get(symbol);
        }
        else if (parent != null) {
            return parent.resolveSymbol(symbol, params);
        }
        else {
            return null;
        }
    }

    public void setParent(Scope parent) {
        this.parent = parent;
    }

    public void addFunction(String symbol, Function func) {
        functions.put(symbol, func);
    }
}