Add hidden etruebins parameter to csphagen that allows setting of number of true…

… energy bins per decade.

Add hidden etruebins parameter to csphagen that allows setting of number of true…
… energy bins per decade.
Jürgen Knödlseder
1 parent 07c8ec4a
Showing 3 changed files with 65 additions and 3 deletions
cscripts/csphagen.par
cscripts/csphagen.py
doc/source/users/reference_manual/csphagen.rst
@@ -42,6 +42,7 @@ bkgmethod, s, h, REFLECTED,REFLECTED,, &quot;Method for background estimation&quot;
 bkgregmin, i, h, 2,1,1000, "Minimum number of background regions"
 maxoffset, r, h, 4.0,,, "Maximum offset of source from camera center to accept observation"
 stack,     b, a, no,,, "Stack multiple observations into single PHA, ARF and RMF files?"
+etruebins, i, h, 30,,, "Number of bins per decade for true energy bins"
  
 #
 # Standard parameters
@@ -121,7 +121,7 @@ class csphagen(ctools.csobservation):
  
         Parameters
         ----------
-        obs : `~gammalib.GObservation()`
+        obs : `~gammalib.GCTAObservation()`
             CTA observation
  
         Returns
@@ -147,7 +147,7 @@ class csphagen(ctools.csobservation):
             if self._srcshape == 'CIRCLE':
                 # angular separation of reflected regions wrt camera center
                 # and number
-                alpha = 1.05 * 2 * self._rad / offset
+                alpha = 1.05 * 2.0 * self._rad / offset
                 # 1.05 ensures background regions do not overlap due to
                 # numerical precision issues
                 N = int(2.0 * math.pi / alpha)
@@ -179,6 +179,60 @@ class csphagen(ctools.csobservation):
         # Return reflected regions
         return regions
  
+    def _set_models(self, obs):
+        """
+        Set models in observation container
+
+        Parameters
+        ----------
+        obs : `~gammalib.GObservations()`
+            Observation container
+
+        Returns
+        -------
+        obs : `~gammalib.GObservations()`
+            Observation container
+        """
+        # Initialise model container
+        models = gammalib.GModels()
+
+        # Loop over all models and replace CTA background model by CTAOnOff
+        # background model
+        for model in self.obs().models():
+            if 'GCTA' in model.classname() and 'CTA' in model.instruments():
+                model.instruments('CTAOnOff')
+            models.append(model)
+
+        # Append model to observation container
+        obs.models(models)
+
+        # Return observation container
+        return obs
+
+    def _etrue_ebounds(self):
+        """
+        Set true energy boundaries
+
+        Returns
+        -------
+        ebounds : `~gammalib.GEbounds()`
+            True energy boundaries
+        """
+        # Determine minimum and maximum energies
+        emin = self._ebounds.emin()
+        emax = self._ebounds.emax()
+
+        # Determine number of energy bins
+        n_decades = (emax.log10TeV() - emin.log10TeV())
+        n_bins    = int(n_decades * float(self['etruebins'].integer())) + 1
+
+        # Set energy boundaries
+        ebounds = gammalib.GEbounds(n_bins, emin, emax)
+
+        # Return energy boundaries
+        return ebounds
+
+
     # Public methods
     def run(self):
         """
@@ -201,6 +255,7 @@ class csphagen(ctools.csobservation):
         # Initialise run variables
         outobs         = gammalib.GObservations()
         self._bkg_regs = []
+        etrue_ebounds  = self._etrue_ebounds()
  
         # Loop through observations and generate pha, arf, rmf files
         for obs in self.obs():
@@ -217,7 +272,7 @@ class csphagen(ctools.csobservation):
             # and append it to the output container
             if bkg_reg.size() >= self['bkgregmin'].integer():
                 onoff = gammalib.GCTAOnOffObservation(obs, self._src_dir,
-                                                      self._ebounds,
+                                                      etrue_ebounds,
                                                       self._ebounds,
                                                       self._src_reg,
                                                       bkg_reg)
@@ -242,6 +297,9 @@ class csphagen(ctools.csobservation):
             outobs = gammalib.GObservations()
             outobs.append(stacked_obs)
  
+        # Set models in output container
+        outobs = self._set_models(outobs)
+
         # Set observation container
         self.obs(outobs)
  
@@ -108,6 +108,9 @@ General parameters
     Specifies whether multiple observations should be stacked (``yes``) or
     whether run-wise PHA, ARF and RMF files should be produced (``no``).
  
+``(etruebins = 30) [integer]``
+    Number of bins per decade for true energy bins.
+
  
 Standard parameters
 -------------------