"""Basic stick plot"""
from vacumm.config import data_sample
from vacumm.misc.plot import stick2
# Read speed
import cdms2
f = cdms2.open(data_sample('mars3d.t.nc'))
u = f('u')
v = f('v')
f.close()
# Plot
stick2(u,
v,
title='Current in the Iroise Sea',
units='m/s',
bottom=.2,
top=.85,
quiver_headwidth=2,
close=True,
quiverkey_value=.5,
quiver_width=0.002,
quiver_scale=5.,
savefigs=__file__,
figsize=(5.5, 3),
show=False)
"""Basic stick plot"""
from vacumm.config import data_sample
from vacumm.misc.plot import stick2
# Read speed
import cdms2
f = cdms2.open(data_sample('mars3d.t.nc'))
u = f('u')
v = f('v')
f.close()
# Plot
stick2(u, v, title='Current in the Iroise Sea', units='m/s',
bottom=.2, top=.85, quiver_headwidth=2,
quiverkey_value=.5, quiver_width=0.002, quiver_scale=5.,
savefigs=__file__, figsize=(5.5, 3), show=False)
nt = len(u)
# Plots
from vacumm.misc.plot import curve2, stick2, bar2
import matplotlib.pyplot as P
P.rc('font', size=9)
P.figure(figsize=(5.5, 6))
# - vectors
colors = N.array(["#000000"]*nt) # back by default
colors[MV2.sqrt(u**2+v**2).filled(0.)>5.] = '#ff0000' # red if modulus > 4.
qv = stick2(u, v, color=colors.tolist(), xticklabels=False, subplot=211,
shadow=True, right=.85, hspace=.2, left=.14, hldays=True,
key=1, key_size=11, key_color='.3', ylabel_color='r', units=r'$m.s^{-1}$',
mod=True, mod_color='.4', mod_linewidth=2, title='Wind', quiver_scale = 50.,
quiver_headwidth=3,quiver_headlength=3,quiver_headaxislength=2,
quiverkey_pos=(.05, 1.05), quiverkey_color='k',
quiver_width=.006, alpha=.8, show=False)
qv.add_text(1.01, 1, 'North', va='top', size=11, fontweight='heavy')
qv.add_text(1.01, 0, 'South', va='bottom', size=11, fontweight='heavy')
# - precipitations
dr = r.clone() # variation != accumulation
dr[:-1] = N.diff(r) ; dr[-1] = r[-1]-r[-2]
lr = bar2(dr, width=.8, subplot=212, hldays=True,
long_name='Precipitations', shadow=True, ylabel = '%(long_name)s [$%(units)s$]',
title=False, dayhl=True, ylabel_color='#008888',zorder=100,
key=2, key_size=11, key_color='.3', log=True, ymin=0.01,
color='#00ffff', linewidth=.2, edgecolor='#888888', show=False)
