Intrepid2_ProjectionStruct.hpp

Go to the documentation of this file.

// @HEADER
// ************************************************************************
//
//                           Intrepid2 Package
//                 Copyright (2007) Sandia Corporation
//
// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
// license for use of this work by or on behalf of the U.S. Government.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
// 1. Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
//
// 3. Neither the name of the Corporation nor the names of the
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// Questions? Contact Mauro Perego  (mperego@sandia.gov), or
//                    Nate Roberts  (nvrober@sandia.gov)
//
//
// ************************************************************************
// @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