mirror of
https://github.com/OPM/opm-simulators.git
synced 2024-12-22 07:23:27 -06:00
120 lines
4.1 KiB
C++
120 lines
4.1 KiB
C++
#ifndef OPM_WELLSGROUP_HPP
|
|
#define OPM_WELLSGROUP_HPP
|
|
|
|
#include <opm/core/InjectionSpecification.hpp>
|
|
#include <opm/core/ProductionSpecification.hpp>
|
|
#include <opm/core/eclipse/EclipseGridParser.hpp>
|
|
#include <opm/core/grid.h>
|
|
#include <string>
|
|
|
|
|
|
namespace Opm
|
|
{
|
|
|
|
class WellsGroupInterface
|
|
{
|
|
public:
|
|
WellsGroupInterface(const std::string& name,
|
|
ProductionSpecification prod_spec,
|
|
InjectionSpecification inj_spec);
|
|
virtual ~WellsGroupInterface();
|
|
|
|
/// The unique identifier for the well or well group.
|
|
const std::string& name();
|
|
|
|
/// Production specifications for the well or well group.
|
|
const ProductionSpecification& prodSpec() const;
|
|
|
|
/// Injection specifications for the well or well group.
|
|
const InjectionSpecification& injSpec() const;
|
|
|
|
/// Production specifications for the well or well group.
|
|
ProductionSpecification& prodSpec();
|
|
|
|
/// Injection specifications for the well or well group.
|
|
InjectionSpecification& injSpec();
|
|
|
|
|
|
/// \returns true if the object is a leaf node (WellNode), false otherwise.
|
|
virtual bool isLeafNode() const;
|
|
|
|
/// \returns the pointer to the WellsGroupInterface with the given name. NULL if
|
|
/// the name is not found.a
|
|
virtual WellsGroupInterface* findGroup(std::string name_of_node) = 0;
|
|
|
|
void setParent(WellsGroupInterface* parent);
|
|
const WellsGroupInterface* getParent() const;
|
|
|
|
/// Recursively calculate the guide rate for each member of the well group.
|
|
/// This should be called after the guide rates are set to the non-normalized values.
|
|
virtual void calculateGuideRates() = 0;
|
|
|
|
/// Calculates the number of leaf nodes in the given group.
|
|
/// A leaf node is defined to have one leaf node in its group.
|
|
virtual int numberOfLeafNodes() = 0;
|
|
protected:
|
|
WellsGroupInterface* parent_;
|
|
|
|
private:
|
|
std::string name_;
|
|
ProductionSpecification production_specification_;
|
|
InjectionSpecification injection_specification_;
|
|
};
|
|
|
|
|
|
|
|
class WellsGroup : public WellsGroupInterface
|
|
{
|
|
public:
|
|
WellsGroup(const std::string& name,
|
|
ProductionSpecification prod_spec,
|
|
InjectionSpecification inj_spec);
|
|
|
|
virtual WellsGroupInterface* findGroup(std::string name_of_node);
|
|
|
|
void addChild(std::tr1::shared_ptr<WellsGroupInterface> child);
|
|
|
|
bool conditionsMet(const std::vector<double>& well_bhp, const std::vector<double>& well_rate,
|
|
const UnstructuredGrid& grid, const std::vector<double>& saturations, const struct Wells* wells,
|
|
int index_of_well, double epsilon = 1e-8);
|
|
|
|
virtual void calculateGuideRates();
|
|
|
|
|
|
virtual int numberOfLeafNodes();
|
|
private:
|
|
std::vector<std::tr1::shared_ptr<WellsGroupInterface> > children_;
|
|
};
|
|
|
|
|
|
|
|
class WellNode : public WellsGroupInterface
|
|
{
|
|
public:
|
|
WellNode(const std::string& name,
|
|
ProductionSpecification prod_spec,
|
|
InjectionSpecification inj_spec);
|
|
|
|
virtual WellsGroupInterface* findGroup(std::string name_of_node);
|
|
virtual bool conditionsMet(const std::vector<double>& well_bhp, const std::vector<double>& well_rate,
|
|
const UnstructuredGrid& grid, const std::vector<double>& saturations, double epsilon=1e-8);
|
|
virtual bool isLeafNode() const;
|
|
|
|
void setWellsPointer(const struct Wells* wells, int self_index);
|
|
|
|
virtual void calculateGuideRates();
|
|
virtual int numberOfLeafNodes();
|
|
private:
|
|
const struct Wells* wells_;
|
|
int self_index_;
|
|
};
|
|
|
|
/// Doc me!
|
|
std::tr1::shared_ptr<WellsGroupInterface> createWellsGroup(std::string name,
|
|
const EclipseGridParser& deck);
|
|
|
|
|
|
}
|
|
#endif /* OPM_WELLSGROUP_HPP */
|
|
|