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