zorro_plotting.py 文件源码

def plotFFT( self ):
        self.fig.clear()
        self.axes = self.fig.add_axes( [0.0, 0.0, 1.0, 1.0] )
        self.axes.hold(False)

        FFTimage = np.fft.fft2( self.plotDict['image'] )
        FFTimage[0,0] = 1.0 # Clip out zero-frequency pixel

        FFTimage = np.log10( 1.0 + np.abs( np.fft.fftshift( FFTimage )))
        if "lowPass" in self.plotDict:
            FFTimage = ni.gaussian_filter( FFTimage, self.plotDict["lowPass"] )

        FFTclim = zorro.util.ciClim( FFTimage, sigma=2.5 )
        mage = self.axes.imshow( FFTimage, interpolation='bicubic', vmin=FFTclim[0], vmax=FFTclim[1], 
                         cmap=self.plotDict['image_cmap'] )
        if 'pixelsize' in self.plotDict:
            inv_ps = 1.0 / (FFTimage.shape[0] * self.plotDict['pixelsize'] )
            zorro.util.plotScalebar( mage, inv_ps, units=u'nm^{-1}' )
        self.axes.set_axis_off()
        if bool(self.plotDict['colorbar']):
            self.fig.colorbar( mage, fraction=0.046, pad=0.04)
        return self.printPlot( dpi_key=u'image_dpi' )