print 'movie' print n.shape #blob = BlobDB(n,meta=meta) #blob = BlobMovie(n,meta=meta) blob = Blob2D(n,meta=meta) time = np.squeeze(collect("t_array",path=path,xind=[0,0])) #how to quickly make a movie from frame import Frame, FrameMovie ##pp = PdfPages('n0.pdf') fig = plt.figure() frm_n0 = Frame(n0,meta={'stationary':True,'dx':dx}) #frm_n0.render(fig,111) frm_n0.ax = fig.add_subplot(111) frm_n0.img = frm_n0.ax.plot(frm_n0[:,0]) #fig.savefig(pp,format='pdf') fig.savefig('n0.eps') plt.close(fig) ##pp.close() data_c = phi frm_data = Frame(blob.raw_data,meta={'mask':True,'dx':dx,'dy':dy,'title':'n'}) frm_amp = Frame(blob.amp,meta={'title':r'$n_{max}$'}) frm_fft = Frame((blob.power[:,0:30,0:60]),meta={'mask':True,'title':'power', 'ylabel':r'$k_y \rho_s$', 'xlabel':r'$k_x$'})
import matplotlib.ticker as ticker from matplotlib.ticker import FormatStrFormatter lin_formatter = ticker.ScalarFormatter() from pylab import legend lin_formatter.set_powerlimits((1, 1)) #plt.autoscale(axis='x',tight=True) #self.ax.axis('tight') #let's create single frame pp = PdfPages('gamma.pdf') fg = plt.figure() gamma.ax = None gamma.t = nt-2 gamma_th.ax = None gamma.render(fig,111) gamma_th.render(fig,111) gamma.ax.yaxis.set_major_formatter(lin_formatter) plt.setp(gamma_th.img, color='b', linewidth=3.0,alpha=.7) plt.setp(gamma.img, color='r', linewidth=2.0,alpha=.7) plt.autoscale(axis='x',tight=True) print 'gamma.img: ',gamma.img leg = plt.legend([gamma.img,gamma_th.img],('BOUT++', 'analytic'), 'best', shadow=False, fancybox=True) leg.get_frame().set_alpha(0.6) fig.savefig(pp,format='pdf') plt.close(fig)
gamma_th.render(fig,223) print data.shape namp = abs(data).max(1).max(1) namp = Frame(namp,meta={'ticksize':14}) namp.render(fig,224) fig.savefig(pp,format='pdf') plt.close(fig) pp.close() pp = PdfPages('gamma.pdf') fig = plt.figure() #gamma_num.render(fig,111) gamma_th.ax = None #gamma.t = 20 gamma_num.ax = None gamma_th.render(fig,111) gamma_num.render(fig,111) fig.savefig(pp,format='pdf') plt.close(fig) pp.close() #let's make a movie fig = plt.figure() gamma_num.t = 0 gamma_num.ax = fig.add_subplot(111) gamma_th.ax = gamma_num.ax #gamma_th.ax = fig.add_subplot(111)
lam_rough.append(est_lam) lambdafit.append(popt[0]*np.exp(-pos[0][xstart:xstop,:]/popt[1])) offset.append(popt[0]) nmax.append(n.max()) print n.shape,phi.shape print 'time:', time sys.stdout = mystdout = StringIO() pp = PdfPages(save_path+key+'lam.pdf') fig = plt.figure() lam_history = Frame(lam,meta={'dx':tchunk,'stationary':False,'fontsz':18,'ylabel':'', 'xlabel':r'$t$','ticksize':14,'title':r'$\lambda$','xlabel':r't'}) lam_history.ax = fig.add_subplot(111) lam_history.ax.set_ylim([0.0,100.*np.round((lam_history[-1]+100.)/100.)]) lam_history.render(fig,111) print lam_history.x fig.savefig(pp,format='pdf') plt.close(fig) pp.close() pp = PdfPages(save_path+'/'+key+'compare_lam.pdf') fig = plt.figure() lam_history.ax = None lam_history.yscale='linear' #lam_history = Frame(lam,meta={'stationary':False,'title':'','fontsz':18,'ylabel':'','xlabel':r'$t$', # 'ticksize':14,'yscale':'linear'})