trx_toolkit/data_msg.py: add type hints to static methods

Change-Id: I06fd8bc7418700de40467fd63a08da2bc2abcea2
This commit is contained in:
Vadim Yanitskiy 2021-03-02 01:39:25 +01:00
parent 45cc12d665
commit b00601d7ee
1 changed files with 5 additions and 4 deletions

View File

@ -25,6 +25,7 @@ import random
import struct
import abc
from typing import List
from enum import Enum
from gsm_shared import *
@ -187,22 +188,22 @@ class DATAMSG(abc.ABC):
return result
@staticmethod
def usbit2sbit(bits):
def usbit2sbit(bits: List[int]) -> List[int]:
''' Convert unsigned soft-bits {254..0} to soft-bits {-127..127}. '''
return [-127 if (b == 0xff) else 127 - b for b in bits]
@staticmethod
def sbit2usbit(bits):
def sbit2usbit(bits: List[int]) -> List[int]:
''' Convert soft-bits {-127..127} to unsigned soft-bits {254..0}. '''
return [127 - b for b in bits]
@staticmethod
def sbit2ubit(bits):
def sbit2ubit(bits: List[int]) -> List[int]:
''' Convert soft-bits {-127..127} to bits {1..0}. '''
return [int(b < 0) for b in bits]
@staticmethod
def ubit2sbit(bits):
def ubit2sbit(bits: List[int]) -> List[int]:
''' Convert bits {1..0} to soft-bits {-127..127}. '''
return [-127 if b else 127 for b in bits]