/*
 * Copyright © 2006 Ondra Kamenik
 * Copyright © 2019 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/>.
 */

#include "parser/cc/parser_exception.hh"
#include "utils/cc/exception.hh"

#include "dynare_atoms.hh"

#include <string>
#include <cmath>
#include <limits>
#include <sstream>
#include <iomanip>

using namespace ogdyn;
using std::string;

void
DynareStaticAtoms::register_name(string name)
{
  if (varnames.query(name))
    throw ogp::ParserException(string("The name ")+name+" is not unique.", 0);
  StaticAtoms::register_name(std::move(name));
}

int
DynareStaticAtoms::check_variable(const string &name) const
{
  if (!varnames.query(name))
    throw ogp::ParserException(std::string("Unknown name <")+name+">", 0);
  auto it = vars.find(name);
  if (it == vars.end())
    return -1;
  else
    return it->second;
}

void
DynareDynamicAtoms::parse_variable(const string &in, std::string &out, int &ll) const
{
  ll = 0;
  auto left = in.find_first_of("({");
  if (left != string::npos)
    {
      out = in.substr(0, left);
      left++;
      auto right = in.find_first_of(")}", left);
      if (string::npos == right)
        throw ogp::ParserException(string("Syntax error when parsing Dynare atom <")+in+">.", 0);
      ll = std::stoi(in.substr(left, right-left));
    }
  else
    out = in;
}

void
DynareDynamicAtoms::register_uniq_endo(string name)
{
  FineAtoms::register_uniq_endo(name);
  atom_type.emplace(std::move(name), atype::endovar);
}

void
DynareDynamicAtoms::register_uniq_exo(string name)
{
  FineAtoms::register_uniq_exo(name);
  atom_type.emplace(std::move(name), atype::exovar);
}

void
DynareDynamicAtoms::register_uniq_param(string name)
{
  FineAtoms::register_uniq_param(name);
  atom_type.emplace(std::move(name), atype::param);
}

bool
DynareDynamicAtoms::is_type(const string &name, atype tp) const
{
  auto it = atom_type.find(name);
  if (it != atom_type.end() && it->second == tp)
    return true;
  else
    return false;
}

void
DynareDynamicAtoms::print() const
{
  SAtoms::print();
  std::cout << "Name types:\n";
  for (auto it : atom_type)
    std::cout << "name=" << it.first << " type="
              << (it.second == atype::endovar ? "endovar" : it.second == atype::exovar ? "exovar" : "param")
              << '\n';
}

std::string
DynareDynamicAtoms::convert(int t) const
{
  if (t < ogp::OperationTree::num_constants)
    {
      throw ogu::Exception(__FILE__, __LINE__,
                           "Tree index is a built-in constant in DynareDynamicAtoms::convert");
      return {};
    }
  if (is_constant(t))
    {
      double v = get_constant_value(t);
      std::ostringstream buf;
      buf << std::setprecision(std::numeric_limits<double>::max_digits10)
          << v;
      return buf.str();
    }

  const string &s = name(t);
  if (is_type(s, atype::endovar))
    {
      int ll = lead(t);
      if (ll)
        return s + '(' + std::to_string(ll) + ')';
    }

  return s;
}

void
DynareAtomValues::setValues(ogp::EvalTree &et) const
{
  // set constants
  atoms.setValues(et);

  // set parameteres
  for (unsigned int i = 0; i < atoms.get_params().size(); i++)
    if (atoms.is_referenced(atoms.get_params()[i]))
      {
        const ogp::DynamicAtoms::Tlagmap &lmap = atoms.lagmap(atoms.get_params()[i]);
        for (auto it : lmap)
          {
            int t = it.second;
            et.set_nulary(t, paramvals[i]);
          }
      }

  // set endogenous
  for (unsigned int outer_i = 0; outer_i < atoms.get_endovars().size(); outer_i++)
    if (atoms.is_referenced(atoms.get_endovars()[outer_i]))
      {
        const ogp::DynamicAtoms::Tlagmap &lmap = atoms.lagmap(atoms.get_endovars()[outer_i]);
        for (auto it : lmap)
          {
            int ll = it.first;
            int t = it.second;
            int i = atoms.outer2y_endo()[outer_i];
            if (ll == -1)
              et.set_nulary(t, yym[i-atoms.nstat()]);
            else if (ll == 0)
              et.set_nulary(t, yy[i]);
            else
              et.set_nulary(t, yyp[i-atoms.nstat()-atoms.npred()]);
          }
      }

  // set exogenous
  for (unsigned int outer_i = 0; outer_i < atoms.get_exovars().size(); outer_i++)
    if (atoms.is_referenced(atoms.get_exovars()[outer_i]))
      {
        const ogp::DynamicAtoms::Tlagmap &lmap = atoms.lagmap(atoms.get_exovars()[outer_i]);
        for (auto it : lmap)
          {
            int ll = it.first;
            if (ll == 0) // this is always true because of checks
              {
                int t = it.second;
                int i = atoms.outer2y_exo()[outer_i];
                et.set_nulary(t, xx[i]);
              }
          }
      }
}

void
DynareStaticSteadyAtomValues::setValues(ogp::EvalTree &et) const
{
  // set constants
  atoms_static.setValues(et);

  // set parameters
  for (auto name : atoms_static.get_params())
    {
      int t = atoms_static.index(name);
      if (t != -1)
        {
          int idyn = atoms.name2outer_param(name);
          et.set_nulary(t, paramvals[idyn]);
        }
    }

  // set endogenous
  for (auto name : atoms_static.get_endovars())
    {
      int t = atoms_static.index(name);
      if (t != -1)
        {
          int idyn = atoms.outer2y_endo()[atoms.name2outer_endo(name)];
          et.set_nulary(t, yy[idyn]);
        }
    }

  // set exogenous
  for (auto name : atoms_static.get_exovars())
    {
      int t = atoms_static.index(name);
      if (t != -1)
        et.set_nulary(t, 0.0);
    }
}

DynareSteadySubstitutions::DynareSteadySubstitutions(const ogp::FineAtoms &a,
                                                     const ogp::OperationTree &tree,
                                                     const Tsubstmap &subst,
                                                     const Vector &pvals, Vector &yy)
  : atoms(a), y(yy)
{
  // fill the vector of left and right hand sides
  for (auto it : subst)
    {
      left_hand_sides.push_back(it.first);
      right_hand_sides.push_back(it.second);
    }

  // evaluate right hand sides
  DynareSteadyAtomValues dsav(atoms, pvals, y);
  ogp::FormulaCustomEvaluator fe(tree, right_hand_sides);
  fe.eval(dsav, *this);
}

void
DynareSteadySubstitutions::load(int i, double res)
{
  const string &name = left_hand_sides[i];
  int iouter = atoms.name2outer_endo(name);
  int iy = atoms.outer2y_endo()[iouter];
  if (!std::isfinite(y[iy]))
    y[iy] = res;
}

DynareStaticSteadySubstitutions::
DynareStaticSteadySubstitutions(const ogp::FineAtoms &a, const ogp::StaticFineAtoms &sa,
                                const ogp::OperationTree &tree,
                                const Tsubstmap &subst,
                                const Vector &pvals, Vector &yy)
  : atoms(a), atoms_static(sa), y(yy)
{
  // fill the vector of left and right hand sides
  for (const auto &it : subst)
    {
      left_hand_sides.push_back(it.first);
      right_hand_sides.push_back(it.second);
    }

  // evaluate right hand sides
  DynareStaticSteadyAtomValues dsav(atoms, atoms_static, pvals, y);
  ogp::FormulaCustomEvaluator fe(tree, right_hand_sides);
  fe.eval(dsav, *this);
}

void
DynareStaticSteadySubstitutions::load(int i, double res)
{
  const string &name = left_hand_sides[i];
  int iouter = atoms.name2outer_endo(name);
  int iy = atoms.outer2y_endo()[iouter];
  if (!std::isfinite(y[iy]))
    y[iy] = res;
}