52 #ifndef Intrepid2_BasisValues_h
53 #define Intrepid2_BasisValues_h
68 template<
class Scalar,
typename ExecSpaceType>
74 Kokkos::Array<TensorDataType,Parameters::MaxTensorComponents> tensorDataFamilies_;
77 int numTensorDataFamilies_ = -1;
79 Kokkos::View<ordinal_type*,ExecSpaceType> ordinalFilter_;
84 tensorDataFamilies_({tensorData}),
85 numTensorDataFamilies_(1)
91 numTensorDataFamilies_(tensorDataFamilies.size())
93 for (
int family=0; family<numTensorDataFamilies_; family++)
95 tensorDataFamilies_[family] = tensorDataFamilies[family];
102 vectorData_(vectorData)
108 numTensorDataFamilies_(0)
113 template<typename OtherExecSpaceType, class = typename std::enable_if<!std::is_same<ExecSpaceType, OtherExecSpaceType>::value>::type>
117 numTensorDataFamilies_(basisValues.numTensorDataFamilies())
120 for (
int family=0; family<numTensorDataFamilies_; family++)
124 auto otherOrdinalFilter = basisValues.ordinalFilter();
125 ordinalFilter_ = Kokkos::View<ordinal_type*,ExecSpaceType>(
"BasisValues::ordinalFilter_",otherOrdinalFilter.extent(0));
127 Kokkos::deep_copy(ordinalFilter_, otherOrdinalFilter);
133 int familyStartOrdinal = -1, familyEndOrdinal = -1;
136 for (
int i=0; i<familyCount; i++)
138 const bool startMatches = (fieldsSoFar == fieldStartOrdinal);
139 familyStartOrdinal = startMatches ? i : familyStartOrdinal;
140 fieldsSoFar += numFieldsInFamily(i);
141 const bool endMatches = (fieldsSoFar - fieldStartOrdinal == numFields);
142 familyEndOrdinal = endMatches ? i : familyEndOrdinal;
144 INTREPID2_TEST_FOR_EXCEPTION(familyStartOrdinal == -1, std::invalid_argument,
"fieldStartOrdinal does not align with the start of a family.");
145 INTREPID2_TEST_FOR_EXCEPTION(familyEndOrdinal == -1, std::invalid_argument,
"fieldStartOrdinal + numFields does not align with the end of a family.");
147 const int numFamiliesInFieldSpan = familyEndOrdinal - familyStartOrdinal + 1;
148 if (numTensorDataFamilies_ > 0)
151 for (
int i=familyStartOrdinal; i<=familyEndOrdinal; i++)
153 tensorDataFamilies[i-familyStartOrdinal] = tensorDataFamilies_[i];
160 std::vector< std::vector<TensorData<Scalar,ExecSpaceType> > > vectorComponents(numFamiliesInFieldSpan, std::vector<
TensorData<Scalar,ExecSpaceType> >(componentCount));
161 for (
int i=familyStartOrdinal; i<=familyEndOrdinal; i++)
163 for (
int j=0; j<componentCount; j++)
165 vectorComponents[i-familyStartOrdinal][j] = vectorData_.
getComponent(i,j);
173 KOKKOS_INLINE_FUNCTION
183 for (
int i=0; i<familyOrdinal; i++)
185 offset += tensorDataFamilies_[i].extent_int(0);
195 return tensorDataFamilies_[0];
203 return tensorDataFamilies_[familyOrdinal];
207 KOKKOS_INLINE_FUNCTION
216 return numTensorDataFamilies_;
220 KOKKOS_INLINE_FUNCTION
221 int numTensorDataFamilies()
const
223 return numTensorDataFamilies_;
226 KOKKOS_INLINE_FUNCTION
227 int numFieldsInFamily(
int familyOrdinal)
const
235 return tensorDataFamilies_[familyOrdinal].extent_int(0);
242 return tensorDataFamilies_;
252 KOKKOS_INLINE_FUNCTION
253 Scalar
operator()(
const int &fieldOrdinal,
const int &pointOrdinal)
const
255 const int &tensorFieldOrdinal = (ordinalFilter_.extent(0) > 0) ? ordinalFilter_(fieldOrdinal) : fieldOrdinal;
256 if (numTensorDataFamilies_ == 1)
258 #ifdef HAVE_INTREPID2_DEBUG
261 return tensorDataFamilies_[0](tensorFieldOrdinal, pointOrdinal);
265 int familyForField = -1;
266 int previousFamilyEnd = -1;
267 int fieldAdjustment = 0;
269 for (
int family=0; family<numTensorDataFamilies_; family++)
271 const int familyFieldCount = tensorDataFamilies_[family].extent_int(0);
272 const bool fieldInRange = (tensorFieldOrdinal > previousFamilyEnd) && (tensorFieldOrdinal <= previousFamilyEnd + familyFieldCount);
273 familyForField = fieldInRange ? family : familyForField;
274 fieldAdjustment = fieldInRange ? previousFamilyEnd + 1 : fieldAdjustment;
275 previousFamilyEnd += familyFieldCount;
277 #ifdef HAVE_INTREPID2_DEBUG
280 return tensorDataFamilies_[familyForField](tensorFieldOrdinal-fieldAdjustment,pointOrdinal);
285 KOKKOS_INLINE_FUNCTION
286 Scalar
operator()(
const int &fieldOrdinal,
const int &pointOrdinal,
const int &dim)
const
288 #ifdef HAVE_INTREPID2_DEBUG
291 const int &tensorFieldOrdinal = (ordinalFilter_.extent(0) > 0) ? ordinalFilter_(fieldOrdinal) : fieldOrdinal;
292 return vectorData_(tensorFieldOrdinal, pointOrdinal, dim);
296 KOKKOS_INLINE_FUNCTION
297 Scalar
operator()(
const int &cellOrdinal,
const int &fieldOrdinal,
const int &pointOrdinal,
const int &dim)
const
303 KOKKOS_INLINE_FUNCTION
304 int extent_int(
const int &i)
const
309 if (ordinalFilter_.extent_int(0) == 0)
312 for (
int familyOrdinal=0; familyOrdinal<
numFamilies(); familyOrdinal++)
314 numFields += numFieldsInFamily(familyOrdinal);
320 return ordinalFilter_.extent_int(0);
329 else if (tensorDataFamilies_[0].isValid())
331 return tensorDataFamilies_[0].extent_int(i);
341 KOKKOS_INLINE_FUNCTION
342 size_t extent(
const int &i)
const
344 return static_cast<size_t>(extent_int(i));
347 KOKKOS_INLINE_FUNCTION
352 return vectorData_.
rank();
354 else if (tensorDataFamilies_[0].isValid())
356 return tensorDataFamilies_[0].rank();
364 void setOrdinalFilter(Kokkos::View<ordinal_type*,ExecSpaceType> ordinalFilter)
366 ordinalFilter_ = ordinalFilter;
369 Kokkos::View<ordinal_type*,ExecSpaceType> ordinalFilter()
const
371 return ordinalFilter_;
const Kokkos::Array< TensorDataType, Parameters::MaxTensorComponents > & tensorDataFamilies() const
TensorDataFamilies accessor.
KOKKOS_INLINE_FUNCTION int numFamilies() const
For valid vectorData, returns the number of families in vectorData; otherwise, returns number of Tens...
KOKKOS_INLINE_FUNCTION int numFieldsInFamily(const unsigned &familyOrdinal) const
returns the number of fields in the specified family
KOKKOS_INLINE_FUNCTION Scalar operator()(const int &cellOrdinal, const int &fieldOrdinal, const int &pointOrdinal, const int &dim) const
operator() for (C,F,P,D) data, which arises in CVFEM; at present unimplemented, and only declared her...
BasisValues(const BasisValues< Scalar, OtherExecSpaceType > &basisValues)
copy-like constructor for differing execution spaces. This does a deep copy of underlying views...
#define INTREPID2_TEST_FOR_EXCEPTION_DEVICE_SAFE(test, x, msg)
KOKKOS_INLINE_FUNCTION Scalar operator()(const int &fieldOrdinal, const int &pointOrdinal, const int &dim) const
operator() for (F,P,D) vector data; throws an exception if this is not a vector-valued container ...
KOKKOS_INLINE_FUNCTION int numFamilies() const
returns the number of families
const VectorDataType & vectorData() const
VectorData accessor.
BasisValues< Scalar, ExecSpaceType > basisValuesForFields(const int &fieldStartOrdinal, const int &numFields)
field start and length must align with families in vectorData_ or tensorDataFamilies_ (whichever is v...
The data containers in Intrepid2 that support sum factorization and other reduced-data optimizations ...
BasisValues(VectorDataType vectorData)
Constructor for vector-valued BasisValues.
BasisValues()
Default constructor.
Reference-space field values for a basis, designed to support typical vector-valued bases...
KOKKOS_INLINE_FUNCTION const TensorData< Scalar, DeviceType > & getComponent(const int &componentOrdinal) const
Single-argument component accessor for the axial-component or the single-family case; in this case...
BasisValues(std::vector< TensorDataType > tensorDataFamilies)
Constructor for scalar-valued BasisValues, with potentially multiple families of values. (Used, e.g., for op = DIV and functionSpace = HDIV.)
KOKKOS_INLINE_FUNCTION int extent_int(const int &r) const
Returns the extent in the specified dimension as an int.
const TensorDataType & tensorData(const int &familyOrdinal) const
TensorData accessor for multi-family scalar data.
TensorDataType & tensorData()
TensorData accessor for single-family scalar data.
KOKKOS_INLINE_FUNCTION constexpr bool isValid() const
returns true for containers that have data; false for those that don't (e.g., those that have been co...
KOKKOS_INLINE_FUNCTION int familyFieldOrdinalOffset(const int &familyOrdinal) const
Returns the field ordinal offset for the specified family.
BasisValues(TensorDataType tensorData)
Constructor for scalar-valued BasisValues with a single family of values.
KOKKOS_INLINE_FUNCTION unsigned rank() const
Returns the rank of this container, which is 3.
View-like interface to tensor data; tensor components are stored separately and multiplied together a...
KOKKOS_INLINE_FUNCTION int familyFieldOrdinalOffset(const int &familyOrdinal) const
Returns the field ordinal offset for the specified family.
View-like interface to tensor data; tensor components are stored separately and multiplied together a...
KOKKOS_INLINE_FUNCTION int numComponents() const
returns the number of components
KOKKOS_INLINE_FUNCTION Scalar operator()(const int &fieldOrdinal, const int &pointOrdinal) const
operator() for (F,P) scalar data; throws an exception if this is not a scalar-valued container ...