mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2023-10-10 13:36:59 +02:00
89afedf3d5
* extend image effects add a basic light clock * add alpha values to rainbow * some effect tuning * make seconds hand optional and change color to yellow make blue hours hand always visible
661 lines
16 KiB
C++
661 lines
16 KiB
C++
// Python include
|
|
#include <Python.h>
|
|
|
|
// stl includes
|
|
#include <iostream>
|
|
#include <sstream>
|
|
|
|
// Qt includes
|
|
#include <QDateTime>
|
|
#include <QFile>
|
|
#include <Qt>
|
|
#include <QLinearGradient>
|
|
#include <QConicalGradient>
|
|
#include <QRadialGradient>
|
|
#include <QRect>
|
|
|
|
// effect engin eincludes
|
|
#include "Effect.h"
|
|
#include <utils/Logger.h>
|
|
#include <hyperion/Hyperion.h>
|
|
|
|
// Python method table
|
|
PyMethodDef Effect::effectMethods[] = {
|
|
{"setColor", Effect::wrapSetColor, METH_VARARGS, "Set a new color for the leds."},
|
|
{"setImage", Effect::wrapSetImage, METH_VARARGS, "Set a new image to process and determine new led colors."},
|
|
{"abort", Effect::wrapAbort, METH_NOARGS, "Check if the effect should abort execution."},
|
|
{"imageShow", Effect::wrapImageShow, METH_NOARGS, "set current effect image to hyperion core."},
|
|
{"imageCanonicalGradient", Effect::wrapImageCanonicalGradient, METH_VARARGS, ""},
|
|
{"imageRadialGradient" , Effect::wrapImageRadialGradient, METH_VARARGS, ""},
|
|
{"imageSolidFill" , Effect::wrapImageSolidFill, METH_VARARGS, ""},
|
|
// {"imageSetPixel",Effect::wrapImageShow, METH_VARARGS, "set pixel color of image"},
|
|
// {"imageGetPixel",Effect::wrapImageShow, METH_VARARGS, "get pixel color of image"},
|
|
{NULL, NULL, 0, NULL}
|
|
};
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
// create the hyperion module
|
|
struct PyModuleDef Effect::moduleDef = {
|
|
PyModuleDef_HEAD_INIT,
|
|
"hyperion", /* m_name */
|
|
"Hyperion module", /* m_doc */
|
|
-1, /* m_size */
|
|
Effect::effectMethods, /* m_methods */
|
|
NULL, /* m_reload */
|
|
NULL, /* m_traverse */
|
|
NULL, /* m_clear */
|
|
NULL, /* m_free */
|
|
};
|
|
|
|
PyObject* Effect::PyInit_hyperion()
|
|
{
|
|
return PyModule_Create(&moduleDef);
|
|
}
|
|
#else
|
|
void Effect::PyInit_hyperion()
|
|
{
|
|
Py_InitModule("hyperion", effectMethods);
|
|
}
|
|
#endif
|
|
|
|
void Effect::registerHyperionExtensionModule()
|
|
{
|
|
PyImport_AppendInittab("hyperion", &PyInit_hyperion);
|
|
}
|
|
|
|
Effect::Effect(PyThreadState * mainThreadState, int priority, int timeout, const QString & script, const QString & name, const Json::Value & args)
|
|
: QThread()
|
|
, _mainThreadState(mainThreadState)
|
|
, _priority(priority)
|
|
, _timeout(timeout)
|
|
, _script(script)
|
|
, _name(name)
|
|
, _args(args)
|
|
, _endTime(-1)
|
|
, _interpreterThreadState(nullptr)
|
|
, _abortRequested(false)
|
|
, _imageProcessor(ImageProcessorFactory::getInstance().newImageProcessor())
|
|
, _colors()
|
|
{
|
|
_colors.resize(_imageProcessor->getLedCount(), ColorRgb::BLACK);
|
|
|
|
// disable the black border detector for effects
|
|
_imageProcessor->enableBlackBorderDetector(false);
|
|
|
|
// init effect image for image based effects, size is based on led layout
|
|
_imageSize = Hyperion::getInstance()->getLedGridSize();
|
|
_image = new QImage(_imageSize, QImage::Format_ARGB32_Premultiplied);
|
|
_image->fill(Qt::black);
|
|
_painter = new QPainter(_image);
|
|
|
|
// connect the finished signal
|
|
connect(this, SIGNAL(finished()), this, SLOT(effectFinished()));
|
|
Q_INIT_RESOURCE(EffectEngine);
|
|
}
|
|
|
|
Effect::~Effect()
|
|
{
|
|
delete _painter;
|
|
delete _image;
|
|
}
|
|
|
|
void Effect::run()
|
|
{
|
|
// switch to the main thread state and acquire the GIL
|
|
PyEval_RestoreThread(_mainThreadState);
|
|
|
|
// Initialize a new thread state
|
|
_interpreterThreadState = Py_NewInterpreter();
|
|
|
|
// import the buildtin Hyperion module
|
|
PyObject * module = PyImport_ImportModule("hyperion");
|
|
|
|
// add a capsule containing 'this' to the module to be able to retrieve the effect from the callback function
|
|
PyObject_SetAttrString(module, "__effectObj", PyCapsule_New(this, nullptr, nullptr));
|
|
|
|
// add ledCount variable to the interpreter
|
|
PyObject_SetAttrString(module, "ledCount", Py_BuildValue("i", _imageProcessor->getLedCount()));
|
|
|
|
// add imageWidth variable to the interpreter
|
|
PyObject_SetAttrString(module, "imageWidth", Py_BuildValue("i", _imageSize.width()));
|
|
|
|
// add imageHeight variable to the interpreter
|
|
PyObject_SetAttrString(module, "imageHeight", Py_BuildValue("i", _imageSize.height()));
|
|
|
|
// add a args variable to the interpreter
|
|
PyObject_SetAttrString(module, "args", json2python(_args));
|
|
|
|
// decref the module
|
|
Py_XDECREF(module);
|
|
|
|
// Set the end time if applicable
|
|
if (_timeout > 0)
|
|
{
|
|
_endTime = QDateTime::currentMSecsSinceEpoch() + _timeout;
|
|
}
|
|
|
|
// Run the effect script
|
|
QFile file (_script);
|
|
QByteArray python_code;
|
|
if (file.open(QIODevice::ReadOnly))
|
|
{
|
|
python_code = file.readAll();
|
|
}
|
|
else
|
|
{
|
|
Error(Logger::getInstance("EFFECTENGINE"), "Unable to open script file %s", _script.toUtf8().constData());
|
|
}
|
|
file.close();
|
|
|
|
if (!python_code.isEmpty())
|
|
{
|
|
PyRun_SimpleString(python_code.constData());
|
|
}
|
|
|
|
// Clean up the thread state
|
|
Py_EndInterpreter(_interpreterThreadState);
|
|
_interpreterThreadState = nullptr;
|
|
PyEval_ReleaseLock();
|
|
}
|
|
|
|
int Effect::getPriority() const
|
|
{
|
|
return _priority;
|
|
}
|
|
|
|
bool Effect::isAbortRequested() const
|
|
{
|
|
return _abortRequested;
|
|
}
|
|
|
|
void Effect::abort()
|
|
{
|
|
_abortRequested = true;
|
|
}
|
|
|
|
void Effect::effectFinished()
|
|
{
|
|
emit effectFinished(this);
|
|
}
|
|
|
|
PyObject *Effect::json2python(const Json::Value &json) const
|
|
{
|
|
switch (json.type())
|
|
{
|
|
case Json::nullValue:
|
|
return Py_BuildValue("");
|
|
case Json::realValue:
|
|
return Py_BuildValue("d", json.asDouble());
|
|
case Json::intValue:
|
|
case Json::uintValue:
|
|
return Py_BuildValue("i", json.asInt());
|
|
case Json::booleanValue:
|
|
return Py_BuildValue("i", json.asBool() ? 1 : 0);
|
|
case Json::stringValue:
|
|
return Py_BuildValue("s", json.asCString());
|
|
case Json::objectValue:
|
|
{
|
|
PyObject * dict= PyDict_New();
|
|
for (Json::Value::iterator i = json.begin(); i != json.end(); ++i)
|
|
{
|
|
PyObject * obj = json2python(*i);
|
|
PyDict_SetItemString(dict, i.memberName(), obj);
|
|
Py_XDECREF(obj);
|
|
}
|
|
return dict;
|
|
}
|
|
case Json::arrayValue:
|
|
{
|
|
PyObject * list = PyList_New(json.size());
|
|
for (Json::Value::iterator i = json.begin(); i != json.end(); ++i)
|
|
{
|
|
PyObject * obj = json2python(*i);
|
|
PyList_SetItem(list, i.index(), obj);
|
|
Py_XDECREF(obj);
|
|
}
|
|
return list;
|
|
}
|
|
}
|
|
|
|
assert(false);
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* Effect::wrapSetColor(PyObject *self, PyObject *args)
|
|
{
|
|
// get the effect
|
|
Effect * effect = getEffect();
|
|
|
|
// check if we have aborted already
|
|
if (effect->_abortRequested)
|
|
{
|
|
return Py_BuildValue("");
|
|
}
|
|
|
|
// determine the timeout
|
|
int timeout = effect->_timeout;
|
|
if (timeout > 0)
|
|
{
|
|
timeout = effect->_endTime - QDateTime::currentMSecsSinceEpoch();
|
|
|
|
// we are done if the time has passed
|
|
if (timeout <= 0)
|
|
{
|
|
return Py_BuildValue("");
|
|
}
|
|
}
|
|
|
|
// check the number of arguments
|
|
int argCount = PyTuple_Size(args);
|
|
if (argCount == 3)
|
|
{
|
|
// three seperate arguments for red, green, and blue
|
|
ColorRgb color;
|
|
if (PyArg_ParseTuple(args, "bbb", &color.red, &color.green, &color.blue))
|
|
{
|
|
std::fill(effect->_colors.begin(), effect->_colors.end(), color);
|
|
effect->setColors(effect->_priority, effect->_colors, timeout, false);
|
|
return Py_BuildValue("");
|
|
}
|
|
else
|
|
{
|
|
return nullptr;
|
|
}
|
|
}
|
|
else if (argCount == 1)
|
|
{
|
|
// bytearray of values
|
|
PyObject * bytearray = nullptr;
|
|
if (PyArg_ParseTuple(args, "O", &bytearray))
|
|
{
|
|
if (PyByteArray_Check(bytearray))
|
|
{
|
|
size_t length = PyByteArray_Size(bytearray);
|
|
if (length == 3 * effect->_imageProcessor->getLedCount())
|
|
{
|
|
char * data = PyByteArray_AS_STRING(bytearray);
|
|
memcpy(effect->_colors.data(), data, length);
|
|
effect->setColors(effect->_priority, effect->_colors, timeout, false);
|
|
return Py_BuildValue("");
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Length of bytearray argument should be 3*ledCount");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Argument is not a bytearray");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Function expect 1 or 3 arguments");
|
|
return nullptr;
|
|
}
|
|
|
|
// error
|
|
PyErr_SetString(PyExc_RuntimeError, "Unknown error");
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* Effect::wrapSetImage(PyObject *self, PyObject *args)
|
|
{
|
|
// get the effect
|
|
Effect * effect = getEffect();
|
|
|
|
// check if we have aborted already
|
|
if (effect->_abortRequested)
|
|
{
|
|
return Py_BuildValue("");
|
|
}
|
|
|
|
// determine the timeout
|
|
int timeout = effect->_timeout;
|
|
if (timeout > 0)
|
|
{
|
|
timeout = effect->_endTime - QDateTime::currentMSecsSinceEpoch();
|
|
|
|
// we are done if the time has passed
|
|
if (timeout <= 0)
|
|
{
|
|
return Py_BuildValue("");
|
|
}
|
|
}
|
|
|
|
// bytearray of values
|
|
int width, height;
|
|
PyObject * bytearray = nullptr;
|
|
if (PyArg_ParseTuple(args, "iiO", &width, &height, &bytearray))
|
|
{
|
|
if (PyByteArray_Check(bytearray))
|
|
{
|
|
int length = PyByteArray_Size(bytearray);
|
|
if (length == 3 * width * height)
|
|
{
|
|
Image<ColorRgb> image(width, height);
|
|
char * data = PyByteArray_AS_STRING(bytearray);
|
|
memcpy(image.memptr(), data, length);
|
|
|
|
effect->_imageProcessor->process(image, effect->_colors);
|
|
effect->setColors(effect->_priority, effect->_colors, timeout, false);
|
|
return Py_BuildValue("");
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Length of bytearray argument should be 3*width*height");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Argument 3 is not a bytearray");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
// error
|
|
PyErr_SetString(PyExc_RuntimeError, "Unknown error");
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* Effect::wrapAbort(PyObject *self, PyObject *)
|
|
{
|
|
Effect * effect = getEffect();
|
|
|
|
// Test if the effect has reached it end time
|
|
if (effect->_timeout > 0 && QDateTime::currentMSecsSinceEpoch() > effect->_endTime)
|
|
{
|
|
effect->_abortRequested = true;
|
|
}
|
|
|
|
return Py_BuildValue("i", effect->_abortRequested ? 1 : 0);
|
|
}
|
|
|
|
|
|
PyObject* Effect::wrapImageShow(PyObject *self, PyObject *args)
|
|
{
|
|
Effect * effect = getEffect();
|
|
|
|
// determine the timeout
|
|
int timeout = effect->_timeout;
|
|
if (timeout > 0)
|
|
{
|
|
timeout = effect->_endTime - QDateTime::currentMSecsSinceEpoch();
|
|
|
|
// we are done if the time has passed
|
|
if (timeout <= 0)
|
|
{
|
|
return Py_BuildValue("");
|
|
}
|
|
}
|
|
|
|
int width = effect->_imageSize.width();
|
|
int height = effect->_imageSize.height();
|
|
QImage * qimage = effect->_image;
|
|
|
|
Image<ColorRgb> image(width, height);
|
|
QByteArray binaryImage;
|
|
|
|
for (int i = 0; i <qimage->height(); ++i)
|
|
{
|
|
const QRgb * scanline = reinterpret_cast<const QRgb *>(qimage->scanLine(i));
|
|
for (int j = 0; j < qimage->width(); ++j)
|
|
{
|
|
binaryImage.append((char) qRed(scanline[j]));
|
|
binaryImage.append((char) qGreen(scanline[j]));
|
|
binaryImage.append((char) qBlue(scanline[j]));
|
|
}
|
|
}
|
|
|
|
memcpy(image.memptr(), binaryImage.data(), binaryImage.size());
|
|
effect->_imageProcessor->process(image, effect->_colors);
|
|
effect->setColors(effect->_priority, effect->_colors, timeout, false);
|
|
|
|
return Py_BuildValue("");
|
|
}
|
|
|
|
|
|
PyObject* Effect::wrapImageCanonicalGradient(PyObject *self, PyObject *args)
|
|
{
|
|
Effect * effect = getEffect();
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
PyObject * bytearray = nullptr;
|
|
int centerX, centerY, angle;
|
|
int startX = 0;
|
|
int startY = 0;
|
|
int width = effect->_imageSize.width();
|
|
int height = effect->_imageSize.height();
|
|
|
|
bool argsOK = false;
|
|
|
|
if ( argCount == 8 && PyArg_ParseTuple(args, "iiiiiiiO", &startX, &startY, &width, &height, ¢erX, ¢erY, &angle, &bytearray) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 4 && PyArg_ParseTuple(args, "iiiO", ¢erX, ¢erY, &angle, &bytearray) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
angle = std::max(std::min(angle,360),0);
|
|
|
|
if (argsOK)
|
|
{
|
|
if (PyByteArray_Check(bytearray))
|
|
{
|
|
const int length = PyByteArray_Size(bytearray);
|
|
const unsigned arrayItemLength = 5;
|
|
if (length % arrayItemLength == 0)
|
|
{
|
|
|
|
QPainter * painter = effect->_painter;
|
|
QRect myQRect(startX,startY,width,height);
|
|
QConicalGradient gradient(QPoint(centerX,centerY), angle );
|
|
char * data = PyByteArray_AS_STRING(bytearray);
|
|
|
|
for (int idx=0; idx<length; idx+=arrayItemLength)
|
|
{
|
|
gradient.setColorAt(
|
|
((uint8_t)data[idx])/255.0,
|
|
QColor(
|
|
(uint8_t)(data[idx+1]),
|
|
(uint8_t)(data[idx+2]),
|
|
(uint8_t)(data[idx+3]),
|
|
(uint8_t)(data[idx+4])
|
|
));
|
|
}
|
|
|
|
painter->fillRect(myQRect, gradient);
|
|
|
|
return Py_BuildValue("");
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Length of bytearray argument should multiple of 5");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Argument 8 is not a bytearray");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
|
|
PyObject* Effect::wrapImageRadialGradient(PyObject *self, PyObject *args)
|
|
{
|
|
Effect * effect = getEffect();
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
PyObject * bytearray = nullptr;
|
|
int centerX, centerY, radius, focalX, focalY, focalRadius;
|
|
int startX = 0;
|
|
int startY = 0;
|
|
int width = effect->_imageSize.width();
|
|
int height = effect->_imageSize.height();
|
|
|
|
bool argsOK = false;
|
|
|
|
if ( argCount == 11 && PyArg_ParseTuple(args, "iiiiiiiiiiO", &startX, &startY, &width, &height, ¢erX, ¢erY, &radius, &focalX, &focalY, &focalRadius, &bytearray) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 8 && PyArg_ParseTuple(args, "iiiiiiiO", &startX, &startY, &width, &height, ¢erX, ¢erY, &radius, &bytearray) )
|
|
{
|
|
argsOK = true;
|
|
focalX = centerX;
|
|
focalY = centerY;
|
|
focalRadius = radius;
|
|
}
|
|
if ( argCount == 7 && PyArg_ParseTuple(args, "iiiiiiO", ¢erX, ¢erY, &radius, &focalX, &focalY, &focalRadius, &bytearray) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 4 && PyArg_ParseTuple(args, "iiiO", ¢erX, ¢erY, &radius, &bytearray) )
|
|
{
|
|
argsOK = true;
|
|
focalX = centerX;
|
|
focalY = centerY;
|
|
focalRadius = radius;
|
|
}
|
|
|
|
if (argsOK)
|
|
{
|
|
if (PyByteArray_Check(bytearray))
|
|
{
|
|
int length = PyByteArray_Size(bytearray);
|
|
if (length % 4 == 0)
|
|
{
|
|
|
|
QPainter * painter = effect->_painter;
|
|
QRect myQRect(startX,startY,width,height);
|
|
QRadialGradient gradient(QPoint(centerX,centerY), std::max(radius,0) );
|
|
char * data = PyByteArray_AS_STRING(bytearray);
|
|
|
|
for (int idx=0; idx<length; idx+=4)
|
|
{
|
|
gradient.setColorAt(
|
|
((uint8_t)data[idx])/255.0,
|
|
QColor(
|
|
(uint8_t)(data[idx+1]),
|
|
(uint8_t)(data[idx+2]),
|
|
(uint8_t)(data[idx+3])
|
|
));
|
|
}
|
|
|
|
//gradient.setSpread(QGradient::ReflectSpread);
|
|
painter->fillRect(myQRect, gradient);
|
|
|
|
return Py_BuildValue("");
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Length of bytearray argument should multiple of 4");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Last argument is not a bytearray");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
PyObject* Effect::wrapImageSolidFill(PyObject *self, PyObject *args)
|
|
{
|
|
Effect * effect = getEffect();
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int r, g, b;
|
|
int a = 255;
|
|
int startX = 0;
|
|
int startY = 0;
|
|
int width = effect->_imageSize.width();
|
|
int height = effect->_imageSize.height();
|
|
|
|
bool argsOK = false;
|
|
|
|
if ( argCount == 8 && PyArg_ParseTuple(args, "iiiiiiii", &startX, &startY, &width, &height, &r, &g, &b, &a) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 7 && PyArg_ParseTuple(args, "iiiiiii", &startX, &startY, &width, &height, &r, &g, &b) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 4 && PyArg_ParseTuple(args, "iiii",&r, &g, &b, &a) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 3 && PyArg_ParseTuple(args, "iii",&r, &g, &b) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
|
|
if (argsOK)
|
|
{
|
|
QRect myQRect(startX,startY,width,height);
|
|
effect->_painter->fillRect(myQRect, QColor(r,g,b,a));
|
|
return Py_BuildValue("");
|
|
}
|
|
else
|
|
{
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
|
|
Effect * Effect::getEffect()
|
|
{
|
|
// extract the module from the runtime
|
|
PyObject * module = PyObject_GetAttrString(PyImport_AddModule("__main__"), "hyperion");
|
|
|
|
if (!PyModule_Check(module))
|
|
{
|
|
// something is wrong
|
|
Py_XDECREF(module);
|
|
Error(Logger::getInstance("EFFECTENGINE"), "Unable to retrieve the effect object from the Python runtime");
|
|
return nullptr;
|
|
}
|
|
|
|
// retrieve the capsule with the effect
|
|
PyObject * effectCapsule = PyObject_GetAttrString(module, "__effectObj");
|
|
Py_XDECREF(module);
|
|
|
|
if (!PyCapsule_CheckExact(effectCapsule))
|
|
{
|
|
// something is wrong
|
|
Py_XDECREF(effectCapsule);
|
|
Error(Logger::getInstance("EFFECTENGINE"), "Unable to retrieve the effect object from the Python runtime");
|
|
return nullptr;
|
|
}
|
|
|
|
// Get the effect from the capsule
|
|
Effect * effect = reinterpret_cast<Effect *>(PyCapsule_GetPointer(effectCapsule, nullptr));
|
|
Py_XDECREF(effectCapsule);
|
|
return effect;
|
|
}
|