OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

Implement Interior Cell Counts for Grid Views

Browse Source 
This is mostly to prepare for implementing partitionCells() for
grid views instead of grids.
This commit is contained in:
Bård Skaflestad
2023-09-12 12:37:10 +02:00
parent 1d33e7caf0
commit a2d43cd0f5
1 changed files with 35 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
opm/simulators/flow/countGlobalCells.hpp
View File

@@ -28,41 +28,55 @@
#include <dune/grid/common/gridview.hh>
#include <any>
#include <algorithm>
#include <vector>
namespace Opm {
namespace detail {
/// \brief Get the number of local interior cells in a grid view.
///
/// \tparam GridView Type of DUNE grid view.
///
/// \param[in] gridView Grid view for which to count the number of
/// interior cells.
///
/// \return The number of interior cell in the partition of the grid
/// stored on this process.
template <class GridView>
std::size_t countLocalInteriorCellsGridView(const GridView& gridView)
{
if (gridView.comm().size() == 1) {
return gridView.size(0);
}
return std::distance(gridView.template begin<0, Dune::Interior_Partition>(),
gridView.template end<0, Dune::Interior_Partition>());
}
/// \brief Get the number of local interior cells in a grid.
/// \tparam The type of the DUNE grid.
/// \param grid The grid which cells we count
/// \return The number of interior cell in the partition of the
/// grid stored on this process.
///
/// \tparam Grid Type of DUNE grid.
///
/// \param[in] grid Grid for which to count interior cells.
///
/// \return The number of interior cells in the partition of the
/// grid stored on this process.
template<class Grid>
std::size_t countLocalInteriorCells(const Grid& grid)
{
if ( grid.comm().size() == 1)
{
return grid.size(0);
}
std::size_t count = 0;
const auto& gridView = grid.leafGridView();
for(auto cell = gridView.template begin<0, Dune::Interior_Partition>(),
endCell = gridView.template end<0, Dune::Interior_Partition>();
cell != endCell; ++cell)
{
++count;
}
return count;
return countLocalInteriorCellsGridView(grid.leafGridView());
}
/// \brief Get the number of cells of a global grid.
///
/// In a parallel run this is the number of cells that a grid would
/// have if the whole grid was stored on one process only.
/// \tparam The type of the DUNE grid.
/// \param grid The grid which cells we count
///
/// \tparam Grid Type of DUNE grid.
///
/// \param[in] grid Grid for which to count global cells.
///
/// \return The global number of cells.
template<class Grid>
std::size_t countGlobalCells(const Grid& grid)
@@ -71,7 +85,7 @@ namespace detail {
{
return grid.size(0);
}
std::size_t count = countLocalInteriorCells(grid);
const std::size_t count = countLocalInteriorCellsGridView(grid.leafGridView());
return grid.comm().sum(count);
}