#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 );

}