Intrepid2_OrientationOperator.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_ORIENTATION_OPERATOR_HPP__
#define __INTREPID2_ORIENTATION_OPERATOR_HPP__
 
#include "Intrepid2_ConfigDefs.hpp"
#include "Intrepid2_Types.hpp"
#include "Intrepid2_Utils.hpp"
 
#include "Shards_CellTopology.hpp"
 
namespace Intrepid2 {
 
  template<class DeviceType>
  class OrientationOperator {
  public:
    template<class ValueType>
    struct RawView1D {
      ValueType* ptr;
      size_type extent0;
 
      KOKKOS_INLINE_FUNCTION
      RawView1D()
      : ptr(nullptr), extent0(0)
      {}
 
      KOKKOS_INLINE_FUNCTION
      RawView1D(ValueType* ptr_, const size_type extent0_)
      : ptr(ptr_), extent0(extent0_)
      {}
 
      KOKKOS_INLINE_FUNCTION
      ValueType& operator()(const ordinal_type i) const {
        return ptr[i];
      }
 
      KOKKOS_INLINE_FUNCTION
      ValueType& operator[](const ordinal_type i) const {
        return ptr[i];
      }
 
      KOKKOS_INLINE_FUNCTION
      ordinal_type extent_int(const ordinal_type i) const {
        return (i == 0) ? static_cast<ordinal_type>(extent0) : 1;
      }
 
      KOKKOS_INLINE_FUNCTION
      size_type extent(const ordinal_type i) const {
        return (i == 0) ? extent0 : 1;
      }
 
      KOKKOS_INLINE_FUNCTION
      ValueType* data() const {
        return ptr;
      }
    };
 
    using UnmanagedOrdinalView = RawView1D<ordinal_type>;
    using UnmanagedDoubleView  = RawView1D<double>;
    
    // only stores deviations from the identity
    UnmanagedOrdinalView rowIndices;           // index in basis (the field ordinal)
    UnmanagedOrdinalView offsetsForRowOrdinal; // argument is same as _rowIndices; offset gives index for _packedColumnIndices and _packedWeights
    UnmanagedOrdinalView packedColumnIndices;  // ordinal is the index from offsets
    UnmanagedDoubleView packedWeights;         // ordinal is the index from offsets
    
    // offsetsForRowOrdinal may be empty; if it is, isPermutation is set to true; otherwise it is false
    // if isWeightedPermutation is true, the offsets are all unit-spaced (0,1,…), and packedColumnIndices and packedWeights have the same length as rowIndices
    bool isWeightedPermutation;
  public:
    KOKKOS_INLINE_FUNCTION
    OrientationOperator(UnmanagedOrdinalView rowIndices_,
                        UnmanagedOrdinalView offsetsForRowOrdinal_,
                        UnmanagedOrdinalView packedColumnIndices_,
                        UnmanagedDoubleView       packedWeights_);
    
    KOKKOS_INLINE_FUNCTION
    OrientationOperator(UnmanagedOrdinalView rowIndices_,
                        UnmanagedOrdinalView packedColumnIndices_,
                        UnmanagedDoubleView       packedWeights_);
    
    KOKKOS_INLINE_FUNCTION
    OrientationOperator();
  };
}
 
// include templated function definitions
#include "Intrepid2_OrientationOperatorDef.hpp"
 
#endif