Changeset 12953 for fact/tools/pyscripts

Timestamp:

02/28/12 18:29:56 (13 years ago)

Author:

neise

Message:

implemented correcto handling for 2D data arrays ... 3D will still fail.

File:

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

fact/tools/pyscripts/pyfact/fir_filter.py

@@ -23 +23 @@
     def __call__(self, data):
         length = max(len(self.a),len(self.b))-1
-        initial = np.ones(length)
-        initial *= data[0]
+        
+        #print data
+        #print data.ndim
+        #print data.shape
+        if ( data.ndim == 1):
+            initial = np.ones(length)
+            initial *= data[0]
+        elif ( data.ndim == 2):
+            initial = np.ones( (data.shape[0], length) )
+            for i in range(data.shape[0]):
+                initial[i,:] *= data[i,0]
+        else:
+            print 'HELP.'
+            pass
+        
+        #print 'initial:\n' , initial
         filtered, zf = signal.lfilter(self.b, self.a, data, zi=initial)
+        #filtered= signal.lfilter(self.b, self.a, data)
         return filtered