implement origin for effects (#408)

* implement rigin for efx

* implement origin for effects and other components
add experimental adalight firmware for arduino with upto 5 pwm channels

* cleanup

* origin ip now with dns lookup

* fix compile

* move some code
This commit is contained in:
redPanther 2017-03-01 15:23:53 +01:00 committed by GitHub
parent 31f352e7ce
commit 221af075a4
24 changed files with 376 additions and 98 deletions

View File

@ -0,0 +1,250 @@
#include "FastLED.h"
#define ANALOG_MODE_AVERAGE 0
#define ANALOG_MODE_LAST_LED 1
/**************************************
S E T U P
set following values to your needs
**************************************/
// Number of leds in your strip. set to 1 and ANALOG_OUTPUT_ENABLED to true to activate analog only
#define NUM_LEDS 100
#define LED_TYPE WS2812B // type of your led controller, possible values, see below
// 3 wire (pwm): NEOPIXEL BTM1829 TM1812 TM1809 TM1804 TM1803 UCS1903 UCS1903B UCS1904 UCS2903 WS2812 WS2852
// S2812B SK6812 SK6822 APA106 PL9823 WS2811 WS2813 APA104 WS2811_40 GW6205 GW6205_40 LPD1886 LPD1886_8BIT
// 4 wire (spi): LPD8806 WS2801 WS2803 SM16716 P9813 APA102 SK9822 DOTSTAR
// For 3 wire led stripes line Neopixel/Ws2812, which have a data line, ground, and power, you just need to define DATA_PIN.
// For led chipsets that are SPI based (four wires - data, clock, ground, and power), both defines DATA_PIN and CLOCK_PIN are needed
// DATA_PIN, or DATA_PIN, CLOCK_PIN
#define LED_PINS MOSI // 3 wire leds
//#define LED_PINS MOSI, SCK // 4 wire leds
#define COLOR_ORDER GRB // colororder of the stripe, set RGB in hyperion
#define OFF_TIMEOUT 15000 // ms to switch off after no data was received, set 0 to deactivate
// analog rgb uni color led stripe - using of hyperion smoothing is recommended
#define ANALOG_MODE ANALOG_MODE_LAST_LED // use ANALOG_MODE_AVERAGE or ANALOG_MODE_LAST_LED
// overall color adjustments
#define ANALOG_BRIGHTNESS_RED 255 // maximum brightness for analog 0-255
#define ANALOG_BRIGHTNESS_GREEN 255 // maximum brightness for analog 0-255
#define ANALOG_BRIGHTNESS_BLUE 255 // maximum brightness for analog 0-255
#define BRIGHTNESS 128 // maximum brightness 0-255
#define DITHER_MODE BINARY_DITHER // BINARY_DITHER or DISABLE_DITHER
#define COLOR_TEMPERATURE CRGB(255,255,255) // RGB value describing the color temperature
#define COLOR_CORRECTION TypicalLEDStrip // predefined fastled color correction
//#define COLOR_CORRECTION CRGB(255,255,255) // or RGB value describing the color correction
// Baudrate, higher rate allows faster refresh rate and more LEDs (defined in /etc/boblight.conf)
#define serialRate 115200 // use 115200 for ftdi based boards
//#define serialRate 460800 // use 115200 for ftdi based boards
// ATTENTION this pin config is default for atmega328 based arduinos, others might work to
// if you have flickering analog leds this might be caused by unsynced pwm signals
// try other pins is more or less the only thing that helps
#if defined(__AVR_ATmega1280__) || defined(__AVR_ATmega2560__)
// 2 3 4 5 6 7 8 9 10 11 12 13 44 45 46
// R B G R B G R B G R B G R B G
#define ANALOG_PINS_MAX 15
#define ANALOG_RGB_STRIPES 4
const byte ANALOG_PINS[ANALOG_PINS_MAX] = {2,4,3,5,7,6,8,10,9,11,13,12,44,46,45};
#else
// 9 10 11
// R B G
#define ANALOG_PINS_MAX 3
#define ANALOG_RGB_STRIPES 1
const byte ANALOG_PINS[ANALOG_PINS_MAX] = {9,11,10};
#endif
/**************************************
A D A L I G H T C O D E
no user changes needed
**************************************/
// Adalight sends a "Magic Word" (defined in /etc/boblight.conf) before sending the pixel data
uint8_t prefix[] = {'A', 'd', 'a'}, hi, lo, chk, i;
unsigned long endTime;
// Define the array of leds
CRGB leds[NUM_LEDS];
// set rgb to analog led stripe
void showAnalogRGB(const CRGB& led, const short stripeId=-1) {
if (ANALOG_RGB_STRIPES > 0) {
byte r = map(led.r, 0,255,0,ANALOG_BRIGHTNESS_RED);
byte g = map(led.g, 0,255,0,ANALOG_BRIGHTNESS_GREEN);
byte b = map(led.b, 0,255,0,ANALOG_BRIGHTNESS_BLUE);
if (stripeId<0) {
for (byte i=0;i<ANALOG_RGB_STRIPES;i++) {
analogWrite(ANALOG_PINS[3*i+1], r);
analogWrite(ANALOG_PINS[3*i] , g);
analogWrite(ANALOG_PINS[3*i+2], b);
}
} else {
analogWrite(ANALOG_PINS[stripeId*3] , g);
analogWrite(ANALOG_PINS[stripeId*3+1], r);
analogWrite(ANALOG_PINS[stripeId*3+2], b);
}
}
}
// set color to all leds
void showColor(const CRGB& led) {
#if NUM_LEDS > ANALOG_RGB_STRIPES
LEDS.showColor(led);
#endif
showAnalogRGB(led);
}
// function to check if serial data is available
// if timeout occured leds switch of, if configured
bool checkIncommingData() {
boolean dataAvailable = true;
while (!Serial.available()) {
if ( OFF_TIMEOUT > 0 && endTime < millis()) {
showColor(CRGB(0,0,0)); // leds off
dataAvailable = false;
endTime = millis() + OFF_TIMEOUT;
}
}
return dataAvailable;
}
// main function that setups and runs the code
void setup() {
// analog output
if (ANALOG_RGB_STRIPES > 0) {
for (byte i=0;i<ANALOG_PINS_MAX;i++) {
pinMode(ANALOG_PINS[i], OUTPUT);
}
}
// Uncomment/edit one of the following lines for your leds arrangement.
int ledCount = NUM_LEDS;
if (ANALOG_MODE == ANALOG_MODE_LAST_LED) {
ledCount -= ANALOG_RGB_STRIPES;
}
#if NUM_LEDS > ANALOG_RGB_STRIPES
FastLED.addLeds<LED_TYPE, LED_PINS, COLOR_ORDER>(leds, ledCount);
#endif
// color adjustments
FastLED.setBrightness ( BRIGHTNESS );
FastLED.setTemperature( COLOR_TEMPERATURE );
FastLED.setCorrection ( COLOR_CORRECTION );
FastLED.setDither ( DITHER_MODE );
// initial RGB flash
showColor(CRGB(255, 0, 0)); delay(400);
showColor(CRGB(0, 255, 0)); delay(400);
showColor(CRGB(0, 0, 255)); delay(400);
showColor(CRGB(0, 0, 0));
Serial.begin(serialRate);
while (!Serial) {
; // wait for serial port to connect. Needed for native USB
}
Serial.print("Ada\n"); // Send "Magic Word" string to host
boolean transmissionSuccess;
unsigned long sum_r, sum_g, sum_b;
// loop() is avoided as even that small bit of function overhead
// has a measurable impact on this code's overall throughput.
while (true) {
// wait for first byte of Magic Word
for (i = 0; i < sizeof prefix; ++i) {
// If next byte is not in Magic Word, the start over
if (!checkIncommingData() || prefix[i] != Serial.read()) {
i = 0;
}
}
// Hi, Lo, Checksum
if (!checkIncommingData()) continue;
hi = Serial.read();
if (!checkIncommingData()) continue;
lo = Serial.read();
if (!checkIncommingData()) continue;
chk = Serial.read();
// if checksum does not match go back to wait
if (chk != (hi ^ lo ^ 0x55)) continue;
memset(leds, 0, NUM_LEDS * sizeof(struct CRGB));
transmissionSuccess = true;
sum_r = 0;
sum_g = 0;
sum_b = 0;
// read the transmission data and set LED values
for (uint8_t idx = 0; idx < NUM_LEDS; idx++) {
byte r, g, b;
if (!checkIncommingData()) {
transmissionSuccess = false;
break;
}
r = Serial.read();
if (!checkIncommingData()) {
transmissionSuccess = false;
break;
}
g = Serial.read();
if (!checkIncommingData()) {
transmissionSuccess = false;
break;
}
b = Serial.read();
leds[idx].r = r;
leds[idx].g = g;
leds[idx].b = b;
#if ANALOG_OUTPUT_ENABLED == true && ANALOG_MODE == ANALOG_MODE_AVERAGE
sum_r += r;
sum_g += g;
sum_b += b;
#endif
}
// shows new values
if (transmissionSuccess) {
endTime = millis() + OFF_TIMEOUT;
#if NUM_LEDS > ANALOG_RGB_STRIPES
FastLED.show();
#endif
#if ANALOG_RGB_STRIPES > 0
#if ANALOG_MODE == ANALOG_MODE_LAST_LED
for ( byte x=1; x<=ANALOG_RGB_STRIPES; x++) {
showAnalogRGB(leds[NUM_LEDS-x], x-1);
}
#else
showAnalogRGB(CRGB(sum_r/NUM_LEDS, sum_g/NUM_LEDS, sum_b/NUM_LEDS));
#endif
#endif
}
}
} // end of setup
void loop() {
// Not used. See note in setup() function.
}

View File

@ -45,13 +45,10 @@ public:
public slots: public slots:
/// Run the specified effect on the given priority channel and optionally specify a timeout /// Run the specified effect on the given priority channel and optionally specify a timeout
int runEffect(const QString &effectName, int priority, int timeout = -1) int runEffect(const QString &effectName, int priority, int timeout = -1, const QString &origin="System");
{
return runEffect(effectName, QJsonObject(), priority, timeout);
};
/// Run the specified effect on the given priority channel and optionally specify a timeout /// Run the specified effect on the given priority channel and optionally specify a timeout
int runEffect(const QString &effectName, const QJsonObject & args, int priority, int timeout = -1, QString pythonScript = ""); int runEffect(const QString &effectName, const QJsonObject & args, int priority, int timeout = -1, const QString &pythonScript = "", const QString &origin = "System");
/// Clear any effect running on the provided channel /// Clear any effect running on the provided channel
void channelCleared(int priority); void channelCleared(int priority);
@ -68,7 +65,7 @@ private:
bool loadEffectSchema(const QString & path, const QString & effectSchemaFile, EffectSchema &effectSchema); bool loadEffectSchema(const QString & path, const QString & effectSchemaFile, EffectSchema &effectSchema);
/// Run the specified effect on the given priority channel and optionally specify a timeout /// Run the specified effect on the given priority channel and optionally specify a timeout
int runEffectScript(const QString &script, const QString &name, const QJsonObject & args, int priority, int timeout = -1); int runEffectScript(const QString &script, const QString &name, const QJsonObject & args, int priority, int timeout = -1, const QString & origin="System");
private: private:
Hyperion * _hyperion; Hyperion * _hyperion;

View File

@ -52,7 +52,7 @@ class Hyperion : public QObject
public: public:
/// Type definition of the info structure used by the priority muxer /// Type definition of the info structure used by the priority muxer
typedef PriorityMuxer::InputInfo InputInfo; typedef PriorityMuxer::InputInfo InputInfo;
typedef std::map<std::string,int> PriorityRegister; typedef std::map<QString,int> PriorityRegister;
/// ///
/// RGB-Color channel enumeration /// RGB-Color channel enumeration
@ -132,12 +132,13 @@ public:
/// register a input source to a priority channel /// register a input source to a priority channel
/// @param name uniq name of input source /// @param name uniq name of input source
/// @param origin External setter
/// @param priority priority channel /// @param priority priority channel
void registerPriority(const std::string name, const int priority); void registerPriority(const QString &name, const int priority);
/// unregister a input source to a priority channel /// unregister a input source to a priority channel
/// @param name uniq name of input source /// @param name uniq name of input source
void unRegisterPriority(const std::string name); void unRegisterPriority(const QString &name);
/// gets current priority register /// gets current priority register
/// @return the priority register /// @return the priority register
@ -243,14 +244,15 @@ public slots:
/// @param effectName Name of the effec to run /// @param effectName Name of the effec to run
/// @param priority The priority channel of the effect /// @param priority The priority channel of the effect
/// @param timeout The timeout of the effect (after the timout, the effect will be cleared) /// @param timeout The timeout of the effect (after the timout, the effect will be cleared)
int setEffect(const QString & effectName, int priority, int timeout = -1); int setEffect(const QString & effectName, int priority, int timeout = -1, const QString & origin="System");
/// Run the specified effect on the given priority channel and optionally specify a timeout /// Run the specified effect on the given priority channel and optionally specify a timeout
/// @param effectName Name of the effec to run /// @param effectName Name of the effec to run
/// @param args arguments of the effect script /// @param args arguments of the effect script
/// @param priority The priority channel of the effect /// @param priority The priority channel of the effect
/// @param timeout The timeout of the effect (after the timout, the effect will be cleared) /// @param timeout The timeout of the effect (after the timout, the effect will be cleared)
int setEffect(const QString & effectName, const QJsonObject & args, int priority, int timeout = -1, QString pythonScript = ""); int setEffect(const QString & effectName, const QJsonObject & args, int priority,
int timeout = -1, const QString & pythonScript = "", const QString & origin="System");
/// sets the methode how image is maped to leds /// sets the methode how image is maped to leds
void setLedMappingType(int mappingType); void setLedMappingType(int mappingType);

View File

@ -63,7 +63,7 @@ private slots:
/// Slot which is called when a client tries to create a new connection /// Slot which is called when a client tries to create a new connection
/// ///
void readPendingDatagrams(); void readPendingDatagrams();
void processTheDatagram(const QByteArray * _datagram); void processTheDatagram(const QByteArray * datagram, const QHostAddress * sender);
private: private:
/// Hyperion instance /// Hyperion instance
@ -78,7 +78,7 @@ private:
/// hyperion priority /// hyperion priority
int _priority; int _priority;
/// hyperion priority /// hyperion timeout
int _timeout; int _timeout;
/// Logger instance /// Logger instance

View File

@ -18,7 +18,8 @@ enum Components
COMP_GRABBER, COMP_GRABBER,
COMP_V4L, COMP_V4L,
COMP_COLOR, COMP_COLOR,
COMP_EFFECT COMP_EFFECT,
COMP_PROTOSERVER
}; };
inline const char* componentToString(Components c) inline const char* componentToString(Components c)
@ -35,6 +36,7 @@ inline const char* componentToString(Components c)
case COMP_V4L: return "V4L capture device"; case COMP_V4L: return "V4L capture device";
case COMP_COLOR: return "Solid color"; case COMP_COLOR: return "Solid color";
case COMP_EFFECT: return "Effect"; case COMP_EFFECT: return "Effect";
case COMP_PROTOSERVER: return "Proto Server";
default: return ""; default: return "";
} }
} }
@ -53,6 +55,7 @@ inline const char* componentToIdString(Components c)
case COMP_V4L: return "V4L"; case COMP_V4L: return "V4L";
case COMP_COLOR: return "COLOR"; case COMP_COLOR: return "COLOR";
case COMP_EFFECT: return "EFFECT"; case COMP_EFFECT: return "EFFECT";
case COMP_PROTOSERVER: return "PROTOSERVER";
default: return ""; default: return "";
} }
} }
@ -70,8 +73,9 @@ inline Components stringToComponent(QString component)
if (component == "V4L") return COMP_V4L; if (component == "V4L") return COMP_V4L;
if (component == "COLOR") return COMP_COLOR; if (component == "COLOR") return COMP_COLOR;
if (component == "EFFECT") return COMP_EFFECT; if (component == "EFFECT") return COMP_EFFECT;
if (component == "PROTOSERVER") return COMP_PROTOSERVER;
return COMP_INVALID; return COMP_INVALID;
} }
} }; // end of namespace

View File

@ -11,7 +11,7 @@
#include <map> #include <map>
#include <QVector> #include <QVector>
#include <utils/global_defines.h>
// standard log messages // standard log messages
//#define _FUNCNAME_ __PRETTY_FUNCTION__ //#define _FUNCNAME_ __PRETTY_FUNCTION__
@ -99,3 +99,5 @@ protected:
QVector<Logger::T_LOG_MESSAGE> _logMessageBuffer; QVector<Logger::T_LOG_MESSAGE> _logMessageBuffer;
const int _loggerMaxMsgBufferSize; const int _loggerMaxMsgBufferSize;
}; };
Q_DECLARE_METATYPE(Logger::T_LOG_MESSAGE);

View File

@ -0,0 +1,4 @@
#pragma once
#define QSTRING_CSTR(str) str.toLocal8Bit().constData()

View File

@ -12,6 +12,7 @@
// Qt includes // Qt includes
#include <QResource> #include <QResource>
#include <QDateTime> #include <QDateTime>
#include <QHostInfo>
// hyperion util includes // hyperion util includes
#include "hyperion/ImageProcessorFactory.h" #include "hyperion/ImageProcessorFactory.h"
@ -32,6 +33,7 @@ BoblightClientConnection::BoblightClientConnection(QTcpSocket *socket, const int
, _priority(priority) , _priority(priority)
, _ledColors(Hyperion::getInstance()->getLedCount(), ColorRgb::BLACK) , _ledColors(Hyperion::getInstance()->getLedCount(), ColorRgb::BLACK)
, _log(Logger::getInstance("BOBLIGHT")) , _log(Logger::getInstance("BOBLIGHT"))
, _clientAddress(QHostInfo::fromName(socket->peerAddress().toString()).hostName())
{ {
// initalize the locale. Start with the default C-locale // initalize the locale. Start with the default C-locale
_locale.setNumberOptions(QLocale::OmitGroupSeparator | QLocale::RejectGroupSeparator); _locale.setNumberOptions(QLocale::OmitGroupSeparator | QLocale::RejectGroupSeparator);
@ -165,7 +167,7 @@ void BoblightClientConnection::handleMessage(const QString & message)
// send current color values to hyperion if this is the last led assuming leds values are send in order of id // send current color values to hyperion if this is the last led assuming leds values are send in order of id
if ((ledIndex == _ledColors.size() -1) && _priority < 255) if ((ledIndex == _ledColors.size() -1) && _priority < 255)
{ {
_hyperion->setColors(_priority, _ledColors, -1, hyperion::COMP_BOBLIGHTSERVER); _hyperion->setColors(_priority, _ledColors, -1, true, hyperion::COMP_BOBLIGHTSERVER, _clientAddress);
} }
return; return;
@ -203,38 +205,33 @@ void BoblightClientConnection::handleMessage(const QString & message)
// send current color values to hyperion // send current color values to hyperion
if (_priority < 255) if (_priority < 255)
{ {
_hyperion->setColors(_priority, _ledColors, -1, hyperion::COMP_BOBLIGHTSERVER); _hyperion->setColors(_priority, _ledColors, -1, true, hyperion::COMP_BOBLIGHTSERVER, _clientAddress);
} }
return; return;
} }
} }
Debug(_log, "unknown boblight message: %s", message.toStdString().c_str()); Debug(_log, "unknown boblight message: %s", QSTRING_CSTR(message));
} }
void BoblightClientConnection::sendMessage(const std::string & message) void BoblightClientConnection::sendMessage(const QByteArray & message)
{ {
//std::cout << "send boblight message: " << message; //std::cout << "send boblight message: " << message;
_socket->write(message.c_str(), message.size()); _socket->write(message);
}
void BoblightClientConnection::sendMessage(const char * message, int size)
{
//std::cout << "send boblight message: " << std::string(message, size);
_socket->write(message, size);
} }
void BoblightClientConnection::sendLightMessage() void BoblightClientConnection::sendLightMessage()
{ {
char buffer[256]; char buffer[256];
int n = snprintf(buffer, sizeof(buffer), "lights %d\n", _hyperion->getLedCount());
sendMessage(buffer, n);
int n = snprintf(buffer, sizeof(buffer), "lights %d\n", _hyperion->getLedCount());
sendMessage(QByteArray(buffer, n));
double h0, h1, v0, v1;
for (unsigned i = 0; i < _hyperion->getLedCount(); ++i) for (unsigned i = 0; i < _hyperion->getLedCount(); ++i)
{ {
double h0, h1, v0, v1;
_imageProcessor->getScanParameters(i, h0, h1, v0, v1); _imageProcessor->getScanParameters(i, h0, h1, v0, v1);
n = snprintf(buffer, sizeof(buffer), "light %03d scan %f %f %f %f\n", i, 100*v0, 100*v1, 100*h0, 100*h1); n = snprintf(buffer, sizeof(buffer), "light %03d scan %f %f %f %f\n", i, 100*v0, 100*v1, 100*h0, 100*h1);
sendMessage(buffer, n); sendMessage(QByteArray(buffer, n));
} }
} }

View File

@ -1,8 +1,5 @@
#pragma once #pragma once
// stl includes
#include <string>
// Qt includes // Qt includes
#include <QByteArray> #include <QByteArray>
#include <QTcpSocket> #include <QTcpSocket>
@ -65,15 +62,7 @@ private:
/// ///
/// @param message The boblight message to send /// @param message The boblight message to send
/// ///
void sendMessage(const std::string &message); void sendMessage(const QByteArray &message);
///
/// Send a message to the connected client
///
/// @param message The boblight message to send
/// @param size The size of the message
///
void sendMessage(const char * message, int size);
/// ///
/// Send a lights message the to connected client /// Send a lights message the to connected client
@ -104,4 +93,7 @@ private:
/// logger instance /// logger instance
Logger * _log; Logger * _log;
/// address of client
QString _clientAddress;
}; };

View File

@ -66,7 +66,7 @@ void Effect::registerHyperionExtensionModule()
PyImport_AppendInittab("hyperion", &PyInit_hyperion); PyImport_AppendInittab("hyperion", &PyInit_hyperion);
} }
Effect::Effect(PyThreadState * mainThreadState, int priority, int timeout, const QString & script, const QString & name, const QJsonObject & args) Effect::Effect(PyThreadState * mainThreadState, int priority, int timeout, const QString & script, const QString & name, const QJsonObject & args, const QString & origin)
: QThread() : QThread()
, _mainThreadState(mainThreadState) , _mainThreadState(mainThreadState)
, _priority(priority) , _priority(priority)
@ -79,6 +79,7 @@ Effect::Effect(PyThreadState * mainThreadState, int priority, int timeout, const
, _abortRequested(false) , _abortRequested(false)
, _imageProcessor(ImageProcessorFactory::getInstance().newImageProcessor()) , _imageProcessor(ImageProcessorFactory::getInstance().newImageProcessor())
, _colors() , _colors()
, _origin(origin)
{ {
_colors.resize(_imageProcessor->getLedCount(), ColorRgb::BLACK); _colors.resize(_imageProcessor->getLedCount(), ColorRgb::BLACK);
@ -146,7 +147,7 @@ void Effect::run()
} }
else else
{ {
Error(Logger::getInstance("EFFECTENGINE"), "Unable to open script file %s", _script.toUtf8().constData()); Error(Logger::getInstance("EFFECTENGINE"), "Unable to open script file %s.", QSTRING_CSTR(_script));
} }
file.close(); file.close();
@ -265,7 +266,7 @@ PyObject* Effect::wrapSetColor(PyObject *self, PyObject *args)
if (PyArg_ParseTuple(args, "bbb", &color.red, &color.green, &color.blue)) if (PyArg_ParseTuple(args, "bbb", &color.red, &color.green, &color.blue))
{ {
std::fill(effect->_colors.begin(), effect->_colors.end(), color); std::fill(effect->_colors.begin(), effect->_colors.end(), color);
effect->setColors(effect->_priority, effect->_colors, timeout, false, hyperion::COMP_EFFECT); effect->setColors(effect->_priority, effect->_colors, timeout, false, hyperion::COMP_EFFECT, effect->_origin);
return Py_BuildValue(""); return Py_BuildValue("");
} }
else else
@ -286,7 +287,7 @@ PyObject* Effect::wrapSetColor(PyObject *self, PyObject *args)
{ {
char * data = PyByteArray_AS_STRING(bytearray); char * data = PyByteArray_AS_STRING(bytearray);
memcpy(effect->_colors.data(), data, length); memcpy(effect->_colors.data(), data, length);
effect->setColors(effect->_priority, effect->_colors, timeout, false, hyperion::COMP_EFFECT); effect->setColors(effect->_priority, effect->_colors, timeout, false, hyperion::COMP_EFFECT, effect->_origin);
return Py_BuildValue(""); return Py_BuildValue("");
} }
else else
@ -356,7 +357,7 @@ PyObject* Effect::wrapSetImage(PyObject *self, PyObject *args)
memcpy(image.memptr(), data, length); memcpy(image.memptr(), data, length);
effect->_imageProcessor->process(image, effect->_colors); effect->_imageProcessor->process(image, effect->_colors);
effect->setColors(effect->_priority, effect->_colors, timeout, false, hyperion::COMP_EFFECT); effect->setColors(effect->_priority, effect->_colors, timeout, false, hyperion::COMP_EFFECT, effect->_origin);
return Py_BuildValue(""); return Py_BuildValue("");
} }
else else
@ -432,7 +433,7 @@ PyObject* Effect::wrapImageShow(PyObject *self, PyObject *args)
memcpy(image.memptr(), binaryImage.data(), binaryImage.size()); memcpy(image.memptr(), binaryImage.data(), binaryImage.size());
effect->_imageProcessor->process(image, effect->_colors); effect->_imageProcessor->process(image, effect->_colors);
effect->setColors(effect->_priority, effect->_colors, timeout, false, hyperion::COMP_EFFECT); effect->setColors(effect->_priority, effect->_colors, timeout, false, hyperion::COMP_EFFECT, effect->_origin);
return Py_BuildValue(""); return Py_BuildValue("");
} }

View File

@ -18,7 +18,7 @@ class Effect : public QThread
Q_OBJECT Q_OBJECT
public: public:
Effect(PyThreadState * mainThreadState, int priority, int timeout, const QString & script, const QString & name, const QJsonObject & args = QJsonObject()); Effect(PyThreadState * mainThreadState, int priority, int timeout, const QString & script, const QString & name, const QJsonObject & args = QJsonObject(), const QString & origin="System");
virtual ~Effect(); virtual ~Effect();
virtual void run(); virtual void run();
@ -43,7 +43,7 @@ public slots:
signals: signals:
void effectFinished(Effect * effect); void effectFinished(Effect * effect);
void setColors(int priority, const std::vector<ColorRgb> &ledColors, const int timeout_ms, bool clearEffects, hyperion::Components component); void setColors(int priority, const std::vector<ColorRgb> &ledColors, const int timeout_ms, bool clearEffects, hyperion::Components componentconst, QString origin);
private slots: private slots:
void effectFinished(); void effectFinished();
@ -102,5 +102,6 @@ private:
QImage * _image; QImage * _image;
QPainter * _painter; QPainter * _painter;
QString _origin;
}; };

View File

@ -342,11 +342,16 @@ void EffectEngine::readEffects()
} }
} }
int EffectEngine::runEffect(const QString &effectName, const QJsonObject &args, int priority, int timeout, QString pythonScript) int EffectEngine::runEffect(const QString &effectName, int priority, int timeout, const QString &origin)
{
return runEffect(effectName, QJsonObject(), priority, timeout, "", origin);
}
int EffectEngine::runEffect(const QString &effectName, const QJsonObject &args, int priority, int timeout, const QString &pythonScript, const QString &origin)
{ {
Info( _log, "run effect %s on channel %d", effectName.toUtf8().constData(), priority); Info( _log, "run effect %s on channel %d", effectName.toUtf8().constData(), priority);
if (pythonScript == "") if (pythonScript.isEmpty())
{ {
const EffectDefinition * effectDefinition = nullptr; const EffectDefinition * effectDefinition = nullptr;
for (const EffectDefinition & e : _availableEffects) for (const EffectDefinition & e : _availableEffects)
@ -364,24 +369,24 @@ int EffectEngine::runEffect(const QString &effectName, const QJsonObject &args,
return -1; return -1;
} }
return runEffectScript(effectDefinition->script, effectName, args.isEmpty() ? effectDefinition->args : args, priority, timeout); return runEffectScript(effectDefinition->script, effectName, (args.isEmpty() ? effectDefinition->args : args), priority, timeout, origin);
} else }
return runEffectScript(pythonScript, effectName, args, priority, timeout); return runEffectScript(pythonScript, effectName, args, priority, timeout, origin);
} }
int EffectEngine::runEffectScript(const QString &script, const QString &name, const QJsonObject &args, int priority, int timeout) int EffectEngine::runEffectScript(const QString &script, const QString &name, const QJsonObject &args, int priority, int timeout, const QString & origin)
{ {
// clear current effect on the channel // clear current effect on the channel
channelCleared(priority); channelCleared(priority);
// create the effect // create the effect
Effect * effect = new Effect(_mainThreadState, priority, timeout, script, name, args); Effect * effect = new Effect(_mainThreadState, priority, timeout, script, name, args, origin);
connect(effect, SIGNAL(setColors(int,std::vector<ColorRgb>,int,bool,hyperion::Components)), _hyperion, SLOT(setColors(int,std::vector<ColorRgb>,int,bool,hyperion::Components)), Qt::QueuedConnection); connect(effect, SIGNAL(setColors(int,std::vector<ColorRgb>,int,bool,hyperion::Components,const QString)), _hyperion, SLOT(setColors(int,std::vector<ColorRgb>,int,bool,hyperion::Components,const QString)), Qt::QueuedConnection);
connect(effect, SIGNAL(effectFinished(Effect*)), this, SLOT(effectFinished(Effect*))); connect(effect, SIGNAL(effectFinished(Effect*)), this, SLOT(effectFinished(Effect*)));
_activeEffects.push_back(effect); _activeEffects.push_back(effect);
// start the effect // start the effect
_hyperion->registerPriority(name.toStdString(), priority); _hyperion->registerPriority(name, priority);
effect->start(); effect->start();
return 0; return 0;
@ -426,5 +431,5 @@ void EffectEngine::effectFinished(Effect *effect)
// cleanup the effect // cleanup the effect
effect->deleteLater(); effect->deleteLater();
_hyperion->unRegisterPriority(effect->getName().toStdString()); _hyperion->unRegisterPriority(effect->getName());
} }

View File

@ -4,7 +4,6 @@
#include <hyperion/GrabberWrapper.h> #include <hyperion/GrabberWrapper.h>
#include <HyperionConfig.h> #include <HyperionConfig.h>
#define QSTRING_CSTR(str) str.toLocal8Bit().constData()
GrabberWrapper::GrabberWrapper(QString grabberName, const int priority, hyperion::Components grabberComponentId) GrabberWrapper::GrabberWrapper(QString grabberName, const int priority, hyperion::Components grabberComponentId)
: _grabberName(grabberName) : _grabberName(grabberName)
, _hyperion(Hyperion::getInstance()) , _hyperion(Hyperion::getInstance())
@ -37,7 +36,7 @@ bool GrabberWrapper::start()
{ {
// Start the timer with the pre configured interval // Start the timer with the pre configured interval
_timer.start(); _timer.start();
_hyperion->registerPriority(_grabberName.toStdString(), _priority); _hyperion->registerPriority(_grabberName, _priority);
return _timer.isActive(); return _timer.isActive();
} }
@ -46,7 +45,7 @@ void GrabberWrapper::stop()
{ {
// Stop the timer, effectivly stopping the process // Stop the timer, effectivly stopping the process
_timer.stop(); _timer.stop();
_hyperion->unRegisterPriority(_grabberName.toStdString()); _hyperion->unRegisterPriority(_grabberName);
} }
void GrabberWrapper::componentStateChanged(const hyperion::Components component, bool enable) void GrabberWrapper::componentStateChanged(const hyperion::Components component, bool enable)

View File

@ -504,22 +504,22 @@ bool Hyperion::configWriteable()
} }
void Hyperion::registerPriority(const std::string name, const int priority) void Hyperion::registerPriority(const QString &name, const int priority/*, const QString &origin*/)
{ {
Info(_log, "Register new input source named '%s' for priority channel '%d'", name.c_str(), priority ); Info(_log, "Register new input source named '%s' for priority channel '%d'", QSTRING_CSTR(name), priority );
for(auto const &entry : _priorityRegister) for(auto const &entry : _priorityRegister)
{ {
WarningIf( ( entry.first != name && entry.second == priority), _log, WarningIf( ( entry.first != name && entry.second == priority), _log,
"Input source '%s' uses same priority channel (%d) as '%s'.", name.c_str(), priority, entry.first.c_str()); "Input source '%s' uses same priority channel (%d) as '%s'.", QSTRING_CSTR(name), priority, QSTRING_CSTR(entry.first));
} }
_priorityRegister.emplace(name,priority); _priorityRegister.emplace(name,priority);
} }
void Hyperion::unRegisterPriority(const std::string name) void Hyperion::unRegisterPriority(const QString &name)
{ {
Info(_log, "Unregister input source named '%s' from priority register", name.c_str()); Info(_log, "Unregister input source named '%s' from priority register", QSTRING_CSTR(name));
_priorityRegister.erase(name); _priorityRegister.erase(name);
} }
@ -688,14 +688,14 @@ const std::list<EffectSchema> & Hyperion::getEffectSchemas()
return _effectEngine->getEffectSchemas(); return _effectEngine->getEffectSchemas();
} }
int Hyperion::setEffect(const QString &effectName, int priority, int timeout) int Hyperion::setEffect(const QString &effectName, int priority, int timeout, const QString & origin)
{ {
return _effectEngine->runEffect(effectName, priority, timeout); return _effectEngine->runEffect(effectName, priority, timeout, origin);
} }
int Hyperion::setEffect(const QString &effectName, const QJsonObject &args, int priority, int timeout, QString pythonScript) int Hyperion::setEffect(const QString &effectName, const QJsonObject &args, int priority, int timeout, const QString & pythonScript, const QString & origin)
{ {
return _effectEngine->runEffect(effectName, args, priority, timeout, pythonScript); return _effectEngine->runEffect(effectName, args, priority, timeout, pythonScript, origin);
} }
void Hyperion::setLedMappingType(int mappingType) void Hyperion::setLedMappingType(int mappingType)

View File

@ -27,6 +27,7 @@
#include <QByteArray> #include <QByteArray>
#include <QIODevice> #include <QIODevice>
#include <QDateTime> #include <QDateTime>
#include <QHostInfo>
// hyperion util includes // hyperion util includes
#include <hyperion/ImageProcessorFactory.h> #include <hyperion/ImageProcessorFactory.h>
@ -57,6 +58,7 @@ JsonClientConnection::JsonClientConnection(QTcpSocket *socket)
, _forwarder_enabled(true) , _forwarder_enabled(true)
, _streaming_logging_activated(false) , _streaming_logging_activated(false)
, _image_stream_timeout(0) , _image_stream_timeout(0)
, _clientAddress(socket->peerAddress())
{ {
// connect internal signals and slots // connect internal signals and slots
connect(_socket, SIGNAL(disconnected()), this, SLOT(socketClosed())); connect(_socket, SIGNAL(disconnected()), this, SLOT(socketClosed()));
@ -365,7 +367,7 @@ void JsonClientConnection::handleColorCommand(const QJsonObject& message, const
// extract parameters // extract parameters
int priority = message["priority"].toInt(); int priority = message["priority"].toInt();
int duration = message["duration"].toInt(-1); int duration = message["duration"].toInt(-1);
QString origin = message["origin"].toString(); QString origin = message["origin"].toString() + "@"+QHostInfo::fromName(_clientAddress.toString()).hostName();
std::vector<ColorRgb> colorData(_hyperion->getLedCount()); std::vector<ColorRgb> colorData(_hyperion->getLedCount());
const QJsonArray & jsonColor = message["color"].toArray(); const QJsonArray & jsonColor = message["color"].toArray();
@ -438,18 +440,19 @@ void JsonClientConnection::handleEffectCommand(const QJsonObject& message, const
// extract parameters // extract parameters
int priority = message["priority"].toInt(); int priority = message["priority"].toInt();
int duration = message["duration"].toInt(-1); int duration = message["duration"].toInt(-1);
QString pythonScript = message["pythonScript"].toString(""); QString pythonScript = message["pythonScript"].toString();
QString origin = message["origin"].toString() + "@"+_clientAddress.toString();
const QJsonObject & effect = message["effect"].toObject(); const QJsonObject & effect = message["effect"].toObject();
const QString & effectName = effect["name"].toString(); const QString & effectName = effect["name"].toString();
// set output // set output
if (effect.contains("args")) if (effect.contains("args"))
{ {
_hyperion->setEffect(effectName, effect["args"].toObject(), priority, duration, pythonScript); _hyperion->setEffect(effectName, effect["args"].toObject(), priority, duration, pythonScript, origin);
} }
else else
{ {
_hyperion->setEffect(effectName, priority, duration); _hyperion->setEffect(effectName, priority, duration, origin);
} }
// send reply // send reply
@ -607,7 +610,7 @@ void JsonClientConnection::handleServerInfoCommand(const QJsonObject&, const QSt
{ {
if (entry.second == priority) if (entry.second == priority)
{ {
item["owner"] = QString::fromStdString(entry.first); item["owner"] = entry.first;
priorityRegister.erase(entry.first); priorityRegister.erase(entry.first);
break; break;
} }
@ -665,7 +668,7 @@ void JsonClientConnection::handleServerInfoCommand(const QJsonObject&, const QSt
item["priority"] = entry.second; item["priority"] = entry.second;
item["active"] = false; item["active"] = false;
item["visible"] = false; item["visible"] = false;
item["owner"] = QString::fromStdString(entry.first); item["owner"] = entry.first;
priorities.append(item); priorities.append(item);
} }

View File

@ -7,6 +7,7 @@
#include <QByteArray> #include <QByteArray>
#include <QTcpSocket> #include <QTcpSocket>
#include <QMutex> #include <QMutex>
#include <QHostAddress>
// Hyperion includes // Hyperion includes
#include <hyperion/Hyperion.h> #include <hyperion/Hyperion.h>
@ -349,6 +350,9 @@ private:
/// timeout for live video refresh /// timeout for live video refresh
volatile qint64 _image_stream_timeout; volatile qint64 _image_stream_timeout;
/// address of client
QHostAddress _clientAddress;
// masks for fields in the basic header // masks for fields in the basic header
static uint8_t const BHB0_OPCODE = 0x0F; static uint8_t const BHB0_OPCODE = 0x0F;
static uint8_t const BHB0_RSV3 = 0x10; static uint8_t const BHB0_RSV3 = 0x10;

View File

@ -16,6 +16,10 @@
"maximum" : 253, "maximum" : 253,
"required": true "required": true
}, },
"origin": {
"type": "string",
"required": true
},
"duration": { "duration": {
"type": "integer", "type": "integer",
"required": false "required": false

View File

@ -11,6 +11,7 @@
#include <QRgb> #include <QRgb>
#include <QResource> #include <QResource>
#include <QDateTime> #include <QDateTime>
#include <QHostInfo>
// hyperion util includes // hyperion util includes
#include "hyperion/ImageProcessorFactory.h" #include "hyperion/ImageProcessorFactory.h"
@ -27,13 +28,13 @@ ProtoClientConnection::ProtoClientConnection(QTcpSocket *socket)
, _hyperion(Hyperion::getInstance()) , _hyperion(Hyperion::getInstance())
, _receiveBuffer() , _receiveBuffer()
, _priority(-1) , _priority(-1)
, _priorityChannelName("Proto-Server")
, _clientAddress(QHostInfo::fromName(socket->peerAddress().toString()).hostName())
{ {
// connect internal signals and slots // connect internal signals and slots
connect(_socket, SIGNAL(disconnected()), this, SLOT(socketClosed())); connect(_socket, SIGNAL(disconnected()), this, SLOT(socketClosed()));
connect(_socket, SIGNAL(readyRead()), this, SLOT(readData())); connect(_socket, SIGNAL(readyRead()), this, SLOT(readData()));
connect(_hyperion, SIGNAL(imageToLedsMappingChanged(int)), _imageProcessor, SLOT(setLedMappingType(int))); connect(_hyperion, SIGNAL(imageToLedsMappingChanged(int)), _imageProcessor, SLOT(setLedMappingType(int)));
_priorityChannelName = "proto@"+ _socket->peerAddress().toString().toStdString();
} }
ProtoClientConnection::~ProtoClientConnection() ProtoClientConnection::~ProtoClientConnection()
@ -198,7 +199,7 @@ void ProtoClientConnection::handleImageCommand(const proto::ImageRequest &messag
// process the image // process the image
std::vector<ColorRgb> ledColors = _imageProcessor->process(image); std::vector<ColorRgb> ledColors = _imageProcessor->process(image);
_hyperion->setColors(_priority, ledColors, duration); _hyperion->setColors(_priority, ledColors, duration, true, hyperion::COMP_PROTOSERVER , "proto@"+_clientAddress);
_hyperion->setImage(_priority, image, duration); _hyperion->setImage(_priority, image, duration);
// send reply // send reply

View File

@ -7,6 +7,7 @@
#include <QByteArray> #include <QByteArray>
#include <QTcpSocket> #include <QTcpSocket>
#include <QStringList> #include <QStringList>
#include <QString>
// Hyperion includes // Hyperion includes
#include <hyperion/Hyperion.h> #include <hyperion/Hyperion.h>
@ -140,5 +141,8 @@ private:
int _priority; int _priority;
std::string _priorityChannelName; QString _priorityChannelName;
/// address of client
QString _clientAddress;
}; };

View File

@ -109,13 +109,13 @@ void UDPListener::readPendingDatagrams()
_server->readDatagram(datagram.data(), datagram.size(), &sender, &senderPort); _server->readDatagram(datagram.data(), datagram.size(), &sender, &senderPort);
processTheDatagram(&datagram); processTheDatagram(&datagram, &sender);
} }
} }
void UDPListener::processTheDatagram(const QByteArray * datagram) void UDPListener::processTheDatagram(const QByteArray * datagram, const QHostAddress * sender)
{ {
int packetLedCount = datagram->size()/3; int packetLedCount = datagram->size()/3;
int hyperionLedCount = Hyperion::getInstance()->getLedCount(); int hyperionLedCount = Hyperion::getInstance()->getLedCount();
@ -130,6 +130,8 @@ void UDPListener::processTheDatagram(const QByteArray * datagram)
rgb.blue = datagram->at(ledIndex*3+2); rgb.blue = datagram->at(ledIndex*3+2);
} }
_hyperion->setColors(_priority, _ledColors, _timeout, -1, hyperion::COMP_UDPLISTENER); _hyperion->setColors(_priority, _ledColors, _timeout, -1, hyperion::COMP_UDPLISTENER, sender->toString());
} }

View File

@ -25,6 +25,7 @@ SET(Utils_HEADERS
${CURRENT_HEADER_DIR}/RgbToRgbw.h ${CURRENT_HEADER_DIR}/RgbToRgbw.h
${CURRENT_HEADER_DIR}/jsonschema/QJsonFactory.h ${CURRENT_HEADER_DIR}/jsonschema/QJsonFactory.h
${CURRENT_HEADER_DIR}/jsonschema/QJsonSchemaChecker.h ${CURRENT_HEADER_DIR}/jsonschema/QJsonSchemaChecker.h
${CURRENT_HEADER_DIR}/global_defines.h
) )
SET(Utils_SOURCES SET(Utils_SOURCES

View File

@ -18,6 +18,7 @@ LoggerManager* LoggerManager::_instance = nullptr;
Logger* Logger::getInstance(QString name, Logger::LogLevel minLevel) Logger* Logger::getInstance(QString name, Logger::LogLevel minLevel)
{ {
qRegisterMetaType<Logger::T_LOG_MESSAGE>();
std::string loggerName = name.toStdString(); std::string loggerName = name.toStdString();
Logger* log = nullptr; Logger* log = nullptr;
if (LoggerMap == nullptr) if (LoggerMap == nullptr)

View File

@ -9,6 +9,7 @@
#include <QJsonObject> #include <QJsonObject>
#include <QJsonArray> #include <QJsonArray>
#include <QJsonDocument> #include <QJsonDocument>
#include <QHostInfo>
// hyperion-remote includes // hyperion-remote includes
#include "JsonConnection.h" #include "JsonConnection.h"
@ -51,6 +52,7 @@ void JsonConnection::setColor(std::vector<QColor> colors, int priority, int dura
// create command // create command
QJsonObject command; QJsonObject command;
command["command"] = QString("color"); command["command"] = QString("color");
command["origin"] = QString("hyperion-remote");
command["priority"] = priority; command["priority"] = priority;
QJsonArray rgbValue; QJsonArray rgbValue;
for (const QColor & color : colors) for (const QColor & color : colors)
@ -97,6 +99,7 @@ void JsonConnection::setImage(QImage &image, int priority, int duration)
QJsonObject command; QJsonObject command;
command["command"] = QString("image"); command["command"] = QString("image");
command["priority"] = priority; command["priority"] = priority;
command["origin"] = QString("hyperion-remote");
command["imagewidth"] = image.width(); command["imagewidth"] = image.width();
command["imageheight"] = image.height(); command["imageheight"] = image.height();
command["imagedata"] = QString(base64Image.data()); command["imagedata"] = QString(base64Image.data());
@ -119,6 +122,7 @@ void JsonConnection::setEffect(const QString &effectName, const QString & effect
// create command // create command
QJsonObject command, effect; QJsonObject command, effect;
command["command"] = QString("effect"); command["command"] = QString("effect");
command["origin"] = QString("hyperion-remote");
command["priority"] = priority; command["priority"] = priority;
effect["name"] = effectName; effect["name"] = effectName;