Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F100635307
pplotdmp
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Sat, Feb 1, 08:54
Size
9 KB
Mime Type
text/x-python
Expires
Mon, Feb 3, 08:54 (2 d)
Engine
blob
Format
Raw Data
Handle
24004113
Attached To
rGTOOLS Gtools
pplotdmp
View Options
#!/usr/bin/env python
from optparse import OptionParser
import Ptools as pt
from pNbody import *
from pNbody import units
from pNbody import ctes
from pNbody import thermodyn
import string
from scipy import optimize
def parse_options():
usage = "usage: %prog [options] file"
parser = OptionParser(usage=usage)
parser = pt.add_postscript_options(parser)
parser = pt.add_limits_options(parser)
parser = pt.add_log_options(parser)
parse = pt.add_units_options(parser)
parser.add_option("--x",
action="store",
dest="x",
type="string",
default = 'r',
help="x value to plot",
metavar=" STRING")
parser.add_option("--y",
action="store",
dest="y",
type="string",
default = 'T',
help="y value to plot",
metavar=" STRING")
parser.add_option("--z",
action="store",
dest="z",
type="string",
default = None,
help="z value to plot",
metavar=" STRING")
parser.add_option("--legend",
action="store_true",
dest="legend",
default = False,
help="add a legend")
parser.add_option("--colorbar",
action="store_true",
dest="colorbar",
default = False,
help="add a colorbar")
parser.add_option("--nopoints",
action="store_true",
dest="nopoints",
default = False,
help="do not plot points")
parser.add_option("--map",
action="store_true",
dest="map",
default = False,
help="plot map instead of points")
parser.add_option("--accumulate",
action="store_true",
dest="accumulate",
default = False,
help="integrate histogramm")
parser.add_option("--data",
action="store",
type="string",
dest="data",
default = None,
help="data")
parser.add_option("--mode",
action="store",
type="string",
dest="mode",
default = 'Velocities',
help="mode")
(options, args) = parser.parse_args()
pt.check_files_number(args)
files = args
return files,options
def get_value_and_label(vals,val):
if val == 'R':
label = r'$\rm{Radius}\,\left[ kpc \right]$'
out_units = units.UnitSystem('local',[units.Unit_kpc,units.Unit_Ms,units.Unit_Myr,units.Unit_K])
val = vals['R']
return val,label
if val == 'vc':
label = r'$\rm{Radius}\,\left[ kpc \right]$'
out_units = units.UnitSystem('local',[units.Unit_kpc,units.Unit_Ms,units.Unit_Myr,units.Unit_K])
val = vals['vc']
return val,label
else:
label = r'%s'%val
print "val = %s"%vals[val]
exec("val = %s"%vals[val])
return val, label
#######################################
# MakePlot
#######################################
def MakePlot(files,opt):
# some inits
colors = pt.Colors(n=len(files))
datas = []
# read files
for file in files:
vals = pt.io.read_dmp(file)
################
# units
################
# define local units
unit_params = pt.do_units_options(opt)
if unit_params!=None:
localsystem_of_units = units.Set_SystemUnits_From_Params(unit_params)
else:
if vals.has_key('localsystem_of_units')==0:
localsystem_of_units=units.cgs
else:
localsystem_of_units = vals['localsystem_of_units']
################
# some info
################
print "---------------------------------------------------------"
localsystem_of_units.info()
print "---------------------------------------------------------"
################
# get values
################
if opt.mode=='Numbers':
xlabel = r'$\rm{Radius}\,\left[ kpc \right]$'
ylabel = r'$\rm{Number}\,\rm{of}\,\rm{points}\,\rm{per}\,\rm{bins}$'
x = vals['R']
y = vals['nm']
data = pt.DataPoints(x,y,color=colors.get(),label=r'nn',tpe='line')
datas.append(data)
if opt.mode=='Dispersions':
xlabel = r'$\rm{Radius}\,\left[ kpc \right]$'
ylabel = r'$\rm{Velocity}\,\left[ km/s \right]$'
x = vals['R']
out_units_y = units.UnitSystem('local',[units.Unit_km,units.Unit_Ms,units.Unit_s,units.Unit_K])
tokms = localsystem_of_units.convertionFactorTo(out_units_y.UnitVelocity)
# vct
y = vals['vct']*tokms
data = pt.DataPoints(x,y,color='k',label=r'$V_{c,tot}$',tpe='line')
datas.append(data)
# vc
#y = vals['vc']*tokms
#data = pt.DataPoints(x,y,color='c',label=r'$V_c$',tpe='line')
#datas.append(data)
# vm
y = vals['vm']*tokms
data = pt.DataPoints(x,y,color='r',label=r'$V_m$',tpe='line')
datas.append(data)
# sr
y = vals['sr']*tokms
data = pt.DataPoints(x,y,color='b',label=r'$\sigma_r$',tpe='line')
datas.append(data)
# sp
y = vals['sp']*tokms
data = pt.DataPoints(x,y,color='y',label=r'$\sigma_\theta$',tpe='line')
datas.append(data)
# sz
y = vals['sz']*tokms
data = pt.DataPoints(x,y,color='g',label=r'$\sigma_z$',tpe='line')
datas.append(data)
if opt.mode=='Velocities':
xlabel = r'$\rm{Radius}\,\left[ kpc \right]$'
ylabel = r'$\rm{Velocity}\,\left[ km/s \right]$'
x = vals['R']
out_units_y = units.UnitSystem('local',[units.Unit_km,units.Unit_Ms,units.Unit_s,units.Unit_K])
tokms = localsystem_of_units.convertionFactorTo(out_units_y.UnitVelocity)
if file==files[0]: # skip black
colors.get()
# vct
y = vals['vct']*tokms
data = pt.DataPoints(x,y,color='k',label=r"%s"%'total',tpe='line')
datas.append(data)
# vc
y = vals['vc']*tokms
data = pt.DataPoints(x,y,color=colors.get(),label=r"%s"%file,tpe='line')
datas.append(data)
if opt.mode=='Q':
xlabel = r'$\rm{Radius}\,\left[ kpc \right]$'
ylabel = r'$Q$'
x = vals['R']
y = vals['Q']
data = pt.DataPoints(x,y,color=colors.get(),label=r"%s"%file,tpe='line')
datas.append(data)
if opt.mode=='Energy':
xlabel = r'$\rm{Radius}\,\left[ kpc \right]$'
ylabel = r'$\rm{Energy}$'
x = vals['R']
out_units_y = units.UnitSystem('local',[units.Unit_km,units.Unit_Ms,units.Unit_s,units.Unit_K])
tokms = localsystem_of_units.convertionFactorTo(out_units_y.UnitVelocity)
# vct
y = vals['vct']**2
data = pt.DataPoints(x,y,color='k',label=r'$V_{c,tot}^2$',tpe='line')
datas.append(data)
# total
y = (vals['vm'])**2 + (vals['sr'])**2 + (vals['sp'])**2 + (vals['sz'])**2 + vals['u']
data = pt.DataPoints(x,y,color='c',label=r'$u+V_m^2+\sigma_r^2+\sigma_\theta^2+\sigma_z^2$',tpe='line')
datas.append(data)
if opt.mode=='srQ1':
xlabel = r'$\rm{Radius}\,\left[ kpc \right]$'
ylabel = r'$\sigma$'
out_units_y = units.UnitSystem('local',[units.Unit_km,units.Unit_Ms,units.Unit_s,units.Unit_K])
tokms = localsystem_of_units.convertionFactorTo(out_units_y.UnitVelocity)
x = vals['R']
y = vals['srQ1'] * tokms
data = pt.DataPoints(x,y,color=colors.get(),label=r"%s"%file,tpe='line')
datas.append(data)
if opt.mode=='lambdaj':
xlabel = r'$\rm{Radius}\,\left[ kpc \right]$'
ylabel = r'$\lambda_{\rm{J}}\,\rm{\left[ kpc \right]}$'
x = vals['R']
y = vals['lambdaj']
data = pt.DataPoints(x,y,color=colors.get(),label=r"%s"%file,tpe='line')
datas.append(data)
norm = None
# now, plot
for d in datas:
if d.tpe=='points' or d.tpe=='both':
if opt.log!=None:
if string.find(opt.log,'z')!=-1:
#norm = pt.colors.LogNorm(opt.zmin,opt.zmax)
norm = pt.colors.LogNorm()
else:
norm = None
cmap = pt.GetColormap('rainbow4',revesed=True)
pt.scatter(d.x,d.y,c=d.color,s=1,linewidths=1,marker='o',vmin=opt.zmin,vmax=opt.zmax,norm=norm,cmap=cmap)
pt.plot(d.x,d.y,c=d.color)
if d.tpe=='line' or d.tpe=='both':
pt.plot(d.x,d.y,color=d.color)
# set limits and draw axis
xmin,xmax,ymin,ymax = pt.SetLimitsFromDataPoints(opt.xmin,opt.xmax,opt.ymin,opt.ymax,datas,opt.log)
# plot axis
pt.SetAxis(xmin,xmax,ymin,ymax,log=opt.log)
pt.xlabel(xlabel,fontsize=pt.labelfont)
pt.ylabel(ylabel,fontsize=pt.labelfont)
pt.grid(False)
if opt.legend:
pt.LegendFromDataPoints(datas)
if opt.colorbar:
#lev = array([-2,0,15])
#l_f = pt.LogFormatter(10, labelOnlyBase=False)
#cb = pt.colorbar(ticks=lev, format = l_f)
cb = pt.colorbar()
cb.set_label(zlabel)
########################################################################
# MAIN
########################################################################
if __name__ == '__main__':
files,opt = parse_options()
pt.InitPlot(files,opt)
pt.pcolors
MakePlot(files,opt)
pt.EndPlot(files,opt)
Event Timeline
Log In to Comment