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'})
