mirror of
https://github.com/OPM/ResInsight.git
synced 2025-02-25 18:55:39 -06:00
0b327ca83b
Avoid resize if data is already present Improve functions for getting grid data and accessing single cell values Add example for read and write of grid data
122 lines
3.4 KiB
Python
122 lines
3.4 KiB
Python
# pylint: disable=too-few-public-methods
|
|
|
|
"""
|
|
Module containing the Grid class, containing information
|
|
about Case grids.
|
|
"""
|
|
|
|
import Case_pb2
|
|
import Grid_pb2
|
|
import Grid_pb2_grpc
|
|
|
|
|
|
class Grid:
|
|
"""Grid Information. Created by methods in Case
|
|
:meth:`rips.case.grid()`
|
|
:meth:`rips.case.grids()`
|
|
"""
|
|
|
|
def __init__(self, index, case, channel):
|
|
self.__channel = channel
|
|
self.__stub = Grid_pb2_grpc.GridStub(self.__channel)
|
|
|
|
self.case = case
|
|
self.index = index
|
|
self.cached_dimensions = None
|
|
|
|
def dimensions(self):
|
|
"""The dimensions in i, j, k direction
|
|
|
|
Returns:
|
|
Vec3i: class with integer attributes i, j, k giving extent in all three dimensions.
|
|
"""
|
|
case_request = Case_pb2.CaseRequest(id=self.case.id)
|
|
|
|
if self.cached_dimensions is None:
|
|
self.cached_dimensions = self.__stub.GetDimensions(
|
|
Grid_pb2.GridRequest(case_request=case_request, grid_index=self.index)
|
|
).dimensions
|
|
|
|
return self.cached_dimensions
|
|
|
|
def cell_centers_async(self):
|
|
"""The cells center for all cells in given grid async.
|
|
|
|
Returns:
|
|
Iterator to a list of Vec3d: class with double attributes x, y, x giving cell centers
|
|
"""
|
|
case_request = Case_pb2.CaseRequest(id=self.case.id)
|
|
chunks = self.__stub.GetCellCenters(
|
|
Grid_pb2.GridRequest(case_request=case_request, grid_index=self.index)
|
|
)
|
|
for chunk in chunks:
|
|
yield chunk
|
|
|
|
def cell_centers(self):
|
|
"""The cell center for all cells in given grid
|
|
|
|
Returns:
|
|
List of Vec3d: class with double attributes x, y, x giving cell centers
|
|
"""
|
|
centers = []
|
|
chunks = self.cell_centers_async()
|
|
for chunk in chunks:
|
|
for center in chunk.centers:
|
|
centers.append(center)
|
|
return centers
|
|
|
|
def cell_corners_async(self):
|
|
"""The cell corners for all cells in given grid, async.
|
|
|
|
Returns:
|
|
iterator to a list of CellCorners: a class with Vec3d for each corner (c0, c1.., c7)
|
|
"""
|
|
case_request = Case_pb2.CaseRequest(id=self.case.id)
|
|
chunks = self.__stub.GetCellCorners(
|
|
Grid_pb2.GridRequest(case_request=case_request, grid_index=self.index)
|
|
)
|
|
|
|
for chunk in chunks:
|
|
yield chunk
|
|
|
|
def cell_corners(self):
|
|
"""The cell corners for all cells in given grid
|
|
|
|
Returns:
|
|
list of CellCorners: a class with Vec3d for each corner (c0, c1.., c7)
|
|
"""
|
|
corners = []
|
|
chunks = self.cell_corners_async()
|
|
for chunk in chunks:
|
|
for center in chunk.cells:
|
|
corners.append(center)
|
|
return corners
|
|
|
|
def property_data_index_from_ijk(self, i, j, k):
|
|
"""Compute property index from 1-based IJK cell address. Cell Property Result data is organized by I, J and K.
|
|
|
|
property_data_index = dims.i * dims.j * (k - 1) + dims.i * (j - 1) + (i - 1)
|
|
|
|
Returns:
|
|
int: Cell property result index from IJK
|
|
"""
|
|
|
|
dims = self.dimensions()
|
|
|
|
property_data_index = dims.i * dims.j * (k - 1) + dims.i * (j - 1) + (i - 1)
|
|
|
|
return property_data_index
|
|
|
|
def cell_count(self):
|
|
"""Cell count in grid
|
|
|
|
Returns:
|
|
int: Cell count in grid
|
|
"""
|
|
|
|
dims = self.dimensions()
|
|
|
|
count = dims.i * dims.j * dims.k
|
|
|
|
return count
|