43 #include "PanzerAdaptersSTK_config.hpp"
45 #ifdef PANZER_HAVE_EPETRA_STACK
50 #include "Kokkos_DynRankView.hpp"
52 #include <stk_mesh/base/FieldBase.hpp>
54 namespace panzer_stk {
57 const Epetra_Vector & x,
const std::vector<std::size_t> & localCellIds,
58 std::map<std::string,Kokkos::DynRankView<double,PHX::Device> > & fc);
61 std::map<std::string,
Teuchos::RCP<std::vector<std::size_t> > > & localIds);
65 std::vector<std::string> blocks;
69 for(std::size_t eb=0;eb<blocks.size();eb++) {
70 const std::string & blockId = blocks[eb];
73 std::vector<stk::mesh::Entity> elements;
77 for(std::size_t el=0;el<elements.size();el++) {
79 double * solnData = stk::mesh::field_data(*field,elements[el]);
81 solnData[0] = data[localId];
88 write_solution_data(dofMngr,mesh,*x(0),prefix,postfix);
93 typedef Kokkos::DynRankView<double,PHX::Device> FieldContainer;
96 std::map<std::string,Teuchos::RCP<std::vector<std::size_t> > > localIds;
97 build_local_ids(mesh,localIds);
100 for(
const auto & itr : localIds) {
101 const auto blockId = itr.first;
102 const auto & localCellIds = *(itr.second);
104 std::map<std::string,FieldContainer> data;
107 gather_in_block(blockId,dofMngr,x,localCellIds,data);
110 for(
const auto & dataItr : data)
116 const Epetra_Vector & x,
const std::vector<std::size_t> & localCellIds,
117 std::map<std::string,Kokkos::DynRankView<double,PHX::Device> > & fc)
121 for(std::size_t fieldIndex=0;fieldIndex<fieldNums.size();fieldIndex++) {
122 int fieldNum = fieldNums[fieldIndex];
127 fc[fieldStr] = Kokkos::DynRankView<double,PHX::Device>(
"fc",localCellIds.size(),elmtOffset.size());
128 auto field = Kokkos::create_mirror_view(fc[fieldStr]);
132 for(std::size_t worksetCellIndex=0;worksetCellIndex<localCellIds.size();++worksetCellIndex) {
133 std::vector<panzer::GlobalOrdinal> GIDs;
134 std::vector<int> LIDs;
135 std::size_t cellLocalId = localCellIds[worksetCellIndex];
140 LIDs.resize(GIDs.size());
141 for(std::size_t i=0;i<GIDs.size();i++)
142 LIDs[i] = x.Map().LID(GIDs[i]);
145 for(std::size_t basis=0;basis<elmtOffset.size();basis++) {
146 int offset = elmtOffset[basis];
147 int lid = LIDs[offset];
148 field(worksetCellIndex,basis) = x[lid];
151 Kokkos::deep_copy(fc[fieldStr], field);
156 std::map<std::string,
Teuchos::RCP<std::vector<std::size_t> > > & localIds)
159 std::vector<std::string> blockIds;
162 std::vector<std::string>::const_iterator idItr;
163 for(idItr=blockIds.begin();idItr!=blockIds.end();++idItr) {
164 std::string blockId = *idItr;
166 localIds[blockId] =
Teuchos::rcp(
new std::vector<std::size_t>);
167 std::vector<std::size_t> & localBlockIds = *localIds[blockId];
170 std::vector<stk::mesh::Entity> blockElmts;
173 std::vector<stk::mesh::Entity>::const_iterator itr;
174 for(itr=blockElmts.begin();itr!=blockElmts.end();++itr)
177 std::sort(localBlockIds.begin(),localBlockIds.end());
183 #endif // PANZER_HAVE_EPETRA_STACK
virtual const std::string & getFieldString(int num) const =0
Reverse lookup of the field string from a field number.
void setSolutionFieldData(const std::string &fieldName, const std::string &blockId, const std::vector< std::size_t > &localElementIds, const ArrayT &solutionValues, double scaleValue=1.0)
void getElementBlockNames(std::vector< std::string > &names) const
stk::mesh::Field< double > SolutionFieldType
std::size_t elementLocalId(stk::mesh::Entity elmt) const
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
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.
#define TEUCHOS_ASSERT(assertion_test)
void getMyElements(std::vector< stk::mesh::Entity > &elements) const
stk::mesh::Field< double > * getCellField(const std::string &fieldName, const std::string &blockId) const