Peter M. Groen / openbr

Commit 6e3bec09fa8ef387f6a146fe80af45b0015b246a

Authored by bhklein
1 parent 3fcbaa15

group images by metadata key in subdirectories in WriteTransform

Inline Side-by-side
Showing 1 changed file with 19 additions and 3 deletions
openbr/plugins/io/write.cpp
  View file @6e3bec0
@@ -10,6 +10,11 @@ namespace br @@ -10,6 +10,11 @@ namespace br
10 * \ingroup transforms 10 * \ingroup transforms
11 * \brief Write all mats to disk as images. 11 * \brief Write all mats to disk as images.
12 * \author Brendan Klare \cite bklare 12 * \author Brendan Klare \cite bklare
  13 + * \br_property QString outputDirectory Top level directory to write images to.
  14 + * \br_property QString underscore
  15 + * \br_property QString imgExtension Extension to save images as.
  16 + * \br_property int padding Zero pad the image filenames.
  17 + * \br_property QString subDir Optional group images by a metadata key in subdirectories.
13 */ 18 */
14 class WriteTransform : public TimeVaryingTransform 19 class WriteTransform : public TimeVaryingTransform
15 { 20 {
@@ -18,12 +23,15 @@ class WriteTransform : public TimeVaryingTransform @@ -18,12 +23,15 @@ class WriteTransform : public TimeVaryingTransform
18 Q_PROPERTY(QString underscore READ get_underscore WRITE set_underscore RESET reset_underscore STORED false) 23 Q_PROPERTY(QString underscore READ get_underscore WRITE set_underscore RESET reset_underscore STORED false)
19 Q_PROPERTY(QString imgExtension READ get_imgExtension WRITE set_imgExtension RESET reset_imgExtension STORED false) 24 Q_PROPERTY(QString imgExtension READ get_imgExtension WRITE set_imgExtension RESET reset_imgExtension STORED false)
20 Q_PROPERTY(int padding READ get_padding WRITE set_padding RESET reset_padding STORED false) 25 Q_PROPERTY(int padding READ get_padding WRITE set_padding RESET reset_padding STORED false)
  26 + Q_PROPERTY(QString subDir READ get_subDir WRITE set_subDir RESET reset_subDir STORED false)
21 BR_PROPERTY(QString, outputDirectory, "Temp") 27 BR_PROPERTY(QString, outputDirectory, "Temp")
22 BR_PROPERTY(QString, underscore, "") 28 BR_PROPERTY(QString, underscore, "")
23 BR_PROPERTY(QString, imgExtension, "jpg") 29 BR_PROPERTY(QString, imgExtension, "jpg")
24 BR_PROPERTY(int, padding, 5) 30 BR_PROPERTY(int, padding, 5)
  31 + BR_PROPERTY(QString, subDir, "")
25 32
26 int cnt; 33 int cnt;
  34 + QMap<QString, int> numImages;
27 35
28 void init() { 36 void init() {
29 cnt = 0; 37 cnt = 0;
@@ -34,10 +42,18 @@ class WriteTransform : public TimeVaryingTransform @@ -34,10 +42,18 @@ class WriteTransform : public TimeVaryingTransform
34 void projectUpdate(const Template &src, Template &dst) 42 void projectUpdate(const Template &src, Template &dst)
35 { 43 {
36 dst = src; 44 dst = src;
37 - QString path = QString("%1/image%2%3.%4").arg(outputDirectory).arg(cnt++, padding, 10, QChar('0')).arg(underscore.isEmpty() ? "" : "_" + underscore).arg(imgExtension);  
38 - OpenCVUtils::saveImage(dst.m(), path); 45 + if (!subDir.isEmpty()) {
  46 + QString dir = src.file.get<QString>(subDir, "Temp");
  47 + QString path = QString("%1/%2/").arg(outputDirectory, dir);
  48 + int value = numImages.value(dir, 0);
  49 + path += QString("%1_%2.%3").arg(src.file.get<QString>(subDir, "Image")).arg(value, padding, 10, QChar('0')).arg(imgExtension);
  50 + numImages[dir] = ++value;
  51 + OpenCVUtils::saveImage(dst.m(), path);
  52 + } else {
  53 + QString path = QString("%1/image%2%3.%4").arg(outputDirectory).arg(cnt++, padding, 10, QChar('0')).arg(underscore.isEmpty() ? "" : "_" + underscore).arg(imgExtension);
  54 + OpenCVUtils::saveImage(dst.m(), path);
  55 + }
39 } 56 }
40 -  
41 }; 57 };
42 58
43 BR_REGISTER(Transform, WriteTransform) 59 BR_REGISTER(Transform, WriteTransform)