dynare/dynare++/parser/cc/static_atoms.cc

// Copyright © 2006, Ondra Kamenik

// $Id: static_atoms.cpp 1360 2007-07-10 11:44:20Z kamenik $

#include "static_atoms.hh"
#include "utils/cc/exception.hh"

using namespace ogp;

StaticAtoms::StaticAtoms(const StaticAtoms &a)
  : Atoms(), Constants(a), varnames(a.varnames),
    varorder(), vars(), indices()
{
  // fill varorder
  for (auto i : a.varorder)
    {
      const char *s = varnames.query(i);
      varorder.push_back(s);
    }

  // fill vars
  for (auto var : a.vars)
    {
      const char *s = varnames.query(var.first);
      vars.insert(Tvarmap::value_type(s, var.second));
    }

  // fill indices
  for (auto indice : a.indices)
    {
      const char *s = varnames.query(indice.second);
      indices.insert(Tinvmap::value_type(indice.first, s));
    }
}

void
StaticAtoms::import_atoms(const DynamicAtoms &da, OperationTree &otree, Tintintmap &tmap)
{
  Constants::import_constants(da, otree, tmap);

  for (int i = 0; i < da.get_name_storage().num(); i++)
    {
      const char *name = da.get_name_storage().get_name(i);
      register_name(name);
      int tnew = otree.add_nulary();
      assign(name, tnew);
      if (da.is_referenced(name))
        {
          const DynamicAtoms::Tlagmap &lmap = da.lagmap(name);
          for (auto it : lmap)
            {
              int told = it.second;
              tmap.insert(Tintintmap::value_type(told, tnew));
            }
        }
    }
}

int
StaticAtoms::check(const char *name) const
{
  if (DynamicAtoms::is_string_constant(name))
    {
      return Constants::check(name);
    }
  else
    {
      return check_variable(name);
    }
}

int
StaticAtoms::index(const char *name) const
{
  auto it = vars.find(name);
  if (it == vars.end())
    return -1;
  else
    return (*it).second;
}

const char *
StaticAtoms::inv_index(int t) const
{
  auto it = indices.find(t);
  if (it == indices.end())
    return nullptr;
  else
    return (*it).second;
}

void
StaticAtoms::assign(const char *name, int t)
{
  if (DynamicAtoms::is_string_constant(name))
    {
      double val;
      sscanf(name, "%lf", &val);
      add_constant(t, val);
    }
  else
    {
      const char *ss = varnames.insert(name);
      vars.insert(Tvarmap::value_type(ss, t));
      indices.insert(Tinvmap::value_type(t, ss));
    }
}

vector<int>
StaticAtoms::variables() const
{
  vector<int> res;
  for (auto var : vars)
    {
      res.push_back(var.second);
    }
  return res;
}

void
StaticAtoms::register_name(const char *name)
{
  const char *ss = varnames.insert(name);
  varorder.push_back(ss);
}

void
StaticAtoms::print() const
{
  printf("constants:\n");
  Constants::print();
  printf("variable names:\n");
  varnames.print();
  printf("map to tree indices:\n");
  for (auto var : vars)
    printf("%s\t->\t%d\n", var.first, var.second);
}