Homec4science
Diffusion Gear (master) Browse

Gear/PyCool/scripts/pycool_cooling_with_metals_plot_cooling_vs_Temperature
dfd738c8ff58master

Last Change
Raw File

pycool_cooling_with_metals_plot_cooling_vs_Temperature
View Options

#!/usr/bin/env python
from PyCool import cooling
from numpy import *
import sys
from optparse import OptionParser
import Ptools as pt
ZHsol = 0.02 # (MZ/MH)_sun
FeHsol = 0.00176604 # (MFe/MH)_sun
def parse_options():
usage = "usage: %prog [options] file"
parser = OptionParser(usage=usage)
parser = pt.add_postscript_options(parser)
parser.add_option("--xmin",
action="store",
dest="xmin",
type="float",
default=1e1,
help="min value in x")
parser.add_option("--xmax",
action="store",
dest="xmax",
type="float",
default=1e9,
help="max value in x")
parser.add_option("--ymin",
action="store",
dest="ymin",
type="float",
default=1e-26,
help="min value in y")
parser.add_option("--ymax",
action="store",
dest="ymax",
type="float",
default=1e-20,
help="max value in y")
parser.add_option("--log",
action="store",
dest="log",
type="string",
default='xy',
help="Use log to plot values (x,y,xy)")
parser.add_option("--legend",
action="store_true",
dest="legend",
default = False,
help="add a legend")
parser.add_option("--FeHSolar",
action="store",
dest="FeHSolar",
type="float",
default = FeHsol,
help="Fe solar abundance",
metavar=" FLOAT")
parser.add_option("--ZHSolar",
action="store",
dest="ZHSolar",
type="float",
default = ZHsol,
help="Z solar abundance",
metavar=" FLOAT")
parser.add_option("--Z",
action="store",
dest="Z",
type="float",
default = ZHsol,
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")
parser.add_option("--Density",
action="store",
dest="Density",
type="float",
default = 100,
help="Density [a/cc]",
metavar=" FLOAT")
parser.add_option("--UV",
action="store",
dest="UV",
type="int",
default = 1,
help="UV on/off",
metavar=" FLOAT")
parser.add_option("--Coolingfile",
action="store",
dest="Coolingfile",
default = "cooling.dat",
help="cooling file")
(options, args) = parser.parse_args()
#pt.check_files_number(args)
files = args
return files,options
########################################################################
# MAIN
########################################################################
files,opt = parse_options()
param = {}
param["CoolingType"]=2
param["CutofCoolingTemperature"]=1e1
param["InitGasMetallicity"]=-10
param["CoolingParameters_FeHSolar"]=opt.FeHSolar
param["CoolingParameters_ZHSolar"] =opt.ZHSolar
param["HubbleParam"]=1.0
files,opt = parse_options()
# init cooling using cooling file
cooling.init_cooling(opt.Coolingfile,param)
# init cooling
cooling.init_cooling_with_metals()
# get FeHSolar, ZHSolar
FeHSolar = cooling.get_FeHSolar_from_table_with_metals()
ZHSolar = cooling.get_ZHSolar_from_table_with_metals()
# define MFe/MH from Z (simple scaling)
Metal = opt.Z/ZHSolar * FeHSolar
print "[Fe/H] = %g"%(log10((Metal/FeHsol)))
# initialize temperature and coolingfct
Ts = 10**arange(1,9,0.02)
Ls = zeros(len(Ts))
for i,T in enumerate(Ts):
Ls[i] = cooling.cooling_function_with_metals(T,Metal)
########################################################################
# plot
########################################################################
pt.xlabel('$T$ [K]')
pt.ylabel('$\\Lambda_n$ [erg s$^{-1}$ cm$^{3}$]')
pt.plot(Ts,fabs(Ls),label="total cooling")
pt.SetAxis(opt.xmin,opt.xmax,opt.ymin,opt.ymax,log=opt.log)
#pt.axis([10**2,10**9,10**-26,10**-20])
#pt.loglog()
pt.legend(loc=4)
pt.show()
c4science · Help