openwebrx-clone/csdr/chain/__init__.py
2021-07-20 00:44:41 +02:00

102 lines
3.0 KiB
Python

from pycsdr.modules import Buffer
class Chain:
def __init__(self, *workers):
self.input = None
self.output = None
self.workers = list(workers)
for i in range(1, len(self.workers)):
self._connect(self.workers[i - 1], self.workers[i])
def _connect(self, w1, w2):
if isinstance(w1, Chain):
buffer = w1.getOutput()
else:
buffer = Buffer(w1.getOutputFormat())
w1.setOutput(buffer)
w2.setInput(buffer)
def stop(self):
for w in self.workers:
w.stop()
self.setInput(None)
if self.output is not None:
self.output.stop()
def setInput(self, buffer):
if self.input == buffer:
return
self.input = buffer
if self.workers:
self.workers[0].setInput(buffer)
else:
self.output = self.input
def getOutput(self):
if self.output is None:
if self.workers:
lastWorker = self.workers[-1]
if isinstance(lastWorker, Chain):
self.output = lastWorker.getOutput()
else:
self.output = Buffer(self.getOutputFormat())
self.workers[-1].setOutput(self.output)
else:
self.output = self.input
return self.output
def getOutputFormat(self):
if self.workers:
return self.workers[-1].getOutputFormat()
else:
return self.input.getOutputFormat()
def replace(self, index, newWorker):
if index >= len(self.workers):
raise IndexError("Index {} does not exist".format(index))
self.workers[index].stop()
self.workers[index] = newWorker
if index == 0:
newWorker.setInput(self.input)
else:
previousWorker = self.workers[index - 1]
if isinstance(previousWorker, Chain):
newWorker.setInput(previousWorker.getOutput())
else:
buffer = Buffer(previousWorker.getOutputFormat())
previousWorker.setOutput(buffer)
newWorker.setInput(buffer)
if index < len(self.workers) - 1:
nextWorker = self.workers[index + 1]
if isinstance(newWorker, Chain):
nextWorker.setInput(newWorker.getOutput())
else:
buffer = Buffer(newWorker.getOutputFormat())
newWorker.setOutput(buffer)
nextWorker.setInput(buffer)
else:
newWorker.setOutput(self.output)
def pump(self, write):
output = self.getOutput()
def copy():
run = True
while run:
data = None
try:
data = output.read()
except ValueError:
pass
if data is None or (isinstance(data, bytes) and len(data) == 0):
run = False
else:
write(data)
return copy