def test_compute_thresholds_from_crab_data():
"""Obs read from file"""
arffile = "$GAMMAPY_DATA/joint-crab/spectra/hess/arf_obs23523.fits"
rmffile = "$GAMMAPY_DATA/joint-crab/spectra/hess/rmf_obs23523.fits"
aeff = EffectiveAreaTable.read(arffile)
edisp = EnergyDispersion.read(rmffile)
thresh_lo, thresh_hi = compute_energy_thresholds(
aeff=aeff,
edisp=edisp,
method_lo="energy_bias",
method_hi="none",
bias_percent_lo=10,
bias_percent_hi=10,
)
assert_allclose(thresh_lo.to("TeV").value, 0.9174, rtol=1e-4)
assert_allclose(thresh_hi.to("TeV").value, 100.0, rtol=1e-4)
def test_EffectiveAreaTable(tmp_path, aeff):
arf = aeff.to_effective_area_table(offset=0.3 * u.deg)
assert_quantity_allclose(arf.data.evaluate(), arf.data.data)
with mpl_plot_check():
arf.plot()
arf.write(tmp_path / "tmp.fits")
arf2 = EffectiveAreaTable.read(tmp_path / "tmp.fits")
assert_quantity_allclose(arf.data.evaluate(), arf2.data.evaluate())
test_aeff = 0.6 * arf.max_area
node_above = np.where(arf.data.data > test_aeff)[0][0]
energy = arf.data.axes["energy_true"]
ener_above = energy.center[node_above]
ener_below = energy.center[node_above - 1]
test_ener = arf.find_energy(test_aeff)
assert ener_below < test_ener and test_ener < ener_above
elo_threshold = arf.find_energy(0.1 * arf.max_area)
assert elo_threshold.unit == "TeV"
assert_allclose(elo_threshold.value, 0.554086, rtol=1e-3)
ehi_threshold = arf.find_energy(0.9 * arf.max_area,
emin=30 * u.TeV,
emax=100 * u.TeV)
assert ehi_threshold.unit == "TeV"
assert_allclose(ehi_threshold.value, 53.347217, rtol=1e-3)
# Test evaluation outside safe range
data = [np.nan, np.nan, 0, 0, 1, 2, 3, np.nan, np.nan]
energy_axis_true = MapAxis.from_energy_bounds("1 TeV",
"10 TeV",
nbin=9,
name="energy_true")
aeff = EffectiveAreaTable(data=data, energy_axis_true=energy_axis_true)
vals = aeff.evaluate_fill_nan()
assert vals[1] == 0
assert vals[-1] == 3
def test_EffectiveAreaTable(tmpdir, aeff):
arf = aeff.to_effective_area_table(offset=0.3 * u.deg)
assert_quantity_allclose(arf.data.evaluate(), arf.data.data)
with mpl_plot_check():
arf.plot()
filename = str(tmpdir / "effarea_test.fits")
arf.write(filename)
arf2 = EffectiveAreaTable.read(filename)
assert_quantity_allclose(arf.data.evaluate(), arf2.data.evaluate())
test_aeff = 0.6 * arf.max_area
node_above = np.where(arf.data.data > test_aeff)[0][0]
energy = arf.data.axis("energy")
ener_above = energy.center[node_above]
ener_below = energy.center[node_above - 1]
test_ener = arf.find_energy(test_aeff)
assert ener_below < test_ener and test_ener < ener_above
elo_threshold = arf.find_energy(0.1 * arf.max_area)
assert elo_threshold.unit == "TeV"
assert_allclose(elo_threshold.value, 0.554086, rtol=1e-3)
ehi_threshold = arf.find_energy(0.9 * arf.max_area,
emin=30 * u.TeV,
emax=100 * u.TeV)
assert ehi_threshold.unit == "TeV"
assert_allclose(ehi_threshold.value, 53.347217, rtol=1e-3)
# Test evaluation outside safe range
data = [np.nan, np.nan, 0, 0, 1, 2, 3, np.nan, np.nan]
energy = np.logspace(0, 10, 10) * u.TeV
aeff = EffectiveAreaTable(data=data,
energy_lo=energy[:-1],
energy_hi=energy[1:])
vals = aeff.evaluate_fill_nan()
assert vals[1] == 0
assert vals[-1] == 3
def from_ogip_files(cls, filename):
"""Read `~gammapy.spectrum.SpectrumDatasetOnOff` from OGIP files.
BKG file, ARF, and RMF must be set in the PHA header and be present in
the same folder.
The naming scheme is fixed to the following scheme
* PHA file is named pha_obs{name}.fits
* BKG file is named bkg_obs{name}.fits
* ARF file is named arf_obs{name}.fits
* RMF file is named rmf_obs{name}.fits
with {name} the dataset name.
Parameters
----------
filename : str
OGIP PHA file to read
"""
filename = make_path(filename)
dirname = filename.parent
with fits.open(filename, memmap=False) as hdulist:
data = _read_ogip_hdulist(hdulist)
counts = CountsSpectrum(energy_hi=data["energy_hi"],
energy_lo=data["energy_lo"],
data=data["data"])
phafile = filename.name
try:
rmffile = phafile.replace("pha", "rmf")
energy_dispersion = EnergyDispersion.read(dirname / rmffile)
except OSError:
# TODO : Add logger and echo warning
energy_dispersion = None
try:
bkgfile = phafile.replace("pha", "bkg")
with fits.open(dirname / bkgfile, memmap=False) as hdulist:
data_bkg = _read_ogip_hdulist(hdulist)
counts_off = CountsSpectrum(
energy_hi=data_bkg["energy_hi"],
energy_lo=data_bkg["energy_lo"],
data=data_bkg["data"],
)
acceptance_off = data_bkg["backscal"]
except OSError:
# TODO : Add logger and echo warning
counts_off, acceptance_off = None, None
arffile = phafile.replace("pha", "arf")
aeff = EffectiveAreaTable.read(dirname / arffile)
mask_safe = np.logical_not(data["quality"])
return cls(
counts=counts,
aeff=aeff,
counts_off=counts_off,
edisp=energy_dispersion,
livetime=data["livetime"],
mask_safe=mask_safe,
acceptance=data["backscal"],
acceptance_off=acceptance_off,
name=str(data["obs_id"]),
gti=data["gti"],
)
def from_ogip_files(cls, filename):
"""Read `~gammapy.spectrum.SpectrumDatasetOnOff` from OGIP files.
BKG file, ARF, and RMF must be set in the PHA header and be present in
the same folder.
The naming scheme is fixed to the following scheme:
* PHA file is named ``pha_obs{name}.fits``
* BKG file is named ``bkg_obs{name}.fits``
* ARF file is named ``arf_obs{name}.fits``
* RMF file is named ``rmf_obs{name}.fits``
with ``{name}`` the dataset name.
Parameters
----------
filename : str
OGIP PHA file to read
"""
filename = make_path(filename)
dirname = filename.parent
with fits.open(filename, memmap=False) as hdulist:
counts = RegionNDMap.from_hdulist(hdulist, format="ogip")
acceptance = RegionNDMap.from_hdulist(hdulist,
format="ogip",
ogip_column="BACKSCAL")
if "GTI" in hdulist:
gti = GTI(Table.read(hdulist["GTI"]))
else:
gti = None
mask_safe = RegionNDMap.from_hdulist(hdulist,
format="ogip",
ogip_column="QUALITY")
mask_safe.data = np.logical_not(mask_safe.data)
phafile = filename.name
try:
rmffile = phafile.replace("pha", "rmf")
kernel = EDispKernel.read(dirname / rmffile)
edisp = EDispKernelMap.from_edisp_kernel(kernel, geom=counts.geom)
except OSError:
# TODO : Add logger and echo warning
edisp = None
try:
bkgfile = phafile.replace("pha", "bkg")
with fits.open(dirname / bkgfile, memmap=False) as hdulist:
counts_off = RegionNDMap.from_hdulist(hdulist, format="ogip")
acceptance_off = RegionNDMap.from_hdulist(
hdulist, ogip_column="BACKSCAL")
except OSError:
# TODO : Add logger and echo warning
counts_off, acceptance_off = None, None
arffile = phafile.replace("pha", "arf")
aeff = EffectiveAreaTable.read(dirname / arffile)
return cls(
counts=counts,
aeff=aeff,
counts_off=counts_off,
edisp=edisp,
livetime=counts.meta["EXPOSURE"] * u.s,
mask_safe=mask_safe,
acceptance=acceptance,
acceptance_off=acceptance_off,
name=str(counts.meta["OBS_ID"]),
gti=gti,
)
