FilterClass.py

from scipy import signal
import numpy as np

class Filter: 
    
    def __init__(self, zerosReal=[],zerosImg=[], polesReal=[],polesImg=[],gain = 1):
        if zerosImg != [] and not isinstance(zerosImg[0],complex):
            for i in range(len(zerosImg)):
                zerosImg[i] *= 1j
        if polesImg != [] and not isinstance(polesImg[0],complex):
            for i in range(len(polesImg)):
                polesImg[i] *= 1j
        self.zeros = []
        for i in range(len(zerosReal)):
            self.zeros.append(zerosReal[i]+zerosImg[i])
        self.poles = []
        for i in range(len(polesReal)):
            self.poles.append(polesReal[i]+polesImg[i])
        self.gain = gain
    def setZeros(self,zeros):
        self.zeros = zeros
    def setPoles(self,poles):
        self.poles = poles
    def getZeros(self):
        return self.zeros
    
    def getPoles(self):
        return self.poles
    
    def getGain(self):
        return self.gain
    
    def getFreqAndComplexGain(self):
        print(self.getZeros(),self.getPoles())
        freq,complexGain = signal.freqz_zpk(self.getZeros(),self.getPoles(),self.getGain())
        return freq,complexGain
     
    def getMagInLogAndPhase(self):
        _,complexGain = self.getFreqAndComplexGain()
        magInLog = 20*np.log10(abs(complexGain))
        phase = np.unwrap(np.angle(complexGain))
        return magInLog,phase
    
    def getOutput(self,input):
        print('zp'*50)
        print(self.getZeros(),self.getPoles())
        num, den = signal.zpk2tf(self.getZeros(),self.getPoles(), self.getGain())
        output_signal =  signal.lfilter(num, den, input).real
        print(input,output_signal)
        return output_signal
    def getFilterOrder(self):
        order = max(len(self.getZeros()), len(self.getPoles()))
        return order