Skip to content

pyshimmer provides a Python API to work with the wearable sensor devices produced by Shimmer.

License

Notifications You must be signed in to change notification settings

seemoo-lab/pyshimmer

Repository files navigation

pyshimmer: Unofficial Python API for Shimmer Sensor devices

pyshimmer provides a Python API to work with the wearable sensor devices produced by Shimmer. The API is divided into three major components:

  • The Bluetooth API: An interface to communicate with the Shimmer LogAndStream firmware via Bluetooth
  • The Dock API: An interface to communicate with the Shimmer while they are placed in a dock
  • The Reader API: An interface to read the binary files produced by the Shimmer devices

Please note that the following README does not provide a general introduction to the Shimmer devices. For this, please consult the corresponding documentation page of the vendor and take a closer look at:

  • The Shimmer User Manual
  • The LogAndStream Manual
  • The SDLog Manual

All code in this repository was produced as part of my Master thesis. This means that the API is not complete. Especially the Bluetooth and Dock API do not feature all calls supported by the devices. However, the code provides a solid foundation to extend it where necessary. Please feel free to make contributions in case the code is missing required calls.

Please submit pull requests against the develop branch. When a new version is released, the current state of the develop branch is merged into master to produce the new version.

The targeted plattform for this library is Linux. It has not been tested on other operating systems. In order to use all aspects of the library, you need to install the package itself, set up the Bluetooth interface, and possibly configure udev rules to ensure that the device names are consistent.

In order to install the package itself, clone it and use pip to install it:

git clone https://github.com/seemoo-lab/pyshimmer.git
cd pyshimmer
pip install .

If you want to run the tests, instead install the package with test extras:

pip install .[test]

You can then run the tests from the repository root by simply issuing:

pytest

As of version v0.15.4 of the Shimmer3 firmware the Python API is fully compatible to the firmware. Older versions of the vanilla firmware exhibit several bugs and are incompatible. If you intend to use a firmware version older than 0.15.4, you will need to compile and run a custom patched version of the firmware. In the following table, the firmware versions and their compatibility are listed.

Firmware Type Version Compatibility Issues
LogAndStream v0.15.4 Compatible to v0.4.0 You will need to use pyshimmer v0.4 or newer
LogAndStream v0.11.0 Incompatible
SDLog v0.21.0 Compatible Untested
SDLog v0.19.0 Compatible  

It is recommended to use the newest LogAndStream firmware that is compatible to the API. If you want to use an older version with the pyshimmer library, you will need to compile and program a patched version of the firmware. We provide a forked repository which features the necessary fixes here. It also contains instructions on how to compile and program the firmware.

Starting with Firmware version v0.15.4, the race condition issue in the Bluetooth stack has been fixed. Additionally, the Shimmer3 now supports an additional command to disable the unsolicited status acknowledgment byte (see Issue 10). The pyshimmer Bluetooth API tries to automatically detect if the Shimmer3 runs a firmware newer or equal to v0.15.4 and automatically issues the command to disable the unsolicited status acknowledgment at startup. You can optionally disable this feature in the constructor. With this new command, the state machine in the Bluetooth API of pyshimmer is compatible to the vanilla firmware version.

When plugging a Shimmer dock into the host, Linux will detect two new serial interfaces and a block device representing the internal SD card of the Shimmer:

  • /dev/ttyUSB0 for the serial interface to the bootloader,
  • /dev/tty/USB1 for the serial interface to the device itself,
  • /dev/sdX for the block device.

When working with multiple docks and devices, keeping track of the names of the serial interfaces can be quite cumbersome, since udev simply names the devices in the order they are plugged in to the system. You can use udev rules to assign persistent names to the device files. Note that the rules do not actually match the Shimmer but the dock that it is located in. This means that you should always place the Shimmer in the same dock.

The following section provides an example of how to handle two Shimmer docks, one of which holds an ECG and the other a PPG device:

Distinguishing the Shimmer booloader and device interfaces based on their udev attributes is somewhat difficult because the distinguishing attributes are spread across multiple devices in the USB device tree. The rules first check the bInterfaceNumber of the tty device that is being processed. If the device is the bootloader device, its bInterfaceNumber is equal to 00. If the device is the interface to the Shimmer itself, bInterfaceNumber is equal to 01. In a second step, the rule set differentiates between the ECG dock and the PPG dock based on the serial number of the device. The entire udev ruleset is shown in the following code snippet:

SUBSYSTEMS=="usb" ATTRS{bInterfaceNumber}!="00" GOTO="is_secondary_interface"
SUBSYSTEM=="tty" ATTRS{idVendor}=="<id_vendor1>" ATTRS{idProduct}=="<id_product1>" ATTRS{serial}=="<id_serial1>" SYMLINK+="ttyPPGbl"
SUBSYSTEM=="tty" ATTRS{idVendor}=="<id_vendor2>" ATTRS{idProduct}=="<id_product2>" ATTRS{serial}=="<id_serial2>" SYMLINK+="ttyECGbl"
GOTO="end"

LABEL="is_secondary_interface"
SUBSYSTEM=="tty" ATTRS{idVendor}=="<id_vendor1>" ATTRS{idProduct}=="<id_product1>" ATTRS{serial}=="<id_serial1>" SYMLINK+="ttyPPGdev"
SUBSYSTEM=="tty" ATTRS{idVendor}=="<id_vendor2>" ATTRS{idProduct}=="<id_product2>" ATTRS{serial}=="<id_serial2>" SYMLINK+="ttyECGdev"
GOTO="end"

LABEL="end"

You can also find the example file in conf/udev/10-shimmer.rules.example.

In order to create a custom ruleset for your devices, create a new udev rule file /etc/udev/rules.d/10-shimmer.rules and add the above contents. In the file, you need to replace the <id_vendor1>, <id_product1>, and <id_serial1> of the first device, and the <id_vendor2>, <id_product2>, and <id_serial2> of the second device. You can find the values by scanning the dmesg command after plugging in a Shimmer device. Here is an example:

[144366.290357] usb 1-4.3: new full-speed USB device number 34 using xhci_hcd
[144366.386661] usb 1-4.3: New USB device found, idVendor=<id_vendor>, idProduct=<id_product>, bcdDevice= 5.00
[144366.386668] usb 1-4.3: New USB device strings: Mfr=1, Product=2, SerialNumber=3
[144366.386674] usb 1-4.3: Product: SHIMMER DOCK
[144366.386679] usb 1-4.3: Manufacturer: FTDI
[144366.386684] usb 1-4.3: SerialNumber: <id_serial>

Save the file and reload the rules for them to take effect:

udevadm control --reload-rules && udevadm trigger

You should now have two named device files for each Shimmer dock:

  • /dev/ttyPPGbl and /dev/ttyPPGdev for the PPG Shimmer bootloader and device interfaces,
  • /dev/ttyECGbl and /dev/ttyECGdev for the ECG Shimmer bootloader and device interfaces.

The library uses a tty serial interface to communicate with the Shimmer over Bluetooth. Before you can use the library, you need to set up the serial channel appropriately. This has only been tested under Arch Linux, but other Linux distributions should work as well.

Requirements:

  • Functioning Bluetooth stack
  • The rfcomm commandline tool. For Arch Linux, use the bluez-rfcomm AUR package
  • The hcitool commandline tool. For Arch Linux, use the bluez-hcitool AUR package
  • A Shimmer device with LogAndStream firmware

Scan for the device to find out its MAC address:

hcitool scan

The MAC address of the listed Shimmer device should end with the BT Radio ID imprinted on the back of the device. Next, you can try and ping the device:

hcitool name <mac_addr>

The command should complete with the name listed previously during the scan. Now you can pair the device as follows:

rfcomm <bind_id> <mac_address>

where <bind_id> is an arbitrary integer of your choosing. The command will create a new serial interface node with the following name: /dev/rfcomm<bind_id>. The file acts as a regular serial device and allows you to communicate with the Shimmer. The file is also used by the library.

If you want to connect to the Bluetooth interface, use the ShimmerBluetooth class. The API only offers blocking calls.

import time

from serial import Serial

from pyshimmer import ShimmerBluetooth, DEFAULT_BAUDRATE, DataPacket, EChannelType


def handler(pkt: DataPacket) -> None:
    cur_value = pkt[EChannelType.INTERNAL_ADC_13]
    print(f'Received new data point: {cur_value}')


if __name__ == '__main__':
    serial = Serial('/dev/rfcomm42', DEFAULT_BAUDRATE)
    shim_dev = ShimmerBluetooth(serial)

    shim_dev.initialize()

    dev_name = shim_dev.get_device_name()
    print(f'My name is: {dev_name}')

    shim_dev.add_stream_callback(handler)

    shim_dev.start_streaming()
    time.sleep(5.0)
    shim_dev.stop_streaming()

    shim_dev.shutdown()

The example shows how to make simple calls and how to use the Bluetooth streaming capabilities of the device.

from serial import Serial

from pyshimmer import ShimmerDock, DEFAULT_BAUDRATE, fmt_hex

if __name__ == '__main__':
    serial = Serial('/dev/ttyPPGdev', DEFAULT_BAUDRATE)
    shim_dock = ShimmerDock(serial)

    mac = shim_dock.get_mac_address()
    print(f'Device MAC: {fmt_hex(mac)}')

    shim_dock.close()

Using the Dock API works very similar to the Bluetooth API. However, it does not require a separate initialization call because it does not use a background thread to decode incoming messages.

from pyshimmer import ShimmerReader, EChannelType

if __name__ == '__main__':

    with open('test/reader/resources/ecg.bin', 'rb') as f:
        reader = ShimmerReader(f)

        # Read the file contents into memory
        reader.load_file_data()

        print(f'Available data channels: {reader.channels}')
        print(f'Sampling rate: {reader.sample_rate} Hz')
        print()

        ts = reader[EChannelType.TIMESTAMP]
        ecg_ch1 = reader[EChannelType.EXG_ADS1292R_1_CH1_24BIT]
        assert len(ts) == len(ecg_ch1)

        print(f'Timestamp: {ts.shape}')
        print(f'ECG Channel: {ecg_ch1.shape}')
        print()

        exg_reg = reader.exg_reg1
        print(f'ECG Chip Sampling Rate: {exg_reg.data_rate} Hz')
        print(f'ECG Chip Gain: {exg_reg.ch1_gain}')

If the data was recorded using the SDLog firmware and features synchronization information, the API automatically interpolates the data to the common timestamp information of the master.

Note: Please be aware that although you have configured a sampling frequency f for your measurements, it can happen that observations are missing. Usually the observed time difference is a multiple of the sampling period 1 / f. However, this is not the case for the time difference between the first two observations. Please take this caveat into consideration when you design your code.

About

pyshimmer provides a Python API to work with the wearable sensor devices produced by Shimmer.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages