// STL includes #include #include // Local LedDevice includes #include "LedDeviceTinkerforge.h" static const unsigned MAX_NUM_LEDS = 320; static const unsigned MAX_NUM_LEDS_SETTABLE = 16; LedDeviceTinkerforge::LedDeviceTinkerforge(const QJsonObject &deviceConfig) : LedDevice() , _ipConnection(nullptr) , _ledStrip(nullptr) , _colorChannelSize(0) { init(deviceConfig); } LedDeviceTinkerforge::~LedDeviceTinkerforge() { // Close the device (if it is opened) if (_ipConnection != nullptr && _ledStrip != nullptr) { switchOff(); } // Clean up claimed resources delete _ipConnection; delete _ledStrip; } bool LedDeviceTinkerforge::init(const QJsonObject &deviceConfig) { _host = deviceConfig["output"].toString("127.0.0.1").toStdString(); _port = deviceConfig["port"].toInt(4223); _uid = deviceConfig["uid"].toString().toStdString(); _interval = deviceConfig["rate"].toInt(); if ((unsigned)_ledCount > MAX_NUM_LEDS) { Error(_log,"Invalid attempt to write led values. Not more than %d leds are allowed.", MAX_NUM_LEDS); return -1; } if (_colorChannelSize < (unsigned)_ledCount) { _redChannel.resize(_ledCount, uint8_t(0)); _greenChannel.resize(_ledCount, uint8_t(0)); _blueChannel.resize(_ledCount, uint8_t(0)); } _colorChannelSize = _ledCount; return true; } LedDevice* LedDeviceTinkerforge::construct(const QJsonObject &deviceConfig) { return new LedDeviceTinkerforge(deviceConfig); } int LedDeviceTinkerforge::open() { // Check if connection is already createds if (_ipConnection != nullptr) { Error(_log, "Attempt to open existing connection; close before opening"); return -1; } // Initialise a new connection _ipConnection = new IPConnection; ipcon_create(_ipConnection); int connectionStatus = ipcon_connect(_ipConnection, _host.c_str(), _port); if (connectionStatus < 0) { Warning(_log, "Attempt to connect to master brick (%s:%d) failed with status %d", _host.c_str(), _port, connectionStatus); return -1; } // Create the 'LedStrip' _ledStrip = new LEDStrip; led_strip_create(_ledStrip, _uid.c_str(), _ipConnection); int frameStatus = led_strip_set_frame_duration(_ledStrip, _interval); if (frameStatus < 0) { Error(_log,"Attempt to connect to led strip bricklet (led_strip_set_frame_duration()) failed with status %d", frameStatus); return -1; } return 0; } int LedDeviceTinkerforge::write(const std::vector &ledValues) { auto redIt = _redChannel.begin(); auto greenIt = _greenChannel.begin(); auto blueIt = _blueChannel.begin(); for (const ColorRgb &ledValue : ledValues) { *redIt = ledValue.red; ++redIt; *greenIt = ledValue.green; ++greenIt; *blueIt = ledValue.blue; ++blueIt; } return transferLedData(_ledStrip, 0, _colorChannelSize, _redChannel.data(), _greenChannel.data(), _blueChannel.data()); } int LedDeviceTinkerforge::transferLedData(LEDStrip *ledStrip, unsigned index, unsigned length, uint8_t *redChannel, uint8_t *greenChannel, uint8_t *blueChannel) { if (length == 0 || index >= length || length > MAX_NUM_LEDS) { return E_INVALID_PARAMETER; } uint8_t reds[MAX_NUM_LEDS_SETTABLE]; uint8_t greens[MAX_NUM_LEDS_SETTABLE]; uint8_t blues[MAX_NUM_LEDS_SETTABLE]; for (unsigned i=index; i length) ? length - i : MAX_NUM_LEDS_SETTABLE; memcpy(reds, redChannel + i, copyLength); memcpy(greens, greenChannel + i, copyLength); memcpy(blues, blueChannel + i, copyLength); const int status = led_strip_set_rgb_values(ledStrip, i, copyLength, reds, greens, blues); if (status != E_OK) { Warning(_log, "Setting led values failed with status %d", status); return status; } } return E_OK; }