rpi_pcm_ws281x.i

// SWIG interface file to define rpi_pcm_ws281x library python wrapper.
// Author: Tony DiCola (tony@tonydicola.com), Jeremy Garff (jer@jers.net)

// Define module name rpi_pcm_ws281x.  This will actually be imported under
// the name _rpi_pcm_ws281x following the SWIG & Python conventions.
%module rpi_pcm_ws281x

// Include standard SWIG types & array support for support of uint32_t
// parameters and arrays.
%include "stdint.i"
%include "carrays.i"

// Declare functions which will be exported as anything in the ws2811.h header.
%{
#include "lib/ws2811-pcm.h"
%}

// Process ws2811.h header and export all included functions.
%include "lib/ws2811-pcm.h"

%inline %{
    uint32_t ws2811_led_get(ws2811_channel_t *channel, int lednum)
    {
        if (lednum >= channel->count)
        {
            return -1;
        }

        return channel->leds[lednum];
    }

    int ws2811_led_set(ws2811_channel_t *channel, int lednum, uint32_t color)
    {
        if (lednum >= channel->count)
        {
            return -1;
        }

        channel->leds[lednum] = color;

        return 0;
    }
%}