summaryrefslogtreecommitdiff
path: root/gr-fec/python/fec/polar/channel_construction_bec.py
diff options
context:
space:
mode:
Diffstat (limited to 'gr-fec/python/fec/polar/channel_construction_bec.py')
-rw-r--r--gr-fec/python/fec/polar/channel_construction_bec.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/gr-fec/python/fec/polar/channel_construction_bec.py b/gr-fec/python/fec/polar/channel_construction_bec.py
index cd61f463f6..4d3654d010 100644
--- a/gr-fec/python/fec/polar/channel_construction_bec.py
+++ b/gr-fec/python/fec/polar/channel_construction_bec.py
@@ -87,6 +87,12 @@ def get_frozen_bit_indices_from_capacities(chan_caps, nfrozen):
return np.sort(indexes)
+def get_bec_frozen_indices(nblock, kfrozen, eta):
+ bec_caps = calculate_bec_channel_capacities(eta, nblock)
+ positions = get_frozen_bit_indices_from_capacities(bec_caps, kfrozen)
+ return positions
+
+
def bec_channel_contruction_tests():
n = 2 ** 10
k = n // 2