def impz(self, signal_in, a, b):
impulse = numpy.repeat(0., len(signal_in))
impulse[numpy.where(signal_in == 1)] = 1.
response = signal.lfilter(b, a, impulse)
return response
def stepz(self, signal_in, a, b):
step = numpy.repeat(0., len(signal_in))
step[numpy.where(signal_in >= 0)] = 1.
response = signal.lfilter(b, a, step)
return response
def filter(self, signal_in, coef_a, coef_b):
signal_out = signal.lfilter(coef_b, coef_a, signal_in)
return signal_out