Line data    Source code

       1            0 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : #pragma once
       5              : 
       6              : #include <array>
       7              : 
       8              : #include "DataStructures/DataBox/Prefixes.hpp"
       9              : #include "DataStructures/DataVector.hpp"
      10              : #include "DataStructures/Tensor/Tensor.hpp"
      11              : #include "DataStructures/Tensor/TypeAliases.hpp"
      12              : #include "Evolution/Systems/RadiationTransport/M1Grey/Tags.hpp"
      13              : #include "PointwiseFunctions/GeneralRelativity/TagsDeclarations.hpp"
      14              : #include "Utilities/Gsl.hpp"
      15              : #include "Utilities/TMPL.hpp"
      16              : 
      17              : namespace RadiationTransport {
      18              : namespace M1Grey {
      19              : 
      20              : // Implementation of the M1 fluxes for individual neutrino species
      21              : namespace detail {
      22              : void compute_fluxes_impl(
      23              :     gsl::not_null<tnsr::I<DataVector, 3, Frame::Inertial>*> tilde_e_flux,
      24              :     gsl::not_null<tnsr::Ij<DataVector, 3, Frame::Inertial>*> tilde_s_flux,
      25              :     gsl::not_null<tnsr::I<DataVector, 3, Frame::Inertial>*> tilde_s_M,
      26              :     const Scalar<DataVector>& tilde_e,
      27              :     const tnsr::i<DataVector, 3, Frame::Inertial>& tilde_s,
      28              :     const tnsr::II<DataVector, 3, Frame::Inertial>& tilde_p,
      29              :     const Scalar<DataVector>& lapse,
      30              :     const tnsr::I<DataVector, 3, Frame::Inertial>& shift,
      31              :     const tnsr::ii<DataVector, 3, Frame::Inertial>& spatial_metric,
      32              :     const tnsr::II<DataVector, 3, Frame::Inertial>& inv_spatial_metric);
      33              : }  // namespace detail
      34              : 
      35              : /*!
      36              :  * \brief The fluxes of the conservative variables in the M1 scheme
      37              :  *
      38              :  * \f{align*}
      39              :  * F^i({\tilde E}) = &~ \alpha \gamma^{ij} {\tilde S}_j - \beta^j {\tilde E} \\
      40              :  * F^i({\tilde S}_j) = &~ \alpha {\tilde P}^{ik} \gamma_{kj} - \beta^i {\tilde
      41              :  * S}_j \f}
      42              :  *
      43              :  * where the conserved variables \f${\tilde E}\f$, \f${\tilde S}_i\f$,
      44              :  * are a generalized mass-energy density and momentum density.
      45              :  * Furthermore, \f${\tilde P^{ij}}\f$ is the pressure tensor density of the
      46              :  * radiation field, \f$\alpha\f$ is the lapse, \f$\beta^i\f$ is the shift,
      47              :  * \f$\gamma_{ij}\f$ the 3-metric, and \f$\gamma^{ij}\f$ its inverse.
      48              :  *
      49              :  * In the main function, we loop over all neutrino species, and then call
      50              :  * the actual implementation of the fluxes.
      51              :  */
      52              : template <typename... NeutrinoSpecies>
      53            1 : struct ComputeFluxes {
      54            0 :   using return_tags =
      55              :       tmpl::list<::Tags::Flux<Tags::TildeE<Frame::Inertial, NeutrinoSpecies>,
      56              :                               tmpl::size_t<3>, Frame::Inertial>...,
      57              :                  ::Tags::Flux<Tags::TildeS<Frame::Inertial, NeutrinoSpecies>,
      58              :                               tmpl::size_t<3>, Frame::Inertial>...>;
      59              : 
      60            0 :   using argument_tags =
      61              :       tmpl::list<Tags::TildeE<Frame::Inertial, NeutrinoSpecies>...,
      62              :                  Tags::TildeS<Frame::Inertial, NeutrinoSpecies>...,
      63              :                  Tags::TildeP<Frame::Inertial, NeutrinoSpecies>...,
      64              :                  gr::Tags::Lapse<DataVector>, gr::Tags::Shift<DataVector, 3>,
      65              :                  gr::Tags::SpatialMetric<DataVector, 3>,
      66              :                  gr::Tags::InverseSpatialMetric<DataVector, 3>>;
      67              : 
      68            0 :   static void apply(
      69              :       const gsl::not_null<typename ::Tags::Flux<
      70              :           Tags::TildeE<Frame::Inertial, NeutrinoSpecies>, tmpl::size_t<3>,
      71              :           Frame::Inertial>::type*>... tilde_e_flux,
      72              :       const gsl::not_null<typename ::Tags::Flux<
      73              :           Tags::TildeS<Frame::Inertial, NeutrinoSpecies>, tmpl::size_t<3>,
      74              :           Frame::Inertial>::type*>... tilde_s_flux,
      75              :       const typename Tags::TildeE<Frame::Inertial,
      76              :                                   NeutrinoSpecies>::type&... tilde_e,
      77              :       const typename Tags::TildeS<Frame::Inertial,
      78              :                                   NeutrinoSpecies>::type&... tilde_s,
      79              :       const typename Tags::TildeP<Frame::Inertial,
      80              :                                   NeutrinoSpecies>::type&... tilde_p,
      81              :       const Scalar<DataVector>& lapse,
      82              :       const tnsr::I<DataVector, 3, Frame::Inertial>& shift,
      83              :       const tnsr::ii<DataVector, 3, Frame::Inertial>& spatial_metric,
      84              :       const tnsr::II<DataVector, 3, Frame::Inertial>& inv_spatial_metric) {
      85              :     // Allocate memory for tildeS^i
      86              :     tnsr::I<DataVector, 3, Frame::Inertial> tilde_s_M(get(lapse).size());
      87              :     EXPAND_PACK_LEFT_TO_RIGHT(detail::compute_fluxes_impl(
      88              :         tilde_e_flux, tilde_s_flux, &tilde_s_M, tilde_e, tilde_s, tilde_p,
      89              :         lapse, shift, spatial_metric, inv_spatial_metric));
      90              :   }
      91              : };
      92              : }  // namespace M1Grey
      93              : }  // namespace RadiationTransport