summaryrefslogtreecommitdiff
path: root/plugin/src/Caelestia/cutils.cpp
blob: 1285c55ed7f3cde1809c4d2b72dbe9af701b1f9b (plain)
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
#include "cutils.hpp"

#include <qobject.h>
#include <QtQuick/QQuickItem>
#include <QtQuick/QQuickItemGrabResult>
#include <QThreadPool>
#include <QQmlEngine>

void CUtils::saveItem(QQuickItem* target, const QUrl& path) {
    this->saveItem(target, path, QRect(), QJSValue(), QJSValue());
}

void CUtils::saveItem(QQuickItem* target, const QUrl& path, const QRect& rect) {
	this->saveItem(target, path, rect, QJSValue(), QJSValue());
}

void CUtils::saveItem(QQuickItem* target, const QUrl& path, QJSValue onSaved) {
	this->saveItem(target, path, QRect(), onSaved, QJSValue());
}

void CUtils::saveItem(QQuickItem* target, const QUrl& path, QJSValue onSaved, QJSValue onFailed) {
	this->saveItem(target, path, QRect(), onSaved, onFailed);
}

void CUtils::saveItem(QQuickItem* target, const QUrl& path, const QRect& rect, QJSValue onSaved) {
	this->saveItem(target, path, rect, onSaved, QJSValue());
}

void CUtils::saveItem(QQuickItem* target, const QUrl& path, const QRect& rect, QJSValue onSaved, QJSValue onFailed) {
	if (!target) {
		qWarning() << "CUtils::saveItem: a target is required";
		return;
	}

	if (!path.isLocalFile()) {
		qWarning() << "CUtils::saveItem:" << path << "is not a local file";
		return;
	}

	QSharedPointer<QQuickItemGrabResult> grabResult = target->grabToImage();

	QObject::connect(
	    grabResult.data(),
	    &QQuickItemGrabResult::ready,
	    this,
	    [grabResult, rect, path, onSaved, onFailed, this]() {
		    QThreadPool::globalInstance()->start([grabResult, rect, path, onSaved, onFailed, this] {
			    QImage image = grabResult->image();

			    if (!rect.isEmpty()) {
				    image = image.copy(rect);
			    }

				const QString file = path.toLocalFile();
			    if (image.save(file)) {
					if (onSaved.isCallable()) {
						onSaved.call({ QJSValue(file), qmlEngine(this)->toScriptValue(QVariant::fromValue(path)) });
					}
				} else if (onFailed.isCallable()) {
					onFailed.call({ qmlEngine(this)->toScriptValue(QVariant::fromValue(path)) });
				}
		    });
	    }
	);
}