init_static_hmc.hpp

Go to the documentation of this file.

#ifndef STAN__COMMON__INIT_STATIC_HMC_HPP
#define STAN__COMMON__INIT_STATIC_HMC_HPP

#include <stan/mcmc/base_mcmc.hpp>
#include <stan/gm/arguments/argument.hpp>
#include <stan/gm/arguments/categorical_argument.hpp>
#include <stan/gm/arguments/singleton_argument.hpp>

namespace stan {
  namespace common {
    
    template<class Sampler>
    bool init_static_hmc(stan::mcmc::base_mcmc* sampler, 
                         stan::gm::argument* algorithm) {

      stan::gm::categorical_argument* hmc 
        = dynamic_cast<stan::gm::categorical_argument*>
        (algorithm->arg("hmc"));
      
      stan::gm::categorical_argument* base 
        = dynamic_cast<stan::gm::categorical_argument*>
        (algorithm->arg("hmc")->arg("engine")->arg("static"));
      
      double epsilon 
        = dynamic_cast<stan::gm::real_argument*>
        (hmc->arg("stepsize"))->value();
      double epsilon_jitter 
        = dynamic_cast<stan::gm::real_argument*>
        (hmc->arg("stepsize_jitter"))->value();
      double int_time 
        = dynamic_cast<stan::gm::real_argument*>(base->arg("int_time"))->value();
      
      dynamic_cast<Sampler*>(sampler)->set_nominal_stepsize_and_T(epsilon, int_time);
      dynamic_cast<Sampler*>(sampler)->set_stepsize_jitter(epsilon_jitter);
      
      return true;
    }
    
  }
}

#endif