Homec4science
Diffusion Gear (master) Browse

Gear/PyChem/scripts/pychem_generate_hdf5_parameters
b9a4fa236396master

Last Change
Raw File

pychem_generate_hdf5_parameters
View Options

#!/usr/bin/env python
from optparse import OptionParser
from PyChem import liblifetime
from PyChem import libimf
from PyChem import libSNIa
from PyChem import libSNII
from PyChem import libDYIN
from PyChem import libmetalsSNII
from PyChem import libmetalsSNIa
from PyChem import libmetalsDYIN
from PyChem import libSolarAbundances
import sys
import h5py
import getpass
import time
def parse_options():
usage = "usage: %prog [options] file"
parser = OptionParser(usage=usage)
parser.add_option("-o","--outputfile",
action="store",
dest="outputfile",
type="string",
default = 'mychimie.dat',
help="Output file",
metavar=" NAME")
parser.add_option("-p","--parameterfile",
action="store",
dest="parameterfile",
type="string",
default = 'chimieparam.py',
help="Parameter file",
metavar=" NAME")
(options, args) = parser.parse_args()
files = args
return files,options
###########################################################
# Main
###########################################################
version = "v0.0"
files,opt = parse_options()
f = h5py.File(opt.outputfile,'w')
# compute cmdline
cmdline = ""
for elt in sys.argv:
cmdline = cmdline + " " + elt
cmdline = cmdline[1:]
try:
nelts
except NameError:
nelts=None
######################
# header
######################
headerGroup = f.create_group("Header")
# some attributes
headerGroup.attrs["version"] = version
headerGroup.attrs["author"] = getpass.getuser()
headerGroup.attrs["date"] = time.ctime()
headerGroup.attrs["cmdline"] = cmdline
ListOfDYINZs=None
ListOfDYINYieldsFiles=None
ListOfDYINHeliumCoreFiles=None
execfile(opt.parameterfile)
######################
# data
######################
dataGroup = f.create_group("Data")
# Lifetime
if not 'StellarLifeTimeModel' in locals():
StellarLifeTimeModel="Poirier" # set default model
liblifetime.WriteH5Params(dataGroup,model=StellarLifeTimeModel)
# IMF
libimf.WriteH5Params(dataGroup,Mmin,Mmax,m_s,a_s)
# SNII
print "SNII"
libSNII.WriteH5Params(dataGroup,SNII_Mmin,SNII_Mmax)
# DYIN
print "DYIN"
libDYIN.WriteH5Params(dataGroup,DYIN_Mmin,DYIN_Mmax)
# SNIa
print "SNIa"
libSNIa.WriteH5Params(dataGroup,SNIa_Mpl,SNIa_Mpu,SNIa_a,SNIa_Mdl1,SNIa_Mdu1,SNIa_bb1,SNIa_Mdl2,SNIa_Mdu2,SNIa_bb2)
# Metal
print "SNII Metal"
SNIIelts = libmetalsSNII.WriteH5Params(dataGroup,Mmin,Mmax,m_s,a_s,SNIIYieldsFile,SNIIHeliumCoreFile,elts,nelts,Mmin1SNII,Mmax1SNII,Mmin2SNII,Mmax2SNII,dMSNII,nSNII)
# Metal
print "SNIa Metal"
SNIaelts = libmetalsSNIa.WriteH5Params(dataGroup,SNIaFile,MeanWDMass,elts,nelts)
if (SNIIelts!=SNIaelts):
raise "SNIIelts !=SNIaelts"
elts = SNIaelts
# Metal
print "DYIN Metal"
SNIIelts = libmetalsDYIN.WriteH5Params(dataGroup,Mmin,Mmax,m_s,a_s,DYINYieldsFile,DYINHeliumCoreFile,elts,nelts,Mmin1DYIN,Mmax1DYIN,Mmin2DYIN,Mmax2DYIN,dMDYIN,nDYIN)
if ListOfDYINZs!=None and ListOfDYINYieldsFiles!=None and ListOfDYINHeliumCoreFiles!=None:
print "DYIN Metal Z"
SNIIelts = libmetalsDYIN.WriteH5ParamsWithZ(dataGroup,Mmin,Mmax,m_s,a_s,ListOfDYINYieldsFiles,ListOfDYINHeliumCoreFiles,ListOfDYINZs,elts,nelts,Mmin1DYIN,Mmax1DYIN,Mmin2DYIN,Mmax2DYIN,dMDYIN,nDYIN,ZminDYIN,ZmaxDYIN,nZDYIN)
# SolarAbundances
print "Solar Abundances"
SolarMassAbundances = libSolarAbundances.Get(SolarAbundancesFile,elts)
######################
# data attributes
nelts=len(elts)
# add attribute to the data
dataGroup.attrs["nelts"] = nelts
dataGroup.attrs["elts"] = elts
dataGroup.attrs["MeanWDMass"] = MeanWDMass
dataGroup.attrs["SolarMassAbundances"] = SolarMassAbundances
dataGroup.attrs["SNIIYieldsFile"] = str(SNIIYieldsFile)
dataGroup.attrs["SNIIHeliumCoreFile"] = str(SNIIHeliumCoreFile)
dataGroup.attrs["DYINYieldsFile"] = str(DYINYieldsFile)
dataGroup.attrs["DYINHeliumCoreFile"] = str(DYINHeliumCoreFile)
dataGroup.attrs["SNIaFile"] = str(SNIaFile)
dataGroup.attrs["SolarAbundancesFile"] = str(SolarAbundancesFile)
f.close()
c4science · Help