def swapaxes(self):
"""Swap the axes inline vs xline, keep origin."""
ncol = _cxtgeo.new_intpointer()
nrow = _cxtgeo.new_intpointer()
yflip = _cxtgeo.new_intpointer()
xinc = _cxtgeo.new_doublepointer()
yinc = _cxtgeo.new_doublepointer()
rota = _cxtgeo.new_doublepointer()
_cxtgeo.intpointer_assign(ncol, self._ncol)
_cxtgeo.intpointer_assign(nrow, self._nrow)
_cxtgeo.intpointer_assign(yflip, self._yflip)
_cxtgeo.doublepointer_assign(xinc, self._xinc)
_cxtgeo.doublepointer_assign(yinc, self._yinc)
_cxtgeo.doublepointer_assign(rota, self._rotation)
values1d = self.values.reshape(-1)
traceid1d = self._traceidcodes.reshape(-1)
ier = _cxtgeo.cube_swapaxes(
ncol,
nrow,
self.nlay,
yflip,
self.xori,
xinc,
self.yori,
yinc,
rota,
values1d,
traceid1d,
0,
)
if ier != 0:
raise Exception
self._ncol = _cxtgeo.intpointer_value(ncol)
self._nrow = _cxtgeo.intpointer_value(nrow)
self._yflip = _cxtgeo.intpointer_value(yflip)
self._xinc = _cxtgeo.doublepointer_value(xinc)
self._yinc = _cxtgeo.doublepointer_value(yinc)
self._rotation = _cxtgeo.doublepointer_value(rota)
ilines = self._xlines.copy()
xlines = self._ilines.copy()
self._xlines = xlines
self._ilines = ilines
self._traceidcodes = traceid1d.reshape((self._ncol, self._nrow))
self._values = values1d.reshape((self._ncol, self._nrow, self.nlay))
def test_grd3d_export_grdeclprop_no_file():
fname = "random_file_name"
with open(fname, "w"):
pass
os.chmod(fname, stat.S_IREAD)
with pytest.raises(xtgeo.XTGeoCLibError, match="Could not open file"):
# The input here is not very important, what is important
# is that "existing_file" can not be opened.
_cxtgeo.grd3d_export_grdeclprop2(
1,
1,
1,
1,
_cxtgeo.new_intpointer(),
_cxtgeo.new_floatpointer(),
_cxtgeo.new_doublepointer(),
"name",
" %d",
fname,
1,
0,
)
os.chmod(fname, stat.S_IWRITE)
os.remove(fname)
def _export_roff_discrete(self,
pfile,
name,
append=False,
last=True,
binary=True):
carray = _gridprop_lowlevel.update_carray(self, undef=-999)
ptr_idum = _cxtgeo.new_intpointer()
ptr_ddum = _cxtgeo.new_doublepointer()
# codes:
ptr_codes = _cxtgeo.new_intarray(256)
ncodes = self.ncodes
codenames = ""
logger.info("Keys: %s", self.codes.keys())
for inum, ckey in enumerate(sorted(self.codes.keys())):
if ckey is not None:
codenames += str(self.codes[ckey])
codenames += "|"
_cxtgeo.intarray_setitem(ptr_codes, inum, int(ckey))
else:
logger.warning("For some odd reason, None is a key. Check!")
mode = 0
if not binary:
mode = 1
if not append:
_cxtgeo.grd3d_export_roff_pstart(mode, self._ncol, self._nrow,
self._nlay, pfile, XTGDEBUG)
nsub = 0
isub_to_export = 0
_cxtgeo.grd3d_export_roff_prop(
mode,
self._ncol,
self._nrow,
self._nlay,
nsub,
isub_to_export,
ptr_idum,
name,
"int",
carray,
ptr_ddum,
ncodes,
codenames,
ptr_codes,
pfile,
XTGDEBUG,
)
if last:
_cxtgeo.grd3d_export_roff_end(mode, pfile, XTGDEBUG)
_gridprop_lowlevel.delete_carray(self, carray)
def swapaxes(self):
"""Swap the axes columns vs rows, keep origin. Will change yflip."""
ncol = _cxtgeo.new_intpointer()
nrow = _cxtgeo.new_intpointer()
yflip = _cxtgeo.new_intpointer()
xinc = _cxtgeo.new_doublepointer()
yinc = _cxtgeo.new_doublepointer()
rota = _cxtgeo.new_doublepointer()
_cxtgeo.intpointer_assign(ncol, self._ncol)
_cxtgeo.intpointer_assign(nrow, self._nrow)
_cxtgeo.intpointer_assign(yflip, self._yflip)
_cxtgeo.doublepointer_assign(xinc, self._xinc)
_cxtgeo.doublepointer_assign(yinc, self._yinc)
_cxtgeo.doublepointer_assign(rota, self._rotation)
val = self.get_values1d(fill_value=xtgeo.UNDEF)
ier = _cxtgeo.surf_swapaxes(
ncol, nrow, yflip, self.xori, xinc, self.yori, yinc, rota, val, 0
)
if ier != 0:
raise RuntimeError(
"Unspecied runtime error from {}: Code: {}".format(__name__, ier)
)
self._ncol = _cxtgeo.intpointer_value(ncol)
self._nrow = _cxtgeo.intpointer_value(nrow)
self._yflip = _cxtgeo.intpointer_value(yflip)
self._xinc = _cxtgeo.doublepointer_value(xinc)
self._yinc = _cxtgeo.doublepointer_value(yinc)
self._rotation = _cxtgeo.doublepointer_value(rota)
ilines = self._xlines.copy()
xlines = self._ilines.copy()
self._ilines = ilines
self._xlines = xlines
self.values = val # reshaping and masking is done in self.values
def vectorinfo2(x1, x2, y1, y2, option=1):
"""
Get length and angles from 2 points in space (2D plane).
Option = 1 gives normal school angle (counterclock from X)
"""
# _cxtgeo.xtg_verbose_file("NONE")
lenp = _cxtgeo.new_doublepointer()
radp = _cxtgeo.new_doublepointer()
degp = _cxtgeo.new_doublepointer()
_cxtgeo.x_vector_info2(x1, x2, y1, y2, lenp, radp, degp, option, DBG)
llen = _cxtgeo.doublepointer_value(lenp)
rad = _cxtgeo.doublepointer_value(radp)
deg = _cxtgeo.doublepointer_value(degp)
return llen, rad, deg
def export_grdecl(self, pfile, name, append=False, binary=False, dtype=None):
logger.info("Exporting %s to file %s, GRDECL format", name, pfile)
if dtype is None:
if self._isdiscrete:
dtype = "int32"
else:
dtype = "float32"
carray = _gridprop_lowlevel.update_carray(self, dtype=dtype)
iarr = _cxtgeo.new_intpointer()
farr = _cxtgeo.new_floatpointer()
darr = _cxtgeo.new_doublepointer()
if "double" in str(carray):
ptype = 3
darr = carray
elif "float" in str(carray):
ptype = 2
farr = carray
else:
ptype = 1
iarr = carray
mode = 0
if not binary:
mode = 1
appendmode = 0
if append:
appendmode = 1
_cxtgeo.grd3d_export_grdeclprop2(
self._ncol,
self._nrow,
self._nlay,
ptype,
iarr,
farr,
darr,
name,
pfile,
mode,
appendmode,
XTGDEBUG,
)
_gridprop_lowlevel.delete_carray(self, carray)
def test_grd3d_export_grdeclprop_no_file(unreadable_file):
with pytest.raises(xtgeo.XTGeoCLibError, match="Could not open file"):
# The input here is not very important, what is important
# is that "existing_file" can not be opened.
_cxtgeo.grd3d_export_grdeclprop2(
1,
1,
1,
1,
_cxtgeo.new_intpointer(),
_cxtgeo.new_floatpointer(),
_cxtgeo.new_doublepointer(),
"name",
" %d",
unreadable_file,
1,
0,
)
def _import_roff_v1(self, pfile, name):
"""Import ROFF format, version 1"""
# pylint: disable=too-many-locals
# there is a todo here to get it more robust for various cases,
# e.g. that a ROFF file may contain both a grid an numerous
# props
logger.info("Looking for %s in file %s", name, pfile)
ptr_ncol = _cxtgeo.new_intpointer()
ptr_nrow = _cxtgeo.new_intpointer()
ptr_nlay = _cxtgeo.new_intpointer()
ptr_ncodes = _cxtgeo.new_intpointer()
ptr_type = _cxtgeo.new_intpointer()
ptr_idum = _cxtgeo.new_intpointer()
ptr_ddum = _cxtgeo.new_doublepointer()
# read with mode 0, to scan for ncol, nrow, nlay and ndcodes, and if
# property is found...
ier, _codenames = _cxtgeo.grd3d_imp_prop_roffbin(
pfile,
0,
ptr_type,
ptr_ncol,
ptr_nrow,
ptr_nlay,
ptr_ncodes,
name,
ptr_idum,
ptr_ddum,
ptr_idum,
0,
XTGDEBUG,
)
if ier == -1:
msg = "Cannot find property name {}".format(name)
logger.warning(msg)
raise SystemExit("Error from ROFF import")
self._ncol = _cxtgeo.intpointer_value(ptr_ncol)
self._nrow = _cxtgeo.intpointer_value(ptr_nrow)
self._nlay = _cxtgeo.intpointer_value(ptr_nlay)
self._ncodes = _cxtgeo.intpointer_value(ptr_ncodes)
ptype = _cxtgeo.intpointer_value(ptr_type)
ntot = self._ncol * self._nrow * self._nlay
if self._ncodes <= 1:
self._ncodes = 1
self._codes = {0: "undef"}
logger.debug("Number of codes: %s", self._ncodes)
# allocate
if ptype == 1: # float, assign to double
ptr_pval_v = _cxtgeo.new_doublearray(ntot)
ptr_ival_v = _cxtgeo.new_intarray(1)
self._isdiscrete = False
self._dtype = "float64"
elif ptype > 1:
ptr_pval_v = _cxtgeo.new_doublearray(1)
ptr_ival_v = _cxtgeo.new_intarray(ntot)
self._isdiscrete = True
self._dtype = "int32"
# number of codes and names
ptr_ccodes_v = _cxtgeo.new_intarray(self._ncodes)
# NB! note the SWIG trick to return modified char values; use cstring.i
# inn the config and %cstring_bounded_output(char *p_codenames_v, NN);
# Then the argument for *p_codevalues_v in C is OMITTED here!
ier, cnames = _cxtgeo.grd3d_imp_prop_roffbin(
pfile,
1,
ptr_type,
ptr_ncol,
ptr_nrow,
ptr_nlay,
ptr_ncodes,
name,
ptr_ival_v,
ptr_pval_v,
ptr_ccodes_v,
0,
XTGDEBUG,
)
if self._isdiscrete:
_gridprop_lowlevel.update_values_from_carray(self,
ptr_ival_v,
np.int32,
delete=True)
else:
_gridprop_lowlevel.update_values_from_carray(self,
ptr_pval_v,
np.float64,
delete=True)
# now make dictionary of codes
if self._isdiscrete:
cnames = cnames.replace(";", "")
cname_list = cnames.split("|")
cname_list.pop() # some rubbish as last entry
ccodes = []
for ino in range(0, self._ncodes):
ccodes.append(_cxtgeo.intarray_getitem(ptr_ccodes_v, ino))
self._codes = dict(zip(ccodes, cname_list))
self._name = name
def _import_segy_xtgeo(sfile,
scanheadermode=False,
scantracemode=False,
outfile=None):
"""Import SEGY via XTGeo's C library. OLD NOT UPDATED!!
Args:
sfile (str): File name of SEGY file
scanheadermode (bool, optional): If true, will scan header
scantracemode (bool, optional): If true, will scan trace headers
outfile (str, optional): Output file for scan dump (default None)
Returns:
A dictionary with relevant data.
"""
# pylint: disable=too-many-statements, too-many-locals
sdata = dict()
logger.info("Import SEGY via XTGeo CLIB")
if outfile is None:
outfile = "/dev/null"
ptr_gn_bitsheader = _cxtgeo.new_intpointer()
ptr_gn_formatcode = _cxtgeo.new_intpointer()
ptr_gf_segyformat = _cxtgeo.new_floatpointer()
ptr_gn_samplespertrace = _cxtgeo.new_intpointer()
ptr_gn_measuresystem = _cxtgeo.new_intpointer()
option = 0
if scantracemode:
option = 0
if scanheadermode:
option = 1
_cxtgeo.cube_scan_segy_hdr(
sfile,
ptr_gn_bitsheader,
ptr_gn_formatcode,
ptr_gf_segyformat,
ptr_gn_samplespertrace,
ptr_gn_measuresystem,
option,
outfile,
)
# get values
gn_bitsheader = _cxtgeo.intpointer_value(ptr_gn_bitsheader)
gn_formatcode = _cxtgeo.intpointer_value(ptr_gn_formatcode)
gf_segyformat = _cxtgeo.floatpointer_value(ptr_gf_segyformat)
gn_samplespertrace = _cxtgeo.intpointer_value(ptr_gn_samplespertrace)
if scanheadermode:
logger.info("Scan SEGY header ... %s bytes ... DONE", gn_bitsheader)
return None
# next is to scan first and last trace, in order to allocate
# cube size
ptr_ncol = _cxtgeo.new_intpointer()
ptr_nrow = _cxtgeo.new_intpointer()
ptr_nlay = _cxtgeo.new_intpointer()
ptr_xori = _cxtgeo.new_doublepointer()
ptr_yori = _cxtgeo.new_doublepointer()
ptr_zori = _cxtgeo.new_doublepointer()
ptr_xinc = _cxtgeo.new_doublepointer()
ptr_yinc = _cxtgeo.new_doublepointer()
ptr_zinc = _cxtgeo.new_doublepointer()
ptr_rotation = _cxtgeo.new_doublepointer()
ptr_minval = _cxtgeo.new_doublepointer()
ptr_maxval = _cxtgeo.new_doublepointer()
ptr_dummy = _cxtgeo.new_floatpointer()
ptr_yflip = _cxtgeo.new_intpointer()
ptr_zflip = _cxtgeo.new_intpointer()
optscan = 1
if scantracemode:
option = 1
logger.debug("Scan via C wrapper...")
_cxtgeo.cube_import_segy(
sfile,
# input
gn_bitsheader,
gn_formatcode,
gf_segyformat,
gn_samplespertrace,
# result (as pointers)
ptr_ncol,
ptr_nrow,
ptr_nlay,
ptr_dummy,
ptr_xori,
ptr_xinc,
ptr_yori,
ptr_yinc,
ptr_zori,
ptr_zinc,
ptr_rotation,
ptr_yflip,
ptr_zflip,
ptr_minval,
ptr_maxval,
# options
optscan,
option,
outfile,
)
logger.debug("Scan via C wrapper... done")
ncol = _cxtgeo.intpointer_value(ptr_ncol)
nrow = _cxtgeo.intpointer_value(ptr_nrow)
nlay = _cxtgeo.intpointer_value(ptr_nlay)
if scantracemode:
return None
nrcl = ncol * nrow * nlay
ptr_cval_v = _cxtgeo.new_floatarray(nrcl)
# next is to do the actual import of the cube
optscan = 0
logger.debug("Import via C wrapper...")
_cxtgeo.cube_import_segy(
sfile,
# input
gn_bitsheader,
gn_formatcode,
gf_segyformat,
gn_samplespertrace,
# result (as pointers)
ptr_ncol,
ptr_nrow,
ptr_nlay,
ptr_cval_v,
ptr_xori,
ptr_xinc,
ptr_yori,
ptr_yinc,
ptr_zori,
ptr_zinc,
ptr_rotation,
ptr_yflip,
ptr_zflip,
ptr_minval,
ptr_maxval,
# options
optscan,
option,
outfile,
)
logger.debug("Import via C wrapper...")
sdata["ncol"] = ncol
sdata["nrow"] = nrow
sdata["nlay"] = nlay
sdata["xori"] = _cxtgeo.doublepointer_value(ptr_xori)
sdata["yori"] = _cxtgeo.doublepointer_value(ptr_yori)
sdata["zori"] = _cxtgeo.doublepointer_value(ptr_zori)
sdata["xinc"] = _cxtgeo.doublepointer_value(ptr_xinc)
sdata["yinc"] = _cxtgeo.doublepointer_value(ptr_yinc)
sdata["zinc"] = _cxtgeo.doublepointer_value(ptr_zinc)
sdata["yflip"] = _cxtgeo.intpointer_value(ptr_yflip)
sdata["zflip"] = _cxtgeo.intpointer_value(ptr_zflip)
sdata["rotation"] = _cxtgeo.doublepointer_value(ptr_rotation)
sdata["minval"] = _cxtgeo.doublepointer_value(ptr_minval)
sdata["maxval"] = _cxtgeo.doublepointer_value(ptr_maxval)
sdata["zmin"] = sdata["zori"]
sdata["zmax"] = sdata["zori"] + sdata["zflip"] * sdata["zinc"] * (nlay - 1)
# the pointer to 1D C array
sdata["cvalues"] = ptr_cval_v
sdata["values"] = None
return sdata
def _get_geometrics_v1(self,
allcells=False,
cellcenter=True,
return_dict=False):
ptr_x = []
for i in range(13):
ptr_x.append(_cxtgeo.new_doublepointer())
option1 = 1
if allcells:
option1 = 0
option2 = 1
if not cellcenter:
option2 = 0
quality = _cxtgeo.grd3d_geometrics(
self._ncol,
self._nrow,
self._nlay,
self._coordsv,
self._zcornsv,
self._actnumsv,
ptr_x[0],
ptr_x[1],
ptr_x[2],
ptr_x[3],
ptr_x[4],
ptr_x[5],
ptr_x[6],
ptr_x[7],
ptr_x[8],
ptr_x[9],
ptr_x[10],
ptr_x[11],
ptr_x[12],
option1,
option2,
)
glist = []
for i in range(13):
glist.append(_cxtgeo.doublepointer_value(ptr_x[i]))
glist.append(quality)
logger.info("Cell geometrics done")
if return_dict:
gdict = {}
gkeys = [
"xori",
"yori",
"zori",
"xmin",
"xmax",
"ymin",
"ymax",
"zmin",
"zmax",
"avg_rotation",
"avg_dx",
"avg_dy",
"avg_dz",
"grid_regularity_flag",
]
for i, key in enumerate(gkeys):
gdict[key] = glist[i]
return gdict
return tuple(glist)
