updated init and load calls with extension keyword

Author: eblur

specutils/spectra/xrayspectrum1d.py

@@ -186,7 +186,7 @@ def apply_response(self, mflux, exposure=None):
 ## ----  Supporting response file objects
 
 class ResponseMatrix(object):
-    def __init__(self, filename):
+    def __init__(self, filename, extension=None):
         """
         Load a response matrix file (RMF) from a FITS file.
 
@@ -244,7 +244,7 @@ def __init__(self, filename):
         self.energ_hi = None
         self.energ_unit = None
         self.detchans = None
-        self._load_rmf(filename)
+        self._load_rmf(filename, extension=extension)
 
     def _load_rmf(self, filename, extension=None):
         # open the FITS file and extract the MATRIX extension
@@ -463,7 +463,7 @@ def apply_rmf(self, spec):
 
 class AreaResponse(object):
 
-    def __init__(self, filename):
+    def __init__(self, filename, extension=None):
         """
         Load an area response file (ARF) from a FITS file.
 
@@ -472,6 +472,10 @@ def __init__(self, filename):
         filename : str
             The file name with the ARF file
 
+        extension : str (default None)
+            FITS file extension keyword, if the spectral response is stored
+            under an extension other than "SPECRESP"
+
         Attributes
         ----------
         filename : str
@@ -506,7 +510,7 @@ def __init__(self, filename):
         self.specresp = None
         self.fracexpo = None
         self.exposure = None
-        self._load_arf(filename)
+        self._load_arf(filename, extension=extension)
 
     def _load_arf(self, filename, extension=None):
         # open the FITS file and extract the SPECRESP extension