1 files changed, 41 insertions, 0 deletions
diff --git a/qdeduper/sigdb_qt.hpp b/qdeduper/sigdb_qt.hpp
new file mode 100644
index 0000000..8a5178a
--- /dev/null
+++ b/qdeduper/sigdb_qt.hpp
@@ -0,0 +1,41 @@
+#ifndef SIGDB_QT_HPP
+#define SIGDB_QT_HPP
+
+#include <filesystem>
+#include <map>
+#include <unordered_map>
+#include <utility>
+#include <vector>
+
+#include <QObject>
+
+#include "signature_db.hpp"
+
+namespace fs = std::filesystem;
+
+class SignatureDB : public QObject
+{
+    Q_OBJECT
+private:
+    signature_db *sdb;
+    std::unordered_map<size_t, fs::path> fmap;
+    std::unordered_map<fs::path, size_t> frmap;
+    std::map<std::pair<size_t, size_t>, double> distmap;
+    std::vector<std::vector<size_t>> groups;
+public:
+    SignatureDB();
+    SignatureDB(const fs::path& dbpath);
+    ~SignatureDB();
+
+    void scan_files(const std::vector<fs::path> &files, int njobs);
+    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);
+
+    fs::path get_image_path(size_t id);
+    size_t get_path_id(const fs::path& p);
+Q_SIGNALS:
+    void image_scanned(size_t n);
+};
+
+#endif