Working subject viewers

Scott Klum
1 parent 42ae191d
Showing 5 changed files with 106 additions and 9 deletions
openbr/gui/faceviewer.cpp
openbr/gui/subjectviewer.cpp
openbr/gui/subjectviewer.h
openbr/gui/subjectviewergrid.cpp
openbr/gui/subjectviewergrid.h
@@ -73,8 +73,7 @@ void FaceViewer::setFile(const File &amp;file_)
         landmarks.append(QPointF());
     nearestLandmark = -1;
-    FaceViewer::refreshImage();
-    //QtConcurrent::run(this, &FaceViewer::refreshImage);
+    QtConcurrent::run(this, &FaceViewer::refreshImage);
 }
 void FaceViewer::refreshImage()
 #include "subjectviewer.h"
-SubjectViewer::SubjectViewer()
+using namespace br;
+
+SubjectViewer::SubjectViewer(QWidget *parent)
+    : TemplateViewer(parent)
+{
+    files = FileList();
+    currentIndex = 0;
+}
+
+void SubjectViewer::setFiles(const FileList &files)
+{
+    if (files.isEmpty()) return;
+
+    currentIndex = 0;
+    this->files = files;
+    TemplateViewer::setFile(this->files[currentIndex]);
+}
+
+void SubjectViewer::wheelEvent(QWheelEvent *event)
 {
+    QPoint numDegrees = event->angleDelta();
+
+    if (numDegrees.y() < 0 && currentIndex > 0)
+        TemplateViewer::setFile(this->files[--currentIndex]);
+     else if (numDegrees.y() > 0 && currentIndex < files.size()-1)
+        TemplateViewer::setFile(this->files[++currentIndex]);
+
+    event->accept();
 }
 #ifndef SUBJECTVIEWER_H
 #define SUBJECTVIEWER_H
-class SubjectViewer : public FaceViewer
+#include <QList>
+
+#include <openbr/openbr_plugin.h>
+#include <openbr/gui/templateviewer.h>
+
+namespace br
+{
+
+class BR_EXPORT SubjectViewer : public TemplateViewer
 {
+    FileList files;
+    int currentIndex;
+
 public:
-    SubjectViewer();
+    explicit SubjectViewer(QWidget *parent = 0);
+
+public slots:
+    void setFiles(const FileList &files);
+
+protected slots:
+    void wheelEvent(QWheelEvent *);
 };
+}
+
 #endif // SUBJECTVIEWER_H
 #include "subjectviewergrid.h"
+using namespace br;
+
 SubjectViewerGrid::SubjectViewerGrid(QWidget *parent) :
-    TemplateViewerGrid(parent)
+    QWidget(parent)
+{
+    setLayout(&gridLayout);
+}
+
+void SubjectViewerGrid::setFiles(const QList<FileList> &files)
 {
+    const int size = std::max(1, (int)ceil(sqrt((float)files.size())));
+    while (subjectViewers.size() < size*size) {
+        subjectViewers.append(QSharedPointer<SubjectViewer>(new SubjectViewer()));
+        connect(subjectViewers.last().data(), SIGNAL(newInput(br::File)), this, SIGNAL(newInput(br::File)));
+        connect(subjectViewers.last().data(), SIGNAL(newInput(QImage)), this, SIGNAL(newInput(QImage)));
+        connect(subjectViewers.last().data(), SIGNAL(newMousePoint(QPointF)), this, SIGNAL(newMousePoint(QPointF)));
+        connect(subjectViewers.last().data(), SIGNAL(selectedInput(br::File)), this, SIGNAL(selectedInput(br::File)));
+    }
+
+    { // Clear layout
+        QLayoutItem *child;
+        while ((child = gridLayout.takeAt(0)) != 0)
+            child->widget()->setVisible(false);
+    }
+
+    for (int i=0; i<subjectViewers.size(); i++) {
+        if (i < size*size) {
+            gridLayout.addWidget(subjectViewers[i].data(), i/size, i%size, 1, 1);
+            subjectViewers[i]->setVisible(true);
+        }
+
+        if (i < files.size()) {
+            subjectViewers[i]->setFiles(files[i]);
+        } else {
+            subjectViewers[i]->setDefaultText("<b>"+ (size > 1 ? QString() : QString("Drag Photo or Folder Here")) +"</b>");
+            subjectViewers[i]->setFiles(FileList());
+        }
+
+        subjectViewers[i]->setEditable(files.size() == 1);
+    }
 }
 #ifndef SUBJECTVIEWERGRID_H
 #define SUBJECTVIEWERGRID_H
-class SubjectViewerGrid : public TemplateViewerGrid
+#include <QObject>
+#include <QGridLayout>
+#include <QSharedPointer>
+
+#include <openbr/openbr_plugin.h>
+
+#include "subjectviewer.h"
+
+namespace br
+{
+
+class BR_EXPORT SubjectViewerGrid : public QWidget
 {
     Q_OBJECT
+
+    QGridLayout gridLayout;
+    QList< QSharedPointer<SubjectViewer> >subjectViewers;
+
 public:
     explicit SubjectViewerGrid(QWidget *parent = 0);
-signals:
-
 public slots:
+    void setFiles(const QList<br::FileList> &file);
 };
+}
+
 #endif // SUBJECTVIEWERGRID_H