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 "Time/Tags/TimeStepper.hpp"
       8             : #include "Time/TimeSteppers/TimeStepper.hpp"
       9             : #include "Utilities/ErrorHandling/Error.hpp"
      10             : #include "Utilities/Gsl.hpp"
      11             : #include "Utilities/ProtocolHelpers.hpp"
      12             : #include "Utilities/TMPL.hpp"
      13             : 
      14             : /// \cond
      15             : namespace Parallel {
      16             : template <typename Metavariables>
      17             : struct GlobalCache;
      18             : }  // namespace Parallel
      19             : namespace Tags {
      20             : struct Time;
      21             : }  // namespace Tags
      22             : namespace db {
      23             : template <typename TagsList>
      24             : class DataBox;
      25             : }  // namespace db
      26             : namespace imex::Tags {
      27             : template <typename ImplicitSector>
      28             : struct ImplicitHistory;
      29             : }  // namespace imex::Tags
      30             : namespace tuples {
      31             : template <class... Tags>
      32             : class TaggedTuple;
      33             : }  // namespace tuples
      34             : /// \endcond
      35             : 
      36             : namespace imex {
      37             : /// Mutator to apply the implicit portion of dense output, intended
      38             : /// for use in `RunEventsAndDenseTriggers`.
      39             : template <typename ImexSystem>
      40           1 : struct ImplicitDenseOutput {
      41             :   static_assert(tt::assert_conforms_to_v<ImexSystem, protocols::ImexSystem>);
      42             : 
      43           0 :   using return_tags = tmpl::list<typename ImexSystem::variables_tag>;
      44           0 :   using argument_tags = tmpl::push_front<
      45             :       tmpl::transform<typename ImexSystem::implicit_sectors,
      46             :                       tmpl::bind<Tags::ImplicitHistory, tmpl::_1>>,
      47             :       ::Tags::TimeStepper<TimeStepper>, ::Tags::Time>;
      48             : 
      49             :   template <typename... Histories>
      50           0 :   static void apply(
      51             :       const gsl::not_null<typename ImexSystem::variables_tag::type*> variables,
      52             :       const TimeStepper& time_stepper, const double time,
      53             :       const Histories&... histories) {
      54             :     const auto update_sector = [&](auto sector_v, auto history) {
      55             :       using sector = tmpl::type_from<decltype(sector_v)>;
      56             :       auto sector_variables =
      57             :           variables->template reference_subset<typename sector::tensors>();
      58             :       const bool success = time_stepper.dense_update_u(
      59             :           make_not_null(&sector_variables), history, time);
      60             :       if (not success) {
      61             :         ERROR("IMEX dense output failed with history:\n" << history);
      62             :       }
      63             :     };
      64             :     tmpl::as_pack<typename ImexSystem::implicit_sectors>(
      65             :         [&](auto... sectors_v) {
      66             :           EXPAND_PACK_LEFT_TO_RIGHT(update_sector(sectors_v, histories));
      67             :         });
      68             :   }
      69             : 
      70             :   template <typename DbTagsList, typename... InboxTags, typename Metavariables,
      71             :             typename ArrayIndex, typename ParallelComponent>
      72           0 :   static bool is_ready(
      73             :       const gsl::not_null<db::DataBox<DbTagsList>*> /*box*/,
      74             :       const gsl::not_null<tuples::TaggedTuple<InboxTags...>*> /*inboxes*/,
      75             :       Parallel::GlobalCache<Metavariables>& /*cache*/,
      76             :       const ArrayIndex& /*array_index*/,
      77             :       const ParallelComponent* /*component*/) {
      78             :     return true;
      79             :   }
      80             : };
      81             : }  // namespace imex