Slight style changes to the pull of the P9813 device

Former-commit-id: 77892a5cfbcc4793061d518a761931be26ef842b
This commit is contained in:
T. van der Zwan 2014-01-18 22:49:32 +00:00
parent cb59a0e951
commit fd6d247506
4 changed files with 38 additions and 28 deletions

View File

@ -54,16 +54,16 @@ if(ENABLE_SPIDEV)
${CURRENT_SOURCE_DIR}/LedSpiDevice.h
${CURRENT_SOURCE_DIR}/LedDeviceLpd6803.h
${CURRENT_SOURCE_DIR}/LedDeviceLpd8806.h
${CURRENT_SOURCE_DIR}/LedDeviceWs2801.h
${CURRENT_SOURCE_DIR}/LedDeviceP9813.h
${CURRENT_SOURCE_DIR}/LedDeviceWs2801.h
)
SET(Leddevice_SOURCES
${Leddevice_SOURCES}
${CURRENT_SOURCE_DIR}/LedSpiDevice.cpp
${CURRENT_SOURCE_DIR}/LedDeviceLpd6803.cpp
${CURRENT_SOURCE_DIR}/LedDeviceLpd8806.cpp
${CURRENT_SOURCE_DIR}/LedDeviceWs2801.cpp
${CURRENT_SOURCE_DIR}/LedDeviceP9813.cpp
${CURRENT_SOURCE_DIR}/LedDeviceWs2801.cpp
)
endif(ENABLE_SPIDEV)

View File

@ -9,8 +9,8 @@
#ifdef ENABLE_SPIDEV
#include "LedDeviceLpd6803.h"
#include "LedDeviceLpd8806.h"
#include "LedDeviceWs2801.h"
#include "LedDeviceP9813.h"
#include "LedDeviceWs2801.h"
#endif
#include "LedDeviceAdalight.h"

View File

@ -13,44 +13,44 @@
LedDeviceP9813::LedDeviceP9813(const std::string& outputDevice, const unsigned baudrate) :
LedSpiDevice(outputDevice, baudrate, 0),
mLedCount(0)
_ledCount(0)
{
// empty
}
int LedDeviceP9813::write(const std::vector<ColorRgb> &ledValues)
{
mLedCount = ledValues.size();
if (_ledCount != ledValues.size())
{
_ledBuf.resize(ledValues.size() * 4 + 8, 0x00);
_ledCount = ledValues.size();
}
const unsigned dataLen = ledValues.size() * 4 + 8;
uint8_t data[dataLen];
uint8_t * dataPtr = _ledBuf.data();
for (const ColorRgb & color : ledValues)
{
*dataPtr++ = calculateChecksum(color);
*dataPtr++ = color.blue;
*dataPtr++ = color.green;
*dataPtr++ = color.red;
}
memset(data, 0x00, dataLen);
int j = 4;
for (unsigned i = 0; i < mLedCount; i++){
data[j++] = calculateChecksum(ledValues[i]);
data[j++] = ledValues[i].blue;
data[j++] = ledValues[i].green;
data[j++] = ledValues[i].red;
}
return writeBytes(dataLen, data);
return writeBytes(_ledBuf.size(), _ledBuf.data());
}
int LedDeviceP9813::switchOff()
{
return write(std::vector<ColorRgb>(mLedCount, ColorRgb{0,0,0}));
return write(std::vector<ColorRgb>(_ledCount, ColorRgb{0,0,0}));
}
const uint8_t LedDeviceP9813::calculateChecksum(const ColorRgb color)
uint8_t LedDeviceP9813::calculateChecksum(const ColorRgb & color) const
{
uint8_t res = 0;
uint8_t res = 0;
res |= (uint8_t)0x03 << 6;
res |= (uint8_t)(~(color.blue >> 6) & 0x03) << 4;
res |= (uint8_t)(~(color.green >> 6) & 0x03) << 2;
res |= (uint8_t)(~(color.red >> 6) & 0x03);
res |= (uint8_t)0x03 << 6;
res |= (uint8_t)(~(color.blue >> 6) & 0x03) << 4;
res |= (uint8_t)(~(color.green >> 6) & 0x03) << 2;
res |= (uint8_t)(~(color.red >> 6) & 0x03);
return res;
return res;
}

View File

@ -35,6 +35,16 @@ public:
private:
/// the number of leds
size_t mLedCount;
const uint8_t calculateChecksum(const ColorRgb color);
size_t _ledCount;
/// Buffer for writing/written led data
std::vector<uint8_t> _ledBuf;
///
/// Calculates the required checksum for one led
///
/// @param color The color of the led
/// @return The checksum for the led
///
uint8_t calculateChecksum(const ColorRgb & color) const;
};