aboutsummaryrefslogtreecommitdiff
path: root/qdeduper
diff options
context:
space:
mode:
Diffstat (limited to 'qdeduper')
-rw-r--r--qdeduper/imageitem.cpp1
-rw-r--r--qdeduper/mingui.cpp69
-rw-r--r--qdeduper/mingui.hpp21
-rw-r--r--qdeduper/sigdb_qt.cpp16
-rw-r--r--qdeduper/sigdb_qt.hpp3
5 files changed, 96 insertions, 14 deletions
diff --git a/qdeduper/imageitem.cpp b/qdeduper/imageitem.cpp
index f18d32c..9ee7c6f 100644
--- a/qdeduper/imageitem.cpp
+++ b/qdeduper/imageitem.cpp
@@ -9,7 +9,6 @@
#include <QFontMetrics>
#include <QPainter>
#include <QLocale>
-#include <qnamespace.h>
#define DEBUGPAINT 0
diff --git a/qdeduper/mingui.cpp b/qdeduper/mingui.cpp
index ba802c7..f93fda3 100644
--- a/qdeduper/mingui.cpp
+++ b/qdeduper/mingui.cpp
@@ -126,11 +126,11 @@ DeduperMainWindow::DeduperMainWindow()
auto &k = keys[i];
QAction *a = new QAction();
a->setShortcut(QKeySequence(k));
- QObject::connect(a, &QAction::triggered, [this, i](){this->mark_toggle(i);});
+ QObject::connect(a, &QAction::triggered, std::bind(&DeduperMainWindow::mark_toggle, this, i));
selhk.push_back(a);
QAction *sa = new QAction();
sa->setShortcut(QKeySequence(Qt::Modifier::SHIFT | k));
- QObject::connect(sa, &QAction::triggered, [this, i](){this->mark_all_but(i);});
+ QObject::connect(sa, &QAction::triggered, std::bind(&DeduperMainWindow::mark_all_but, this, i));
selhk.push_back(a);
}
this->addActions(selhk);
@@ -220,7 +220,31 @@ void DeduperMainWindow::setup_menu()
this->addAction(loadlist);
file->addSeparator();
- file->addAction("Search for Image...");
+ QAction *search_img = file->addAction("Search for Image...");
+ QObject::connect(search_img, &QAction::triggered, [this]{
+ QString fpath = QFileDialog::getOpenFileName(this, "Select Image", QString(), "Image file");
+ if (fpath.isNull()) return;
+ auto sim = this->sdb->search_file(qstring_to_path(fpath));
+ if (sim.empty())
+ {
+ this->sb->showMessage("No similar image found.", 2000);
+ return;
+ }
+ this->vm = ViewMode::view_searchresult;
+ std::vector<fs::path> ps;
+ std::map<std::pair<size_t, size_t>, double> dm;
+ for (size_t i = 0; i < sim.size(); ++i)
+ {
+ auto &s = sim[i];
+ ps.push_back(this->sdb->get_image_path(s.first));
+ dm[std::make_pair(0, i)] = s.second;
+ }
+ this->show_images(ps);
+ this->update_distances(dm);
+ this->sb->showMessage("Use next group / previous group to go back.");
+ this->permamsg->setText("Viewing image search result");
+ });
+ menuact["search_image"] = search_img;
file->addSeparator();
file->addAction("Preferences...");
file->addAction("Exit");
@@ -230,6 +254,7 @@ void DeduperMainWindow::setup_menu()
menuact["next_group"] = nxtgrp;
nxtgrp->setShortcut(QKeySequence(Qt::Key::Key_M));
QObject::connect(nxtgrp, &QAction::triggered, [this] {
+ if (this->vm == ViewMode::view_searchresult) { this->show_group(curgroup); return; }
if (this->sdb && curgroup + 1 < this->sdb->num_groups())
this->show_group(++curgroup);
});
@@ -240,6 +265,7 @@ void DeduperMainWindow::setup_menu()
menuact["prev_group"] = prvgrp;
prvgrp->setShortcut(QKeySequence(Qt::Key::Key_Z));
QObject::connect(prvgrp, &QAction::triggered, [this] {
+ if (this->vm == ViewMode::view_searchresult) { this->show_group(curgroup); return; }
if (this->sdb && curgroup > 0)
this->show_group(--curgroup);
});
@@ -306,14 +332,21 @@ void DeduperMainWindow::update_actions()
menuact["save_db"]->setEnabled(false);
menuact["load_list"]->setEnabled(false);
menuact["save_list"]->setEnabled(false);
+ menuact["search_image"]->setEnabled(false);
return;
}
menuact["skip_group"]->setEnabled(true);
menuact["prev_group"]->setEnabled(curgroup > 0);
menuact["next_group"]->setEnabled(curgroup + 1 < sdb->num_groups());
+ menuact["search_image"]->setEnabled(true);
menuact["save_db"]->setEnabled(true);
menuact["load_list"]->setEnabled(true);
menuact["save_list"]->setEnabled(true);
+ if (vm == ViewMode::view_searchresult)
+ {
+ menuact["next_group"]->setEnabled(true);
+ menuact["prev_group"]->setEnabled(true);
+ }
}
void DeduperMainWindow::show_images(const std::vector<fs::path> &fns)
@@ -350,7 +383,10 @@ void DeduperMainWindow::update_distances(const std::map<std::pair<size_t, size_t
ka = QKeySequence(keys[p.first.first]).toString();
if (p.first.second < keys.size())
kb = QKeySequence(keys[p.first.second]).toString();
- r += QString("%1 <-> %2: %3\n").arg(ka).arg(kb).arg(QString::number(p.second));
+ if (vm == ViewMode::view_normal)
+ r += QString("%1 <-> %2: %3\n").arg(ka).arg(kb).arg(QString::number(p.second));
+ else
+ r += QString("%1 : %2\n").arg(kb).arg(QString::number(p.second));;
}
infopanel->setText(r);
}
@@ -444,6 +480,7 @@ void DeduperMainWindow::scan_dirs(std::vector<std::pair<fs::path, bool>> paths)
}
int flsize = fs->file_list().size() - 1;
QMetaObject::invokeMethod(this->pd, [flsize, this] {this->pd->setMaximum(flsize);}, Qt::ConnectionType::QueuedConnection);
+ if (this->sdb) delete this->sdb;
this->sdb = new SignatureDB();
QObject::connect(this->sdb, &SignatureDB::image_scanned, this, [this](size_t n) {
static auto lt = std::chrono::steady_clock::now();
@@ -470,6 +507,7 @@ void DeduperMainWindow::scan_dirs(std::vector<std::pair<fs::path, bool>> paths)
this->pd->reset();
this->pd->close();
this->curgroup = 0;
+ this->vm = ViewMode::view_normal;
this->show_group(this->curgroup);
}, Qt::ConnectionType::QueuedConnection);
QObject::connect(pd, &QProgressDialog::canceled, [this] {
@@ -482,6 +520,7 @@ void DeduperMainWindow::show_group(size_t gid)
{
if (!sdb || gid >= sdb->num_groups())
return;
+ this->vm = ViewMode::view_normal;
auto g = sdb->get_group(gid);
current_set.clear();
std::for_each(g.begin(), g.end(), [this](auto id){current_set.push_back(sdb->get_image_path(id));});
@@ -493,6 +532,12 @@ void DeduperMainWindow::show_group(size_t gid)
void DeduperMainWindow::mark_toggle(size_t x)
{
+ if (vm == ViewMode::view_searchresult)
+ {
+ mark_none(false);
+ sb->showMessage("Marking images in search result is disabled.", 2000);
+ return;
+ }
if (x < marks.size())
{
marks[x] = !marks[x];
@@ -507,6 +552,12 @@ void DeduperMainWindow::mark_toggle(size_t x)
void DeduperMainWindow::mark_all_but(size_t x)
{
+ if (vm == ViewMode::view_searchresult)
+ {
+ mark_none(false);
+ sb->showMessage("Marking images in search result is disabled.", 2000);
+ return;
+ }
if (x < marks.size())
{
for (size_t i = 0; i < marks.size(); ++i)
@@ -524,6 +575,12 @@ void DeduperMainWindow::mark_all_but(size_t x)
void DeduperMainWindow::mark_all()
{
+ if (vm == ViewMode::view_searchresult)
+ {
+ mark_none(false);
+ sb->showMessage("Marking images in search result is disabled.", 2000);
+ return;
+ }
for (size_t i = 0; i < marks.size(); ++i)
{
marks[i] = true;
@@ -532,7 +589,7 @@ void DeduperMainWindow::mark_all()
mark_view_update();
}
-void DeduperMainWindow::mark_none()
+void DeduperMainWindow::mark_none(bool msg)
{
for (size_t i = 0; i < marks.size(); ++i)
{
@@ -540,7 +597,7 @@ void DeduperMainWindow::mark_none()
if (marked.find(current_set[i]) != marked.end())
marked.erase(marked.find(current_set[i]));
}
- mark_view_update();
+ mark_view_update(msg);
}
void DeduperMainWindow::mark_view_update(bool update_msg)
diff --git a/qdeduper/mingui.hpp b/qdeduper/mingui.hpp
index e4b1c3f..9f3ff2a 100644
--- a/qdeduper/mingui.hpp
+++ b/qdeduper/mingui.hpp
@@ -28,6 +28,12 @@ class ImageItemDelegate;
namespace fs = std::filesystem;
+enum ViewMode
+{
+ view_normal,
+ view_searchresult
+};
+
class DeduperMainWindow : public QMainWindow
{
Q_OBJECT
@@ -46,12 +52,13 @@ private:
SignatureDB *sdb = nullptr;
FileScanner *fsc = nullptr;
- std::size_t curgroup;
+ size_t curgroup;
+ ViewMode vm;
bool nohotkeywarn;
- void mark_toggle(std::size_t x);
- void mark_all_but(std::size_t x);
+ void mark_toggle(size_t x);
+ void mark_all_but(size_t x);
void mark_all();
- void mark_none();
+ void mark_none(bool msg = true);
void mark_view_update(bool update_msg = true);
fs::path::string_type common_prefix(const std::vector<fs::path> &fns);
std::vector<bool> marks;
@@ -65,8 +72,8 @@ public:
void setup_menu();
void show_images(const std::vector<fs::path> &fns);
- void update_distances(const std::map<std::pair<std::size_t, std::size_t>, double> &d);
- void update_viewstatus(std::size_t cur, std::size_t size);
+ void update_distances(const std::map<std::pair<size_t, size_t>, double> &d);
+ void update_viewstatus(size_t cur, size_t size);
void save_list();
void load_list();
@@ -78,7 +85,7 @@ public Q_SLOTS:
Q_SIGNALS:
void next();
void prev();
- void switch_group(std::size_t group);
+ void switch_group(size_t group);
};
#endif
diff --git a/qdeduper/sigdb_qt.cpp b/qdeduper/sigdb_qt.cpp
index 67cc3e6..733a198 100644
--- a/qdeduper/sigdb_qt.cpp
+++ b/qdeduper/sigdb_qt.cpp
@@ -97,6 +97,22 @@ void SignatureDB::interrupt_scan()
sdb->populate_interrupt();
}
+std::vector<std::pair<size_t, double>> SignatureDB::search_file(const fs::path &files)
+{
+ populate_cfg_t pcfg = {
+ 3,
+ 3,
+ cfg_full,
+ cfg_subslice,
+ 0.3,
+ nullptr,
+ 0
+ };
+ if(sdb)
+ return sdb->search_image(files, pcfg, false);
+ return {};
+}
+
size_t SignatureDB::num_groups()
{
return groups.size();
diff --git a/qdeduper/sigdb_qt.hpp b/qdeduper/sigdb_qt.hpp
index 112ffa9..772c264 100644
--- a/qdeduper/sigdb_qt.hpp
+++ b/qdeduper/sigdb_qt.hpp
@@ -33,6 +33,9 @@ public:
void scan_files(const std::vector<fs::path> &files, int njobs);
void interrupt_scan();
+
+ std::vector<std::pair<size_t, double>> search_file(const fs::path &files);
+
size_t num_groups();
std::vector<size_t> get_group(size_t gid);
std::map<std::pair<size_t, size_t>, double> group_distances(size_t gid);
generated by cgit v1.2.3 (git 2.39.1) at 2024-12-26 19:50:36 +0000