43 #include "PanzerAdaptersSTK_config.hpp" 44 #ifdef PANZER_HAVE_TEKO 53 ParameterListCallbackBlocked::ParameterListCallbackBlocked(
54 const Teuchos::RCP<const panzer_stk::STKConnManager> & connManager,
55 const Teuchos::RCP<const panzer::BlockedDOFManager> & blocked_ugi,
56 const Teuchos::RCP<const panzer::BlockedDOFManager> & aux_blocked_ugi)
57 : connManager_(connManager), blocked_ugi_(blocked_ugi), aux_blocked_ugi_(aux_blocked_ugi)
60 Teuchos::RCP<Teuchos::ParameterList>
61 ParameterListCallbackBlocked::request(
const Teko::RequestMesg & rm)
63 TEUCHOS_ASSERT(handlesRequest(rm));
66 Teuchos::RCP<Teuchos::ParameterList> outputPL = rcp(
new Teuchos::ParameterList);
67 Teuchos::RCP<const Teuchos::ParameterList> inputPL = rm.getParameterList();
68 Teuchos::ParameterList::ConstIterator itr;
69 for(itr=inputPL->begin();itr!=inputPL->end();++itr) {
70 std::string * str_ptr = 0;
71 std::string
field = inputPL->entry(itr).getValue(str_ptr);
72 setFieldByKey(itr->first,
field,*outputPL);
78 bool ParameterListCallbackBlocked::handlesRequest(
const Teko::RequestMesg & rm)
82 if(rm.getName()==
"Parameter List") {
83 bool isHandled =
true;
84 Teuchos::RCP<const Teuchos::ParameterList> pl = rm.getParameterList();
86 if(pl->isType<std::string>(
"x-coordinates")) {
87 field = pl->get<std::string>(
"x-coordinates");
92 if(pl->isType<std::string>(
"y-coordinates")) {
94 if(
field != pl->get<std::string>(
"y-coordinates")) {
98 if(pl->isType<std::string>(
"z-coordinates")) {
100 if(
field != pl->get<std::string>(
"z-coordinates")) {
104 if(pl->isType<std::string>(
"Coordinates")){
105 field = pl->get<std::string>(
"Coordinates");
107 if(pl->isType<std::string>(
"Coordinates-Epetra")){
108 field = pl->get<std::string>(
"Coordinates-Epetra");
116 void ParameterListCallbackBlocked::preRequest(
const Teko::RequestMesg & rm)
118 TEUCHOS_ASSERT(handlesRequest(rm));
120 const std::string&
field(getHandledField(*rm.getParameterList()));
125 std::vector<Teuchos::RCP<panzer::GlobalIndexer>>
126 fieldDOFMngrs = blocked_ugi_->getFieldDOFManagers();
127 for (
int b(0); b < static_cast<int>(fieldDOFMngrs.size()); ++b)
129 for (
int f(0); f < fieldDOFMngrs[b]->getNumFields(); ++f)
131 if (fieldDOFMngrs[b]->getFieldString(f) ==
field)
139 aux_blocked_ugi_->getFieldBlock(aux_blocked_ugi_->getFieldNum(
field));
141 block = blocked_ugi_->getFieldBlock(blocked_ugi_->getFieldNum(
field));
144 if (rm.getParameterList()->isType<std::string>(
"Coordinates-Epetra")) {
145 buildArrayToVectorEpetra(block,
field, useAux);
146 buildCoordinatesEpetra(
field, useAux);
148 buildArrayToVectorTpetra(block,
field, useAux);
149 buildCoordinatesTpetra(
field, useAux);
153 void ParameterListCallbackBlocked::setFieldByKey(
const std::string & key,
const std::string &
field,Teuchos::ParameterList & pl)
const 156 if(key==
"x-coordinates") {
157 double * x =
const_cast<double *
>(&getCoordinateByField(0,
field)[0]);
158 pl.set<
double*>(key,x);
160 else if(key==
"y-coordinates") {
161 double * y =
const_cast<double *
>(&getCoordinateByField(1,
field)[0]);
162 pl.set<
double*>(key,y);
164 else if(key==
"z-coordinates") {
165 double * z =
const_cast<double *
>(&getCoordinateByField(2,
field)[0]);
166 pl.set<
double*>(key,z);
167 }
else if(key ==
"Coordinates") {
168 pl.set<Teuchos::RCP<Tpetra::MultiVector<double,int,panzer::GlobalOrdinal,panzer::TpetraNodeType> > >(key,coordsVecTp_);
169 }
else if(key ==
"Coordinates-Epetra") {
170 pl.set<Teuchos::RCP<Epetra_MultiVector> >(
"Coordinates",coordsVecEp_);
174 TEUCHOS_TEST_FOR_EXCEPTION(
true,std::runtime_error,
175 "ParameterListCallback cannot handle key=\"" << key <<
"\"");
178 void ParameterListCallbackBlocked::buildArrayToVectorTpetra(
int block,
const std::string &
field,
const bool useAux)
180 if(arrayToVectorTpetra_[
field]==Teuchos::null) {
181 Teuchos::RCP<const panzer::GlobalIndexer> ugi;
183 ugi = aux_blocked_ugi_->getFieldDOFManagers()[block];
185 ugi = blocked_ugi_->getFieldDOFManagers()[block];
190 void ParameterListCallbackBlocked::buildArrayToVectorEpetra(
int block,
const std::string &
field,
const bool useAux)
192 if(arrayToVectorEpetra_[
field]==Teuchos::null) {
193 Teuchos::RCP<const panzer::GlobalIndexer> ugi;
195 ugi = aux_blocked_ugi_->getFieldDOFManagers()[block];
197 ugi = blocked_ugi_->getFieldDOFManagers()[block];
202 void ParameterListCallbackBlocked::buildCoordinatesTpetra(
const std::string &
field,
const bool useAux)
204 std::map<std::string,Kokkos::DynRankView<double,PHX::Device> > data;
206 Teuchos::RCP<const panzer::Intrepid2FieldPattern> fieldPattern = getFieldPattern(
field,useAux);
208 std::vector<std::string> elementBlocks;
210 aux_blocked_ugi_->getElementBlockIds(elementBlocks);
212 blocked_ugi_->getElementBlockIds(elementBlocks);
213 for(std::size_t i=0;i<elementBlocks.size();++i) {
214 std::string blockId = elementBlocks[i];
215 std::vector<std::size_t> localCellIds;
218 Kokkos::DynRankView<double,PHX::Device> & fieldData = data[blockId];
219 fieldData = Kokkos::DynRankView<double,PHX::Device>(
"fieldData",connManager_->getElementBlock(blockId).size(),fieldPattern->numberIds());
221 if(fieldPattern->supportsInterpolatoryCoordinates()) {
223 connManager_->getDofCoords(blockId,*fieldPattern,localCellIds,fieldData);
226 Teuchos::FancyOStream out(Teuchos::rcpFromRef(std::cout));
227 out.setOutputToRootOnly(-1);
228 out <<
"WARNING: In ParameterListCallback::buildCoordinates(), the Intrepid2::FieldPattern in " 229 <<
"block \"" << blockId <<
"\" does not support interpolatory coordinates. " 230 <<
"This may be fine if coordinates are not actually needed. However if they are then bad things " 231 <<
"will happen. Enjoy!" << std::endl;
237 coordsVecTp_ = arrayToVectorTpetra_[
field]->template getDataVector<double>(
field,data);
239 switch(coordsVecTp_->getNumVectors()) {
241 zcoords_[
field].resize(coordsVecTp_->getLocalLength());
242 coordsVecTp_->getVector(2)->get1dCopy(Teuchos::arrayViewFromVector(zcoords_[
field]));
245 ycoords_[
field].resize(coordsVecTp_->getLocalLength());
246 coordsVecTp_->getVector(1)->get1dCopy(Teuchos::arrayViewFromVector(ycoords_[
field]));
249 xcoords_[
field].resize(coordsVecTp_->getLocalLength());
250 coordsVecTp_->getVector(0)->get1dCopy(Teuchos::arrayViewFromVector(xcoords_[
field]));
253 TEUCHOS_TEST_FOR_EXCEPTION(
true,std::logic_error,
254 "ParameterListCallback::buildCoordinates: Constructed multivector has nonphysical dimensions.");
259 void ParameterListCallbackBlocked::buildCoordinatesEpetra(
const std::string &
field,
const bool useAux)
261 std::map<std::string,Kokkos::DynRankView<double,PHX::Device> > data;
263 Teuchos::RCP<const panzer::Intrepid2FieldPattern> fieldPattern = getFieldPattern(
field,useAux);
265 std::vector<std::string> elementBlocks;
267 aux_blocked_ugi_->getElementBlockIds(elementBlocks);
269 blocked_ugi_->getElementBlockIds(elementBlocks);
270 for(std::size_t i=0;i<elementBlocks.size();++i) {
271 std::string blockId = elementBlocks[i];
272 std::vector<std::size_t> localCellIds;
275 Kokkos::DynRankView<double,PHX::Device> & fieldData = data[blockId];
276 fieldData = Kokkos::DynRankView<double,PHX::Device>(
"fieldData",connManager_->getElementBlock(blockId).size(),fieldPattern->numberIds());
278 if(fieldPattern->supportsInterpolatoryCoordinates()) {
280 connManager_->getDofCoords(blockId,*fieldPattern,localCellIds,fieldData);
283 Teuchos::FancyOStream out(Teuchos::rcpFromRef(std::cout));
284 out.setOutputToRootOnly(-1);
285 out <<
"WARNING: In ParameterListCallback::buildCoordinates(), the Intrepid2::FieldPattern in " 286 <<
"block \"" << blockId <<
"\" does not support interpolatory coordinates. " 287 <<
"This may be fine if coordinates are not actually needed. However if they are then bad things " 288 <<
"will happen. Enjoy!" << std::endl;
294 coordsVecEp_ = arrayToVectorEpetra_[
field]->template getDataVector<double>(
field,data);
297 std::string ParameterListCallbackBlocked::
298 getHandledField(
const Teuchos::ParameterList & pl)
const 301 if(pl.isType<std::string>(
"x-coordinates"))
302 return pl.get<std::string>(
"x-coordinates");
303 else if(pl.isType<std::string>(
"Coordinates"))
304 return pl.get<std::string>(
"Coordinates");
305 else if(pl.isType<std::string>(
"Coordinates-Epetra"))
306 return pl.get<std::string>(
"Coordinates-Epetra");
308 TEUCHOS_TEST_FOR_EXCEPTION(
true,std::logic_error,
"Neither x-coordinates nor Coordinates or Coordinates-Epetra field provided.");
312 const std::vector<double> & ParameterListCallbackBlocked::
313 getCoordinateByField(
int dim,
const std::string &
field)
const 315 TEUCHOS_ASSERT(dim>=0);
316 TEUCHOS_ASSERT(dim<=2);
319 const std::map<std::string,std::vector<double> > * coord =
nullptr;
320 if(dim==0) coord = &xcoords_;
321 else if(dim==1) coord = &ycoords_;
322 else if(dim==2) coord = &zcoords_;
324 TEUCHOS_TEST_FOR_EXCEPTION(
true,std::runtime_error,
325 "ParameterListCallbackBlocked::getCoordinateByField: dim is not in range of [0,2] for field \"" 329 std::map<std::string,std::vector<double> >::const_iterator itr;
330 itr = coord->find(
field);
332 TEUCHOS_TEST_FOR_EXCEPTION(itr==coord->end(),std::runtime_error,
333 "ParameterListCallbackBlocked::getCoordinateByField: Coordinates for field \"" +
field +
334 "\" dimension " << dim <<
" have not been built!");
339 Teuchos::RCP<const panzer::Intrepid2FieldPattern> ParameterListCallbackBlocked
340 ::getFieldPattern(
const std::string & fieldName,
const bool useAux)
const 342 std::vector<std::string> elementBlocks;
344 aux_blocked_ugi_->getElementBlockIds(elementBlocks);
346 blocked_ugi_->getElementBlockIds(elementBlocks);
348 for(std::size_t e=0;e<elementBlocks.size();e++) {
349 std::string blockId = elementBlocks[e];
351 if(blocked_ugi_->fieldInBlock(fieldName,blockId))
352 return Teuchos::rcp_dynamic_cast<const panzer::Intrepid2FieldPattern>(blocked_ugi_->getFieldPattern(blockId,fieldName),
true);
354 if(aux_blocked_ugi_->fieldInBlock(fieldName,blockId))
355 return Teuchos::rcp_dynamic_cast<const panzer::Intrepid2FieldPattern>(aux_blocked_ugi_->getFieldPattern(blockId,fieldName),
true);
358 return Teuchos::null;
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...