mirror of
https://github.com/OPM/ResInsight.git
synced 2025-02-25 18:55:39 -06:00
75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
"""
|
|
ResInsight SimulationWell
|
|
"""
|
|
import grpc
|
|
|
|
import SimulationWell_pb2
|
|
import SimulationWell_pb2_grpc
|
|
|
|
import Properties_pb2
|
|
import Properties_pb2_grpc
|
|
|
|
from .resinsight_classes import SimulationWell
|
|
|
|
from .pdmobject import PdmObjectBase, add_method
|
|
|
|
import rips.case
|
|
|
|
|
|
@add_method(SimulationWell)
|
|
def __custom_init__(self, pb2_object, channel):
|
|
self._simulation_well_stub = SimulationWell_pb2_grpc.SimulationWellStub(channel)
|
|
|
|
|
|
@add_method(SimulationWell)
|
|
def status(self, timestep):
|
|
"""Get simulation well status
|
|
|
|
**SimulationWellStatus class description**::
|
|
|
|
Parameter | Description | Type
|
|
----------- | ------------------------------------------------------------- | -----
|
|
well_type | Well type as string | string
|
|
is_open | True if simulation well is open at the specified time step | bool
|
|
|
|
Arguments:
|
|
timestep(int): Time step index
|
|
|
|
"""
|
|
sim_well_request = SimulationWell_pb2.SimulationWellRequest(
|
|
case_id=self.case().id, well_name=self.name, timestep=timestep
|
|
)
|
|
return self._simulation_well_stub.GetSimulationWellStatus(sim_well_request)
|
|
|
|
|
|
@add_method(SimulationWell)
|
|
def cells(self, timestep):
|
|
"""Get reservoir cells the simulation well is defined for
|
|
|
|
**SimulationWellCellInfo class description**::
|
|
|
|
Parameter | Description | Type
|
|
----------- | --------------------------------------------------------- | -----
|
|
ijk | Cell IJK location | Vec3i
|
|
grid_index | Grid index | int
|
|
is_open | True if connection to is open at the specified time step | bool
|
|
branch_id | | int
|
|
segment_id | | int
|
|
|
|
Arguments:
|
|
timestep(int): Time step index
|
|
|
|
Returns:
|
|
List of SimulationWellCellInfo
|
|
|
|
"""
|
|
sim_well_request = SimulationWell_pb2.SimulationWellRequest(
|
|
case_id=self.case().id, well_name=self.name, timestep=timestep
|
|
)
|
|
return self._simulation_well_stub.GetSimulationWellCells(sim_well_request).data
|
|
|
|
|
|
@add_method(SimulationWell)
|
|
def case(self):
|
|
return self.ancestor(rips.case.Case)
|