Line data    Source code

       1            1 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : /// \file
       5              : /// Defines metafunctions for manipulating Tags that refer to Tensors
       6              : 
       7              : #pragma once
       8              : 
       9              : #include "DataStructures/Tensor/IndexType.hpp"
      10              : #include "NumericalAlgorithms/LinearOperators/PartialDerivatives.hpp"
      11              : #include "Utilities/TMPL.hpp"
      12              : 
      13              : namespace TensorMetafunctions {
      14              : /// \ingroup TensorGroup
      15              : /// \brief Replaces Tag with an equivalent Tag but in frame NewFrame
      16              : template <typename Tag, typename NewFrame>
      17            1 : struct replace_frame_in_tag {
      18              :   // Base definition, for tags with no template parameters,
      19              :   // like some scalars.
      20            0 :   using type = Tag;
      21              : };
      22              : // Specialization for tensors in gr::Tags / gh::Tags
      23              : template <template <typename, size_t, typename> typename Tag, size_t Dim,
      24              :           typename Frame, typename DataType, typename NewFrame>
      25            0 : struct replace_frame_in_tag<Tag<DataType, Dim, Frame>, NewFrame> {
      26            0 :   using type = Tag<DataType, Dim, NewFrame>;
      27              : };
      28              : // Specialization for scalars in gr::Tags (which have a DataType).
      29              : template <template <typename> typename Tag, typename DataType,
      30              :           typename NewFrame>
      31            0 : struct replace_frame_in_tag<Tag<DataType>, NewFrame> {
      32            0 :   using type = Tag<DataType>;
      33              : };
      34              : // Specializations for Tags::deriv<Tag> with Tag in gh::Tags
      35              : template <template <typename, size_t, typename> typename Tag, typename DataType,
      36              :           size_t Dim, typename Frame, typename NewFrame>
      37            0 : struct replace_frame_in_tag<
      38              :     ::Tags::deriv<Tag<DataType, Dim, Frame>, tmpl::size_t<Dim>, Frame>,
      39              :     NewFrame> {
      40            0 :   using type =
      41              :       ::Tags::deriv<Tag<DataType, Dim, NewFrame>, tmpl::size_t<Dim>, NewFrame>;
      42              : };
      43              : template <template <typename> typename Tag, typename DataType, size_t Dim,
      44              :           typename Frame, typename NewFrame>
      45            0 : struct replace_frame_in_tag<
      46              :     ::Tags::deriv<Tag<DataType>, tmpl::size_t<Dim>, Frame>, NewFrame> {
      47            0 :   using type = ::Tags::deriv<Tag<DataType>, tmpl::size_t<Dim>, NewFrame>;
      48              : };
      49              : // Specialization for inverse jacobian
      50              : template <template <size_t, typename, typename> typename Tag, size_t Dim,
      51              :           typename SrcFrame, typename TargetFrame, typename NewTargetFrame>
      52            0 : struct replace_frame_in_tag<Tag<Dim, SrcFrame, TargetFrame>, NewTargetFrame> {
      53            0 :   using type = Tag<Dim, SrcFrame, NewTargetFrame>;
      54              : };
      55              : 
      56              : /// \ingroup TensorGroup
      57              : /// \brief Replaces Tag with an equivalent Tag but in frame NewFrame
      58              : template <typename Tag, typename NewFrame>
      59            1 : using replace_frame_in_tag_t =
      60              :     typename replace_frame_in_tag<Tag, NewFrame>::type;
      61              : 
      62              : /// \ingroup TensorGroup
      63              : /// \brief Replaces every Tag in Taglist with an equivalent Tag
      64              : /// but in frame NewFrame
      65              : template <typename TagList, typename NewFrame>
      66            1 : using replace_frame_in_taglist =
      67              :     tmpl::transform<TagList,
      68              :                     tmpl::bind<replace_frame_in_tag_t, tmpl::_1, NewFrame>>;
      69              : }  // namespace TensorMetafunctions