Page MenuHomec4science
Files F81631330

addmetals.py
No OneTemporary
Actions

File Metadata

Created
Sat, Sep 7, 03:24

addmetals.py
View Options

#!/usr/bin/env python
from pNbody import *
from optparse import OptionParser
########################################
#
# parser
#
########################################
def parse_options():
usage = "usage: %prog [options] file"
parser = OptionParser(usage=usage)
parser.add_option("-t",
action="store",
dest="ftype",
type="string",
default = 'gadget',
help="type of the file",
metavar=" TYPE")
parser.add_option("-o",
action="store",
dest="outputfile",
type="string",
default = None,
help="outputfile name",
metavar=" STING")
parser.add_option("--exec",
action="store",
dest="execline",
type="string",
default = None,
help="give command to execute before",
metavar=" STRING")
(options, args) = parser.parse_args()
if len(args) == 0:
print "you must specify a filename"
sys.exit(0)
file = None
else:
file = args
return file,options
################################################################################
#
# MAIN
#
################################################################################
file,opt = parse_options()
nb = Nbody(file,ftype=opt.ftype)
if opt.execline != None:
exec(opt.execline)
NELEMENTS = 5
nb.flag_metals=NELEMENTS
nb.rho = ones(nb.nbody).astype(float32)
nb.rsp = zeros(nb.nbody).astype(float32)
nb.metals = zeros((nb.nbody,NELEMENTS)).astype(float32)
# Fe
Fe = -1 * ones(nb.nbody).astype(float32)
nb.metals[:,0] = nb.ChimieSolarAbundances['Fe'] * (10**Fe - 1.0e-20)
if opt.outputfile!=None:
nb.rename(opt.outputfile)
nb.write()

Event Timeline

c4science · Help