erosita/coma
1
0
Fork 0
forked from erosita/uds
Code Pull Requests Activity

transfer

Browse Source
This commit is contained in:
Roman Krivonos
2024-11-14 11:28:13 +03:00
parent 13046181ea
commit b276a4e533
4 changed files with 56 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/01_init_events.py
View File

@@ -21,7 +21,10 @@ el=root_path+'/data/evtlists/'
pr=root_path+'/data/processed/'
create_folder(pr)
region="box({},{},8d,8d,0)".format(ra_cen,de_cen) # Selection region
region="box({},{},12d,12d,0)".format(ra_cen,de_cen) # Selection region
for m in [1,2,3,4,5,6,7]:
print("tm{}_".format(m))
@@ -57,9 +60,11 @@ for i in range(len(partII)-1):
do_evtool_esass(evlist=el+'tm{}_partII.txt'.format(m), outfile=pr+'tm{}_scan{}.fits'.format(m,scan), gti='{} {}'.format(tstart,tstop), emin=0.2, emax=10.0, region=region, rmlock=True)
scan=scan+1
"""
### Survey ###
for m in [1,2,3,4,5,6,7]:
do_evtool_esass(evlist=el+'tm{}_survey.txt'.format(m), outfile=pr+'tm{}_survey.fits'.format(m), emin=0.2, emax=10.0, region=region, rmlock=True)
do_evtool_esass(evlist=el+'tm{}_survey.txt'.format(m), outfile=pr+'tm{}_survey.fits'.format(m), emin=0.2, emax=10.0, rmlock=True)
"""

44
scripts/02_merge_events.py
View File

@@ -28,48 +28,54 @@ ftools does not like long file path names, for this reason, we use relative path
root_path='..'
print("UDS root path: {}".format(root_path))
print("Coma root path: {}".format(root_path))
infile_dir=root_path+'/data/processed'
outfile_dir=root_path+'/products'
create_folder(outfile_dir)
index=5 # select energy band
do_init = True
do_merge = False
do_init = False
do_merge = True
do_rate = False
do_adapt = False # requires CIAO
vign=False
index=5 # select energy band
width=7000
vign=True
vignetting = 'vign' if (vign==True) else 'novign'
#print(ra_cen, de_cen)
#sys.exit()
events=[]
expmaps=[]
bkgmaps=[]
for tmkey in keylist_tm.keys():
print("TM{} in work... init events".format(tmkey))
for datakey in keylist_tm[tmkey]:
#if not ("scan" in datakey):
# continue
if ("survey" in datakey):
continue
print("--> {}".format(datakey))
""" Подготавливаем списки событий индивидуальных наблюдений """
outfile_evtool,outfile_expmap=init_events(key=datakey, eband_index=eband[index],
infile_dir=infile_dir,
outfile_dir=outfile_dir,
do_init=do_init,
do_obsmode=False,
do_center=False,
do_obsmode=True, # also controlled by lock file!
do_center=True, # also controlled by lock file!
do_evtool=True,
do_expmap=True,
vign=vign,
ra_cen=ra_cen, de_cen=de_cen,
ra_cen=ra_cen, de_cen=de_cen, width=width,
emin_kev=emin_kev[index],
emax_kev=emax_kev[index])
events.append(outfile_evtool)
expmaps.append(outfile_expmap)
bkgmaps.append("{}_BackMap3_en{}.fits".format(os.path.join(outfile_dir,datakey), eband[0]))
""" Собираем общий список событий """
outfile_evtool="{}_EventList_en{}.fits".format(os.path.join(outfile_dir,outkey), eband[index])
@@ -78,9 +84,21 @@ outfile_expmap="{}_ExposureMap_en{}.{}.fits".format(os.path.join(outfile_dir,out
outfile_bkgmap="{}_BackMap_en{}.{}.fits".format(os.path.join(outfile_dir,outkey), eband[index], vignetting)
if(do_merge==True):
do_evtool_esass(events=events, outfile=outfile_evtool)
do_fimgmerge_ftools(maps=expmaps, outfile=outfile_expmap)
do_fimgmerge_ftools(maps=bkgmaps, outfile=outfile_bkgmap)
evlist="{}.evlist.txt".format(os.getpid())
f = open(evlist, "w")
for s in events:
f.write("{}\n".format(s))
f.close()
do_evtool_esass(evlist=evlist, outfile=outfile_evtool, width=width)
# old version, does not work with long list of files
#do_evtool_esass(events=events, outfile=outfile_evtool)
#do_fimgmerge_ftools(maps=expmaps, outfile=outfile_expmap)
#do_fimgmerge_ftools(maps=bkgmaps, outfile=outfile_bkgmap)
if(os.path.isfile(evlist)==True):
os.remove(evlist)
outfile_rate="{}_RateMap_en{}.{}.fits".format(os.path.join(outfile_dir,outkey), eband[index], vignetting)
if(do_rate==True):