Intrepid2_ProjectionStruct.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_PROJECTIONSTRUCT_HPP__
#define __INTREPID2_PROJECTIONSTRUCT_HPP__

#include "Intrepid2_ConfigDefs.hpp"
#include "Intrepid2_Types.hpp"

#include "Intrepid2_NodalBasisFamily.hpp"

#include <array>

namespace Intrepid2 {

ordinal_type
KOKKOS_INLINE_FUNCTION
range_size(const Kokkos::pair<ordinal_type, ordinal_type>& range) {
  return range.second - range.first;
}

template<typename DeviceType, typename ValueType>
class ProjectionStruct {
public:

  enum EvalPointsType {BASIS, TARGET};

  using range_type = Kokkos::pair<ordinal_type,ordinal_type>;
  using HostExecutionSpaceType = Kokkos::DefaultHostExecutionSpace;
  using HostMemorySpaceType = Kokkos::HostSpace;
  using MemSpaceType = typename DeviceType::memory_space;
  using HostDeviceType = Kokkos::Device<HostExecutionSpaceType,HostMemorySpaceType>;
  using view_type = Kokkos::DynRankView<ValueType,DeviceType >;
  using host_view_type = Kokkos::DynRankView<ValueType,HostDeviceType >;
  using range_tag = Kokkos::View<range_type**,HostDeviceType>;
  static constexpr int numberSubCellDims = 4; //{0 for vertex, 1 for edges, 2 for faces, 3 for volumes}
  //max of numVertices, numEdges, numFaces for a reference cell.
  //12 is the number of edges in a Hexahderon.
  //We'll need to change this if we consider generic polyhedra
  static constexpr int maxSubCellsCount = 12;
  using view_tag = std::array<std::array<host_view_type, maxSubCellsCount>, numberSubCellDims>;
  using key_tag = Kokkos::View<unsigned**,HostDeviceType >;


  ordinal_type getNumBasisEvalPoints() {
    return numBasisEvalPoints;
  }

  ordinal_type getNumBasisDerivEvalPoints() {
    return numBasisDerivEvalPoints;
  }

  ordinal_type getNumTargetEvalPoints() {
    return numTargetEvalPoints;
  }

  ordinal_type getNumTargetDerivEvalPoints() {
    return numTargetDerivEvalPoints;
  }

  ordinal_type getMaxNumDerivPoints(const EvalPointsType type) const {
    if(type == BASIS)
      return maxNumBasisDerivEvalPoints;
    else
      return maxNumTargetDerivEvalPoints;
  }

  ordinal_type getMaxNumEvalPoints(const EvalPointsType type) const {
    if(type == BASIS)
      return maxNumBasisEvalPoints;
    else
      return maxNumTargetEvalPoints;
  }

  host_view_type getBasisEvalPoints(const ordinal_type subCellDim, const ordinal_type subCellId) {
    return basisCubPoints[subCellDim][subCellId];
  }


  host_view_type getBasisDerivEvalPoints(const ordinal_type subCellDim, const ordinal_type subCellId) {
    return basisDerivCubPoints[subCellDim][subCellId];
  }


  host_view_type getTargetEvalPoints(const ordinal_type subCellDim, const ordinal_type subCellId) {
    return targetCubPoints[subCellDim][subCellId];
  }


  host_view_type getTargetDerivEvalPoints(const ordinal_type subCellDim, const ordinal_type subCellId) {
    return targetDerivCubPoints[subCellDim][subCellId];
  }


  host_view_type getEvalPoints(const ordinal_type subCellDim, const ordinal_type subCellId, EvalPointsType type) const{
    if(type == BASIS)
      return basisCubPoints[subCellDim][subCellId];
    else
      return targetCubPoints[subCellDim][subCellId];
  }

  view_type getAllEvalPoints(EvalPointsType type = TARGET) const{
    if(type == BASIS)
      return allBasisEPoints;
    else
      return allTargetEPoints;
  }

  host_view_type getDerivEvalPoints(const ordinal_type subCellDim, const ordinal_type subCellId, EvalPointsType type) const{
    if(type == BASIS)
      return basisDerivCubPoints[subCellDim][subCellId];
    else
      return targetDerivCubPoints[subCellDim][subCellId];
  }

  view_type getAllDerivEvalPoints(EvalPointsType type = TARGET) const{
    if(type == BASIS)
      return allBasisDerivEPoints;
    else
      return allTargetDerivEPoints;
  }


  host_view_type getBasisEvalWeights(const ordinal_type subCellDim, const ordinal_type subCellId) {
    return basisCubWeights[subCellDim][subCellId];
  }


  host_view_type getBasisDerivEvalWeights(const ordinal_type subCellDim, const ordinal_type subCellId) {
    return basisDerivCubWeights[subCellDim][subCellId];
  }


  host_view_type getTargetEvalWeights(const ordinal_type subCellDim, const ordinal_type subCellId) {
    return targetCubWeights[subCellDim][subCellId];
  }


  host_view_type getTargetDerivEvalWeights(const ordinal_type subCellDim, const ordinal_type subCellId) {
    return targetDerivCubWeights[subCellDim][subCellId];
  }


  const range_tag getBasisPointsRange() const {
    return basisPointsRange;
  }


  const range_tag getPointsRange(const EvalPointsType type) const {
    if(type == BASIS)
      return basisPointsRange;
    else
      return targetPointsRange;
  }


  const range_tag getBasisDerivPointsRange() const {
    return basisDerivPointsRange;
  }

  const range_tag getDerivPointsRange(const EvalPointsType type) const {
    if(type == BASIS)
      return basisDerivPointsRange;
    else
      return targetDerivPointsRange;
  }


  const range_tag getTargetPointsRange() const {
    return targetPointsRange;
  }


  const range_tag getTargetDerivPointsRange() const {
    return targetDerivPointsRange;
  }

  const key_tag getTopologyKey() const {
    return subCellTopologyKey;
  }

  template<typename BasisPtrType>
  void createL2ProjectionStruct(const BasisPtrType cellBasis,
      const ordinal_type targetCubDegree);


  template<typename BasisPtrType>
  void createL2DGProjectionStruct(const BasisPtrType cellBasis,
      const ordinal_type targetCubDegree) {
      createHVolProjectionStruct(cellBasis, targetCubDegree);
  }


  template<typename BasisPtrType>
  void createHGradProjectionStruct(const BasisPtrType cellBasis,
      const ordinal_type targetCubDegree,
      const ordinal_type targetGradCubDegre);


  template<typename BasisPtrType>
  void createHCurlProjectionStruct(const BasisPtrType cellBasis,
      const ordinal_type targetCubDegree,
      const ordinal_type targetCurlCubDegre);


  template<typename BasisPtrType>
  void createHDivProjectionStruct(const BasisPtrType cellBasis,
      const ordinal_type targetCubDegree,
      const ordinal_type targetDivCubDegre);

  template<typename BasisPtrType>
  void createHVolProjectionStruct(const BasisPtrType cellBasis,
      const ordinal_type targetCubDegree);

  key_tag subCellTopologyKey;
  range_tag basisPointsRange;
  range_tag basisDerivPointsRange;
  range_tag targetPointsRange;
  range_tag targetDerivPointsRange;
  view_tag basisCubPoints;
  view_tag basisCubWeights;
  view_tag basisDerivCubPoints;
  view_tag basisDerivCubWeights;
  view_tag targetCubPoints;
  view_tag targetCubWeights;
  view_tag targetDerivCubPoints;
  view_tag targetDerivCubWeights;
  view_type allBasisEPoints;
  view_type allBasisDerivEPoints;
  view_type allTargetEPoints;
  view_type allTargetDerivEPoints;
  ordinal_type numBasisEvalPoints;
  ordinal_type numBasisDerivEvalPoints;
  ordinal_type numTargetEvalPoints;
  ordinal_type numTargetDerivEvalPoints;
  ordinal_type maxNumBasisEvalPoints;
  ordinal_type maxNumTargetEvalPoints;
  ordinal_type maxNumBasisDerivEvalPoints;
  ordinal_type maxNumTargetDerivEvalPoints;
};

} // Intrepid2 namespace
#include "Intrepid2_ProjectionStructDef.hpp"
#endif