def test_merca_map():
grid = salem.grids.local_mercator_grid(center_ll=(11.38, 47.26),
extent=(2000000, 2000000))
m1 = Map(grid)
grid = salem.grids.local_mercator_grid(center_ll=(11.38, 47.26),
extent=(2000000, 2000000),
order='ul')
m2 = Map(grid)
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(12, 6))
m1.visualize(ax=ax1, addcbar=False)
m2.visualize(ax=ax2, addcbar=False)
plt.tight_layout()
m1.set_lonlat_contours(add_tick_labels=False)
m1.visualize()
def test_oceans():
f = os.path.join(get_demo_file('wrf_tip_d1.nc'))
grid = GeoNetcdf(f).grid
m = Map(grid, countries=False)
m.set_shapefile(rivers=True, linewidths=2)
m.set_shapefile(oceans=True, edgecolor='k', linewidth=3)
fig, ax = plt.subplots(1, 1)
m.visualize(ax=ax, addcbar=False)
plt.tight_layout()
def test_gmap():
g = GoogleCenterMap(center_ll=(10.762660, 46.794221), zoom=13,
size_x=640, size_y=640)
m = Map(g.grid, countries=False, nx=640)
m.set_lonlat_countours(interval=0.025)
m.set_shapefile(get_demo_file('Hintereisferner.shp'),
linewidths=2, edgecolor='darkred')
m.set_rgb(g.get_vardata())
m.visualize(addcbar=False)
dem = salem.GeoTiff(get_demo_file('hef_srtm.tif'))
dem.set_subset(margin=-100)
dem = salem.grids.local_mercator_grid(center_ll=(10.76, 46.798444),
extent=(10000, 7000))
i, j = dem.ij_coordinates
g = GoogleVisibleMap(x=i, y=j, src=dem, size_x=500, size_y=400)
img = g.get_vardata()
m = Map(dem, countries=False)
assert_raises(ValueError, m.set_data, img)
m.set_lonlat_countours(interval=0.025)
m.set_shapefile(get_demo_file('Hintereisferner.shp'),
linewidths=2, edgecolor='darkred')
m.set_rgb(img, g.grid)
m.visualize(addcbar=False)
def test_hef():
grid = salem.grids.local_mercator_grid(center_ll=(10.76, 46.798444),
extent=(10000, 7000))
c = Map(grid, countries=False)
c.set_lonlat_countours(interval=10)
c.set_shapefile(get_demo_file('Hintereisferner_UTM.shp'))
c.set_topography(get_demo_file('hef_srtm.tif'),
interp='linear')
c.visualize(addcbar=False, title='linear')
c.set_topography(get_demo_file('hef_srtm.tif'),
interp='spline', ks=2)
c.visualize(addcbar=False, title='spline deg 2')
c.set_topography(get_demo_file('hef_srtm.tif'))
c.visualize(addcbar=False, title='Default: spline deg 3')
h = c.set_topography(get_demo_file('hef_srtm.tif'),
interp='spline', ks=5)
c.visualize(addcbar=False, title='spline deg 5')
dem = salem.GeoTiff(get_demo_file('hef_srtm.tif'))
mytopo = dem.get_vardata()
c.set_topography(mytopo, crs=dem.grid, interp='spline')
c.visualize(addcbar=False, title='From array')
c.set_lonlat_countours()
c.set_cmap(cleo.get_cm('topo'))
c.set_plot_params(nlevels=256)
c.set_data(h)
c.visualize()
def test_geometries():
# UL Corner
g = Grid(nxny=(5, 4), dxdy=(10, 10), ll_corner=(-20, -15), proj=wgs84,
pixel_ref='corner')
c = Map(g, ny=4)
c.set_lonlat_countours(interval=10., colors='crimson')
c.set_geometry(shpg.Point(10, 10), color='darkred', markersize=60)
c.set_geometry(shpg.Point(5, 5), s=500, marker='s',
facecolor='green', hatch='||||')
s = np.array([(-5, -10), (0., -5), (-5, 0.), (-10, -5)])
l1 = shpg.LineString(s)
l2 = shpg.LinearRing(s+3)
c.set_geometry(l1)
c.set_geometry(l2, color='pink', linewidth=3)
s += 20
p = shpg.Polygon(shpg.LineString(s), [shpg.LineString(s/4 + 10)])
c.set_geometry(p, facecolor='red', edgecolor='k', linewidth=3, alpha=0.5)
p1 = shpg.Point(20, 10)
p2 = shpg.Point(20, 20)
p3 = shpg.Point(10, 20)
mpoints = shpg.MultiPoint([p1, p2, p3])
c.set_geometry(mpoints, s=250, marker='s',
c='purple', hatch='||||')
c.visualize(addcbar=False)
c.set_geometry()
assert_true(len(c._geometries) == 0)
def test_simple_map():
a = np.zeros((4, 5))
a[0, 0] = -1
a[1, 1] = 1.1
a[2, 2] = 2.2
a[2, 4] = 1.9
a[3, 3] = 9
a_inv = a[::-1, :]
fs = _create_dummy_shp('fs.shp')
# UL Corner
g1 = Grid(nxny=(5, 4), dxdy=(1, -1), ul_corner=(-1, 3), proj=wgs84,
pixel_ref='corner')
c1 = Map(g1, ny=4, countries=False)
# LL Corner
g2 = Grid(nxny=(5, 4), dxdy=(1, 1), ll_corner=(-1, -1), proj=wgs84,
pixel_ref='corner')
c2 = Map(g2, ny=4, countries=False)
# Settings
for c, data in zip([c1, c2], [a_inv, a]):
c.set_cmap(mpl.cm.get_cmap('jet'))
c.set_plot_params(levels=[0, 1, 2, 3])
c.set_data(data)
c.set_shapefile(fs)
c.set_lonlat_countours(interval=0.5)
fig = plt.figure(figsize=(9, 8))
ax1 = fig.add_subplot(321)
ax2 = fig.add_subplot(322)
c1.visualize(ax1)
c2.visualize(ax2)
# UL Corner
c1 = Map(g1, ny=400, countries=False)
c2 = Map(g2, ny=400, countries=False)
# Settings
for c, data, g in zip([c1, c2], [a_inv, a], [g1, g2]):
c.set_cmap(mpl.cm.get_cmap('jet'))
c.set_data(data, crs=g)
c.set_shapefile(fs)
c.set_plot_params(nlevels=256)
c.set_lonlat_countours(interval=2)
ax1 = fig.add_subplot(323)
ax2 = fig.add_subplot(324)
c1.visualize(ax1)
c2.visualize(ax2)
# Settings
for c, data in zip([c1, c2], [a_inv, a]):
c.set_plot_params(nlevels=256, vmax=3)
c.set_lonlat_countours(interval=1)
c.set_data(data, interp='linear')
ax1 = fig.add_subplot(325)
ax2 = fig.add_subplot(326)
c1.visualize(ax1)
c2.visualize(ax2)
fig.tight_layout()
if os.path.exists(testdir):
shutil.rmtree(testdir)
def test_text():
# UL Corner
g = Grid(nxny=(5, 4), dxdy=(10, 10), ll_corner=(-20, -15), proj=wgs84,
pixel_ref='corner')
c = Map(g, ny=4, countries=False)
c.set_lonlat_contours(interval=5., colors='crimson')
c.set_text(-5, -5, 'Less Middle', color='green', style='italic', size=25)
c.set_geometry(shpg.Point(-10, -10), s=500, marker='o',
text='My point', text_delta=[0, 0])
shape = salem.utils.read_shapefile_to_grid(files['world_borders'], c.grid)
had_c = set()
for index, row in shape.iloc[::-1].iterrows():
if row.CNTRY_NAME in had_c:
c.set_geometry(row.geometry, crs=c.grid)
else:
c.set_geometry(row.geometry, text=row.CNTRY_NAME, crs=c.grid,
text_kwargs=dict(horizontalalignment='center',
verticalalignment='center', clip_on=True,
color='gray'), text_delta=[0, 0])
had_c.add(row.CNTRY_NAME)
c.set_points([20, 20, 10], [10, 20, 20], s=250, marker='s',
c='purple', hatch='||||', text='baaaaad', text_delta=[0, 0],
text_kwargs=dict(horizontalalignment='center',
verticalalignment='center', color='red'))
c.visualize(addcbar=False)
c.set_text()
assert_true(len(c._text) == 0)
c.set_geometry()
assert_true(len(c._geometries) == 0)
def test_contourf():
a = np.zeros((4, 5))
a[0, 0] = -1
a[1, 1] = 1.1
a[2, 2] = 2.2
a[2, 4] = 1.9
a[3, 3] = 9
s = a * 0.
s[2, 2] = 1
# UL Corner
g = Grid(nxny=(5, 4), dxdy=(1, -1), ul_corner=(-1, 3), proj=wgs84,
pixel_ref='corner')
c = Map(g, ny=400, countries=False)
c.set_cmap(mpl.cm.get_cmap('viridis'))
c.set_plot_params(levels=[0, 1, 2, 3])
c.set_data(a)
c.set_contourf(s, interp='linear', hatches=['xxx'], colors='none',
levels=[0.5, 1.5])
c.set_lonlat_contours(interval=0.5)
c.visualize()
# remove it
c.set_contourf()
c.visualize()
def test_map(self):
a = np.zeros((4, 5))
a[0, 0] = -1
a[1, 1] = 1.1
a[2, 2] = 2.2
a[2, 4] = 1.9
a[3, 3] = 9
cmap = copy.deepcopy(mpl.cm.get_cmap('jet'))
# ll_corner (type geotiff)
g = Grid(nxny=(5, 4), dxdy=(1, 1), ll_corner=(0, 0), proj=wgs84,
pixel_ref='corner')
c = Map(g, ny=4, countries=False)
c.set_cmap(cmap)
c.set_plot_params(levels=[0, 1, 2, 3])
c.set_data(a)
rgb1 = c.to_rgb()
c.set_data(a, crs=g)
assert_array_equal(rgb1, c.to_rgb())
c.set_data(a, interp='linear')
rgb1 = c.to_rgb()
c.set_data(a, crs=g, interp='linear')
assert_array_equal(rgb1, c.to_rgb())
# centergrid (type WRF)
g = Grid(nxny=(5, 4), dxdy=(1, 1), ll_corner=(0.5, 0.5), proj=wgs84,
pixel_ref='center')
c = Map(g, ny=4, countries=False)
c.set_cmap(cmap)
c.set_plot_params(levels=[0, 1, 2, 3])
c.set_data(a)
rgb1 = c.to_rgb()
c.set_data(a, crs=g)
assert_array_equal(rgb1, c.to_rgb())
c.set_data(a, interp='linear')
rgb1 = c.to_rgb()
c.set_data(a, crs=g.corner_grid, interp='linear')
assert_array_equal(rgb1, c.to_rgb())
c.set_data(a, crs=g.center_grid, interp='linear')
assert_array_equal(rgb1, c.to_rgb())
# More pixels
c = Map(g, ny=500, countries=False)
c.set_cmap(cmap)
c.set_plot_params(levels=[0, 1, 2, 3])
c.set_data(a)
rgb1 = c.to_rgb()
c.set_data(a, crs=g)
assert_array_equal(rgb1, c.to_rgb())
c.set_data(a, interp='linear')
rgb1 = c.to_rgb()
c.set_data(a, crs=g, interp='linear')
rgb2 = c.to_rgb()
# The interpolation is conservative with the grid...
srgb = np.sum(rgb2[..., 0:3], axis=2)
pok = np.nonzero(srgb != srgb[0, 0])
rgb1 = rgb1[np.min(pok[0]):np.max(pok[0]),
np.min(pok[1]):np.max(pok[1]),...]
rgb2 = rgb2[np.min(pok[0]):np.max(pok[0]),
np.min(pok[1]):np.max(pok[1]),...]
assert_array_equal(rgb1, rgb2)
cmap.set_bad('pink')
# Add masked arrays
a[1, 1] = np.NaN
c.set_data(a)
rgb1 = c.to_rgb()
c.set_data(a, crs=g)
assert_array_equal(rgb1, c.to_rgb())
# Interp?
c.set_data(a, interp='linear')
rgb1 = c.to_rgb()
c.set_data(a, crs=g, interp='linear')
rgb2 = c.to_rgb()
