Updated empty gallery to use deepest folder as subject label

Scott Klum
1 parent a2c6d47d
Showing 3 changed files with 27 additions and 4 deletions
openbr/core/utility.cpp
openbr/core/utility.h
openbr/plugins/gallery/empty.cpp
@@ -19,6 +19,25 @@ QStringList br::getFiles(QDir dir, bool recursive)
     return files;
 }
+QList<QPair<QString,QStringList> > br::getFilesWithLabels(QDir dir)
+{
+    dir = QDir(dir.canonicalPath());
+
+    QStringList files;
+    foreach (const QString &file, QtUtils::naturalSort(dir.entryList(QDir::Files)))
+        files.append(dir.absoluteFilePath(file));
+
+    QList<QPair<QString,QStringList> > filesWithLabels;
+    filesWithLabels.append(QPair<QString,QStringList>(dir.dirName(),files));
+
+    foreach (const QString &folder, QtUtils::naturalSort(dir.entryList(QDir::Dirs | QDir::NoDotAndDotDot))) {
+        QDir subdir(dir);
+        bool success = subdir.cd(folder); if (!success) qFatal("cd failure.");
+        filesWithLabels.append(getFilesWithLabels(subdir));
+    }
+    return filesWithLabels;
+}
+
 QStringList br::getFiles(const QString &regexp)
 {
     QFileInfo fileInfo(regexp);
@@ -11,6 +11,7 @@ namespace br
 {
 BR_EXPORT QStringList getFiles(QDir dir, bool recursive);
+BR_EXPORT QList<QPair<QString,QStringList> > getFilesWithLabels(QDir dir);
 BR_EXPORT QStringList getFiles(const QString &regexp);
 } // namespace br
@@ -20,6 +20,8 @@
 #include <openbr/core/qtutils.h>
 #include <openbr/core/utility.h>
+typedef QPair<QString,QStringList> FilesWithLabel;
+
 namespace br
 {
@@ -113,10 +115,11 @@ class EmptyGallery : public Gallery
     static TemplateList getTemplates(const QDir &dir)
     {
-        const QStringList files = getFiles(dir, true);
-        TemplateList templates; templates.reserve(files.size());
-        foreach (const QString &file, files)
-            templates.append(File(file, dir.dirName()));
+        const QList<FilesWithLabel> filesWithLabels = getFilesWithLabels(dir);
+        TemplateList templates; templates.reserve(filesWithLabels.size());
+        foreach (const FilesWithLabel &filesWithLabel, filesWithLabels)
+            foreach (const QString &file, filesWithLabel.second)
+                templates.append(File(file, filesWithLabel.first));
         return templates;
     }
 };