Intrepid2
Intrepid2_DataTools.hpp
Go to the documentation of this file.
1// @HEADER
2// ************************************************************************
3//
4// Intrepid2 Package
5// Copyright (2007) Sandia Corporation
6//
7// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
8// license for use of this work by or on behalf of the U.S. Government.
9//
10// Redistribution and use in source and binary forms, with or without
11// modification, are permitted provided that the following conditions are
12// met:
13//
14// 1. Redistributions of source code must retain the above copyright
15// notice, this list of conditions and the following disclaimer.
16//
17// 2. Redistributions in binary form must reproduce the above copyright
18// notice, this list of conditions and the following disclaimer in the
19// documentation and/or other materials provided with the distribution.
20//
21// 3. Neither the name of the Corporation nor the names of the
22// contributors may be used to endorse or promote products derived from
23// this software without specific prior written permission.
24//
25// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36//
37// Questions? Contact Mauro Perego (mperego@sandia.gov) or
38// Nate Roberts (nvrober@sandia.gov)
39//
40// ************************************************************************
41// @HEADER
42
43#ifndef Intrepid2_DataTools_h
44#define Intrepid2_DataTools_h
45
46#include "Intrepid2_Data.hpp"
47
52
53namespace Intrepid2 {
54
56{
57public:
63 template<class Scalar, class DeviceType>
64 static void multiplyByCPWeights(Data<Scalar,DeviceType> &resultMatrixData, const Data<Scalar,DeviceType> &matrixDataIn, const Data<Scalar,DeviceType> &scalarDataIn)
65 {
66 const ordinal_type rank = scalarDataIn.rank();
67 auto extents = scalarDataIn.getExtents();
68 auto variationTypes = scalarDataIn.getVariationTypes();
69 extents[rank] = matrixDataIn.extent_int(rank);
70 extents[rank+1] = matrixDataIn.extent_int(rank+1);
71 variationTypes[rank] = CONSTANT;
72 variationTypes[rank+1] = CONSTANT;
73
74 auto scalarDataInExtended = scalarDataIn.shallowCopy(rank + 2, extents, variationTypes);
75 resultMatrixData.storeInPlaceProduct(matrixDataIn,scalarDataInExtended);
76 }
77
83 template<class Scalar, class DeviceType>
85 {
86 const ordinal_type rank = scalarDataIn.rank();
87 auto extents = scalarDataIn.getExtents();
88 auto variationTypes = scalarDataIn.getVariationTypes();
89 extents[rank] = matrixDataIn.extent_int(rank);
90 extents[rank+1] = matrixDataIn.extent_int(rank+1);
91 variationTypes[rank] = CONSTANT;
92 variationTypes[rank+1] = CONSTANT;
93
94 auto scalarDataInExtended = scalarDataIn.shallowCopy(rank + 2, extents, variationTypes);
95
96 auto result = Data<Scalar,DeviceType>::allocateInPlaceCombinationResult(scalarDataInExtended, matrixDataIn);
97
98 result.storeInPlaceProduct(matrixDataIn,scalarDataInExtended);
99 return result;
100 }
101};
102}
103
104#endif /* Intrepid2_DataTools_h */
Defines the Data class, a wrapper around a Kokkos::View that allows data that is constant or repeatin...
static Data< Scalar, DeviceType > multiplyByCPWeights(const Data< Scalar, DeviceType > &matrixDataIn, const Data< Scalar, DeviceType > &scalarDataIn)
static void multiplyByCPWeights(Data< Scalar, DeviceType > &resultMatrixData, const Data< Scalar, DeviceType > &matrixDataIn, const Data< Scalar, DeviceType > &scalarDataIn)
Wrapper around a Kokkos::View that allows data that is constant or repeating in various logical dimen...
void storeInPlaceProduct(const Data< DataScalar, DeviceType > &A, const Data< DataScalar, DeviceType > &B)
stores the in-place (entrywise) product, A .* B, into this container.
KOKKOS_INLINE_FUNCTION int extent_int(const int &r) const
Returns the logical extent in the specified dimension.
static Data< DataScalar, DeviceType > allocateInPlaceCombinationResult(const Data< DataScalar, DeviceType > &A, const Data< DataScalar, DeviceType > &B)
KOKKOS_INLINE_FUNCTION const Kokkos::Array< DataVariationType, 7 > & getVariationTypes() const
Returns an array with the variation types in each logical dimension.
KOKKOS_INLINE_FUNCTION Kokkos::Array< int, 7 > getExtents() const
Returns an array containing the logical extents in each dimension.
Data shallowCopy(const int rank, const Kokkos::Array< int, 7 > &extents, const Kokkos::Array< DataVariationType, 7 > &variationTypes) const
Creates a new Data object with the same underlying view, but with the specified logical rank,...
KOKKOS_INLINE_FUNCTION unsigned rank() const
Returns the logical rank of the Data container.