NDSolver.hh

Go to the documentation of this file.

#pragma once
 
#include "Storage.hh"
#include "Compare.hh"
#include "Exceptions.hh"
#include "NTensor.hh"
#include "NEvaluator.hh"
 
namespace cadabra {
 
    
    class NDSolver {
        public:
            // Initialise with an Ex containing the ODEs in standard form
            // (that is, only 1st order derivatives on the lhs).
            NDSolver(const Ex&);
 
            typedef std::vector<double> state_type;
 
            // Integration function.
            void operator()(const state_type &x, state_type &dxdt, const double t);
 
            // Observer function.
            void operator()( const state_type &x , double t );
 
            // Set and check ODE(s).
            void set_ODEs(const Ex&);
 
            // Set initial values.
            void set_initial_value(const Ex&, double val);
 
            // Set integration range for the independent variable.
            void set_range(const Ex&, double from, double to);
            
            // Entry point.
            std::vector<NTensor> integrate();
            
        private:
            // Exception used by the observer function to terminate
            // iteration in case an event happens.
            
            class EventException : public CadabraException {
                public:
                    EventException(std::string="");
            };
            
            Ex ODEs;
 
            // Information extracted from `ODEs`.
            std::vector<Ex>           variables;
 
            // For each function in the ODEs we have one evaluator.
            std::vector<NEvaluator>   evaluators;
 
            // Extract from `ODEs` the right-hand side expressions as
            // well as the names of the functions to solve for.
            void extract_from_ODEs();
 
            // Storage of the result of the integration.
            std::vector<std::vector<double>> states;
            std::vector<double>              times;
            std::map<Ex, double, tree_exact_less_no_wildcards_obj>  initial_values;
            Ex                               time_variable;
            double                           range_from, range_to;
};
    
};