0bc150794d
python: exposes Schedule. python: added constructor for Schedule. test_schedule: reintroduced most tests. python test_schedule: all test works. python test_wells. opened up some tests. test_wells: opened up most. test_wells.py ok.
50 lines
1.6 KiB
C++
50 lines
1.6 KiB
C++
#include <tuple>
|
|
|
|
#include <opm/parser/eclipse/EclipseState/Schedule/Well/Well2.hpp>
|
|
#include <pybind11/stl.h>
|
|
#include "export.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 Well2::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" );
|
|
}
|
|
}
|
|
|
|
std::tuple<int, int, double> get_pos( const Well2& w ) {
|
|
return std::make_tuple(w.getHeadI(), w.getHeadJ(), w.getRefDepth());
|
|
}
|
|
|
|
}
|
|
|
|
void python::common::export_Well(py::module& module) {
|
|
|
|
py::class_< Well2 >( module, "Well")
|
|
.def_property_readonly( "name", &Well2::name )
|
|
.def_property_readonly( "preferred_phase", &preferred_phase )
|
|
.def( "pos", &get_pos )
|
|
.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 );
|
|
|
|
}
|