adopt to updated api
This commit is contained in:
@ -1,11 +1,10 @@
|
||||
from pycsdr.modules import Buffer
|
||||
from pycsdr.api import Flow
|
||||
|
||||
import logging
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class Chain(Flow):
|
||||
class Chain:
|
||||
def __init__(self, *workers):
|
||||
self.input = None
|
||||
self.output = None
|
||||
@ -14,7 +13,7 @@ class Chain(Flow):
|
||||
self._connect(self.workers[i - 1], self.workers[i])
|
||||
|
||||
def _connect(self, w1, w2):
|
||||
buffer = Buffer()
|
||||
buffer = Buffer(w1.getOutputFormat())
|
||||
w1.setOutput(buffer)
|
||||
w2.setInput(buffer)
|
||||
|
||||
@ -36,9 +35,12 @@ class Chain(Flow):
|
||||
self.output = buffer
|
||||
self.workers[-1].setOutput(buffer)
|
||||
|
||||
def getOutputFormat(self):
|
||||
return self.workers[-1].getOutputFormat()
|
||||
|
||||
def pump(self, write):
|
||||
if self.output is None:
|
||||
self.setOutput(Buffer())
|
||||
self.setOutput(Buffer(self.getOutputFormat()))
|
||||
|
||||
def copy():
|
||||
run = True
|
||||
|
@ -1,5 +1,5 @@
|
||||
from csdr.chain import Chain
|
||||
from pycsdr.modules import Fft, LogPower, LogAveragePower, FftExchangeSides, CompressFftAdpcm
|
||||
from pycsdr.modules import Fft, LogPower, LogAveragePower, FftSwap, FftAdpcm
|
||||
|
||||
import logging
|
||||
logger = logging.getLogger(__name__)
|
||||
@ -37,7 +37,7 @@ class FftAverager(Chain):
|
||||
if self.input is not None:
|
||||
self.worker.setInput(self.input)
|
||||
else:
|
||||
self.worker.setFftAverages(avg_number=fft_averages)
|
||||
self.worker.setAvgNumber(avg_number=fft_averages)
|
||||
self.workers = [self.worker]
|
||||
self.fftAverages = fft_averages
|
||||
|
||||
@ -53,7 +53,7 @@ class FftChain(Chain):
|
||||
|
||||
self.fft = Fft(size=self.size, every_n_samples=self.blockSize)
|
||||
self.averager = FftAverager(fft_size=self.size, fft_averages=10)
|
||||
self.fftExchangeSides = FftExchangeSides(fft_size=self.size)
|
||||
self.fftExchangeSides = FftSwap(fft_size=self.size)
|
||||
workers = [
|
||||
self.fft,
|
||||
self.averager,
|
||||
@ -61,7 +61,7 @@ class FftChain(Chain):
|
||||
]
|
||||
self.compressFftAdpcm = None
|
||||
if fft_compression == "adpcm":
|
||||
self.compressFftAdpcm = CompressFftAdpcm(fft_size=self.size)
|
||||
self.compressFftAdpcm = FftAdpcm(fft_size=self.size)
|
||||
workers += [self.compressFftAdpcm]
|
||||
|
||||
self._updateParameters()
|
||||
|
Reference in New Issue
Block a user