gr-gsm/python/qa_tch_h_decoder.py

191 lines
10 KiB
Python
Executable File

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @file
# @author (C) 2018 by Vasil Velichkov <vvvelichkov@gmail.com>
# @section LICENSE
#
# Gr-gsm is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3, or (at your option)
# any later version.
#
# Gr-gsm is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with gr-gsm; see the file COPYING. If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street,
# Boston, MA 02110-1301, USA.
#
#
import numpy as np
from gnuradio import gr, gr_unittest, blocks
import grgsm_swig as grgsm
import pmt
class qa_tch_h_decoder (gr_unittest.TestCase):
def setUp (self):
self.tb = gr.top_block ()
self.b = [
"0001010001011010010000111100010101011011110000001011100101110010011101011000001001110100100010011111111000001001011000001100100010001011011000010000",
"0000000110100000010101000111011101101100010000101101000011000010011101011000001001110100001111000111000010100010111111011000001010111101110001000000",
"0000001101101000100100101101000011110011010110111100110000100010011101011000001001110100001110011100011000010000000111110100111110001101101100100000",
"0000011111010101110111001101111001001111010010000011111111010010011101011000001001110100010110000010000000000011100101111111010101111101111011101000",
"0001001110111000101001110001010011001101101111000010111100010010011101011000001001110100111000111011000111111100001011011011111001011100110111001000",
"0000110110000101000011001001011110110110001110001000010001000010011101011000001001110100000010100001010001011000000110001110011110001000111100110000",
"0000001101100010101011110100000010110100111110010010001100010010011101011000001001110100100000110001101011011101011110000110100010111100101011011000",
"0001111110100000011111010011100100001101011101000011011000000010011101011000001001110100100111110010000011100011000011001110001111110000001010011000",
"0001010100101010111010111001000100010100000010101011010011011010011101011000001001110100010100011100101110101110000000000111101011111111100011000000",
"0000000011011011101011000101000001001110001011101000100001011010011101011000001001110100000100011110101010100011010101010100101011101010000100000000",
"0001010110101001110001111001110100011010000110100000111001110010011101011000001001110100100100010101000100001100100000111111000001010111100011000000",
"0001100000100001100001100101101001101110001000101011100001010010011101011000001001110100110010110101000100100011011101111100000011100010000100000000",
"0001010110100000010010111001000100011110101010100001010001110010011101011000001001110100110100010100000110001100000000100111000011011111100011000000",
"0000101001110011000001000101001001101110001001100000100001110010011101011000001001110100100110110100001000100011111101111110101011100000100100000000",
#FACCH framces 14 - 21
"0001011000000010000000001000010111001000101111010000000010110010011101011000001001110101100000100000010010100000101000011000100011100010100000110000",
"0000010101111010111010000001100001110101111100001010101001100010011101011000001001110101111101111101101110000010110011111111110101010000001000110000",
"0000000000111110101010100001000000001000101110100010100100011010011101011000001001110101001010101011101010100000010100000000101111101000000000010000",
"0000001010010101011111011011101010101000000001011111111110101010011101011000001001110101101000010000010101111011111010100001010001011101111010101000",
"0001010100101010111010111001000100010100000010101011010011011010011101011000001001110100010100011100101110101110000000000111101011111111100011000000",
"0000000011011011101011000101000001001110001011101000100001011010011101011000001001110100000100011110101010100011010101010100101011101010000100000000",
"0001010110101001110001111001110100011010000110100000111001110010011101011000001001110100100100010101000100001100100000111111000001010111100011000000",
"0001100000100001100001100101101001101110001000101011100001010010011101011000001001110100110010110101000100100011011101111100000011100010000100000000",
]
def tearDown (self):
self.tb = None
def tchh_multirate (self, frames, timeslots, bursts, multirate, subchan):
"""
Common TCH/H MultiRate test code
"""
src = grgsm.burst_source(frames, timeslots, bursts)
decoder = grgsm.tch_h_decoder(subchan, multirate, False);
dst = blocks.message_debug()
self.tb.msg_connect(src, "out", decoder, "bursts")
self.tb.msg_connect(decoder, "voice", dst, "store")
self.tb.run ()
result = []
for i in range(0, dst.num_messages()):
pdu = dst.get_message(i)
self.assertEqual(True, pmt.is_pair(pdu))
data = pmt.cdr(pdu)
self.assertEqual(True, pmt.is_blob(data))
result.append(pmt.to_python(data).tolist())
return result
def test_amr7_40 (self):
"""
TCH/H MultiRate AMR 7.40
"""
b = self.b
self.assertListEqual(self.tchh_multirate(
multirate = "28111a40",
subchan = 0,
frames = [259215, 259217, 259220, 259222, 259220, 259222, 259224, 259226],
timeslots = [ 6, 6, 6, 6, 6, 6, 6, 6],
bursts = [ b[8], b[9], b[10], b[11], b[10], b[11], b[12], b[13]]),
[
[0x23,0x21,0x41,0x4d,0x52,0x0a],
[0x20,0xff,0x3c,0x67,0xe0,0x00,0x1f,0x3d,0x01,0xf0,0xfc,0x3f,0x77,0x18,0x61,0x86,0x00,0x00,0x00,0x00]
])
def test_4_75 (self):
"""
TCH/H MultiRate AMR 4.75
"""
b = self.b
self.assertListEqual(self.tchh_multirate(
multirate = "28111a40",
subchan = 0,
frames = [259666, 259668, 259670, 259672, 259670, 259672, 259675, 259677],
timeslots = [ 6, 6, 6, 6, 6, 6, 6, 6],
bursts = [ b[2], b[3], b[4], b[5], b[4], b[5], b[6], b[7]]),
[
[0x23,0x21,0x41,0x4d,0x52,0x0a],
[0x00,0x67,0x19,0x24,0xd5,0x1b,0xd1,0x29,0x3f,0xa1,0x50,0x5f,0x3e]
])
def test_amr7_40_and_4_75 (self):
"""
TCH/H MultiRate AMR 7.40 and 4.75
Two 7.40 followed by two 4.75 frames
"""
b = self.b
self.assertListEqual(self.tchh_multirate(
multirate= "28111a40",
subchan = 0,
frames = [259657, 259659, 259662, 259664,
259662, 259664, 259666, 259668,
259666, 259668, 259670, 259672,
259670, 259672, 259675, 259677,
259675, 259677, 259679, 259681],
timeslots= [ 6, 6, 6, 6,
6, 6, 6, 6,
6, 6, 6, 6,
6, 6, 6, 6,
6, 6, 6, 6],
bursts = [ b[0], b[1], b[2], b[3],
b[0], b[1], b[2], b[3],
b[2], b[3], b[4], b[5],
b[2], b[3], b[4], b[5],
b[4], b[5], b[6], b[7]]),
[
[0x23,0x21,0x41,0x4d,0x52,0x0a],
[0x20,0xe0,0x27,0x2a,0x00,0x21,0x30,0x38,0x75,0xf8,0x14,0x3c,0xec,0xde,0x0b,0x47,0x6f,0x9c,0xc6,0x70],
[0x20,0xe0,0x27,0x2a,0x00,0x21,0x30,0x38,0x75,0xf8,0x14,0x3c,0xec,0xde,0x0b,0x47,0x6f,0x9c,0xc6,0x70],
[0x00,0x67,0x19,0x24,0xd5,0x1b,0xd1,0x29,0x3f,0xa1,0x50,0x5f,0x3e],
[0x00,0x67,0x19,0x24,0xd5,0x1b,0xd1,0x29,0x3f,0xa1,0x50,0x5f,0x3e],
])
def facch_test (self, frames, timeslots, bursts):
'''
Common FACCH/TH test code
'''
src = grgsm.burst_source(frames, timeslots, bursts)
decoder = grgsm.tch_h_decoder(0, "28111a40", False);
dst = blocks.message_debug()
facch = grgsm.message_sink()
self.tb.msg_connect(src, "out", decoder, "bursts")
self.tb.msg_connect(decoder, "voice", dst, "store")
self.tb.msg_connect(decoder, "msgs", facch, "in")
self.tb.run ()
self.assertEqual(dst.num_messages(), 0)
return list(facch.get_messages())
def test_facch_th (self):
"""
FACCH/TH test
"""
b = self.b
self.assertEqual(self.facch_test(
frames= [259207, 259209, 259211, 259213, 259211, 259213, 259215, 259217],
timeslots = [ 6, 6, 6, 6, 6, 6, 6, 6],
bursts = [ b[14], b[15], b[16], b[17], b[16], b[17], b[18], b[19]]),
['02 04 01 06 00 00 00 00 00 03 f4 8b 06 00 00 00 03 60 09 03 0f 2b 2b 2b 2b 2b 2b 2b 2b 2b 2b 2b 2b 2b 2b 2b 2b 2b 2b'])
def test_facch_th_error (self):
"""
FACCH/TH Error handling test
"""
b = self.b
self.assertEqual(self.facch_test(
frames= [259207, 259209, 259211, 259213, 259211, 259213, 259215, 259217],
timeslots = [ 6, 6, 6, 6, 6, 6, 6, 6],
bursts = [ b[16], b[17], b[18], b[19], b[18], b[19], b[20], b[21]]),
[]) #Must return an empty array
if __name__ == '__main__':
gr_unittest.run(qa_tch_h_decoder)