GeometryPartitioner.hpp

Go to the documentation of this file.

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
   This file is part of CEPS.
 
   CEPS is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.
 
   CEPS 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 General Public License for more details.
 
   You should have received a copy of the GNU General Public License
   along with CEPS (see file LICENSE at root of project).
   If not, see <https://www.gnu.org/licenses/>.
 
 
   Copyright 2019-2024 Inria, Universite de Bordeaux
 
   Authors, in alphabetical order:
 
     Pierre-Elliott BECUE, Florian CARO, Yves COUDIERE(*), Andjela DAVIDOVIC, 
     Charlie DOUANLA-LONTSI, Marc FUENTES, Mehdi JUHOOR, Michael LEGUEBE(*), 
     Pauline MIGERDITICHAN, Valentin PANNETIER(*), Nejib ZEMZEMI.
     * : currently active authors
 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
#pragma once
 
#include "common/CepsCommon.hpp"
#include "geometry/Geometry.hpp"
#include "geometry/partitioning/PartitioningMethod.hpp"
 
class GeometryPartitioner : public CepsObject
{
public:
  GeometryPartitioner () = delete;
 
  explicit GeometryPartitioner (Geometry *geometry);
 
  ~GeometryPartitioner () override;
 
  virtual CepsUInt
  computePartition() = 0;
 
protected:
  void
  readMeshOnMaster();
 
  void
  broadcastMeshData();
 
  void
  setUpHalo (
    const CepsVector<CepsInt> &nodeOwnership,
    const CepsVector<CepsInt> *cellOwnership = nullptr
  );
 
  void
  buildNodeAdjacency();
 
  void
  buildCellAdjacency();
 
  void
  nodeAttribution(const CepsVector<CepsInt> &nodeOwnership);
 
  void
  cellAttribution();
 
  template <CepsUInt _Dim>
  void
  addCellToMesh (CepsUInt index, CepsBool isOnBoundary, Mesh *mesh);
 
  CepsUInt
  getDimensionByNodeId (CepsUInt i) const;
 
  CepsUInt
  getDimensionByCellId (CepsUInt i) const;
 
 
protected:
  Geometry *m_geom;                           
 
  CepsSet<CepsCellGlobalIndex> m_ownedCells;  
  CepsSet<CepsNodeGlobalIndex> m_ownedNodes;  
  CepsSet<CepsNodeGlobalIndex> m_haloNodes;   
 
  CepsArray4<CepsUInt> m_nodeOffset;          
  CepsArray4<CepsUInt> m_cellOffset;          
 
  // Node data
  CepsUInt                  m_nbNodes;         
  CepsVector<CepsReal>      m_coords;          
  CepsVector<CepsAttribute> m_nodeAttr;        
  CepsVector<CepsUInt>      m_nodeAttrOffset;  
  CepsUInt                  m_nodeAttrSize;    
 
  CepsMap<CepsNodeGlobalIndex,CepsSet<CepsNodeGlobalIndex>> m_nodeNodeAdj;  
  CepsMap<CepsNodeGlobalIndex,CepsSet<CepsCellGlobalIndex>> m_nodeCellAdj;  
  CepsMap<CepsCellGlobalIndex,CepsSet<CepsCellGlobalIndex>> m_cellCellAdj;  
  CepsSet<CepsCellGlobalIndex>                              m_markedCells;  
 
  CepsMap<CepsCellGlobalIndex,CepsMathVertex> m_cellBarycenters; 
 
  CepsMultiMap<CepsNodeGlobalIndex, CepsNodeGlobalIndex>
    m_coupledNodes;  
 
  // Cell data
  CepsUInt m_nbCells;  
  CepsVector<CepsInt>
    m_nodeIndices;     
  CepsVector<CepsUInt>      m_nodeIndicesOffset;  
  CepsUInt                  m_nodeIndicesSize;    
  CepsVector<CepsAttribute> m_cellAttr;           
  CepsVector<CepsUInt>      m_cellAttrOffset;     
  CepsUInt                  m_cellAttrSize;       
 
  CepsVector<CepsChar> m_isBoundary;
};