integrator_Leapfrog.cpp

Go to the documentation of this file.

#include "integrator_Leapfrog.h"

#ifdef USE_PARAMETERS_FACTORY
#include "parameters_factory.h"
#endif

//- parameter entries
namespace {
  void append_entry(Parameters& param)
  {
    param.Register_int("level", 0);
    param.Register_int("number_of_steps", 0);

    param.Register_string("verbose_level", "NULL");
  }


#ifdef USE_PARAMETERS_FACTORY
  bool init_param = ParametersFactory::Register("Integrator.Leapfrog", append_entry);
#endif
}
//- end

//- parameters class
Parameters_Integrator_Leapfrog::Parameters_Integrator_Leapfrog() { append_entry(*this); }
//- end

const std::string Integrator_Leapfrog::class_name = "Integrator_Leapfrog";

//====================================================================
void Integrator_Leapfrog::set_parameters(const Parameters& params)
{
  const string str_vlevel = params.get_string("verbose_level");

  m_vl = vout.set_verbose_level(str_vlevel);

  //- fetch and check input parameters
  int level;
  int Nstep;

  int err = 0;
  err += params.fetch_int("level", level);
  err += params.fetch_int("number_of_steps", Nstep);

  if (err) {
    vout.crucial(m_vl, "%s: fetch error, input parameter not found.\n", class_name.c_str());
    exit(EXIT_FAILURE);
  }

  set_parameters(level, Nstep);
}


//====================================================================
void Integrator_Leapfrog::set_parameters(int level, int Nstep)
{
  //- print input parameters
  vout.general(m_vl, "Parameters of %s:\n", class_name.c_str());
  vout.general(m_vl, "  Level: %4d\n", level);
  vout.general(m_vl, "  Nstep    = %4d\n", Nstep);

  //- range check
  // NB. level,Estep,Nstep == 0 is allowed.

  //- store values
  m_Nstep = Nstep;
  m_level = level;
}


//====================================================================
void Integrator_Leapfrog::set_parameter_level(const int level)
{
  m_level = level;
}


//====================================================================
void Integrator_Leapfrog::set_parameter_nstep(const int nstep)
{
  m_Nstep = nstep;
}


//====================================================================
void Integrator_Leapfrog::set_parameter_nsteps(const std::vector<int>& nsteps)
{
  if (nsteps.size() > 0) {
    set_parameter_nstep(nsteps[0]);

    // transfer to lower levels
    if (m_update_u && (nsteps.size() > 1)) {
      std::vector<int> next_steps(nsteps.begin() + 1, nsteps.end());
      m_update_u->set_parameter_nsteps(next_steps);
    }
  }
}


//====================================================================
void Integrator_Leapfrog::evolve(const double step_size, Field_G& iP, Field_G& U)
{
  int Nin  = U.nin();
  int Nvol = U.nvol();
  int Nex  = U.nex();
  int Nc   = CommonParameters::Nc();

  vout.general(m_vl, "Integration level-%d start.\n", m_level);

  if (m_Nstep > 0) {
    double estep = step_size / m_Nstep;

    // initial half step
    m_update_p->evolve(estep * 0.5, iP, U);

    for (int istep = 1; istep <= m_Nstep; ++istep) {
      vout.general(m_vl, "istep = %d\n", istep);

      m_update_u->evolve(estep, iP, U);

      if (istep < m_Nstep) {
        m_update_p->evolve(estep, iP, U);
      }
    }

    // last half step
    m_update_p->evolve(estep * 0.5, iP, U);
  } else {
    vout.general(m_vl, "Nstep is zero. skip.\n");
  }

  vout.general(m_vl, "Integration level-%d finished.\n", m_level);
}


//====================================================================
//============================================================END=====