Skip to content
Snippets Groups Projects
Select Git revision
  • c56d58822e6c581926ac765b7604b29e1ecdf557
  • master default protected
  • pylib
  • 5.x
  • wasm
  • julia
  • 4.6
  • uop
  • rework_pac
  • aux_vars_fix
  • julia-6.2.0
  • created_preprocessor_repo
12 results

SymbolList.hh

Blame
  • Forked from Dynare / preprocessor
    Source project has a limited visibility.
    Shocks.hh 8.60 KiB
    /*
     * Copyright © 2003-2022 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 <https://www.gnu.org/licenses/>.
     */
    
    #ifndef _SHOCKS_HH
    #define _SHOCKS_HH
    
    #include <string>
    #include <vector>
    #include <map>
    
    #include "Statement.hh"
    #include "SymbolTable.hh"
    #include "ExprNode.hh"
    
    using namespace std;
    
    class AbstractShocksStatement : public Statement
    {
    public:
      // The tuple is (period1, period2, value)
      using det_shocks_t = map<int, vector<tuple<int, int, expr_t>>>;
    protected:
      //! Is this statement a "mshocks" statement ? (instead of a "shocks" statement)
      const bool mshocks;
      //! Does this "shocks" statement replace the previous ones?
      const bool overwrite;
      const det_shocks_t det_shocks;
      const SymbolTable &symbol_table;
      void writeDetShocks(ostream &output) const;
      void writeJsonDetShocks(ostream &output) const;
    
      AbstractShocksStatement(bool mshocks_arg, bool overwrite_arg,
                              det_shocks_t det_shocks_arg,
                              const SymbolTable &symbol_table_arg);
    };
    
    class ShocksStatement : public AbstractShocksStatement
    {
    public:
      using var_and_std_shocks_t = map<int, expr_t>;
      using covar_and_corr_shocks_t = map<pair<int, int>, expr_t>;
    private:
      const var_and_std_shocks_t var_shocks, std_shocks;
      const covar_and_corr_shocks_t covar_shocks, corr_shocks;
      void writeVarOrStdShock(ostream &output, const pair<int, expr_t> &it, bool stddev) const;
      void writeVarAndStdShocks(ostream &output) const;
      void writeCovarOrCorrShock(ostream &output, const pair<pair<int, int>, expr_t> &it, bool corr) const;
      void writeCovarAndCorrShocks(ostream &output) const;
      bool has_calibrated_measurement_errors() const;
    public:
      ShocksStatement(bool overwrite_arg,
                      det_shocks_t det_shocks_arg,
                      var_and_std_shocks_t var_shocks_arg,
                      var_and_std_shocks_t std_shocks_arg,
                      covar_and_corr_shocks_t covar_shocks_arg,
                      covar_and_corr_shocks_t corr_shocks_arg,
                      const SymbolTable &symbol_table_arg);
      void checkPass(ModFileStructure &mod_file_struct, WarningConsolidation &warnings) override;
      void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
      void writeJsonOutput(ostream &output) const override;
    };
    
    class MShocksStatement : public AbstractShocksStatement
    {
    public:
      MShocksStatement(bool overwrite_arg,
                       det_shocks_t det_shocks_arg,
                       const SymbolTable &symbol_table_arg);
      void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
      void writeJsonOutput(ostream &output) const override;
    };
    
    /* Represents a shocks(surprise) block.
       Given the differences with the plain “shocks” block, it was easier to make
       it a separate class. */
    class ShocksSurpriseStatement : public Statement
    {
    public:
      //! Does this "shocks(surprise)" statement replace the previous ones?
      const bool overwrite;
      const AbstractShocksStatement::det_shocks_t surprise_shocks;
    private:
      const SymbolTable &symbol_table;
    public:
      ShocksSurpriseStatement(bool overwrite_arg,
                              AbstractShocksStatement::det_shocks_t surprise_shocks_arg,
                              const SymbolTable &symbol_table_arg);
      void checkPass(ModFileStructure &mod_file_struct, WarningConsolidation &warnings) override;
      void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
      void writeJsonOutput(ostream &output) const override;
    };
    
    /* Represents a shocks(learnt_in=…) block.
       Given the differences with the plain “shocks” block, it was easier to make
       it a separate class. */
    class ShocksLearntInStatement : public Statement
    {
    public:
      const int learnt_in_period;
      //! Does this "shocks(learnt_in=…)" statement replace the previous ones?
      const bool overwrite;
      enum class LearntShockType
        {
          level,
          add,
          multiply
        };
      // The tuple is (type, period1, period2, value)
      using learnt_shocks_t = map<int, vector<tuple<LearntShockType, int, int, expr_t>>>;
      const learnt_shocks_t learnt_shocks;
    private:
      const SymbolTable &symbol_table;
      static string typeToString(LearntShockType type);
    public:
      ShocksLearntInStatement(int learnt_in_period_arg, bool overwrite_arg,
                              learnt_shocks_t learnt_shocks_arg,
                              const SymbolTable &symbol_table_arg);
      void checkPass(ModFileStructure &mod_file_struct, WarningConsolidation &warnings) override;
      void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
      void writeJsonOutput(ostream &output) const override;
    };
    
    class ConditionalForecastPathsStatement : public Statement
    {
    private:
      const AbstractShocksStatement::det_shocks_t paths;
      const SymbolTable &symbol_table;
      const int path_length;
    public:
      ConditionalForecastPathsStatement(AbstractShocksStatement::det_shocks_t paths_arg,
                                        const SymbolTable &symbol_table_arg);
      void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
      void writeJsonOutput(ostream &output) const override;
      static int computePathLength(const AbstractShocksStatement::det_shocks_t &paths);
    };
    
    class MomentCalibration : public Statement
    {
    public:
      struct Constraint
      {
        int endo1, endo2;
        string lags;
        expr_t lower_bound, upper_bound;
      };
      using constraints_t = vector<Constraint>;
    private:
      constraints_t constraints;
      const SymbolTable &symbol_table;
    public:
      MomentCalibration(constraints_t constraints_arg,
                        const SymbolTable &symbol_table_arg);
      void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
      void writeJsonOutput(ostream &output) const override;
    };
    
    class IrfCalibration : public Statement
    {
    public:
      struct Constraint
      {
        int endo;
        int exo;
        string periods;
        expr_t lower_bound, upper_bound;
      };
      using constraints_t = vector<Constraint>;
    private:
      constraints_t constraints;
      const SymbolTable &symbol_table;
      const OptionsList options_list;
    public:
      IrfCalibration(constraints_t constraints_arg,
                     const SymbolTable &symbol_table_arg,
                     OptionsList options_list_arg);
      void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
      void writeJsonOutput(ostream &output) const override;
    };
    
    class ShockGroupsStatement : public Statement
    {
    public:
      struct Group
      {
        string name;
        vector<string> list;
      };
      using group_t = vector<Group>;
    private:
      group_t shock_groups;
      string name;
    public:
      ShockGroupsStatement(group_t shock_groups_arg, string name_arg);
      void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
      void writeJsonOutput(ostream &output) const override;
    };
    
    class Init2shocksStatement : public Statement
    {
    private:
      const vector<pair<int, int>> init2shocks;
      const string name;
      const SymbolTable &symbol_table;
    public:
      Init2shocksStatement(vector<pair<int, int>> init2shocks_arg, string name_arg, const SymbolTable &symbol_table_arg);
      void checkPass(ModFileStructure &mod_file_struct, WarningConsolidation &warnings) override;
      void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
      void writeJsonOutput(ostream &output) const override;
    };
    
    class HeteroskedasticShocksStatement : public Statement
    {
    public:
      // Maps exo symb_id to list of tuples (period1, period2, value/scale)
      using heteroskedastic_shocks_t = map<int, vector<tuple<int, int, expr_t>>>;
    private:
      const bool overwrite;
      const heteroskedastic_shocks_t values, scales;
      const SymbolTable &symbol_table;
    public:
      HeteroskedasticShocksStatement(bool overwrite_arg, const heteroskedastic_shocks_t &values_arg,
                                     const heteroskedastic_shocks_t &scales_arg,
                                     const SymbolTable &symbol_table_arg);
      void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
      void writeJsonOutput(ostream &output) const override;
    };
    
    #endif