Changeset 12948 for fact/tools/pyscripts/pyfact

Timestamp:

02/28/12 17:37:03 (13 years ago)

Author:

neise

Message:

filters use the 1st sample as initial values, instead of zero.

File:

• fact/tools/pyscripts/pyfact/fir_filter.py (modified) (4 diffs)

fact/tools/pyscripts/pyfact/fir_filter.py

@@ -17 +17 @@
         
         """
-        
         self.a = a
         self.b = b
@@ -23 +22 @@
         
     def __call__(self, data):
-        return signal.lfilter(self.b, self.a, data)
+        length = max(len(self.a),len(self.b))-1
+        initial = np.ones(length)
+        initial *= data[0]
+        filtered, zf = signal.lfilter(self.b, self.a, data, zi=initial)
+        return filtered
 
     def __str__(self):
         s = self.name + '\n'
+        s += 'initial condition for filter: signal@rest = 1st sample\n'
         s += 'filter, coefficients:\n'
         s += 'nominator ' + str(self.b) + '\n'
@@ -42 +46 @@
     
         """
-
         b = np.ones(length)
-        a = len(b)
+        a = np.zeros(length)
+        a[0] = len(b)
         FirFilter.__init__(self, b, a, 'sliding average')
             
@@ -70 +74 @@
         
         b = np.array((-0.5, 1., -0.5)) 
-        a = 1.
+        a = np.zeros(len(b))
+        a[0] = 1.0
         FirFilter.__init__(self, b, a, 'remove signal')