Source code for itm.grid.data
import cpo_tools
import type_helper
import numpy
class ScalarData:
'''Wraps a complexgrid_scalar data structure to provide high-level operations.'''
[docs]
def __init__(self, cpo, data, path = None):
#assert data is None | type_helper.is_complexgrid_scalar(data)
[docs] self._cpo = cpo
self._data = data
self._path = path
def __str__(self):
return str(self.cpo) + self._path
[docs]
@property
def cpo(self):
return self._cpo
[docs]
@property
def data(self):
return self._data
[docs]
@property
def path(self):
return self._path
[docs]
@property
def grid(self):
'''Return grid for this data object.'''
[docs] return self._cpo.grid
@property
def subgrid_index(self):
'''Return subgrid index for this data object.'''
[docs] return self._data.subgrid
@property
def subgrid(self):
'''Return subgrid for this data object.'''
[docs] return self._cpo.grid.subgrid(self._data.subgrid)
@property
def values(self):
# TODO: handle data representations
[docs] return self._data.scalar
def values_for_objects(self, objs):
'''Return values of this data object for a sequence of objects as 1d numpy array.
[docs]
If for any of the object no data is available, None is returned.'''
# TODO: this currently assumes default data representation.
try:
values = numpy.zeros([len(objs)])
indexMethod = self.subgrid.index
for iobj, obj in enumerate(objs):
lInd = indexMethod(obj)
values[iobj] = self.values[lInd]
except ValueError:
# This happens if one object cannot be found in the subgrid
return None
return values
#===============================================================================
# @property
# def arg_range(self):
# pass
#
#
# def eval(self, x):
# '''Evaluate variable at position x.'''
# pass
#
# def plot(self, fig=None):
# '''Plot data.'''
# pass
#===============================================================================
#===============================================================================
# class VectorData:
#
# def __init__(self, cpo, vdata=None):
# assert vdata is None | type_helper.is_complexgrid_vector(vdata)
#
# self._cpo = cpo
# self._vdata = vdata
#===============================================================================
class ScalarDataList:
'''Wrap a list of complexgrid_scalar data structures to provide high-level operations.'''
[docs]
def __init__(self, cpo, dataList, path):
self._cpo = cpo
[docs] self._dataList = dataList
self._path = path
def __len__(self):
return len(self._dataList)
[docs]
def __getitem__(self, key):
return ScalarData(self._cpo, self._dataList[key])
[docs]
def __str__(self):
return str(self.cpo) + self._path
[docs]
@property
def cpo(self):
return self._cpo
[docs]
@property
def data_list(self):
return self._dataList
[docs]
@property
def path(self):
return self._path
[docs]
def for_subgrid_index(self, sgInd):
for cgs_data in self._dataList:
[docs] if cgs_data.subgrid == sgInd:
return ScalarData(self._cpo, cgs_data)
return None
def for_subgrid_id(self, sgId):
return self.for_subgrid_index(self._cpo.grid.subgrid_for_id(sgId).subgrid_index)
[docs]
def values_for_objects(self, objs):
'''Return data values for the given objects. If values for the objects are stored in multiple
[docs] entries of the data list, the data from the first entry that can serve all objects is returned.
If no data entry is found that can serve all objects, None is returned.'''
for data in self:
values = data.values_for_objects(objs)
if values is not None:
return values
return None