Convert triple_alpha_plus_cago to new reactions scheme

networks/triple_alpha_plus_cago/Make.package

@@ -1,11 +1,9 @@
 CEXE_headers += network_properties.H
 
 ifeq ($(USE_REACT),TRUE)
-  CEXE_sources += actual_network_data.cpp
   CEXE_headers += actual_network.H
-
-  CEXE_sources += actual_rhs_data.cpp
   CEXE_headers += actual_rhs.H
 
+  USE_RATES = TRUE
   USE_SCREENING = TRUE
 endif

networks/triple_alpha_plus_cago/actual_network.H

@@ -1,33 +1,187 @@
 #ifndef actual_network_H
 #define actual_network_H
 
+#define NEW_NETWORK_IMPLEMENTATION
+
 #include <AMReX_REAL.H>
 #include <AMReX_Vector.H>
 #include <AMReX_Array.H>
 
 #include <fundamental_constants.H>
 #include <network_properties.H>
+#include <rhs_type.H>
+#include <aprox_rates.H>
 
 using namespace amrex;
 
-void actual_network_init();
+AMREX_INLINE
+void actual_network_init() {}
 
 const std::string network_name = "triple_alpha_plus_cago";
 
 namespace network
 {
-    extern AMREX_GPU_MANAGED amrex::Array1D<amrex::Real, 1, NumSpec> bion;
+    template<int spec>
+    AMREX_GPU_HOST_DEVICE AMREX_INLINE
+    constexpr Real bion () {
+        using namespace Species;
+
+        static_assert(spec >= 1 && spec <= NumSpec);
+
+        // Set the binding energy of the element
+        if constexpr (spec == He4) {
+            return 28.29603_rt;
+        }
+        else if constexpr (spec == C12) {
+            return 92.16294_rt;
+        }
+        else if constexpr (spec == O16) {
+            return 127.62093_rt;
+        }
+        else if constexpr (spec == Fe56) {
+            return 492.25389_rt;
+        }
+
+        // Return zero if we don't recognize the species.
+
+        return 0.0_rt;
+    }
+
+    template<int spec>
+    AMREX_GPU_HOST_DEVICE AMREX_INLINE
+    constexpr Real mion () {
+        static_assert(spec >= 1 && spec <= NumSpec);
+
+        constexpr Real A = NetworkProperties::aion(spec);
+        constexpr Real Z = NetworkProperties::zion(spec);
+
+        return (A - Z) * C::Legacy::m_n + Z * (C::Legacy::m_p + C::Legacy::m_e) - bion<spec>() * C::Legacy::MeV2gr;
+    }
+
+    // Legacy (non-templated) interfaces
+
+    AMREX_GPU_HOST_DEVICE AMREX_INLINE
+    Real bion (int spec) {
+        using namespace Species;
+
+        Real b = 0.0_rt;
+
+        // Set the binding energy of the element
+        constexpr_for<1, NumSpec+1>([&] (auto n) {
+            if (n == spec) {
+                b = bion<n>();
+            }
+        });
+
+        return b;
+    }
+
+    AMREX_GPU_HOST_DEVICE AMREX_INLINE
+    Real mion (int spec) {
+        using namespace Species;
+
+        Real m = 0.0_rt;
+
+        constexpr_for<1, NumSpec+1>([&] (auto n) {
+            if (n == spec) {
+                m = mion<n>();
+            }
+        });
+
+        return m;
+    }
 }
 
 namespace Rates
 {
     enum NetworkRates {
-        ir3a   = 1,
-        ircago = 2,
-        NumRates = ircago
+        He4_He4_He4_to_C12 = 1,
+        C12_He4_to_O16,
+        NumRates = C12_He4_to_O16
     };
-
-    const int NumGroups = 2;
 };
 
+namespace RHS {
+
+    AMREX_GPU_HOST_DEVICE AMREX_INLINE
+    constexpr rhs_t rhs_data (int rate)
+    {
+        using namespace Species;
+        using namespace Rates;
+
+        rhs_t data;
+
+        switch (rate) {
+
+        case He4_He4_He4_to_C12:
+            data.species_A = He4;
+            data.species_D = C12;
+
+            data.number_A = 3;
+            data.number_D = 1;
+
+            data.exponent_A = 3;
+            data.exponent_D = 1;
+
+            // This network doesn't allow the reverse reaction
+            data.reverse_branching_ratio = 0.0_rt;
+            break;
+
+        case C12_He4_to_O16:
+            data.species_A = C12;
+            data.species_B = He4;
+            data.species_D = O16;
+
+            data.number_A = 1;
+            data.number_B = 1;
+            data.number_D = 1;
+
+            data.exponent_A = 1;
+            data.exponent_B = 1;
+            data.exponent_D = 1;
+
+            // This network doesn't allow the reverse reaction
+            data.reverse_branching_ratio = 0.0_rt;
+            break;
+
+        }
+
+        return data;
+    }
+
+    template<int rate>
+    AMREX_GPU_HOST_DEVICE AMREX_INLINE
+    void evaluate_analytical_rate (const rhs_state_t& state, rate_t& rates)
+    {
+        using namespace Species;
+        using namespace Rates;
+
+        if constexpr (rate == He4_He4_He4_to_C12) {
+            rate_triplealf(state.tf, 1.0_rt, rates.fr, rates.frdt, rates.rr, rates.rrdt);
+        }
+        else if constexpr (rate == C12_He4_to_O16) {
+            rate_c12ag(state.tf, 1.0_rt, rates.fr, rates.frdt, rates.rr, rates.rrdt);
+        }
+    }
+
+    template<int rate>
+    AMREX_GPU_HOST_DEVICE AMREX_INLINE
+    void postprocess_rate ([[maybe_unused]] const rhs_state_t& state, rate_t& rates,
+                           rate_t& rates1, rate_t& rates2, rate_t& rates3)
+    {
+        using namespace Species;
+        using namespace Rates;
+
+        // Nothing to do for this network.
+    }
+
+    template<int spec>
+    AMREX_GPU_HOST_DEVICE AMREX_INLINE
+    Real ener_gener_rate (Real const& dydt)
+    {
+        return dydt * network::mion<spec>() * C::Legacy::enuc_conv2;
+    }
+
+} // namespace RHS
+
 #endif