Intrepid2_CellData.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                           Intrepid2 Package
//
// Copyright 2007 NTESS and the Intrepid2 contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef __INTREPID2_CELLDATA_HPP__
#define __INTREPID2_CELLDATA_HPP__

#include "Intrepid2_ConfigDefs.hpp"

#include "Shards_CellTopology.hpp"

#include "Intrepid2_Types.hpp"
#include "Intrepid2_Utils.hpp"
#include "Intrepid2_Kernels.hpp"

namespace Intrepid2 {

//============================================================================================//
//                                                                                            //
//                                RefSubcellParametrization                                   //
//                                                                                            //
//============================================================================================//

template<typename DeviceType>
class RefSubcellParametrization {
public:
  using ConstViewType = Kokkos::DynRankView<const double,DeviceType>;

  RefSubcellParametrization() = default;

  ~RefSubcellParametrization() = default;


  static inline bool
  isSupported( const unsigned cellTopoKey );


  static inline
  ConstViewType
  get( const ordinal_type          subcellDim,
      const unsigned              parentCellKey );

private:

  static void set();

  template <typename HostViewType>
  static void
  set(       HostViewType          subcellParam,
      const ordinal_type          subcellDim,
      const shards::CellTopology  parentCell );

  using ViewType = Kokkos::DynRankView<double,DeviceType>;
  static ViewType lineEdgesParam;  // edge maps for 2d non-standard cells; shell line and beam
  static ViewType triEdgesParam, quadEdgesParam; // edge maps for 2d standard cells
  static ViewType shellTriEdgesParam, shellQuadEdgesParam; // edge maps for 3d non-standard cells; shell tri and quad
  static ViewType tetEdgesParam, hexEdgesParam, pyrEdgesParam, wedgeEdgesParam; // edge maps for 3d standard cells
  static ViewType shellTriFacesParam, shellQuadFacesParam; // face maps for 3d non-standard cells
  static ViewType tetFacesParam, hexFacesParam, pyrFacesParam, wedgeFacesParam; // face maps for 3d standard cells

  static bool isSubcellParametrizationSet_;


};



//============================================================================================//
//                                                                                            //
//                                        RefCellNodes                                        //
//                                                                                            //
//============================================================================================//


template<typename DeviceType>
class RefCellNodes {
public:
  using ConstViewType = Kokkos::DynRankView<const double,DeviceType>;

  RefCellNodes() = default;

  ~RefCellNodes() = default;


  static inline
  ConstViewType
  get(const unsigned      cellTopoKey);

private:
  static void set();

  using ViewType = Kokkos::DynRankView<double,DeviceType>;
  static ViewType lineNodes, line3Nodes;
  static ViewType triangleNodes, triangle4Nodes, triangle6Nodes;
  static ViewType quadrilateralNodes, quadrilateral8Nodes, quadrilateral9Nodes;
  static ViewType tetrahedronNodes, tetrahedron8Nodes, tetrahedron10Nodes, tetrahedron11Nodes;
  static ViewType hexahedronNodes, hexahedron20Nodes, hexahedron27Nodes;
  static ViewType pyramidNodes, pyramid13Nodes, pyramid14Nodes;
  static ViewType wedgeNodes, wedge15Nodes, wedge18Nodes;


  struct ReferenceNodeDataStatic {
    double line[2][3], line_3[3][3];
    double triangle[3][3], triangle_4[4][3], triangle_6[6][3];
    double quadrilateral[4][3], quadrilateral_8[8][3], quadrilateral_9[9][3];
    double tetrahedron[4][3], tetrahedron_8[8][3], tetrahedron_10[10][3], tetrahedron_11[10][3];
    double hexahedron[8][3], hexahedron_20[20][3], hexahedron_27[27][3];
    double pyramid[5][3], pyramid_13[13][3], pyramid_14[14][3];
    double wedge[6][3], wedge_15[15][3], wedge_18[18][3];
  };

  static const ReferenceNodeDataStatic refNodeDataStatic_;

  static bool isReferenceNodeDataSet_;

};

//============================================================================================//
//                                                                                            //
//                                       RefCellCenter                                        //
//                                                                                            //
//============================================================================================//


template<typename DeviceType>
class RefCellCenter {
public:
  using ConstViewType = Kokkos::DynRankView<const double,DeviceType>;


  RefCellCenter() = default;

  ~RefCellCenter() = default;


  static inline
  ConstViewType
  get(const unsigned      cellTopoKey);

private:
  static void set();

  using ViewType = Kokkos::DynRankView<double,DeviceType>;
  static ViewType lineCenter;
  static ViewType triangleCenter;
  static ViewType quadrilateralCenter;
  static ViewType tetrahedronCenter;
  static ViewType hexahedronCenter;
  static ViewType pyramidCenter;
  static ViewType wedgeCenter;

  struct ReferenceCenterDataStatic {
    double line[3];
    double triangle[3];
    double quadrilateral[3];
    double tetrahedron[3];
    double hexahedron[3];
    double pyramid[3];
    double wedge[3];
  };

  static  const ReferenceCenterDataStatic refCenterDataStatic_;

  static bool isReferenceCellCenterDataSet_;
};

//============================================================================================//
//                                                                                            //
//                                       PointInclusion                                       //
//                                                                                            //
//============================================================================================//


template<unsigned CellTopologyKey>
  struct PointInclusion;
  
  template<>
  struct PointInclusion<shards::Line<>::key> {
    template<typename PointViewType, typename ScalarType>
    KOKKOS_INLINE_FUNCTION
    static bool
    check(const PointViewType &point, const ScalarType threshold);   
  };
  
  template<>
  struct PointInclusion<shards::Triangle<>::key> {
    template<typename PointViewType, typename ScalarType>
    KOKKOS_INLINE_FUNCTION
    static bool
    check(const PointViewType &point, const ScalarType threshold);
  };
  
  template<>
  struct PointInclusion<shards::Quadrilateral<>::key> {

    template<typename PointViewType, typename ScalarType>
    KOKKOS_INLINE_FUNCTION
    static bool
    check(const PointViewType &point, const ScalarType threshold);
  };
    
  template<>
  struct PointInclusion<shards::Tetrahedron<>::key> {
    template<typename PointViewType, typename ScalarType>
    KOKKOS_INLINE_FUNCTION
    static bool
    check(const PointViewType &point, const ScalarType threshold);
  };

  template<>
  struct PointInclusion<shards::Hexahedron<>::key> {
    template<typename PointViewType, typename ScalarType>
    KOKKOS_INLINE_FUNCTION
    static bool
    check(const PointViewType &point, const ScalarType threshold);
  };
  
  template<>
  struct PointInclusion<shards::Pyramid<>::key> {
    template<typename PointViewType, typename ScalarType>
    KOKKOS_INLINE_FUNCTION
    static bool
    check(const PointViewType &point, const ScalarType threshold);
  };

  template<>
  struct PointInclusion<shards::Wedge<>::key> {
    template<typename PointViewType, typename ScalarType>
    KOKKOS_INLINE_FUNCTION
    static bool
    check(const PointViewType &point, const ScalarType threshold);
  };

  const CellTopologyData* getCellTopologyData(const unsigned& cellTopologyKey);
  
}

#include "Intrepid2_CellDataDef.hpp"

#endif