57 lines
2.0 KiB
C++
57 lines
2.0 KiB
C++
#include <opm/parser/eclipse/EclipseState/Schedule/Well/Well2.hpp>
|
|
#include <pybind11/stl.h>
|
|
#include "common.hpp"
|
|
|
|
|
|
namespace {
|
|
|
|
std::vector<Connection> connections( const Well2& w ) {
|
|
const auto& well_connections = w.getConnections();
|
|
return std::vector<Connection>(well_connections.begin(), well_connections.end());
|
|
}
|
|
|
|
std::string status( const Well2& w ) {
|
|
return WellCommon::Status2String( w.getStatus() );
|
|
}
|
|
|
|
std::string preferred_phase( const Well2& w ) {
|
|
switch( w.getPreferredPhase() ) {
|
|
case Phase::OIL: return "OIL";
|
|
case Phase::GAS: return "GAS";
|
|
case Phase::WATER: return "WATER";
|
|
default: throw std::logic_error( "Unhandled enum value" );
|
|
}
|
|
}
|
|
|
|
int (Well2::*headI)() const = &Well2::getHeadI;
|
|
int (Well2::*headJ)() const = &Well2::getHeadI;
|
|
double (Well2::*refD)() const = &Well2::getRefDepth;
|
|
|
|
int (Well2::*headI_at)() const = &Well2::getHeadI;
|
|
int (Well2::*headJ_at)() const = &Well2::getHeadI;
|
|
double (Well2::*refD_at)() const = &Well2::getRefDepth;
|
|
|
|
}
|
|
|
|
void opmcommon_python::export_Well(py::module& module) {
|
|
|
|
py::class_< Well2 >( module, "Well")
|
|
.def_property_readonly( "name", &Well2::name )
|
|
.def_property_readonly( "preferred_phase", &preferred_phase )
|
|
.def( "I", headI )
|
|
.def( "I", headI_at )
|
|
.def( "J", headJ )
|
|
.def( "J", headJ_at )
|
|
.def( "ref", refD )
|
|
.def( "ref", refD_at )
|
|
.def( "status", &status )
|
|
.def( "isdefined", &Well2::hasBeenDefined )
|
|
.def( "isinjector", &Well2::isInjector )
|
|
.def( "isproducer", &Well2::isProducer )
|
|
.def( "group", &Well2::groupName )
|
|
.def( "guide_rate", &Well2::getGuideRate )
|
|
.def( "available_gctrl", &Well2::isAvailableForGroupControl )
|
|
.def( "_connections", &connections );
|
|
|
|
}
|