OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2024-12-18 21:43:27 -06:00
Code Issues Packages Projects Releases Wiki Activity

WellState: move mapping from well name to well index to here from autodiff's WellStateFullyImplicitBlackoil.

Browse Source
This commit is contained in:
Robert Kloefkorn 2015-09-15 13:00:01 +02:00
parent 99368d73d2
commit 14d7ce43a6
1 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
opm/core/simulator/WellState.hpp
View File

@ -23,6 +23,8 @@
#include <opm/core/wells.h>
#include <opm/core/well_controls.h>
#include <vector>
#include <array>
#include <map>
#include <cassert>
#include <cstddef>
@ -32,6 +34,10 @@ namespace Opm
/// The state of a set of wells.
class WellState
{
protected:
typedef std::array< int, 3 > mapentry_t;
typedef std::map< std::string, mapentry_t > WellMapType;
public:
/// Allocate and initialize if wells is non-null.
/// Also tries to give useful initial values to the bhp() and
@ -41,6 +47,9 @@ namespace Opm
template <class State>
void init(const Wells* wells, const State& state)
{
// clear old name mapping
wellMap_.clear();
if (wells) {
const int nw = wells->number_of_wells;
const int np = wells->number_of_phases;
@ -51,6 +60,19 @@ namespace Opm
for (int w = 0; w < nw; ++w) {
assert((wells->type[w] == INJECTOR) || (wells->type[w] == PRODUCER));
const WellControls* ctrl = wells->ctrls[w];
// setup wellname -> well index mapping
{
std::string name( wells->name[ w ] );
assert( name.size() > 0 );
mapentry_t& wellMapEntry = wellMap_[name];
wellMapEntry[ 0 ] = w;
wellMapEntry[ 1 ] = wells->well_connpos[w];
// also store the number of perforations in this well
const int num_perf_this_well = wells->well_connpos[w + 1] - wells->well_connpos[w];
wellMapEntry[ 2 ] = num_perf_this_well;
}
if (well_controls_well_is_stopped(ctrl)) {
// Stopped well:
// 1. Rates: assign zero well rates.
@ -170,6 +192,21 @@ namespace Opm
return getRestartTemperatureOffset() + temperature_.size();
}
const WellMapType& wellMap() const { return wellMap_; }
WellMapType& wellMap() { return wellMap_; }
/// The number of wells present.
int numWells() const
{
return bhp().size();
}
/// The number of phases present.
int numPhases() const
{
return wellRates().size() / numWells();
}
private:
std::vector<double> bhp_;
std::vector<double> thp_;
@ -177,6 +214,8 @@ namespace Opm
std::vector<double> wellrates_;
std::vector<double> perfrates_;
std::vector<double> perfpress_;
WellMapType wellMap_;
};
} // namespace Opm