trx_toolkit: define TRXDv2 PDUs using declarative codec
Change-Id: If356d285006c0b9b57879d0499b8144eca820cab Related: OS#4006, SYS#4895
This commit is contained in:
parent
7c3a566b52
commit
d4900f93ee
|
@ -30,12 +30,20 @@ import codec
|
|||
class Header(codec.BitFieldSet):
|
||||
''' Header constructor for TRXD PDUs. '''
|
||||
|
||||
def __init__(self, ver: int):
|
||||
codec.BitFieldSet.__init__(self, len=1, set=(
|
||||
codec.BitField('ver', bl=4, val=ver),
|
||||
codec.BitField.Pad(bl=1), # RFU
|
||||
def __init__(self, ver: int, batched: bool = False):
|
||||
f = [ # Dynamically generated field list
|
||||
codec.BitField('ver', bl=4, val=ver) if not batched
|
||||
else codec.BitField.Spare(bl=4), # RFU
|
||||
codec.BitField.Spare(bl=1),
|
||||
codec.BitField('tn', bl=3),
|
||||
))
|
||||
]
|
||||
|
||||
if ver >= 2: # TRXDv2 and higher
|
||||
f.append(codec.BitField('batch', bl=1))
|
||||
f.append(codec.BitField.Spare(bl=1))
|
||||
f.append(codec.BitField('trxn', bl=6))
|
||||
|
||||
codec.BitFieldSet.__init__(self, set=tuple(f))
|
||||
|
||||
class MTS(codec.BitFieldSet):
|
||||
''' Modulation and Training Sequence. '''
|
||||
|
@ -53,8 +61,10 @@ class MTS(codec.BitFieldSet):
|
|||
|
||||
GMSK_BURST_LEN = 148
|
||||
|
||||
if (mod >> 2) in (0b00, 0b11): # GMSK or AQPSK
|
||||
if (mod >> 2) == 0b00: # GMSK
|
||||
return 1 * GMSK_BURST_LEN
|
||||
elif (mod >> 2) == 0b11: # AQPSK
|
||||
return 2 * GMSK_BURST_LEN
|
||||
elif (mod >> 1) == 0b010: # 8-PSK
|
||||
return 3 * GMSK_BURST_LEN
|
||||
elif (mod >> 1) == 0b100: # 16QAM
|
||||
|
@ -65,6 +75,35 @@ class MTS(codec.BitFieldSet):
|
|||
return 1 * GMSK_BURST_LEN
|
||||
raise ValueError('Unknown modulation type')
|
||||
|
||||
class Power(codec.Codec):
|
||||
''' SCPIR and Tx power reduction (TRXDv2 and higher).
|
||||
|
||||
+-----------------+---------------------------------+
|
||||
| 7 6 5 4 3 2 1 0 | Description |
|
||||
+-----------------+---------------------------------+
|
||||
| . . . . x x x x | Power REDuction (in 2 dB steps) |
|
||||
+-----------------+---------------------------------+
|
||||
| . x x x . . . . | SCPIR value (in 2 dB steps) |
|
||||
+-----------------+---------------------------------+
|
||||
| x . . . . . . . | SCPIR sign indicator |
|
||||
+-----------------+---------------------------------+
|
||||
|
||||
'''
|
||||
|
||||
def from_bytes(self, vals: dict, data: bytes) -> int:
|
||||
blob = ord(data) # Convert a byte to an int
|
||||
vals['red'] = (blob & 0b1111) * 2
|
||||
vals['scpir'] = ((blob >> 4) & 0b111) * 2
|
||||
if blob & (1 << 7): # negative sign
|
||||
vals['scpir'] *= -1
|
||||
return 1
|
||||
|
||||
def to_bytes(self, vals: dict) -> bytes:
|
||||
blob = (vals['red'] & 0b1111) \
|
||||
| (abs(vals['scpir']) << 4) // 2 \
|
||||
| (0x80 if (vals['scpir'] < 0) else 0x00)
|
||||
return bytes((blob,))
|
||||
|
||||
class BurstBits(codec.Buf):
|
||||
''' Soft-/hard-bits with variable length. '''
|
||||
|
||||
|
@ -118,3 +157,46 @@ class PDUv1Tx(PDUv0Tx):
|
|||
def __init__(self, *args, **kw):
|
||||
PDUv0Tx.__init__(self, *args, **kw)
|
||||
self.STRUCT[0]._fields[0].val = 1
|
||||
|
||||
|
||||
class PDUv2Rx(codec.Envelope):
|
||||
class BPDU(codec.Envelope):
|
||||
''' Batched PDU part. '''
|
||||
STRUCT = (
|
||||
Header(ver=2, batched=True),
|
||||
MTS(),
|
||||
codec.Uint('rssi', mult=-1),
|
||||
codec.Int16BE('toa256'),
|
||||
codec.Int16BE('cir'),
|
||||
BurstBits('soft-bits'),
|
||||
)
|
||||
|
||||
STRUCT = (
|
||||
Header(ver=2),
|
||||
MTS(),
|
||||
codec.Uint('rssi', mult=-1),
|
||||
codec.Int16BE('toa256'),
|
||||
codec.Int16BE('cir'),
|
||||
codec.Uint32BE('fn'),
|
||||
BurstBits('soft-bits'),
|
||||
codec.Sequence(item=BPDU()).f('bpdu'),
|
||||
)
|
||||
|
||||
class PDUv2Tx(codec.Envelope):
|
||||
class BPDU(codec.Envelope):
|
||||
''' Batched PDU part. '''
|
||||
STRUCT = (
|
||||
Header(ver=2, batched=True),
|
||||
MTS(),
|
||||
Power(),
|
||||
BurstBits('hard-bits'),
|
||||
)
|
||||
|
||||
STRUCT = (
|
||||
Header(ver=2),
|
||||
MTS(),
|
||||
Power(),
|
||||
codec.Uint32BE('fn'),
|
||||
BurstBits('hard-bits'),
|
||||
codec.Sequence(item=BPDU()).f('bpdu'),
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue