coupled_ode_system.hpp

Go to the documentation of this file.

#ifndef STAN__MATH__ODE__COUPLED_ODE_SYSTEM_HPP
#define STAN__MATH__ODE__COUPLED_ODE_SYSTEM_HPP

#include <ostream>
#include <vector>

#include <stan/math/error_handling/matrix/check_matching_sizes.hpp>

namespace stan {

  namespace math {

    template <typename F, typename T1, typename T2>
    struct coupled_ode_system {
    };


    template <typename F>
    struct coupled_ode_system<F, double, double> {

      const F& f_;
      const std::vector<double>& y0_dbl_;
      const std::vector<double>& theta_dbl_;
      const std::vector<double>& x_;
      const std::vector<int>& x_int_;
      const int N_;
      const int M_;
      const int size_;
      std::ostream* msgs_;
      
      coupled_ode_system(const F& f,
                         const std::vector<double>& y0,
                         const std::vector<double>& theta,
                         const std::vector<double>& x,
                         const std::vector<int>& x_int,
                         std::ostream* msgs)
        : f_(f),  
          y0_dbl_(y0),
          theta_dbl_(theta),
          x_(x), 
          x_int_(x_int), 
          N_(y0.size()),
          M_(theta.size()),
          size_(N_),
          msgs_(msgs) {
      }

      void operator()(const std::vector<double>& y,
                      std::vector<double>& dy_dt,
                      double t) {
        dy_dt = f_(t,y,theta_dbl_,x_,x_int_,msgs_);
        stan::math::check_matching_sizes("coupled_ode_system(%1%)",
                                         y,"y",dy_dt,"dy_dt",
                                         static_cast<double*>(0));
      }

      int size() const {
        return size_;
      }
      
      std::vector<double> initial_state() {
        std::vector<double> state(size_, 0.0);
        for (int n = 0; n < N_; n++)
          state[n] = y0_dbl_[n];
        return state;
      }

      std::vector<std::vector<double> > 
      decouple_states(const std::vector<std::vector<double> >& y) {
        return y;
      }

    };

  }

}

#endif