def append_fhl(models, bmax,
bin_models, bin_dict, bin_distx, bin_disty, bin_radr, bin_frlog,
snr_models, snr_dict, snr_distx, snr_disty, snr_radr, snr_frlog,
isnr_models, isnr_dict, isnr_distx, isnr_disty, isnr_radr, isnr_frlog,
dist_sigma=3., sig50_thresh=3., eph_thresh=100.):
"""
Append missing models from Fermi high-energy catalog to gammalib model container
:param models: ~gammalib.GModels, gammalib model container
:param bmax: float, maximum latitude of models to include
:param dist_sigma: float, minimum distance in sigma's from pre-existing source to add as new
:param sig50_thresh: float, threshold sigma on significance above 50 GeV to apply
:param eph_thresh: float, minimum energy of detected photons required
:return: result: dictionary
"""
# filter FHL table based on latitude and hardness
# hard sources are seleted based on significance above 50 GeV and highest photon energy
# calculate significance above 50 GeV
sig50 = np.sqrt(np.sum(np.power(fhl['Sqrt_TS_Band'][:,2:], 2),axis=1))
# filter
fsources = fhl[(np.abs(fhl['GLAT']) <= bmax) & (sig50 >= sig50_thresh) & (
fhl['HEP_Energy'] >= eph_thresh)]
# prepare new gammalib model container
newpt = 0
newext = 0
n_bin_del = 0
n_snr_del = 0
n_isnr_del = 0
newmodels = gammalib.GModels()
# keep track also of artificial cutoffs
ecut_pwn = []
ecut_snr = []
ecut_unid = []
ecut_agn = []
n_ecut_pwn = 0
n_ecut_snr = 0
n_ecut_agn = 0
n_ecut_unid = 0
msg = ''
# loop over fermi sources
for fsource in fsources:
# assume source is new
new = 1
# fermi source position as gammalib.GSkyDir
fdir = gammalib.GSkyDir()
ra = np.double(fsource['RAJ2000'])
dec = np.double(fsource['DEJ2000'])
fdir.radec_deg(ra, dec)
######################################### treat pointlike sources for Fermi ###########
if fsource['Extended_Source_Name'] == '':
# case of sources pointlike for Fermi
# loop over gammalib container and determine closest neighbor
# initialize at 1000
dist_min = 1000.
for source in models:
dir = get_model_dir(source)
dist = fdir.dist_deg(dir)
dist_min = np.minimum(dist, dist_min)
if dist_min < dist_sigma * fsource['Conf_95_SemiMajor'] / 2:
# closeby source foud in container, source will not be added
new = 0
else:
# source will be added to container, set spatial model
spatial = gammalib.GModelSpatialPointSource(fdir)
newpt += 1
######################################### treat extended sources for Fermi ############
else:
# retrieve Fermi extended source radius
ext_fsource = ext_fhl[ext_fhl['Source_Name'] == fsource['Extended_Source_Name']][0]
fradius = np.double(ext_fsource['Model_SemiMajor'])
# "corrected" distance, i.e, centre distance - radius of source
# initialize at 1000
dist_corr = 1000.
for source in models:
dir = get_model_dir(source)
dist = fdir.dist_deg(dir)
# get source radius according to model used
radius = get_model_radius(source)
# use as radius the max between Fermi and model
radius = np.maximum(fradius,radius)
# subtract from distance the max radius
dist -= radius
dist_corr = np.minimum(dist,dist_corr)
if dist_corr < 0.:
# overlapping source foud in container, source will not be added
new = 0
else:
# source will be added to container, set spatial model
fradius2 = np.double(ext_fsource['Model_SemiMinor'])
fpangle = np.double(ext_fsource['Model_PosAng'])
# retrieve Fermi spatial model to set it in container
if ext_fsource['Model_Form'] == 'Disk':
if fradius2 == fradius:
spatial = gammalib.GModelSpatialRadialDisk(fdir,fradius)
else:
spatial = gammalib.GModelSpatialEllipticalDisk(fdir, fradius, fradius2, fpangle)
elif ext_fsource['Model_Form'] == '2D Gaussian':
if fradius2 == fradius:
spatial = gammalib.GModelSpatialRadialGauss(fdir,fradius)
else:
spatial = gammalib.GModelSpatialEllipticalGauss(fdir, fradius, fradius2, fpangle)
elif ext_fsource['Model_Form'] == 'Ring':
spatial = gammalib.GModelSpatialRadialShell(fdir,fradius, fradius2)
elif ext_fsource['Model_Form'] == 'Map':
print('{} modeled by spatial template, which is not implemented, skip'.format(fsource['Source_Name']))
new = 0
else:
print('{} modeled by model type {}, which is not implemented, skip'.format(fsource['Source_Name'],ext_fsource['Model_Form']))
new = 0
if new == 1:
newext +=1
######################################### spectra #####################################
if new == 1:
# spectral model
eref = gammalib.GEnergy(np.double(fsource['Pivot_Energy']), 'GeV')
norm = np.double(fsource['Flux_Density'])
norm *= 1.e-3 # ph GeV-1 -> ph MeV-1
# use power law model only if signifcance of curvature < 1
# this avoids extrapolating hard power laws not justified by the data
if fsource['Signif_Curve'] < 1:
index = np.double(fsource['PowerLaw_Index'])
if index < 2.4:
# correction for fake pevatrons
if fsource['CLASS'] == 'PWN' or fsource['CLASS'] == 'pwn':
# dummy model to obtain search radius
mod = gammalib.GModelSky(spatial, gammalib.GModelSpectralPlaw())
# search radius
rad = get_model_radius(mod) + 0.2
# set cutoff
ecut = get_cutoff(ra, dec, 'PSR', rad_search=rad)
ecut_pwn.append(ecut)
n_ecut_pwn += 1
elif fsource['CLASS'] == 'SNR' or fsource['CLASS'] == 'snr':
gname = fsource['ASSOC1']
if 'SNR' in gname:
pass
else:
gname = None
# compute cutoff
# we use default value for particle spectral index because measurements are highly uncertain
ecut = get_cutoff(ra, dec, 'SNR', name=gname)
ecut_snr.append(ecut)
n_ecut_snr += 1
elif fsource['CLASS'] == 'bll' or fsource['CLASS'] == 'bcu' or fsource['CLASS'] == 'fsrq':
ecut = get_cutoff(ra, dec, 'AGN', z = fsource['Redshift'])
ecut_agn.append(ecut)
n_ecut_agn += 1
else:
if fsource['CLASS'] == '' or fsource['CLASS'] == 'unknown':
pass
else:
# set warning if we have hard source of unexpected type
msg += 'FHL source {} of type {} has an unxepctedly hard spectrum ' \
'with index {}. We are setting a random artificial cutoff\n'.format(
fsource['Source_Name'], fsource['CLASS'], index)
print(msg)
ecut = get_cutoff(ra, dec, 'UNID')
ecut_unid.append(ecut)
n_ecut_unid += 1
ecut = gammalib.GEnergy(np.double(ecut), 'TeV')
spectral = gammalib.GModelSpectralExpPlaw(norm, -index, eref, ecut)
else:
spectral = gammalib.GModelSpectralPlaw(norm, -index, eref)
else:
index = np.double(fsource['Spectral_Index'])
curvature = np.double(fsource['beta'])
spectral = gammalib.GModelSpectralLogParabola(norm, -index, eref, curvature)
# assemble model and append to container
model = gammalib.GModelSky(spatial, spectral)
model.name(fsource['Source_Name'])
newmodels.append(model)
# delete synthetic sources as needed
if fsource['CLASS'] == 'HMB' or fsource['CLASS'] == 'hmb' or fsource['CLASS'] == 'BIN' or fsource['CLASS'] == 'bin':
rname, bin_dict, distx, disty, radr, frlog = find_source_to_delete(bin_dict, fdir.l_deg(),
fdir.b_deg(), get_model_radius(model),
flux_Crab(model,1.,1000.))
bin_models.remove(rname)
n_bin_del += 1
bin_distx.append(distx)
bin_disty.append(disty)
bin_radr.append(radr)
bin_frlog.append(frlog)
elif fsource['CLASS'] == 'PWN' or fsource['CLASS'] == 'pwn' or fsource['CLASS'] == 'spp':
# implement synthetic PWNe here
pass
elif fsource['CLASS'] == 'SNR' or fsource['CLASS'] == 'snr':
# determine if snr is interacting
assoc = snr_class[snr_class['ASSOC1'] == fsource['ASSOC1']]
if assoc[0]['is int'] == 'yes':
# interacting
rname, isnr_dict, distx, disty, radr, frlog = find_source_to_delete(isnr_dict,
fdir.l_deg(),
fdir.b_deg(), get_model_radius(model),
flux_Crab(model, 1.,1000.))
isnr_models.remove(rname)
n_isnr_del += 1
isnr_distx.append(distx)
isnr_disty.append(disty)
isnr_radr.append(radr)
isnr_frlog.append(frlog)
else:
# young
rname, snr_dict, distx, disty, radr, frlog = find_source_to_delete(snr_dict,
fdir.l_deg(),
fdir.b_deg(), get_model_radius(model),
flux_Crab(model,1.,1000.))
snr_models.remove(rname)
n_snr_del += 1
snr_distx.append(distx)
snr_disty.append(disty)
snr_radr.append(radr)
snr_frlog.append(frlog)
else:
pass
else:
# source already present, skip
pass
# add new models to original container
for model in newmodels:
models.append(model)
# assemble output in dictionary
result = { 'models' : models, 'newpt' : newpt, 'newext' : newext,
'ecut_pwn' : ecut_pwn, 'ecut_snr' : ecut_snr, 'ecut_agn' : ecut_agn,
'ecut_unid' : ecut_unid, 'n_ecut_pwn' : n_ecut_pwn, 'n_ecut_snr' : n_ecut_snr,
'n_ecut_agn' : n_ecut_agn, 'n_ecut_unid' : n_ecut_unid, 'msg' : msg,
'bin_models' : bin_models, 'bin_dict' : bin_dict, 'n_bin_del' : n_bin_del,
'bin_distx' : bin_distx, 'bin_disty' : bin_disty, 'bin_radr' : bin_radr, 'bin_frlog' : bin_frlog,
'snr_models': snr_models, 'snr_dict': snr_dict, 'n_snr_del': n_snr_del,
'snr_distx': snr_distx, 'snr_disty': snr_disty, 'snr_radr': snr_radr, 'snr_frlog': snr_frlog,
'isnr_models': isnr_models, 'isnr_dict': isnr_dict, 'n_isnr_del': n_isnr_del,
'isnr_distx': isnr_distx, 'isnr_disty': isnr_disty, 'isnr_radr': isnr_radr, 'isnr_frlog': isnr_frlog
}
return result
def compact_sources(model_tot, source_dict, tscalc=True):
"""
Build a ctools model for the compact sources.
Parameters
----------
- model_tot: a gammalib.GModels object
- source_dict (dictionary): the source dictionary
as defined by the CompactSource class
- tscalc (bool): request TS computation or not
Outputs
--------
- the model is updated to include the sources
"""
Nsource = len(source_dict.name)
for ips in range(Nsource):
#----- Spatial model
# PointSource
if source_dict.spatial[ips]['type'] == 'PointSource':
RA = source_dict.spatial[ips]['param']['RA']['value'].to_value(
'deg')
Dec = source_dict.spatial[ips]['param']['DEC']['value'].to_value(
'deg')
spatial = gammalib.GModelSpatialPointSource(RA, Dec)
# Error
else:
raise ValueError('Spatial model not avaiable')
#----- Spectral model
# PowerLaw
if source_dict.spectral[ips]['type'] == 'PowerLaw':
prefact = source_dict.spectral[ips]['param']['Prefactor'][
'value'].to_value('cm-2 s-1 MeV-1')
index = source_dict.spectral[ips]['param']['Index']['value']
pivot = gammalib.GEnergy(
source_dict.spectral[ips]['param']['PivotEnergy']
['value'].to_value('TeV'), 'TeV')
spectral = gammalib.GModelSpectralPlaw(prefact, index, pivot)
# PowerLawExpCutoff
elif source_dict.spectral[ips]['type'] == 'PowerLawExpCutoff':
prefact = source_dict.spectral[ips]['param']['Prefactor'][
'value'].to_value('cm-2 s-1 MeV-1')
index = source_dict.spectral[ips]['param']['Index']['value']
pivot = gammalib.GEnergy(
source_dict.spectral[ips]['param']['PivotEnergy']
['value'].to_value('TeV'), 'TeV')
cutoff = gammalib.GEnergy(
source_dict.spectral[ips]['param']['Cutoff']['value'].to_value(
'TeV'), 'TeV')
spectral = gammalib.GModelSpectralExpPlaw(prefact, index, pivot,
cutoff)
# Error
else:
raise ValueError('Spectral model not available')
#----- Temporal model
# Constant
if source_dict.temporal[ips]['type'] == 'Constant':
temporal = gammalib.GModelTemporalConst(
source_dict.temporal[ips]['param']['Normalization']['value'])
# Error
else:
raise ValueError('Temporal model not available')
#----- Parameter management
spatial = manage_parameters(source_dict.spatial[ips]['param'], spatial)
spectral = manage_parameters(source_dict.spectral[ips]['param'],
spectral)
temporal = manage_parameters(source_dict.temporal[ips]['param'],
temporal)
#----- Overal model for each source
model = gammalib.GModelSky(spatial, spectral, temporal)
model.name(source_dict.name[ips])
model.tscalc(tscalc)
#----- Append model for each source
model_tot.append(model)
def set_source_model(srcmodel,
spec='plaw',
alpha=1.0,
mapname='../map_RXJ1713.fits'):
"""
Set source model
Parameters
----------
srcmodel : str
Source model name
spec : str, optional
Spectral model
alpha : float, optional
Map scaling factor
mapname : str, optional
Sky map name
Returns
-------
source : `~gammalib.GModelSky()`
Source model
"""
# Set spectral component
if spec == 'plaw':
spectral = gammalib.GModelSpectralPlaw(1.0e-17, -2.0,
gammalib.GEnergy(1.0, 'TeV'))
spectral['Prefactor'].min(1.0e-25)
elif spec == 'eplaw':
spectral = gammalib.GModelSpectralExpPlaw(
1.0e-17, -2.0, gammalib.GEnergy(1.0, 'TeV'),
gammalib.GEnergy(10.0, 'TeV'))
spectral['CutoffEnergy'].min(1.0e6)
spectral['CutoffEnergy'].max(1.0e8)
spectral['Prefactor'].min(1.0e-25)
elif spec == 'inveplaw':
spectral = gammalib.GModelSpectralExpInvPlaw(
1.0e-17, -2.0, gammalib.GEnergy(1.0, 'TeV'),
gammalib.GEnergy(10.0, 'TeV'))
spectral['Prefactor'].min(1.0e-25)
elif spec == 'logparabola':
spectral = gammalib.GModelSpectralLogParabola(
1.0e-17, -2.0, gammalib.GEnergy(1.0, 'TeV'), -0.3)
spectral['Prefactor'].min(1.0e-25)
elif spec == 'abdalla2018':
spectral = gammalib.GModelSpectralExpPlaw(
2.3e-17, -2.06, gammalib.GEnergy(1.0, 'TeV'),
gammalib.GEnergy(12.9, 'TeV'))
spectral['Prefactor'].fix()
spectral['Index'].fix()
spectral['CutoffEnergy'].fix()
elif spec == 'abdalla2018Ec':
spectral = gammalib.GModelSpectralExpPlaw(
2.3e-17, -2.06, gammalib.GEnergy(1.0, 'TeV'),
gammalib.GEnergy(12.9, 'TeV'))
spectral['CutoffEnergy'].fix()
spectral['Prefactor'].min(1.0e-25)
elif spec == 'aharonian2007':
spectral = gammalib.GModelSpectralLogParabola(
2.06e-17, -2.02, gammalib.GEnergy(1.0, 'TeV'), -0.29)
spectral['Prefactor'].fix()
spectral['Index'].fix()
spectral['Curvature'].fix()
# Set spatial component
if 'map' in srcmodel:
filename = '%s' % mapname
map = gammalib.GSkyMap(filename)
map = scale_map(map, alpha)
filename = 'map_RXJ1713_%.2f.fits' % alpha
map.save(filename, True)
spatial = gammalib.GModelSpatialDiffuseMap(filename)
elif 'disk' in srcmodel:
dir = gammalib.GSkyDir()
dir.radec_deg(258.3, -39.7)
spatial = gammalib.GModelSpatialRadialDisk(dir, 0.5)
spatial['RA'].free()
spatial['DEC'].free()
elif 'gauss' in srcmodel:
dir = gammalib.GSkyDir()
dir.radec_deg(258.3, -39.7)
spatial = gammalib.GModelSpatialRadialGauss(dir, 0.5)
spatial['RA'].free()
spatial['DEC'].free()
elif 'shell' in srcmodel:
dir = gammalib.GSkyDir()
dir.radec_deg(258.3, -39.7)
spatial = gammalib.GModelSpatialRadialShell(dir, 0.4, 0.2)
spatial['RA'].free()
spatial['DEC'].free()
# Set source model
source = gammalib.GModelSky(spatial, spectral)
source.name('RX J1713.7-3946')
source.tscalc(True)
# Return source model
return source
n_ecut_snr += 1
# otherwise if unidentified
else:
if 'unid' in source['classes']:
pass
else:
# set warning if we have hard source of unexpected type
msg = 'Gamma-cat source {} of type {} has an unxepctedly hard spectrum ' \
'with index {}. We are setting a random artificial cutoff\n'.format(source['common_name'],source['classes'],index)
print(msg)
outfile.write(msg)
ecut = get_cutoff(ra, dec, 'UNID')
ecut_unid.append(ecut)
n_ecut_unid += 1
ecut = gammalib.GEnergy(np.double(ecut), 'TeV')
spectral = gammalib.GModelSpectralExpPlaw(
norm, -index, eref, ecut)
else:
spectral = gammalib.GModelSpectralPlaw(norm, -index, eref)
elif source['spec_type'] == 'ecpl':
index = source['spec_ecpl_index']
norm = source['spec_ecpl_norm']
eref = gammalib.GEnergy(np.double(source['spec_ecpl_e_ref']),
'TeV')
ecut = gammalib.GEnergy(np.double(source['spec_ecpl_e_cut']),
'TeV')
# convert norm from ph cm-2 s-1 TeV-1 (gamma-cat) to ph cm-2 s-1 MeV-1 (gammalib)
norm *= 1.e-6
index = np.double(index)
norm = np.double(norm)
spectral = gammalib.GModelSpectralExpPlaw(norm, -index, eref, ecut)
else:
def background(model_tot, bkg_dict_in, setID=True):
"""
Build a ctools model for the background.
Parameters
----------
- model_tot: a gammalib.GModels object
- bkg_dict (dictionary): the dictionary that
contain the background properties (from class
Background). In case of multiple background this can be a list.
- setID (bool): decide to set or not an ID in the bkg model
Outputs
--------
- the model is updated to include the background
"""
#---------- Get the number of background
if type(bkg_dict_in) == list:
Nbkg = len(bkg_dict_in)
else:
Nbkg = 1
#---------- Add all bkg models
for i in range(Nbkg):
#---------- Select the background from the list or not
if type(bkg_dict_in) == list:
bkg_dict = bkg_dict_in[i]
else:
bkg_dict = bkg_dict_in
#----- Spectral model
# PowerLaw
if bkg_dict.spectral['type'] == 'PowerLaw':
prefact = bkg_dict.spectral['param']['Prefactor']['value']
index = bkg_dict.spectral['param']['Index']['value']
pivot = gammalib.GEnergy(
bkg_dict.spectral['param']['PivotEnergy']['value'].to_value(
'TeV'), 'TeV')
spectral = gammalib.GModelSpectralPlaw(prefact, index, pivot)
# PowerLawExpCutoff
elif bkg_dict.spectral['type'] == 'PowerLawExpCutoff':
prefact = bkg_dict.spectral['param']['Prefactor']['value']
index = bkg_dict.spectral['param']['Index']['value']
pivot = gammalib.GEnergy(
bkg_dict.spectral['param']['PivotEnergy']['value'].to_value(
'TeV'), 'TeV')
cutoff = gammalib.GEnergy(
bkg_dict.spectral['param']['Cutoff']['value'].to_value('TeV'),
'TeV')
spectral = gammalib.GModelSpectralExpPlaw(prefact, index, pivot,
cutoff)
# Error
else:
raise ValueError('Spectral model not available')
# Parameter management
spectral = manage_parameters(bkg_dict.spectral['param'], spectral)
#----- Spatial model
# CTAIrfBackground
if bkg_dict.spatial['type'] == 'CTAIrfBackground':
#----- Overal model for each source
model = gammalib.GCTAModelIrfBackground(spectral)
# Error
else:
raise ValueError('Spatial model not avaiable')
#----- Append model
model.name(bkg_dict.name)
model.instruments(bkg_dict.instrument)
if setID:
if bkg_dict.obsid is not None: model.ids(bkg_dict.obsid)
model_tot.append(model)
def dealWithModelFile(cfg):
"""
Import infos from configuration file and create a model file
for the analysis
"""
# Create model container
models = gammalib.GModels()
# coordinates
src_dir = gammalib.GSkyDir()
src_dir.radec_deg(cfg.getValue('model', 'coords', 'ra'),
cfg.getValue('model', 'coords', 'dec'))
# spatial model
spatial = None
spectral = None
opt_spatial = cfg.getValue('model', 'spatial')
if opt_spatial == 'plike':
spatial = gammalib.GModelSpatialPointSource(src_dir)
spatial['RA'].min(-360.)
spatial['RA'].max(360.)
spatial['DEC'].min(-90.)
spatial['DEC'].max(90.)
if cfg.getValue('model', 'coords', 'fitposition') is True:
spatial['RA'].free()
spatial['DEC'].free()
else: # TODO
pass
# spectral model
opt_spectral = cfg.getValue('model', 'spectral')
if opt_spectral == 'pwl':
spectral = gammalib.GModelSpectralPlaw()
# handle prefactor
prefactor = cfg.getValue('model', 'pwl', 'prefactor')
spectral['Prefactor'].value(prefactor * 1.e-17)
spectral['Prefactor'].min(1.e-24)
spectral['Prefactor'].max(1.e-14)
spectral['Prefactor'].scale(1.e-17)
# handle scale
scale = cfg.getValue('model', 'pwl', 'scale')
spectral['PivotEnergy'].value(scale * 1.e6)
spectral['PivotEnergy'].scale(1.e6)
spectral['PivotEnergy'].min(1.e4)
spectral['PivotEnergy'].max(1.e9)
# handle index
index = cfg.getValue('model', 'pwl', 'index')
spectral['Index'].value(index)
spectral['Index'].scale(-1.)
spectral['Index'].min(-0)
spectral['Index'].max(-5)
elif opt_spectral == 'logpwl':
spectral = gammalib.GModelSpectralLogParabola()
# handle prefactor
prefactor = cfg.getValue('model', 'logpwl', 'prefactor')
spectral['Prefactor'].value(prefactor * 1.e-17)
spectral['Prefactor'].min(1.e-24)
spectral['Prefactor'].max(1.e-14)
spectral['Prefactor'].scale(1.e-17)
# handle scale
scale = cfg.getValue('model', 'logpwl', 'scale')
spectral['PivotEnergy'].value(scale * 1.e6)
spectral['PivotEnergy'].scale(1.e6)
spectral['PivotEnergy'].min(1.e4)
spectral['PivotEnergy'].max(1.e9)
# handle index
index = cfg.getValue('model', 'logpwl', 'index')
spectral['Index'].value(index)
spectral['Index'].scale(-1.)
spectral['Index'].min(-0)
spectral['Index'].max(-5)
# handle curvature
curvature = cfg.getValue('model', 'logpwl', 'curvature')
spectral['Curvature'].value(curvature)
spectral['Curvature'].scale(-1.)
spectral['Curvature'].min(5)
spectral['Curvature'].max(-5)
elif opt_spectral == 'exppwl':
spectral = gammalib.GModelSpectralExpPlaw()
# handle prefactor
prefactor = cfg.getValue('model', 'exppwl', 'prefactor')
spectral['Prefactor'].value(prefactor * 1.e-17)
spectral['Prefactor'].min(1.e-24)
spectral['Prefactor'].max(1.e-14)
spectral['Prefactor'].scale(1.e-17)
# handle scale
scale = cfg.getValue('model', 'exppwl', 'scale')
spectral['PivotEnergy'].value(scale * 1.e6)
spectral['PivotEnergy'].scale(1.e6)
spectral['PivotEnergy'].min(1.e4)
spectral['PivotEnergy'].max(1.e9)
# handle index
index = cfg.getValue('model', 'exppwl', 'index')
spectral['Index'].value(index)
spectral['Index'].scale(-1.)
spectral['Index'].min(-0)
spectral['Index'].max(-5)
# handle index
cutoff = cfg.getValue('model', 'exppwl', 'cutoff')
spectral['CutoffEnergy'].value(cutoff * 1.e6)
spectral['CutoffEnergy'].scale(1.e6)
spectral['CutoffEnergy'].min(1.e4)
spectral['CutoffEnergy'].max(1.e9)
else:
pass
model = gammalib.GModelSky(spatial, spectral)
models.append(model)
outputdir = cfg.getValue('general', 'outputdir')
models.save(outputdir + '/' + cfg.getValue('model', 'output'))
# print(models)
del models
def grb_simulation(sim_in, config_in, model_xml, fits_header_0, counter):
"""
Function to handle the GRB simulation.
:param sim_in: the yaml file for the simulation (unpacked as a dict of dicts)
:param config_in: the yaml file for the job handling (unpacked as a dict of dicts)
:param model_xml: the XML model name for the source under analysis
:param fits_header_0: header for the fits file of the GRB model to use. Used in the visibility calculation
:param counter: integer number. counts the id of the source realization
:return: significance obtained with the activated detection methods
"""
src_name = model_xml.split('/')[-1].split('model_')[1][:-4]
print(src_name, counter)
ctools_pipe_path = create_path(config_in['exe']['software_path'])
ctobss_params = sim_in['ctobssim']
seed = int(counter)*10
# PARAMETERS FROM THE CTOBSSIM
sim_t_min = u.Quantity(ctobss_params['time']['t_min']).to_value(u.s)
sim_t_max = u.Quantity(ctobss_params['time']['t_max']).to_value(u.s)
sim_e_min = u.Quantity(ctobss_params['energy']['e_min']).to_value(u.TeV)
sim_e_max = u.Quantity(ctobss_params['energy']['e_max']).to_value(u.TeV)
sim_rad = ctobss_params['radius']
models = sim_in['source']
source_type = models['type']
if source_type == "GRB":
phase_path = "/" + models['phase']
elif source_type == "GW":
phase_path = ""
output_path = create_path(sim_in['output']['path'] + phase_path + '/' + src_name)
save_simulation = ctobss_params['save_simulation']
with open(f"{output_path}/GRB-{src_name}_seed-{seed}.txt", "w") as f:
f.write(f"GRB,seed,time_start,time_end,sigma_lima,sqrt_TS_onoff,sqrt_TS_std\n")
# VISIBILITY PART
# choose between AUTO mode (use visibility) and MANUAL mode (manually insert IRF)
simulation_mode = sim_in['IRF']['mode']
if simulation_mode == "auto":
print("using visibility to get IRFs")
# GRB information from the fits header
ra = fits_header_0['RA']
dec = fits_header_0['DEC']
t0 = Time(fits_header_0['GRBJD'])
irf_dict = sim_in['IRF']
site = irf_dict['site']
obs_condition = Observability(site=site)
obs_condition.set_irf(irf_dict)
t_zero_mode = ctobss_params['time']['t_zero'].lower()
if t_zero_mode == "VIS":
# check if the source is visible one day after the onset of the source
print("time starts when source becomes visible")
obs_condition.Proposal_obTime = 86400
condition_check = obs_condition.check(RA=ra, DEC=dec, t_start=t0)
elif t_zero_mode == "ONSET":
print("time starts from the onset of the GRB")
condition_check = obs_condition.check(RA=ra, DEC=dec, t_start=t0, t_min=sim_t_min, t_max=sim_t_max)
else:
print(f"Choose some proper mode between 'VIS' and 'ONSET'. {t_zero_mode} is not a valid one.")
sys.exit()
# NO IRF in AUTO mode ==> No simulation! == EXIT!
if len(condition_check) == 0:
f.write(f"{src_name},{seed}, -1, -1, -1, -1, -1\n")
sys.exit()
elif simulation_mode == "manual":
print("manual picking IRF")
# find proper IRF name
irf = IRFPicker(sim_in, ctools_pipe_path)
name_irf = irf.irf_pick()
backgrounds_path = create_path(ctobss_params['bckgrnd_path'])
fits_background_list = glob.glob(
f"{backgrounds_path}/{irf.prod_number}_{irf.prod_version}_{name_irf}/background*.fits")
if len(fits_background_list) == 0:
print(f"No background for IRF {name_irf}")
sys.exit()
fits_background_list = sorted(fits_background_list, key=sort_background)
background_fits = fits_background_list[int(counter) - 1]
obs_back = gammalib.GCTAObservation(background_fits)
else:
print(f"wrong input for IRF - mode. Input is {simulation_mode}. Use 'auto' or 'manual' instead")
sys.exit()
if irf.prod_number == "3b" and irf.prod_version == 0:
caldb = "prod3b"
else:
caldb = f'prod{irf.prod_number}-v{irf.prod_version}'
# source simulation
sim = ctools.ctobssim()
sim['inmodel'] = model_xml
sim['caldb'] = caldb
sim['irf'] = name_irf
sim['ra'] = 0.0
sim['dec'] = 0.0
sim['rad'] = sim_rad
sim['tmin'] = sim_t_min
sim['tmax'] = sim_t_max
sim['emin'] = sim_e_min
sim['emax'] = sim_e_max
sim['seed'] = seed
sim.run()
obs = sim.obs()
# # move the source photons from closer to (RA,DEC)=(0,0), where the background is located
# for event in obs[0].events():
# # ra_evt = event.dir().dir().ra()
# dec_evt = event.dir().dir().dec()
# ra_evt_deg = event.dir().dir().ra_deg()
# dec_evt_deg = event.dir().dir().dec_deg()
#
# ra_corrected = (ra_evt_deg - ra_pointing)*np.cos(dec_evt)
# dec_corrected = dec_evt_deg - dec_pointing
# event.dir().dir().radec_deg(ra_corrected, dec_corrected)
# append all background events to GRB ones ==> there's just one observation and not two
for event in obs_back.events():
obs[0].events().append(event)
# ctselect to save data on disk
if save_simulation:
event_list_path = create_path(f"{ctobss_params['output_path']}/{src_name}/")
#obs.save(f"{event_list_path}/event_list_source-{src_name}_seed-{seed:03}.fits")
select_time = ctools.ctselect(obs)
select_time['rad'] = sim_rad
select_time['tmin'] = sim_t_min
select_time['tmax'] = sim_t_max
select_time['emin'] = sim_e_min
select_time['emax'] = sim_e_max
select_time['outobs'] = f"{event_list_path}/event_list_source-{src_name}_{seed:03}.fits"
select_time.run()
sys.exit()
# delete all 70+ models from the obs def file...not needed any more
obs.models(gammalib.GModels())
# CTSELECT
select_time = sim_in['ctselect']['time_cut']
slices = int(select_time['t_slices'])
if slices == 0:
times = [sim_t_min, sim_t_max]
times_start = times[:-1]
times_end = times[1:]
elif slices > 0:
time_mode = select_time['mode']
if time_mode == "log":
times = np.logspace(np.log10(sim_t_min), np.log10(sim_t_max), slices + 1, endpoint=True)
elif time_mode == "lin":
times = np.linspace(sim_t_min, sim_t_max, slices + 1, endpoint=True)
else:
print(f"{time_mode} not valid. Use 'log' or 'lin' ")
sys.exit()
if select_time['obs_mode'] == "iter":
times_start = times[:-1]
times_end = times[1:]
elif select_time['obs_mode'] == "cumul":
times_start = np.repeat(times[0], slices) # this is to use the same array structure for the loop
times_end = times[1:]
elif select_time['obs_mode'] == "all":
begins, ends = np.meshgrid(times[:-1], times[1:])
mask_times = begins < ends
times_start = begins[mask_times].ravel()
times_end = ends[mask_times].ravel()
else:
print(f"obs_mode: {select_time['obs_mode']} not supported")
sys.exit()
else:
print(f"value {slices} not supported...check yaml file")
sys.exit()
# ------------------------------------
# ----- TIME LOOP STARTS HERE --------
# ------------------------------------
ctlike_mode = sim_in['detection']
mode_1 = ctlike_mode['counts']
mode_2 = ctlike_mode['ctlike-onoff']
mode_3 = ctlike_mode['ctlike-std']
for t_in, t_end in zip(times_start, times_end):
sigma_onoff = 0
sqrt_ts_like_onoff = 0
sqrt_ts_like_std = 0
print("-----------------------------")
print(f"t_in: {t_in:.2f}, t_end: {t_end:.2f}")
# different ctlikes (onoff or std) need different files.
# will be appended here and used later on for the final likelihood
dict_obs_select_time = {}
# perform time selection for this specific time bin
select_time = ctools.ctselect(obs)
select_time['rad'] = sim_rad
select_time['tmin'] = t_in
select_time['tmax'] = t_end
select_time['emin'] = sim_e_min
select_time['emax'] = sim_e_max
select_time.run()
if mode_1:
fits_temp_title = f"skymap_{seed}_{t_in:.2f}_{t_end:.2f}.fits"
pars_counts = ctlike_mode['pars_counts']
scale = float(pars_counts['scale'])
npix = 2*int(sim_rad/scale)
skymap = ctools.ctskymap(select_time.obs().copy())
skymap['emin'] = sim_e_min
skymap['emax'] = sim_e_max
skymap['nxpix'] = npix
skymap['nypix'] = npix
skymap['binsz'] = scale
skymap['proj'] = 'TAN'
skymap['coordsys'] = 'CEL'
skymap['xref'] = 0
skymap['yref'] = 0
skymap['bkgsubtract'] = 'RING'
skymap['roiradius'] = pars_counts['roiradius']
skymap['inradius'] = pars_counts['inradius']
skymap['outradius'] = pars_counts['outradius']
skymap['iterations'] = pars_counts['iterations']
skymap['threshold'] = pars_counts['threshold']
skymap['outmap'] = fits_temp_title
skymap.execute()
input_fits = fits.open(fits_temp_title)
datain = input_fits[2].data
datain[np.isnan(datain)] = 0.0
datain[np.isinf(datain)] = 0.0
sigma_onoff = np.max(datain)
os.remove(fits_temp_title)
if mode_3:
dict_obs_select_time['std'] = select_time.obs().copy()
if mode_2:
onoff_time_sel = cscripts.csphagen(select_time.obs().copy())
onoff_time_sel['inmodel'] = 'NONE'
onoff_time_sel['ebinalg'] = 'LOG'
onoff_time_sel['emin'] = sim_e_min
onoff_time_sel['emax'] = sim_e_max
onoff_time_sel['enumbins'] = 30
onoff_time_sel['coordsys'] = 'CEL'
onoff_time_sel['ra'] = 0.0
onoff_time_sel['dec'] = 0.5
onoff_time_sel['rad'] = 0.2
onoff_time_sel['bkgmethod'] = 'REFLECTED'
onoff_time_sel['use_model_bkg'] = False
onoff_time_sel['stack'] = False
onoff_time_sel.run()
dict_obs_select_time['onoff'] = onoff_time_sel.obs().copy()
del onoff_time_sel
# print(f"sigma ON/OFF: {sigma_onoff:.2f}")
if mode_2 or mode_3:
# Low Energy PL fitting
# to be saved in this dict
dict_pl_ctlike_out = {}
e_min_pl_ctlike = 0.030
e_max_pl_ctlike = 0.080
# simple ctobssim copy and select for ctlike-std
select_pl_ctlike = ctools.ctselect(select_time.obs().copy())
select_pl_ctlike['rad'] = 3
select_pl_ctlike['tmin'] = t_in
select_pl_ctlike['tmax'] = t_end
select_pl_ctlike['emin'] = e_min_pl_ctlike
select_pl_ctlike['emax'] = e_max_pl_ctlike
select_pl_ctlike.run()
# create test source
src_dir = gammalib.GSkyDir()
src_dir.radec_deg(0, 0.5)
spatial = gammalib.GModelSpatialPointSource(src_dir)
# create and append source spectral model
spectral = gammalib.GModelSpectralPlaw()
spectral['Prefactor'].value(5.5e-16)
spectral['Prefactor'].scale(1e-16)
spectral['Index'].value(-2.6)
spectral['Index'].scale(-1.0)
spectral['PivotEnergy'].value(50000)
spectral['PivotEnergy'].scale(1e3)
model_src = gammalib.GModelSky(spatial, spectral)
model_src.name('PL_fit_temp')
model_src.tscalc(True)
spectral_back = gammalib.GModelSpectralPlaw()
spectral_back['Prefactor'].value(1.0)
spectral_back['Prefactor'].scale(1.0)
spectral_back['Index'].value(0)
spectral_back['PivotEnergy'].value(300000)
spectral_back['PivotEnergy'].scale(1e6)
if mode_2:
back_model = gammalib.GCTAModelIrfBackground()
back_model.instruments('CTAOnOff')
back_model.name('Background')
back_model.spectral(spectral_back.copy())
onoff_pl_ctlike_lima = cscripts.csphagen(select_pl_ctlike.obs().copy())
onoff_pl_ctlike_lima['inmodel'] = 'NONE'
onoff_pl_ctlike_lima['ebinalg'] = 'LOG'
onoff_pl_ctlike_lima['emin'] = e_min_pl_ctlike
onoff_pl_ctlike_lima['emax'] = e_max_pl_ctlike
onoff_pl_ctlike_lima['enumbins'] = 30
onoff_pl_ctlike_lima['coordsys'] = 'CEL'
onoff_pl_ctlike_lima['ra'] = 0.0
onoff_pl_ctlike_lima['dec'] = 0.5
onoff_pl_ctlike_lima['rad'] = 0.2
onoff_pl_ctlike_lima['bkgmethod'] = 'REFLECTED'
onoff_pl_ctlike_lima['use_model_bkg'] = False
onoff_pl_ctlike_lima['stack'] = False
onoff_pl_ctlike_lima.run()
onoff_pl_ctlike_lima.obs().models(gammalib.GModels())
onoff_pl_ctlike_lima.obs().models().append(model_src.copy())
onoff_pl_ctlike_lima.obs().models().append(back_model.copy())
like_pl = ctools.ctlike(onoff_pl_ctlike_lima.obs())
like_pl['refit'] = True
like_pl.run()
dict_pl_ctlike_out['onoff'] = like_pl.obs().copy()
del onoff_pl_ctlike_lima
del like_pl
if mode_3:
models_ctlike_std = gammalib.GModels()
models_ctlike_std.append(model_src.copy())
back_model = gammalib.GCTAModelIrfBackground()
back_model.instruments('CTA')
back_model.name('Background')
back_model.spectral(spectral_back.copy())
models_ctlike_std.append(back_model)
# save models
xmlmodel_PL_ctlike_std = 'test_model_PL_ctlike_std.xml'
models_ctlike_std.save(xmlmodel_PL_ctlike_std)
del models_ctlike_std
like_pl = ctools.ctlike(select_pl_ctlike.obs().copy())
like_pl['inmodel'] = xmlmodel_PL_ctlike_std
like_pl['refit'] = True
like_pl.run()
dict_pl_ctlike_out['std'] = like_pl.obs().copy()
del like_pl
del spatial
del spectral
del model_src
del select_pl_ctlike
# EXTENDED CTLIKE
for key in dict_obs_select_time.keys():
likelihood_pl_out = dict_pl_ctlike_out[key]
selected_data = dict_obs_select_time[key]
pref_out_pl = likelihood_pl_out.models()[0]['Prefactor'].value()
index_out_pl = likelihood_pl_out.models()[0]['Index'].value()
pivot_out_pl = likelihood_pl_out.models()[0]['PivotEnergy'].value()
expplaw = gammalib.GModelSpectralExpPlaw()
expplaw['Prefactor'].value(pref_out_pl)
expplaw['Index'].value(index_out_pl)
expplaw['PivotEnergy'].value(pivot_out_pl)
expplaw['CutoffEnergy'].value(80e3)
if key == "onoff":
selected_data.models()[0].name(src_name)
selected_data.models()[0].tscalc(True)
selected_data.models()[0].spectral(expplaw.copy())
like = ctools.ctlike(selected_data)
like['refit'] = True
like.run()
ts = like.obs().models()[0].ts()
if ts > 0:
sqrt_ts_like_onoff = np.sqrt(like.obs().models()[0].ts())
else:
sqrt_ts_like_onoff = 0
del like
if key == "std":
models_fit_ctlike = gammalib.GModels()
# create test source
src_dir = gammalib.GSkyDir()
src_dir.radec_deg(0, 0.5)
spatial = gammalib.GModelSpatialPointSource(src_dir)
# append spatial and spectral models
model_src = gammalib.GModelSky(spatial, expplaw.copy())
model_src.name('Source_fit')
model_src.tscalc(True)
models_fit_ctlike.append(model_src)
# create and append background
back_model = gammalib.GCTAModelIrfBackground()
back_model.instruments('CTA')
back_model.name('Background')
spectral_back = gammalib.GModelSpectralPlaw()
spectral_back['Prefactor'].value(1.0)
spectral_back['Prefactor'].scale(1.0)
spectral_back['Index'].value(0)
spectral_back['PivotEnergy'].value(300000)
spectral_back['PivotEnergy'].scale(1e6)
back_model.spectral(spectral_back)
models_fit_ctlike.append(back_model)
# save models
input_ctlike_xml = "model_GRB_fit_ctlike_in.xml"
models_fit_ctlike.save(input_ctlike_xml)
del models_fit_ctlike
like = ctools.ctlike(selected_data)
like['inmodel'] = input_ctlike_xml
like['refit'] = True
like.run()
ts = like.obs().models()[0].ts()
if ts > 0:
sqrt_ts_like_std = np.sqrt(like.obs().models()[0].ts())
else:
sqrt_ts_like_std = 0
del like
# E_cut_off = like.obs().models()[0]['CutoffEnergy'].value()
# E_cut_off_error = like.obs().models()[0]['CutoffEnergy'].error()
# print(f"sqrt(TS) {key}: {np.sqrt(ts_like):.2f}")
# print(f"E_cut_off {key}: {E_cut_off:.2f} +- {E_cut_off_error:.2f}")
del dict_pl_ctlike_out
f.write(f"{src_name},{seed},{t_in:.2f},{t_end:.2f},{sigma_onoff:.2f},{sqrt_ts_like_onoff:.2f},{sqrt_ts_like_std:.2f}\n")
del dict_obs_select_time
del select_time
def set_source_model(srcmodel, spec='plaw'):
"""
Set source model
Parameters
----------
srcmodel : str
Source model name
spec : str, optional
Spectral model
Returns
-------
source : `~gammalib.GModelSky()`
Source model
"""
# Set spectral component
if spec == 'plaw':
spectral = gammalib.GModelSpectralPlaw(7.0e-17, -3.0,
gammalib.GEnergy(1.0,'TeV'))
spectral['Prefactor'].min(1.0e-25)
elif spec == 'eplaw':
spectral = gammalib.GModelSpectralExpPlaw(7.0e-17, -3.0,
gammalib.GEnergy(1.0,'TeV'),
gammalib.GEnergy(3.0,'TeV'))
spectral['Prefactor'].min(1.0e-25)
spectral['CutoffEnergy'].min(1.0e5)
spectral['CutoffEnergy'].max(1.0e8)
elif spec == 'logparabola':
spectral = gammalib.GModelSpectralLogParabola(7.0e-17, -3.0,
gammalib.GEnergy(1.0,'TeV'),
-0.8)
spectral['Prefactor'].min(1.0e-25)
elif spec == 'fplaw200':
spectral = gammalib.GModelSpectralPlawPhotonFlux(1.0e-10, -2.0,
gammalib.GEnergy(0.2,'TeV'),
gammalib.GEnergy(100.0,'TeV'))
spectral['PhotonFlux'].min(1.0e-25)
elif spec == 'fplaw300':
spectral = gammalib.GModelSpectralPlawPhotonFlux(1.0e-10, -2.0,
gammalib.GEnergy(0.3,'TeV'),
gammalib.GEnergy(100.0,'TeV'))
spectral['PhotonFlux'].min(1.0e-25)
elif spec == 'fplaw700':
spectral = gammalib.GModelSpectralPlawPhotonFlux(1.0e-10, -3.3,
gammalib.GEnergy(0.7,'TeV'),
gammalib.GEnergy(100.0,'TeV'))
spectral['PhotonFlux'].min(1.0e-25)
spectral['Index'].min(-5.0)
spectral['Index'].max(-1.5)
# Set spatial component
dir = gammalib.GSkyDir()
dir.radec_deg(329.71694, -30.22559)
if srcmodel == 'ptsrc':
spatial = gammalib.GModelSpatialPointSource(dir)
elif srcmodel == 'gauss':
spatial = gammalib.GModelSpatialRadialGauss(dir, 0.1)
spatial['Sigma'].min(0.0001)
spatial['Sigma'].free()
spatial['RA'].free()
spatial['DEC'].free()
# Set source model
source = gammalib.GModelSky(spatial, spectral)
source.name('PKS 2155-304')
source.tscalc(True)
# Return source model
return source
def compact_sources(model_tot,
source_dict,
work_dir,
tscalc=True,
EBL_model='dominguez',
energy=np.logspace(-1, 5, 1000) * u.GeV):
"""
Build a ctools model for the compact sources.
Parameters
----------
- model_tot: a gammalib.GModels object
- source_dict (dictionary): the source dictionary
as defined by the CompactSource class
- tscalc (bool): request TS computation or not
- EBL_model (string): the EBL model to use
Outputs
--------
- The model is updated to include the sources
"""
Nsource = len(source_dict.name)
for ips in range(Nsource):
#----- Spatial model
# PointSource
if source_dict.spatial[ips]['type'] == 'PointSource':
RA = source_dict.spatial[ips]['param']['RA']['value'].to_value(
'deg')
Dec = source_dict.spatial[ips]['param']['DEC']['value'].to_value(
'deg')
spatial = gammalib.GModelSpatialPointSource(RA, Dec)
# Error
else:
raise ValueError('Spatial model not avaiable')
#----- Spectral model
# PowerLaw
if source_dict.spectral[ips]['type'] == 'PowerLaw':
prefact = source_dict.spectral[ips]['param']['Prefactor'][
'value'].to_value('cm-2 s-1 MeV-1')
index = source_dict.spectral[ips]['param']['Index']['value']
pivot = gammalib.GEnergy(
source_dict.spectral[ips]['param']['PivotEnergy']
['value'].to_value('TeV'), 'TeV')
spectral0 = gammalib.GModelSpectralPlaw(prefact, index, pivot)
# PowerLawExpCutoff
elif source_dict.spectral[ips]['type'] == 'PowerLawExpCutoff':
prefact = source_dict.spectral[ips]['param']['Prefactor'][
'value'].to_value('cm-2 s-1 MeV-1')
index = source_dict.spectral[ips]['param']['Index']['value']
pivot = gammalib.GEnergy(
source_dict.spectral[ips]['param']['PivotEnergy']
['value'].to_value('TeV'), 'TeV')
cutoff = gammalib.GEnergy(
source_dict.spectral[ips]['param']['Cutoff']['value'].to_value(
'TeV'), 'TeV')
spectral0 = gammalib.GModelSpectralExpPlaw(prefact, index, pivot,
cutoff)
# Error
else:
raise ValueError('Spectral model not available')
# EBL absorb
if EBL_model is not None and source_dict.redshift[ips] is not None:
absorb = cluster_spectra.get_ebl_absorb(energy.to_value('GeV'),
source_dict.redshift[ips],
EBL_model)
absmin = np.amin(absorb[absorb > 0])
absorb[absorb == 0] = np.amin(np.array([absmin, 1e-16
])) # does not accept 0
ebl_abs_file = work_dir + '/EBLmodel_' + source_dict.name[
ips] + '.txt'
save_txt_file(ebl_abs_file, energy.to_value('MeV'), absorb,
'energy (MeV)', 'Absorb')
ebl_spec = gammalib.GModelSpectralFunc(ebl_abs_file, 1.0)
ebl_spec['Normalization'].fix()
doEBL = True
else:
doEBL = False
#----- Temporal model
# Constant
if source_dict.temporal[ips]['type'] == 'Constant':
temporal = gammalib.GModelTemporalConst(
source_dict.temporal[ips]['param']['Normalization']['value'])
# Error
else:
raise ValueError('Temporal model not available')
#----- Parameter management
spatial = manage_parameters(source_dict.spatial[ips]['param'], spatial)
spectral0 = manage_parameters(source_dict.spectral[ips]['param'],
spectral0)
temporal = manage_parameters(source_dict.temporal[ips]['param'],
temporal)
#----- EBL x initial spectrum
if doEBL:
spectral = gammalib.GModelSpectralMultiplicative()
spectral.append(spectral0)
spectral.append(ebl_spec)
else:
spectral = spectral0
#----- Overal model for each source
model = gammalib.GModelSky(spatial, spectral, temporal)
model.name(source_dict.name[ips])
model.tscalc(tscalc)
#----- Append model for each source
model_tot.append(model)
def set_source_model(srcmodel, spec='plaw'):
"""
Set source model
Parameters
----------
srcmodel : str
Source model name
spec : str, optional
Spectral model
Returns
-------
source : `~gammalib.GModelSky()`
Source model
"""
# Set spectral component
if spec == 'plaw':
spectral = gammalib.GModelSpectralPlaw(1.0e-17, -2.0,
gammalib.GEnergy(1.0, 'TeV'))
spectral['Prefactor'].min(1.0e-25)
elif spec == 'plaw145':
spectral = gammalib.GModelSpectralPlaw(1.0e-17, -2.0,
gammalib.GEnergy(1.45, 'TeV'))
spectral['Prefactor'].min(1.0e-25)
elif spec == 'eplaw':
spectral = gammalib.GModelSpectralExpPlaw(
1.0e-17, -2.0, gammalib.GEnergy(1.0, 'TeV'),
gammalib.GEnergy(10.0, 'TeV'))
spectral['Prefactor'].min(1.0e-25)
spectral['CutoffEnergy'].min(3.0e6)
spectral['CutoffEnergy'].max(1.0e12)
elif spec == 'inveplaw':
spectral = gammalib.GModelSpectralExpInvPlaw(
1.0e-17, -2.0, gammalib.GEnergy(1.0, 'TeV'),
gammalib.GEnergy(10.0, 'TeV'))
spectral['Prefactor'].min(1.0e-25)
elif spec == 'logparabola':
spectral = gammalib.GModelSpectralLogParabola(
1.0e-17, -2.0, gammalib.GEnergy(1.0, 'TeV'), -0.3)
spectral['Prefactor'].min(1.0e-25)
elif spec == 'fplaw':
spectral = gammalib.GModelSpectralPlawPhotonFlux(
1.0e-11, -2.0, gammalib.GEnergy(1.0, 'TeV'),
gammalib.GEnergy(100.0, 'TeV'))
spectral['PhotonFlux'].min(1.0e-25)
elif spec == 'aharonian2006':
spectral = gammalib.GModelSpectralExpPlaw(
3.84e-17, -2.41, gammalib.GEnergy(1.0, 'TeV'),
gammalib.GEnergy(15.1, 'TeV'))
spectral['Prefactor'].fix()
spectral['Index'].fix()
spectral['CutoffEnergy'].fix()
elif spec == 'nigro2019':
spectral = gammalib.GModelSpectralLogParabola(
4.47e-17, -2.39, gammalib.GEnergy(1.0, 'TeV'), -0.16)
spectral['Prefactor'].fix()
spectral['Index'].fix()
spectral['Curvature'].fix()
# Set spatial component
dir = gammalib.GSkyDir()
dir.radec_deg(83.6331, 22.0145)
if srcmodel == 'ptsrc':
spatial = gammalib.GModelSpatialPointSource(dir)
elif srcmodel == 'gauss':
spatial = gammalib.GModelSpatialRadialGauss(dir, 0.1)
spatial['Sigma'].min(0.0001)
spatial['Sigma'].free()
elif srcmodel == 'egauss':
spatial = gammalib.GModelSpatialEllipticalGauss(dir, 0.02, 0.01, 0.0)
spatial['MinorRadius'].min(0.0001)
spatial['MinorRadius'].free()
spatial['MajorRadius'].min(0.0001)
spatial['MajorRadius'].free()
spatial['PA'].free()
spatial['RA'].free()
spatial['DEC'].free()
# Set source model
source = gammalib.GModelSky(spatial, spectral)
source.name('Crab')
source.tscalc(True)
# Return source model
return source
# positions of the clusters from SIMBAD
glons = [80.22, 339.55]
glats = [0.80, -0.40]
# resolution of output maps in deg
out_res = 0.05
# set output in the templates directory
outdir = '../known-sources/templates/'
# spectral models
# the spectrum for the Cygnus cocoon comes from the joint ARGO+Fermi fit in Bartoli et al. 2014ApJ...790..152B
# with the cutoff at 40 TeV suggested by Milagro (Fig. 3)
spectrum_cyg = gammalib.GModelSpectralExpPlaw(3.5e-15, -2.16,
gammalib.GEnergy(0.1, 'TeV'),
gammalib.GEnergy(40, 'TeV'))
# the spectrum of Westerlund 1 comes from HESS collaboration 2011 A&A...537A.114A
spectrum_wd1 = gammalib.GModelSpectralPlaw(9.e-18, -2.19,
gammalib.GEnergy(1, 'TeV'))
spectral_models = [spectrum_cyg, spectrum_wd1]
for s, name in enumerate(names):
# first, create map
# create wcs for output map
npix = 1 + int(2 * rad_max[s] / out_res)
out_wcs = wcs.WCS(naxis=2)
out_wcs.wcs.crpix = [(npix - 1.) / 2 + 1., (npix - 1) / 2 + 1.]
out_wcs.wcs.cdelt = [-out_res, out_res]
out_wcs.wcs.crval = [glons[s], glats[s]] # centered on cluster
def set_source_model(srcmodel, spec='plaw'):
"""
Set source model
Parameters
----------
srcmodel : str
Source model name
spec : str, optional
Spectral model
Returns
-------
source : `~gammalib.GModelSky()`
Source model
"""
# Set spectral component
if spec == 'plaw':
spectral = gammalib.GModelSpectralPlaw(1.0e-18, -3.0,
gammalib.GEnergy(1.0, 'TeV'))
spectral['Prefactor'].min(1.0e-25)
elif spec == 'eplaw':
spectral = gammalib.GModelSpectralExpPlaw(
1.0e-18, -3.0, gammalib.GEnergy(1.0, 'TeV'),
gammalib.GEnergy(10.0, 'TeV'))
spectral['Prefactor'].min(1.0e-25)
spectral['CutoffEnergy'].min(3.0e6)
spectral['CutoffEnergy'].max(1.0e12)
elif spec == 'logparabola':
spectral = gammalib.GModelSpectralLogParabola(
1.0e-18, -3.0, gammalib.GEnergy(1.0, 'TeV'), -0.3)
spectral['Prefactor'].min(1.0e-25)
elif spec == 'fplaw':
spectral = gammalib.GModelSpectralPlawPhotonFlux(
1.0e-11, -3.0, gammalib.GEnergy(1.0, 'TeV'),
gammalib.GEnergy(100.0, 'TeV'))
spectral['PhotonFlux'].min(1.0e-25)
elif spec == 'aharonian2005':
spectral = gammalib.GModelSpectralPlaw(5.7e-18, -2.27,
gammalib.GEnergy(1.0, 'TeV'))
spectral['Prefactor'].fix()
spectral['Index'].fix()
elif spec == 'aharonian2005-fitk0':
spectral = gammalib.GModelSpectralPlaw(5.7e-18, -2.27,
gammalib.GEnergy(1.0, 'TeV'))
spectral['Prefactor'].min(1.0e-25)
spectral['Index'].fix()
elif spec == 'aharonian2005-fit':
spectral = gammalib.GModelSpectralPlaw(5.7e-18, -2.27,
gammalib.GEnergy(1.0, 'TeV'))
spectral['Prefactor'].min(1.0e-25)
# Set spatial component
dir = gammalib.GSkyDir()
dir.radec_deg(228.55, -59.17)
if srcmodel == 'ptsrc':
spatial = gammalib.GModelSpatialPointSource(dir)
spatial['RA'].free()
spatial['DEC'].free()
elif srcmodel == 'gauss':
spatial = gammalib.GModelSpatialRadialGauss(dir, 0.1)
spatial['Sigma'].min(0.0001)
spatial['Sigma'].free()
spatial['RA'].free()
spatial['DEC'].free()
elif srcmodel == 'egauss':
spatial = gammalib.GModelSpatialEllipticalGauss(dir, 0.05, 0.10, 60.0)
spatial['MinorRadius'].min(0.0001)
spatial['MinorRadius'].free()
spatial['MajorRadius'].min(0.0001)
spatial['MajorRadius'].free()
spatial['PA'].free()
spatial['RA'].free()
spatial['DEC'].free()
elif srcmodel == 'edisk':
spatial = gammalib.GModelSpatialEllipticalDisk(dir, 0.10, 0.20, 60.0)
spatial['MinorRadius'].min(0.0001)
spatial['MinorRadius'].free()
spatial['MajorRadius'].min(0.0001)
spatial['MajorRadius'].free()
spatial['PA'].free()
spatial['RA'].free()
spatial['DEC'].free()
elif srcmodel == 'aharonian2005':
dir = gammalib.GSkyDir()
dir.radec_deg(228.529, -59.159)
spatial = gammalib.GModelSpatialEllipticalGauss(
dir, 0.03833, 0.10666, 49.0)
spatial['MinorRadius'].fix()
spatial['MajorRadius'].fix()
spatial['PA'].fix()
spatial['RA'].fix()
spatial['DEC'].fix()
elif srcmodel == 'aharonian2005-fitpos':
dir = gammalib.GSkyDir()
dir.radec_deg(228.529, -59.159)
spatial = gammalib.GModelSpatialEllipticalGauss(
dir, 0.03833, 0.10666, 49.0)
spatial['MinorRadius'].fix()
spatial['MajorRadius'].fix()
spatial['PA'].fix()
spatial['RA'].free()
spatial['DEC'].free()
elif srcmodel == 'aharonian2005-fitpa':
dir = gammalib.GSkyDir()
dir.radec_deg(228.529, -59.159)
spatial = gammalib.GModelSpatialEllipticalGauss(
dir, 0.03833, 0.10666, 49.0)
spatial['MinorRadius'].fix()
spatial['MajorRadius'].fix()
spatial['PA'].free()
spatial['RA'].fix()
spatial['DEC'].fix()
# Set source model
source = gammalib.GModelSky(spatial, spectral)
source.name('MSH 15-52')
source.tscalc(True)
# Return source model
return source