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)
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)
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())
def plot_erroneous_ne_to_latlon():
import gmtpy
import random
import subprocess
import time
while True:
w, h = 20, 15
gsize = random.uniform(0., 1.) * 4. * 10.**random.uniform(4., 7.)
north_grid, east_grid = num.meshgrid(
num.linspace(-gsize / 2., gsize / 2., 11),
num.linspace(-gsize / 2., gsize / 2., 11))
north_grid = north_grid.flatten()
east_grid = east_grid.flatten()
lat_delta = gsize / earthradius * r2d * 2.
lon = random.uniform(-180., 180.)
lat = random.uniform(-90., 90.)
print(gsize / 1000.)
lat_grid, lon_grid = orthodrome.ne_to_latlon(lat, lon, north_grid,
east_grid)
lat_grid_alt, lon_grid_alt = \
orthodrome.ne_to_latlon_alternative_method(
lat, lon, north_grid, east_grid)
maxerrlat = num.max(num.abs(lat_grid - lat_grid_alt))
maxerrlon = num.max(num.abs(lon_grid - lon_grid_alt))
eps = 1.0e-8
if maxerrlon > eps or maxerrlat > eps:
print(lat, lon, maxerrlat, maxerrlon)
gmt = gmtpy.GMT(
config={
'PLOT_DEGREE_FORMAT': 'ddd.xxxF',
'PAPER_MEDIA': 'Custom_%ix%i' %
(w * gmtpy.cm, h * gmtpy.cm),
'GRID_PEN_PRIMARY': 'thinnest/0/50/0'
})
south = max(-85., lat - 0.5 * lat_delta)
north = min(85., lat + 0.5 * lat_delta)
lon_delta = lat_delta / math.cos(lat * d2r)
delta = lat_delta / 360. * earthradius * 2. * math.pi
scale_km = gmtpy.nice_value(delta / 10.) / 1000.
west = lon - 0.5 * lon_delta
east = lon + 0.5 * lon_delta
x, y = (west, east), (south, north)
xax = gmtpy.Ax(mode='min-max', approx_ticks=4.)
yax = gmtpy.Ax(mode='min-max', approx_ticks=4.)
scaler = gmtpy.ScaleGuru(data_tuples=[(x, y)], axes=(xax, yax))
scaler['R'] = '-Rg'
layout = gmt.default_layout()
mw = 2.5 * gmtpy.cm
layout.set_fixed_margins(mw, mw, mw / gmtpy.golden_ratio,
mw / gmtpy.golden_ratio)
widget = layout.get_widget()
# widget['J'] = ('-JT%g/%g' % (lon, lat)) + '/%(width)gp'
widget['J'] = (
'-JE%g/%g/%g' % (lon, lat, min(lat_delta/2., 180.)))\
+ '/%(width)gp'
aspect = gmtpy.aspect_for_projection(*(widget.J() + scaler.R()))
widget.set_aspect(aspect)
gmt.psbasemap(B='5g5',
L=('x%gp/%gp/%g/%g/%gk' %
(widget.width() / 2., widget.height() / 7., lon,
lat, scale_km)),
*(widget.JXY() + scaler.R()))
gmt.psxy(in_columns=(lon_grid, lat_grid),
S='x10p',
W='1p/200/0/0',
*(widget.JXY() + scaler.R()))
gmt.psxy(in_columns=(lon_grid_alt, lat_grid_alt),
S='c10p',
W='1p/0/0/200',
*(widget.JXY() + scaler.R()))
gmt.save('orthodrome.pdf')
subprocess.call(['xpdf', '-remote', 'ortho', '-reload'])
time.sleep(2)
else:
print('ok', gsize, lat, lon)
layout = gmt.default_layout()
palette_layout = gmtpy.GridLayout(3, 1)
layout.set_widget('center', palette_layout)
widget = palette_layout.get_widget(0, 0)
spacer = palette_layout.get_widget(1, 0)
palette_widget = palette_layout.get_widget(2, 0)
spacer.set_horizontal(0.5 * gmtpy.cm)
palette_widget.set_horizontal(0.5 * gmtpy.cm)
w, h = gmt.page_size_points()
palette_layout.set_policy((w / gmtpy.golden_ratio, 0.), (0., 0.),
aspect=1. / gmtpy.golden_ratio)
xx, yy, zz = to_flat_xyz(x, y, z)
xax = gmtpy.Ax(label='Distance from Yoda')
yax = gmtpy.Ax(label='Height over Yoda')
zax = gmtpy.Ax(snap=True, label='The Force')
guru = gmtpy.ScaleGuru([(xx, yy, zz)], axes=(xax, yax, zax))
grdfile = gmt.tempfilename()
cptfile = gmt.tempfilename()
par = guru.get_params()
inc_interpol = ((par['xmax'] - par['xmin']) /
(widget.width() / gmtpy.inch * 50.),
(par['ymax'] - par['ymin']) /
(widget.height() / gmtpy.inch * 50.))
rxyj = guru.R() + widget.XYJ()
'GRID_PEN_PRIMARY': 'thinnest/0/50/0'
})
south = max(-85., lat - 0.5 * lat_delta)
north = min(85., lat + 0.5 * lat_delta)
lon_delta = lat_delta / math.cos(lat * d2r)
delta = lat_delta / 360. * config.earthradius * 2. * math.pi
scale_km = gmtpy.nice_value(delta / 10.) / 1000.
west = lon - 0.5 * lon_delta
east = lon + 0.5 * lon_delta
x, y = (west, east), (south, north)
xax = gmtpy.Ax(mode='min-max', approx_ticks=4.)
yax = gmtpy.Ax(mode='min-max', approx_ticks=4.)
scaler = gmtpy.ScaleGuru(data_tuples=[(x, y)], axes=(xax, yax))
scaler['R'] = '-Rg'
layout = gmt.default_layout()
mw = 2.5 * gmtpy.cm
layout.set_fixed_margins(mw, mw, mw / gmtpy.golden_ratio,
mw / gmtpy.golden_ratio)
widget = layout.get_widget()
# widget['J'] = ('-JT%g/%g' % (lon, lat)) + '/%(width)gp'
widget['J'] = (
'-JE%g/%g/%g' %
(lon, lat, min(lat_delta / 2., 180.))) + '/%(width)gp'
aspect = gmtpy.aspect_for_projection(*(widget.J() + scaler.R()))
widget.set_aspect(aspect)
