amrplot.rgplot Class Reference

As polyplot, but use regridded data to display. More...

Inheritance diagram for amrplot.rgplot:

Public Member Functions
def	show (self, var=None, data=None, min=None, max=None, reset=None, fixrange=None, filenameout=None)
	Draw the plotting-window. More...
Public Member Functions inherited from amrplot.polyplot
def	__init__ (self, value, data, nlevels=256, grid=None, blocks=None, blockWidth=8, blockHeight=8, nlevel1=0, cmap='jet', min=None, max=None, xrange=None, yrange=None, orientation='vertical', right=True, fixzoom=None, fixrange=None, fig=None, axis=None, filenameout=None, clear=True, edgecolor='k', smooth=0, swap=0, kwargs)
def	setValue (self, value, min=None, max=None)
	Sets the min and max values of the data to saturate the display. More...
def	update (self, var=None, data=None, min=None, max=None, reset=None, fixrange=None, filenameout=None)
	Prepare to re-draw the window, check if data was updated. More...
def	info (self)
	Print info to the console. More...
def	show (self, var=None, data=None, min=None, max=None, reset=None, fixrange=None, filenameout=None)
	Draw the plotting-window. More...
def	colorbar (self, cax=None)
	Draw the colorbar. More...
def	save (self, filenameout=None)
	Save the figure. More...
def	onkey (self, event)

Public Attributes
	viewXrange
	viewYrange
	ax
	fixzoom
	valueClip
	swap
	image
	filenameout
Public Attributes inherited from amrplot.polyplot
	swap
	nlevels
	grid
	blocks
	blockWidth
	blockHeight
	nlevel1
	cmap
	orientation
	right
	cbarwidth
	cbarpad
	fixzoom
	fixrange
	filenameout
	clear
	fontsize
	fig_w
	fig_h
	dpi
	maxXticks
	maxYticks
	cbarticks
	edgecolor
	smooth
	xrange
	yrange
	figure
	ax
	value
	min
	max
	data
	viewXrange
	viewYrange
	valueClip
	xlist
	ylist
	xlistspecial
	ylistspecial
	cax
	cbar
	selection

Detailed Description

As polyplot, but use regridded data to display.

Definition at line 336 of file amrplot.py.

Member Function Documentation

show()

def amrplot.rgplot.show	(		self,
			var = None,
			data = None,
			min = None,
			max = None,
			reset = None,
			fixrange = None,
			filenameout = None
	)

Draw the plotting-window.

Definition at line 339 of file amrplot.py.

        '''Draw the plotting-window'''

        t0 = default_timer()

        self.update(var=var,data=data,min=min,max=max,reset=reset,fixrange=fixrange,filenameout=filenameout)

        self.viewXrange=self.ax.xaxis.get_view_interval()
        self.viewYrange=self.ax.yaxis.get_view_interval()
        self.figure.clf()
        self.ax = self.figure.gca()        
        self.ax.set_rasterization_zorder(-9)
        self.info()

        if self.fixzoom==None:
            self.ax.set_xlim(self.xrange[0],self.xrange[1])
            self.ax.set_ylim(self.yrange[0],self.yrange[1])
        else:
            self.ax.set_xlim(self.viewXrange)
            self.ax.set_ylim(self.viewYrange)
        self.ax.set_aspect('equal')


        valueRange=self.max-self.min
        if valueRange == 0.:
            valueRange = 1

        self.valueClip = np.clip(self.value,self.min,self.max)

# Now regrid and imshow:
        tdata0= default_timer()
        xrange=[self.ax.get_xlim()[0],self.ax.get_xlim()[1]]
        yrange=[self.ax.get_ylim()[0],self.ax.get_ylim()[1]]

        nregrid = [self.dpi*self.fig_w,self.dpi*self.fig_h]

        if (self.swap == 0):
            CC=self.data.getCenterPoints()
        else:
            CC=self.data.getCenterPoints()[:,[1,0]]
        tmp0=np.complex(0,nregrid[0])
        tmp1=np.complex(0,nregrid[1])
        grid_x, grid_y = np.mgrid[xrange[0]:xrange[1]:tmp0, yrange[0]:yrange[1]:tmp1]

# regrid with linear interpolation and fall back to nearest neighbor at NaN, which should just be the borders.
        gridvar = griddata(CC, self.valueClip, (grid_x, grid_y), method='cubic',fill_value=float(np.NaN))
        isnan=np.isnan(gridvar)
        gridvar[isnan] = griddata(CC, self.value, (grid_x[isnan], grid_y[isnan]), method='nearest',fill_value=float(np.NaN))

# smooth the data slightly:
        gridvar = ndimage.gaussian_filter(gridvar, sigma=self.smooth)

        extent = xrange[0], xrange[1], yrange[0], yrange[1]
        gridvarClip = np.clip(gridvar,self.min,self.max)
        self.image=gridvarClip
        tdata1=default_timer()


        im2 = self.ax.imshow(gridvarClip.transpose(), cmap=plt.cm.get_cmap(self.cmap, self.nlevels), interpolation='bicubic',extent=extent,origin='lower',zorder=-10)
        norm = colors.Normalize(vmin=self.min, vmax=self.max)
        im2.set_norm(norm) 
#        Q=velovect(self.data.u1,self.data.u2,self.data,nvect=[20,20],scale=30,fig=self)

        if self.grid != None:
            if (self.swap == 0):
                [myxlist,myylist]=self.data.getPointList()
            else:
                [myylist,myxlist]=self.data.getPointList()
            self.ax.fill(myxlist,myylist, 
                     facecolor='none', edgecolor=self.edgecolor,aa=True,linewidth=0.2,alpha=0.8)

        if self.blocks != None:
            [myxBlockList,myyBlockList] = self.data.getPieces(self.blockWidth,self.blockHeight,self.nlevel1)
            self.ax.fill(myxBlockList,myyBlockList, 
                    facecolor='none', edgecolor=self.edgecolor,aa=True,linewidth=0.4,alpha=0.5)

        if self.orientation != None:
            self.colorbar() 

# ticks:
        if self.maxXticks != None:
            self.ax.xaxis.set_major_locator(MaxNLocator(self.maxXticks-1))
        if self.maxYticks != None:
            self.ax.yaxis.set_major_locator(MaxNLocator(self.maxYticks-1))

        for tick in self.ax.xaxis.get_major_ticks():
            tick.label1.set_fontsize(self.fontsize)
        for tick in self.ax.yaxis.get_major_ticks():
            tick.label1.set_fontsize(self.fontsize)
        self.ax.xaxis.get_offset_text().set_fontsize(self.fontsize-2)
        self.ax.yaxis.get_offset_text().set_fontsize(self.fontsize-2)

        tend = default_timer()
        print('time for arranging the data= %f sec' % (tdata1-tdata0))
        print('Execution time = %f sec' % (tend-t0))
        print('=======================================================')
        if self.filenameout == None:
            plt.draw()

# Make the main axis active:
        plt.sca(self.ax)

#=============================================================================

Member Data Documentation

ax

amrplot.rgplot.ax

Definition at line 348 of file amrplot.py.

filenameout

amrplot.rgplot.filenameout

Definition at line 434 of file amrplot.py.

fixzoom

amrplot.rgplot.fixzoom

Definition at line 352 of file amrplot.py.

image

amrplot.rgplot.image

Definition at line 392 of file amrplot.py.

swap

amrplot.rgplot.swap

Definition at line 374 of file amrplot.py.

valueClip

amrplot.rgplot.valueClip

Definition at line 365 of file amrplot.py.

viewXrange

amrplot.rgplot.viewXrange

Definition at line 345 of file amrplot.py.

viewYrange

amrplot.rgplot.viewYrange

Definition at line 346 of file amrplot.py.

---

The documentation for this class was generated from the following file:

• amrplot.py