# @param gp global parameters
if __name__ == '__main__':
parser = OptionParser()
parser.add_option("-i", "--investigation", dest="investigate", default="", help="investigation to plot")
parser.add_option("-c", "--case", dest="case", default=-1, help="case to plot")
parser.add_option("-l", "--latest", help='plot latest one', dest = 'latest', default = False, action = 'store_true')
parser.add_option("-t", "--timestamps", help='take all models from following timestamps', dest='timestamps', default=[], action='append')
# parser.add_option("-t", "--timestamp", dest="timestamp", default=-1, help="timestamp of run to plot. Overrides -l")
# parser.add_option("-a", "--action", dest="action", default="p", help="action to take: p: print, k: kill")
(options, args) = parser.parse_args()
# gh.LOG(1, 'plot_profiles '+str(options.investigate)+' '+str(options.case)+' '+str(options.latest))
if len(options.timestamps) == 0:
import select_run as sr
timestamp, basename = sr.run(options.investigate, options.case, options.latest)
options.timestamps=[timestamp]
else:
import gi_base as gb
basepath = gb.get_basepath()
basename = os.path.abspath(basepath+'DT'+options.investigate+'/'+str(options.case))+'/'
# use last timestamp for sample gp
import import_path as ip
tt = options.timestamps[-1]
ip.insert_sys_path(basename+tt+'/programs/')
ip.insert_sys_path(basename+tt+'/programs/sphere')
import gi_params as gip
gp = gip.Params(tt)
gp.pops = sr.get_pops(basename+tt+'/')
if prof=='rho' or prof=='nu':
gpl.startlog(); gpl.xscale('log')
else:
gpl.start()
plot_labels(prof, pop)
if prof=='nr':
gpl.xscale('log')
# gpl.xlim([0.,2.*gp.xipol[-1]])
gp.xipol = np.hstack([gp.xipol[0]/2., gp.xipol, gp.rinfty*gp.xipol[-1]])
if prof=='nu' or prof=='sig':
plot_data(gp.xipol, prof, pop)
plot_analytic(gp.xepol, prof, pop)
radfill = gp.xepol
if prof == 'sig':
radfill = gp.xipol
fill_nice(radfill, M95lo, M68lo, Mmedi, M68hi, M95hi)
gpl.savefig(basename+'/prof_'+prof+'_'+str(pop)+'.png')
gpl.ion(); gpl.show(); gpl.ioff()
return
## \fn run()
# main functionality
if __name__=="__main__":
import select_run
basename, prof, pop = select_run.run()
run(basename, prof)
gpl.show()
#!/usr/bin/python # (c) 2013 Pascal Steger, [email protected] import numpy as np from pylab import * import pdb from scipy.interpolate import Rbf, InterpolatedUnivariateSpline from gl_analytic import Mwalkertot, rhowalkertot_3D from matplotlib.backends.backend_pdf import PdfPages # Walker data sets import select_run basename, prof = select_run.run() def M_anf(r): return r**2/(r+1.)**2 def ipol(xin,yin,xout,smooth=1.e-9): rbf = Rbf(xin, yin, smooth=smooth) return rbf(xout) def ipollog(xin,yin,xout,smooth=1.e-9): rbf = Rbf(xin, np.log10(yin), smooth=smooth) return 10**(rbf(xout)) def setlabels(ax,xtext,ytext): ax.set_xlabel(r'$'+xtext+'$') ax.set_ylabel(r'$'+ytext+'$') def setlims(ax,xlim,ylim): ax.set_xlim(xlim)
