APA102: Initial implementation

Signed-off-by: Yannick Adam <yannick.adam@gmail.com>
Signed-off-by: Abhishek Malik <abhishek.malik@intel.com>

src/apa102/CMakeLists.txt

@@ -0,0 +1,5 @@
+set (libname "apa102")
+set (libdescription "upm apa102 led strip spi output module")
+set (module_src ${libname}.cxx)
+set (module_h ${libname}.h)
+upm_module_init()

src/apa102/apa102.cxx

@@ -0,0 +1,166 @@
+/*
+ * Author: Yannick Adam <yannick.adam@gmail.com>
+ * Copyright (c) 2016 Yannick Adam
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+ * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+ * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+#include <cstring>
+#include <iostream>
+#include <stdexcept>
+#include <stdlib.h>
+#include <unistd.h>
+
+#include "apa102.h"
+
+using namespace upm;
+
+APA102::APA102(uint16_t ledCount, uint8_t spiBus, bool batchMode, int8_t csn)
+        : m_ledCount(ledCount), m_batchMode(batchMode)
+{
+    mraa::Result res = mraa::SUCCESS;
+    m_leds = NULL;
+
+    // Optional chip select pin
+    m_csnPinCtx = NULL;
+    if (csn > -1) {
+        m_csnPinCtx = new mraa::Gpio(csn);
+        res = m_csnPinCtx->dir(mraa::DIR_OUT);
+        if (res != mraa::SUCCESS) {
+            throw std::invalid_argument(std::string(__FUNCTION__) +
+                                        ": GPIO failed to set direction");
+        }
+    }
+
+    CSOff();
+    // Initialize SPI
+    m_spi = new mraa::Spi(spiBus);
+
+    // Initialize LED array
+    uint16_t endFrameLength = (m_ledCount + 15) / 16; // End frame should be (leds/2) bits
+    m_frameLength = endFrameLength + (m_ledCount + 1) * 4;
+    if ((m_leds = (uint8_t*) malloc(m_frameLength))) {
+        memset(m_leds, 0x00, m_frameLength - 4);                               // Clear state
+        memset(&m_leds[m_frameLength - endFrameLength], 0xFF, endFrameLength); // Frame End
+
+        // Need to set the brightness to "0" for each Led
+        for (int i = 1; i <= m_ledCount; i++) {
+            m_leds[i * 4] = 224;
+        }
+
+    } else {
+        throw std::runtime_error(std::string(__FUNCTION__) +
+                                 ": Failed to allocate memory for LED Strip");
+    }
+}
+
+APA102::~APA102()
+{
+    // Clear leds
+    if (m_leds) {
+        free(m_leds);
+    }
+
+    // Clear SPI
+    if (m_spi) {
+        delete m_spi;
+    }
+
+    // Clear GPIO
+    if (m_csnPinCtx) {
+        delete m_csnPinCtx;
+    }
+}
+
+
+void
+APA102::setLed(uint16_t ledIdx, uint8_t brightness, uint8_t r, uint8_t g, uint8_t b)
+{
+    setLeds(ledIdx, ledIdx, brightness, r, g, b);
+}
+
+void
+APA102::setAllLeds(uint8_t brightness, uint8_t r, uint8_t g, uint8_t b)
+{
+    setLeds(0, m_ledCount - 1, brightness, r, g, b);
+}
+
+void
+APA102::setLeds(uint16_t startIdx, uint16_t endIdx, uint8_t brightness, uint8_t r, uint8_t g, uint8_t b)
+{
+    uint16_t s_idx = (startIdx + 1) * 4;
+    uint16_t e_idx = (endIdx + 1) * 4;
+
+    for (uint16_t i = s_idx; i <= e_idx; i += 4) {
+        m_leds[i] = brightness | 224;
+        m_leds[i + 1] = b;
+        m_leds[i + 2] = g;
+        m_leds[i + 3] = r;
+    }
+
+    if (!m_batchMode) {
+        pushState();
+    }
+}
+
+void
+APA102::setLeds(uint16_t startIdx, uint16_t endIdx, uint8_t* colors)
+{
+    uint16_t s_idx = (startIdx + 1) * 4;
+    memcpy(&m_leds[s_idx], colors, (endIdx - startIdx + 1) * 4);
+
+    if (!m_batchMode) {
+        pushState();
+    }
+}
+
+void
+APA102::pushState(void)
+{
+    CSOn();
+    m_spi->write(m_leds, m_frameLength);
+    CSOff();
+}
+
+/*
+ * **************
+ *  private area
+ * **************
+ */
+
+mraa::Result
+APA102::CSOn()
+{
+    if (m_csnPinCtx) {
+        return m_csnPinCtx->write(HIGH);
+    }
+
+    return mraa::ERROR_FEATURE_NOT_SUPPORTED;
+}
+
+mraa::Result
+APA102::CSOff()
+{
+    if (m_csnPinCtx) {
+        return m_csnPinCtx->write(LOW);
+    }
+
+    return mraa::ERROR_FEATURE_NOT_SUPPORTED;
+}

src/apa102/apa102.h

@@ -0,0 +1,142 @@
+/*
+ * Author: Yannick Adam <yannick.adam@gmail.com>
+ * Copyright (c) 2016 Yannick Adam
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+ * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+ * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+#pragma once
+
+#include <mraa/gpio.hpp>
+#include <mraa/spi.hpp>
+#include <string>
+
+#define HIGH 1
+#define LOW 0
+
+namespace upm
+{
+/**
+ * @brief APA102 RGB LED Strip driver library
+ * @defgroup apa102 libupm-apa102
+ * @ingroup spi led
+ */
+
+/**
+ * @library apa102
+ * @sensor apa102
+ * @comname APA102/DotStar LED Strip
+ * @type led
+ * @man adafruit
+ * @con spi
+ *
+ * @brief API for controlling APA102/DotStar RGB LED Strips
+ *
+ * APA102 LED Strips provide individually controllable LEDs through a SPI interface.
+ * For each LED, brightness (0-31) and RGB (0-255) values can be set.
+ *
+ * @image html apa102.jpg
+ */
+class APA102
+{
+  public:
+    /**
+     * Instantiates a new APA102 LED Strip
+     *
+     * @param ledCount 	Number of APA102 leds in the strip
+     * @param spiBus 	SPI Bus number
+     * @param batchMode (optional) Immediatly write to SPI (false, default) or wait for a pushState
+     * call (true)
+     * @param csn 		(optional) Chip Select Pin
+     */
+    APA102(uint16_t ledCount, uint8_t spiBus, bool batchMode = false, int8_t csn = -1);
+
+    /**
+     * APA102 destructor
+     */
+    ~APA102();
+
+    /**
+     * Change the color for a single led
+     *
+     * @param ledIdx		Index of the LED in the strip (0 based)
+     * @param brightness	Brightness value (0-31)
+     * @param r				Red component (0-255)
+     * @param g				Green component (0-255)
+     * @param b				Blue component (0-255)
+     */
+    void setLed(uint16_t ledIdx, uint8_t brightness, uint8_t r, uint8_t g, uint8_t b);
+
+    /**
+     * Change the color for all leds
+     *
+     * @param brightness	Brightness value (0-31)
+     * @param r				Red component (0-255)
+     * @param g				Green component (0-255)
+     * @param b				Blue component (0-255)
+     */
+    void setAllLeds(uint8_t brightness, uint8_t r, uint8_t g, uint8_t b);
+
+    /**
+     * Change the color for a range of leds
+     *
+     * @param startIdx		Start index of the range of LEDs in the strip (0 based)
+     * @param endIdx		End index of the range of LEDs in the strip (0 based)
+     * @param brightness	Brightness value (0-31)
+     * @param r				Red component (0-255)
+     * @param g				Green component (0-255)
+     * @param b				Blue component (0-255)
+     */
+    void
+    setLeds(uint16_t startIdx, uint16_t endIdx, uint8_t brightness, uint8_t r, uint8_t g, uint8_t b);
+
+    /**
+     * (Advanced) Manually control the colors of a range of LEDS
+     * Best used to maximize performance
+     *
+     * @param startIdx		Start index of the range of LEDs to update (0 based)
+     * @param endIdx		End index of the range of LEDs to update (0 based)
+     * @param colors		Pointer to an array of bytes. Each color is described as the following:
+     *						B1: Brightness (224-255) B2: Blue (0-255) B3: Green (0-255) B4: Red
+     *(0-255)
+     *						No check done on the boundaries
+     */
+    void setLeds(uint16_t startIdx, uint16_t endIdx, uint8_t* colors);
+
+    /**
+     * Outputs the current LED data to the SPI bus
+     * Note: Only required if batch mode is set to TRUE
+     *
+     */
+    void pushState();
+
+  private:
+    mraa::Spi* m_spi;
+    mraa::Gpio* m_csnPinCtx;
+
+    uint16_t m_ledCount;
+    uint8_t* m_leds;
+    uint16_t m_frameLength;
+
+    bool m_batchMode;
+
+    mraa::Result CSOn();
+    mraa::Result CSOff();
+};
+}

src/apa102/javaupm_apa102.i

@@ -0,0 +1,30 @@
+%module javaupm_apa102
+
+%include "../upm.i"
+%include "typemaps.i"
+
+%typemap(jtype) (uint8_t *colors) "byte[]"
+%typemap(jstype) (uint8_t *colors) "byte[]"
+%typemap(jni) (uint8_t *colors) "jbyteArray"
+%typemap(javain) (uint8_t *colors) "$javainput"
+
+%typemap(in) (uint8_t *colors) {
+  $1 = (uint8_t*)JCALL2(GetByteArrayElements, jenv, $input, NULL);
+}
+
+%{
+    #include "apa102.h"
+%}
+
+%include "apa102.h"
+
+%pragma(java) jniclasscode=%{
+    static {
+        try {
+            System.loadLibrary("javaupm_apa102");
+        } catch (UnsatisfiedLinkError e) {
+            System.err.println("Native code library failed to load. \n" + e);
+            System.exit(1);
+        }
+    }
+%}

src/apa102/jsupm_apa102.i

@@ -0,0 +1,19 @@
+%module jsupm_apa102
+ 
+%include "../upm.i"
+%inline %{
+    #include <node_buffer.h>
+%}
+
+%typemap(in) (uint8_t *colors) {
+  if (!node::Buffer::HasInstance($input)) {
+      SWIG_exception_fail(SWIG_ERROR, "Expected a node Buffer");
+  }
+  $1 = (uint8_t*) node::Buffer::Data($input);
+}
+
+%include "apa102.h" 
+
+%{
+    #include "apa102.h"
+%}

src/apa102/pyupm_apa102.i

@@ -0,0 +1,25 @@
+// Include doxygen-generated documentation
+%include "pyupm_doxy2swig.i"
+ 
+%module pyupm_apa102
+ 
+%include "../upm.i"
+ 
+%feature("autodoc", "3");
+ 
+// setLeds 
+%typemap(in) (uint8_t *colors) {
+  if (PyByteArray_Check($input)) {
+    $1 = (uint8_t*) PyByteArray_AsString($input);
+  } else {
+    PyErr_SetString(PyExc_ValueError, "bytearray expected");
+    return NULL;
+  }
+}
+ 
+%include "apa102.h"
+ 
+%{
+ 
+    #include "apa102.h"
+%}