DynareMain.cc

/*
 * Copyright (C) 2003-2008 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/>.
 */

using namespace std;

#include <iostream>
#include <sstream>
#include <fstream>

#include <cctype>    // for tolower()
#include <algorithm> // for transform()

#include "macro/MacroDriver.hh"

/* Prototype for second part of main function
   Splitting main() in two parts was necessary because ParsingDriver.h and MacroDriver.h can't be
   included simultaneously (because of Bison limitations).
*/
void main2(stringstream &in, string &basename, bool debug, bool clear_all);

int
main(int argc, char** argv)
{
  if (argc < 2)
    {
      cerr << "Missing model file!" << endl;
      cerr << "Dynare usage: dynare mod_file [debug] [noclearall] [savemacro]" << endl;
      exit(-1);
    }

  bool clear_all = true;
  bool save_macro = false;
  bool debug = false;

  // Parse options
  for (int arg = 2; arg < argc; arg++)
    {
      if (string(argv[arg]) == string("debug"))
        debug = true;
      else if (string(argv[arg]) == string("noclearall"))
        clear_all = false;
      else if (string(argv[arg]) == string("savemacro"))
        save_macro = true;
    }

  cout << "Starting Dynare ..." << endl;
  cout << "Parsing your model file ..." << endl;

  // Construct basename (check file extension is correct then remove it)
  string basename = argv[1];
  string ext = basename.substr(basename.size() - 4);
  transform(ext.begin(), ext.end(), ext.begin(), (int(*)(int)) tolower); // Convert ext to lowercase
  if (ext != string(".mod") && ext != string(".dyn"))
    {
      cerr << "mod_file extension must be .mod or .dyn!" << endl;
      exit(-1);
    }
  basename.erase(basename.size() - 4, 4);

  // Do macro processing
  MacroDriver m;

  stringstream macro_output;
  m.parse(argv[1], macro_output, debug);
  if (save_macro)
    {
      ofstream macro_output_file((basename + "-macroexp.mod").c_str());
      macro_output_file << macro_output.str();
      macro_output_file.close();
    }

  // Do the rest
  main2(macro_output, basename, debug, clear_all);

  return 0;
}