engine_approx8.h

Go to the documentation of this file.

/* ***********************************************************************
//
//   Copyright (C) 2025 -- The 4D-STAR Collaboration
//   File Author: Emily Boudreaux
//   Last Modified: March 21, 2025
//
//   4DSSE is free software; you can use it and/or modify
//   it under the terms and restrictions the GNU General Library Public
//   License version 3 (GPLv3) as published by the Free Software Foundation.
//
//   4DSSE 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 Library General Public License for more details.
//
//   You should have received a copy of the GNU Library General Public License
//   along with this software; if not, write to the Free Software
//   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
//
// *********************************************************************** */
#pragma once
 
#include <array>
 
#include <boost/numeric/odeint.hpp>
 
#include "gridfire/network.h"

 
 
namespace gridfire::approx8{

    typedef boost::numeric::ublas::vector< double > vector_type;

    typedef boost::numeric::ublas::matrix< double > matrix_type;

    typedef std::array<double,7> vec7;

    struct Approx8Net{
        static constexpr int ih1=0;
        static constexpr int ihe3=1;
        static constexpr int ihe4=2;
        static constexpr int ic12=3;
        static constexpr int in14=4;
        static constexpr int io16=5;
        static constexpr int ine20=6;
        static constexpr int img24=7;
 
        static constexpr int iTemp=img24+1;
        static constexpr int iDensity =iTemp+1;
        static constexpr int iEnergy=iDensity+1;
 
        static constexpr int nIso=img24+1; // number of isotopes
        static constexpr int nVar=iEnergy+1; // number of variables
 
        static constexpr std::array<int,nIso> aIon = {
            1,
            3,
            4,
            12,
            14,
            16,
            20,
            24
        };
 
        static constexpr std::array<double,nIso> mIon = {
            1.67262164e-24, 
            5.00641157e-24, 
            6.64465545e-24, 
            1.99209977e-23,
            2.32462686e-23,
            2.65528858e-23,
            3.31891077e-23,
            3.98171594e-23
        };
 
    };

    double sum_product( const vec7 &a, const vec7 &b);

    vec7 get_T9_array(const double &T);

    double rate_fit(const vec7 &T9, const vec7 &coef);

    double pp_rate(const vec7 &T9);

    double dp_rate(const vec7 &T9);

    double he3he3_rate(const vec7 &T9);

    double he3he4_rate(const vec7 &T9);

    double triple_alpha_rate(const vec7 &T9);

    double c12p_rate(const vec7 &T9);

    double c12a_rate(const vec7 &T9);

    double n14p_rate(const vec7 &T9);

    double n14a_rate(const vec7 &T9);

    double n15pa_rate(const vec7 &T9);

    double n15pg_rate(const vec7 &T9);

    double n15pg_frac(const vec7 &T9);

    double o16p_rate(const vec7 &T9);

    double o16a_rate(const vec7 &T9);

    double ne20a_rate(const vec7 &T9);

    double c12c12_rate(const vec7 &T9);

    double c12o16_rate(const vec7 &T9);

    struct Jacobian {
        void operator() ( const vector_type &y, matrix_type &J, double /* t */, vector_type &dfdt ) const;
    };

    struct ODE {
        void operator() ( const vector_type &y, vector_type &dydt, double /* t */) const;
    };

    class Approx8Network final : public Network {
    public:
        Approx8Network();

        NetOut evaluate(const NetIn &netIn) override;

        void setStiff(bool stiff) override;

        bool isStiff() const override { return m_stiff; }
    private:
        vector_type m_y;
        double m_tMax = 0;
        double m_dt0 = 0;
        bool m_stiff = false;

        static vector_type convert_netIn(const NetIn &netIn);
    };
 
 
} // namespace nnApprox8