Computando o CRC CRC8-ATM em Python

O polinômio CRC8-ATM de 8 bits é usado em muitas aplicações embarcadas, incluindo drivers de motor de passo controlados por UART da Trinamic como o TMC2209:

$$\text{CRC} = x^8 + x^2 + x^1 + x^0$$

O código a seguir fornece um exemplo de como computar este tipo de CRC em Python:

crc8_atm.py

def compute_crc8_atm(datagram, initial_value=0):
    crc = initial_value
    # Iterar os bytes nos dados
    for byte in datagram:
        # Iterar os bits no byte
        for _ in range(0, 8):
            if (crc >> 7) ^ (byte & 0x01):
                crc = ((crc << 1) ^ 0x07) & 0xFF
            else:
                crc = (crc << 1) & 0xFF
            # Deslocar para o próximo bit
            byte = byte >> 1
    return crc

Este código foi verificado em campo para o TMC2209.