Home > Software engineering >  How can I convert 8 bit array to 16 bit array in Python?
How can I convert 8 bit array to 16 bit array in Python?

Time:12-16

I'm transferring 16 bit numbers from a STM32 (from an ADC) over SPI to raspberry pi 4. On the pi side, I have a script that runs in a loop, waits for a GPIO pin to go high to act as a "detect" for my system to then enable the raspberry pi to initiate the SPI transfer. Unfortunately the raspberry pi hardware only supports 8 bit SPI and so the numbers I'm getting back are split across 2 x 8 bits. The data I'm transferring:

uint16_t spi_test_16[64] = {1023, 19, 38, 47, 51, 52, 53, 59, 76, 91, 99, 119, 172, 174,
    179, 205, 215, 218, 225, 235, 242, 264, 284, 308, 316, 334, 404, 431, 434,
    442, 442, 457, 468, 479, 490, 496, 508, 509, 512, 523, 524, 524, 525, 589, 598,
    642, 674, 711, 716, 749, 761, 767, 771, 788, 790, 818, 854, 928, 935, 947, 964,
    976, 987, 996}

and the data I'm receiving:

(255, 3, 19, 0, 38, 0, 47, 0, 51, 0, 52, 0, 53, 0, 59, 0, 76, 0, 91, 0, 99, 0, 119, 0, 172, 0  174, 0, 179, 0, 205, 0, 215, 0, 218, 0, 225, 0, 235, 0, 242, 0, 8, 1, 28, 1, 52, 1, 60, 1, 78, 1, 148, 1, 175, 1, 178, 1, 186, 1, 186, 1, 201, 1, 212, 1, 223, 1, 234, 1, 240, 1, 252, 1, 253, 1, 0, 2, 11, 2, 12, 2, 12, 2, 13, 2, 77, 2, 86, 2, 130, 2, 162, 2, 199, 2, 204, 2, 237, 2, 249, 2, 255, 2, 3, 3, 20, 3, 22, 3, 50, 3, 86, 3, 160, 3, 167, 3, 179, 3, 196, 3, 208, 3, 219, 3, 228, 3)

As you can see, it's getting the right numbers but they're being formatted incorrectly because it's in an 2 x 8 bit format. I want help in order to get the values back in to the original format using Python. My code:

    #a simple test for spi
    
    import spidev
    import time
    import RPi.GPIO as GPIO
    import numpy
    
    GPIO.setmode(GPIO.BOARD)
    GPIO.setup(26, GPIO.IN)
    GPIO.setup(40, GPIO.OUT)
    #using spi0
    bus = 0
    
    #device is chip select.
    device = 0
    
    #enable
    spi = spidev.SpiDev()
    
    #open a connection to a specific bus and device (CS pin)
    spi.open(bus, device)
    
    #set spi speed and mode
    spi.max_speed_hz = 100000
    spi.mode = 0
    spi.bits_per_word = 8
    #msg = ["Hello! "]
    msg = [72, 101, 108, 108, 111, 33, 32]
    buffer_size = 128
    spi_buffer = [0]*buffer_size
    msgRx = 0;

    flag = 0

print ("python spi test program with STM32")
#GPIO.output(7,0)
GPIO.output(40,1)
time.sleep (1)
while True :
    
    if GPIO.input(26) == 1:
        #print ("Message receiving")
        time.sleep(0.1)
        if GPIO.input(26) == 1:
            GPIO.output(40,0)
            #msgRx = spi.readbytes(160)
            msgRx = spi.xfer3(spi_buffer)
            GPIO.output(40,1)
            print (msgRx)
            ##listofzeroes = [0]*50000
            print("msgRx reset")
            msgRx = numpy.asarray(msgRx, dtype=numpy.uint16)
            for index, value in enumerate(msgRx):
                msgRx[index] >> 8

            print (msgRx)
            time.sleep(1)
            #flag = 1
    #if flag == 1:        
        #print("Message Complete")
        #flag = 0

I've attempted a couple of things, but currently in the main While loop I'm converting the received list into a 16bit unsigned integer array and then bit shifting to the left. But then how do I add the second cell of the array to the first etc?

TDLR; how can I convert a list of 2 x 8 bit integers into a list of 1 x 16 bit integers?

CodePudding user response:

Use numpy:

In [9]: data = (255, 3, 19, 0, 38, 0, 47, 0, 51, 0, 52, 0, 53, 0, 59, 0, 76, 0, 91, 0, 99, 0, 119, 0, 172, 0,  174, 0, 179, 0, 205, 0, 215, 0, 218, 0, 225, 0, 235, 0, 242, 0, 8, 1, 28, 1, 52, 1, 60, 1, 78, 1, 148, 1, 175, 1, 178, 1,
   ...: 186, 1, 186, 1, 201, 1, 212, 1, 223, 1, 234, 1, 240, 1, 252, 1, 253, 1, 0, 2, 11, 2, 12, 2, 12, 2, 13, 2, 77, 2, 86, 2, 130, 2, 162, 2, 199, 2, 204, 2, 237, 2, 249, 2, 255, 2, 3, 3, 20, 3, 22, 3, 50, 3, 86, 3, 160, 3, 167, 3,
   ...:  179, 3, 196, 3, 208, 3, 219, 3, 228, 3)
   ...:

In [10]: arr = np.array(data, np.uint8)

It looks like in your tuple, you have unsigned 8-bit integers. So then just create a view of that as unsigned 16-bit integers:

In [11]: arr = arr.view(np.uint16)

In [12]: arr
Out[12]:
array([1023,   19,   38,   47,   51,   52,   53,   59,   76,   91,   99,
        119,  172,  174,  179,  205,  215,  218,  225,  235,  242,  264,
        284,  308,  316,  334,  404,  431,  434,  442,  442,  457,  468,
        479,  490,  496,  508,  509,  512,  523,  524,  524,  525,  589,
        598,  642,  674,  711,  716,  749,  761,  767,  771,  788,  790,
        818,  854,  928,  935,  947,  964,  976,  987,  996], dtype=uint16)

As noted in the comment, to make the above portable, you can specify the endianess using the following notation:

arr = arr.view('<H'). 

The < means 'little-endian', and H is the character code for unsigned 16 bit integers. See the docs for more on specifying dtypes.

Note, I don't know the library you are using, but you have a commented out line:

msgRx = spi.readbytes(160)

Probably, you just want to use that, but read 128 bytes (for the 64 unsigned 16-bit integers) then use np.frombuffer:

msgRx = spi.readbytes(128)
msgRx = numpy.frombuffer(msgRx, dtype=numpy.uint16)

CodePudding user response:

This is working:

def to_bin(int_num):
    return '{0:08b}'.format(int_num)


def to_int(bina):
    return int(bina, 2)


def transform(int_num_1, int_num_2):
    print(f"{int_num_1 = }")
    print(f"{int_num_2 = }")
    bina_1 = to_bin(int_num_1)
    bina_2 = to_bin(int_num_2)
    print(f"{bina_1 = }")
    print(f"{bina_2 = }")
    trame = bina_2   bina_1
    print(f"{trame = }")
    result = to_int(trame)
    print(f"{result = }")
    return result


input_list = [255, 3, 19, 0, 38, 0, 47, 0, 51, 0, 52, 0, 53, 0, 59, 0, 76, 0, 91, 0, 99, 0, 119, 0, 172, 0,  174, 0, 179, 0, 205, 0, 215, 0, 218, 0, 225, 0, 235, 0, 242, 0, 8, 1, 28, 1, 52, 1, 60, 1, 78, 1, 148, 1, 175, 1, 178, 1, 186, 1, 186, 1, 201, 1, 212, 1, 223, 1, 234, 1, 240, 1, 252, 1, 253, 1, 0, 2, 11, 2, 12, 2, 12, 2, 13, 2, 77, 2, 86, 2, 130, 2, 162, 2, 199, 2, 204, 2, 237, 2, 249, 2, 255, 2, 3, 3, 20, 3, 22, 3, 50, 3, 86, 3, 160, 3, 167, 3, 179, 3, 196, 3, 208, 3, 219, 3, 228, 3]
for i in range(len(input_list)//2):
    transform(input_list[i*2], input_list[i*2 1])

The trick is to convert to binaries, concatenate them and reconvert into integers.

Results for first 3:

int_num_1 = 255
int_num_2 = 3
bina_1 = '11111111'
bina_2 = '00000011'
trame = '0000001111111111'
result = 1023

int_num_1 = 19
int_num_2 = 0
bina_1 = '00010011'
bina_2 = '00000000'
trame = '0000000000010011'
result = 19

int_num_1 = 38
int_num_2 = 0
bina_1 = '00100110'
bina_2 = '00000000'
trame = '0000000000100110'
result = 38

EDIT: Use the answer of @juanpa.arrivillaga (much faster). Take this answer to understand how it's work!

  • Related