gsl_ode_solver.h

Go to the documentation of this file.

//
//
//--------------------------------------------------------------------------
//--------------------------------------------------------------------------
//  
//  libMoM Library - a library for stochastic simulations in engineering.
// 
//  Copyright (C) 2010 Rochan R. Upadhyay
// 
//  This library is free software; you can redistribute it and/or
//  modify it under the terms of the Version 2.1 GNU Lesser General
//  Public License as published by the Free Software Foundation.
// 
//  This library is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
//  Lesser General Public License for more details.
// 
//  You should have received a copy of the GNU Lesser General Public
//  License along with this library; if not, write to the Free Software
//  Foundation, Inc. 51 Franklin Street, Fifth Floor, 
//  Boston, MA  02110-1301  USA
// 
//--------------------------------------------------------------------------


#ifndef _GSL_ODE_SOLVER_H
#define _GSL_ODE_SOLVER_H

#include <gsl/gsl_odeiv.h>
#include "gsl_vector_double.h"

typedef unsigned int uint;

class ode_function
{
    public:

//
//    
    
    virtual vector operator()(double t, const vector& x);       
    
//
//    
    
    static int func(double t, const double y[], double f[], void *p);

//
//    
    
    void set_params(const vector &vpass);

//

    double *params; 

//
//

    void set_dimension(uint dim);

//

    uint dim;
    
};

//

class ode_solver 
{

    public:

//
    
    
    ode_solver(uint _dim, double _etol_abs, double _etol_rel,
                        const gsl_odeiv_step_type* type=gsl_odeiv_step_gear2);

//
//
    
    ~ode_solver();
    
//
//
    
    void set(ode_function& function);
    
//
//

    int evolve(double& t, double t1, double& step_size, vector& solution_y);
    
    private:
    
//

    void assert_set();
    
//    
    
    uint dim;
    
//    

        double etol_abs;

//

        double etol_rel;
        
//      

        bool isSet;
        
//      
        
        bool isConverged;
        
//
    
    gsl_odeiv_step * s;

//

    gsl_odeiv_control * c;

//

    gsl_odeiv_evolve * e;

//
    gsl_odeiv_system sys; 
    
};

#endif // _GSL_ODE_SOLVERER_H