Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
  • python-codegen
  • 6.x protected
  • llvm-15
  • julia-6.4.0
  • julia
  • julia-meson
  • 5.x protected
  • julia-6.3.0
  • 4.6 protected
  • uop
  • rework_pac
  • aux_vars_fix
  • julia-6.2.0
  • created_preprocessor_repo
15 results

Statement.cc

Blame
  • Statement.cc 10.36 KiB
    /*
     * Copyright (C) 2006-2018 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/>.
     */
    
    #include "Statement.hh"
    #include <boost/xpressive/xpressive.hpp>
    #include <utility>
    
    ModFileStructure::ModFileStructure() :
      dsge_var_calibrated("")
    {
    }
    
    Statement::~Statement()
    = default;
    
    void
    Statement::checkPass(ModFileStructure &mod_file_struct, WarningConsolidation &warnings)
    {
    }
    
    void
    Statement::writeCOutput(ostream &output, const string &basename)
    {
    }
    
    void
    Statement::writeJuliaOutput(ostream &output, const string &basename)
    {
    }
    
    void
    Statement::writeJsonOutput(ostream &output) const
    {
    }
    
    void
    Statement::computingPass()
    {
    }
    
    NativeStatement::NativeStatement(string native_statement_arg) :
      native_statement(move(native_statement_arg))
    {
    }
    
    void
    NativeStatement::writeOutput(ostream &output, const string &basename, bool minimal_workspace) const
    {
      using namespace boost::xpressive;
      string date_regex = "(-?\\d+([YyAa]|[Mm]([1-9]|1[0-2])|[Qq][1-4]|[Ww]([1-9]{1}|[1-4]\\d|5[0-2])))";
      sregex regex_lookbehind = sregex::compile("(?<!\\$|\\d|[a-zA-Z_]|\\')" + date_regex);
      sregex regex_dollar = sregex::compile("(\\$)"+date_regex);
    
      string ns = regex_replace(native_statement, regex_lookbehind, "dates('$&')");
      ns = regex_replace(ns, regex_dollar, "$2"); //replace $DATE with DATE
      output << ns << endl;
    }
    
    void
    NativeStatement::writeJsonOutput(ostream &output) const
    {
      output << "{\"statementName\": \"native\""
             << ", \"string\": \"" << native_statement << "\""
             << "}";
    }
    
    VerbatimStatement::VerbatimStatement(string verbatim_statement_arg) :
      verbatim_statement(move(verbatim_statement_arg))
    {
    }
    
    void
    VerbatimStatement::writeOutput(ostream &output, const string &basename, bool minimal_workspace) const
    {
      output << verbatim_statement << endl;
    }
    
    void
    VerbatimStatement::writeJsonOutput(ostream &output) const
    {
      output << "{\"statementName\": \"verbatim\""
             << ", \"string\": \"" << verbatim_statement << "\""
             << "}";
    }
    
    void
    OptionsList::writeOutput(ostream &output) const
    {
      for (const auto & num_option : num_options)
        output << "options_." << num_option.first << " = " << num_option.second << ";" << endl;
    
      for (const auto & paired_num_option : paired_num_options)
        output << "options_." << paired_num_option.first << " = [" << paired_num_option.second.first << "; "
               << paired_num_option.second.second << "];" << endl;
    
      for (const auto & string_option : string_options)
        output << "options_." << string_option.first << " = '" << string_option.second << "';" << endl;
    
      for (const auto & date_option : date_options)
        output << "options_." << date_option.first << " = " << date_option.second << ";" << endl;
    
      for (const auto & symbol_list_option : symbol_list_options)
        symbol_list_option.second.writeOutput("options_." + symbol_list_option.first, output);
    
      for (const auto & vector_int_option : vector_int_options)
        {
          output << "options_." << vector_int_option.first << " = ";
          if (vector_int_option.second.size() > 1)
            {
              output << "[";
              for (auto viit = vector_int_option.second.begin();
                   viit != vector_int_option.second.end(); viit++)
                output << *viit << ";";
              output << "];" << endl;
            }
          else
            output << vector_int_option.second.front() << ";" << endl;
        }
    
      for (const auto & vector_str_option : vector_str_options)
        {
          output << "options_." << vector_str_option.first << " = ";
          if (vector_str_option.second.size() > 1)
            {
              output << "{";
              for (auto viit = vector_str_option.second.begin();
                   viit != vector_str_option.second.end(); viit++)
                output << "'" << *viit << "';";
              output << "};" << endl;
            }
          else
            output << vector_str_option.second.front() << ";" << endl;
        }
    }
    
    void
    OptionsList::writeOutput(ostream &output, const string &option_group) const
    {
      // Initialize option_group as an empty struct iff the field does not exist!
      unsigned idx = option_group.find_last_of(".");
      if (idx < UINT_MAX)
        {
          output << "if ~isfield(" << option_group.substr(0, idx) << ",'" << option_group.substr(idx+1) << "')" << endl;
          output << "    " << option_group << " = struct();" << endl;
          output << "end" << endl;
        }
      else
        output << option_group << " = struct();" << endl;
    
      for (const auto & num_option : num_options)
        output << option_group << "." << num_option.first << " = " << num_option.second << ";" << endl;
    
      for (const auto & paired_num_option : paired_num_options)
        output << option_group << "." << paired_num_option.first << " = [" << paired_num_option.second.first << "; "
               << paired_num_option.second.second << "];" << endl;
    
      for (const auto & string_option : string_options)
        output << option_group << "." << string_option.first << " = '" << string_option.second << "';" << endl;
    
      for (const auto & date_option : date_options)
        output << option_group << "." << date_option.first << " = " << date_option.second << ";" << endl;
    
      for (const auto & symbol_list_option : symbol_list_options)
        symbol_list_option.second.writeOutput(option_group + "." + symbol_list_option.first, output);
    
      for (const auto & vector_int_option : vector_int_options)
        {
          output << option_group << "." << vector_int_option.first << " = ";
          if (vector_int_option.second.size() > 1)
            {
              output << "[";
              for (auto viit = vector_int_option.second.begin();
                   viit != vector_int_option.second.end(); viit++)
                output << *viit << ";";
              output << "];" << endl;
            }
          else
            output <<  vector_int_option.second.front() << ";" << endl;
        }
    
      for (const auto & vector_str_option : vector_str_options)
        {
          output << option_group << "." << vector_str_option.first << " = ";
          if (vector_str_option.second.size() > 1)
            {
              output << "{";
              for (auto viit = vector_str_option.second.begin();
                   viit != vector_str_option.second.end(); viit++)
                output << "'" << *viit << "';";
              output << "};" << endl;
            }
          else
            output <<  vector_str_option.second.front() << ";" << endl;
        }
    }
    
    void
    OptionsList::writeJsonOutput(ostream &output) const
    {
      if (getNumberOfOptions() == 0)
        return;
    
      output << "\"options\": {";
      for (auto it = num_options.begin();
           it != num_options.end();)
        {
          output << "\""<< it->first << "\": " << it->second;
          it++;
          if (it != num_options.end()
              || !(paired_num_options.empty()
                   && string_options.empty()
                   && date_options.empty()
                   && symbol_list_options.empty()
                   && vector_int_options.empty()))
            output << ", ";
        }
    
      for (auto it = paired_num_options.begin();
           it != paired_num_options.end();)
        {
          output << "\""<< it->first << "\": [" << it->second.first << " " << it->second.second << "]";
          it++;
          if (it != paired_num_options.end()
              || !(string_options.empty()
                   && date_options.empty()
                   && symbol_list_options.empty()
                   && vector_int_options.empty()))
            output << ", ";
        }
    
      for (auto it = string_options.begin();
           it != string_options.end();)
        {
          output << "\""<< it->first << "\": \"" << it->second << "\"";
          it++;
          if (it != string_options.end()
              || !(date_options.empty()
                   && symbol_list_options.empty()
                   && vector_int_options.empty()))
            output << ", ";
        }
    
      for (auto it = date_options.begin();
           it != date_options.end();)
        {
          output << "\""<< it->first << "\": \"" << it->second << "\"";
          it++;
          if (it != date_options.end()
              || !(symbol_list_options.empty()
                   && vector_int_options.empty()))
            output << ", ";
        }
    
      for (auto it = symbol_list_options.begin();
           it != symbol_list_options.end(); it++)
        {
          output << "\""<< it->first << "\":";
          it->second.writeJsonOutput(output);
          it++;
          if (it != symbol_list_options.end()
              || !vector_int_options.empty())
            output << ", ";
        }
    
      for (auto it = vector_int_options.begin();
           it != vector_int_options.end();)
        {
          output << "\""<< it->first << "\": [";
          if (it->second.size() > 1)
            {
              for (auto viit = it->second.begin();
                   viit != it->second.end();)
                {
                  output << *viit;
                  viit++;
                  if (viit != it->second.end())
                    output << ", ";
                }
            }
          else
            output << it->second.front() << endl;
          output << "]";
          it++;
          if (it != vector_int_options.end())
            output << ", ";
        }
    
    
      for (auto it = vector_str_options.begin();
           it != vector_str_options.end();)
        {
          output << "\""<< it->first << "\": [";
          if (it->second.size() > 1)
            {
              for (auto viit = it->second.begin();
                   viit != it->second.end();)
                {
                  output << "\"" << *viit << "\"";
                  viit++;
                  if (viit != it->second.end())
                    output << ", ";
                }
            }
          else
            output << it->second.front() << endl;
          output << "]";
          it++;
          if (it != vector_str_options.end())
            output << ", ";
        }
    
      output << "}";
    }
    
    void
    OptionsList::clear()
    {
      num_options.clear();
      paired_num_options.clear();
      string_options.clear();
      date_options.clear();
      symbol_list_options.clear();
      vector_int_options.clear();
      vector_str_options.clear();
    }
    
    int
    OptionsList::getNumberOfOptions() const
    {
      return num_options.size()
        + paired_num_options.size()
        + string_options.size()
        + date_options.size()
        + symbol_list_options.size()
        + vector_int_options.size()
        + vector_str_options.size();
    }