33 lines
1.4 KiB
Python
33 lines
1.4 KiB
Python
from csdr.chain import Chain
|
|
from pycsdr import Fft, LogAveragePower, FftExchangeSides, CompressFftAdpcm
|
|
|
|
import logging
|
|
logger = logging.getLogger(__name__)
|
|
logger.setLevel(logging.INFO)
|
|
|
|
|
|
class FftChain(Chain):
|
|
def __init__(self, fft_size, fft_block_size, fft_averages, fft_compression):
|
|
logger.debug("new fft: fft_size={0}, fft_block_size={1}, fft_averages={2}, fft_compression={3}".format(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 = [
|
|
self.fft,
|
|
self.logAveragePower,
|
|
self.fftExchangeSides,
|
|
]
|
|
self.compressFftAdpcm = None
|
|
if fft_compression == "adpcm":
|
|
self.compressFftAdpcm = CompressFftAdpcm(fft_size=fft_size)
|
|
workers += [self.compressFftAdpcm]
|
|
super().__init__(*workers)
|
|
|
|
def setFftAverages(self, fft_averages):
|
|
logger.debug("setting fft_averages={0}".format(fft_averages))
|
|
self.logAveragePower.setFftAverages(avg_number=fft_averages)
|
|
|
|
def setFftBlockSize(self, fft_block_size):
|
|
logger.debug("setting fft_block_size={0}".format(fft_block_size))
|
|
self.fft.setEveryNSamples(int(fft_block_size))
|