def contourf(self, var="depth", it=None, tiles=False, **kwargs):
x = kwargs.get("x", self._obj.SCHISM_hgrid_node_x[:].values)
y = kwargs.get("y", self._obj.SCHISM_hgrid_node_y[:].values)
try:
t = kwargs.get("t", self._obj.time.values)
except:
pass
tes = kwargs.get("tri3",
self._obj.SCHISM_hgrid_face_nodes.values[:, :4])
# sort out quads
try:
mask = np.isnan(tes)[:, 3]
tr3 = tes[mask][:, :3]
tr3_ = quads_to_tris(tes[~mask])
if tr3_:
tri3 = np.append(tr3, tr3_, axis=0).astype(int)
else:
tri3 = tr3.astype(int)
except:
tri3 = tes.astype(int)
if tri3.min() > 0:
tri3 = tri3 - 1
z = kwargs.get("z", self._obj[var].values[it, :].flatten())
nodes = pd.DataFrame({
"longitude": x,
"latitude": y,
"{}".format(var): z
})
elems = pd.DataFrame(tri3, columns=["a", "b", "c"])
width = kwargs.get("width", 800)
height = kwargs.get("height", 600)
opts.defaults(opts.WMTS(width=width, height=height))
tile = gv.WMTS("https://b.tile.openstreetmap.org/{Z}/{X}/{Y}.png")
points = gv.operation.project_points(
gv.Points(nodes, vdims=["{}".format(var)]))
trimesh = gv.TriMesh((elems, points))
if tiles:
return tile * rasterize(trimesh, aggregator="mean").opts(
colorbar=True, cmap="Viridis", padding=0.1, tools=["hover"])
else:
return rasterize(trimesh, aggregator="mean").opts(
colorbar=True,
cmap="Viridis",
padding=0.1,
tools=["hover"],
width=width,
height=height,
)
def frames(self, var="depth", tiles=False, **kwargs):
x = kwargs.get("x", self._obj.SCHISM_hgrid_node_x[:].values)
y = kwargs.get("y", self._obj.SCHISM_hgrid_node_y[:].values)
try:
t = kwargs.get("t", self._obj.time.values)
except:
pass
tes = kwargs.get("tri3",
self._obj.SCHISM_hgrid_face_nodes.values[:, :4])
# sort out quads
try:
mask = np.isnan(tes)[:, 3]
tr3 = tes[mask][:, :3]
tr3_ = quads_to_tris(tes[~mask])
if tr3_:
tri3 = np.append(tr3, tr3_, axis=0).astype(int)
else:
tri3 = tr3.astype(int)
except:
tri3 = tes.astype(int)
times = kwargs.get("times", self._obj.time.values)
z = kwargs.get("z", self._obj[var].values[0, :].flatten())
zmin = self._obj[var].values.min()
zmax = self._obj[var].values.max()
nodes = pd.DataFrame({
"longitude": x,
"latitude": y,
"{}".format(var): z
})
elems = pd.DataFrame(tri3, columns=["a", "b", "c"])
width = kwargs.get("width", 800)
height = kwargs.get("height", 600)
opts.defaults(opts.WMTS(width=width, height=height))
tile = gv.WMTS("https://b.tile.openstreetmap.org/{Z}/{X}/{Y}.png")
points = gv.operation.project_points(
gv.Points(nodes, vdims=["{}".format(var)]))
trimesh = gv.TriMesh((elems, points))
def time_mesh(time):
points.data[var] = self._obj[var].sel(time=time).values
return gv.TriMesh((elems, points)) # , crs=ccrs.GOOGLE_MERCATOR)
meshes = hv.DynamicMap(time_mesh,
kdims=["Time"]).redim.values(Time=times)
imesh = rasterize(meshes, aggregator="mean").opts(cmap="viridis",
colorbar=True,
padding=0.1,
tools=["hover"],
clim=(zmin, zmax))
if tiles:
return hv.output(tile * imesh, holomap="scrubber", fps=1)
else:
return hv.output(imesh.opts(width=width, height=height),
holomap="scrubber",
fps=1)
def frames(self, **kwargs):
cr = kwargs.get('coastlines', None)
c_attrs = kwargs.get('coastlines_attrs', {})
x = kwargs.get('x', self._obj.SCHISM_hgrid_node_x[:].values)
y = kwargs.get('y', self._obj.SCHISM_hgrid_node_y[:].values)
t = kwargs.get('t', self._obj.time.values)
tes = kwargs.get('tes',
self._obj.SCHISM_hgrid_face_nodes.values[:, :4])
# sort out quads
try:
mask = np.isnan(tes)[:, 3]
tr3 = tes[mask][:, :3]
tr3_ = quads_to_tris(tes[~mask])
if tr3_:
tri3 = np.append(tr3, tr3_, axis=0).astype(int)
else:
tri3 = tr3.astype(int)
except:
tri3 = tes.astype(int)
var = kwargs.get('var', 'depth')
z = kwargs.get('z', self._obj[var].values)
# set figure size
xr = x.max() - x.min()
yr = y.max() - y.min()
ratio = yr / xr
xf = 12
yf = np.ceil(12 * ratio).astype(int)
fig = plt.figure(figsize=(xf, yf))
vmin = kwargs.get('vmin', z.min())
vmax = kwargs.get('vmax', z.max())
nv = kwargs.get('nv', 10)
title = kwargs.get('title', None)
vrange = np.linspace(vmin, vmax, nv, endpoint=True)
#optional mask for the data
mask = kwargs.get('mask', None)
if 'mask' in kwargs:
z = np.ma.masked_array(z, mask)
z = z.filled(fill_value=-99999)
## CHOOSE YOUR PROJECTION
# ax = plt.axes(projection=ccrs.Orthographic(grid_x.mean(), grid_y.mean()))
# ax = plt.axes(projection=ccrs.PlateCarree())
# ax.background_patch.set_facecolor('k')
# Limit the extent of the map to a small longitude/latitude range.
# ax.set_extent([x.min(), x.max(), y.min(), y.max()])
ax = plt.axes()
ax.set_aspect('equal')
ims = []
for i in range(len(t)):
im = ax.tricontourf(x,
y,
tri3,
z[i, :],
vrange,
vmin=vmin,
vmax=vmax) #, transform=ccrs.PlateCarree())
# im = ax.contourf(x,y,z[i,:,:],v,vmin=v1,vmax=v2,latlon=True)
add_arts = im.collections
text = 'time={}'.format(t[i])
#te = ax.text(90, 90, text)
an = ax.annotate(text, xy=(0.05, -.1), xycoords='axes fraction')
ims.append(add_arts + [an])
# if cr is not None: TO DO
# try:
# coastl = gp.GeoDataFrame.from_file(cr)
# except:
# coastl = gp.GeoDataFrame(cr)
# coastl.plot(ax=ax, **c_attrs)
if title: ax.set_title(title)
#ax.set_global()
#ax.coastlines('50m')
#cbar_ax = fig.add_axes([0.05, 0.05, 0.85, 0.05])
cbar = fig.colorbar(im,
ticks=vrange,
orientation='vertical',
fraction=0.017,
pad=0.04)
#plt.colorbar()
v = animation.ArtistAnimation(fig,
ims,
interval=200,
blit=False,
repeat=False)
plt.close()
return v
def contourf(self, it=None, **kwargs):
x = kwargs.get('x', self._obj.SCHISM_hgrid_node_x[:].values)
y = kwargs.get('y', self._obj.SCHISM_hgrid_node_y[:].values)
try:
t = kwargs.get('t', self._obj.time.values)
except:
pass
tes = kwargs.get('tes',
self._obj.SCHISM_hgrid_face_nodes.values[:, :4])
# sort out quads
try:
mask = np.isnan(tes)[:, 3]
tr3 = tes[mask][:, :3]
tr3_ = quads_to_tris(tes[~mask])
if tr3_:
tri3 = np.append(tr3, tr3_, axis=0).astype(int)
else:
tri3 = tr3.astype(int)
except:
tri3 = tes.astype(int)
var = kwargs.get('var', 'depth')
z = kwargs.get('z', self._obj[var].values[it, :].flatten())
vmin = kwargs.get('vmin', z.min())
vmax = kwargs.get('vmax', z.max())
nv = kwargs.get('nv', 10)
title = kwargs.get('title', 'contourf plot for {}'.format(var))
vrange = np.linspace(vmin, vmax, nv, endpoint=True)
## CHOOSE YOUR PROJECTION
# ax = plt.axes(projection=ccrs.Orthographic(grid_x.mean(), grid_y.mean()))
# [fig,ax] = kwargs.get('figure',[plt.figure(figsize=(12,8)),plt.axes(projection=ccrs.PlateCarree())])
# ax.set_extent([x.min(), x.max(), y.min(), y.max()])
# ax.background_patch.set_facecolor('k')
fig = plt.figure(figsize=(12, 8))
ax = plt.axes()
#optional mask for the data
mask = kwargs.get('mask', None)
if 'mask' in kwargs:
z = np.ma.masked_array(z, mask)
z = z.filled(fill_value=-99999)
xy = kwargs.get('xy', (.3, 1.05))
for val in [
'x', 'y', 't', 'it', 'z', 'vmin', 'vmax', 'title', 'nv', 'tes',
'mask', 'xy', 'var', 'figure'
]:
try:
del kwargs[val]
except:
pass
ax.set_aspect('equal')
p = ax.tricontourf(x,
y,
tri3,
z,
vrange,
vmin=vmin,
vmax=vmax,
**kwargs) #, transform=ccrs.PlateCarree() )
cbar = fig.colorbar(p, ticks=vrange, orientation='vertical')
if it:
text = 'time={}'.format(t[it])
an = ax.annotate(text, xy=xy, xycoords='axes fraction')
ax.set_title(title, pad=30)
ax.set_xlabel('Longitude (degrees)')
ax.set_ylabel('Latitude (degrees)')
return p #fig, ax
def contour(self, it=None, **kwargs):
x = kwargs.get("x", self._obj.SCHISM_hgrid_node_x[:].values)
y = kwargs.get("y", self._obj.SCHISM_hgrid_node_y[:].values)
try:
t = kwargs.get("t", self._obj.time.values)
except:
pass
tes = kwargs.get("tes",
self._obj.SCHISM_hgrid_face_nodes.values[:, :4])
var = kwargs.get("var", "depth")
z = kwargs.get("z", self._obj[var].values[it, :].flatten())
# sort out quads
try:
mask = np.isnan(tes)[:, 3]
tr3 = tes[mask][:, :3]
tr3_ = quads_to_tris(tes[~mask])
if tr3_:
tri3 = np.append(tr3, tr3_, axis=0).astype(int)
else:
tri3 = tr3.astype(int)
except:
tri3 = tes.astype(int)
fig, ax = plt.subplots(figsize=(12, 8))
vmin = kwargs.get("vmin", z.min())
vmax = kwargs.get("vmax", z.max())
nv = kwargs.get("nv", 10)
xy = kwargs.get("xy", (0.3, 1.05))
title = kwargs.get("title", "contour plot for {}".format(var))
vrange = np.linspace(vmin, vmax, nv, endpoint=True)
## CHOOSE YOUR PROJECTION
# ax = plt.axes(projection=ccrs.Orthographic(x.mean(), y.mean()))
# ax = plt.axes(projection=ccrs.PlateCarree())
# ax.background_patch.set_facecolor('k')
ax = plt.axes()
# optional mask for the data
mask = kwargs.get("mask", None)
if "mask" in kwargs:
z = np.ma.masked_array(z, mask)
z = z.filled(fill_value=-99999)
for val in [
"x", "y", "t", "it", "vmin", "vmax", "title", "nv", "tes",
"mask", "xy", "z", "var"
]:
try:
del kwargs[val]
except:
pass
ax.set_aspect("equal")
p = plt.tricontour(x,
y,
tri3,
z,
vrange,
vmin=vmin,
vmax=vmax,
**kwargs)
cbar = fig.colorbar(p, ticks=vrange, orientation="vertical")
if it:
text = "time={}".format(t[it])
an = ax.annotate(text, xy=xy, xycoords="axes fraction")
ax.set_title(title, pad=30)
plt.xlabel("Longitude (degrees)")
plt.ylabel("Latitude (degrees)")
return p # , ax
