from csdr.chain.demodulator import FixedIfSampleRateChain, BaseDemodulatorChain, FixedAudioRateChain, DialFrequencyReceiver, HdAudio, MetaProvider, AudioServiceSelector
from csdr.module.hdradio import HdRadioModule
from pycsdr.modules import Convert, Agc, Downmix, Writer, Buffer, Throttle
from pycsdr.types import Format
from typing import Optional

import logging

logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)


class HdRadio(BaseDemodulatorChain, FixedIfSampleRateChain, FixedAudioRateChain, HdAudio, MetaProvider, DialFrequencyReceiver, AudioServiceSelector):
    def __init__(self, program: int = 0):
        self.hdradio = HdRadioModule(program = program)
        workers = [
            Agc(Format.COMPLEX_FLOAT),
            Convert(Format.COMPLEX_FLOAT, Format.COMPLEX_SHORT),
            self.hdradio,
            Throttle(Format.SHORT, 44100 * 2),
            Downmix(Format.SHORT),
        ]
        super().__init__(workers)

    def getFixedIfSampleRate(self) -> int:
        return self.hdradio.getFixedAudioRate()

    def getFixedAudioRate(self) -> int:
        return 44100

    def supportsSquelch(self) -> bool:
        return False

    # Set metadata consumer
    def setMetaWriter(self, writer: Writer) -> None:
        self.hdradio.setMetaWriter(writer)

    # Change program
    def setAudioServiceId(self, serviceId: int) -> None:
        self.hdradio.setProgram(serviceId)

    def setDialFrequency(self, frequency: int) -> None:
        self.hdradio.setFrequency(frequency)

    def _connect(self, w1, w2, buffer: Optional[Buffer] = None) -> None:
        if isinstance(w2, Throttle):
            # Audio data comes in in bursts, so we use a throttle
            # and 10x the default buffer size here
            buffer = Buffer(Format.SHORT, 2621440)
        return super()._connect(w1, w2, buffer)