From 22bcc888b99c8ceb6b8f8181ace5e2c81b78e0d4 Mon Sep 17 00:00:00 2001
From: Albert S <mail@quitesimple.org>
Date: Sun, 18 Aug 2019 23:23:11 +0200
Subject: [PATCH] new future: highlight searched words in pdf preview

---
 gui/mainwindow.cpp | 20 +++++++++++++++++---
 gui/mainwindow.h   |  2 ++
 gui/pdfworker.cpp  | 22 ++++++++++++++++++----
 gui/pdfworker.h    |  3 ++-
 4 files changed, 39 insertions(+), 8 deletions(-)

diff --git a/gui/mainwindow.cpp b/gui/mainwindow.cpp
index 923fe47..a0789f9 100644
--- a/gui/mainwindow.cpp
+++ b/gui/mainwindow.cpp
@@ -16,7 +16,6 @@
 #include "clicklabel.h"
 #include "../shared/sqlitesearch.h"
 #include "../shared/qssgeneralexception.h"
-#include "../shared/qssquery.h"
 MainWindow::MainWindow(QWidget *parent) : QMainWindow(parent), ui(new Ui::MainWindow)
 {
 	ui->setupUi(this);
@@ -164,7 +163,8 @@ void MainWindow::lineEditReturnPressed()
 		[&, q]()
 		{
 			SqliteSearch searcher(db);
-			return searcher.search(QSSQuery::build(q));
+			this->currentQuery = QSSQuery::build(q);
+			return searcher.search(this->currentQuery);
 		});
 	searchWatcher.setFuture(searchFuture);
 }
@@ -215,8 +215,22 @@ void MainWindow::makePdfPreview()
 	processedPdfPreviews = 0;
 	QString scaleText = ui->comboScale->currentText();
 	scaleText.chop(1);
+
+	QVector<QString> wordsToHighlight;
+	for(const Token &token : this->currentQuery.getTokens())
+	{
+		if(token.type == FILTER_CONTENT_CONTAINS)
+		{
+			auto splitted = token.value.split(" ");
+			for(QString &str : splitted)
+			{
+				wordsToHighlight.append(str);
+			}
+		}
+	}
 	PdfWorker worker;
-	this->pdfWorkerWatcher.setFuture(worker.generatePreviews(this->pdfSearchResults, scaleText.toInt() / 100.));
+	this->pdfWorkerWatcher.setFuture(
+		worker.generatePreviews(this->pdfSearchResults, wordsToHighlight, scaleText.toInt() / 100.));
 	ui->pdfProcessBar->setMaximum(this->pdfWorkerWatcher.progressMaximum());
 	ui->pdfProcessBar->setMinimum(this->pdfWorkerWatcher.progressMinimum());
 }
diff --git a/gui/mainwindow.h b/gui/mainwindow.h
index 72a167d..ad9aa50 100644
--- a/gui/mainwindow.h
+++ b/gui/mainwindow.h
@@ -9,6 +9,7 @@
 #include <QFutureWatcher>
 #include <QSqlDatabase>
 #include "pdfworker.h"
+#include "../shared/qssquery.h"
 namespace Ui
 {
 class MainWindow;
@@ -41,6 +42,7 @@ class MainWindow : public QMainWindow
 	unsigned int processedPdfPreviews;
 	void handleSearchResults(const QVector<SearchResult> &results);
 	void handleSearchError(QString error);
+	QSSQuery currentQuery;
   private slots:
 	void lineEditReturnPressed();
 	void treeSearchItemActivated(QTreeWidgetItem *item, int i);
diff --git a/gui/pdfworker.cpp b/gui/pdfworker.cpp
index 9fe7270..cd708d6 100644
--- a/gui/pdfworker.cpp
+++ b/gui/pdfworker.cpp
@@ -6,6 +6,7 @@
 #include <QMutexLocker>
 #include <QtConcurrent/QtConcurrent>
 #include <QtConcurrent/QtConcurrentMap>
+#include <QPainter>
 #include <atomic>
 #include "pdfworker.h"
 
@@ -19,12 +20,13 @@ struct Renderer
 	QHash<QString, Poppler::Document *> documentcache;
 	qsizetype maxTotalPreviewImageMemUsage;
 	std::atomic<qsizetype> currentTotalPreviewImageMemUsage{0};
-
-	Renderer(double scaleX, double scaleY, qsizetype maxPreviewImageMemUsage)
+	QVector<QString> wordsToHighlight;
+	Renderer(double scaleX, double scaleY, qsizetype maxPreviewImageMemUsage, QVector<QString> wordsToHighlight)
 	{
 		this->scaleX = scaleX;
 		this->scaleY = scaleY;
 		this->maxTotalPreviewImageMemUsage = maxPreviewImageMemUsage;
+		this->wordsToHighlight = wordsToHighlight;
 	}
 
 	/*we need this one because std::atomic has none, but this is only a functor for
@@ -34,6 +36,7 @@ struct Renderer
 		this->scaleX = o.scaleX;
 		this->scaleY = o.scaleY;
 		this->maxTotalPreviewImageMemUsage = o.maxTotalPreviewImageMemUsage;
+		this->wordsToHighlight = o.wordsToHighlight;
 	}
 
 	~Renderer()
@@ -79,6 +82,16 @@ struct Renderer
 			}
 			Poppler::Page *pdfPage = doc->page(p);
 			QImage img = pdfPage->renderToImage(scaleX, scaleY);
+			for(QString &word : wordsToHighlight)
+			{
+				QList<QRectF> rects = pdfPage->search(word);
+				for(QRectF &rect : rects)
+				{
+					QPainter painter(&img);
+					painter.scale(scaleX / 72.0, scaleY / 72.0);
+					painter.fillRect(rect, QColor(255, 255, 0, 64));
+				}
+			}
 			result.previewImage = img;
 			currentTotalPreviewImageMemUsage += img.sizeInBytes();
 		}
@@ -86,7 +99,8 @@ struct Renderer
 	}
 };
 
-QFuture<PdfPreview> PdfWorker::generatePreviews(const QVector<SearchResult> paths, double scalefactor)
+QFuture<PdfPreview> PdfWorker::generatePreviews(const QVector<SearchResult> paths, QVector<QString> wordsToHighlight,
+												double scalefactor)
 {
 	QVector<PdfPreview> previews;
 
@@ -106,5 +120,5 @@ QFuture<PdfPreview> PdfWorker::generatePreviews(const QVector<SearchResult> path
 
 	QSettings setting;
 	qsizetype maxPreviewImageMemUsage = setting.value("maxpreviewimagesmemory", 1024 * 1024 * 1024).toLongLong();
-	return QtConcurrent::mapped(previews, Renderer(scaleX, scaleY, maxPreviewImageMemUsage));
+	return QtConcurrent::mapped(previews, Renderer(scaleX, scaleY, maxPreviewImageMemUsage, wordsToHighlight));
 }
diff --git a/gui/pdfworker.h b/gui/pdfworker.h
index efb5900..434d4d6 100644
--- a/gui/pdfworker.h
+++ b/gui/pdfworker.h
@@ -16,7 +16,8 @@ class PdfWorker : public QObject
 {
 	Q_OBJECT
   public:
-	QFuture<PdfPreview> generatePreviews(const QVector<SearchResult> paths, double scalefactor);
+	QFuture<PdfPreview> generatePreviews(const QVector<SearchResult> paths, QVector<QString> wordsToHighlight,
+										 double scalefactor);
 };
 
 #endif // PDFWORKER_H