Homec4science
Diffusion Gear (master) Browse

Gear/PyCool/scripts/pycool_cooling_wiersma_plot_cooling_vs_Temperature
1a49d5806c83master

Last Change
Raw File

pycool_cooling_wiersma_plot_cooling_vs_Temperature
View Options

#!/usr/bin/env python
import PyCool.CoolingWiersma as CoolingTables
from numpy import *
import sys
from optparse import OptionParser
import Ptools as pt
def parse_options():
usage = "usage: %prog [options] file"
parser = OptionParser(usage=usage)
parser = pt.add_postscript_options(parser)
parser = pt.add_limits_options(parser)
parser = pt.add_log_options(parser)
parser.add_option("--x",
action="store",
dest="x",
type="string",
default = 'r',
help="x value to plot",
metavar=" STRING")
parser.add_option("--y",
action="store",
dest="y",
type="string",
default = 'T',
help="y value to plot",
metavar=" STRING")
parser.add_option("--z",
action="store",
dest="z",
type="string",
default = None,
help="z value to plot",
metavar=" STRING")
parser.add_option("--legend",
action="store_true",
dest="legend",
default = False,
help="add a legend")
parser.add_option("--coolingdir",
action="store",
dest="coolingdir",
default = "/home/epfl/revaz/code/gear/PyCool/tables_wiersma/coolingtables/",
help="cooling directory")
parser.add_option("--HeMfaction",
action="store",
dest="HeMfaction",
type="float",
default = 0.2806,
help="Helium mass fraction",
metavar=" FLOAT")
parser.add_option("--nH",
action="store",
dest="nH",
type="float",
default = 1e-3,
help="Hydrogen density",
metavar=" FLOAT")
parser.add_option("--Z",
action="store",
dest="Z",
type="float",
default = 0.02,
help="Metallicity",
metavar=" FLOAT")
parser.add_option("--Redshift",
action="store",
dest="Redshift",
type="float",
default = 0.0,
help="Redshift",
metavar=" FLOAT")
parser.add_option("--Temperature",
action="store",
dest="Temperature",
type="float",
default = 1e5,
help="Temperature",
metavar=" FLOAT")
(options, args) = parser.parse_args()
#pt.check_files_number(args)
files = args
return files,options
########################################################################
# MAIN
########################################################################
files,opt = parse_options()
FeHSolar=0.00176604
Zsol =0.02
HeMfaction = opt.HeMfaction
nH = opt.nH
Z = opt.Z;
Redshift=opt.Redshift
CoolingTables.Init(opt.coolingdir)
CoolingTables.setTablesFromRedshift(Redshift)
Ts = 10**arange(-1,9,0.01)
Lms = zeros(len(Ts))
Lfs = zeros(len(Ts))
Ls = zeros(len(Ts))
for i,T in enumerate(Ts):
Ls[i] = CoolingTables.compute_LambdaTotal(T,nH,HeMfaction,Z,Zsol)
########################################################################
# plot
########################################################################
pt.xlabel('log T')
pt.ylabel('log Lambda/nH2')
pt.axis([1,8,-26,-20])
# total metal
#pt.plot(Ts,Lms)
#pt.plot(h5Ts,h5Lms)
# metal free
pt.plot(Ts,fabs(Ls),label="total")
pt.axis([10**2,10**9,10**-26,10**-21])
pt.loglog()
pt.legend(loc=4)
pt.show()
c4science · Help