Add support for reading Fermi/gbm fits file in Eventlist.read

Author: ankitkhushwaha

stingray/events.py

@@ -14,7 +14,7 @@
 from .base import StingrayTimeseries
 from .filters import get_deadtime_mask
 from .gti import generate_indices_of_boundaries
-from .io import pi_to_energy, get_file_extension
+from .io import pi_to_energy, get_file_extension, read_header_key, read_gbm_data
 from .io import FITSTimeseriesReader
 from .lightcurve import Lightcurve
 from .utils import simon, njit
@@ -627,6 +627,14 @@ def read(cls, filename, fmt=None, rmf_file=None, **kwargs):
                 if fits_ext in get_file_extension(filename).lower():
                     fmt = "hea"
                     break
+
+        source = read_header_key(filename, "INSTRUME")
+        if source == "GBM":
+            emin = kwargs.pop("emin")
+            emax = kwargs.pop("emax")
+            evt = read_gbm_data(filename, emin=emin, emax=emax)
+            return evt
+
         if fmt is not None and fmt.lower() in ("hea", "ogip"):
             additional_columns = kwargs.pop("additional_columns", None)
 

stingray/io.py

@@ -16,6 +16,7 @@
 
 import stingray.utils as utils
 from stingray.loggingconfig import setup_logger
+from stingray.events import EventList
 
 
 from .utils import (
@@ -1550,3 +1551,51 @@ def _can_serialize_meta(probe_file: str, fmt: str) -> bool:
         )
         yes_it_can = False
     return yes_it_can
+
+
+def read_gbm_data(filename, emin=None, emax=None):
+    """
+    Read Fermi/GBM detector data.
+
+    Parameters
+    ----------
+    filename: str
+        Path to the fits file, Only TTE files are supported.
+
+    emin, emax : float
+        The minimum and maximum energies to use
+
+    Returns
+    -------
+    ev: :class:`EventList` object
+        The :class:`EventList` object reconstructed from file
+    """
+    hdulist = fits.open(filename)
+    header = hdulist[0].header
+
+    elower = hdulist[1].data.field("E_MIN")
+    ehigher = hdulist[1].data.field("E_MAX")
+    emid = elower + (ehigher - elower) / 2.0
+
+    time = hdulist[2].data.field("TIME")
+    channels = hdulist[2].data.field("PHA")
+
+    if emin is None:
+        emin = np.min(elower)
+    if emax is None:
+        emax = np.max(ehigher)
+    for c in channels:
+        emid[c]
+
+    energy = np.array([emid[c] for c in channels])
+    mask = (energy > emin) & (energy < emax)
+
+    time = time[mask]
+    energy = energy[mask]
+    trigtime = hdulist[0].header["TRIGTIME"]
+
+    hdulist.close()
+
+    evt = EventList(time, energy, header)
+    evt.trigtime = trigtime
+    return evt