Intrepid2
Intrepid2_HGRAD_LINE_Cn_FEM_JACOBI.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 Kyungjoo Kim (kyukim@sandia.gov), or
38// Mauro Perego (mperego@sandia.gov)
39//
40// ************************************************************************
41// @HEADER
42
49#ifndef __INTREPID2_HGRAD_LINE_CN_FEM_JACOBI_HPP__
50#define __INTREPID2_HGRAD_LINE_CN_FEM_JACOBI_HPP__
51
52#include "Intrepid2_Basis.hpp"
53#include "Intrepid2_Polylib.hpp"
54
55namespace Intrepid2 {
56
131 namespace Impl {
132
137 public:
138
142 template<EOperator opType>
143 struct Serial {
144 template<typename OutputViewType,
145 typename inputViewType>
146 KOKKOS_INLINE_FUNCTION
147 static void
148 getValues( OutputViewType output,
149 const inputViewType input,
150 const ordinal_type order,
151 const double alpha,
152 const double beta,
153 const ordinal_type opDn = 0 );
154 };
155
156 template<typename ExecSpaceType, ordinal_type numPtsPerEval,
157 typename outputValueValueType, class ...outputValueProperties,
158 typename inputPointValueType, class ...inputPointProperties>
159 static void
160 getValues( Kokkos::DynRankView<outputValueValueType,outputValueProperties...> outputValues,
161 const Kokkos::DynRankView<inputPointValueType, inputPointProperties...> inputPoints,
162 const ordinal_type order,
163 const double alpha,
164 const double beta,
165 const EOperator operatorType );
166
170 template<typename outputValueViewType,
171 typename inputPointViewType,
172 EOperator opType,
173 ordinal_type numPtsEval>
174 struct Functor {
175 outputValueViewType _outputValues;
176 const inputPointViewType _inputPoints;
177 const ordinal_type _order;
178 const double _alpha, _beta;
179 const ordinal_type _opDn;
180
181 KOKKOS_INLINE_FUNCTION
182 Functor( outputValueViewType outputValues_,
183 inputPointViewType inputPoints_,
184 const ordinal_type order_,
185 const double alpha_,
186 const double beta_,
187 const ordinal_type opDn_ = 0 )
188 : _outputValues(outputValues_), _inputPoints(inputPoints_),
189 _order(order_), _alpha(alpha_), _beta(beta_), _opDn(opDn_) {}
190
191 KOKKOS_INLINE_FUNCTION
192 void operator()(const size_type iter) const {
193 const auto ptBegin = Util<ordinal_type>::min(iter*numPtsEval, _inputPoints.extent(0));
194 const auto ptEnd = Util<ordinal_type>::min(ptBegin+numPtsEval, _inputPoints.extent(0));
195
196 const auto ptRange = Kokkos::pair<ordinal_type,ordinal_type>(ptBegin, ptEnd);
197 const auto input = Kokkos::subview( _inputPoints, ptRange, Kokkos::ALL() );
198
199 if (input.extent(0)) {
200 switch (opType) {
201 case OPERATOR_VALUE : {
202 auto output = Kokkos::subview( _outputValues, Kokkos::ALL(), ptRange );
203 Serial<opType>::getValues( output, input, _order, _alpha, _beta );
204 break;
205 }
206 case OPERATOR_GRAD : {
207 auto output = Kokkos::subview( _outputValues, Kokkos::ALL(), ptRange, Kokkos::ALL() );
208 Serial<opType>::getValues( output, input, _order, _alpha, _beta );
209 break;
210 }
211 case OPERATOR_Dn: {
212 auto output = Kokkos::subview( _outputValues, Kokkos::ALL(), ptRange, Kokkos::ALL() );
213 Serial<opType>::getValues( output, input, _order, _alpha, _beta, _opDn );
214 break;
215 }
216 default: {
217 INTREPID2_TEST_FOR_ABORT( true,
218 ">>> ERROR: (Intrepid2::Basis_HGRAD_LINE_Cn_FEM_JACOBI::Functor) operator is not supported");
219
220 }
221 }
222 }
223 }
224 };
225 };
226 }
227
228 template<typename DeviceType = void,
229 typename outputValueType = double,
230 typename pointValueType = double>
232 : public Basis<DeviceType,outputValueType,pointValueType> {
233 public:
234 typedef double value_type;
238
241 Basis_HGRAD_LINE_Cn_FEM_JACOBI( const ordinal_type order,
242 const double alpha = 0,
243 const double beta = 0 );
244
248
249 using Basis<DeviceType,outputValueType,pointValueType>::getValues;
250
251 virtual
252 void
253 getValues( OutputViewType outputValues,
254 const PointViewType inputPoints,
255 const EOperator operatorType = OPERATOR_VALUE ) const override {
256#ifdef HAVE_INTREPID2_DEBUG
258 inputPoints,
259 operatorType,
260 this->getBaseCellTopology(),
261 this->getCardinality() );
262#endif
263 constexpr ordinal_type numPtsPerEval = 1;
264 Impl::Basis_HGRAD_LINE_Cn_FEM_JACOBI::
265 getValues<DeviceType,numPtsPerEval>( outputValues,
266 inputPoints,
267 this->getDegree(),
268 this->alpha_,
269 this->beta_,
270 operatorType );
271 }
272
273 private:
274 double alpha_, beta_;
275
276 };
277
278}
279
281
282#endif
Header file for the abstract base class Intrepid2::Basis.
void getValues_HGRAD_Args(const outputValueViewType outputValues, const inputPointViewType inputPoints, const EOperator operatorType, const shards::CellTopology cellTopo, const ordinal_type basisCard)
Runtime check of the arguments for the getValues method in an HGRAD-conforming FEM basis....
Definition file for FEM orthogonal basis functions of degree n for H(grad) functions on LINE.
Header file for Intrepid2::Polylib class providing orthogonal polynomial calculus and interpolation.
Implementation of the locally H(grad)-compatible FEM basis of variable order on the [-1,...
virtual void getValues(OutputViewType outputValues, const PointViewType inputPoints, const EOperator operatorType=OPERATOR_VALUE) const override
Evaluation of a FEM basis on a reference cell.
An abstract base class that defines interface for concrete basis implementations for Finite Element (...
Kokkos::DynRankView< PointValueType, Kokkos::LayoutStride, DeviceType > PointViewType
View type for input points.
Kokkos::DynRankView< OutputValueType, Kokkos::LayoutStride, DeviceType > OutputViewType
View type for basis value output.
Kokkos::View< ordinal_type ***, typename ExecutionSpace::array_layout, Kokkos::HostSpace > OrdinalTypeArray3DHost
View type for 3d host array.
ordinal_type getDegree() const
Returns the degree of the basis.
ordinal_type getCardinality() const
Returns cardinality of the basis.
Kokkos::DynRankView< scalarType, Kokkos::LayoutStride, DeviceType > ScalarViewType
View type for scalars.
Device DeviceType
(Kokkos) Device type on which Basis is templated. Does not necessarily return true for Kokkos::is_dev...
Kokkos::View< ordinal_type **, typename ExecutionSpace::array_layout, Kokkos::HostSpace > OrdinalTypeArray2DHost
View type for 2d host array.
shards::CellTopology getBaseCellTopology() const
Returns the base cell topology for which the basis is defined. See Shards documentation https://trili...
Kokkos::View< ordinal_type *, typename ExecutionSpace::array_layout, Kokkos::HostSpace > OrdinalTypeArray1DHost
View type for 1d host array.
small utility functions