Commit a1597e04 authored by Modellers Operational's avatar Modellers Operational

Fix to allow plotting of spd

parent 9ff75c2b
......@@ -53,7 +53,13 @@ print(fname)
cmap = pt.plotting.pmlcmaps(var)
pool_size = 4
fvcom = pf.read.FileReader(fname, [var])
if var == 'uv':
fvcom = pf.read.FileReader(fname, ['u', 'v'])
setattr(fvcom.data, 'uv', np.sqrt(fvcom.data.u**2 + fvcom.data.v**2))
else:
fvcom = pf.read.FileReader(fname, [var])
label = labels[var]
extension = pt.plotting.colourbar_extension(*clim, getattr(fvcom.data, var).min(), getattr(fvcom.data, var).max())
level = 0
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment