Homec4science
Diffusion pNbody (master) Browse

pNbody/examples/ic/addmetals/addmetals-gas+stars.py
339ab77f7b61master

Last Change
Raw File

addmetals-gas+stars.py
View Options

#!/usr/bin/env python
from pNbody import *
from optparse import OptionParser
from PyChem import chemistry
########################################
#
# 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")
parser.add_option("-p",
action="store",
dest="chimieparameterfile",
type="string",
default = "chimie.dat",
help="chimie parameter file",
metavar=" STING")
parser.add_option("--NumberOfTables",
action="store",
dest="NumberOfTables",
type="int",
default = 1,
help="NumberOfTables",
metavar=" INT")
parser.add_option("--DefaultTable",
action="store",
dest="DefaultTable",
type="int",
default = 0,
help="DefaultTable",
metavar=" INT")
(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()
#########################
# open nbody file
nb = Nbody(file,ftype=opt.ftype)
if opt.execline != None:
exec(opt.execline)
####################################
# convert disk and bulge to stars1
nb0 = nb.select(0)
nb1 = nb.select(1)
nb2 = nb.select(2)
nb3 = nb.select(3)
nb4 = nb.select(4)
nb5 = nb.select(5)
nb3.set_tpe(1)
nb4.set_tpe(1)
nb5.set_tpe(1)
nb = nb0 + nb3 + nb4 + nb5 + nb2
del nb0,nb1,nb2,nb3,nb4,nb5
nb.init()
#########################
# open chimie file
chemistry.init_chimie(opt.chimieparameterfile,opt.NumberOfTables,opt.DefaultTable)
#########################
# set the chimie extra-header
nb.flag_chimie_extraheader = 1
nb.ChimieNelements = int(chemistry.get_nelts())
nb.ChimieElements = chemistry.get_elts_labels()
nb.ChimieSolarMassAbundances = chemistry.get_elts_SolarMassAbundances()
NELEMENTS = nb.ChimieNelements
nb.flag_metals=NELEMENTS
#########################
# global init
nb.rho = ones(nb.nbody).astype(float32)
nb.rsp = zeros(nb.nbody).astype(float32)
nb.metals = zeros((nb.nbody,NELEMENTS)).astype(float32)
#########################
# variables for the gas
# Fe
Fe = -1 + nb.rxy()*(-1/10.)
Fe = Fe.astype(float32)
metals = nb.ChimieSolarMassAbundances['Fe'] * (10**Fe - 1.0e-20)
nb.metals[:,0] = where(nb.tpe==0, metals,nb.metals[:,0] )
#########################
# variables for the stars
# Fe
Fe = -1.5 + nb.rxy()*(-1/10.)
Fe = Fe.astype(float32)
metals = nb.ChimieSolarMassAbundances['Fe'] * (10**Fe - 1.0e-20)
nb.metals[:,0] = where(nb.tpe==1, metals,nb.metals[:,0] )
#########################
# write outputs
if opt.outputfile!=None:
nb.rename(opt.outputfile)
nb.write()
c4science · Help