move fft calculations to dsp class

This commit is contained in:
Jakob Ketterl
2020-12-19 16:41:48 +01:00
parent efe80a75f5
commit 40c07ebb57
3 changed files with 35 additions and 19 deletions

View File

@ -4,11 +4,16 @@ from pycsdr import Fft, LogAveragePower, FftExchangeSides, CompressFftAdpcm
class FftChain(Chain):
def __init__(self, fft_size, fft_block_size, fft_averages, fft_compression):
self.fft = Fft(size=fft_size, every_n_samples=int(fft_block_size))
self.logAveragePower = LogAveragePower(add_db=-70, fft_size=fft_size, avg_number=fft_averages)
self.fftExchangeSides = FftExchangeSides(fft_size=fft_size)
workers = [
Fft(size=fft_size, every_n_samples=int(fft_block_size)),
LogAveragePower(add_db=-70, fft_size=fft_size, avg_number=fft_averages),
FftExchangeSides(fft_size=fft_size),
self.fft,
self.logAveragePower,
self.fftExchangeSides,
]
self.compressFftAdpcm = None
if fft_compression == "adpcm":
workers += [CompressFftAdpcm(fft_size=fft_size)]
self.compressFftAdpcm = CompressFftAdpcm(fft_size=fft_size)
workers += [self.compressFftAdpcm]
super().__init__(*workers)