Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F121376015
gmasses
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
Thu, Jul 10, 09:48
Size
3 KB
Mime Type
text/x-python
Expires
Sat, Jul 12, 09:48 (2 d)
Engine
blob
Format
Raw Data
Handle
27294724
Attached To
rGTOOLS Gtools
gmasses
View Options
#!/usr/bin/env python
'''
Extract and plot masses values contained in the
output Gadget file called by default "energy.txt".
Yves Revaz
mar sep 26 10:16:33 CEST 2006
'''
from
numarray
import
*
from
Nbody
import
*
import
SM
import
string
import
sys
import
os
from
Nbody.libutil
import
histogram
from
optparse
import
OptionParser
from
Gtools
import
*
from
Gtools
import
io
def
parse_options
():
usage
=
"usage: %prog [options] file"
parser
=
OptionParser
(
usage
=
usage
)
parser
=
add_postscript_options
(
parser
)
parser
=
add_color_options
(
parser
)
parser
=
add_limits_options
(
parser
)
parser
=
add_log_options
(
parser
)
parser
=
add_units_options
(
parser
)
parser
.
add_option
(
"-o"
,
action
=
"store"
,
dest
=
"obs"
,
type
=
"string"
,
default
=
'MassComp5'
,
help
=
"observable name"
,
metavar
=
" NAME"
)
parser
.
add_option
(
"--relative"
,
action
=
"store_true"
,
dest
=
"relative"
,
default
=
0
,
help
=
"plot relative value"
)
(
options
,
args
)
=
parser
.
parse_args
()
if
options
.
colors
!=
None
:
exec
(
"options.colors = array([
%s
])"
%
(
options
.
colors
))
if
len
(
args
)
==
0
:
print
"you must specify a filename"
sys
.
exit
(
0
)
files
=
args
return
files
,
options
#############################
# graph
#############################
# get options
files
,
options
=
parse_options
()
ps
=
options
.
ps
col
=
options
.
colors
xmin
=
options
.
xmin
xmax
=
options
.
xmax
ymin
=
options
.
ymin
ymax
=
options
.
ymax
log
=
options
.
log
obs
=
options
.
obs
rel
=
options
.
relative
localsystem
=
Set_SystemUnits_From_Options
(
options
)
#######################################
# define output system of unit
#######################################
outputunits
=
UnitSystem
(
'mks'
,[
Unit_kpc
,
Unit_Ms
,
Unit_yr
,
Unit_K
,
Unit_mol
,
Unit_C
])
#######################################
# open sm
#######################################
g
=
Graph_Init
(
ps
)
Graph_SetDefaultsGraphSettings
(
g
)
colors
=
Graph_SetColorsForFiles
(
files
,
col
)
#######################################
# LOOP
#######################################
# read files
for
file
in
files
:
try
:
vals
=
io
.
read_new_energy
(
file
)
except
"NotNewEgyFileError"
:
vals
=
io
.
read_energy
(
file
,
iobs
=
None
)
x
=
vals
[
'Time'
]
if
obs
==
'MassTot'
:
y
=
0.
nrjvals
=
[
'MassComp1'
,
'MassComp2'
,
'MassComp3'
,
'MassComp4'
,
'MassComp5'
,
'MassComp6'
]
for
nrjval
in
nrjvals
:
if
vals
.
has_key
(
nrjval
):
print
nrjval
y
=
y
+
vals
[
nrjval
]
elif
vals
.
has_key
(
obs
):
y
=
vals
[
obs
]
else
:
print
"unknown observable
%s
"
%
(
obs
)
sys
.
exit
()
# output units
FACT
=
PhysCte
(
1
,
localsystem
.
UnitDic
[
'kg'
])
FACT
=
FACT
.
into
(
outputunits
)
y
=
y
*
FACT
if
rel
:
y
=
100
*
(
fabs
(
y
-
y
[
0
]))
/
y
[
0
]
y
=
fabs
(
y
)
# use log
if
log
!=
None
:
x
,
y
=
Graph_UseLog
(
x
,
y
,
log
)
if
file
==
files
[
0
]:
xmin
,
xmax
,
ymin
,
ymax
=
Graph_SetLimits
(
g
,
xmin
,
xmax
,
ymin
,
ymax
,
x
,
y
)
g
.
box
()
# plot points
g
.
ctype
(
colors
[
file
])
g
.
connect
(
x
,
y
)
# labels
g
.
ctype
(
0
)
g
.
xlabel
(
'T'
)
g
.
ylabel
(
'
%s
'
%
obs
)
g
.
ctype
(
0
)
if
log
==
'xy'
or
log
==
'yx'
:
g
.
xlabel
(
'log T'
)
g
.
ylabel
(
'log
%s
'
%
obs
)
elif
log
==
'x'
:
g
.
xlabel
(
'log T'
)
g
.
ylabel
(
'
%s
'
%
obs
)
elif
log
==
'y'
:
g
.
xlabel
(
'T'
)
g
.
ylabel
(
'log
%s
'
%
obs
)
else
:
g
.
xlabel
(
'T'
)
g
.
ylabel
(
'
%s
'
%
obs
)
# -- end ---
Graph_End
(
g
,
ps
)
Event Timeline
Log In to Comment