def show(self,filter =True,quick=False,pdfname='output2.pdf',debug=False,spectrum_movie=False): colors = ['b','g','r','c','m','y','k','b','g','r','c','m','y','k'] pp = PdfPages('output.pdf') #start by removing modes above the maxN threshold modelist =[] [modelist.append(list(self.modeid[p])) for p in range(self.nmodes) if self.mn[p][1] <= self.maxN[p] ] s = subset(self.db,'modeid',modelist) try: #fig = Figure(figsize=(6,6)) # fig = plt.figure() dz0 = list(set(s.dz).union())[0] ss = subset(s.db,'dz',[dz0]) # show initial condition and the first step after s.plotvsK(pp,yscale='log',xscale='log',t=[0,1,-1],overplot=False,comp='amp',trans=True) if spectrum_movie: ss.savemovie() except: print 'no scatter' #2D true NM spectrum with color code and boxes around spectral res regions log scale plt.figure() i = 0 for j in list(set(s.dz).union()): #looping over runs, over unique 'dz' key values ss = subset(s.db,'dz',[j]) #subset where dz = j plt.scatter(ss.MN[:,1],ss.MN[:,0],c=colors[i]) plt.annotate(str(j),(ss.MN[0,1],ss.MN[0,0])) i+=1 plt.title(' Ni spectrum at t=0, all x') plt.ylabel('M -parallel') plt.xlabel('N - axisymmteric') plt.xscale('log') plt.grid(True,linestyle='-',color='.75') try: plt.savefig(pp, format='pdf') except: print 'FAILED TO save 1st part' plt.close() # for elem in self.meta['evolved']['v']: # s.plotnl(pp if self.meta['nonlinear']['v'] == 'true': self.plotnlrhs(pp) if self.meta['transport'] == 'true': self.plotnlrms(pp) for elem in self.meta['evolved']: s.plotmodes(pp,yscale='symlog',comp='phase',linestyle='.',field=elem,summary=False) s.plotmodes(pp,yscale='symlog',field=elem,summary=False) print elem try: s.plotmodes(pp,yscale='symlog',field=elem,comp='gamma_i',summary=False) except: print 'gamma_i plot for '+elem+' failed' #s.plotmodes(pp,yscale='symlog',summary=False) modelist = [] # maxZ = #[modelist.append([1,p+1]) for p in range(maxZ-1)] [modelist.append(list(self.modeid[p])) for p in range(self.nmodes) if self.mn[p][1] <= self.maxN[p] ] ss = subset(s.db,'mn',modelist) if debug: #just a few problematic slides fig1 = plt.figure() pp_bug = PdfPages('debug.pdf') #ss.plotmodes(pp_bug,yscale='symlog',comp='phase',summary=False) s.plotfreq2(pp_bug,xscale='log',yscale='symlog',overplot=True) ss.plotgamma(pp_bug,xscale='log',yscale='symlog',overplot=True) ss.plottheory(pp_bug) ss.plottheory(pp_bug,comp='freq') fig1.savefig(pp_bug, format='pdf') pp_bug.close() pp.close() return 0 dir(ss) ss.plotmodes(pp,yscale='log',debug=True,summary=False) ss.plotmodes(pp,yscale='symlog',comp='phase',summary=False) ss.plotmodes(pp,yscale='symlog',comp='phase',field='rho',summary=False) print dir(ss) #ss.plotmodes(pp,yscale='log',comp='phase',clip=True) #ss.plotfreq2(pp,xscale='log',yscale='linear',overplot=False) for elem in self.meta['evolved']: ss.plotfreq2(pp,xscale='log',yscale='symlog',field=elem, overplot=True,trans=True) #ss.plotfreq2(pp,xscale='log',yscale='symlog',field='rho',overplot=True) if quick==True: pp.close() s.printmeta(pp) #plt.savefig(pp, format='pdf') return 0 all_fields = list(set(s.field).union()) s.plotgamma(pp,xscale='log',yscale='linear',overplot=True,trans=True) s.plotgamma(pp,yscale='symlog',xscale='log',overplot=True) s.plotgamma(pp,yscale='symlog',xscale='log',field='rho',overplot=True) try: s.plotfreq2(pp,xscale='log',yscale='linear',overplot=True) #s.plotfreq2(pp,xscale='log',yscale='symlog',overplot=False) s.plotfreq2(pp,xscale='log',yscale='symlog',field='rho',overplot=True) #s.plotfreq2(pp,xscale='log',yscale='linear') except: print 'something terrible' s.plotradeigen(pp,yscale='linear') #s.plotradeigen(pp,field ='Vi',yscale='linear') s.plotradeigen(pp,field='rho',yscale='log') pp.close() s.printmeta(pp,filename=pdfname) #append a metadata header
def show(self, filter=True, quick=False, pdfname='output2.pdf', debug=False, spectrum_movie=False): colors = [ 'b', 'g', 'r', 'c', 'm', 'y', 'k', 'b', 'g', 'r', 'c', 'm', 'y', 'k' ] pp = PdfPages('output.pdf') #start by removing modes above the maxN threshold modelist = [] [ modelist.append(list(self.modeid[p])) for p in range(self.nmodes) if self.mn[p][1] <= self.maxN[p] ] s = subset(self.db, 'modeid', modelist) try: #fig = Figure(figsize=(6,6)) # fig = plt.figure() dz0 = list(set(s.dz).union())[0] ss = subset(s.db, 'dz', [dz0]) # show initial condition and the first step after s.plotvsK(pp, yscale='log', xscale='log', t=[0, 1, -1], overplot=False, comp='amp', trans=True) if spectrum_movie: ss.savemovie() except: print 'no scatter' #2D true NM spectrum with color code and boxes around spectral res regions log scale plt.figure() i = 0 for j in list(set(s.dz).union() ): #looping over runs, over unique 'dz' key values ss = subset(s.db, 'dz', [j]) #subset where dz = j plt.scatter(ss.MN[:, 1], ss.MN[:, 0], c=colors[i]) plt.annotate(str(j), (ss.MN[0, 1], ss.MN[0, 0])) i += 1 plt.title(' Ni spectrum at t=0, all x') plt.ylabel('M -parallel') plt.xlabel('N - axisymmteric') plt.xscale('log') plt.grid(True, linestyle='-', color='.75') try: plt.savefig(pp, format='pdf') except: print 'FAILED TO save 1st part' plt.close() # for elem in self.meta['evolved']['v']: # s.plotnl(pp if self.meta['nonlinear']['v'] == 'true': self.plotnlrhs(pp) if self.meta['transport'] == 'true': self.plotnlrms(pp) for elem in self.meta['evolved']: s.plotmodes(pp, yscale='symlog', comp='phase', linestyle='.', field=elem, summary=False) s.plotmodes(pp, yscale='symlog', field=elem, summary=False) print elem try: s.plotmodes(pp, yscale='symlog', field=elem, comp='gamma_i', summary=False) except: print 'gamma_i plot for ' + elem + ' failed' #s.plotmodes(pp,yscale='symlog',summary=False) modelist = [] # maxZ = #[modelist.append([1,p+1]) for p in range(maxZ-1)] [ modelist.append(list(self.modeid[p])) for p in range(self.nmodes) if self.mn[p][1] <= self.maxN[p] ] ss = subset(s.db, 'mn', modelist) if debug: #just a few problematic slides fig1 = plt.figure() pp_bug = PdfPages('debug.pdf') #ss.plotmodes(pp_bug,yscale='symlog',comp='phase',summary=False) s.plotfreq2(pp_bug, xscale='log', yscale='symlog', overplot=True) ss.plotgamma(pp_bug, xscale='log', yscale='symlog', overplot=True) ss.plottheory(pp_bug) ss.plottheory(pp_bug, comp='freq') fig1.savefig(pp_bug, format='pdf') pp_bug.close() pp.close() return 0 dir(ss) ss.plotmodes(pp, yscale='log', debug=True, summary=False) ss.plotmodes(pp, yscale='symlog', comp='phase', summary=False) ss.plotmodes(pp, yscale='symlog', comp='phase', field='rho', summary=False) print dir(ss) #ss.plotmodes(pp,yscale='log',comp='phase',clip=True) #ss.plotfreq2(pp,xscale='log',yscale='linear',overplot=False) for elem in self.meta['evolved']: ss.plotfreq2(pp, xscale='log', yscale='symlog', field=elem, overplot=True, trans=True) #ss.plotfreq2(pp,xscale='log',yscale='symlog',field='rho',overplot=True) if quick == True: pp.close() s.printmeta(pp) #plt.savefig(pp, format='pdf') return 0 all_fields = list(set(s.field).union()) s.plotgamma(pp, xscale='log', yscale='linear', overplot=True, trans=True) s.plotgamma(pp, yscale='symlog', xscale='log', overplot=True) s.plotgamma(pp, yscale='symlog', xscale='log', field='rho', overplot=True) try: s.plotfreq2(pp, xscale='log', yscale='linear', overplot=True) #s.plotfreq2(pp,xscale='log',yscale='symlog',overplot=False) s.plotfreq2(pp, xscale='log', yscale='symlog', field='rho', overplot=True) #s.plotfreq2(pp,xscale='log',yscale='linear') except: print 'something terrible' s.plotradeigen(pp, yscale='linear') #s.plotradeigen(pp,field ='Vi',yscale='linear') s.plotradeigen(pp, field='rho', yscale='log') pp.close() s.printmeta(pp, filename=pdfname) #append a metadata header
def show(self,filter =True,quick=False,pdfname='output2.pdf',debug=False): colors = ['b','g','r','c','m','y','k','b','g','r','c','m','y','k'] pp = PdfPages('output.pdf') #start by removing modes above the maxN threshold modelist =[] [modelist.append(list(self.modeid[p])) for p in range(self.nmodes) if self.mn[p][1] <= self.maxN[p] ] s = subset(self.db,'modeid',modelist) try: fig = Figure(figsize=(6,6)) a_amp = np.array([s.amp[i][1,int(s.nx/2)] for i in range(s.nmodes)]) plt.scatter(s.mn[:,1],s.mn[:,0],s = 5*s.mn.max()* s._amp(1,int(s.nx/2))/s._amp(1,int(s.nx/2)).mean()) plt.annotate(str(list(s.mn[0,:])),tuple(s.mn[0,:]+.2)) plt.title('n-m spectrum at t=0') plt.xlabel('n') plt.ylabel('m') plt.grid(True,linestyle='-',color='.75') plt.savefig(pp, format='pdf') plt.close() except: print 'no scatter' #2D true NM spectrum with color code and boxes around spectral res regions log scale plt.figure() i = 0 for j in list(set(s.dz).union()): #looping over runs, over unique 'dz' key values ss = subset(s.db,'dz',[j]) #subset where dz = j plt.scatter(ss.MN[:,1],ss.MN[:,0],c=colors[i]) plt.annotate(str(j),(ss.MN[0,1],ss.MN[0,0])) i+=1 plt.title(' Ni spectrum at t=0, all x') plt.ylabel('M -parallel') plt.xlabel('N - axisymmteric') plt.xscale('log') plt.grid(True,linestyle='-',color='.75') try: plt.savefig(pp, format='pdf') except: print 'FAILED TO save 1st part' plt.close() for elem in self.meta['evolved']['v']: s.plotmodes(pp,yscale='symlog',comp='phase',linestyle='.',field=elem,summary=False) s.plotmodes(pp,yscale='symlog',field=elem,summary=False) #s.plotmodes(pp,yscale='symlog',summary=False) modelist = [] [modelist.append([1,p+1]) for p in range(7)] ss = subset(s.db,'mn',modelist) if debug: #just a few problematic slides fig1 = plt.figure() pp_bug = PdfPages('debug.pdf') #ss.plotmodes(pp_bug,yscale='symlog',comp='phase',summary=False) s.plotfreq2(pp_bug,xscale='log',yscale='symlog',overplot=True) ss.plotgamma(pp_bug,xscale='log',yscale='symlog',overplot=True) ss.plottheory(pp_bug) ss.plottheory(pp_bug,comp='freq') fig1.savefig(pp_bug, format='pdf') pp_bug.close() pp.close() return 0 ss.plotmodes(pp,yscale='log',debug=True,summary=False) ss.plotmodes(pp,yscale='symlog',comp='phase',summary=False) ss.plotmodes(pp,yscale='symlog',comp='phase',field='rho',summary=False) #ss.plotmodes(pp,yscale='log',comp='phase',clip=True) #ss.plotfreq2(pp,xscale='log',yscale='linear',overplot=False) for elem in self.meta['evolved']['v']: ss.plotfreq2(pp,xscale='log',yscale='symlog',field=elem, overplot=True,trans=True) #ss.plotfreq2(pp,xscale='log',yscale='symlog',field='rho',overplot=True) if quick==True: pp.close() s.printmeta(pp) #plt.savefig(pp, format='pdf') return 0 all_fields = list(set(s.field).union()) s.plotgamma(pp,xscale='log',yscale='linear',overplot=True,trans=True) s.plotgamma(pp,yscale='symlog',xscale='log',overplot=True) s.plotgamma(pp,yscale='symlog',xscale='log',field='rho',overplot=True) try: s.plotfreq2(pp,xscale='log',yscale='linear',overplot=True) s.plotfreq2(pp,xscale='log',yscale='symlog',overplot=False) s.plotfreq2(pp,xscale='log',yscale='symlog',field='rho',overplot=True) s.plotfreq2(pp,xscale='log',yscale='linear') except: print 'something terrible' # s.plotradeigen(pp,yscale='linear') # s.plotradeigen(pp,field ='Vi',yscale='linear') # s.plotradeigen(pp,field='rho',yscale='log') pp.close() s.printmeta(pp,filename=pdfname) #append a metadata header