#!/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)
    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):
    errors = False
    count = 0
    column = bitarray(13)
    
    while True:
        for col in xrange(15):
            pos = col + 1
            for index in xrange(13):
                column[index] = _data[pos]
                pos += 15
            
            if hamming.dec_1393(col)
        if not errors or count < 5: break
    return _data
    
# 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
    data = '\x44\x4d\x52\x44\x00\x2f\x9b\xe5\x00\x0c\x30\x00\x04\xc2\xc4\xa1\xa1\x99\x48\x6e\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\x00\x00'
    data = data[20:53]
    
    t0 = time()
    bin_data = to_binary_19696(data)
    deint_data = deinterleave_19696(bin_data)
    err_corrected = error_check_19696(deint_data)
    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')
    print(len(ext_data), 'bytes')
    print(h(ext_data))