Mesh.cpp

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
 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
#include "geometry/Mesh.hpp"
 
Mesh::Mesh(CepsUInt dim) :
  m_dim                  (dim),
  m_geom                 (nullptr),
  m_nbCells              (0),
  m_nbBoundaryCells      (0),
  m_nbNodes              (0),
  m_connectivityComputed (false),
  m_maxCellDiameter      (0.e0),
  m_maxNodeConnectivity  (0)
{
}
 
Mesh::~Mesh()
{
  m_cells     .destroyObjects();
  m_bCells    .destroyObjects();
  m_ownedNodes.destroyObjects();
  m_haloNodes .destroyObjects();
 
  ceps::destroyObject(m_cells     );
  ceps::destroyObject(m_bCells    );
  ceps::destroyObject(m_ownedNodes);
  ceps::destroyObject(m_haloNodes );
}
 
CepsUInt
Mesh::getDimension() const
{
  return m_dim;
}
 
void
Mesh::addCell(GeomCell *c)
{
  CEPS_ABORT_IF(not c,"adding a nullptr cell to the mesh");
  CEPS_ABORT_IF(c->getDimension () != m_dim,
    "trying to add a cell of dimension " << c->getDimension() 
          << "\nin a mesh of dimension " << m_dim
  );
 
  CepsCellGlobalIndex cID = c->getGlobalIndex();
  CEPS_ABORT_IF(m_cells.hasGlobal(cID),
    "Cell #" << cID << " was already added to the mesh"
  );
 
  // Register the cell
  m_cells.append(c,getCellHash(c),cID);
 
  m_maxCellDiameter = std::max(m_maxCellDiameter,c->getDiameter());
 
  return;
}
 
void
Mesh::addBoundaryCell(GeomCell *c)
{
  CEPS_ABORT_IF(not c, "adding a nullptr cell to the mesh");
  CEPS_ABORT_IF(c->getDimension () != m_dim-1,
    "trying to add a cell of dimension " << c->getDimension () 
          << "\nin a mesh of dimension " << m_dim
  );
 
  CepsCellGlobalIndex cID = c->getGlobalIndex ();
  CEPS_ABORT_IF(m_bCells.hasGlobal(cID),
    "Cell #" << cID << " was already added to the mesh"
  );
 
  // Register the cell
  m_bCells.append(c,getCellHash(c),cID);
  m_maxCellDiameter = std::max(m_maxCellDiameter,c->getDiameter());
  return;
}
 
void Mesh::addNode(GeomNode *n)
{
  CEPS_ABORT_IF(not n, "adding a nullptr node to the mesh");
  CepsNodeGlobalIndex nID = n->getGlobalIndex();
  CEPS_ABORT_IF(m_ownedNodes.hasGlobal(nID),
    "trying to add node " << nID << " that is already added to the mesh"
  );
 
  // Register the node
  m_ownedNodes.append(n,getNodeHash(n),nID);
  return;
}
 
void
Mesh::addHaloNode(GeomNode *n)
{
  CEPS_ABORT_IF(not n,
    "adding a nullptr node to the mesh"
  );
  CepsNodeGlobalIndex nID = n->getGlobalIndex();
  CEPS_ABORT_IF(m_haloNodes.hasGlobal(nID),
    "trying to add node " << nID << " that is already added to the mesh"
  );
  // Register the node
  m_haloNodes.append(n,getNodeHash(n),nID);
  return;
}
 
void
Mesh::refreshCells()
{
  for (GeomCell *&cell : m_cells.all())
    cell->refresh ();
  for (GeomCell *&boundaryCell : m_bCells.all())
    boundaryCell->refresh();
  return;
}
 
void
Mesh::setNbCells(CepsUInt n)
{
  m_nbCells = n;
}
 
void
Mesh::setNbBoundaryCells(CepsUInt n)
{
  m_nbBoundaryCells = n;
}
 
void Mesh::setNbNodes(CepsUInt n)
{
  m_nbNodes = n;
}
 
GeomNode*
Mesh::getNode(CepsNodeGlobalIndex geomId) const
{
  return m_ownedNodes.atGlobal(geomId,nullptr);
}
 
GeomNode*
Mesh::getHaloNode(CepsNodeGlobalIndex geomId) const
{
  return m_haloNodes.atGlobal(geomId,nullptr);
}
 
GeomNode*
Mesh::getNodeOrHaloNode(CepsNodeGlobalIndex geomId) const
{
  return m_ownedNodes.atGlobal(geomId,getHaloNode(geomId));
}
 
CepsInt
Mesh::getLocalIndexOfCell(CepsCellGlobalIndex geomId) const
{
  return m_cells.hasGlobal(geomId) ? m_cells.globalToLocal(geomId) : -1;
}
 
CepsInt
Mesh::getLocalIndexOfBoundaryCell(CepsCellGlobalIndex geomId) const
{
  return m_bCells.hasGlobal(geomId) ? m_bCells.globalToLocal(geomId) : -1;
}
 
GeomCell*
Mesh::getCell(CepsCellGlobalIndex geomId) const
{
  return m_cells.atGlobal(geomId,nullptr);
}
 
GeomCell*
Mesh::getBoundaryCell(CepsCellGlobalIndex geomId) const
{
  return m_bCells.atGlobal(geomId,nullptr);
}
 
const CepsVector<GeomCell*>&
Mesh::getCells()
{
  return m_cells.all();
}
 
const CepsVector<GeomCell*>&
Mesh::getBoundaryCells()
{
  return m_bCells.all();
}
 
const CepsVector<GeomNode*>&
Mesh::getOwnedNodes()
{
  return m_ownedNodes.all();
}
 
const CepsVector<GeomNode*>&
Mesh::getHaloNodes()
{
  return m_haloNodes.all();
}
 
CepsUInt
Mesh::getNbCells() const
{
  return m_nbCells;
}
 
CepsUInt
Mesh::getNbBoundaryCells() const
{
  return m_nbBoundaryCells;
}
 
CepsUInt
Mesh::getNbNodes() const
{
  return m_nbNodes;
}
 
CepsUInt
Mesh::getLocalNbCells() const
{
  return m_cells.size();
}
 
CepsUInt
Mesh::getLocalNbBoundaryCells() const
{
  return m_bCells.size();
}
 
CepsUInt
Mesh::getLocalNbNodes() const
{
  return m_ownedNodes.size ();
}
 
CepsUInt
Mesh::getLocalNbHaloNodes() const
{
  return m_haloNodes.size();
}
 
CepsUInt
Mesh::getMaxNodeConnectivity()
{
  if (m_connectivityComputed)
    return m_maxNodeConnectivity;
  return this->computeMaxNodeConnectivity();
}
 
CepsReal
Mesh::getMaxCellDiameter() const
{
  return m_maxCellDiameter;
}
 
void
Mesh::setGeometry(Geometry *geom)
{
  m_geom = geom;
}
 
Geometry*
Mesh::getGeometry() const
{
  return m_geom;
}
 
void
Mesh::scale(CepsReal scaleFactor)
{
  for (GeomNode* node: m_ownedNodes.all())
    node->scale(scaleFactor);
  for (GeomNode* node: m_haloNodes.all())
    node->scale(scaleFactor);
  // Recompute all jacobians 
  refreshCells ();
}
 
CepsHash
Mesh::getNodeHash(GeomNode* n) 
{
  return n->getGlobalIndex();
}
 
CepsHash
Mesh::getCellHash(GeomCell* c) 
{
  return c->getGlobalIndex();
}
 
CepsUInt
Mesh::computeMaxNodeConnectivity()
{
  m_maxNodeConnectivity = 0;
 
  // iterate through our nodes and halo node
  for (GeomNode *node : m_ownedNodes.all())
    m_maxNodeConnectivity = std::max (m_maxNodeConnectivity, node->getNumberOfNodes());
  for (GeomNode *node : m_haloNodes.all())
    m_maxNodeConnectivity = std::max (m_maxNodeConnectivity, node->getNumberOfNodes());
 
  m_connectivityComputed = true;
  return m_maxNodeConnectivity;
}
 
std::ostream&
operator<<(std::ostream& os, const Mesh& m)
{
  os << "Cells: #" << m.m_cells.size() << std::endl;
 
  for (GeomCell *el : m.m_cells.all())
  {
    os << el->getGlobalIndex () << " - nodes : ";
    for (GeomNode *node : el->getNodes())
      os << node->getGlobalIndex () << " ";
    os << std::endl;
  }
  os << CEPS_STRING_SEPARATOR << std::endl;
 
  os << "Boundary Cells: #" << m.m_bCells.size() << std::endl;
  for (GeomCell *el : m.m_bCells.all())
  {
    os << el->getGlobalIndex() << " - nodes : ";
    for (GeomNode *node : el->getNodes ())
      os << node->getGlobalIndex() << " ";
    os << std::endl;
  }
 
  os << CEPS_STRING_SEPARATOR << std::endl;
 
  os << "Owned Nodes: #" << m.m_ownedNodes.size() << std::endl;
  for (GeomNode *node : m.m_ownedNodes.all())
    os << node->getGlobalIndex() << " - coordinates : " << *node << std::endl;
 
  os << "Halo Nodes: #" << m.m_haloNodes.size() << std::endl;
  for (GeomNode *node : m.m_haloNodes.all())
    os << node->getGlobalIndex() << " - coordinates : " << *node << std::endl;
 
  return os;
}