This is merely a historical archive of years 2008-2021, before the migration to mailman3.
A maintained and still updated list archive can be found at https://lists.osmocom.org/hyperkitty/list/gerrit-log@lists.osmocom.org/.
Vadim Yanitskiy gerrit-no-reply at lists.osmocom.org
Review at https://gerrit.osmocom.org/1585
utils/conv_gen.py: add test vector generation feature
Change-Id: Ie10c47ee952f253b1ba77ecf6e79f2c033545bc1
---
M utils/conv_gen.py
1 file changed, 81 insertions(+), 5 deletions(-)
git pull ssh://gerrit.osmocom.org:29418/libosmocore refs/changes/85/1585/1
diff --git a/utils/conv_gen.py b/utils/conv_gen.py
index 57e9ca6..34e164a 100644
--- a/utils/conv_gen.py
+++ b/utils/conv_gen.py
@@ -30,13 +30,16 @@
class ConvolutionalCode(object):
def __init__(self, block_len, polys, name,
- description = None, puncture = [], term_type = None):
+ description = None, puncture = [], term_type = None,
+ vec_in = None, vec_out = None):
# Save simple params
self.block_len = block_len
self.k = 1
self.puncture = puncture
self.rate_inv = len(polys)
self.term_type = term_type
+ self.vec_in = vec_in
+ self.vec_out = vec_out
# Infos
self.name = name
@@ -226,6 +229,44 @@
fi.write("\t.puncture = %s_puncture,\n" % self.name)
fi.write("};\n\n")
+ def calc_out_len(self):
+ out_len = self.block_len * self.rate_inv
+
+ # By default CONV_TERM_FLUSH
+ if self.term_type is None:
+ out_len += self.rate_inv * (self.k - 1)
+
+ if len(self.puncture):
+ out_len -= len(self.puncture) - 1
+
+ return out_len
+
+ def gen_test_vector(self, fi, prefix):
+ code_name = "%s_%s" % (prefix, self.name)
+
+ fi.write("\t{\n")
+ fi.write("\t\t.name = \"%s\",\n" % code_name)
+ fi.write("\t\t.code = &%s,\n" % code_name)
+
+ fi.write("\t\t.in_len = %d,\n" % self.block_len)
+ fi.write("\t\t.out_len = %d,\n" % self.calc_out_len())
+
+ # Print pre computed vectors if preset
+ if self.vec_in is not None and self.vec_out is not None:
+ fi.write("\t\t.has_vec = 1,\n")
+ fi.write("\t\t.vec_in = {\n")
+ print_formatted(self.vec_in, "0x%02x, ", 8, fi, indent = "\t\t\t")
+ fi.write("\t\t},\n")
+ fi.write("\t\t.vec_out = {\n")
+ print_formatted(self.vec_out, "0x%02x, ", 8, fi, indent = "\t\t\t")
+ fi.write("\t\t},\n")
+ else:
+ fi.write("\t\t.has_vec = 0,\n")
+ fi.write("\t\t.vec_in = { },\n")
+ fi.write("\t\t.vec_out = { },\n")
+
+ fi.write("\t},\n")
+
poly = lambda *args: sum([(1 << x) for x in args])
def combine(src, sel, nb):
@@ -233,15 +274,15 @@
fn_xor = lambda x, y: x ^ y
return reduce(fn_xor, [(x >> n) & 1 for n in range(nb)])
-def print_formatted(items, format, count, fi):
+def print_formatted(items, format, count, fi, indent = "\t"):
counter = 0
# Print initial indent
- fi.write("\t")
+ fi.write(indent)
for item in items:
if counter > 0 and counter % count == 0:
- fi.write("\n\t")
+ fi.write("\n" + indent)
fi.write(format % item)
counter += 1
@@ -281,11 +322,37 @@
code.gen_tables(prefix, f, shared_tables = shared)
+def generate_vectors(codes, path, prefix, inc = None):
+ # Open a new file for writing
+ f = open(os.path.join(path, prefix + "_vectors.c"), 'w')
+ f.write(mod_license + "\n")
+ f.write("#include <osmocom/core/conv.h>\n")
+ if inc is not None:
+ for item in inc:
+ f.write("%s\n" % item)
+ f.write("\n")
+
+ sys.stderr.write("Generating test vectors...\n")
+
+ vec_count = len(codes.conv_codes)
+ f.write("int %s_vectors_count = %d;\n\n"
+ % (prefix, vec_count))
+ f.write("const struct conv_test_vector %s_vectors[%d] = {\n"
+ % (prefix, vec_count))
+
+ # Generate the vectors one by one
+ for code in codes.conv_codes:
+ sys.stderr.write("Generate '%s' test vector\n" % code.name)
+ code.gen_test_vector(f, prefix)
+
+ f.write("};\n")
+
def print_help(error = None):
print("Usage: python %s action code_type [path]" % sys.argv[0])
print("\nAvailable actions:")
- print(" gen_codes - generate convolutional code definitions")
+ print(" gen_codes - generate convolutional code definitions")
+ print(" gen_vectors - generate test vectors")
print("\nAvailable code types:")
print(" gsm - GSM specific codes")
@@ -302,9 +369,16 @@
code_type = sys.argv[2]
path = sys.argv[3] if len(sys.argv) > 3 else os.getcwd()
+ # Code family specific includes
+ inc = None
+
# Determine convolutional code type
if code_type == "gsm":
codes = conv_codes_gsm
+ inc = [
+ "#include <osmocom/gsm/gsm0503.h>",
+ "#include \"conv_test.h\""
+ ]
else:
print_help("Error: Unknown code type!")
sys.exit(1)
@@ -312,6 +386,8 @@
# What to generate?
if action == "gen_codes":
generate_codes(codes, path, "gsm0503")
+ elif action == "gen_vectors":
+ generate_vectors(codes, path, "gsm0503", inc)
else:
print_help("Error: Unknown action!")
sys.exit(1)
--
To view, visit https://gerrit.osmocom.org/1585
To unsubscribe, visit https://gerrit.osmocom.org/settings
Gerrit-MessageType: newchange
Gerrit-Change-Id: Ie10c47ee952f253b1ba77ecf6e79f2c033545bc1
Gerrit-PatchSet: 1
Gerrit-Project: libosmocore
Gerrit-Branch: master
Gerrit-Owner: Vadim Yanitskiy <axilirator at gmail.com>