SteadyStateModel.hh 2.25 KB
Newer Older
1
/*
2
 * Copyright (C) 2010-2014 Dynare Team
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
 *
 * 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 _STEADY_STATE_MODEL_HH
#define _STEADY_STATE_MODEL_HH

#include "DataTree.hh"
24
#include "StaticModel.hh"
25
#include "WarningConsolidation.hh"
26
27
28
29

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

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

36
public:
37
  SteadyStateModel(SymbolTable &symbol_table_arg, NumericalConstants &num_constants, ExternalFunctionsTable &external_functions_table_arg, const StaticModel &static_model_arg);
38
  //! Add an expression of the form "var = expr;"
39
  void addDefinition(int symb_id, expr_t expr);
40
41
  //! Add an expression of the form "[ var1, var2, ... ] = expr;"
  void addMultipleDefinitions(const vector<int> &symb_ids, expr_t expr);
42
43
  //! Checks that definitions are in a recursive order, and that no variable is declared twice
  /*!
44
    \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
45
  */
46
  void checkPass(bool ramsey_model, WarningConsolidation &warnings) const;
47
  //! Write the steady state file
48
  /*!
49
    \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
50
  */
51
52
  void writeSteadyStateFile(const string &basename, bool ramsey_model) const;
  void writeSteadyStateFileCC(const string &basename, bool ramsey_model, bool cuda) const;
53
54
55
};

#endif