add YSF chain; re-introduce RRC filters

csdr/chain/digiham.py

@@ -1,23 +1,25 @@
 from csdr.chain import Chain
 from pycsdr.modules import FmDemod, Agc, Writer
 from pycsdr.types import Format
-from digiham.modules import DstarDecoder, DcBlock, FskDemodulator, GfskDemodulator, DigitalVoiceFilter, MbeSynthesizer, NarrowRrcFilter, NxdnDecoder, DmrDecoder
+from digiham.modules import DstarDecoder, DcBlock, FskDemodulator, GfskDemodulator, DigitalVoiceFilter, MbeSynthesizer, NarrowRrcFilter, NxdnDecoder, DmrDecoder, WideRrcFilter, YsfDecoder
 from digiham.ambe import Modes
 
 
 class DigihamChain(Chain):
-    def __init__(self, codecserver: str = ""):
+    def __init__(self, fskDemodulator, decoder, mbeMode, filter=None, codecserver: str = ""):
+        self.decoder = decoder
         if codecserver is None:
             codecserver = ""
         agc = Agc(Format.SHORT)
         agc.setMaxGain(30)
         agc.setInitialGain(3)
-        workers = [
-            FmDemod(),
-            DcBlock(),
-            self.fskDemodulator,
-            self.decoder,
-            MbeSynthesizer(self.mbeMode, codecserver),
+        workers = [FmDemod(), DcBlock()]
+        if filter is not None:
+            workers += [filter]
+        workers += [
+            fskDemodulator,
+            decoder,
+            MbeSynthesizer(mbeMode, codecserver),
             DigitalVoiceFilter(),
             agc
         ]
@@ -29,26 +31,45 @@ class DigihamChain(Chain):
 
 class Dstar(DigihamChain):
     def __init__(self, codecserver: str = ""):
-        self.fskDemodulator = FskDemodulator(samplesPerSymbol=10)
-        self.decoder = DstarDecoder()
-        self.mbeMode = Modes.DStarMode
-        super().__init__(codecserver)
+        super().__init__(
+            fskDemodulator=FskDemodulator(samplesPerSymbol=10),
+            decoder=DstarDecoder(),
+            mbeMode=Modes.DStarMode,
+            codecserver=codecserver
+        )
 
 
 class Nxdn(DigihamChain):
     def __init__(self, codecserver: str = ""):
-        self.fskDemodulator = GfskDemodulator(samplesPerSymbol=20)
-        self.decoder = NxdnDecoder()
-        self.mbeMode = Modes.NxdnMode
-        super().__init__(codecserver)
+        super().__init__(
+            fskDemodulator=GfskDemodulator(samplesPerSymbol=20),
+            decoder=NxdnDecoder(),
+            mbeMode=Modes.NxdnMode,
+            filter=NarrowRrcFilter(),
+            codecserver=codecserver
+        )
 
 
 class Dmr(DigihamChain):
     def __init__(self, codecserver: str = ""):
-        self.fskDemodulator = GfskDemodulator(samplesPerSymbol=10)
-        self.decoder = DmrDecoder()
-        self.mbeMode = Modes.DmrMode
-        super().__init__(codecserver)
+        super().__init__(
+            fskDemodulator=GfskDemodulator(samplesPerSymbol=10),
+            decoder=DmrDecoder(),
+            mbeMode=Modes.DmrMode,
+            filter=WideRrcFilter(),
+            codecserver=codecserver,
+        )
 
-    def setSlotFilter(self, filter: int) -> None:
-        self.decoder.setSlotFilter(filter)
+    def setSlotFilter(self, slotFilter: int) -> None:
+        self.decoder.setSlotFilter(slotFilter)
+
+
+class Ysf(DigihamChain):
+    def __init__(self, codecserver: str = ""):
+        super().__init__(
+            fskDemodulator=GfskDemodulator(samplesPerSymbol=10),
+            decoder=YsfDecoder(),
+            mbeMode=Modes.YsfMode,
+            filter=WideRrcFilter(),
+            codecserver=codecserver
+        )