Line data    Source code

       1           0 : // Distributed under the MIT License.
       2             : // See LICENSE.txt for details.
       3             : 
       4             : #pragma once
       5             : 
       6             : #include <cstddef>
       7             : 
       8             : #include "DataStructures/Tensor/EagerMath/FrameTransform.hpp"
       9             : #include "DataStructures/Tensor/Metafunctions.hpp"
      10             : #include "DataStructures/Tensor/TypeAliases.hpp"
      11             : #include "DataStructures/Variables.hpp"
      12             : #include "Utilities/Gsl.hpp"
      13             : 
      14             : namespace transform {
      15             : 
      16             : /// Tags to represent the result of frame-transforming Variables
      17           1 : namespace Tags {
      18             : /// The `Tag` with the first index transformed to a different frame
      19             : template <typename Tag, typename FirstIndexFrame>
      20           1 : struct TransformedFirstIndex : db::SimpleTag {
      21           0 :   using type = TensorMetafunctions::prepend_spatial_index<
      22             :       TensorMetafunctions::remove_first_index<typename Tag::type>,
      23             :       tmpl::front<typename Tag::type::index_list>::dim, UpLo::Up,
      24             :       FirstIndexFrame>;
      25             : };
      26             : }  // namespace Tags
      27             : 
      28             : /// @{
      29             : /// Transforms the first index of all tensors in the Variables to a different
      30             : /// frame
      31             : ///
      32             : /// See single-Tensor overload for details.
      33             : template <typename... ResultTags, typename... InputTags, size_t Dim,
      34             :           typename SourceFrame, typename TargetFrame>
      35           1 : void first_index_to_different_frame(
      36             :     const gsl::not_null<Variables<tmpl::list<ResultTags...>>*> result,
      37             :     const Variables<tmpl::list<InputTags...>>& input,
      38             :     const InverseJacobian<DataVector, Dim, SourceFrame, TargetFrame>&
      39             :         inv_jacobian) {
      40             :   EXPAND_PACK_LEFT_TO_RIGHT(
      41             :       first_index_to_different_frame(make_not_null(&get<ResultTags>(*result)),
      42             :                                      get<InputTags>(input), inv_jacobian));
      43             : }
      44             : 
      45             : template <typename... InputTags, size_t Dim, typename SourceFrame,
      46             :           typename TargetFrame,
      47             :           typename ResultVars = Variables<tmpl::list<
      48             :               Tags::TransformedFirstIndex<InputTags, SourceFrame>...>>>
      49           1 : ResultVars first_index_to_different_frame(
      50             :     const Variables<tmpl::list<InputTags...>>& input,
      51             :     const InverseJacobian<DataVector, Dim, SourceFrame, TargetFrame>&
      52             :         inv_jacobian) {
      53             :   ResultVars result{input.number_of_grid_points()};
      54             :   first_index_to_different_frame(make_not_null(&result), input, inv_jacobian);
      55             :   return result;
      56             : }
      57             : /// @}
      58             : 
      59             : }  // namespace transform