Panzer_STK_SetupUtilities.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_STK_SetupUtilities_hpp__
#define __Panzer_STK_SetupUtilities_hpp__

#include "Panzer_STK_Interface.hpp"


#include "Panzer_Workset.hpp"
#include "Panzer_WorksetNeeds.hpp"

#include "Teuchos_RCP.hpp"

#include <vector>
#include <map>
#include <string>

namespace panzer_stk { 

Teuchos::RCP<std::vector<panzer::Workset> >  
buildWorksets(const panzer_stk::STK_Interface & mesh,
              const std::string & eBlock,
              const panzer::WorksetNeeds & needs);

Teuchos::RCP<std::vector<panzer::Workset> >  
buildWorksets(const panzer_stk::STK_Interface & mesh,
              const panzer::WorksetNeeds & needs,
              const std::string & sideset,
              const std::string & eBlock,
              bool useCascade=false);

Teuchos::RCP<std::map<unsigned,panzer::Workset> >
buildBCWorksets(const panzer_stk::STK_Interface & mesh,
                const panzer::WorksetNeeds & needs_a,
                const std::string & blockid_a,
                const panzer::WorksetNeeds & needs_b,
                const std::string & blockid_b,
                const std::string & sideset);

Teuchos::RCP<std::map<unsigned,panzer::Workset> >
buildBCWorksets(const panzer_stk::STK_Interface & mesh,
                const panzer::WorksetNeeds & needs_a,const std::string & eblock_a,
                const panzer::WorksetNeeds & needs_b,const std::string & eblock_b,
                const std::string & sideset);

Teuchos::RCP<std::map<unsigned,panzer::Workset> >
buildBCWorksets(const panzer_stk::STK_Interface & mesh,
                const panzer::WorksetNeeds & needs,
                const std::string & eblockID,
                const std::string & sidesetID);

// namespace may not be neccssary in the future, currently avoids
// collisions with previously implemented code in tests
namespace workset_utils { 


template<typename ArrayT>
void getIdsAndVertices(const panzer_stk::STK_Interface& mesh,
           std::string blockId,
           std::vector<std::size_t>& localIds,
           ArrayT& vertices);


template<typename ArrayT>
void getIdsAndNodes(const panzer_stk::STK_Interface& mesh,
           std::string blockId,
           std::vector<std::size_t>& localIds,
           ArrayT& nodes);

void getSubcellElements(const panzer_stk::STK_Interface & mesh,
            const std::string & blockId, 
            const std::vector<stk::mesh::Entity> & entities,
            std::vector<std::size_t> & localEntityIds, 
            std::vector<stk::mesh::Entity> & elements);

void getUniversalSubcellElements(const panzer_stk::STK_Interface & mesh,
         const std::string & blockId, 
         const std::vector<stk::mesh::Entity> & entities,
         std::vector<std::size_t> & localEntityIds, 
         std::vector<stk::mesh::Entity> & elements,
         std::vector<std::size_t> & missingElementIndices);

void getSideElements(const panzer_stk::STK_Interface & mesh,
           const std::string & blockId, 
           const std::vector<stk::mesh::Entity> & sides,
           std::vector<std::size_t> & localSideIds, 
           std::vector<stk::mesh::Entity> & elements);

void getSideElements(const panzer_stk::STK_Interface & mesh,
                     const std::string & blockId_a, 
                     const std::string & blockId_b, 
                     const std::vector<stk::mesh::Entity> & sides,
                     std::vector<std::size_t> & localSideIds_a, 
                     std::vector<stk::mesh::Entity> & elements_a,
                     std::vector<std::size_t> & localSideIds_b, 
                     std::vector<stk::mesh::Entity> & elements_b);

void getNodeElements(const panzer_stk::STK_Interface & mesh,
           const std::string & blockId, 
           const std::vector<stk::mesh::Entity> & nodes,
           std::vector<std::size_t> & localNodeIds, 
           std::vector<stk::mesh::Entity> & elements);

void getSideElementCascade(const panzer_stk::STK_Interface & mesh,
                           const std::string & blockId, 
                           const std::vector<stk::mesh::Entity> & sides,
                           std::vector<std::size_t> & localSubcellDim, 
                           std::vector<std::size_t> & subcellIds, 
                           std::vector<stk::mesh::Entity> & elements);

void getSubcellEntities(const panzer_stk::STK_Interface & mesh,
            const std::vector<stk::mesh::Entity> & entities,
            std::vector<std::vector<stk::mesh::Entity> > & subcells);

}
}

#include "Panzer_STK_SetupUtilities_impl.hpp"

#endif