Environment.hh 2.24 KB
Newer Older
Houtan Bastani's avatar
Houtan Bastani committed
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
/*
 * Copyright © 2019 Dynare Team
 *
 * This file is part of Dynare.
 *
 * Dynare is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Dynare is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Dynare.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef _ENVIRONMENT_HH
#define _ENVIRONMENT_HH

#include "ForwardDeclarationsAndEnums.hh"

#include <map>
26
#include <vector>
Houtan Bastani's avatar
Houtan Bastani committed
27
28
29
30
31
32
33
34
35
36

namespace macro
{
  class Environment
  {
  private:
    const Environment *parent;
    map<string, ExpressionPtr> variables;
    map<string, tuple<FunctionPtr, ExpressionPtr>> functions;
  public:
Sébastien Villemot's avatar
Sébastien Villemot committed
37
38
39
40
41
42
    Environment() : parent{nullptr}
    {
    }
    Environment(const Environment *parent_arg) : parent{parent_arg}
    {
    }
Houtan Bastani's avatar
Houtan Bastani committed
43
44
45
46
47
48
49
    void define(VariablePtr var, ExpressionPtr value);
    void define(FunctionPtr func, ExpressionPtr value);
    ExpressionPtr getVariable(const string &name) const;
    tuple<FunctionPtr, ExpressionPtr> getFunction(const string &name) const;
    codes::BaseType getType(const string &name);
    bool isVariableDefined(const string &name) const noexcept;
    bool isFunctionDefined(const string &name) const noexcept;
Sébastien Villemot's avatar
Sébastien Villemot committed
50
51
52
53
54
    inline bool
    isSymbolDefined(const string &name) const noexcept
    {
      return isVariableDefined(name) || isFunctionDefined(name);
    }
55
    void print(ostream &output, const vector<string> &vars, int line = -1, bool save = false) const;
Sébastien Villemot's avatar
Sébastien Villemot committed
56
57
58
59
60
61
62
63
64
65
66
67
    void printVariable(ostream &output, const string &name, int line, bool save) const;
    void printFunction(ostream &output, const tuple<FunctionPtr, ExpressionPtr> &function, int line, bool save) const;
    inline size_t
    size() const noexcept
    {
      return variables.size() + functions.size();
    }
    inline const Environment *
    getGlobalEnv() const noexcept
    {
      return parent == nullptr ? this : parent->getGlobalEnv();
    }
Houtan Bastani's avatar
Houtan Bastani committed
68
69
70
  };
}
#endif