43 #include "PanzerAdaptersSTK_config.hpp" 48 #include "Kokkos_DynRankView.hpp" 50 #include <stk_mesh/base/FieldBase.hpp> 55 const Epetra_Vector & x,
const std::vector<std::size_t> & localCellIds,
56 std::map<std::string,Kokkos::DynRankView<double,PHX::Device> > & fc);
59 std::map<std::string,Teuchos::RCP<std::vector<std::size_t> > > & localIds);
63 std::vector<std::string> blocks;
67 for(std::size_t eb=0;eb<blocks.size();eb++) {
68 const std::string & blockId = blocks[eb];
71 std::vector<stk::mesh::Entity> elements;
75 for(std::size_t el=0;el<elements.size();el++) {
77 double * solnData = stk::mesh::field_data(*
field,elements[el]);
78 TEUCHOS_ASSERT(solnData!=0);
79 solnData[0] = data[localId];
91 typedef Kokkos::DynRankView<double,PHX::Device> FieldContainer;
94 std::map<std::string,Teuchos::RCP<std::vector<std::size_t> > > localIds;
98 for(
const auto & itr : localIds) {
99 const auto blockId = itr.first;
100 const auto & localCellIds = *(itr.second);
102 std::map<std::string,FieldContainer> data;
108 for(
const auto & dataItr : data)
114 const Epetra_Vector & x,
const std::vector<std::size_t> & localCellIds,
115 std::map<std::string,Kokkos::DynRankView<double,PHX::Device> > & fc)
119 for(std::size_t fieldIndex=0;fieldIndex<fieldNums.size();fieldIndex++) {
120 int fieldNum = fieldNums[fieldIndex];
125 fc[fieldStr] = Kokkos::DynRankView<double,PHX::Device>(
"fc",localCellIds.size(),elmtOffset.size());
126 auto field = Kokkos::create_mirror_view(fc[fieldStr]);
130 for(std::size_t worksetCellIndex=0;worksetCellIndex<localCellIds.size();++worksetCellIndex) {
131 std::vector<panzer::GlobalOrdinal> GIDs;
132 std::vector<int> LIDs;
133 std::size_t cellLocalId = localCellIds[worksetCellIndex];
138 LIDs.resize(GIDs.size());
139 for(std::size_t i=0;i<GIDs.size();i++)
140 LIDs[i] = x.
Map().
LID(GIDs[i]);
143 for(std::size_t basis=0;basis<elmtOffset.size();basis++) {
144 int offset = elmtOffset[basis];
145 int lid = LIDs[offset];
146 field(worksetCellIndex,basis) = x[lid];
149 Kokkos::deep_copy(fc[fieldStr],
field);
154 std::map<std::string,Teuchos::RCP<std::vector<std::size_t> > > & localIds)
157 std::vector<std::string> blockIds;
160 std::vector<std::string>::const_iterator idItr;
161 for(idItr=blockIds.begin();idItr!=blockIds.end();++idItr) {
162 std::string blockId = *idItr;
164 localIds[blockId] = Teuchos::rcp(
new std::vector<std::size_t>);
165 std::vector<std::size_t> & localBlockIds = *localIds[blockId];
168 std::vector<stk::mesh::Entity> blockElmts;
171 std::vector<stk::mesh::Entity>::const_iterator itr;
172 for(itr=blockElmts.begin();itr!=blockElmts.end();++itr)
175 std::sort(localBlockIds.begin(),localBlockIds.end());
virtual const std::string & getFieldString(int num) const =0
Reverse lookup of the field string from a field number.
const Epetra_BlockMap & Map() const
void setSolutionFieldData(const std::string &fieldName, const std::string &blockId, const std::vector< std::size_t > &localElementIds, const ArrayT &solutionValues, double scaleValue=1.0)
std::size_t elementLocalId(stk::mesh::Entity elmt) const
stk::mesh::Field< double > SolutionFieldType
void getMyElements(std::vector< stk::mesh::Entity > &elements) const
void getElementBlockNames(std::vector< std::string > &names) const
static void gather_in_block(const std::string &blockId, const panzer::GlobalIndexer &dofMngr, const Epetra_Vector &x, const std::vector< std::size_t > &localCellIds, std::map< std::string, Kokkos::DynRankView< double, PHX::Device > > &fc)
stk::mesh::Field< double > * getCellField(const std::string &fieldName, const std::string &blockId) const
void write_solution_data(const panzer::GlobalIndexer &dofMngr, panzer_stk::STK_Interface &mesh, const Epetra_MultiVector &x, const std::string &prefix, const std::string &postfix)
static void build_local_ids(const panzer_stk::STK_Interface &mesh, std::map< std::string, Teuchos::RCP< std::vector< std::size_t > > > &localIds)
void write_cell_data(panzer_stk::STK_Interface &mesh, const std::vector< double > &data, const std::string &fieldName)
PHX::MDField< ScalarT, panzer::Cell, panzer::BASIS > field
A field to which we'll contribute, or in which we'll store, the result of computing this integral...
virtual const std::vector< int > & getGIDFieldOffsets(const std::string &blockId, int fieldNum) const =0
Use the field pattern so that you can find a particular field in the GIDs array.
virtual const std::vector< int > & getBlockFieldNumbers(const std::string &blockId) const =0
virtual void getElementGIDs(panzer::LocalOrdinal localElmtId, std::vector< panzer::GlobalOrdinal > &gids, const std::string &blockIdHint="") const =0
Get the global IDs for a particular element. This function overwrites the gids variable.