Line data    Source code

       1            0 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : #pragma once
       5              : 
       6              : #include "Evolution/Imex/Protocols/ImexSystem.hpp"
       7              : #include "Evolution/Imex/Tags/ImplicitHistory.hpp"
       8              : #include "Evolution/Imex/Tags/Mode.hpp"
       9              : #include "Evolution/Imex/Tags/SolveFailures.hpp"
      10              : #include "Evolution/Imex/Tags/SolveTolerance.hpp"
      11              : #include "Time/History.hpp"
      12              : #include "Time/Tags/HistoryEvolvedVariables.hpp"
      13              : #include "Utilities/Gsl.hpp"
      14              : #include "Utilities/MakeWithValue.hpp"
      15              : #include "Utilities/ProtocolHelpers.hpp"
      16              : #include "Utilities/TMPL.hpp"
      17              : 
      18              : namespace imex {
      19              : /// Create the IMEX structures and options.
      20              : template <typename System, typename = typename System::implicit_sectors>
      21            1 : struct Initialize;
      22              : 
      23              : /// \cond
      24              : template <typename System, typename... Sectors>
      25              : struct Initialize<System, tmpl::list<Sectors...>> {
      26              :   static_assert(tt::assert_conforms_to_v<System, protocols::ImexSystem>);
      27              : 
      28              :   using variables_tag = typename System::variables_tag;
      29              :   using example_tensor_tag =
      30              :       tmpl::front<typename tmpl::front<tmpl::list<Sectors...>>::tensors>;
      31              : 
      32              :   using const_global_cache_tags = tmpl::list<Tags::Mode, Tags::SolveTolerance>;
      33              :   using mutable_global_cache_tags = tmpl::list<>;
      34              :   using simple_tags_from_options = tmpl::list<>;
      35              :   using simple_tags = tmpl::list<Tags::ImplicitHistory<Sectors>...,
      36              :                                  Tags::SolveFailures<Sectors>...>;
      37              :   using compute_tags = tmpl::list<>;
      38              : 
      39              :   using return_tags = simple_tags;
      40              :   using argument_tags =
      41              :       tmpl::list<::Tags::HistoryEvolvedVariables<variables_tag>,
      42              :                  example_tensor_tag>;
      43              : 
      44              :   static void apply(
      45              :       const gsl::not_null<
      46              :           typename Tags::ImplicitHistory<Sectors>::type*>... histories,
      47              :       const gsl::not_null<
      48              :           typename Tags::SolveFailures<Sectors>::type*>... solve_failures,
      49              :       const TimeSteppers::History<typename variables_tag::type>&
      50              :           explicit_history,
      51              :       const typename example_tensor_tag::type& example_tensor) {
      52              :     const auto order = explicit_history.integration_order();
      53              :     expand_pack((histories->integration_order(order), 0)...);
      54              :     expand_pack(*solve_failures = make_with_value<Scalar<DataVector>>(
      55              :                     example_tensor, 0.0)...);
      56              :   }
      57              : };
      58              : /// \endcond
      59              : }  // namespace imex