mirror of
https://codeberg.org/Yael-II/MSc1-Internship-Stellar-Cluster.git
synced 2026-03-15 11:16:27 +01:00
144 lines
6.5 KiB
Python
144 lines
6.5 KiB
Python
"""
|
|
* 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)
|