1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
from opm._common import EModel, eclArrType, EModel_add_filter_docstring
from opm._common import calc_cell_vol
def emodel_add_filter(self, key, operator, val1, val2 = None):
if key in ["I", "J", "K", "ROW", "COLUMN", "LAYER"]:
arrType=eclArrType.INTE
else:
arrList = self.get_list_of_arrays()
arrNameList = [item[0] for item in arrList]
arrTypeList = [item[1] for item in arrList]
if key not in arrNameList:
raise ValueError("{} array not found in EModel".format(key))
ind = arrNameList.index(key)
arrType = arrTypeList[ind]
if not val2:
if arrType == eclArrType.INTE:
self.__add_filter(key, operator, int(val1))
elif arrType == eclArrType.REAL:
self.__add_filter(key, operator, float(val1))
else:
if arrType == eclArrType.INTE:
self.__add_filter(key, operator, int(val1), int(val2))
elif arrType == eclArrType.REAL:
self.__add_filter(key, operator, float(val1), float(val2))
emodel_add_filter.__doc__ = EModel_add_filter_docstring
setattr(EModel, "add_filter", emodel_add_filter)
|
