Implemented querying of rectangular region parameters.

Andreas Specovius · Jürgen Knödlseder
1 parent 17366f6b
Showing 1 changed file with 22 additions and 8 deletions
cscripts/csphagen.py
@@ -184,21 +184,35 @@ class csphagen(ctools.csobservation):
         # Get source shape
         self._srcshape = self['srcshape'].string()
-        # Set source region (so far only CIRCLE is supported)
-        if self._srcshape == 'CIRCLE':
+        # Query source direction
+        self._query_src_direction()
-            # Query source direction
-            self._query_src_direction()
+        # Query minimum number of background regions and number of
+        # background regions to skip next to On region
+        self['bkgregmin'].integer()
+        self['bkgregskip'].integer()
-            # Query minimum number of background regions and number of
-            # background regions to skip next to On region
-            self['bkgregmin'].integer()
-            self['bkgregskip'].integer()
+        # Set source region (so far only CIRCLE is supported)
+        if self._srcshape == 'CIRCLE':
             # Set circular source region
             self._rad = self['rad'].real()
             self._src_reg.append(gammalib.GSkyRegionCircle(self._src_dir, self._rad))
+        elif self._srcshape == 'RECT':
+
+            # Set rectangular source region
+            self._reg_width  = self['width'].real()
+            self._reg_height = self['height'].real()
+            self._reg_posang = self['posang'].real()
+            self._src_reg.append(gammalib.GSkyRegionRect(self._src_dir
+                                                        ,self._reg_width
+                                                        ,self._reg_height
+                                                        ,self._reg_posang
+                                                        )
+                                )
+
+
         # Query usage of background model
         self['use_model_bkg'].boolean()