Intrepid2_SerendipityBasisFamily.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 Kyungjoo Kim  (kyukim@sandia.gov),
//                    Mauro Perego  (mperego@sandia.gov), or
//                    Nate Roberts  (nvrober@sandia.gov)
//
// ************************************************************************
// @HEADER

#ifndef Intrepid2_SerendipityBasisFamily_h
#define Intrepid2_SerendipityBasisFamily_h

#include "Intrepid2_Basis.hpp"

#include "Intrepid2_DerivedBasis_HGRAD_QUAD.hpp"
#include "Intrepid2_DerivedBasis_HCURL_QUAD.hpp"
#include "Intrepid2_DerivedBasis_HDIV_QUAD.hpp"
#include "Intrepid2_DerivedBasis_HVOL_QUAD.hpp"

#include "Intrepid2_DerivedBasis_HGRAD_HEX.hpp"
#include "Intrepid2_DerivedBasis_HCURL_HEX.hpp"
#include "Intrepid2_DerivedBasis_HDIV_HEX.hpp"
#include "Intrepid2_DerivedBasis_HVOL_HEX.hpp"

namespace Intrepid2 {
  template<class FullBasis, int numPolyOrderArgs>
  class SerendipityBasisWrapper
  :
  public SerendipityBasis<typename FullBasis::BasisBase>
  {
  public:
    using BasisBase = typename FullBasis::BasisBase;
    using BasisPtr  = Teuchos::RCP<BasisBase>;
    using DeviceType = typename BasisBase::DeviceType;
    using ExecutionSpace  = typename BasisBase::ExecutionSpace;
    using OutputValueType = typename BasisBase::OutputValueType;
    using PointValueType  = typename BasisBase::PointValueType;
    
    using OrdinalTypeArray1D     = typename BasisBase::OrdinalTypeArray1D;
    using OrdinalTypeArray1DHost = typename BasisBase::OrdinalTypeArray1DHost;
    using OrdinalTypeArray2DHost = typename BasisBase::OrdinalTypeArray2DHost;
    using OutputViewType         = typename BasisBase::OutputViewType;
    using PointViewType          = typename BasisBase::PointViewType;
    
    SerendipityBasisWrapper(int polyOrder, const EPointType pointType=POINTTYPE_DEFAULT)
    :
    SerendipityBasis<typename FullBasis::BasisBase>(Teuchos::rcp(new FullBasis(polyOrder)) )
    {}
    
    template<bool M=(numPolyOrderArgs==2)>
    SerendipityBasisWrapper(int polyOrder_x, int polyOrder_y, const EPointType pointType=POINTTYPE_DEFAULT, typename std::enable_if<M>::type* = 0)
    :
    SerendipityBasis<typename FullBasis::BasisBase>(Teuchos::rcp(new FullBasis(polyOrder_x, polyOrder_y)) )
    {}
    
    template<bool M=(numPolyOrderArgs==3)>
    SerendipityBasisWrapper(int polyOrder_x, int polyOrder_y, int polyOrder_z, const EPointType pointType=POINTTYPE_DEFAULT, typename std::enable_if<M>::type* = 0)
    :
    SerendipityBasis<typename FullBasis::BasisBase>(Teuchos::rcp(new FullBasis(polyOrder_x, polyOrder_y, polyOrder_z)) )
    {}
  };

  template<class LineBasisHGRAD, class LineBasisHVOL, class TriangleBasisFamily, class TetrahedronBasisFamily >
  class DerivedSerendipityBasisFamily
  {
  public:
    using ExecutionSpace  = typename LineBasisHGRAD::ExecutionSpace;
    using OutputValueType = typename LineBasisHGRAD::OutputValueType;
    using PointValueType  = typename LineBasisHGRAD::PointValueType;
    
    using Basis      = typename LineBasisHGRAD::BasisBase;
    using BasisPtr   = Teuchos::RCP<Basis>;
    using DeviceType = typename Basis::DeviceType;
    
    // line bases
    using HGRAD_LINE = LineBasisHGRAD;
    using HVOL_LINE  = LineBasisHVOL;
    
    // quadrilateral bases
    using HGRAD_QUAD = SerendipityBasisWrapper<Basis_Derived_HGRAD_QUAD<HGRAD_LINE>,            2>;
    using HCURL_QUAD = SerendipityBasisWrapper<Basis_Derived_HCURL_QUAD<HGRAD_LINE, HVOL_LINE>, 2>;
    using HDIV_QUAD  = SerendipityBasisWrapper<Basis_Derived_HDIV_QUAD <HGRAD_LINE, HVOL_LINE>, 2>;
    using HVOL_QUAD  = SerendipityBasisWrapper<Basis_Derived_HVOL_QUAD <HVOL_LINE>,             2>;
    
    // hexahedron bases
    using HGRAD_HEX = SerendipityBasisWrapper<Basis_Derived_HGRAD_HEX<HGRAD_LINE>,            3>;
    using HCURL_HEX = SerendipityBasisWrapper<Basis_Derived_HCURL_HEX<HGRAD_LINE, HVOL_LINE>, 3>;
    using HDIV_HEX  = SerendipityBasisWrapper<Basis_Derived_HDIV_HEX <HGRAD_LINE, HVOL_LINE>, 3>;
    using HVOL_HEX  = SerendipityBasisWrapper<Basis_Derived_HVOL_HEX <HVOL_LINE>,             3>;
    
    // triangle bases
    using HGRAD_TRI = typename TriangleBasisFamily::HGRAD;
    using HCURL_TRI = typename TriangleBasisFamily::HCURL;
    using HDIV_TRI  = typename TriangleBasisFamily::HDIV;
    using HVOL_TRI  = typename TriangleBasisFamily::HVOL;
    
    // tetrahedron bases
    using HGRAD_TET = typename TetrahedronBasisFamily::HGRAD;
    using HCURL_TET = typename TetrahedronBasisFamily::HCURL;
    using HDIV_TET  = typename TetrahedronBasisFamily::HDIV;
    using HVOL_TET  = typename TetrahedronBasisFamily::HVOL;
  };

  template<typename DeviceType,
           typename OutputScalar = double,
           typename PointScalar  = double>
  using SerendipityBasisFamily = DerivedSerendipityBasisFamily< IntegratedLegendreBasis_HGRAD_LINE<DeviceType,OutputScalar,PointScalar,true>,
                                                                LegendreBasis_HVOL_LINE<DeviceType,OutputScalar,PointScalar>,
                                                                HierarchicalTriangleBasisFamily<DeviceType,OutputScalar,PointScalar,true>,
                                                                HierarchicalTetrahedronBasisFamily<DeviceType,OutputScalar,PointScalar,true>
                                                              >;

  template<typename DeviceType,
           typename OutputScalar = double,
           typename PointScalar  = double>
  using DGSerendipityBasisFamily = DerivedSerendipityBasisFamily< IntegratedLegendreBasis_HGRAD_LINE<DeviceType,OutputScalar,PointScalar,false>,
                                                                  LegendreBasis_HVOL_LINE<DeviceType,OutputScalar,PointScalar>,
                                                                  HierarchicalTriangleBasisFamily<DeviceType,OutputScalar,PointScalar,false>,
                                                                  HierarchicalTetrahedronBasisFamily<DeviceType,OutputScalar,PointScalar,false>
                                                                >;
}

#endif /* Intrepid2_HierarchicalBasisFamily_h */