# Copyright (c) 2018, TU Wien, Department of Geodesy and Geoinformation
# All rights reserved.
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# * Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of TU Wien, Department of Geodesy and Geoinformation
# nor the names of its contributors may be used to endorse or promote
# products derived from this software without specific prior written
# permission.
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL TU WIEN, DEPARTMENT OF GEODESY AND
# GEOINFORMATION BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
# OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
# OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
import warnings
try:
import matplotlib.pyplot as plt
import matplotlib as mp
except ImportError:
warnings.warn("Matplotlib is necessary for plotting grids.")
try:
from mpl_toolkits.basemap import Basemap
except ImportError:
warnings.warn("Basemap is necessary for plotting grids.")
import numpy as np
import pygeogrids.grids as grids
[docs]def plot_cell_grid_partitioning(
output, cellsize_lon=5.0, cellsize_lat=5.0, figsize=(12, 6)
):
"""
Plot an overview of a global cell partitioning.
Parameters
----------
output: string
output file name
"""
mp.rcParams["font.size"] = 10
mp.rcParams["text.usetex"] = True
plt.figure(figsize=figsize, dpi=300)
ax = plt.axes([0, 0, 1, 1])
map = Basemap(
projection="cyl",
llcrnrlat=-90,
urcrnrlat=90,
llcrnrlon=-180,
urcrnrlon=180,
ax=ax,
)
map.drawparallels(
np.arange(-90, 90, cellsize_lat), labels=[1, 0, 0, 0], linewidth=0.5
)
map.drawmeridians(
np.arange(-180, 180, cellsize_lon),
labels=[0, 0, 0, 1],
rotation="vertical",
linewidth=0.5,
)
# fill continents 'coral' (with zorder=0), color wet areas 'aqua'
map.drawmapboundary(fill_color="aqua")
map.fillcontinents(color="0.6", lake_color="aqua")
label_lats = np.arange(-90 + cellsize_lat / 2.0, 90, cellsize_lat)
label_lons = np.arange(-180 + cellsize_lon / 2.0, 180, cellsize_lon)
lons, lats = np.meshgrid(label_lons, label_lats)
x, y = map(lons.flatten(), lats.flatten())
cells = grids.lonlat2cell(
lons.flatten(),
lats.flatten(),
cellsize_lon=cellsize_lon,
cellsize_lat=cellsize_lat,
)
for xt, yt, cell in zip(x, y, cells):
plt.text(
xt,
yt,
"{:}".format(cell),
fontsize=4,
va="center",
ha="center",
weight="bold",
)
plt.savefig(output, format="png", dpi=300)
plt.close()