forked from commontk/CTK
/
ctkWidgetsUtils.cpp
110 lines (97 loc) · 3.61 KB
/
ctkWidgetsUtils.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/*=========================================================================
Library: CTK
Copyright (c) Kitware Inc.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0.txt
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
=========================================================================*/
// Qt includes
#include <QBuffer>
#include <QImage>
#include <QGLWidget>
#include <QPainter>
#include <QWidget>
// ctkWidgets includes
#include "ctkWidgetsUtils.h"
//----------------------------------------------------------------------------
QString ctk::base64HTMLImageTagSrc(const QImage& image)
{
QByteArray byteArray;
QBuffer buffer(&byteArray);
buffer.open(QIODevice::WriteOnly);
image.save(&buffer, "PNG");
return QString("data:image/png;base64,%1")
.arg(QString(buffer.data().toBase64()));
}
//----------------------------------------------------------------------------
QImage ctk::grabWidget(QWidget* widget, QRect rectangle)
{
if (!widget)
{
return QImage();
}
if (!rectangle.isValid())
{
rectangle = QRect(0,0,widget->width(),widget->height());
}
QPixmap widgetPixmap = QPixmap::grabWidget(widget, rectangle);
QImage widgetImage = widgetPixmap.toImage();
QPainter painter;
painter.begin(&widgetImage);
foreach(QGLWidget* glWidget, widget->findChildren<QGLWidget*>())
{
if (!glWidget->isVisible())
{
continue;
}
QRect subWidgetRect = QRect(glWidget->mapTo(widget, QPoint(0,0)), glWidget->size());
if (!rectangle.intersects(subWidgetRect))
{
continue;
}
QImage subImage = glWidget->grabFrameBuffer();
painter.drawImage(subWidgetRect, subImage);
}
painter.end();
return widgetImage;
}
//----------------------------------------------------------------------------
QImage ctk::kwIconToQImage(const unsigned char *data, int width, int height, int pixelSize, unsigned int bufferLength, int options)
{
Q_UNUSED(options); // not yet supported
QByteArray imageData = QByteArray::fromRawData(
reinterpret_cast<const char *>(data), bufferLength);
unsigned int expectedLength = width * height * pixelSize;
// Start with a zlib header ? If not, then it must be base64 encoded
if (bufferLength != expectedLength &&
static_cast<unsigned char>(imageData[0]) != 0x78 &&
static_cast<unsigned char>(imageData[1]) != 0xDA)
{
imageData = QByteArray::fromBase64(imageData);
bufferLength = imageData.size();
}
if (bufferLength != expectedLength &&
static_cast<unsigned char>(imageData[0]) == 0x78 &&
static_cast<unsigned char>(imageData[1]) == 0xDA)
{
imageData.prepend(static_cast<char>((expectedLength >> 0) & 0xFF));
imageData.prepend(static_cast<char>((expectedLength >> 8) & 0xFF));
imageData.prepend(static_cast<char>((expectedLength >> 16) & 0xFF));
imageData.prepend(static_cast<char>((expectedLength >> 24) & 0xFF));
imageData = qUncompress(imageData);
}
QImage image(reinterpret_cast<unsigned char*>(imageData.data()),
width, height, width * pixelSize,
pixelSize == 4 ? QImage::Format_ARGB32 : QImage::Format_RGB888);
if (pixelSize == 4)
{
return image.rgbSwapped();
}
return image.copy();
}