ModelEquationBlock.hh 3.78 KB
Newer Older
1
/*
Houtan Bastani's avatar
Houtan Bastani committed
2
 * Copyright (C) 2010-2018 Dynare Team
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
 *
 * 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/>.
 */

20
21
#ifndef _MODEL_EQUATION_BLOCK_HH
#define _MODEL_EQUATION_BLOCK_HH
22
23

#include "DataTree.hh"
24
#include "Statement.hh"
25
#include "StaticModel.hh"
Houtan Bastani's avatar
Houtan Bastani committed
26
#include "DynamicModel.hh"
27
#include "WarningConsolidation.hh"
28
29
30
31

class SteadyStateModel : public DataTree
{
private:
32
  //! Associates a set of symbol IDs (the variable(s) assigned in a given statement) to an expression (their assigned value)
33
  vector<pair<vector<int>, expr_t>> def_table;
34

35
36
37
  //! Reference to static model (for writing auxiliary equations)
  const StaticModel &static_model;

38
public:
39
40
41
42
  SteadyStateModel(SymbolTable &symbol_table_arg,
                   NumericalConstants &num_constants_arg,
                   ExternalFunctionsTable &external_functions_table_arg,
                   const StaticModel &static_model_arg);
43
44
45
46
47
48

  SteadyStateModel(const SteadyStateModel &m);
  SteadyStateModel(SteadyStateModel &&) = delete;
  SteadyStateModel & operator=(const SteadyStateModel &m);
  SteadyStateModel & operator=(SteadyStateModel &&) = delete;

49
  //! Add an expression of the form "var = expr;"
50
  void addDefinition(int symb_id, expr_t expr);
51
52
  //! Add an expression of the form "[ var1, var2, ... ] = expr;"
  void addMultipleDefinitions(const vector<int> &symb_ids, expr_t expr);
53
54
  //! Checks that definitions are in a recursive order, and that no variable is declared twice
  /*!
55
    \param[in] ramsey_model Is there a Ramsey model in the MOD file? If yes, then disable the check on the recursivity of the declarations
56
  */
57
  void checkPass(ModFileStructure &mod_file_struct, WarningConsolidation &warnings) const;
58
  //! Write the steady state file
59
  /*!
60
    \param[in] ramsey_model Is there a Ramsey model in the MOD file? If yes, then use the "ys" in argument of the steady state file as initial values
61
  */
Houtan Bastani's avatar
Houtan Bastani committed
62
  void writeSteadyStateFile(const string &basename, bool ramsey_model, bool julia) const;
63
64
  //! Writes LaTeX file with the equations of the dynamic model (for the steady state model)
  void writeLatexSteadyStateFile(const string &basename) const;
65
66
  //! Writes JSON output
  void writeJsonSteadyStateFile(ostream &output, bool transformComputingPass) const;
67
68
};

Houtan Bastani's avatar
Houtan Bastani committed
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
class Epilogue : public DynamicModel
{
private:
  //! Associates a set of symbol IDs (the variable(s) assigned in a given statement) to an expression (their assigned value)
  vector<pair<int, expr_t>> def_table;

  //! List of variables found in block
  set<int> endogs;
  set<int> exogs;
public:
  Epilogue(SymbolTable &symbol_table_arg,
           NumericalConstants &num_constants_arg,
           ExternalFunctionsTable &external_functions_table_arg,
           TrendComponentModelTable &trend_component_model_table_arg,
           VarModelTable &var_model_table_arg);

85
86
87
88
89
  Epilogue(const Epilogue &m);
  Epilogue(Epilogue &&) = delete;
  Epilogue & operator=(const Epilogue &m);
  Epilogue & operator=(Epilogue &&) = delete;

Houtan Bastani's avatar
Houtan Bastani committed
90
91
92
93
94
95
96
97
98
99
100
  //! Add an expression of the form "var = expr;"
  void addDefinition(int symb_id, expr_t expr);

  //! Checks that no variable is declared twice
  void checkPass(WarningConsolidation &warnings) const;

  //! Write the steady state file
  void writeEpilogueFile(const string &basename) const;
};


101
#endif