44 #include <Teuchos_TimeMonitor.hpp> 45 #include <PanzerAdaptersSTK_config.hpp> 65 PANZER_FUNC_TIME_MONITOR(
"panzer::SquareTriMeshFactory::buildMesh()");
71 mesh->initialize(parallelMach);
81 PANZER_FUNC_TIME_MONITOR(
"panzer::SquareTriMeshFactory::buildUncomittedMesh()");
85 machRank_ = stk::parallel_machine_rank(parallelMach);
86 machSize_ = stk::parallel_machine_size(parallelMach);
98 const int maxFactor = 100;
101 int factors[maxFactor];
102 for (
int jj = 0; jj < maxFactor; jj++) factors[jj] = 0;
103 for (
int jj = 2; jj < maxFactor; jj++) {
106 int temp = ProcTemp/jj;
107 if (temp*jj == ProcTemp) {
117 for (
int jj = maxFactor-1; jj > 0; jj--) {
118 while (factors[jj] != 0) {
132 "Cannot build SquareTriMeshFactory, the product of \"X Procs\" and \"Y Procs\"" 133 " must equal the number of processors.");
146 PANZER_FUNC_TIME_MONITOR(
"panzer::SquareTriMeshFactory::completeMeshConstruction()");
180 setMyParamList(paramList);
182 x0_ = paramList->get<
double>(
"X0");
183 y0_ = paramList->get<
double>(
"Y0");
185 xf_ = paramList->get<
double>(
"Xf");
186 yf_ = paramList->get<
double>(
"Yf");
188 xBlocks_ = paramList->get<
int>(
"X Blocks");
189 yBlocks_ = paramList->get<
int>(
"Y Blocks");
191 nXElems_ = paramList->get<
int>(
"X Elements");
192 nYElems_ = paramList->get<
int>(
"Y Elements");
194 xProcs_ = paramList->get<
int>(
"X Procs");
195 yProcs_ = paramList->get<
int>(
"Y Procs");
206 static RCP<Teuchos::ParameterList> defaultParams;
209 if(defaultParams == Teuchos::null) {
210 defaultParams = rcp(
new Teuchos::ParameterList);
212 defaultParams->set<
double>(
"X0",0.0);
213 defaultParams->set<
double>(
"Y0",0.0);
215 defaultParams->set<
double>(
"Xf",1.0);
216 defaultParams->set<
double>(
"Yf",1.0);
218 defaultParams->set<
int>(
"X Blocks",1);
219 defaultParams->set<
int>(
"Y Blocks",1);
221 defaultParams->set<
int>(
"X Procs",-1);
222 defaultParams->set<
int>(
"Y Procs",1);
224 defaultParams->set<
int>(
"X Elements",5);
225 defaultParams->set<
int>(
"Y Elements",5);
228 defaultParams->set<
bool>(
"Create Edge Blocks",
false,
"Create edge blocks in the mesh");
230 Teuchos::ParameterList & bcs = defaultParams->sublist(
"Periodic BCs");
231 bcs.set<
int>(
"Count",0);
234 return defaultParams;
240 RCP<Teuchos::ParameterList> validParams = rcp(
new Teuchos::ParameterList(*
getValidParameters()));
248 typedef shards::Triangle<> TriTopo;
249 const CellTopologyData * ctd = shards::getCellTopologyData<TriTopo>();
250 const CellTopologyData * side_ctd = shards::CellTopology(ctd).getBaseCellTopologyData(1,0);
259 std::stringstream ebPostfix;
260 ebPostfix <<
"-" << bx <<
"_" << by;
279 const CellTopologyData * edge_ctd = shards::CellTopology(ctd).getBaseCellTopologyData(1,0);
288 for(
int xBlock=0;xBlock<
xBlocks_;xBlock++) {
289 for(
int yBlock=0;yBlock<
yBlocks_;yBlock++) {
302 int myXElems_start = sizeAndStartX.first;
303 int myXElems_end = myXElems_start+sizeAndStartX.second;
304 int myYElems_start = sizeAndStartY.first;
305 int myYElems_end = myYElems_start+sizeAndStartY.second;
309 double deltaX = (
xf_-
x0_)/
double(totalXElems);
310 double deltaY = (
yf_-
y0_)/
double(totalYElems);
312 std::vector<double> coord(2,0.0);
315 for(
int nx=myXElems_start;nx<myXElems_end+1;++nx) {
317 for(
int ny=myYElems_start;ny<myYElems_end+1;++ny) {
320 mesh.
addNode(ny*(totalXElems+1)+nx+1,coord);
324 std::stringstream blockName;
325 blockName <<
"eblock-" << xBlock <<
"_" << yBlock;
329 for(
int nx=myXElems_start;nx<myXElems_end;++nx) {
330 for(
int ny=myYElems_start;ny<myYElems_end;++ny) {
331 stk::mesh::EntityId gid_a = 2*(totalXElems*ny+nx+1)-1;
332 stk::mesh::EntityId gid_b = gid_a+1;
333 std::vector<stk::mesh::EntityId> nodes(3);
334 stk::mesh::EntityId sw,se,ne,nw;
335 sw = nx+1+ny*(totalXElems+1);
337 ne = se+(totalXElems+1);
356 unsigned int minElements =
nXElems_/size;
357 unsigned int extra =
nXElems_ - minElements*size;
359 TEUCHOS_ASSERT(minElements>0);
365 nume = minElements+1;
366 start = xProcLoc*(minElements+1);
370 start = extra*(minElements+1)+(xProcLoc-extra)*minElements;
373 return std::make_pair(start+
nXElems_*xBlock,nume);
379 unsigned int minElements =
nYElems_/size;
380 unsigned int extra =
nYElems_ - minElements*size;
382 TEUCHOS_ASSERT(minElements>0);
388 nume = minElements+1;
389 start = yProcLoc*(minElements+1);
393 start = extra*(minElements+1)+(yProcLoc-extra)*minElements;
396 return std::make_pair(start+
nYElems_*yBlock,nume);
407 stk::mesh::Part * left = mesh.
getSideset(
"left");
408 stk::mesh::Part * right = mesh.
getSideset(
"right");
409 stk::mesh::Part * top = mesh.
getSideset(
"top");
410 stk::mesh::Part * bottom = mesh.
getSideset(
"bottom");
412 std::vector<stk::mesh::Entity> localElmts;
416 std::vector<stk::mesh::Entity>::const_iterator itr;
417 for(itr=localElmts.begin();itr!=localElmts.end();++itr) {
418 stk::mesh::Entity element = (*itr);
421 bool lower = (gid%2 != 0);
422 std::size_t block = lower ? (gid+1)/2 : gid/2;
424 ny = (block-1) / totalXElems;
425 nx = block-ny*totalXElems-1;
430 if(nx+1==totalXElems && lower) {
438 if(nx==0 && !lower) {
457 if(ny+1==totalYElems && !lower) {
474 stk::mesh::Part * origin = mesh.
getNodeset(
"origin");
476 Teuchos::RCP<stk::mesh::BulkData> bulkData = mesh.
getBulkData();
479 stk::mesh::Entity node = bulkData->get_entity(mesh.
getNodeRank(),1);
494 Teuchos::RCP<stk::mesh::BulkData> bulkData = mesh.
getBulkData();
495 Teuchos::RCP<stk::mesh::MetaData> metaData = mesh.
getMetaData();
497 std::vector<stk::mesh::Entity> edges;
498 bulkData->get_entities(mesh.
getEdgeRank(),metaData->locally_owned_part(),edges);
513 return Teuchos::tuple(i,j);
Teuchos::RCP< stk::mesh::MetaData > getMetaData() const
stk::mesh::Part * getElementBlockPart(const std::string &name) const
get the block part
void addNodeset(const std::string &name)
Teuchos::Tuple< std::size_t, 2 > procTuple_
void addEntityToNodeset(stk::mesh::Entity entity, stk::mesh::Part *nodeset)
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
From ParameterListAcceptor.
void setParameterList(const Teuchos::RCP< Teuchos::ParameterList > ¶mList)
From ParameterListAcceptor.
void addEdgeBlocks(STK_Interface &mesh) const
void getMyElements(std::vector< stk::mesh::Entity > &elements) const
void addEntityToSideset(stk::mesh::Entity entity, stk::mesh::Part *sideset)
stk::mesh::EntityRank getNodeRank() const
void buildMetaData(stk::ParallelMachine parallelMach, STK_Interface &mesh) const
Teuchos::RCP< stk::mesh::BulkData > getBulkData() const
Teuchos::Tuple< std::size_t, 2 > procRankToProcTuple(std::size_t procRank) const
what is the 2D tuple describe this processor distribution
stk::mesh::Entity findConnectivityById(stk::mesh::Entity src, stk::mesh::EntityRank tgt_rank, unsigned rel_id) const
void buildBlock(stk::ParallelMachine machRank, int xBlock, int yBlock, STK_Interface &mesh) const
stk::mesh::Part * getEdgeBlock(const std::string &name) const
get the block part
stk::mesh::EntityRank getEdgeRank() const
stk::mesh::Part * getSideset(const std::string &name) const
void buildElements(stk::ParallelMachine parallelMach, STK_Interface &mesh) const
~SquareTriMeshFactory()
Destructor.
void addElement(const Teuchos::RCP< ElementDescriptor > &ed, stk::mesh::Part *block)
stk::mesh::EntityId elementGlobalId(std::size_t lid) const
SquareTriMeshFactory()
Constructor.
void initialize(stk::ParallelMachine parallelMach, bool setupIO=true, const bool buildRefinementSupport=false)
double getMeshCoord(const int nx, const double deltaX, const double x0) const
bool isInitialized() const
Has initialize been called on this mesh object?
static const std::string edgeBlockString
std::pair< int, int > determineYElemSizeAndStart(int yBlock, unsigned int size, unsigned int rank) const
virtual void completeMeshConstruction(STK_Interface &mesh, stk::ParallelMachine parallelMach) const
void addNode(stk::mesh::EntityId gid, const std::vector< double > &coord)
void addEntitiesToEdgeBlock(std::vector< stk::mesh::Entity > entities, stk::mesh::Part *edgeblock)
void buildSubcells()
force the mesh to build subcells: edges and faces
std::pair< int, int > determineXElemSizeAndStart(int xBlock, unsigned int size, unsigned int rank) const
void addSideSets(STK_Interface &mesh) const
void addSideset(const std::string &name, const CellTopologyData *ctData)
void addEdgeBlock(const std::string &name, const CellTopologyData *ctData)
Teuchos::RCP< STK_Interface > buildMesh(stk::ParallelMachine parallelMach) const
Build the mesh object.
std::vector< Teuchos::RCP< const PeriodicBC_MatcherBase > > periodicBCVec_
void buildLocalElementIDs()
unsigned entityOwnerRank(stk::mesh::Entity entity) const
void addNodeSets(STK_Interface &mesh) const
void rebalance(STK_Interface &mesh) const
void addElementBlock(const std::string &name, const CellTopologyData *ctData)
virtual Teuchos::RCP< STK_Interface > buildUncommitedMesh(stk::ParallelMachine parallelMach) const
stk::mesh::Part * getNodeset(const std::string &name) const
static void parsePeriodicBCList(const Teuchos::RCP< Teuchos::ParameterList > &pl, std::vector< Teuchos::RCP< const PeriodicBC_MatcherBase > > &periodicBC)
void initializeWithDefaults()