141 lines
5.5 KiB
Python
Executable File
141 lines
5.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
# This work is licensed under the Creative Attribution-NonCommercial-ShareAlike
|
|
# 3.0 Unported License.To view a copy of this license, visit
|
|
# http://creativecommons.org/licenses/by-nc-sa/3.0/ or send a letter to
|
|
# Creative Commons, 444 Castro Street, Suite 900, Mountain View,
|
|
# California, 94041, USA.
|
|
|
|
from __future__ import print_function
|
|
from bitarray import bitarray
|
|
import hamming
|
|
|
|
# Does anybody read this stuff? There's a PEP somewhere that says I should do this.
|
|
__author__ = 'Cortney T. Buffington, N0MJS'
|
|
__copyright__ = 'Copyright (c) 2016 Cortney T. Buffington, N0MJS and the K0USY Group'
|
|
__credits__ = 'Jonathan Naylor, G4KLX; Ian Wraith'
|
|
__license__ = 'Creative Commons Attribution-NonCommercial-ShareAlike 3.0 Unported'
|
|
__maintainer__ = 'Cort Buffington, N0MJS'
|
|
__email__ = 'n0mjs@me.com'
|
|
|
|
#------------------------------------------------------------------------------
|
|
# BPTC(196,96) Decoding routings
|
|
#------------------------------------------------------------------------------
|
|
|
|
INDEX_181 = (
|
|
0, 181, 166, 151, 136, 121, 106, 91, 76, 61, 46, 31, 16, 1, 182, 167, 152, 137,
|
|
122, 107, 92, 77, 62, 47, 32, 17, 2, 183, 168, 153, 138, 123, 108, 93, 78, 63,
|
|
48, 33, 18, 3, 184, 169, 154, 139, 124, 109, 94, 79, 64, 49, 34, 19, 4, 185, 170,
|
|
155, 140, 125, 110, 95, 80, 65, 50, 35, 20, 5, 186, 171, 156, 141, 126, 111, 96,
|
|
81, 66, 51, 36, 21, 6, 187, 172, 157, 142, 127, 112, 97, 82, 67, 52, 37, 22, 7,
|
|
188, 173, 158, 143, 128, 113, 98, 83, 68, 53, 38, 23, 8, 189, 174, 159, 144, 129,
|
|
114, 99, 84, 69, 54, 39, 24, 9, 190, 175, 160, 145, 130, 115, 100, 85, 70, 55, 40,
|
|
25, 10, 191, 176, 161, 146, 131, 116, 101, 86, 71, 56, 41, 26, 11, 192, 177, 162,
|
|
147, 132, 117, 102, 87, 72, 57, 42, 27, 12, 193, 178, 163, 148, 133, 118, 103, 88,
|
|
73, 58, 43, 28, 13, 194, 179, 164, 149, 134, 119, 104, 89, 74, 59, 44, 29, 14,
|
|
195, 180, 165, 150, 135, 120, 105, 90, 75, 60, 45, 30, 15)
|
|
|
|
# Converts a DMR frame using 98-68-98 (info-sync/EMB-info) into 196 bit array
|
|
def to_binary_19696(_data):
|
|
_bits = bitarray(endian='big')
|
|
_bits.frombytes(_data)
|
|
return _bits[:98] + _bits[-98:]
|
|
|
|
# Applies interleave indecies de-interleave 196 bit array
|
|
def deinterleave_19696(_data):
|
|
deint = bitarray(196, endian='big')
|
|
for index in xrange(196):
|
|
deint[index] = _data[INDEX_181[index]] # the real math is slower: deint[index] = _data[(index * 181) % 196]
|
|
return deint
|
|
|
|
# Applies BTPC error detection/correction routines (INCOMPLETE)
|
|
def error_check_19696(_data):
|
|
count = 0
|
|
column = bitarray(13, endian='big')
|
|
|
|
while True:
|
|
errors = False
|
|
for col in xrange(15):
|
|
pos = col + 1
|
|
for index in xrange(13):
|
|
column[index] = _data[pos]
|
|
pos += 15
|
|
|
|
result_1393 = hamming.dec_1393(column)
|
|
if result_1393[1]:
|
|
pos = col + 1
|
|
for index in xrange(13):
|
|
_data[pos] = result_1393[0][index]
|
|
pos += 15
|
|
errors = True
|
|
print('fixing error in column {}'.format(col))
|
|
|
|
for index in xrange(9):
|
|
pos = (index*15) + 1
|
|
result_15113 = hamming.dec_15113(_data[pos:(pos+15)])
|
|
if result_15113[1]:
|
|
errors = True
|
|
_data[pos:(pos+15)] = result_15113[0]
|
|
print('fixing error in row {}'.format(index))
|
|
|
|
count += 1
|
|
print('pass count is {}'.format(count))
|
|
if not errors or count > 4: break
|
|
return (errors)
|
|
|
|
# Returns useable LC data - 9 bytes info + 3 bytes RS(12,9) ECC
|
|
def to_bytes_19696(_data):
|
|
databits = _data[4:12]+_data[16:27]+_data[31:42]+_data[46:57]+_data[61:72]+_data[76:87]+_data[91:102]+_data[106:117]+_data[121:132]
|
|
return databits.tobytes()
|
|
|
|
|
|
#------------------------------------------------------------------------------
|
|
# BPTC(196,96) Decoding routings
|
|
#------------------------------------------------------------------------------
|
|
# not yet implemented
|
|
|
|
|
|
#------------------------------------------------------------------------------
|
|
# Used to execute the module directly to run built-in tests
|
|
#------------------------------------------------------------------------------
|
|
|
|
if __name__ == '__main__':
|
|
|
|
from binascii import b2a_hex as h
|
|
from time import time
|
|
|
|
# Validation Example
|
|
# Good Data
|
|
data = '\x2b\x60\x04\x10\x1f\x84\x2d\xd0\x0d\xf0\x7d\x41\x04\x6d\xff\x57\xd7\x5d\xf5\xde\x30\x15\x2e\x20\x70\xb2\x0f\x80\x3f\x88\xc6\x95\xe2'
|
|
# Bad Data
|
|
data = '\xff\xff\x14\x10\x1f\x84\x2d\xd0\x0d\xf0\x7d\x41\x04\x6d\xff\x57\xd7\x5d\xf5\xde\x30\x15\x2e\x20\x70\xb2\x0f\x80\x3f\x88\xc6\x95\xe2'
|
|
|
|
t0 = time()
|
|
bin_data = to_binary_19696(data)
|
|
deint_data = deinterleave_19696(bin_data)
|
|
err_corrected = error_check_19696(deint_data) # This corrects deint_data in place -- it does not return a new array!!!
|
|
if err_corrected:
|
|
print('WARNING DATA COULD NOT BE CORRECTED')
|
|
ext_data = to_bytes_19696(deint_data)
|
|
t1 = time()
|
|
print('TIME: ', t1-t0, '\n')
|
|
|
|
print('original 33 byte data block:')
|
|
print(h(data))
|
|
print(len(data), 'bytes')
|
|
print()
|
|
|
|
print('binary data (discarding sync)')
|
|
print(len(bin_data), 'bits')
|
|
print(bin_data)
|
|
print()
|
|
|
|
print('deinterleaved binary data')
|
|
print(len(deint_data), 'bits')
|
|
print(deint_data)
|
|
print(h(deint_data.tobytes()))
|
|
print()
|
|
|
|
print('decoded hex data should be: 001020000c302f9be5dad45a')
|
|
print('decoded data is: ', h(ext_data))
|
|
print(len(ext_data), 'bytes') |