double_ode.cpp

View page source

Double Speed: Ode Solution

Specifications

See link_ode .

Implementation

# include <cstring>
# include <cppad/utility/vector.hpp>
# include <cppad/speed/ode_evaluate.hpp>
# include <cppad/speed/uniform_01.hpp>

// Note that CppAD uses global_option["memory"] at the main program level
# include <map>
extern std::map<std::string, bool> global_option;

bool link_ode(
   size_t                     size       ,
   size_t                     repeat     ,
   CppAD::vector<double>      &x         ,
   CppAD::vector<double>      &jacobian
)
{
   if(global_option["onetape"]||global_option["atomic"]||global_option["optimize"])
      return false;
   // -------------------------------------------------------------
   // setup
   assert( x.size() == size );

   size_t n = size;

   size_t m = 0;
   CppAD::vector<double> f(n);

   while(repeat--)
   {  // choose next x value
      uniform_01(n, x);

      // evaluate function
      CppAD::ode_evaluate(x, m, f);

   }
   size_t i;
   for(i = 0; i < n; i++)
      jacobian[i] = f[i];
   return true;
}