Changeset 13028 for fact/tools/pyscripts/pyfact/extractor.py

Timestamp:

03/09/12 10:23:42 (13 years ago)

Author:

neise

Message:

added class WindowIntegrator, which integrate around a give position in a given window

File:

• fact/tools/pyscripts/pyfact/extractor.py (modified) (1 diff)

fact/tools/pyscripts/pyfact/extractor.py

@@ -41 +41 @@
         pass
 
+
+class WindowIntegrator(object):
+    """ Integrates in a given intergration window around the given position
+    """
+    
+    def __init__(self, min=13, max=23 , name = 'WindowIntegrator'):
+        """ initialize integration Window
+        """
+        self.min = min
+        self.max = max
+        self.name = name
+        
+    def __call__(self, data, pos):
+        integral = np.empty( data.shape[0] )
+        for pixel in range( data.shape[0] ):
+            integral[pixel] = data[pixel, (pos[pixel]-self.min):(pos[pixel]+self.max)].sum()
+        return integral
+
+    def __str__(self):
+        s = self.name + '\n'
+        s += 'window:\n'
+        s += '(min,max) = (' + str(self.min) + ',' + str(self.max) + ')'
+        return s    
 
 class FixedWindowIntegrator(object):