Skip to main content

Redshift Labs Pty Ltd RSL Communication Python Driver

Project description

RSL Comm Python Driver

PyPI version test and package

TL;DR: "Swiss army knife" for using the UM7, UM8, and shearwater orientation sensors with Python 3 (Python 3.7+, including Python 3.10).

UM7 originally came with the "Serial Software Interface" for handling / communicating with the sensor, which is currently available for Windows only.

The python3 driver provided here is designed to keep you up and running on different platforms (Linux, Windows, Mac). If you have the UM7, UM8, or shearwater board and want to use it on Linux (e.g. Ubuntu, Debian, Raspbian, Yocto, Suse, etc.), Windows, or Mac, this repo provides driver code to send / receive individual packets and receive broadcasts, as well example code how to create a sensor communication object.

The driver has the following capabilities:

  • read / write single registers to/from UM7, UM8, shearwater over SPI;

  • read / write single registers to/from UM7, UM8, shearwater over UART;

  • receive broadcast packets from the UM7, UM8, shearwater sensor over UART;

  • register map and interpretation of the sensor registers for UM7, UM8, shearwater.

Checking out the repo with submodules:

This repo contains git submodules, in particularly rsl_xml_svd project is included as a submodule. In order to check out the project, including submodules use the following command:

git clone --recurse-submodules git@github.com:RedshiftLabsPtyLtd/rsl_comm_py.git

If the repo is already cloned without submodules, then in the existing repo:

git submodule update --recursive --init

If using GUI client like git kraken, the following post might be useful.

After syncing the sub-modules, the rsl_xml_svd folder should appear inside the repo, pointing exactly to the commit specified in this repository.

Repository structure

To get started, you need to know that communication with the UM7, UM8, and shearwater is organized as follows. For each sensor, there is a file with register description, and separate files with UART and SPI communication. The register description code interprets the received payload, and sensor files handle constructing, transmitting and receiving packets.

E.g. for the UM7 the UART communication is coded in um7_serial.py file, where the UM7Serial class is defined. Information about UM7 registers comes to um7_serial.py from the um7_registers.py file, where the accessing to the UM7 registers are stored. Since it is possible to access the UM7 register map over UART and SPI, the register data (e.g. addresses, fields, and their meaning) is stored in a separate file. In the examples folder we store the examples how to communicate with the sensor.

The UM7, UM8, and shearwater register descriptions are stored in SVD files, (um7.svd, um8.svd, shearwater.svd) and is parsed by the rsl_svd_parser.py. The parser extracts the information from the XML file and fills in python data classes.

Below we outline the repo structure:

HW Prerequisites

UM7, UM8, shearwater boards provide serial (UART) and SPI interfaces, hence the two main ways to access the sensor data are UART (serial) or SPI. The differences in short: UART provides broadcast functionality, i.e. when packets can transmitted by the board with a specified frequency (transmission frequencies are set up in configuration registers), and it is possible to issue sensor commands (i.e. accessing command registers). SPI access the sensor register on demand (i.e. no broadcast functionality), and only configuration and data registers can be accessed. Accessing commands is only supported over UART.

Serial connection (UART)

When using UM7, UM8, shearwater over serial, it is possible to connect to the target system (i.e. user's target):

  • to the serial port directly (e.g. when serial pins are wired out as on the Raspberry PI, NVIDIA Jetson Nano, or other board computers with GPIO and UART pins wired out);

  • to the USB port using the USB Expansion Board, which performs USB to serial conversion.

SPI connection

When using the UM7, UM8, shearwater over SPI, there are also a couple of possibilities:

  • to the SPI pins directly (e.g. Raspberry PI, NVIDIA Jetson Nano), i.e. the pins are wired to the SoC directly;

  • to the USB port using USB to SPI converter, e.g. USB-ISS.

The difference between the two, that in the first case SoC pins support the SPI directly (on the hardware level, which also mirrors in the OS level), then the OS is likely to have the SPI device driver built-in (e.g. Raspberry PI). In the second case, using external converter (e.g. USB-ISS), the device will be shown as a so-called cdc_acm (communication device class), and low-level SPI communication will be done by the converter, yet to the OS the converter will be shown as Abstract Control Model (ACM) USB Device.

Installation

pip install rsl-comm-py

Python dependencies

TL;DR: install (i) pyserial, jinja2.

If you want to use SPI: if using on Linux and use SPI bus directly, install spidev, otherwise if using USB-ISS install usb_iss python package.

Alternatively, one may use environment.yml to create conda environment with dependencies resolved.

If you are using python3.6, you need to install dataclasses (the module is included in standard library since python3.7, and for 3.6 needs to be installed separately).

Python driver 101

The python driver for Redshift Labs Pty Ltd orientation sensors works as follows: the *.svd (e.g. um7.svd) file describes the register map, and the fields of registers. From the *.svd file the [sensor]_registers.py (e.g. um7_registers.py or shearwater_registers.py) is generated. The files have generated methods for reading / writing single registers, and how to interpret the payload. In addition, the files (e.g. um7_registers.py or shearwater_registers.py) provide abstract methods connect, read_register, and write_register.

The [sensor]_serial.py files (e.g. shearwater_serial.py or um7_serial.py) implement required functionality to read / write registers via the UART interface. In addition to this the [sensor]_serial.py files also implement decoding of broadcast packets. Connection to the UART might be done via USB Expansion Board and for this the so-called device file might be used. One can also connect to the port directly using the port argument.

The [sensor]_spi.py files (e.g. shearwater_spi.py or um7_spi.py) implement required functionality to read / write registers via SPI interface. In particularly two ways of connecting to SPI are supported: either using USB-ISS adapter, and corresponding usb_iss library, or connecting to the SPI port directly, e.g. if Linux kernel have SPI functionality activated, and the SPI interface is wired directly to the SoC (e.g. as for Raspberry PI). This way the communication is done using the spidev python library.

OS Prerequisites

  1. When plugged in on a Linux system, the UM7, shearwater, UM8 sensor should appear as /dev/ttyUSB* device (this is only true when you are using USB Expansion Board which in turn uses USB-to-serial FTDI chip.

  2. If you are on a Linux machine, your Linux user must be a member of the dialout group (e.g. see this thread) to be able to read/write ttyUSB* devices without root privileges.

Device autodetect

To facilitate discovering RSL sensors (UM7, UM8, shearwater) among other USB-to-serial devices (and store the configuration when re-plugging and/or adding other devices), we provide rsl_autodetect method and the rsl_autodetect.py script (both are equivalent).

These methods create rsl_[SERIAL_NUM].json configuration file, which is then used to match the USB-to-serial converter to which the sensor is connected.

Using rsl_autodetect method

  1. Launch rsl_autodetect method:
from rsl_comm_py import rsl_autodetect
rsl_autodetect()

At this point, the rsl_[SERIAL_NUM].json files is created in the current directory.

Using rsl_autodetect.py script

  1. Obtain the rsl_autodetect.py script from rsl_comm_py package:
from rsl_comm_py import serve_autodetect_script
serve_autodetect_script()
  1. Launch the script and follow instructions:
./rsl_autodetect.py --help

At this point, the rsl_[SERIAL_NUM].json file is created in the current directory.

Important: The created rsl_[SERIAL_NUM].json configuration file should be used as a device file, when creating instance of UM7Communication class:

from rsl_comm_py import ShearWaterSerial
shearwater_sensor = ShearWaterSerial(device='rsl_[MY_SERIAL_NUM].json')

Quick start for UM7

Create UM7 serial communication object, UM7 connected to a port /dev/ttyUSB0, and read the firmware version:

from rsl_comm_py import UM7Serial
um7_serial = UM7Serial(port_name='/dev/ttyUSB0')
print(f"um7 firmware revision: {um7_serial.get_fw_revision}")

Reading all types of broadcast packets from UM7, 1000 packets in total:

from rsl_comm_py import UM7Serial
um7_serial = UM7Serial(port_name='/dev/ttyUSB0')
for packet in um7_serial.recv_broadcast(num_packets=1000):
    print(f"packet: {packet}")

Reading the raw sensor data broadcast packets from UM7, not limiting number of packets:

from rsl_comm_py import UM7Serial
um7_serial = UM7Serial(port_name='/dev/ttyUSB0')
for packet in um7_serial.recv_all_raw_broadcast():
    print(f"packet: {packet}")

Reading 100 processed sensor data broadcast packets from UM7:

from rsl_comm_py import UM7Serial
um7_serial = UM7Serial(port_name='/dev/ttyUSB0')
for packet in um7_serial.recv_all_proc_broadcast(num_packets=100):
    print(f"packet: {packet}")

Reading the Euler angles broadcast packets from UM7:

from rsl_comm_py import UM7Serial
um7_serial = UM7Serial(port_name='/dev/ttyUSB0')
for packet in um7_serial.recv_euler_broadcast():
    print(f"packet: {packet}")

Reading the CREG_COM_SETTINGS configuration register from UM7:

from rsl_comm_py import UM7Serial
um7_serial = UM7Serial(port_name='/dev/ttyUSB0')
print(f"received value: {um7_serial.creg_com_settings}")

Writing 40 (changing ALL_RAW_RATE to 40 Hz) to the CREG_COM_RATES2 register:

from rsl_comm_py import UM7Serial
um7_serial = UM7Serial(port_name='/dev/ttyUSB0')
um7_serial.creg_com_rates2 = 40

Quick start for shearwater

Read shearwater firmware build id:

from rsl_comm_py import ShearWaterSerial
shearwater = ShearWaterSerial(device='rsl_A500CNHH.json')
print(f"get_fw_build_id : {shearwater.get_fw_build_id}")

Create UM7 serial communication object, UM7 connected to a port /dev/ttyUSB0, and read the firmware version:

from rsl_comm_py import ShearWaterSerial
shearwater = ShearWaterSerial(port_name='/dev/ttyUSB0')
print(f"shearwater firmware revision: {shearwater.get_fw_build_version}")

Slow start

Take a look at the available examples.

In order to use the python driver functionality one first needs to create a communication object (e.g. UM7Serial, UM7SpiLinuxPort, or ShearWaterSerial, ShearWaterSpiUsbIss).

The construction of the UART communication object can be done either by specifying a port_name directly (e.g. /dev/ttyS0), or by specifying the device file that stores USB2Serial config (e.g. rsl_A500CNP8.json). The device argument shall only be used, when RSL board (UM7, UM8, shearwater) is connected via the USB Expansion Board, the device stores properties of the expansion board. Why? The issue to keep in mind that when the sensor is re-plugged, it might appear to the OS as different serial connection, i.e. when first plugged in the OS detects the device as /dev/ttyS0, then after re-plugging exactly the same sensor might appear by different port name, e.g. /dev/ttyS1, which means the user code needs to be changed. If using the device, we store properties of the USB Expansion Board in a JSON file (e.g. converter chip ID) and search connection which match with the properties, and in this case connecting to the sensor, even if is shown as a different serial connection by the OS.

So the communication object can either be created with specifying the port_name:

from rsl_comm_py import UM7Serial
um7_serial = UM7Serial(port_name='/dev/ttyUSB0')

Or with specifying the device:

from rsl_comm_py import UM7Serial
um7 = UM7Serial(device='um7_A500CNP8.json')

The two options are exclusive, i.e. specifying both port_name and device will not work.

Accessing to the individual registers is done via python properties. Properties for register names are all lower-case, split by _.

For example, reading the CREG_COM_RATES1:

from rsl_comm_py import UM7Serial
um7 = UM7Serial(device='rsl_A500CNP8.json')
um7.creg_com_rates1

Note, that reading single register is quite a slow operation, since one first constructs and sends a packet, and then parses output for response. Reading single registers is not recommended for reading sensor data, since it might happen, that data from different sensor registers come from different measurements. We strongly advice to use broadcast messages for reading sensor and fusion data.

UM7 Data Packets

UM7 sends different types of broadcast messages over the UART. These messages are e.g. HEALTH packet (i.e. the DREG_HEALTH register), raw sensor data (raw gyro, accelerometer, and magnetometer, and temperature), processed sensor data (processed gyro, accelerometer, and magnetometer), Euler angles, quaternions.

These data packets are stored in the repo as dataclasses in the file um7_broadcast_packets.py. Note, that only payload stored in the dataclasses, and all the checks (e.g. checksum, data length) is done during broadcast reception.

For example, the raw data broadcast message has the following payload:

from dataclasses import dataclass

@dataclass
class UM7AllRawPacket:
    gyro_raw_x: int
    gyro_raw_y: int
    gyro_raw_z: int
    gyro_raw_time: float
    accel_raw_x: int
    accel_raw_y: int
    accel_raw_z: int
    accel_raw_time: float
    mag_raw_x: int
    mag_raw_y: int
    mag_raw_z: int
    mag_raw_time: float
    temperature: float
    temperature_time: float

shearwater Data Packets

shearwater, similar to UM7 sends broadcast messages over the UART. These are e.g. raw sensor data (raw gyro, accelerometer, and magnetometer, and temperature), processed sensor data (processed gyro, accelerometer, and magnetometer), Euler angles, quaternions.

These data packets are defined in in the file shearwater_broadcast_packets.py.

For example, the data broadcast message with the processed sensor data has the following payload:

from dataclasses import dataclass


@dataclass
class ShearWaterAllProcPacket:
    gyro_1_proc_x: float
    gyro_1_proc_y: float
    gyro_1_proc_z: float
    gyro_1_proc_time: float
    gyro_2_proc_x: float
    gyro_2_proc_y: float
    gyro_2_proc_z: float
    gyro_2_proc_time: float
    accel_1_proc_x: float
    accel_1_proc_y: float
    accel_1_proc_z: float
    accel_1_proc_time: float
    mag_1_proc_x: float
    mag_1_proc_y: float
    mag_1_proc_z: float
    mag_1_norm: float
    mag_1_proc_time: float
    mag_2_proc_x: float
    mag_2_proc_y: float
    mag_2_proc_z: float
    mag_2_norm: float
    mag_2_proc_time: float

Acknowledgement

We are so grateful for the open source community for creating open source UM7 driver versions and sharing it with a world! We are inspired by your work, and at the same time want to improve: provide UART and SPI communication, in detail documentation and explanations to facilitate the start for new users.

The acknowledgments go to:

  • Daniel Kurek and his um7 repository, for implementing the first driver for interfacing with UM7;

  • Till Busch and his um7 fork of Daniel's Kurek repo, for extending on the Daniel's work and adding new functionality.

Maintainer

Dr. Konstantin Selyunin, for suggestions / questions / comments please contact: selyunin [dot] k [dot] v [at] gmail [dot] com

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

rsl_comm_py-0.1.11.tar.gz (146.8 kB view hashes)

Uploaded Source

Built Distribution

rsl_comm_py-0.1.11-py3-none-any.whl (175.8 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page