Context Navigation

source: trunk/UTIL/PYTHON/planetoplot_v2/ppplot.py @ 928

Last change on this file since 928 was 928, checked in by aslmd, 12 years ago
UTIL PYTHON planetoplot_v2. possibility to draw negative-like plots.
File size: 26.6 KB

Rev	Line
[910]	1	###############################################
	2	## PLANETOPLOT ##
	3	## --> PPPLOT ##
	4	###############################################
	5	## Author: Aymeric Spiga. 02-03/2013 ##
	6	###############################################
	7	# python built-in librairies
	8	import time as timelib
	9	# added librairies
	10	import numpy as np
	11	import matplotlib.pyplot as mpl
	12	from matplotlib.cm import get_cmap
	13	from mpl_toolkits.basemap import Basemap
	14	# personal librairies
	15	import ppcompute
	16	###############################################
	17
	18	#################################
	19	# global variables and settings #
	20	#################################
	21
	22	# matplotlib settings
	23	# http://matplotlib.org/users/customizing.html
	24	# -------------------------------
	25	mpl.rcParams['font.family'] = "serif"
	26	mpl.rcParams['axes.color_cycle'] = "r,g,b,k"
[928]	27	mpl.rcParams['contour.negative_linestyle'] = "dashed" # ou "solid"
[910]	28	mpl.rcParams['verbose.level'] = "silent"
	29	mpl.rcParams['lines.linewidth'] = 1.5
	30	mpl.rcParams['lines.markersize'] = 10
	31	mpl.rcParams['xtick.major.pad'] = 10
	32	mpl.rcParams['ytick.major.pad'] = 10
	33
	34	# global variables
	35	# -------------------------------
	36	# - where settings files are located
	37	# (None means planetoplot_v2 in PYTHONPATH)
	38	whereset = None
	39	# - some good default settings.
	40	# (bounds)
	41	how_many_sigma = 3.0
	42	# (contours)
	43	ccol = 'black'
	44	cline = 0.55
	45	# (vectors)
	46	widthvec = 0.002
	47	reducevec = 30.
	48	# (colorbar)
	49	zeorientation="vertical"
	50	zefrac = 0.05
	51	# (save figures)
	52	pad_inches_value=0.25
[928]	53	# (negative mode)
	54	def_negative = False
	55	def_negative = True
[910]	56	###############################################
	57
[928]	58	### settings for 'negative-like' mode
	59	if def_negative:
	60	mpl.rc('figure', facecolor='k', edgecolor='k')
	61	mpl.rcParams['text.color'] = 'w'
	62	mpl.rc('axes',labelcolor='w',facecolor='k',edgecolor='w')
	63	mpl.rcParams['xtick.color'] = 'w'
	64	mpl.rcParams['ytick.color'] = 'w'
	65	mpl.rcParams['grid.color'] = 'w'
	66	mpl.rc('savefig',facecolor='k',edgecolor='k')
	67
[910]	68	##########################
	69	# executed when imported #
	70	##########################
	71	###########################################
	72	# we load user-defined automatic settings #
	73	###########################################
	74	# initialize the warning variable about file not present...
	75	files_not_present = ""
	76	# ... and the whereset variable
	77	whereset = ppcompute.findset(whereset)
	78
	79	# - variable settings
	80	# -------------------------------
	81	zefile = "set_var.txt"
	82	vf = {} ; vc = {} ; vl = {}
	83	try:
	84	f = open(whereset+zefile, 'r')
	85	for line in f:
	86	if "#" in line: pass
	87	else:
	88	var, format, colorb, label = line.strip().split(';')
	89	ind = var.strip()
	90	vf[ind] = format.strip()
	91	vc[ind] = colorb.strip()
	92	vl[ind] = label.strip()
	93	f.close()
	94	except IOError:
	95	files_not_present = files_not_present + zefile + " "
	96
	97	## - file settings
	98	## -------------------------------
	99	#zefile = "set_file.txt"
	100	#prefix_t = {} ; suffix_t = {} ; name_t = {} ; proj_t = {} ; vecx_t = {} ; vecy_t = {}
	101	#try:
	102	# f = open(whereset+zefile, 'r')
	103	# for line in f:
	104	# if "#" in line: pass
	105	# else:
	106	# prefix, suffix, name, proj, vecx, vecy = line.strip().split(';')
	107	# ind = name.strip()
	108	# prefix_t[ind] = prefix.strip()
	109	# suffix_t[ind] = suffix.strip()
	110	# #name_t[ind] = name.strip()
	111	# proj_t[ind] = proj.strip()
	112	# vecx_t[ind] = vecx.strip()
	113	# vecy_t[ind] = vecy.strip()
	114	# f.close()
	115	#except IOError:
	116	# files_not_present = files_not_present + zefile + " "
	117
	118	# - multiplot settings
	119	# -------------------------------
	120	zefile = "set_multiplot.txt"
	121	subv_t = {} ; subh_t = {} ; wspace_t = {} ; hspace_t = {} ; font_t = {}
	122	try:
	123	f = open(whereset+zefile, 'r')
	124	for line in f:
	125	if "#" in line: pass
	126	else:
	127	num, subv, subh, wspace, hspace, font = line.strip().split(';')
	128	ind = int(num.strip())
	129	subv_t[ind] = int(subv.strip())
	130	subh_t[ind] = int(subh.strip())
	131	wspace_t[ind] = float(wspace.strip())
	132	hspace_t[ind] = float(hspace.strip())
	133	font_t[ind] = float(font.strip())
	134	f.close()
	135	except IOError:
	136	files_not_present = files_not_present + zefile + " "
	137
	138	# - background settings
	139	# -------------------------------
	140	zefile = "set_back.txt"
	141	back = {}
	142	try:
	143	f = open(whereset+zefile, 'r')
	144	for line in f:
	145	if "#" in line: pass
	146	else:
	147	name, link = line.strip().split(';')
	148	ind = name.strip()
	149	back[ind] = link.strip()
	150	f.close()
	151	except IOError:
	152	files_not_present = files_not_present + zefile + " "
	153
	154	# - area settings
	155	# -------------------------------
	156	zefile = "set_area.txt"
	157	area = {}
	158	try:
	159	f = open(whereset+zefile, 'r')
	160	for line in f:
	161	if "#" in line: pass
	162	else:
	163	name, wlat1, wlat2, wlon1, wlon2 = line.strip().split(';')
	164	area[name.strip()] = [[float(wlon1.strip()),float(wlon2.strip())],\
	165	[float(wlat1.strip()),float(wlat2.strip())]]
	166	f.close()
	167	except IOError:
	168	files_not_present = files_not_present + zefile + " "
	169	# A note to the user about missing files
	170	if files_not_present != "":
	171	print "warning: files "+files_not_present+" not in "+whereset+" ; those presets will be missing"
	172
	173	# TBD: should change vector color with colormaps
	174	#"gist_heat": "white",\
	175	#"hot": "white",\
	176	#"gray": "red",\
	177
	178	####################
	179	# useful functions #
	180	####################
	181
	182	# a function to define subplot
	183	# ... user can change settings in set_multiplot.txt read above
	184	# -------------------------------
	185	def definesubplot(numplot, fig):
	186	try:
	187	mpl.rcParams['font.size'] = font_t[numplot]
	188	except:
	189	mpl.rcParams['font.size'] = 18
	190	try:
	191	fig.subplots_adjust(wspace = wspace_t[numplot], hspace = hspace_t[numplot])
	192	subv, subh = subv_t[numplot],subh_t[numplot]
	193	except:
	194	print "!! WARNING !! no preset found from set_multiplot.txt, or this setting file was not found."
	195	subv = 1 ; subh = numplot
	196	return subv,subh
	197
	198	# a function to calculate automatically bounds (or simply prescribe those)
	199	# -------------------------------
	200	def calculate_bounds(field,vmin=None,vmax=None):
	201	# prescribed cases first
	202	zevmin = vmin
	203	zevmax = vmax
	204	# computed cases
	205	if zevmin is None or zevmax is None:
	206	# select values
	207	ind = np.where(field < 9e+35)
	208	fieldcalc = field[ ind ] # field must be a numpy array
	209	# calculate stdev and mean
	210	dev = np.std(fieldcalc)*how_many_sigma
	211	damean = ppcompute.mean(fieldcalc)
	212	# fill min/max if needed
	213	if vmin is None: zevmin = damean - dev
	214	if vmax is None: zevmax = damean + dev
	215	# special case: negative values with stddev while field is positive
	216	if zevmin < 0. and ppcompute.min(fieldcalc) > 0.: zevmin = 0.
[920]	217	# check that bounds are not too tight given the field
	218	amin = ppcompute.min(field)
	219	amax = ppcompute.max(field)
	220	cmin = 100.*np.abs((amin - zevmin)/amin)
	221	cmax = 100.*np.abs((amax - zevmax)/amax)
	222	if cmin > 150. or cmax > 150.:
	223	print "!! WARNING !! Bounds are a bit too tight. Might need to reconsider those."
	224	print "!! WARNING !! --> actual",amin,amax,"adopted",zevmin,zevmax
[910]	225	return zevmin, zevmax
	226	#### treat vmin = vmax for continuity
	227	#if vmin == vmax: zevmin = damean - dev ; zevmax = damean + dev
	228
	229	# a function to solve the problem with blank bounds !
	230	# -------------------------------
	231	def bounds(what_I_plot,zevmin,zevmax,miss=9e+35):
	232	small_enough = 1.e-7
	233	if zevmin < 0: what_I_plot[ what_I_plot < zevmin(1.-small_enough) ] = zevmin(1.-small_enough)
	234	else: what_I_plot[ what_I_plot < zevmin(1.+small_enough) ] = zevmin(1.+small_enough)
	235	what_I_plot[ what_I_plot > miss ] = -miss
	236	what_I_plot[ what_I_plot > zevmax ] = zevmax*(1.-small_enough)
	237	return what_I_plot
	238
	239	# a generic function to show (GUI) or save a plot (PNG,EPS,PDF,...)
	240	# -------------------------------
[917]	241	def save(mode="gui",filename="plot",folder="./",includedate=True,res=150,custom=False):
[910]	242	# a few settings
	243	possiblesave = ['eps','ps','svg','png','jpg','pdf']
	244	# now the main instructions
	245	if mode == "gui":
	246	mpl.show()
	247	elif mode in possiblesave:
	248	## name of plot
	249	name = folder+'/'+filename
	250	if includedate:
	251	for ttt in timelib.gmtime():
	252	name = name + "_" + str(ttt)
	253	name = name +"."+mode
	254	## save file
	255	print "**** Saving file in "+mode+" format... Please wait."
[917]	256	if not custom:
	257	# ... regular plots
	258	mpl.savefig(name,dpi=res,pad_inches=pad_inches_value,bbox_inches='tight')
	259	else:
	260	# ... mapping mode, adapted space for labels etc...
	261	mpl.savefig(name,dpi=res)
[910]	262	else:
	263	print "!! ERROR !! File format not supported. Supported: ",possiblesave
	264
	265	##################################
	266	# a generic class to make a plot #
	267	##################################
	268	class plot():
	269
	270	# print out a help string when help is invoked on the object
	271	# -------------------------------
	272	def __repr__(self):
	273	whatprint = 'plot object. \"help(plot)\" for more information\n'
	274	return whatprint
	275
	276	# default settings
	277	# -- user can define settings by two methods.
	278	# -- 1. yeah = plot2d(title="foo")
	279	# -- 2. yeah = pp() ; yeah.title = "foo"
	280	# -------------------------------
	281	def __init__(self,\
	282	var=None,\
	283	field=None,\
	284	absc=None,\
	285	xlabel="",\
	286	ylabel="",\
	287	div=20,\
	288	logx=False,\
	289	logy=False,\
	290	swap=False,\
	291	swaplab=True,\
	292	invert=False,\
	293	xcoeff=1.,\
	294	ycoeff=1.,\
	295	title=""):
	296	## what could be defined by the user
	297	self.var = var
	298	self.field = field
	299	self.absc = absc
	300	self.xlabel = xlabel
	301	self.ylabel = ylabel
	302	self.title = title
	303	self.div = div
	304	self.logx = logx
	305	self.logy = logy
	306	self.swap = swap
	307	self.swaplab = swaplab # NB: swaplab only used if swap=True
	308	self.invert = invert
	309	self.xcoeff = xcoeff
	310	self.ycoeff = ycoeff
	311	## other useful arguments
	312	## ... not used here in ppplot but need to be attached to plot object
	313	self.axisbg = "white"
	314	self.superpose = False
	315
	316	# check
	317	# -------------------------------
	318	def check(self):
	319	if self.field is None: print "!! ERROR !! Please define a field to be plotted" ; exit()
	320
	321	# define_from_var
	322	# ... this uses settings in set_var.txt
	323	# -------------------------------
	324	def define_from_var(self):
	325	if self.var is not None:
	326	if self.var.upper() in vl.keys():
	327	self.title = vl[self.var.upper()]
	328
	329	# make
	330	# this is generic to all plots
	331	# -------------------------------
	332	def make(self):
	333	self.check()
[928]	334	# labels, title, etc...
[910]	335	mpl.xlabel(self.xlabel)
	336	mpl.ylabel(self.ylabel)
	337	if self.swap:
	338	if self.swaplab:
	339	mpl.xlabel(self.ylabel)
	340	mpl.ylabel(self.xlabel)
	341	mpl.title(self.title)
	342	# if masked array, set masked values to filled values (e.g. np.nan) for plotting purposes
	343	if type(self.field).__name__ in 'MaskedArray':
	344	self.field[self.field.mask] = self.field.fill_value
	345
	346	################################
	347	# a subclass to make a 1D plot #
	348	################################
	349	class plot1d(plot):
	350
	351	# print out a help string when help is invoked on the object
	352	# -------------------------------
	353	def __repr__(self):
	354	whatprint = 'plot1d object. \"help(plot1d)\" for more information\n'
	355	return whatprint
	356
	357	# default settings
	358	# -- user can define settings by two methods.
	359	# -- 1. yeah = plot1d(title="foo")
	360	# -- 2. yeah = pp() ; yeah.title = "foo"
	361	# -------------------------------
	362	def __init__(self,\
	363	lstyle='-',\
	364	color='b',\
[923]	365	marker='x',\
	366	label=None):
[910]	367	## get initialization from parent class
	368	plot.__init__(self)
	369	## what could be defined by the user
	370	self.lstyle = lstyle
	371	self.color = color
	372	self.marker = marker
[923]	373	self.label = label
[910]	374
	375	# define_from_var
	376	# ... this uses settings in set_var.txt
	377	# -------------------------------
	378	def define_from_var(self):
	379	# get what is done in the parent class
	380	plot.define_from_var(self)
	381	# add specific stuff
	382	if self.var is not None:
	383	if self.var.upper() in vl.keys():
	384	self.ylabel = vl[self.var.upper()]
	385	self.title = ""
	386
	387	# make
	388	# -------------------------------
	389	def make(self):
	390	# get what is done in the parent class
	391	plot.make(self)
	392	# add specific stuff
	393	mpl.grid(color='grey')
	394	if self.lstyle == "": self.lstyle = " " # to allow for no line at all with ""
	395	# swapping if requested
	396	if self.swap: x = self.field ; y = self.absc
	397	else: x = self.absc ; y = self.field
	398	# coefficients on axis
	399	x=xself.xcoeff ; y=yself.ycoeff
	400	# check axis
	401	if x.size != y.size:
	402	print "!! ERROR !! x and y sizes don't match on 1D plot.", x.size, y.size
	403	exit()
	404	# make the 1D plot
	405	# either request linestyle or let matplotlib decide
	406	if self.lstyle is not None and self.color is not None:
[923]	407	mpl.plot(x,y,self.color+self.lstyle,marker=self.marker,label=self.label)
[910]	408	else:
[923]	409	mpl.plot(x,y,marker=self.marker,label=self.label)
[910]	410	# make log axes and/or invert ordinate
	411	# ... this must be after plot so that axis bounds are well-defined
	412	# ... also inverting must be after making the thing logarithmic
	413	if self.logx: mpl.semilogx()
	414	if self.logy: mpl.semilogy()
	415	if self.invert: ax = mpl.gca() ; ax.set_ylim(ax.get_ylim()[::-1])
[923]	416	# add a label for line(s)
	417	if self.label is not None:
	418	if self.label != "":
	419	mpl.legend(loc="best",fancybox=True)
[910]	420	## TBD: set with .div the number of ticks
	421	## TBD: be able to control plot limits
	422	#ticks = self.div + 1
	423	#ax = mpl.gca()
	424	#ax.get_xaxis().set_ticks(np.linspace(ppcompute.min(x),ppcompute.max(x),ticks/2+1))
	425
	426	################################
	427	# a subclass to make a 2D plot #
	428	################################
	429	class plot2d(plot):
	430
	431	# print out a help string when help is invoked on the object
	432	# -------------------------------
	433	def __repr__(self):
	434	whatprint = 'plot2d object. \"help(plot2d)\" for more information\n'
	435	return whatprint
	436
	437	# default settings
	438	# -- user can define settings by two methods.
	439	# -- 1. yeah = plot2d(title="foo")
	440	# -- 2. yeah = pp() ; yeah.title = "foo"
	441	# -------------------------------
	442	def __init__(self,\
	443	ordi=None,\
	444	mapmode=False,\
	445	proj="cyl",\
	446	back=None,\
	447	colorb="jet",\
	448	trans=1.0,\
	449	addvecx=None,\
	450	addvecy=None,\
	451	addcontour=None,\
	452	fmt="%.2e",\
	453	blon=None,\
	454	blat=None,\
	455	area=None,\
	456	vmin=None,\
	457	vmax=None,\
	458	colorvec="black"):
	459	## get initialization from parent class
	460	plot.__init__(self)
	461	## what could be defined by the user
	462	self.ordi = ordi
	463	self.mapmode = mapmode
	464	self.proj = proj
	465	self.back = back
	466	self.colorb = colorb
	467	self.trans = trans
	468	self.addvecx = addvecx
	469	self.addvecy = addvecy
	470	self.colorvec = colorvec
	471	self.addcontour = addcontour
	472	self.fmt = fmt
	473	self.blon = blon ; self.blat = blat
	474	self.area = area
	475	self.vmin = vmin ; self.vmax = vmax
	476
	477	# define_from_var
	478	# ... this uses settings in set_var.txt
	479	# -------------------------------
	480	def define_from_var(self):
	481	# get what is done in the parent class
	482	plot.define_from_var(self)
	483	# add specific stuff
	484	if self.var is not None:
	485	if self.var.upper() in vl.keys():
	486	self.colorb = vc[self.var.upper()]
	487	self.fmt = vf[self.var.upper()]
	488
	489	# make
	490	# -------------------------------
	491	def make(self):
	492	# get what is done in the parent class...
	493	plot.make(self)
	494	# ... then add specific stuff
	495	############################################################################################
	496	### PRE-SETTINGS
	497	############################################################################################
	498	# transposing if necessary
	499	shape = self.field.shape
	500	if shape[0] != shape[1]:
	501	if len(self.absc) == shape[0] and len(self.ordi) == shape[1]:
	502	print "!! WARNING !! Transposing axes"
	503	self.field = np.transpose(self.field)
	504	# bound field
	505	zevmin, zevmax = calculate_bounds(self.field,vmin=self.vmin,vmax=self.vmax)
	506	what_I_plot = bounds(self.field,zevmin,zevmax)
	507	# define contour field levels. define color palette
	508	ticks = self.div + 1
	509	zelevels = np.linspace(zevmin,zevmax,ticks)
	510	palette = get_cmap(name=self.colorb)
	511	# do the same thing for possible contourline entries
	512	if self.addcontour is not None:
	513	# if masked array, set masked values to filled values (e.g. np.nan) for plotting purposes
	514	if type(self.addcontour).__name__ in 'MaskedArray':
	515	self.addcontour[self.addcontour.mask] = self.addcontour.fill_value
	516	zevminc, zevmaxc = calculate_bounds(self.addcontour)
	517	what_I_contour = bounds(self.addcontour,zevminc,zevmaxc)
	518	ticks = self.div + 1
	519	zelevelsc = np.linspace(zevminc,zevmaxc,ticks)
	520	############################################################################################
	521	### MAIN PLOT
	522	### NB: contour lines are done before contour shades otherwise colorar error
	523	############################################################################################
	524	if not self.mapmode:
	525	## A SIMPLE 2D PLOT
	526	###################
	527	# swapping if requested
	528	if self.swap: x = self.ordi ; y = self.absc
	529	else: x = self.absc ; y = self.ordi
	530	# coefficients on axis
	531	x=xself.xcoeff ; y=yself.ycoeff
	532	# make shaded and line contours
	533	if self.addcontour is not None:
	534	mpl.contour(x, y, what_I_contour, \
	535	zelevelsc, colors = ccol, linewidths = cline)
	536	mpl.contourf(x, y, \
	537	self.field, \
	538	zelevels, cmap=palette)
	539	# make log axes and/or invert ordinate
	540	if self.logx: mpl.semilogx()
	541	if self.logy: mpl.semilogy()
	542	if self.invert: ax = mpl.gca() ; ax.set_ylim(ax.get_ylim()[::-1])
	543	else:
	544	## A 2D MAP USING PROJECTIONS (basemap)
	545	#######################################
	546	mpl.xlabel("") ; mpl.ylabel("")
	547	# additional security in case self.proj is None here
	548	# ... we set cylindrical projection (the simplest one)
	549	if self.proj is None: self.proj = "cyl"
	550	# get lon and lat in 2D version.
	551	# (but first ensure we do have 2D coordinates)
	552	if self.absc.ndim == 1: [self.absc,self.ordi] = np.meshgrid(self.absc,self.ordi)
	553	elif self.absc.ndim > 2: print "!! ERROR !! lon and lat arrays must be 1D or 2D"
	554	# get lat lon intervals and associated settings
	555	wlon = [np.min(self.absc),np.max(self.absc)]
	556	wlat = [np.min(self.ordi),np.max(self.ordi)]
	557	# -- area presets are in set_area.txt
	558	if self.area is not None:
	559	if self.area in area.keys():
	560	wlon, wlat = area[self.area]
	561	# -- settings for meridians and parallels
	562	steplon = abs(wlon[1]-wlon[0])/6.
	563	steplat = abs(wlat[1]-wlat[0])/3.
	564	mertab = np.r_[wlon[0]:wlon[1]:steplon] ; merlab = [0,0,0,1]
	565	partab = np.r_[wlat[0]:wlat[1]:steplat] ; parlab = [1,0,0,0]
	566	format = '%.1f'
	567	# -- center of domain and bounding lats
	568	lon_0 = 0.5*(wlon[0]+wlon[1])
	569	lat_0 = 0.5*(wlat[0]+wlat[1])
	570	# some tests, bug fixes, and good-looking settings
	571	# ... cyl is good for global and regional
	572	if self.proj == "cyl":
[917]	573	format = '%.0f'
[910]	574	# ... global projections
	575	elif self.proj in ["ortho","moll","robin"]:
	576	wlat[0] = None ; wlat[1] = None ; wlon[0] = None ; wlon[1] = None
	577	steplon = 30. ; steplat = 30.
[923]	578	if self.proj in ["robin","moll"]: steplon = 60.
[910]	579	mertab = np.r_[-360.:360.:steplon]
	580	partab = np.r_[-90.:90.:steplat]
	581	if self.proj == "ortho":
	582	merlab = [0,0,0,0] ; parlab = [0,0,0,0]
	583	# in ortho projection, blon and blat can be used to set map center
	584	if self.blon is not None: lon_0 = self.blon
	585	if self.blat is not None: lat_0 = self.blat
	586	elif self.proj == "moll":
	587	merlab = [0,0,0,0]
	588	format = '%.0f'
	589	# ... regional projections
	590	elif self.proj in ["lcc","laea","merc"]:
	591	if self.proj in ["lcc","laea"] and wlat[0] == -wlat[1]:
	592	print "!! ERROR !! with Lambert lat1 must be different than lat2" ; exit()
	593	if wlat[0] < -80. and wlat[1] > 80.:
	594	print "!! ERROR !! set an area (not global)" ; exit()
	595	format = '%.0f'
	596	elif self.proj in ["npstere","spstere"]:
	597	# in polar projections, blat gives the bounding lat
	598	# if not set, set something reasonable
	599	if self.blat is None: self.blat = 60.
	600	# help the user who forgets self.blat would better be negative in spstere
	601	# (this actually serves for the default setting just above)
	602	if self.proj == "spstere" and self.blat > 0: self.blat = -self.blat
	603	# ... unsupported projections
	604	else:
	605	print "!! ERROR !! unsupported projection. supported: "+\
	606	"cyl, npstere, spstere, ortho, moll, robin, lcc, laea, merc"
	607	# finally define projection
	608	m = Basemap(projection=self.proj,\
	609	lat_0=lat_0,lon_0=lon_0,\
	610	boundinglat=self.blat,\
	611	llcrnrlat=wlat[0],urcrnrlat=wlat[1],\
	612	llcrnrlon=wlon[0],urcrnrlon=wlon[1])
	613	# draw meridians and parallels
	614	ft = int(mpl.rcParams['font.size']*3./4.)
	615	m.drawmeridians(mertab,labels=merlab,color='grey',linewidth=0.75,fontsize=ft,fmt=format)
	616	m.drawparallels(partab,labels=parlab,color='grey',linewidth=0.75,fontsize=ft,fmt=format)
	617	# define background (see set_back.txt)
	618	if self.back is not None:
	619	if self.back in back.keys():
	620	print "**** info: loading a background, please wait.",self.back
	621	if back not in ["coast","sea"]: m.warpimage(back[self.back],scale=0.75)
	622	elif back == "coast": m.drawcoastlines()
	623	elif back == "sea": m.drawlsmask(land_color='white',ocean_color='aqua')
	624	else:
	625	print "!! ERROR !! requested background not defined. change name or fill in set_back.txt" ; exit()
	626	# define x and y given the projection
	627	x, y = m(self.absc, self.ordi)
	628	# contour field. first line contour then shaded contour.
	629	if self.addcontour is not None:
	630	m.contour(x, y, what_I_contour, \
	631	zelevelsc, colors = ccol, linewidths = cline)
	632	m.contourf(x, y, what_I_plot, zelevels, cmap = palette, alpha = self.trans)
	633	############################################################################################
	634	### COLORBAR
	635	############################################################################################
	636	if self.trans > 0.:
	637	## draw colorbar. settings are different with projections. or if not mapmode.
[917]	638	if not self.mapmode: orientation=zeorientation ; frac = 0.075 ; pad = 0.03
	639	elif self.proj in ['moll']: orientation="horizontal" ; frac = 0.075 ; pad = 0.03
	640	elif self.proj in ['cyl']: orientation="vertical" ; frac = 0.023 ; pad = 0.03
	641	else: orientation = zeorientation ; frac = zefrac ; pad = 0.03
[910]	642	zelevpal = np.linspace(zevmin,zevmax,num=min([ticks/2+1,21]))
[917]	643	zecb = mpl.colorbar(fraction=frac,pad=pad,\
[910]	644	format=self.fmt,orientation=orientation,\
	645	ticks=zelevpal,\
	646	extend='neither',spacing='proportional')
	647	if zeorientation == "horizontal": zecb.ax.set_xlabel(self.title) ; self.title = ""
	648	############################################################################################
	649	### VECTORS. must be after the colorbar. we could also leave possibility for streamlines.
	650	############################################################################################
	651	### not expecting NaN in self.addvecx and self.addvecy. masked arrays is just enough.
	652	stride = 3
	653	if self.addvecx is not None and self.addvecy is not None and self.mapmode:
	654	## for metwinds only ???
	655	[vecx,vecy] = m.rotate_vector(self.addvecx,self.addvecy,self.absc,self.ordi)
	656	# reference vector is scaled
	657	zescale = ppcompute.mean(np.sqrt(self.addvecxself.addvecx+self.addvecyself.addvecy))
	658	# make vector field
	659	q = m.quiver( x[::stride,::stride],y[::stride,::stride],\
	660	vecx[::stride,::stride],vecy[::stride,::stride],\
	661	angles='xy',color=self.colorvec,pivot='middle',\
	662	scale=zescale*reducevec,width=widthvec )
	663	# make vector key. default is on upper left corner.
[923]	664	keyh = 1.025 ; keyv = 1.05
	665	#keyh = -0.03 ; keyv = 1.08
	666	p = mpl.quiverkey(q,keyh,keyv,\
[910]	667	zescale,str(int(zescale)),\
	668	color='black',labelpos='S',labelsep = 0.07)

Note: See TracBrowser for help on using the repository browser.

Download in other formats: