def test_adjoint_inject_interpolate(self, shape, coords, npoints=19):
"""
Verify that p.inject is the adjoint of p.interpolate for a
devito SparseFunction p
"""
a = unit_box(shape=shape)
a.data[:] = 0.
c = unit_box(shape=shape, name='c', grid=a.grid)
c.data[:] = 27.
assert a.grid == c.grid
# Inject receiver
p = points(a.grid, ranges=coords, npoints=npoints)
p.data[:] = 1.2
expr = p.inject(field=a, expr=p)
# Read receiver
p2 = points(a.grid, name='points2', ranges=coords, npoints=npoints)
expr2 = p2.interpolate(expr=c)
Operator(expr + expr2)(a=a, c=c)
# < P x, y > - < x, P^T y>
# Px => p2
# y => p
# x => c
# P^T y => a
term1 = np.dot(p2.data.reshape(-1), p.data.reshape(-1))
term2 = np.dot(c.data.reshape(-1), a.data.reshape(-1))
assert np.isclose((term1-term2) / term1, 0., atol=1.e-6)
def test_adjoint_inject_interpolate(self, shape, coords, npoints=19):
"""
Verify that p.inject is the adjoint of p.interpolate for a
devito SparseFunction p
"""
a = unit_box(shape=shape)
a.data[:] = 0.
c = unit_box(shape=shape, name='c', grid=a.grid)
c.data[:] = 27.
assert a.grid == c.grid
# Inject receiver
p = points(a.grid, ranges=coords, npoints=npoints)
p.data[:] = 1.2
expr = p.inject(field=a, expr=p)
# Read receiver
p2 = points(a.grid, name='points2', ranges=coords, npoints=npoints)
expr2 = p2.interpolate(expr=c)
Operator(expr + expr2)(a=a, c=c)
# < P x, y > - < x, P^T y>
# Px => p2
# y => p
# x => c
# P^T y => a
term1 = np.dot(p2.data.reshape(-1), p.data.reshape(-1))
term2 = np.dot(c.data.reshape(-1), a.data.reshape(-1))
assert np.isclose((term1-term2) / term1, 0., atol=1.e-6)
def test_inject_array(shape, coords, result, npoints=19):
"""Test point injection with a set of points forming a line
through the middle of the grid.
"""
a = unit_box(shape=shape)
a.data[:] = 0.
p = points(a.grid, ranges=coords, npoints=npoints)
p2 = points(a.grid, ranges=coords, npoints=npoints, name='p2')
p2.data[:] = 1.
expr = p.inject(a, p)
Operator(expr)(a=a, points=p2.data[:])
indices = [slice(4, 6, 1) for _ in coords]
indices[0] = slice(1, -1, 1)
assert np.allclose(a.data[indices], result, rtol=1.e-5)
def test_inject_array(shape, coords, result, npoints=19):
"""Test point injection with a set of points forming a line
through the middle of the grid.
"""
a = unit_box(shape=shape)
a.data[:] = 0.
p = points(a.grid, ranges=coords, npoints=npoints)
p2 = points(a.grid, ranges=coords, npoints=npoints, name='p2')
p2.data[:] = 1.
expr = p.inject(a, p)
Operator(expr)(a=a, points=p2.data[:])
indices = [slice(4, 6, 1) for _ in coords]
indices[0] = slice(1, -1, 1)
assert np.allclose(a.data[indices], result, rtol=1.e-5)
def test_interpolate_array(shape, coords, npoints=20):
"""Test generic point interpolation testing the x-coordinate of an
abitrary set of points going across the grid.
"""
a = unit_box(shape=shape)
p = points(a.grid, coords, npoints=npoints)
xcoords = p.coordinates.data[:, 0]
expr = p.interpolate(a)
Operator(expr)(a=a, points=p.data[:])
assert np.allclose(p.data[:], xcoords, rtol=1e-6)
def test_interpolate_array(shape, coords, npoints=20):
"""Test generic point interpolation testing the x-coordinate of an
abitrary set of points going across the grid.
"""
a = unit_box(shape=shape)
p = points(a.grid, coords, npoints=npoints)
xcoords = p.coordinates.data[:, 0]
expr = p.interpolate(a)
Operator(expr)(a=a, points=p.data[:])
assert np.allclose(p.data[:], xcoords, rtol=1e-6)
def test_inject_from_field(shape, coords, result, npoints=19):
"""Test point injection from a second field along a line
through the middle of the grid.
"""
a = unit_box(shape=shape)
a.data[:] = 0.
b = Function(name='b', grid=a.grid)
b.data[:] = 1.
p = points(a.grid, ranges=coords, npoints=npoints)
expr = p.inject(field=a, expr=b)
Operator(expr)(a=a, b=b)
indices = [slice(4, 6, 1) for _ in coords]
indices[0] = slice(1, -1, 1)
assert np.allclose(a.data[indices], result, rtol=1.e-5)
def test_inject_from_field(shape, coords, result, npoints=19):
"""Test point injection from a second field along a line
through the middle of the grid.
"""
a = unit_box(shape=shape)
a.data[:] = 0.
b = Function(name='b', grid=a.grid)
b.data[:] = 1.
p = points(a.grid, ranges=coords, npoints=npoints)
expr = p.inject(field=a, expr=b)
Operator(expr)(a=a, b=b)
indices = [slice(4, 6, 1) for _ in coords]
indices[0] = slice(1, -1, 1)
assert np.allclose(a.data[indices], result, rtol=1.e-5)
