def test_basic(self):
for version in gmtpy.all_installed_gmt_versions():
width = 8.0 * inch
height = 9.0 * inch
resolution = 72
gmt = gmtpy.GMT(version=version, config_papersize=(width, height))
gmt.pscoast(X=0,
Y=0,
R='g',
J='E32/30/170/8i',
B='10g10',
D='c',
A=10000,
S=(114, 159, 207),
G=(233, 185, 110),
W='thinnest')
gmt.dump('test')
gmt.load('test')
for oversample in (1, 2):
fname = 'gmtpy_test_basic_o%i.png' % oversample
fpath = self.fpath(fname)
gmt.save(fpath, resolution=resolution, oversample=oversample)
self.compare_with_ref(fname, 0.03)
img = image.imread(fpath, format='png')
self.assertEqual(img.shape,
(int(round(resolution * height / inch)),
int(round(resolution * width / inch)), 3))
def test_override_args(self):
x = num.array([0, 0.5, 1, 0])
y = num.array([0, 1, 0, 0])
width = 300
height = 100
config_papersize = (width, height)
for version in gmtpy.all_installed_gmt_versions():
gmt = gmtpy.GMT(
version=version,
config_papersize=config_papersize)
for i, cutoff in enumerate([30, 90]):
gmt.psxy(
in_columns=(i*2+x, y),
W='10p,red',
J='X%gp/%gp' % (width, height),
X=0,
Y=0,
R=(-1, 4, -1, 2),
config={
'PS_MITER_LIMIT': '%i' % cutoff})
fname = 'gmtpy_test_override.png'
fpath = self.fpath(fname)
gmt.save(fpath)
self.compare_with_ref(fname, 0.001, show=False)
def test_grid_layout(self):
for version in gmtpy.all_installed_gmt_versions():
gmt = gmtpy.GMT(version=version, config_papersize='a3')
nx, ny = 2, 5
grid = gmtpy.GridLayout(nx, ny)
layout = gmt.default_layout()
layout.set_widget('center', grid)
widgets = []
for iy in range(ny):
for ix in range(nx):
inner = gmtpy.FrameLayout()
inner.set_fixed_margins(
1.*cm*golden_ratio, 1.*cm*golden_ratio, 1.*cm, 1.*cm)
grid.set_widget(ix, iy, inner)
inner.set_vertical(0, (iy+1.))
widgets.append(inner.get_widget('center'))
gmt.draw_layout(layout)
for widget in widgets:
x = num.linspace(0., 10., 5)
y = num.sin(x)
xax = gmtpy.Ax(approx_ticks=4, snap=True)
yax = gmtpy.Ax(approx_ticks=4, snap=True)
guru = gmtpy.ScaleGuru([(x, y)], axes=(xax, yax))
gmt.psbasemap(*(widget.JXY() + guru.RB(ax_projection=True)))
gmt.psxy(in_columns=(x, y), *(widget.JXY() + guru.R()))
fname = 'gmtpy_test_grid_layout.png'
fpath = self.fpath(fname)
gmt.save(fpath, resolution=75)
self.compare_with_ref(fname, 0.01)
def test_basic2(self):
for version in gmtpy.all_installed_gmt_versions():
if version.startswith('5'):
gmt = gmtpy.GMT(
version=version,
config={'MAP_FRAME_TYPE': 'fancy'},
eps_mode=True)
else:
gmt = gmtpy.GMT(
version=version,
config={'BASEMAP_TYPE': 'fancy'})
layout = gmt.default_layout()
widget = layout.get_widget()
xax = gmtpy.Ax(label='Lon', mode='min-max')
yax = gmtpy.Ax(label='Lat', mode='min-max')
scaler = gmtpy.ScaleGuru([([5, 15], [52, 58])], axes=(xax, yax))
par = scaler.get_params()
lon0 = (par['xmin'] + par['xmax'])/2.
lat0 = (par['ymin'] + par['ymax'])/2.
sll = '%g/%g' % (lon0, lat0)
widget['J'] = '-JM' + sll + '/%(width)gp'
widget['J'] = '-JM' + sll + '/%(width)gp'
scaler['B'] = \
'-B%(xinc)gg%(xinc)g:%(xlabel)s:' \
'/%(yinc)gg%(yinc)g:%(ylabel)s:WSen'
aspect = gmtpy.aspect_for_projection(
version, *(widget.J() + scaler.R()))
aspect = 1.045
widget.set_aspect(aspect)
gmt.pscoast(D='h', W='1p,red', *(widget.JXY() + scaler.R()))
gmt.psbasemap(*(widget.JXY() + scaler.BR()))
fname = 'gmtpy_test_basic2.png'
fpath = self.fpath(fname)
gmt.save(fpath, resolution=75, bbox=layout.bbox())
self.compare_with_ref(fname, 0.01, show=False)
def plotMap(self, outfile, **kwargs):
from pyrocko.plot import gmtpy
lats = self.lats()
lons = self.lons()
s, n, w, e = (lats.min(), lats.max(), lons.min(), lons.max())
def darken(c, f=0.7):
return (c[0] * f, c[1] * f, c[2] * f)
gmt = gmtpy.GMT()
gmt.psbasemap(B='40/20', J='M0/12', R='%f/%f/%f/%f' % (w, e, s, n))
gmt.pscoast(R=True, J=True, D='i', S='216/242/254', A=10000, W='.2p')
gmt.psxy(R=True, J=True, in_columns=[lons, lats], S='c2p', G='black')
gmt.save(outfile)
def test_layout(self):
x = num.linspace(0., math.pi * 6, 1001)
y1 = num.sin(x) * 1e-9
y2 = 2.0 * num.cos(x) * 1e-9
xax = gmtpy.Ax(label='Time', unit='s')
yax = gmtpy.Ax(label='Amplitude',
unit='m',
scaled_unit='nm',
scaled_unit_factor=1e9,
approx_ticks=5,
space=0.05)
guru = gmtpy.ScaleGuru([(x, y1), (x, y2)], axes=(xax, yax))
for version in gmtpy.all_installed_gmt_versions():
width = 8 * inch
height = 3 * inch
gmt = gmtpy.GMT(version=version, config_papersize=(width, height))
layout = gmt.default_layout()
widget = layout.get_widget()
gmt.draw_layout(layout)
gmt.psbasemap(*(widget.JXY() + guru.RB(ax_projection=True)))
gmt.psxy(in_columns=(x, y1),
W='1p,red',
*(widget.JXY() + guru.R()))
gmt.psxy(in_columns=(x, y2),
W='1p,blue',
*(widget.JXY() + guru.R()))
fname = 'gmtpy_test_layout.png'
fpath = self.fpath(fname)
gmt.save(fpath)
self.compare_with_ref(fname, 0.01)
