39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from csdr.chain import Chain
|
|
from csdr.chain.selector import Selector
|
|
from csdr.chain.demodulator import BaseDemodulatorChain, ServiceDemodulator
|
|
from pycsdr.types import Format
|
|
|
|
|
|
class ServiceDemodulatorChain(Chain):
|
|
def __init__(self, demod: BaseDemodulatorChain, secondaryDemod: ServiceDemodulator, sampleRate: int, frequencyOffset: int):
|
|
self.frequency = None
|
|
self.mode = None
|
|
|
|
self.selector = Selector(sampleRate, secondaryDemod.getFixedAudioRate(), withSquelch=False)
|
|
self.selector.setFrequencyOffset(frequencyOffset)
|
|
|
|
workers = [self.selector]
|
|
|
|
# primary demodulator is only necessary if the secondary does not accept IQ input
|
|
if secondaryDemod.getInputFormat() is not Format.COMPLEX_FLOAT:
|
|
workers += [demod]
|
|
|
|
workers += [secondaryDemod]
|
|
|
|
super().__init__(workers)
|
|
|
|
def setBandPass(self, lowCut, highCut):
|
|
self.selector.setBandpass(lowCut, highCut)
|
|
|
|
def setFrequency(self, frequency):
|
|
self.frequency = frequency
|
|
|
|
def setMode(self, mode):
|
|
self.mode = mode
|
|
|
|
def getFrequency(self):
|
|
return self.frequency
|
|
|
|
def getMode(self):
|
|
return self.mode
|