Homec4science
Diffusion akantu (features/mesh-periodic-info-in-python) Browse

rAKA/src/mesh_utils/mesh_utils_distribution.cc
ca5d409e4ee8features/mesh-periodic…

Last Change
Raw File

mesh_utils_distribution.cc
View Options

/**
* @file mesh_utils_distribution.cc
*
* @author Nicolas Richart <nicolas.richart@epfl.ch>
*
* @date Sun Oct 2 19:23:15 2016
*
* @brief Implementation of the methods of mesh utils distribute
*
* @section LICENSE
*
* Copyright (©) 2010-2011 EPFL (Ecole Polytechnique Fédérale de Lausanne)
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
*
* Akantu is free software: you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License as published by the Free
* Software Foundation, either version 3 of the License, or (at your option) any
* later version.
*
* Akantu is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
* A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Akantu. If not, see <http://www.gnu.org/licenses/>.
*
*/
/* -------------------------------------------------------------------------- */
#include "mesh_utils_distribution.hh"
#include "element_info_per_processor.hh"
#include "element_synchronizer.hh"
#include "mesh.hh"
#include "mesh_accessor.hh"
#include "mesh_partition.hh"
#include "mesh_utils.hh"
#include "node_info_per_processor.hh"
#include "node_synchronizer.hh"
/* -------------------------------------------------------------------------- */
namespace akantu {
namespace MeshUtilsDistribution {
/* ------------------------------------------------------------------------ */
void distributeMeshCentralized(Mesh & mesh, UInt,
const MeshPartition & partition) {
MeshAccessor mesh_accessor(mesh);
ElementSynchronizer & element_synchronizer =
mesh_accessor.getElementSynchronizer();
NodeSynchronizer & node_synchronizer = mesh_accessor.getNodeSynchronizer();
const StaticCommunicator & comm = element_synchronizer.getCommunicator();
UInt nb_proc = comm.getNbProc();
UInt my_rank = comm.whoAmI();
mesh_accessor.setNbGlobalNodes(mesh.getNbNodes());
mesh_accessor.getNodesGlobalIds().resize(0);
if (nb_proc == 1) {
UInt nb_nodes = mesh.getNbGlobalNodes();
mesh_accessor.getNodesGlobalIds().resize(nb_nodes);
auto gids = mesh_accessor.getNodesGlobalIds();
for (UInt n = 0; n < nb_nodes; ++n) {
gids(n) = n;
}
return;
}
mesh.synchronizeGroupNames();
AKANTU_DEBUG_ASSERT(
partition.getNbPartition() == nb_proc,
"The number of partition does not match the number of processors: "
<< partition.getNbPartition() << " != " << nb_proc);
/**
* connectivity and communications scheme construction
*/
Mesh::type_iterator it =
mesh.firstType(_all_dimensions, _not_ghost, _ek_not_defined);
Mesh::type_iterator end =
mesh.lastType(_all_dimensions, _not_ghost, _ek_not_defined);
UInt count = 0;
/* --- MAIN LOOP ON TYPES --- */
for (; it != end; ++it) {
ElementType type = *it;
/// \todo change this ugly way to avoid a problem if an element
/// type is present in the mesh but not in the partitions
try {
partition.getPartition(type, _not_ghost);
} catch (...) {
continue;
}
MasterElementInfoPerProc proc_infos(element_synchronizer, count, my_rank,
type, partition);
proc_infos.synchronizeConnectivities();
proc_infos.synchronizePartitions();
proc_infos.synchronizeTags();
proc_infos.synchronizeGroups();
++count;
}
{ /// Ending the synchronization of elements by sending a stop message
MasterElementInfoPerProc proc_infos(element_synchronizer, count, my_rank,
_not_defined, partition);
++count;
}
/**
* Nodes synchronization
*/
MasterNodeInfoPerProc node_proc_infos(node_synchronizer, count, my_rank);
node_proc_infos.synchronizeNodes();
node_proc_infos.synchronizeTypes();
node_proc_infos.synchronizeGroups();
MeshUtils::fillElementToSubElementsData(mesh);
mesh_accessor.setDistributed();
AKANTU_DEBUG_OUT();
}
/* ------------------------------------------------------------------------ */
void distributeMeshCentralized(Mesh & mesh, UInt root) {
MeshAccessor mesh_accessor(mesh);
ElementSynchronizer & element_synchronizer =
mesh_accessor.getElementSynchronizer();
NodeSynchronizer & node_synchronizer = mesh_accessor.getNodeSynchronizer();
const StaticCommunicator & comm = element_synchronizer.getCommunicator();
UInt nb_proc = comm.getNbProc();
mesh_accessor.getNodesGlobalIds().resize(0);
if (nb_proc == 1)
return;
mesh.synchronizeGroupNames();
/**
* connectivity and communications scheme construction on distant
* processors
*/
UInt count = 0;
bool need_synchronize = true;
do {
/* --------<<<<-SIZE--------------------------------------------------- */
SlaveElementInfoPerProc proc_infos(element_synchronizer, count, root);
need_synchronize = proc_infos.needSynchronize();
if (need_synchronize) {
proc_infos.synchronizeConnectivities();
proc_infos.synchronizePartitions();
proc_infos.synchronizeTags();
proc_infos.synchronizeGroups();
}
++count;
} while (need_synchronize);
/**
* Nodes synchronization
*/
SlaveNodeInfoPerProc node_proc_infos(node_synchronizer, count, root);
node_proc_infos.synchronizeNodes();
node_proc_infos.synchronizeTypes();
node_proc_infos.synchronizeGroups();
MeshUtils::fillElementToSubElementsData(mesh);
mesh_accessor.setDistributed();
}
} // MeshUtilsDistribution
} // akantu
c4science · Help