mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2023-10-10 13:36:59 +02:00
1065 lines
28 KiB
C++
1065 lines
28 KiB
C++
#include <cmath>
|
|
|
|
#include <effectengine/Effect.h>
|
|
#include <effectengine/EffectModule.h>
|
|
|
|
// hyperion
|
|
#include <hyperion/Hyperion.h>
|
|
#include <utils/Logger.h>
|
|
|
|
// qt
|
|
#include <QJsonArray>
|
|
#include <QDateTime>
|
|
#include <QImageReader>
|
|
#include <QBuffer>
|
|
|
|
// Get the effect from the capsule
|
|
#define getEffect() static_cast<Effect*>((Effect*)PyCapsule_Import("hyperion.__effectObj", 0))
|
|
|
|
// create the hyperion module
|
|
struct PyModuleDef EffectModule::moduleDef = {
|
|
PyModuleDef_HEAD_INIT,
|
|
"hyperion", /* m_name */
|
|
"Hyperion module", /* m_doc */
|
|
-1, /* m_size */
|
|
EffectModule::effectMethods, /* m_methods */
|
|
NULL, /* m_reload */
|
|
NULL, /* m_traverse */
|
|
NULL, /* m_clear */
|
|
NULL, /* m_free */
|
|
};
|
|
|
|
PyObject* EffectModule::PyInit_hyperion()
|
|
{
|
|
return PyModule_Create(&moduleDef);
|
|
}
|
|
|
|
void EffectModule::registerHyperionExtensionModule()
|
|
{
|
|
PyImport_AppendInittab("hyperion", &PyInit_hyperion);
|
|
}
|
|
|
|
PyObject *EffectModule::json2python(const QJsonValue &jsonData)
|
|
{
|
|
switch (jsonData.type())
|
|
{
|
|
case QJsonValue::Null:
|
|
Py_RETURN_NONE;
|
|
case QJsonValue::Undefined:
|
|
Py_RETURN_NOTIMPLEMENTED;
|
|
case QJsonValue::Double:
|
|
{
|
|
if (std::round(jsonData.toDouble()) != jsonData.toDouble())
|
|
return Py_BuildValue("d", jsonData.toDouble());
|
|
|
|
return Py_BuildValue("i", jsonData.toInt());
|
|
}
|
|
case QJsonValue::Bool:
|
|
return Py_BuildValue("i", jsonData.toBool() ? 1 : 0);
|
|
case QJsonValue::String:
|
|
return Py_BuildValue("s", jsonData.toString().toUtf8().constData());
|
|
case QJsonValue::Object:
|
|
{
|
|
PyObject * dict= PyDict_New();
|
|
QJsonObject objectData = jsonData.toObject();
|
|
for (QJsonObject::iterator i = objectData.begin(); i != objectData.end(); ++i)
|
|
{
|
|
PyObject * obj = json2python(*i);
|
|
PyDict_SetItemString(dict, i.key().toStdString().c_str(), obj);
|
|
Py_XDECREF(obj);
|
|
}
|
|
return dict;
|
|
}
|
|
case QJsonValue::Array:
|
|
{
|
|
QJsonArray arrayData = jsonData.toArray();
|
|
PyObject * list = PyList_New(arrayData.size());
|
|
int index = 0;
|
|
for (QJsonArray::iterator i = arrayData.begin(); i != arrayData.end(); ++i, ++index)
|
|
{
|
|
PyObject * obj = json2python(*i);
|
|
Py_INCREF(obj);
|
|
PyList_SetItem(list, index, obj);
|
|
Py_XDECREF(obj);
|
|
}
|
|
return list;
|
|
}
|
|
}
|
|
|
|
assert(false);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
// Python method table
|
|
PyMethodDef EffectModule::effectMethods[] = {
|
|
{"setColor" , EffectModule::wrapSetColor , METH_VARARGS, "Set a new color for the leds."},
|
|
{"setImage" , EffectModule::wrapSetImage , METH_VARARGS, "Set a new image to process and determine new led colors."},
|
|
{"getImage" , EffectModule::wrapGetImage , METH_VARARGS, "get image data from file."},
|
|
{"abort" , EffectModule::wrapAbort , METH_NOARGS, "Check if the effect should abort execution."},
|
|
{"imageShow" , EffectModule::wrapImageShow , METH_VARARGS, "set current effect image to hyperion core."},
|
|
{"imageLinearGradient" , EffectModule::wrapImageLinearGradient , METH_VARARGS, ""},
|
|
{"imageConicalGradient" , EffectModule::wrapImageConicalGradient , METH_VARARGS, ""},
|
|
{"imageRadialGradient" , EffectModule::wrapImageRadialGradient , METH_VARARGS, ""},
|
|
{"imageSolidFill" , EffectModule::wrapImageSolidFill , METH_VARARGS, ""},
|
|
{"imageDrawLine" , EffectModule::wrapImageDrawLine , METH_VARARGS, ""},
|
|
{"imageDrawPoint" , EffectModule::wrapImageDrawPoint , METH_VARARGS, ""},
|
|
{"imageDrawRect" , EffectModule::wrapImageDrawRect , METH_VARARGS, ""},
|
|
{"imageDrawPolygon" , EffectModule::wrapImageDrawPolygon , METH_VARARGS, ""},
|
|
{"imageDrawPie" , EffectModule::wrapImageDrawPie , METH_VARARGS, ""},
|
|
{"imageSetPixel" , EffectModule::wrapImageSetPixel , METH_VARARGS, "set pixel color of image"},
|
|
{"imageGetPixel" , EffectModule::wrapImageGetPixel , METH_VARARGS, "get pixel color of image"},
|
|
{"imageSave" , EffectModule::wrapImageSave , METH_NOARGS, "adds a new background image"},
|
|
{"imageMinSize" , EffectModule::wrapImageMinSize , METH_VARARGS, "sets minimal dimension of background image"},
|
|
{"imageWidth" , EffectModule::wrapImageWidth , METH_NOARGS, "gets image width"},
|
|
{"imageHeight" , EffectModule::wrapImageHeight , METH_NOARGS, "gets image height"},
|
|
{"imageCRotate" , EffectModule::wrapImageCRotate , METH_VARARGS, "rotate the coordinate system by given angle"},
|
|
{"imageCOffset" , EffectModule::wrapImageCOffset , METH_VARARGS, "Add offset to the coordinate system"},
|
|
{"imageCShear" , EffectModule::wrapImageCShear , METH_VARARGS, "Shear of coordinate system by the given horizontal/vertical axis"},
|
|
{"imageResetT" , EffectModule::wrapImageResetT , METH_NOARGS, "Resets all coords modifications (rotate,offset,shear)"},
|
|
{NULL, NULL, 0, NULL}
|
|
};
|
|
|
|
PyObject* EffectModule::wrapSetColor(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
// determine the timeout
|
|
int timeout = getEffect()->_timeout;
|
|
if (timeout > 0)
|
|
{
|
|
timeout = getEffect()->_endTime - QDateTime::currentMSecsSinceEpoch();
|
|
|
|
// we are done if the time has passed
|
|
if (timeout <= 0) Py_RETURN_NONE;
|
|
}
|
|
|
|
// 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))
|
|
{
|
|
getEffect()->_colors.fill(color);
|
|
getEffect()->setInput(getEffect()->_priority, getEffect()->_colors.toStdVector(), timeout, false);
|
|
Py_RETURN_NONE;
|
|
}
|
|
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 * getEffect()->_hyperion->getLedCount())
|
|
{
|
|
char * data = PyByteArray_AS_STRING(bytearray);
|
|
memcpy(getEffect()->_colors.data(), data, length);
|
|
getEffect()->setInput(getEffect()->_priority, getEffect()->_colors.toStdVector(), timeout, false);
|
|
Py_RETURN_NONE;
|
|
}
|
|
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;
|
|
}
|
|
}
|
|
|
|
PyObject* EffectModule::wrapSetImage(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
// determine the timeout
|
|
int timeout = getEffect()->_timeout;
|
|
if (timeout > 0)
|
|
{
|
|
timeout = getEffect()->_endTime - QDateTime::currentMSecsSinceEpoch();
|
|
|
|
// we are done if the time has passed
|
|
if (timeout <= 0) Py_RETURN_NONE;
|
|
}
|
|
|
|
// 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);
|
|
getEffect()->setInputImage(getEffect()->_priority, image, timeout, false);
|
|
Py_RETURN_NONE;
|
|
}
|
|
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* EffectModule::wrapGetImage(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
QString file;
|
|
QBuffer buffer;
|
|
QImageReader reader;
|
|
|
|
if (getEffect()->_imageData.isEmpty())
|
|
{
|
|
Q_INIT_RESOURCE(EffectEngine);
|
|
|
|
char *source;
|
|
if(!PyArg_ParseTuple(args, "s", &source))
|
|
{
|
|
PyErr_SetString(PyExc_TypeError, "String required");
|
|
return nullptr;
|
|
}
|
|
|
|
file = QString::fromUtf8(source);
|
|
|
|
if (file.mid(0, 1) == ":")
|
|
file = ":/effects/"+file.mid(1);
|
|
|
|
reader.setDecideFormatFromContent(true);
|
|
reader.setFileName(file);
|
|
}
|
|
else
|
|
{
|
|
buffer.setData(QByteArray::fromBase64(getEffect()->_imageData.toUtf8()));
|
|
buffer.open(QBuffer::ReadOnly);
|
|
reader.setDecideFormatFromContent(true);
|
|
reader.setDevice(&buffer);
|
|
}
|
|
|
|
if (reader.canRead())
|
|
{
|
|
PyObject *result = PyList_New(reader.imageCount());
|
|
|
|
for (int i = 0; i < reader.imageCount(); ++i)
|
|
{
|
|
reader.jumpToImage(i);
|
|
if (reader.canRead())
|
|
{
|
|
QImage qimage = reader.read();
|
|
|
|
int width = qimage.width();
|
|
int height = qimage.height();
|
|
|
|
QByteArray binaryImage;
|
|
for (int i = 0; i<height; ++i)
|
|
{
|
|
const QRgb *scanline = reinterpret_cast<const QRgb *>(qimage.scanLine(i));
|
|
for (int j = 0; j< width; ++j)
|
|
{
|
|
binaryImage.append((char) qRed(scanline[j]));
|
|
binaryImage.append((char) qGreen(scanline[j]));
|
|
binaryImage.append((char) qBlue(scanline[j]));
|
|
}
|
|
}
|
|
PyList_SET_ITEM(result, i, Py_BuildValue("{s:i,s:i,s:O}", "imageWidth", width, "imageHeight", height, "imageData", PyByteArray_FromStringAndSize(binaryImage.constData(),binaryImage.size())));
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_TypeError, reader.errorString().toUtf8().constData());
|
|
return nullptr;
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_TypeError, reader.errorString().toUtf8().constData());
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
PyObject* EffectModule::wrapAbort(PyObject *self, PyObject *)
|
|
{
|
|
return Py_BuildValue("i", getEffect()->isInterruptionRequested() ? 1 : 0);
|
|
}
|
|
|
|
|
|
PyObject* EffectModule::wrapImageShow(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
// determine the timeout
|
|
int timeout = getEffect()->_timeout;
|
|
if (timeout > 0)
|
|
{
|
|
timeout = getEffect()->_endTime - QDateTime::currentMSecsSinceEpoch();
|
|
|
|
// we are done if the time has passed
|
|
if (timeout <= 0) Py_RETURN_NONE;
|
|
}
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int imgId = -1;
|
|
bool argsOk = (argCount == 0);
|
|
if (argCount == 1 && PyArg_ParseTuple(args, "i", &imgId))
|
|
{
|
|
argsOk = true;
|
|
}
|
|
|
|
if ( ! argsOk || (imgId>-1 && imgId >= getEffect()->_imageStack.size()))
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
|
|
QImage * qimage = (imgId<0) ? &(getEffect()->_image) : &(getEffect()->_imageStack[imgId]);
|
|
int width = qimage->width();
|
|
int height = qimage->height();
|
|
|
|
Image<ColorRgb> image(width, height);
|
|
QByteArray binaryImage;
|
|
|
|
for (int i = 0; i<height; ++i)
|
|
{
|
|
const QRgb * scanline = reinterpret_cast<const QRgb *>(qimage->scanLine(i));
|
|
for (int j = 0; j< 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());
|
|
getEffect()->setInputImage(getEffect()->_priority, image, timeout, false);
|
|
|
|
return Py_BuildValue("");
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageLinearGradient(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
PyObject * bytearray = nullptr;
|
|
int startRX = 0;
|
|
int startRY = 0;
|
|
int startX = 0;
|
|
int startY = 0;
|
|
int endX, width = getEffect()->_imageSize.width();
|
|
int endY, height = getEffect()->_imageSize.height();
|
|
int spread = 0;
|
|
|
|
bool argsOK = false;
|
|
|
|
if ( argCount == 10 && PyArg_ParseTuple(args, "iiiiiiiiOi", &startRX, &startRY, &width, &height, &startX, &startY, &endX, &endY, &bytearray, &spread) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 6 && PyArg_ParseTuple(args, "iiiiOi", &startX, &startY, &endX, &endY, &bytearray, &spread) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
|
|
if (argsOK)
|
|
{
|
|
if (PyByteArray_Check(bytearray))
|
|
{
|
|
const int length = PyByteArray_Size(bytearray);
|
|
const unsigned arrayItemLength = 5;
|
|
if (length % arrayItemLength == 0)
|
|
{
|
|
QRect myQRect(startRX,startRY,width,height);
|
|
QLinearGradient gradient(QPoint(startX,startY), QPoint(endX,endY));
|
|
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])
|
|
));
|
|
}
|
|
|
|
gradient.setSpread(static_cast<QGradient::Spread>(spread));
|
|
getEffect()->_painter->fillRect(myQRect, gradient);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Length of bytearray argument should multiple of 5");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "No bytearray properly defined");
|
|
return nullptr;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageConicalGradient(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
PyObject * bytearray = nullptr;
|
|
int centerX, centerY, angle;
|
|
int startX = 0;
|
|
int startY = 0;
|
|
int width = getEffect()->_imageSize.width();
|
|
int height = getEffect()->_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 = qMax(qMin(angle,360),0);
|
|
|
|
if (argsOK)
|
|
{
|
|
if (PyByteArray_Check(bytearray))
|
|
{
|
|
const int length = PyByteArray_Size(bytearray);
|
|
const unsigned arrayItemLength = 5;
|
|
if (length % arrayItemLength == 0)
|
|
{
|
|
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])
|
|
));
|
|
}
|
|
|
|
getEffect()->_painter->fillRect(myQRect, gradient);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
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;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
|
|
PyObject* EffectModule::wrapImageRadialGradient(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
PyObject * bytearray = nullptr;
|
|
int centerX, centerY, radius, focalX, focalY, focalRadius, spread;
|
|
int startX = 0;
|
|
int startY = 0;
|
|
int width = getEffect()->_imageSize.width();
|
|
int height = getEffect()->_imageSize.height();
|
|
|
|
bool argsOK = false;
|
|
|
|
if ( argCount == 12 && PyArg_ParseTuple(args, "iiiiiiiiiiOi", &startX, &startY, &width, &height, ¢erX, ¢erY, &radius, &focalX, &focalY, &focalRadius, &bytearray, &spread) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 9 && PyArg_ParseTuple(args, "iiiiiiiOi", &startX, &startY, &width, &height, ¢erX, ¢erY, &radius, &bytearray, &spread) )
|
|
{
|
|
argsOK = true;
|
|
focalX = centerX;
|
|
focalY = centerY;
|
|
focalRadius = radius;
|
|
}
|
|
if ( argCount == 8 && PyArg_ParseTuple(args, "iiiiiiOi", ¢erX, ¢erY, &radius, &focalX, &focalY, &focalRadius, &bytearray, &spread) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 5 && PyArg_ParseTuple(args, "iiiOi", ¢erX, ¢erY, &radius, &bytearray, &spread) )
|
|
{
|
|
argsOK = true;
|
|
focalX = centerX;
|
|
focalY = centerY;
|
|
focalRadius = radius;
|
|
}
|
|
|
|
if (argsOK)
|
|
{
|
|
if (PyByteArray_Check(bytearray))
|
|
{
|
|
int length = PyByteArray_Size(bytearray);
|
|
if (length % 4 == 0)
|
|
{
|
|
|
|
QRect myQRect(startX,startY,width,height);
|
|
QRadialGradient gradient(QPoint(centerX,centerY), qMax(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(static_cast<QGradient::Spread>(spread));
|
|
getEffect()->_painter->fillRect(myQRect, gradient);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
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;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageDrawPolygon(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
PyObject * bytearray = nullptr;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int r, g, b;
|
|
int a = 255;
|
|
|
|
bool argsOK = false;
|
|
|
|
if ( argCount == 5 && PyArg_ParseTuple(args, "Oiiii", &bytearray, &r, &g, &b, &a) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 4 && PyArg_ParseTuple(args, "Oiii", &bytearray, &r, &g, &b) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
|
|
if (argsOK)
|
|
{
|
|
if (PyByteArray_Check(bytearray))
|
|
{
|
|
int length = PyByteArray_Size(bytearray);
|
|
if (length % 2 == 0)
|
|
{
|
|
QVector <QPoint> points;
|
|
char * data = PyByteArray_AS_STRING(bytearray);
|
|
|
|
for (int idx=0; idx<length; idx+=2)
|
|
{
|
|
points.append(QPoint((int)(data[idx]),(int)(data[idx+1])));
|
|
}
|
|
|
|
QPainter * painter = getEffect()->_painter;
|
|
QPen oldPen = painter->pen();
|
|
QPen newPen(QColor(r,g,b,a));
|
|
painter->setPen(newPen);
|
|
painter->setBrush(QBrush(QColor(r,g,b,a), Qt::SolidPattern));
|
|
painter->drawPolygon(points);
|
|
painter->setPen(oldPen);
|
|
Py_RETURN_NONE;
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Length of bytearray argument should multiple of 2");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Argument 1 is not a bytearray");
|
|
return nullptr;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageDrawPie(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
PyObject * bytearray = nullptr;
|
|
|
|
QString brush;
|
|
int argCount = PyTuple_Size(args);
|
|
int radius, centerX, centerY;
|
|
int startAngle = 0;
|
|
int spanAngle = 360;
|
|
int r = 0;
|
|
int g = 0;
|
|
int b = 0;
|
|
int a = 255;
|
|
|
|
bool argsOK = false;
|
|
|
|
if ( argCount == 9 && PyArg_ParseTuple(args, "iiiiiiiii", ¢erX, ¢erY, &radius, &startAngle, &spanAngle, &r, &g, &b, &a) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 8 && PyArg_ParseTuple(args, "iiiiiiii", ¢erX, ¢erY, &radius, &startAngle, &spanAngle, &r, &g, &b) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 7 && PyArg_ParseTuple(args, "iiiiisO", ¢erX, ¢erY, &radius, &startAngle, &spanAngle, &brush, &bytearray) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 5 && PyArg_ParseTuple(args, "iiisO", ¢erX, ¢erY, &radius, &brush, &bytearray) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
|
|
if (argsOK)
|
|
{
|
|
QPainter * painter = getEffect()->_painter;
|
|
startAngle = qMax(qMin(startAngle,360),0);
|
|
spanAngle = qMax(qMin(spanAngle,360),-360);
|
|
|
|
if( argCount == 7 || argCount == 5 )
|
|
{
|
|
a = 0;
|
|
if (PyByteArray_Check(bytearray))
|
|
{
|
|
int length = PyByteArray_Size(bytearray);
|
|
if (length % 5 == 0)
|
|
{
|
|
|
|
QConicalGradient gradient(QPoint(centerX,centerY), startAngle);
|
|
|
|
|
|
char * data = PyByteArray_AS_STRING(bytearray);
|
|
|
|
for (int idx=0; idx<length; idx+=5)
|
|
{
|
|
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->setBrush(gradient);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Length of bytearray argument should multiple of 5");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PyExc_RuntimeError, "Last argument is not a bytearray");
|
|
return nullptr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
painter->setBrush(QBrush(QColor(r,g,b,a), Qt::SolidPattern));
|
|
}
|
|
QPen oldPen = painter->pen();
|
|
QPen newPen(QColor(r,g,b,a));
|
|
painter->setPen(newPen);
|
|
painter->drawPie(centerX - radius, centerY - radius, centerX + radius, centerY + radius, startAngle * 16, spanAngle * 16);
|
|
painter->setPen(oldPen);
|
|
Py_RETURN_NONE;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageSolidFill(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int r, g, b;
|
|
int a = 255;
|
|
int startX = 0;
|
|
int startY = 0;
|
|
int width = getEffect()->_imageSize.width();
|
|
int height = getEffect()->_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);
|
|
getEffect()->_painter->fillRect(myQRect, QColor(r,g,b,a));
|
|
Py_RETURN_NONE;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
|
|
PyObject* EffectModule::wrapImageDrawLine(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int r, g, b;
|
|
int a = 255;
|
|
int startX = 0;
|
|
int startY = 0;
|
|
int thick = 1;
|
|
int endX = getEffect()->_imageSize.width();
|
|
int endY = getEffect()->_imageSize.height();
|
|
|
|
bool argsOK = false;
|
|
|
|
if ( argCount == 9 && PyArg_ParseTuple(args, "iiiiiiiii", &startX, &startY, &endX, &endY, &thick, &r, &g, &b, &a) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 8 && PyArg_ParseTuple(args, "iiiiiiii", &startX, &startY, &endX, &endY, &thick, &r, &g, &b) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
|
|
if (argsOK)
|
|
{
|
|
QPainter * painter = getEffect()->_painter;
|
|
QRect myQRect(startX, startY, endX, endY);
|
|
QPen oldPen = painter->pen();
|
|
QPen newPen(QColor(r,g,b,a));
|
|
newPen.setWidth(thick);
|
|
painter->setPen(newPen);
|
|
painter->drawLine(startX, startY, endX, endY);
|
|
painter->setPen(oldPen);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageDrawPoint(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int r, g, b, x, y;
|
|
int a = 255;
|
|
int thick = 1;
|
|
|
|
bool argsOK = false;
|
|
|
|
if ( argCount == 7 && PyArg_ParseTuple(args, "iiiiiii", &x, &y, &thick, &r, &g, &b, &a) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 6 && PyArg_ParseTuple(args, "iiiiii", &x, &y, &thick, &r, &g, &b) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
|
|
if (argsOK)
|
|
{
|
|
QPainter * painter = getEffect()->_painter;
|
|
QPen oldPen = painter->pen();
|
|
QPen newPen(QColor(r,g,b,a));
|
|
newPen.setWidth(thick);
|
|
painter->setPen(newPen);
|
|
painter->drawPoint(x, y);
|
|
painter->setPen(oldPen);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageDrawRect(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int r, g, b;
|
|
int a = 255;
|
|
int startX = 0;
|
|
int startY = 0;
|
|
int thick = 1;
|
|
int width = getEffect()->_imageSize.width();
|
|
int height = getEffect()->_imageSize.height();
|
|
|
|
bool argsOK = false;
|
|
|
|
if ( argCount == 9 && PyArg_ParseTuple(args, "iiiiiiiii", &startX, &startY, &width, &height, &thick, &r, &g, &b, &a) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
if ( argCount == 8 && PyArg_ParseTuple(args, "iiiiiiii", &startX, &startY, &width, &height, &thick, &r, &g, &b) )
|
|
{
|
|
argsOK = true;
|
|
}
|
|
|
|
if (argsOK)
|
|
{
|
|
QPainter * painter = getEffect()->_painter;
|
|
QRect myQRect(startX,startY,width,height);
|
|
QPen oldPen = painter->pen();
|
|
QPen newPen(QColor(r,g,b,a));
|
|
newPen.setWidth(thick);
|
|
painter->setPen(newPen);
|
|
painter->drawRect(startX, startY, width, height);
|
|
painter->setPen(oldPen);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
|
|
PyObject* EffectModule::wrapImageSetPixel(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int r, g, b, x, y;
|
|
|
|
if ( argCount == 5 && PyArg_ParseTuple(args, "iiiii", &x, &y, &r, &g, &b ) )
|
|
{
|
|
getEffect()->_image.setPixel(x,y,qRgb(r,g,b));
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
|
|
PyObject* EffectModule::wrapImageGetPixel(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int x, y;
|
|
|
|
if ( argCount == 2 && PyArg_ParseTuple(args, "ii", &x, &y) )
|
|
{
|
|
QRgb rgb = getEffect()->_image.pixel(x,y);
|
|
return Py_BuildValue("iii",qRed(rgb),qGreen(rgb),qBlue(rgb));
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageSave(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
QImage img(getEffect()->_image.copy());
|
|
getEffect()->_imageStack.append(img);
|
|
|
|
return Py_BuildValue("i", getEffect()->_imageStack.size()-1);
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageMinSize(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int w, h;
|
|
int width = getEffect()->_imageSize.width();
|
|
int height = getEffect()->_imageSize.height();
|
|
|
|
if ( argCount == 2 && PyArg_ParseTuple(args, "ii", &w, &h) )
|
|
{
|
|
if (width<w || height<h)
|
|
{
|
|
delete getEffect()->_painter;
|
|
|
|
getEffect()->_image = getEffect()->_image.scaled(qMax(width,w),qMax(height,h), Qt::KeepAspectRatioByExpanding, Qt::SmoothTransformation);
|
|
getEffect()->_imageSize = getEffect()->_image.size();
|
|
getEffect()->_painter = new QPainter(&(getEffect()->_image));
|
|
}
|
|
return Py_BuildValue("ii", getEffect()->_image.width(), getEffect()->_image.height());
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageWidth(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
return Py_BuildValue("i", getEffect()->_imageSize.width());
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageHeight(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
return Py_BuildValue("i", getEffect()->_imageSize.height());
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageCRotate(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int argCount = PyTuple_Size(args);
|
|
int angle;
|
|
|
|
if ( argCount == 1 && PyArg_ParseTuple(args, "i", &angle ) )
|
|
{
|
|
angle = qMax(qMin(angle,360),0);
|
|
getEffect()->_painter->rotate(angle);
|
|
Py_RETURN_NONE;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageCOffset(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int offsetX = 0;
|
|
int offsetY = 0;
|
|
int argCount = PyTuple_Size(args);
|
|
|
|
if ( argCount == 2 )
|
|
{
|
|
PyArg_ParseTuple(args, "ii", &offsetX, &offsetY );
|
|
}
|
|
|
|
getEffect()->_painter->translate(QPoint(offsetX,offsetY));
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageCShear(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
int sh,sv;
|
|
int argCount = PyTuple_Size(args);
|
|
|
|
if ( argCount == 2 && PyArg_ParseTuple(args, "ii", &sh, &sv ))
|
|
{
|
|
getEffect()->_painter->shear(sh,sv);
|
|
Py_RETURN_NONE;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* EffectModule::wrapImageResetT(PyObject *self, PyObject *args)
|
|
{
|
|
// check if we have aborted already
|
|
if (getEffect()->isInterruptionRequested()) Py_RETURN_NONE;
|
|
|
|
getEffect()->_painter->resetTransform();
|
|
Py_RETURN_NONE;
|
|
}
|