def test_cube_slice_w_ignore_dead_traces_trilinear():
"""Get cube slice trilinear aka Auto4D input, with scrambled data with
dead traces to be ignored, various YFLIP cases."""
cube1 = Cube(XCUB2)
surf1 = RegularSurface()
surf1.from_cube(cube1, 1000.0)
cells = [(18, 12), (20, 2), (0, 4)]
surf1.slice_cube(cube1,
sampling="trilinear",
snapxy=True,
deadtraces=False)
plotfile = ojn(td, "slice_tri1.png")
title = "Cube with dead traces; trilinear; keep as is at dead traces"
surf1.quickplot(filename=plotfile, minmax=(-10000, 10000), title=title)
for cell in cells:
icell, jcell = cell
assert surf1.values[icell,
jcell] == pytest.approx(cube1.values[icell, jcell,
0],
abs=0.1)
assert ma.count_masked(surf1.values) == 0 # shall be no masked cells
def test_get_surface_from_cube(load_cube_rsgy1):
"""Construct a constant surface from cube."""
surf = RegularSurface()
cube = load_cube_rsgy1
surf.from_cube(cube, 1999.0)
assert surf.xinc == cube.xinc
assert surf.nrow == cube.nrow
tsetup.assert_almostequal(surf.values.mean(), 1999.0, 0.00001)
def test_cube_slice_w_ignore_dead_traces_nearest():
"""Get cube slice nearest aka Auto4D input, with scrambled data with
dead traces, various YFLIP cases, ignore dead traces."""
cube1 = Cube(XCUB2)
surf1 = RegularSurface()
surf1.from_cube(cube1, 1000.1)
cells = ((18, 12), (20, 2), (0, 4))
surf1.slice_cube(cube1, deadtraces=False)
plotfile = ojn(td, "slice_nea1.png")
title = "Cube with dead traces; nearest; use just values as is"
surf1.quickplot(filename=plotfile, minmax=(-10000, 10000), title=title)
for cell in cells:
icell, jcell = cell
assert surf1.values[icell,
jcell] == pytest.approx(cube1.values[icell, jcell,
0],
abs=0.01)
assert ma.count_masked(surf1.values) == 0 # shall be no masked cells
# swap surface
surf2 = surf1.copy()
surf2.values = 1000.1
surf2.swapaxes()
surf2.slice_cube(cube1, deadtraces=False)
assert surf2.values.mean() == pytest.approx(surf1.values.mean(), rel=0.001)
# swap surface and cube
surf2 = surf1.copy()
surf2.values = 1000.1
surf2.swapaxes()
cube2 = cube1.copy()
cube2.swapaxes()
surf2.slice_cube(cube2, deadtraces=False)
assert surf2.values.mean() == pytest.approx(surf1.values.mean(), rel=0.001)
# swap cube only
surf2 = surf1.copy()
surf2.values = 1000.1
cube2 = cube1.copy()
cube2.swapaxes()
surf2.slice_cube(cube2, deadtraces=False)
assert surf2.values.mean() == pytest.approx(surf1.values.mean(), rel=0.001)
def test_cube_slice_w_dead_traces_nearest(tmpdir):
"""Get cube slice nearest aka Auto4D input, with scrambled data with
dead traces, various YFLIP cases, undef at dead traces."""
cube1 = Cube(XCUB2)
surf1 = RegularSurface()
surf1.from_cube(cube1, 1000.1)
cells = ((18, 12),)
surf1.slice_cube(cube1, deadtraces=True, algorithm=1)
plotfile = join(tmpdir, "slice_nea1_dead1.png")
title = "Cube with dead traces; nearest; UNDEF at dead traces"
surf1.quickplot(filename=plotfile, minmax=(-10000, 10000), title=title)
for cell in cells:
icell, jcell = cell
assert surf1.values[icell, jcell] == cube1.values[icell, jcell, 0]
ndead = (cube1.traceidcodes == 2).sum()
assert ma.count_masked(surf1.values) == ndead
surf2 = RegularSurface()
surf2.from_cube(cube1, 1000.1)
surf2.slice_cube(cube1, deadtraces=True, algorithm=2)
plotfile = join(tmpdir, "slice_nea1_dead2.png")
title = "Cube with dead traces; nearest; UNDEF at dead traces algo 2"
surf1.quickplot(filename=plotfile, minmax=(-10000, 10000), title=title)
for cell in cells:
icell, jcell = cell
assert surf2.values[icell, jcell] == cube1.values[icell, jcell, 0]
ndead = (cube1.traceidcodes == 2).sum()
assert ma.count_masked(surf1.values) == ndead
def test_cube_slice_w_dead_traces_nearest():
"""Get cube slice nearest aka Auto4D input, with scrambled data with
dead traces, various YFLIP cases, undef at dead traces."""
cube1 = Cube(xcub2)
surf1 = RegularSurface()
surf1.from_cube(cube1, 1000.1)
cells = ((18, 12), )
surf1.slice_cube(cube1, deadtraces=True)
plotfile = ojn(td, 'slice_nea1_dead.png')
title = 'Cube with dead traces; nearest; UNDEF at dead traces'
surf1.quickplot(filename=plotfile, minmax=(-10000, 10000), title=title)
for cell in cells:
icell, jcell = cell
assert surf1.values[icell, jcell] == cube1.values[icell, jcell, 0]
ndead = (cube1.traceidcodes == 2).sum()
print(ndead)
assert ma.count_masked(surf1.values) == ndead
# swap cube only
surf2 = surf1.copy()
surf2.values = 1000.1
cube2 = cube1.copy()
cube2.swapaxes()
surf2.slice_cube(cube2, deadtraces=True)
plotfile = ojn(td, 'slice_nea1_dead_cubeswap.png')
surf2.quickplot(filename=plotfile, minmax=(-10000, 10000))
assert ma.count_masked(surf2.values) == ndead
assert surf2.values.mean() == surf1.values.mean()
