44#include <Teuchos_TimeMonitor.hpp>
45#include <PanzerAdaptersSTK_config.hpp>
46#include "Teuchos_StandardParameterEntryValidators.hpp"
68 PANZER_FUNC_TIME_MONITOR(
"panzer::SquareQuadMeshFactory::buildMesh()");
74 mesh->initialize(parallelMach);
84 PANZER_FUNC_TIME_MONITOR(
"panzer::SquareQuadMeshFactory::buildUncomittedMesh()");
88 machRank_ = stk::parallel_machine_rank(parallelMach);
89 machSize_ = stk::parallel_machine_size(parallelMach);
101 const int maxFactor = 100;
104 int factors[maxFactor];
105 for (
int jj = 0; jj < maxFactor; jj++) factors[jj] = 0;
106 for (
int jj = 2; jj < maxFactor; jj++) {
109 int temp = ProcTemp/jj;
110 if (temp*jj == ProcTemp) {
120 for (
int jj = maxFactor-1; jj > 0; jj--) {
121 while (factors[jj] != 0) {
136 <<
"' must equal the number of processors = " <<
machSize_
137 <<
"\n\n\t==> Run the simulation with an appropriate number of processors, i.e. #procs = " <<
xProcs_*
yProcs_ <<
".\n");
151 PANZER_FUNC_TIME_MONITOR(
"panzer::SquareQuadMeshFactory::completeMeshConstruction()");
160#ifndef ENABLE_UNIFORM
169#ifndef ENABLE_UNIFORM
189 setMyParamList(paramList);
191 x0_ = paramList->get<
double>(
"X0");
192 y0_ = paramList->get<
double>(
"Y0");
194 xf_ = paramList->get<
double>(
"Xf");
195 yf_ = paramList->get<
double>(
"Yf");
197 xBlocks_ = paramList->get<
int>(
"X Blocks");
198 yBlocks_ = paramList->get<
int>(
"Y Blocks");
200 nXElems_ = paramList->get<
int>(
"X Elements");
201 nYElems_ = paramList->get<
int>(
"Y Elements");
203 xProcs_ = paramList->get<
int>(
"X Procs");
204 yProcs_ = paramList->get<
int>(
"Y Procs");
206 offsetGIDs_ = (paramList->get<std::string>(
"Offset mesh GIDs above 32-bit int limit") ==
"ON") ? true :
false;
217 static RCP<Teuchos::ParameterList> defaultParams;
220 if(defaultParams == Teuchos::null) {
221 defaultParams = rcp(
new Teuchos::ParameterList);
223 defaultParams->set<
double>(
"X0",0.0);
224 defaultParams->set<
double>(
"Y0",0.0);
226 defaultParams->set<
double>(
"Xf",1.0);
227 defaultParams->set<
double>(
"Yf",1.0);
229 defaultParams->set<
int>(
"X Blocks",1);
230 defaultParams->set<
int>(
"Y Blocks",1);
232 defaultParams->set<
int>(
"X Procs",-1);
233 defaultParams->set<
int>(
"Y Procs",1);
235 defaultParams->set<
int>(
"X Elements",5);
236 defaultParams->set<
int>(
"Y Elements",5);
239 defaultParams->set<
bool>(
"Create Edge Blocks",
false,
"Create edge blocks in the mesh");
241 Teuchos::setStringToIntegralParameter<int>(
242 "Offset mesh GIDs above 32-bit int limit",
244 "If 64-bit GIDs are supported, the mesh element and node global indices will start at a value greater than 32-bit limit.",
245 Teuchos::tuple<std::string>(
"OFF",
"ON"),
246 defaultParams.get());
248 Teuchos::ParameterList & bcs = defaultParams->sublist(
"Periodic BCs");
249 bcs.set<
int>(
"Count",0);
252 return defaultParams;
258 RCP<Teuchos::ParameterList> validParams = rcp(
new Teuchos::ParameterList(*
getValidParameters()));
272 typedef shards::Quadrilateral<4> QuadTopo;
273 const CellTopologyData * ctd = shards::getCellTopologyData<QuadTopo>();
274 const CellTopologyData * side_ctd = shards::CellTopology(ctd).getBaseCellTopologyData(1,0);
275 const CellTopologyData * edge_ctd = shards::CellTopology(ctd).getBaseCellTopologyData(1,0);
284 std::stringstream ebPostfix;
285 ebPostfix <<
"-" << bx <<
"_" << by;
300#ifndef ENABLE_UNIFORM
307 std::stringstream ss;
308 ss <<
"vertical_" << bx-1;
312 std::stringstream ss;
313 ss <<
"horizontal_" << by-1;
327 for(
int xBlock=0;xBlock<
xBlocks_;xBlock++) {
328 for(
int yBlock=0;yBlock<
yBlocks_;yBlock++) {
341 int myXElems_start = sizeAndStartX.first;
342 int myXElems_end = myXElems_start+sizeAndStartX.second;
343 int myYElems_start = sizeAndStartY.first;
344 int myYElems_end = myYElems_start+sizeAndStartY.second;
348 double deltaX = (
xf_-
x0_)/
double(totalXElems);
349 double deltaY = (
yf_-
y0_)/
double(totalYElems);
351 std::vector<double> coord(2,0.0);
355 if (std::numeric_limits<panzer::GlobalOrdinal>::max() > std::numeric_limits<unsigned int>::max())
356 offset_ = panzer::GlobalOrdinal(std::numeric_limits<unsigned int>::max()) + 1;
360 for(
int nx=myXElems_start;nx<myXElems_end+1;++nx) {
362 for(
int ny=myYElems_start;ny<myYElems_end+1;++ny) {
369 std::stringstream blockName;
370 blockName <<
"eblock-" << xBlock <<
"_" << yBlock;
374 for(
int nx=myXElems_start;nx<myXElems_end;++nx) {
375 for(
int ny=myYElems_start;ny<myYElems_end;++ny) {
376 stk::mesh::EntityId gid = totalXElems*ny+nx+1 +
offset_;
377 std::vector<stk::mesh::EntityId> nodes(4);
378 nodes[0] = nx+1+ny*(totalXElems+1);
379 nodes[1] = nodes[0]+1;
380 nodes[2] = nodes[1]+(totalXElems+1);
381 nodes[3] = nodes[2]-1;
383 for (
int i=0; i < 4; ++i)
395 unsigned int minElements =
nXElems_/size;
396 unsigned int extra =
nXElems_ - minElements*size;
398 TEUCHOS_ASSERT(minElements>0);
404 nume = minElements+1;
405 start = xProcLoc*(minElements+1);
409 start = extra*(minElements+1)+(xProcLoc-extra)*minElements;
412 return std::make_pair(start+
nXElems_*xBlock,nume);
418 unsigned int minElements =
nYElems_/size;
419 unsigned int extra =
nYElems_ - minElements*size;
421 TEUCHOS_ASSERT(minElements>0);
427 nume = minElements+1;
428 start = yProcLoc*(minElements+1);
432 start = extra*(minElements+1)+(yProcLoc-extra)*minElements;
435 return std::make_pair(start+
nYElems_*yBlock,nume);
446 stk::mesh::Part * left = mesh.
getSideset(
"left");
447 stk::mesh::Part * right = mesh.
getSideset(
"right");
448 stk::mesh::Part * top = mesh.
getSideset(
"top");
449 stk::mesh::Part * bottom = mesh.
getSideset(
"bottom");
451 std::vector<stk::mesh::Part*> vertical;
452 std::vector<stk::mesh::Part*> horizontal;
454 std::stringstream ss;
455 ss <<
"vertical_" << bx-1;
456 vertical.push_back(mesh.
getSideset(ss.str()));
459 std::stringstream ss;
460 ss <<
"horizontal_" << by-1;
461 horizontal.push_back(mesh.
getSideset(ss.str()));
464 std::vector<stk::mesh::Entity> localElmts;
468 std::vector<stk::mesh::Entity>::const_iterator itr;
469 for(itr=localElmts.begin();itr!=localElmts.end();++itr) {
470 stk::mesh::Entity element = (*itr);
477 ny = (gid-1) / totalXElems;
478 nx = gid-ny*totalXElems-1;
483 if(nx+1==totalXElems) {
499 if(nx+1!=totalXElems && ((nx+1) %
nXElems_==0)) {
530 if(ny+1==totalYElems) {
548 if(ny+1!=totalYElems && ((ny+1) %
nYElems_==0)) {
567 stk::mesh::Part * lower_left = mesh.
getNodeset(
"lower_left");
568 stk::mesh::Part * origin = mesh.
getNodeset(
"origin");
573 Teuchos::RCP<stk::mesh::BulkData> bulkData = mesh.
getBulkData();
591 Teuchos::RCP<stk::mesh::BulkData> bulkData = mesh.
getBulkData();
592 Teuchos::RCP<stk::mesh::MetaData> metaData = mesh.
getMetaData();
596 stk::mesh::Selector owned_block = metaData->locally_owned_part();
598 std::vector<stk::mesh::Entity> edges;
599 bulkData->get_entities(mesh.
getEdgeRank(), owned_block, edges);
614 return Teuchos::tuple(i,j);
void buildLocalElementIDs()
stk::mesh::Entity findConnectivityById(stk::mesh::Entity src, stk::mesh::EntityRank tgt_rank, unsigned rel_id) const
void initialize(stk::ParallelMachine parallelMach, bool setupIO=true, const bool buildRefinementSupport=false)
stk::mesh::Part * getElementBlockPart(const std::string &name) const
get the block part
static const std::string edgeBlockString
stk::mesh::EntityId elementGlobalId(std::size_t lid) const
stk::mesh::Part * getEdgeBlock(const std::string &name) const
get the block part
bool isInitialized() const
Has initialize been called on this mesh object?
stk::mesh::EntityRank getNodeRank() const
void addEdgeBlock(const std::string &elemBlockName, const std::string &edgeBlockName, const stk::topology &topology)
void buildSubcells()
force the mesh to build subcells: edges and faces
void addEntityToNodeset(stk::mesh::Entity entity, stk::mesh::Part *nodeset)
void addElement(const Teuchos::RCP< ElementDescriptor > &ed, stk::mesh::Part *block)
void addNode(stk::mesh::EntityId gid, const std::vector< double > &coord)
void addNodeset(const std::string &name)
void addSideset(const std::string &name, const CellTopologyData *ctData)
Teuchos::RCP< stk::mesh::MetaData > getMetaData() const
stk::mesh::Part * getNodeset(const std::string &name) const
unsigned entityOwnerRank(stk::mesh::Entity entity) const
void addEntitiesToEdgeBlock(std::vector< stk::mesh::Entity > entities, stk::mesh::Part *edgeblock)
void addEntityToSideset(stk::mesh::Entity entity, stk::mesh::Part *sideset)
void getMyElements(std::vector< stk::mesh::Entity > &elements) const
void addElementBlock(const std::string &name, const CellTopologyData *ctData)
Teuchos::RCP< stk::mesh::BulkData > getBulkData() const
stk::mesh::EntityRank getEdgeRank() const
stk::mesh::Part * getSideset(const std::string &name) const
void rebalance(STK_Interface &mesh) const
static void parsePeriodicBCList(const Teuchos::RCP< Teuchos::ParameterList > &pl, std::vector< Teuchos::RCP< const PeriodicBC_MatcherBase > > &periodicBC, bool &useBBoxSearch)
double getMeshCoord(const int nx, const double deltaX, const double x0) const
std::vector< Teuchos::RCP< const PeriodicBC_MatcherBase > > periodicBCVec_
SquareQuadMeshFactory(bool enableRebalance=false)
Constructor.
void initializeWithDefaults()
stk::mesh::EntityId offset_
Teuchos::RCP< STK_Interface > buildMesh(stk::ParallelMachine parallelMach) const
Build the mesh object.
~SquareQuadMeshFactory()
Destructor.
virtual Teuchos::RCP< STK_Interface > buildUncommitedMesh(stk::ParallelMachine parallelMach) const
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
From ParameterListAcceptor.
Teuchos::Tuple< std::size_t, 2 > procTuple_
void buildMetaData(stk::ParallelMachine parallelMach, STK_Interface &mesh) const
void addSideSets(STK_Interface &mesh) const
void buildElements(stk::ParallelMachine parallelMach, STK_Interface &mesh) const
bool offsetGIDs_
If true, offset mesh GIDs to exercise 32-bit limits.
void addNodeSets(STK_Interface &mesh) const
std::pair< int, int > determineYElemSizeAndStart(int yBlock, unsigned int size, unsigned int rank) const
std::string edgeBlockName_
virtual void completeMeshConstruction(STK_Interface &mesh, stk::ParallelMachine parallelMach) const
void setParameterList(const Teuchos::RCP< Teuchos::ParameterList > ¶mList)
From ParameterListAcceptor.
Teuchos::Tuple< std::size_t, 2 > procRankToProcTuple(std::size_t procRank) const
what is the 2D tuple describe this processor distribution
void addEdgeBlocks(STK_Interface &mesh) const
std::pair< int, int > determineXElemSizeAndStart(int xBlock, unsigned int size, unsigned int rank) const
void buildBlock(stk::ParallelMachine machRank, int xBlock, int yBlock, STK_Interface &mesh) const