core/colorquant: add imageRect option for cropping image

Author: HigherOrderLogic

src/core/colorquantizer.cpp

@@ -13,6 +13,7 @@
 #include <qnumeric.h>
 #include <qobject.h>
 #include <qqmllist.h>
+#include <qrect.h>
 #include <qrgb.h>
 #include <qthreadpool.h>
 #include <qtmetamacros.h>
@@ -24,9 +25,15 @@ namespace {
 QS_LOGGING_CATEGORY(logColorQuantizer, "quickshell.colorquantizer", QtWarningMsg);
 }
 
-ColorQuantizerOperation::ColorQuantizerOperation(QUrl* source, qreal depth, qreal rescaleSize)
+ColorQuantizerOperation::ColorQuantizerOperation(
+    QUrl* source,
+    qreal depth,
+    QRect imageRect,
+    qreal rescaleSize
+)
     : source(source)
     , maxDepth(depth)
+    , imageRect(imageRect)
     , rescaleSize(rescaleSize) {
 	setAutoDelete(false);
 }
@@ -37,6 +44,11 @@ void ColorQuantizerOperation::quantizeImage(const QAtomicInteger<bool>& shouldCa
 	colors.clear();
 
 	auto image = QImage(source->toLocalFile());
+
+	if (imageRect.isValid()) {
+		image = image.copy(imageRect);
+	}
+
 	if ((image.width() > rescaleSize || image.height() > rescaleSize) && rescaleSize > 0) {
 		image = image.scaled(
 		    static_cast<int>(rescaleSize),
@@ -209,6 +221,15 @@ void ColorQuantizer::setRescaleSize(int rescaleSize) {
 	}
 }
 
+void ColorQuantizer::setImageRect(QRect imageRect) {
+	if (imageRect != mImageRect) {
+		mImageRect = imageRect;
+		emit this->imageRectChanged();
+
+		if (this->componentCompleted) quantizeAsync();
+	}
+}
+
 void ColorQuantizer::operationFinished(const QList<QColor>& result) {
 	bColors = result;
 	this->liveOperation = nullptr;
@@ -219,7 +240,7 @@ void ColorQuantizer::quantizeAsync() {
 	if (this->liveOperation) this->cancelAsync();
 
 	qCDebug(logColorQuantizer) << "Starting color quantization asynchronously";
-	this->liveOperation = new ColorQuantizerOperation(&mSource, mDepth, mRescaleSize);
+	this->liveOperation = new ColorQuantizerOperation(&mSource, mDepth, mImageRect, mRescaleSize);
 
 	QObject::connect(
 	    this->liveOperation,

src/core/colorquantizer.hpp

@@ -5,6 +5,7 @@
 #include <qproperty.h>
 #include <qqmlintegration.h>
 #include <qqmlparserstatus.h>
+#include <qrect.h>
 #include <qrunnable.h>
 #include <qtmetamacros.h>
 #include <qtypes.h>
@@ -16,7 +17,7 @@ class ColorQuantizerOperation
 	Q_OBJECT;
 
 public:
-	explicit ColorQuantizerOperation(QUrl* source, qreal depth, qreal rescaleSize);
+	explicit ColorQuantizerOperation(QUrl* source, qreal depth, QRect imageRect, qreal rescaleSize);
 
 	void run() override;
 	void tryCancel();
@@ -44,6 +45,7 @@ private slots:
 	QList<QColor> colors;
 	QUrl* source;
 	qreal maxDepth;
+	QRect imageRect;
 	qreal rescaleSize;
 };
 
@@ -83,6 +85,9 @@ class ColorQuantizer
 	/// > reccommended to rescale, otherwise the quantization process will take much longer.
 	Q_PROPERTY(qreal rescaleSize READ rescaleSize WRITE setRescaleSize NOTIFY rescaleSizeChanged);
 
+	/// Rectangle that the source image is cropped to.
+	Q_PROPERTY(QRect imageRect READ imageRect WRITE setImageRect NOTIFY imageRectChanged);
+
 public:
 	explicit ColorQuantizer(QObject* parent = nullptr): QObject(parent) {}
 
@@ -100,11 +105,15 @@ class ColorQuantizer
 	[[nodiscard]] qreal rescaleSize() const { return mRescaleSize; }
 	void setRescaleSize(int rescaleSize);
 
+	[[nodiscard]] QRect imageRect() const { return mImageRect; }
+	void setImageRect(QRect imageRect);
+
 signals:
 	void colorsChanged();
 	void sourceChanged();
 	void depthChanged();
 	void rescaleSizeChanged();
+	void imageRectChanged();
 
 public slots:
 	void operationFinished(const QList<QColor>& result);
@@ -118,6 +127,7 @@ public slots:
 	QUrl mSource;
 	qreal mDepth = 0;
 	qreal mRescaleSize = 0;
+	QRect mImageRect = QRect();
 
 	Q_OBJECT_BINDABLE_PROPERTY(
 	    ColorQuantizer,