update fft parameters without restarting

This commit is contained in:
Jakob Ketterl 2020-12-19 17:13:36 +01:00
parent 40c07ebb57
commit 1083d51e18
2 changed files with 38 additions and 9 deletions

View File

@ -1,9 +1,14 @@
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)
@ -17,3 +22,11 @@ class FftChain(Chain):
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))

View File

@ -113,6 +113,7 @@ class dsp(object):
self.secondary_process_fft = None
self.secondary_process_demod = None
self.fft_voverlap_factor = 0
self.fft_block_size = 0
self.pipe_names = {
"bpf_pipe": Pipe.WRITE,
"shift_pipe": Pipe.WRITE,
@ -509,6 +510,8 @@ class dsp(object):
return int(base)
def set_fft_compression(self, what):
if self.fft_compression == what:
return
self.fft_compression = what
def get_fft_bytes_to_read(self):
@ -529,7 +532,8 @@ class dsp(object):
self.samp_rate = samp_rate
self.calculate_decimation()
self.update_fft_averages()
if self.running:
self.update_fft_block_size()
if self.running and self.pycsdr_chain is None:
self.restart()
def calculate_decimation(self):
@ -654,6 +658,7 @@ class dsp(object):
return
self.fft_fps = fft_fps
self.update_fft_averages()
self.update_fft_block_size()
self.restart()
def set_fft_voverlap_factor(self, fft_voverlap_factor):
@ -673,13 +678,24 @@ class dsp(object):
if self.fft_averages == fft_averages:
return
self.fft_averages = fft_averages
self.update_fft_block_size()
if self.pycsdr_chain:
self.pycsdr_chain.setFftAverages(fft_averages)
else:
self.restart()
def fft_block_size(self):
def update_fft_block_size(self):
if self.fft_averages == 0:
return self.samp_rate / self.fft_fps
self.set_fft_block_size(self.samp_rate / self.fft_fps)
else:
return self.samp_rate / self.fft_fps / self.fft_averages
self.set_fft_block_size(self.samp_rate / self.fft_fps / self.fft_averages)
def set_fft_block_size(self, fft_block_size):
if self.fft_block_size == fft_block_size:
return
self.fft_block_size = fft_block_size
if self.pycsdr_chain:
self.pycsdr_chain.setFftBlockSize(fft_block_size)
def set_offset_freq(self, offset_freq):
if offset_freq is None:
@ -798,7 +814,7 @@ class dsp(object):
self.pycsdr_chain = FftChain(
fft_size=self.fft_size,
fft_block_size=self.fft_block_size(),
fft_block_size=self.fft_block_size,
fft_averages=self.fft_averages,
fft_compression=self.fft_compression
)
@ -842,7 +858,7 @@ class dsp(object):
decimation=self.decimation,
last_decimation=self.last_decimation,
fft_size=self.fft_size,
fft_block_size=self.fft_block_size(),
fft_block_size=self.fft_block_size,
fft_averages=self.fft_averages,
bpf_transition_bw=float(self.bpf_transition_bw) / self.if_samp_rate(),
ddc_transition_bw=self.ddc_transition_bw(),