Method filter

pytta/classes/filter.py:121–165

Filter the signal object. For each channel inside the input signalObj, will be generated a new SignalObj with the channel filtered signal. Args: sigobj: SignalObj Return: output: List A list containing one SignalObj

(self, sigobj)

    #     return self._filter(sigobj)

    def filter(self, sigobj):
        """
        Filter the signal object.

        For each channel inside the input signalObj, will be generated a new
        SignalObj with the channel filtered signal.

        Args:
            sigobj: SignalObj

        Return:
            output: List
                A list containing one SignalObj with the filtered data for each
                channel in the original signalObj.

        """
        if self.samplingRate != sigobj.samplingRate:
            raise ValueError("SignalObj must have same sampling rate of filter to be filtered.")
        n = self.sos.shape[2]
        output = []
        chl = []
        for ch in range(sigobj.numChannels):
            sobj = sigobj[ch]
            filtered = np.zeros((sobj.numSamples, n))
            for k in range(n):
                cContigousArray = sobj.timeSignal[:].copy(order='C')
                filtered[:, k] = ss.sosfilt(self.sos[:, :, k].copy(order='C'),
                                            cContigousArray,
                                            axis=0).T
                chl.append(copy(sobj.channels[sobj.channels.mapping[0]]))
                chl[-1].num = k+1
                chl[-1].name = f'Band {k+1}'
                chl[-1].code = f'B{k+1}'
            signalDict = {'signalArray': filtered,
                          'domain': 'time',
                          'samplingRate': self.samplingRate,
                          'freqMin': sigobj.freqMin,
                          'freqMax': sigobj.freqMax,
                          }
            out = SignalObj(**signalDict)
            out.channels = ChannelsList(chl)
            # out.timeSignal = out.timeSignal * out.channels.CFlist()
            output.append(out)
            chl.clear()
        return output