CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
List of all members | Public Member Functions | Private Member Functions | Private Attributes
RK2 Class Reference

#include <ConstantStepOdeSolver.h>

Inheritance diagram for RK2:
ConstantStepOdeSolver

Public Member Functions

const char * methodName () const
 
 RK2 ()
 
 RK2 (const AbsODERHS &rhs)
 
- Public Member Functions inherited from ConstantStepOdeSolver
 ConstantStepOdeSolver ()
 
 ConstantStepOdeSolver (const AbsODERHS &rhs)
 
 ConstantStepOdeSolver (const ConstantStepOdeSolver &r)
 
double getCoordinate (const unsigned which, const unsigned idx) const
 
double getIntegrated (unsigned which, unsigned idx) const
 
double getPeakTime (unsigned which) const
 
const AbsODERHSgetRHS () const
 
AbsODERHSgetRHS ()
 
double getTime (const unsigned idx) const
 
double interpolateCoordinate (unsigned which, double t, bool cubic=false) const
 
double interpolateIntegrated (unsigned which, double t, bool cubic=false) const
 
double lastDeltaT () const
 
unsigned lastDim () const
 
double lastMaxT () const
 
unsigned lastRunLength () const
 
ConstantStepOdeSolveroperator= (const ConstantStepOdeSolver &r)
 
void run (const double *initialConditions, unsigned lenConditions, double dt, unsigned nSteps)
 
void setHistory (double dt, const double *data, unsigned dim, unsigned runLen)
 
void setRHS (const AbsODERHS &rhs)
 
void truncateCoordinate (unsigned which, double minValue, double maxValue)
 
void writeHistory (std::ostream &os, double dt, bool cubic=false) const
 
void writeIntegrated (std::ostream &os, unsigned which, double dt, bool cubic=false) const
 
virtual ~ConstantStepOdeSolver ()
 

Private Member Functions

void step (double t, double dt, const double *x, unsigned lenX, double *coordIncrement) const
 

Private Attributes

std::vector< double > buf_
 

Additional Inherited Members

- Protected Attributes inherited from ConstantStepOdeSolver
AbsODERHSrhs_
 

Detailed Description

Definition at line 149 of file ConstantStepOdeSolver.h.

Constructor & Destructor Documentation

RK2::RK2 ( )
inline

Definition at line 152 of file ConstantStepOdeSolver.h.

RK2::RK2 ( const AbsODERHS rhs)
inlineexplicit

Definition at line 154 of file ConstantStepOdeSolver.h.

Member Function Documentation

const char* RK2::methodName ( ) const
inlinevirtual

Implements ConstantStepOdeSolver.

Definition at line 156 of file ConstantStepOdeSolver.h.

156 {return "2nd order Runge-Kutta";}
void RK2::step ( double  t,
double  dt,
const double *  x,
unsigned  lenX,
double *  coordIncrement 
) const
privatevirtual

Implements ConstantStepOdeSolver.

Definition at line 354 of file ConstantStepOdeSolver.cc.

References buf_, AbsODERHS::calc(), i, and ConstantStepOdeSolver::rhs_.

357 {
358  const double halfstep = dt/2.0;
359  if (buf_.size() < lenX)
360  buf_.resize(lenX);
361  double* midpoint = &buf_[0];
362  rhs_->calc(t, x, lenX, midpoint);
363  for (unsigned i=0; i<lenX; ++i)
364  {
365  midpoint[i] *= halfstep;
366  midpoint[i] += x[i];
367  }
368  rhs_->calc(t + halfstep, midpoint, lenX, coordIncrement);
369  for (unsigned i=0; i<lenX; ++i)
370  coordIncrement[i] *= dt;
371 }
tuple t
Definition: tree.py:139
float dt
Definition: AMPTWrapper.h:126
int i
Definition: DBlmapReader.cc:9
virtual void calc(double t, const double *x, unsigned lenX, double *derivative)=0
std::vector< double > buf_

Member Data Documentation

std::vector<double> RK2::buf_
mutableprivate

Definition at line 163 of file ConstantStepOdeSolver.h.

Referenced by step().