analysis/include/ODESolver.hh

0001 //
0002 // ********************************************************************
0003 // * License and Disclaimer                                           *
0004 // *                                                                  *
0005 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
0006 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
0007 // * conditions of the Geant4 Software License,  included in the file *
0008 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
0009 // * include a list of copyright holders.                             *
0010 // *                                                                  *
0011 // * Neither the authors of this software system, nor their employing *
0012 // * institutes,nor the agencies providing financial support for this *
0013 // * work  make  any representation or  warranty, express or implied, *
0014 // * regarding  this  software system or assume any liability for its *
0015 // * use.  Please see the license in the file  LICENSE  and URL above *
0016 // * for the full disclaimer and the limitation of liability.         *
0017 // *                                                                  *
0018 // * This  code  implementation is the result of  the  scientific and *
0019 // * technical work of the GEANT4 collaboration.                      *
0020 // * By using,  copying,  modifying or  distributing the software (or *
0021 // * any work based  on the software)  you  agree  to acknowledge its *
0022 // * use  in  resulting  scientific  publications,  and indicate your *
0023 // * acceptance of all terms of the Geant4 Software license.          *
0024 // ********************************************************************
0025 //
0026 //
0027 //
0028 /// \file ODESolver.hh
0029 /// \brief Definition of the ODESolver class
0030
0031 #ifndef ODESolver_h
0032 #define ODESolver_h 1
0033
0034 #include <vector>
0035 #include <map>
0036 #include <functional>
0037
0038
0039 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
0040
0041 using myODEs = std::vector<double>(*)(double ,std::vector<double>) ;
0042 std::vector<double> operator*(const std::vector<double> v, double alfa);
0043 std::vector<double> operator+(const std::vector<double> v, double alfa);
0044 std::vector<double> operator+(const std::vector<double> v1, const std::vector<double> v2);
0045
0046 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
0047
0048 class ODESolver
0049 {
0050 public:
0051     ODESolver();
0052     ~ODESolver() = default;
0053     void Embedded_RungeKutta_Fehlberg(
0054         std::function<std::vector<double>(double,std::vector<double>)>, std::vector<double> &y,
0055         double start,double end,double stepsize=-1, double epsilon = 1e-3,
0056         std::vector<double> *time_observer=nullptr,std::vector<std::vector<double>> *state_observer=nullptr);
0057     void SetNstepsForObserver(unsigned int ndt) {fNstepsForObserver = ndt;}
0058     void RungeKutta4(
0059         std::function<std::vector<double>(double,std::vector<double>)>, std::vector<double> &y,
0060         double start,double end,double stepsize=-1,
0061         std::vector<double> *time_observer=nullptr,std::vector<std::vector<double>> *state_observer=nullptr);
0062 private:
0063     double RungeKutta_Fehlberg(std::function<std::vector<double>(double,std::vector<double>)>,
0064             std::vector<double> &y,double t, double stepsize=-1);
0065     void absValuesVector(std::vector<double> &vIn);
0066     unsigned int fNstepsForObserver{1};
0067 };
0068
0069 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
0070
0071 inline void ODESolver::absValuesVector(std::vector<double> &vIn)
0072 {
0073     for (double &val : vIn) {
0074         if (val < 0) val *= -1.;
0075     }
0076 }
0077
0078 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
0079
0080 #endif