mirror of
https://codeberg.org/Yael-II/MSc1-Internship-Stellar-Cluster.git
synced 2026-03-15 03:16:26 +01:00
Add files via upload
This commit is contained in:
Yaël M.
143
COSMIC/source/CEXE_v3.py
Normal file
143
COSMIC/source/CEXE_v3.py
Normal file
@@ -0,0 +1,143 @@
|
||||
"""
|
||||
* CEXE: COSMIC Extension for X-ray Emission
|
||||
* Version 3
|
||||
@ Yaël Moussouni
|
||||
@ Unistra, P&E, MSc1-MoFP
|
||||
@ Observatory of Strasbourg (Intern)
|
||||
"""
|
||||
|
||||
import numpy as np
|
||||
import os
|
||||
import subprocess as sp
|
||||
from astropy import units as u
|
||||
|
||||
directory = "./output/"
|
||||
|
||||
# SIMPUT (sources)
|
||||
val_Elow = 0.1
|
||||
val_Eup = 15
|
||||
val_NBins = 1000
|
||||
val_logEgrid = "yes"
|
||||
val_Emin = 2
|
||||
val_Emax = 10
|
||||
val_nH = 0.2
|
||||
val_flux_min = 2
|
||||
val_flux_max = 10
|
||||
# SIXTE (instrument)
|
||||
val_ra = 0
|
||||
val_dec = 0
|
||||
val_Exposure = 1000
|
||||
XML_file = "TODO" # TODO !!
|
||||
|
||||
# * COSMIC
|
||||
def open_file_terminal():
|
||||
filelist = np.sort([f for f in os.listdir(directory) if "stars" in f])
|
||||
print("\033[36m"+"Available files:"+"\033[0m")
|
||||
for i in range(len(filelist)):
|
||||
print("\033[36m"+"\t{}: ".format(i+1) + "\033[0m" + filelist[i])
|
||||
j = int(input("\033[32m"+"Select a file number: "+"\033[0m"))-1
|
||||
filename = filelist[j]
|
||||
date = filename[filename.find("COSMIC_")+len("COSMIC_") : filename.find("_stars")]
|
||||
time = filename[filename.find("stars_")+len("stars_") : filename.find("_Myr")]
|
||||
print("")
|
||||
file = np.loadtxt(directory+filename, delimiter=",", comments="#")
|
||||
with open(directory+filename) as f: head = f.readline().strip('\n')[1:]
|
||||
return file, head, date, time
|
||||
|
||||
# * XSpec
|
||||
def xspec_build(date, time):
|
||||
with open(directory+"CEXE_XSPEC_{}_{}_Myr_0.6_0.0.sh".format(date, time), "w+") as xspec_file_0600:
|
||||
xspec_file_0600.write("model phabs*TODO\n")
|
||||
xspec_file_0600.write("phabs:nH>{}\n".format(val_nH))
|
||||
xspec_file_0600.write("TODO:...\n")
|
||||
xspec_file_0600.write("flux {} {}\n".format(val_flux_min, val_flux_max))
|
||||
xspec_file_0600.write("save model {}XSPEC_{}_{}_Myr_0.6_0.0.xcm\n".format(directory, date, time))
|
||||
xspec_file_0600.write("quit")
|
||||
with open(directory+"CEXE_XSPEC_{}_{}_Myr_0.5_0.0.sh".format(date, time), "w+") as xspec_file_0500:
|
||||
xspec_file_0500.write("model phabs*TODO\n")
|
||||
xspec_file_0500.write("phabs:nH>{}\n".format(val_nH))
|
||||
xspec_file_0500.write("TODO:...\n")
|
||||
xspec_file_0500.write("flux {} {}\n".format(val_flux_min, val_flux_max))
|
||||
xspec_file_0500.write("save model {}XSPEC_{}_{}_Myr_0.5_0.0.xcm\n".format(directory, date, time))
|
||||
xspec_file_0500.write("quit")
|
||||
with open(directory+"CEXE_XSPEC_{}_{}_Myr_0.4_1.0.sh".format(date, time), "w+") as xspec_file_0410:
|
||||
xspec_file_0410.write("model phabs*TODO\n")
|
||||
xspec_file_0410.write("phabs:nH>{}\n".format(val_nH))
|
||||
xspec_file_0410.write("TODO:...\n")
|
||||
xspec_file_0410.write("flux {} {}\n".format(val_flux_min, val_flux_max))
|
||||
xspec_file_0410.write("save model {}XSPEC_{}_{}_Myr_0.4_1.0.xcm\n".format(directory, date, time))
|
||||
xspec_file_0410.write("quit")
|
||||
with open(directory+"CEXE_XSPEC_{}_{}_Myr_0.2_0.8.sh".format(date, time), "w+") as xspec_file_0208:
|
||||
xspec_file_0208.write("model phabs*TODO\n")
|
||||
xspec_file_0208.write("phabs:nH>{}\n".format(val_nH))
|
||||
xspec_file_0208.write("TODO:...\n")
|
||||
xspec_file_0208.write("flux {} {}\n".format(val_flux_min, val_flux_max))
|
||||
xspec_file_0208.write("save model {}XSPEC_{}_{}_Myr_0.2_0.8.xcm\n".format(directory, date, time))
|
||||
xspec_file_0208.write("quit")
|
||||
return None
|
||||
|
||||
def xspec_exec(date, time):
|
||||
sp.run("echo \"xspec < {}CEXE_XSPEC_{}_{}_Myr_0.6_0.0.sh - TO EXECUTE\"".format(directory, date, time), shell=True)
|
||||
sp.run("echo \"xspec < {}CEXE_XSPEC_{}_{}_Myr_0.5_0.0.sh - TO EXECUTE\"".format(directory, date, time), shell=True)
|
||||
sp.run("echo \"xspec < {}CEXE_XSPEC_{}_{}_Myr_0.4_1.0.sh - TO EXECUTE\"".format(directory, date, time), shell=True)
|
||||
sp.run("echo \"xspec < {}CEXE_XSPEC_{}_{}_Myr_0.2_0.8.sh - TO EXECUTE\"".format(directory, date, time), shell=True)
|
||||
return None
|
||||
|
||||
# * SIMPUT
|
||||
def simput_build(file, col, date, time):
|
||||
with open(directory+"CEXE_SIMPUT_{}_{}_Myr.sh".format(date, time), "w+") as simput_file:
|
||||
Infiles = ""
|
||||
for i in range(len(file)):
|
||||
star_flux = file[i,col["X_luminosity"]] / (4*np.pi*(file[i,col["dist"]] * u.pc).to(u.cm).value**2)
|
||||
star_ra = file[i,col["ra"]]
|
||||
star_dec = file[i,col["dec"]]
|
||||
star_T_0 = file[i,col["X_temperature_0"]]
|
||||
star_T_1 = file[i,col["X_temperature_1"]]
|
||||
|
||||
Infiles += "{}SIMPUT_{}_{}_Myr_{}.fits,".format(directory, date, time, i)
|
||||
|
||||
simput_file.write("$SIXTE/bin/simputfile Simput={}SIMPUT_{}_{}_Myr_{}.fits ".format(directory, date, time, i))
|
||||
simput_file.write("Src_Name = star_{} ".format(i))
|
||||
simput_file.write("RA = {} Dec = {} ".format(star_ra, star_dec))
|
||||
simput_file.write("srcFlux = {} ".format(star_flux))
|
||||
simput_file.write("Elow={} Eup={} NBins={} logEgrid={} Emin={} Emax={} ".format(val_Elow, val_Eup, val_NBins, val_logEgrid, val_Emin, val_Emax))
|
||||
|
||||
simput_file.write("XSPECFile={}XSPEC_{}_{}_Myr_{:2.01f}_{:2.01f}.xcm".format(directory, date, time, star_T_0, star_T_1))
|
||||
simput_file.write("\n")
|
||||
simput_file.write("$SIXTE/bin/simputmerge ")
|
||||
simput_file.write("Infiles={} ".format(Infiles[:-1]))
|
||||
simput_file.write("Outfile={}SIMPUT_{}_{}_Myr.fits".format(directory, date, time))
|
||||
simput_file.write("\n")
|
||||
return 0
|
||||
|
||||
def simput_exec(date, time):
|
||||
sp.run(["chmod", "a+x", "{}CEXE_SIMPUT_{}_{}_Myr.sh".format(directory, date, time)])
|
||||
sp.run(["echo" ,"{}CEXE_SIMPUT_{}_{}_Myr.sh - TO EXECUTE".format(directory, date, time)])
|
||||
return 0
|
||||
|
||||
def sixte_build(date, time):
|
||||
with open(directory+"CEXE_SIXTE_{}_{}_Myr.sh".format(date, time), "w+") as sixte_file:
|
||||
sixte_file.write("$SIXTE/bin/runsixt ")
|
||||
sixte_file.write("XMLFile = {} ".format(XML_file))
|
||||
sixte_file.write("RA = {} DEC = {} ".format(val_ra, val_dec))
|
||||
# sixte_file.write("Prefix = SIXTE_ ")
|
||||
sixte_file.write("Simput = {}SIMPUT_{}_{}_Myr.fits ".format(directory, date, time))
|
||||
sixte_file.write("EvtFile = {}SIXTE_Event_{}_{}_Myr.fits ".format(directory, date, time))
|
||||
sixte_file.write("Exposure = val_Exposure")
|
||||
sixte_file.write("\n")
|
||||
return 0
|
||||
|
||||
def sixte_exec(date, time):
|
||||
sp.run(["chmod", "a+x", "{}CEXE_SIXTE_{}_{}_Myr.sh".format(directory, date, time)])
|
||||
sp.run(["echo", "{}CEXE_SIXTE_{}_{}_Myr.sh - TO EXECUTE".format(directory, date, time)])
|
||||
return 0
|
||||
|
||||
file, head, date, time = open_file_terminal()
|
||||
col = {head.split(",")[i]:i for i in range(len(head.split(",")))}
|
||||
|
||||
xspec_build(date, time)
|
||||
xspec_exec(date, time)
|
||||
simput_build(file, col, date, time)
|
||||
simput_exec(date, time)
|
||||
sixte_build(date, time)
|
||||
sixte_exec(date, time)
|
||||
Reference in New Issue
Block a user