Panzer_IntrepidFieldPattern.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//           Panzer: A partial differential equation assembly
//       engine for strongly coupled complex multiphysics systems
//
// Copyright 2011 NTESS and the Panzer contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef __Panzer_IntrepidFieldPattern_hpp__
#define __Panzer_IntrepidFieldPattern_hpp__

#include "Panzer_FieldPattern.hpp"

// Trilinos includes
#include "Kokkos_Core.hpp"
#include "Kokkos_DynRankView.hpp"
#include "Intrepid2_Basis.hpp"

#include "Phalanx_KokkosDeviceTypes.hpp"
#include "Teuchos_RCP.hpp"
#include <set>

namespace panzer {

  class Intrepid2FieldPattern : public FieldPattern {
  public:
    Intrepid2FieldPattern(const Teuchos::RCP< Intrepid2::Basis<PHX::Device,double,double> > &intrepidBasis);
    Teuchos::RCP<panzer::FieldPattern> clone() const
    {return Teuchos::rcp(new Intrepid2FieldPattern(*this));}

    virtual int getSubcellCount(int dim) const;
    virtual const std::vector<int> & getSubcellIndices(int dim, int cellIndex) const;
    virtual int getDimension() const;
    virtual shards::CellTopology getCellTopology() const;

    virtual void getSubcellClosureIndices(int dim,int cellIndex,std::vector<int> & indices) const;

    // static functions for examining shards objects

    static void buildSubcellClosure(const shards::CellTopology & cellTopo,unsigned dim,unsigned subCell,
                                    std::set<std::pair<unsigned,unsigned> > & closure);

    static void findContainedSubcells(const shards::CellTopology & cellTopo,unsigned dim,
                                      const std::vector<unsigned> & nodes,
                                      std::set<std::pair<unsigned,unsigned> > & subCells);

    static void getSubcellNodes(const shards::CellTopology & cellTopo,unsigned dim,unsigned subCell,
                                std::vector<unsigned> & nodes);

    bool supportsInterpolatoryCoordinates() const;

    void getInterpolatoryCoordinates(Kokkos::DynRankView<double,PHX::Device> & coords) const;

    void getInterpolatoryCoordinates(const Kokkos::DynRankView<double,PHX::Device> & cellNodes,
                                     Kokkos::DynRankView<double,PHX::Device> & coords,
                                     Teuchos::RCP<const shards::CellTopology> meshCellTopology=Teuchos::null) const;

    Teuchos::RCP< Intrepid2::Basis<PHX::Device,double,double> > getIntrepidBasis() const;

  protected:
    Teuchos::RCP< Intrepid2::Basis<PHX::Device,double,double> > intrepidBasis_;

    //mutable std::vector<int> subcellIndices_;
    mutable std::vector<std::vector<std::vector<int> > > subcellIndicies_;
    std::vector<int> empty_;
  };

}

#endif