Loading...
Searching...
No Matches
CollectDataOnIORank.hpp
// -*- mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
// vi: set et ts=4 sw=4 sts=4:
/*
This file is part of the Open Porous Media project (OPM).
OPM 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 2 of the License, or
(at your option) any later version.
OPM 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 OPM. If not, see <http://www.gnu.org/licenses/>.
Consult the COPYING file in the top-level source directory of this
module for the precise wording of the license and the list of
copyright holders.
*/
#ifndef OPM_COLLECT_DATA_ON_IO_RANK_HPP
#define OPM_COLLECT_DATA_ON_IO_RANK_HPP
#include <opm/grid/common/p2pcommunicator.hh>
#include <opm/input/eclipse/Schedule/Well/WellTestState.hpp>
#include <opm/output/data/Aquifer.hpp>
#include <opm/output/data/Cells.hpp>
#include <opm/output/data/Groups.hpp>
#include <opm/output/data/Solution.hpp>
#include <opm/output/data/Wells.hpp>
#include <opm/simulators/flow/FlowsData.hpp>
#include <opm/simulators/flow/InterRegFlows.hpp>
#include <array>
#include <cstddef>
#include <map>
#include <set>
#include <string>
#include <type_traits>
#include <utility>
#include <vector>
namespace Dune {
template<class Grid> class CartesianIndexMapper;
}
namespace Opm {
template <class Grid, class EquilGrid, class GridView>
class CollectDataOnIORank
{
public:
using CollectiveCommunication = typename Grid::Communication;
using P2PCommunicatorType = Dune::Point2PointCommunicator<Dune::SimpleMessageBuffer>;
using IndexMapType = std::vector<int>;
using IndexMapStorageType = std::vector<IndexMapType>;
static constexpr int dimension = Grid::dimension;
enum { ioRank = 0 };
static const bool needsReordering =
!std::is_same<Grid, EquilGrid>::value;
CollectDataOnIORank(const Grid& grid,
const EquilGrid* equilGrid,
const GridView& gridView,
const Dune::CartesianIndexMapper<Grid>& cartMapper,
const Dune::CartesianIndexMapper<EquilGrid>* equilCartMapper,
const std::set<std::string>& fipRegionsInterregFlow = {});
// gather solution to rank 0 for EclipseWriter
void collect(const data::Solution& localCellData,
const std::map<std::pair<std::string, int>, double>& localBlockData,
std::map<std::pair<std::string, int>, double>& localExtraBlockData,
const data::Wells& localWellData,
const data::WellBlockAveragePressures& localWBPData,
const data::GroupAndNetworkValues& localGroupAndNetworkData,
const data::Aquifers& localAquiferData,
const WellTestState& localWellTestState,
const InterRegFlowMap& interRegFlows,
const std::array<FlowsData<double>, 3>& localFlowsn,
const std::array<FlowsData<double>, 3>& localFloresn);
const std::map<std::pair<std::string, int>, double>& globalBlockData() const
{ return globalBlockData_; }
const data::Solution& globalCellData() const
{ return globalCellData_; }
data::Solution& globalCellData()
{ return globalCellData_; }
const data::Wells& globalWellData() const
{ return globalWellData_; }
const data::WellBlockAveragePressures& globalWBPData() const
{ return this->globalWBPData_; }
const data::GroupAndNetworkValues& globalGroupAndNetworkData() const
{ return globalGroupAndNetworkData_; }
const data::Aquifers& globalAquiferData() const
{ return globalAquiferData_; }
const WellTestState& globalWellTestState() const
{ return this->globalWellTestState_; }
InterRegFlowMap& globalInterRegFlows()
{ return this->globalInterRegFlows_; }
const InterRegFlowMap& globalInterRegFlows() const
{ return this->globalInterRegFlows_; }
const std::array<FlowsData<double>, 3>& globalFlowsn() const
{ return globalFlowsn_; }
const std::array<FlowsData<double>, 3>& globalFloresn() const
{ return globalFloresn_; }
bool isIORank() const
{ return toIORankComm_.rank() == ioRank; }
bool isParallel() const
{ return toIORankComm_.size() > 1; }
int localIdxToGlobalIdx(unsigned localIdx) const;
const std::vector<int>& localIdxToGlobalIdxMapping() const
{
return localIdxToGlobalIdx_;
}
bool doesNeedReordering() const
{ return needsReordering;}
std::size_t numCells () const
{ return globalCartesianIndex_.size(); }
const std::vector<int>& globalRanks() const
{ return globalRanks_; }
bool isCartIdxOnThisRank(int cartIdx) const;
protected:
P2PCommunicatorType toIORankComm_;
InterRegFlowMap globalInterRegFlows_;
IndexMapType globalCartesianIndex_;
IndexMapType localIndexMap_;
IndexMapStorageType indexMaps_;
std::vector<int> globalRanks_;
data::Solution globalCellData_;
std::map<std::pair<std::string, int>, double> globalBlockData_;
data::Wells globalWellData_;
data::WellBlockAveragePressures globalWBPData_;
data::GroupAndNetworkValues globalGroupAndNetworkData_;
data::Aquifers globalAquiferData_;
WellTestState globalWellTestState_;
std::vector<int> localIdxToGlobalIdx_;
std::array<FlowsData<double>, 3> globalFlowsn_;
std::array<FlowsData<double>, 3> globalFloresn_;
std::vector<int> sortedCartesianIdx_;
};
} // end namespace Opm
#endif // OPM_COLLECT_DATA_ON_IO_RANK_HPP
InterRegFlows.hpp
MPI-aware facility for converting collection of tuples of region ID pairs and associate flow rates in...
Dune::CartesianIndexMapper
Definition CollectDataOnIORank.hpp:49
Opm::CollectDataOnIORank::sortedCartesianIdx_
std::vector< int > sortedCartesianIdx_
sorted list of cartesian indices present-
Definition CollectDataOnIORank.hpp:170
Opm::InterRegFlowMap
Inter-region flow accumulation maps for all region definition arrays.
Definition InterRegFlows.hpp:179
Opm
This file contains a set of helper functions used by VFPProd / VFPInj.
Definition blackoilbioeffectsmodules.hh:43
Opm::FlowsData
Simple container for FLOWS data.
Definition FlowsData.hpp:32