Reorganize code of SC List Polar codec

Author: grigory

examples/modelling/functions.py

@@ -1,6 +1,6 @@
-from datetime import datetime
 from math import ceil
 from random import shuffle
+from datetime import datetime
 
 import numpy as np
 from pymongo import MongoClient

examples/modelling/runner.py

@@ -1,10 +1,12 @@
 import multiprocessing
-from concurrent import futures
 from datetime import datetime
 from functools import partial
+from concurrent import futures
 
-from examples.modelling.functions import (generate_simulation_parameters,
-                                          simulation_task)
+from examples.modelling.functions import (
+    generate_simulation_parameters,
+    simulation_task,
+)
 from examples.modelling.mongo import DB_NAME
 from python_polar_coding.channels.simple import SimpleBPSKModulationAWGN
 

python_polar_coding/polar_codes/decoders/sc_list_decoder.py

@@ -0,0 +1,2 @@
+from .codec import SCListPolarCodec
+from .decoder import SCListDecoder

python_polar_coding/polar_codes/sc_list.py

@@ -0,0 +1,32 @@
+from typing import Union
+
+from ..base import BasePolarCodec
+from .decoder import SCListDecoder
+
+
+class SCListPolarCodec(BasePolarCodec):
+    """Polar code with SC List decoding algorithm."""
+    decoder_class = SCListDecoder
+
+    def __init__(self, N: int, K: int,
+                 design_snr: float = 0.0,
+                 is_systematic: bool = True,
+                 mask: Union[str, None] = None,
+                 pcc_method: str = BasePolarCodec.BHATTACHARYYA,
+                 L: int = 1):
+
+        self.L = L
+        super().__init__(N=N, K=K,
+                         is_systematic=is_systematic,
+                         design_snr=design_snr,
+                         mask=mask,
+                         pcc_method=pcc_method)
+
+    def init_decoder(self):
+        return self.decoder_class(n=self.n, mask=self.mask,
+                                  is_systematic=self.is_systematic, L=self.L)
+
+    def to_dict(self):
+        d = super().to_dict()
+        d.update({'L': self.L})
+        return d