Homec4science
Diffusion Gear (master) Browse

Gear/PyChem/scripts/pychem_generate_hdf5_parameters.old.03062013
1a49d5806c83master

Last Change
Raw File

pychem_generate_hdf5_parameters.old.03062013
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 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:]
######################
# 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
execfile(opt.parameterfile)
######################
# data
######################
dataGroup = f.create_group("Data")
# Lifetime
liblifetime.WriteH5Params(dataGroup)
# IMF
libimf.WriteH5Params(dataGroup,Mmin,Mmax,m_s,a_s)
# SNII
print "SNII"
libSNII.WriteH5Params(dataGroup,SNII_Mmin)
# 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,MetalFile,HeliumCoreFile,elts,nelts,Mmin1,Mmax1,Mmin2,Mmax2,dM,n)
# 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(f,Mmin,Mmax,m_s,a_s,DyingStarsFile,elts,nelts,Mmin1,Mmax1,Mmin2,Mmax2,dM,n)
# 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["MetalFile"] = MetalFile
dataGroup.attrs["HeliumCoreFile"] = HeliumCoreFile
dataGroup.attrs["SNIaFile"] = SNIaFile
dataGroup.attrs["SolarAbundancesFile"] = SolarAbundancesFile
f.close()
c4science · Help