Line data    Source code

       1            0 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : #pragma once
       5              : 
       6              : #include <algorithm>
       7              : #include <cstddef>
       8              : #include <unordered_set>
       9              : 
      10              : #include "DataStructures/DataVector.hpp"
      11              : #include "DataStructures/Index.hpp"
      12              : #include "DataStructures/Variables.hpp"
      13              : #include "Domain/Structure/Direction.hpp"
      14              : #include "Domain/Structure/DirectionMap.hpp"
      15              : #include "Domain/Structure/DirectionalId.hpp"
      16              : #include "Domain/Structure/DirectionalIdMap.hpp"
      17              : #include "Evolution/DgSubcell/SliceData.hpp"
      18              : #include "Utilities/Gsl.hpp"
      19              : 
      20              : namespace evolution::dg::subcell {
      21              : /// @{
      22              : /*!
      23              :  * \brief Slice a Variables object on subcell mesh for a given direction and
      24              :  * slicing depth (number of ghost points).
      25              :  */
      26              : template <size_t Dim, typename TagList>
      27            1 : void slice_variable(
      28              :     const gsl::not_null<Variables<TagList>*>& sliced_subcell_vars,
      29              :     const Variables<TagList>& volume_subcell_vars,
      30              :     const Index<Dim>& subcell_extents, const size_t ghost_zone_size,
      31              :     const Direction<Dim>& direction,
      32              :     const DirectionalIdMap<Dim, std::optional<intrp::Irregular<Dim>>>&
      33              :         fd_to_neighbor_fd_interpolants) {
      34              :   // check the size of sliced_subcell_vars (output)
      35              :   const size_t num_sliced_pts =
      36              :       subcell_extents.slice_away(direction.dimension()).product() *
      37              :       ghost_zone_size;
      38              :   if (sliced_subcell_vars->size() != num_sliced_pts) {
      39              :     sliced_subcell_vars->initialize(num_sliced_pts);
      40              :   }
      41              : 
      42              :   // Slice volume variables. Note that the return type of the
      43              :   // `slice_data_impl()` function is DataVector.
      44              :   std::unordered_set directions_to_slice{direction};
      45              : 
      46              :   const DataVector sliced_data{detail::slice_data_impl(
      47              :       gsl::make_span(volume_subcell_vars.data(), volume_subcell_vars.size()),
      48              :       subcell_extents, ghost_zone_size, directions_to_slice, 0,
      49              :       fd_to_neighbor_fd_interpolants)[direction]};
      50              : 
      51              :   // copy the returned DataVector data into sliced variables
      52              :   std::copy(sliced_data.begin(), sliced_data.end(),
      53              :             sliced_subcell_vars->data());
      54              : }
      55              : 
      56              : template <size_t Dim, typename TagList>
      57            1 : Variables<TagList> slice_variable(
      58              :     const Variables<TagList>& volume_subcell_vars,
      59              :     const Index<Dim>& subcell_extents, const size_t ghost_zone_size,
      60              :     const Direction<Dim>& direction,
      61              :     const DirectionalIdMap<Dim, std::optional<intrp::Irregular<Dim>>>&
      62              :         fd_to_neighbor_fd_interpolants) {
      63              :   Variables<TagList> sliced_subcell_vars{
      64              :       subcell_extents.slice_away(direction.dimension()).product() *
      65              :       ghost_zone_size};
      66              :   slice_variable(make_not_null(&sliced_subcell_vars), volume_subcell_vars,
      67              :                  subcell_extents, ghost_zone_size, direction,
      68              :                  fd_to_neighbor_fd_interpolants);
      69              :   return sliced_subcell_vars;
      70              : }
      71              : /// @}
      72              : }  // namespace evolution::dg::subcell