def plotPhs(self, vname="eta", k=0, con="M2", phsunits="radians", xlims=None, ylims=None, **kwargs): """ Plots the phase of the constituent on a map """ ii = findCon(con, self.frqnames) if xlims == None or ylims == None: xlims = self.xlims ylims = self.ylims if len(self.Phs[vname].shape) == 2: zP = self.Phs[vname][ii, :] else: zP = self.Phs[vname][ii, k, :].ravel() if phsunits == "radians": clim = [0, 2 * np.pi] self.fig, self.ax, self.patches, self.cb = unsurf(self.xy, zP, xlim=xlims, ylim=ylims, clim=clim, **kwargs) elif phsunits == "degrees": clim = [0, 360.0] self.fig, self.ax, self.patches, self.cb = unsurf( self.xy, zP * 180 / np.pi, xlim=xlims, ylim=ylims, clim=clim, **kwargs )
def plotPhs(self,vname='eta',k=0,con='M2',phsunits='radians',xlims=None,ylims=None,**kwargs): """ Plots the phase of the constituent on a map """ ii = findCon(con,self.frqnames) if xlims==None or ylims==None: xlims=self.xlims ylims=self.ylims if len(self.Phs[vname].shape)==2: zP = self.Phs[vname][ii,:] else: zP = self.Phs[vname][ii,k,:].ravel() if phsunits=='radians': clim = [0,2*np.pi] self.fig,self.ax,self.patches,self.cb=unsurf(self.xy,zP,xlim=xlims,ylim=ylims,\ clim=clim,**kwargs) elif phsunits=='degrees': clim = [0,360.0] self.fig,self.ax,self.patches,self.cb=unsurf(self.xy,zP*180/np.pi,xlim=xlims,ylim=ylims,\ clim=clim,**kwargs)
def plotAmp(self, vname="eta", k=0, con="M2", xlims=None, ylims=None, barotropic=False, cbarpos=None, **kwargs): """ Plots the amplitude of the constituent on a map vname: one of 'eta','ubar','vbar','uc','vc',... Set vname = 'ellmaj','ellmin','ellang','ellphs' to plot ellipse properties """ ii = findCon(con, self.frqnames) if self.clim == None: self.clim = [] self.clim.append(np.min(self.Amp)) self.clim.append(np.max(self.Amp)) if xlims == None or ylims == None: xlims = self.xlims ylims = self.ylims if vname in ["ellmaj", "ellmin", "ellang", "ellphs"]: ell = self.getEllipse(barotropic=barotropic, k=k, con=con) if vname == "ellmaj": zA = ell[0] elif vname == "ellmin": zA = ell[1] elif vname == "ellang": zA = ell[2] elif vname == "ellphs": zA = ell[3] elif len(self.Amp[vname].shape) == 2: zA = self.Amp[vname][ii, :] else: zA = self.Amp[vname][ii, k, :].ravel() self.fig, self.ax, self.patches, self.cb = unsurf( self.xy, zA, xlim=xlims, ylim=ylims, clim=self.clim, colorbar=False, **kwargs ) # Add a decent looking colorbar if not cbarpos == None: cbaxes = self.fig.add_axes(cbarpos) cb = self.fig.colorbar(self.patches, cax=cbaxes, orientation="vertical") # cb.ax.set_title('[$m s^{-1}$]') plt.sca(self.ax) return self.patches
def plotAmp(self,vname='eta',k=0,con='M2',xlims=None,ylims=None,\ barotropic=False,cbarpos=None,**kwargs): """ Plots the amplitude of the constituent on a map vname: one of 'eta','ubar','vbar','uc','vc',... Set vname = 'ellmaj','ellmin','ellang','ellphs' to plot ellipse properties """ ii = findCon(con,self.frqnames) if self.clim==None: self.clim=[] self.clim.append(np.min(self.Amp)) self.clim.append(np.max(self.Amp)) if xlims==None or ylims==None: xlims=self.xlims ylims=self.ylims if vname in ['ellmaj','ellmin','ellang','ellphs']: ell=self.getEllipse(barotropic=barotropic,k=k,con=con) if vname=='ellmaj': zA=ell[0] elif vname=='ellmin': zA=ell[1] elif vname=='ellang': zA=ell[2] elif vname=='ellphs': zA=ell[3] elif len(self.Amp[vname].shape)==2: zA = self.Amp[vname][ii,:] else: zA = self.Amp[vname][ii,k,:].ravel() self.fig,self.ax,self.patches,self.cb=unsurf(self.xy,zA,xlim=xlims,ylim=ylims,\ clim=self.clim,colorbar=False,**kwargs) # Add a decent looking colorbar if not cbarpos==None: cbaxes = self.fig.add_axes(cbarpos) cb = self.fig.colorbar(self.patches,cax = cbaxes,orientation='vertical') #cb.ax.set_title('[$m s^{-1}$]') plt.sca(self.ax) return self.patches
def plotMean(self, vname="eta", k=0, xlims=None, ylims=None, **kwargs): """ Plots the amplitude of the constituent on a map """ if self.clim == None: self.clim = [] self.clim.append(np.min(self.Mean)) self.clim.append(np.max(self.Mean)) if xlims == None or ylims == None: xlims = self.xlims ylims = self.ylims if len(self.Amp[vname].shape) == 2: zA = self.Mean[vname] else: zA = self.Mean[vname][k, :].ravel() self.fig, self.ax, self.patches, self.cb = unsurf(self.xy, zA, xlim=xlims, ylim=ylims, clim=self.clim, **kwargs)
def plotMean(self,vname='eta',k=0,xlims=None,ylims=None,**kwargs): """ Plots the amplitude of the constituent on a map """ if self.clim==None: self.clim=[] self.clim.append(np.min(self.Mean)) self.clim.append(np.max(self.Mean)) if xlims==None or ylims==None: xlims=self.xlims ylims=self.ylims if len(self.Amp[vname].shape)==2: zA = self.Mean[vname] else: zA = self.Mean[vname][k,:].ravel() self.fig,self.ax,self.patches,self.cb=unsurf(self.xy,zA,xlim=xlims,ylim=ylims,\ clim=self.clim,**kwargs)