mirror of
https://github.com/OPM/ResInsight.git
synced 2026-10-26 07:00:17 -05:00
Kristian Bendiksen
7899736f05
* Python: Clean up imports. * Python: Clean up error handling example. * Python: Remove unused local variables * Python: Avoid comparing against True/False. * Python: Remove unused name variable. * Python: Reuse class definition type variable. * Python: Fix more unused imports * Black fixes
74 lines
1.8 KiB
Python
74 lines
1.8 KiB
Python
"""
|
|
Grid Case Group statistics module
|
|
"""
|
|
|
|
from .pdmobject import add_method
|
|
from .view import View as View
|
|
from .case import Case as Case
|
|
|
|
import Commands_pb2
|
|
from .resinsight_classes import GridCaseGroup
|
|
from .resinsight_classes import EclipseView
|
|
from .resinsight_classes import RimStatisticalCalculation
|
|
|
|
|
|
@add_method(GridCaseGroup)
|
|
def statistics_cases(self):
|
|
"""Get a list of all statistics cases in the Grid Case Group
|
|
|
|
Returns:
|
|
List of :class:`rips.generated.generated_classes.EclipseCase`
|
|
|
|
"""
|
|
return self.descendants(RimStatisticalCalculation)
|
|
|
|
|
|
@add_method(GridCaseGroup)
|
|
def views(self):
|
|
"""Get a list of views belonging to a grid case group
|
|
|
|
Returns:
|
|
List of :class:`rips.generated.generated_classes.EclipseView`
|
|
|
|
"""
|
|
resinsight_classes = self.descendants(EclipseView)
|
|
view_list = []
|
|
for pdm_object in resinsight_classes:
|
|
view_list.append(pdm_object)
|
|
return view_list
|
|
|
|
|
|
@add_method(GridCaseGroup)
|
|
def view(self, view_id):
|
|
"""Get a particular view belonging to a case group by providing view id
|
|
|
|
Arguments:
|
|
id(int): view id
|
|
|
|
Returns:
|
|
List of :class:`rips.generated.generated_classes.EclipseView`
|
|
|
|
"""
|
|
views = self.views()
|
|
for view_object in views:
|
|
if view_object.id == view_id:
|
|
return view_object
|
|
return None
|
|
|
|
|
|
@add_method(GridCaseGroup)
|
|
def compute_statistics(self, case_ids=None):
|
|
"""Compute statistics for the given case ids
|
|
|
|
Arguments:
|
|
case_ids(list of integers): List of case ids. If this is None all cases in group are included
|
|
|
|
"""
|
|
if case_ids is None:
|
|
case_ids = []
|
|
return self._execute_command(
|
|
computeCaseGroupStatistics=Commands_pb2.ComputeCaseGroupStatRequest(
|
|
caseIds=case_ids, caseGroupId=self.group_id
|
|
)
|
|
)
|