def test_tf():
"polyprism.tf against prism"
resprism = prism.tf(xp, yp, zp, prismmodel, inc, dec)
respoly = polyprism.tf(xp, yp, zp, model, inc, dec)
diff = np.abs(resprism - respoly)
errormsg = 'max diff: %g | max polyprism: %g | max prism: %g' % (
max(diff), max(respoly), max(resprism))
assert np.all(diff <= precision_mag), errormsg
def test_tf():
"polyprism.tf against prism"
resprism = prism.tf(xp, yp, zp, prismmodel, inc, dec)
respoly = polyprism.tf(xp, yp, zp, model, inc, dec)
diff = np.abs(resprism - respoly)
errormsg = 'max diff: %g | max polyprism: %g | max prism: %g' % (
max(diff), max(respoly), max(resprism))
assert np.all(diff <= precision_mag), errormsg
# The regional field
inc, dec = 30, -15
# Draw each polygonal prism (viewed from above)
bounds = [-5000, 5000, -5000, 5000, 0, 5000]
area = bounds[:4]
axis = mpl.figure().gca()
mpl.axis('scaled')
model = [
mesher.PolygonalPrism(
mpl.draw_polygon(area, axis, xy2ne=True),
# Use only induced magnetization
0, 2000, {'magnetization':2})]
# Calculate the effect
shape = (100, 100)
xp, yp, zp = gridder.regular(area, shape, z=-500)
tf = polyprism.tf(xp, yp, zp, model, inc, dec)
# and plot it
mpl.figure()
mpl.axis('scaled')
mpl.title("Total field anomalyproduced by prism model (nT)")
mpl.contourf(yp, xp, tf, shape, 20)
mpl.colorbar()
for p in model:
mpl.polygon(p, '.-k', xy2ne=True)
mpl.set_area(area)
mpl.m2km()
mpl.show()
# Show the prisms
myv.figure()
myv.polyprisms(model, 'magnetization')
myv.axes(myv.outline(bounds), ranges=[i*0.001 for i in bounds])
# The regional field
inc, dec = 30, -15
# Draw each polygonal prism (viewed from above)
bounds = [-5000, 5000, -5000, 5000, 0, 5000]
area = bounds[:4]
axis = mpl.figure().gca()
mpl.axis('scaled')
model = [
mesher.PolygonalPrism(
mpl.draw_polygon(area, axis, xy2ne=True),
# Use only induced magnetization
0, 2000, {'magnetization': 2})]
# Calculate the effect
shape = (100, 100)
xp, yp, zp = gridder.regular(area, shape, z=-500)
tf = polyprism.tf(xp, yp, zp, model, inc, dec)
# and plot it
mpl.figure()
mpl.axis('scaled')
mpl.title("Total field anomalyproduced by prism model (nT)")
mpl.contourf(yp, xp, tf, shape, 20)
mpl.colorbar()
for p in model:
mpl.polygon(p, '.-k', xy2ne=True)
mpl.set_area(area)
mpl.m2km()
mpl.show()
# Show the prisms
myv.figure()
myv.polyprisms(model, 'magnetization')
myv.axes(myv.outline(bounds), ranges=[i * 0.001 for i in bounds])
def test_tf_numpy():
"polyprism.tf cython vs numpy implementation"
cy = polyprism.tf(xp, yp, zp, model, inc, dec)
py = _polyprism_numpy.tf(xp, yp, zp, model, inc, dec)
diff = np.abs(py - cy)
assert np.all(diff <= precision_mag), 'max diff: %g' % (max(diff))
def test_tf_numpy():
"polyprism.tf cython vs numpy implementation"
cy = polyprism.tf(xp, yp, zp, model, inc, dec)
py = _polyprism_numpy.tf(xp, yp, zp, model, inc, dec)
diff = np.abs(py - cy)
assert np.all(diff <= precision_mag), 'max diff: %g' % (max(diff))
