openwebrx-clone/owrx/modes.py

from owrx.feature import FeatureDetector
from functools import reduce


class Bandpass(object):
    def __init__(self, low_cut, high_cut):
        self.low_cut = low_cut
        self.high_cut = high_cut


class Mode(object):
    def __init__(self, modulation, name, bandpass: Bandpass = None, requirements=None, service=False):
        self.modulation = modulation
        self.name = name
        self.requirements = requirements if requirements is not None else []
        self.service = service
        self.bandpass = bandpass

    def is_available(self):
        fd = FeatureDetector()
        return reduce(
            lambda a, b: a and b, [fd.is_available(r) for r in self.requirements], True
        )

    def is_service(self):
        return self.service


class AnalogMode(Mode):
    pass


class DigitalMode(Mode):
    def __init__(self, modulation, name, underlying, bandpass: Bandpass = None, requirements=None, service=False):
        super().__init__(modulation, name, bandpass, requirements, service)
        self.underlying = underlying


class Modes(object):
    mappings = [
        AnalogMode("nfm", "FM", bandpass=Bandpass(-4000, 4000)),
        AnalogMode("am", "AM", bandpass=Bandpass(-4000, 4000)),
        AnalogMode("lsb", "LSB", bandpass=Bandpass(-3000, -300)),
        AnalogMode("usb", "USB", bandpass=Bandpass(300, 3000)),
        AnalogMode("cw", "CW", bandpass=Bandpass(700, 900)),
        AnalogMode("dmr", "DMR", bandpass=Bandpass(-4000, 4000), requirements=["digital_voice_digiham"]),
        AnalogMode("dstar", "DStar", bandpass=Bandpass(-3250, 3250), requirements=["digital_voice_dsd"]),
        AnalogMode("nxdn", "NXDN", bandpass=Bandpass(-3250, 3250), requirements=["digital_voice_dsd"]),
        AnalogMode("ysf", "YSF", bandpass=Bandpass(-4000, 4000), requirements=["digital_voice_digiham"]),
        DigitalMode("bpsk31", "BPSK31", underlying=["usb"]),
        DigitalMode("bpsk63", "BPSK63", underlying=["usb"]),
        DigitalMode("ft8", "FT8", underlying=["usb"], requirements=["wsjt-x"], service=True),
        DigitalMode("ft4", "FT4", underlying=["usb"], requirements=["wsjt-x"], service=True),
        DigitalMode("jt65", "JT65", underlying=["usb"], requirements=["wsjt-x"], service=True),
        DigitalMode("jt9", "JT9", underlying=["usb"], requirements=["wsjt-x"], service=True),
        DigitalMode("wspr", "WSPR", underlying=["usb"], bandpass=Bandpass(1350, 1650), requirements=["wsjt-x"], service=True),
        DigitalMode("js8", "JS8Call", underlying=["usb"], requirements=["js8call"], service=True),
        DigitalMode("packet", "Packet", underlying=["nfm", "usb", "lsb"], requirements=["packet"], service=True),
        DigitalMode("pocsag", "Pocsag", underlying=["nfm"], bandpass=Bandpass(-6000, 6000), requirements=["pocsag"])
    ]

    @staticmethod
    def getModes():
        return Modes.mappings

    @staticmethod
    def getAvailableModes():
        return [m for m in Modes.getModes() if m.is_available()]

    @staticmethod
    def getAvailableServices():
        return [m for m in Modes.getAvailableModes() if m.is_service()]

    @staticmethod
    def findByModulation(modulation):
        modes = [m for m in Modes.getAvailableModes() if m.modulation == modulation]
        if modes:
            return modes[0]