/*
 * Copyright (C) 2006-2015 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 _STATEMENT_HH
#define _STATEMENT_HH

#include <ostream>
#include <string>
#include <map>
#include <set>

#include "SymbolList.hh"
#include "WarningConsolidation.hh"

class ModFileStructure
{
public:
  ModFileStructure();
  //! Whether check is present
  bool check_present;
  //! Whether steady is present
  bool steady_present;
  //! Whether a perfect_foresight_solver/simul statement is present
  bool perfect_foresight_solver_present;
  //! Whether a stoch_simul statement is present
  bool stoch_simul_present;
  //! Whether an estimation statement is present
  bool estimation_present;
  //! Whether an osr statement is present
  bool osr_present;
  //! Whether an osr params statement is present
  bool osr_params_present;
  //! Whether an optim weight statement is present
  bool optim_weights_present;
  //! Whether a ramsey_model statement is present
  bool ramsey_model_present;
  //! Whether a ramsey_policy statement is present
  bool ramsey_policy_present;
  //! Whether a discretionary_objective statement is present
  bool discretionary_policy_present;
  //! Whether a planner_objective statement is present
  bool planner_objective_present;
  //! Whether an extended_path statement is present
  bool extended_path_present;
  //! The value of the "order" option of stoch_simul, estimation, osr, ramsey_policy
  //! Derivation order
  /*! First initialized to zero. If user sets order option somewhere in the MOD file, it will be equal to the maximum of order options. Otherwise will default to 2 */
  int order_option;
  //! Whether a bvar_density, bvar_forecast, sbvar, ms_sbvar statement is present
  bool bvar_present;
  //! Whether an svar_identification statement is present
  bool svar_identification_present;
  //! Whether an identification statement is present or the identification option of dynare_sensitivity statement is equal to one
  bool identification_present;
  //! Whether the option analytic_derivation is given to estimation
  bool estimation_analytic_derivation;
  //! Whether the option partial_information is given to stoch_simul/estimation/osr/ramsey_policy
  bool partial_information;
  //! Whether a histval bloc is present
  /*! Used for the workaround for trac ticket #157 */
  bool histval_present;
  //! Whether the "k_order_solver" option is used (explictly, or implicitly if order >= 3)
  bool k_order_solver;
  //! Whether there is a calibrated measurement error
  bool calibrated_measurement_errors;
  //! Whether dsge_prior_weight was initialized as a parameter
  bool dsge_prior_weight_initialized;
  //! Whether dsge_prior_weight is in the estimated_params block
  bool dsge_prior_weight_in_estimated_params;
  //! Whether there is a dsge_var, with calibrated prior weight
  string dsge_var_calibrated;
  //! Whether there is a dsge_var, with prior weight that must be estimated
  bool dsge_var_estimated;
  //! Whether there is a bayesian_irf option passed to the estimation statement
  bool bayesian_irf_present;
  //! Whether there is a data statement present
  bool estimation_data_statement_present;
  //! Last chain number for Markov Switching statement
  int last_markov_switching_chain;
  //! Whether a calib_smoother statement is present
  bool calib_smoother_present;
  //! Whether there is an estimated_params_init with use_calibration
  bool estim_params_use_calib;
  //! Set of parameters used within shocks blocks, inside the expressions
  //! defining the values of covariances (stored as symbol ids)
  set<int> parameters_within_shocks_values;
  //! Set of estimated parameters (stored as symbol ids)
  set<int> estimated_parameters;
  //! Whether there is a prior statement present
  bool prior_statement_present;
  //! Whether there is a std prior statement present
  bool std_prior_statement_present;
  //! Whether there is a corr prior statement present
  bool corr_prior_statement_present;
  //! Whether there is a options statement present
  bool options_statement_present;
  //! Whether there is a std options statement present
  bool std_options_statement_present;
  //! Whether there is a corr options statement present
  bool corr_options_statement_present;
  //! Whether a Markov Switching DSGE is present
  bool ms_dsge_present;
  //! Whether occbin is present
  bool occbin_option;
};

class Statement
{
public:
  virtual ~Statement();
  //! Do some internal check, and fill the ModFileStructure class
  /*! Don't forget to update ComputingTasks.hh, Shocks.hh and
    NumericalInitialization.hh if you modify the signature of this
    method. Otherwise the default implementation (i.e. a no-op) will apply and
    some checks won't be run. */
  virtual void checkPass(ModFileStructure &mod_file_struct, WarningConsolidation &warnings);
  virtual void computingPass();
  //! Write Matlab output code
  /*!
    \param output is the output stream of the main matlab file
    \param basename is the name of the modfile (without extension) which can be used to build auxiliary files
  */
  virtual void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const = 0;
  virtual void writeCOutput(ostream &output, const string &basename);
};

class NativeStatement : public Statement
{
private:
  const string native_statement;
public:
  NativeStatement(const string &native_statement_arg);
  virtual void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const;
};

class VerbatimStatement : public Statement
{
private:
  const string verbatim_statement;
public:
  VerbatimStatement(const string &verbatim_statement_arg);
  virtual void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const;
};

class OptionsList
{
public:
  typedef map<string, string> num_options_t;
  typedef map<string, pair<string, string> > paired_num_options_t;
  typedef map<string, string> string_options_t;
  typedef map<string, string> date_options_t;
  typedef map<string, SymbolList> symbol_list_options_t;
  typedef map<string, vector<int> > vec_int_options_t;
  num_options_t num_options;
  paired_num_options_t paired_num_options;
  string_options_t string_options;
  date_options_t date_options;
  symbol_list_options_t symbol_list_options;
  vec_int_options_t vector_int_options;
  void writeOutput(ostream &output) const;
  void writeOutput(ostream &output, const string &option_group) const;
  void clear();
};

#endif // ! _STATEMENT_HH
