Kinect demo update

+ visualization type can be selected on run-time

Change-Id: I2f57ad18a261b264b3b0b958d2d367a1fd2795ef
Change-Id: I2f57ad18a261b264b3b0b958d2d367a1fd2795ef
Reviewed-by: Miikka Heikkinen <miikka.heikkinen@digia.com>

3 files changed, 215 insertions, 135 deletions

diff --git a/tests/kinectsurface/main.cpp b/tests/kinectsurface/main.cpp
index bc82a38e..f428ba2d 100644
--- a/tests/kinectsurface/main.cpp
+++ b/tests/kinectsurface/main.cpp
@@ -37,25 +37,37 @@ int main(int argc, char **argv)
     QHBoxLayout *hLayout = new QHBoxLayout(widget);
     QVBoxLayout *vLayout = new QVBoxLayout();
 
-#if defined(USE_SCATTER)
-    Q3DScatter *surface = new Q3DScatter();
-#elif defined(USE_BARS)
-    Q3DBars *surface = new Q3DBars();
-#else
     Q3DSurface *surface = new Q3DSurface();
-#endif
+    Q3DScatter *scatter = new Q3DScatter();
+    Q3DBars *bars = new Q3DBars();
 
     QSize screenSize = surface->screen()->size();
 
-    QWidget *container = QWidget::createWindowContainer(surface);
-    container->setMinimumSize(QSize(screenSize.width() / 2, screenSize.height() / 2));
-    container->setMaximumSize(screenSize);
-    container->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
-    container->setFocusPolicy(Qt::StrongFocus);
-
-    widget->setWindowTitle(QStringLiteral("Surface mapping from Kinect depth data"));
-
-    hLayout->addWidget(container, 1);
+    QWidget *containerSurface = QWidget::createWindowContainer(surface);
+    containerSurface->setMinimumSize(QSize(screenSize.width() / 2, screenSize.height() / 2));
+    containerSurface->setMaximumSize(screenSize);
+    containerSurface->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
+    containerSurface->setFocusPolicy(Qt::StrongFocus);
+
+    QWidget *containerScatter = QWidget::createWindowContainer(scatter);
+    containerScatter->setMinimumSize(QSize(screenSize.width() / 2, screenSize.height() / 2));
+    containerScatter->setMaximumSize(screenSize);
+    containerScatter->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
+    containerScatter->setFocusPolicy(Qt::StrongFocus);
+    containerScatter->setVisible(false);
+
+    QWidget *containerBars = QWidget::createWindowContainer(bars);
+    containerBars->setMinimumSize(QSize(screenSize.width() / 2, screenSize.height() / 2));
+    containerBars->setMaximumSize(screenSize);
+    containerBars->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
+    containerBars->setFocusPolicy(Qt::StrongFocus);
+    containerBars->setVisible(false);
+
+    widget->setWindowTitle(QStringLiteral("Visualization from Kinect depth data"));
+
+    hLayout->addWidget(containerSurface, 1);
+    hLayout->addWidget(containerScatter, 1);
+    hLayout->addWidget(containerBars, 1);
     hLayout->addLayout(vLayout);
 
     QPushButton *startButton = new QPushButton(widget);
@@ -71,6 +83,12 @@ int main(int argc, char **argv)
     resolutionBox->addItem(QStringLiteral("Max")); // Comment this out if demo machine is low-perf
     resolutionBox->setCurrentIndex(0);
 
+    QComboBox *modeBox = new QComboBox(widget);
+    modeBox->addItem(QStringLiteral("Surface Plot"));
+    modeBox->addItem(QStringLiteral("Scatter Chart"));
+    modeBox->addItem(QStringLiteral("Bar Chart"));
+    modeBox->setCurrentIndex(0);
+
     QSlider *distanceSlider = new QSlider(Qt::Horizontal, widget);
     distanceSlider->setTickInterval(10);
     distanceSlider->setTickPosition(QSlider::TicksBelow);
@@ -78,7 +96,6 @@ int main(int argc, char **argv)
     distanceSlider->setValue(50);
     distanceSlider->setMaximum(200);
 
-#if !defined(USE_SCATTER) && !defined(USE_BARS)
     QLinearGradient gradientOne(0, 0, 200, 1);
     gradientOne.setColorAt(0.0, Qt::black);
     gradientOne.setColorAt(0.33, Qt::blue);
@@ -109,7 +126,6 @@ int main(int argc, char **argv)
     gradientTwoButton->setIcon(QIcon(pm));
     gradientTwoButton->setIconSize(QSize(200, 24));
     gradientTwoButton->setToolTip(QStringLiteral("Colors: Highlight Foreground"));
-#endif
 
     QTextEdit *status = new QTextEdit(QStringLiteral("<b>Ready</b><br>"), widget);
     status->setReadOnly(true);
@@ -118,30 +134,32 @@ int main(int argc, char **argv)
     vLayout->addWidget(stopButton);
     vLayout->addWidget(new QLabel(QStringLiteral("Change resolution")));
     vLayout->addWidget(resolutionBox);
+    vLayout->addWidget(new QLabel(QStringLiteral("Change visualization type")));
+    vLayout->addWidget(modeBox);
     vLayout->addWidget(new QLabel(QStringLiteral("Adjust far distance")));
     vLayout->addWidget(distanceSlider);
-#if !defined(USE_SCATTER) && !defined(USE_BARS)
     vLayout->addWidget(new QLabel(QStringLiteral("Change color scheme")));
     vLayout->addWidget(gradientOneButton);
     vLayout->addWidget(gradientTwoButton);
-#endif
     vLayout->addWidget(status, 1, Qt::AlignBottom);
 
     widget->show();
 
-    SurfaceData *datagen = new SurfaceData(surface, status);
+    SurfaceData *datagen = new SurfaceData(surface, scatter, bars, status);
+    ContainerChanger changer(containerSurface, containerScatter, containerBars,
+                             gradientOneButton, gradientTwoButton);
 
     QObject::connect(startButton, &QPushButton::clicked, datagen, &SurfaceData::start);
     QObject::connect(stopButton, &QPushButton::clicked, datagen, &SurfaceData::stop);
     QObject::connect(distanceSlider, &QSlider::valueChanged, datagen, &SurfaceData::setDistance);
     QObject::connect(resolutionBox, SIGNAL(activated(int)), datagen, SLOT(setResolution(int)));
+    QObject::connect(modeBox, SIGNAL(activated(int)), &changer, SLOT(changeContainer(int)));
+    QObject::connect(modeBox, SIGNAL(activated(int)), datagen, SLOT(changeMode(int)));
     QObject::connect(status, &QTextEdit::textChanged, datagen, &SurfaceData::scrollDown);
-#if !defined(USE_SCATTER) && !defined(USE_BARS)
     QObject::connect(gradientOneButton, &QPushButton::clicked, datagen,
                      &SurfaceData::useGradientOne);
     QObject::connect(gradientTwoButton, &QPushButton::clicked, datagen,
                      &SurfaceData::useGradientTwo);
-#endif
 
     datagen->setDistance(distanceSlider->value());
 
diff --git a/tests/kinectsurface/surfacedata.cpp b/tests/kinectsurface/surfacedata.cpp
index 06d73ec1..66fc705b 100644
--- a/tests/kinectsurface/surfacedata.cpp
+++ b/tests/kinectsurface/surfacedata.cpp
@@ -20,62 +20,26 @@
 
 #include "surfacedata.h"
 #include "QKinectWrapper.h"
-#if defined(USE_SCATTER)
 #include <QtDataVisualization/QScatterDataProxy>
-#elif defined(USE_BARS)
 #include <QtDataVisualization/QBarDataProxy>
-#else
 #include <QtDataVisualization/QHeightMapSurfaceDataProxy>
-#endif
 #include <QtDataVisualization/Q3DValueAxis>
 #include <QScrollBar>
 #include <QSize>
 #include <QImage>
 
-#include <QDebug>
-
 QT_DATAVISUALIZATION_USE_NAMESPACE
 
-#if defined(USE_SCATTER)
-SurfaceData::SurfaceData(Q3DScatter *surface, QTextEdit *statusArea) :
-    m_surface(surface),
-    m_statusArea(statusArea),
-    m_resize(true),
-    m_resolution(QSize(80, 60))
-{
-    // Initialize scatter
-    m_surface->setTheme(QDataVis::ThemeStoneMoss);
-    m_surface->setObjectType(QDataVis::Dots, false);
-    m_surface->setShadowQuality(QDataVis::ShadowSoftLow);
-    m_surface->setCameraPosition(0.0, 85.0, 110);
-    m_surface->axisY()->setMax(255);
-    m_surface->axisX()->setMin(-m_resolution.width() / 2);
-    m_surface->axisX()->setMax(m_resolution.width() / 2);
-    m_surface->axisZ()->setMin(-m_resolution.height() / 2);
-    m_surface->axisZ()->setMax(m_resolution.height() / 2);
-#elif defined(USE_BARS)
-SurfaceData::SurfaceData(Q3DBars *surface, QTextEdit *statusArea) :
+SurfaceData::SurfaceData(Q3DSurface *surface, Q3DScatter *scatter, Q3DBars *bars,
+                         QTextEdit *statusArea) :
     m_surface(surface),
+    m_scatter(scatter),
+    m_bars(bars),
     m_statusArea(statusArea),
     m_resize(true),
-    m_resolution(QSize(80, 60))
-{
-    // Initialize scatter
-    m_surface->setTheme(QDataVis::ThemeQt);
-    m_surface->setBarType(QDataVis::Bars, true);
-#if 1
-    m_surface->setShadowQuality(QDataVis::ShadowLow);
-#else
-    m_surface->setBarSpacing(QSizeF(0.0, 0.0));
-#endif
-    m_surface->setCameraPosition(0.0, 75.0);
-    m_surface->valueAxis()->setMax(255);
-#else
-SurfaceData::SurfaceData(Q3DSurface *surface, QTextEdit *statusArea) :
-    m_surface(surface),
-    m_statusArea(statusArea),
-    m_resize(true),
-    m_resolution(QSize(320, 240))
+    m_resolution(QSize(320, 240)),
+    m_resolutionLevel(0),
+    m_mode(Surface)
 {
     // Initialize surface
     m_surface->setTheme(QDataVis::ThemeIsabelle);
@@ -84,17 +48,41 @@ SurfaceData::SurfaceData(Q3DSurface *surface, QTextEdit *statusArea) :
     gradient.setColorAt(0.33, Qt::blue);
     gradient.setColorAt(0.67, Qt::red);
     gradient.setColorAt(1.0, Qt::yellow);
+    m_surface->setSelectionMode(QDataVis::ModeNone);
+    m_surface->setGridVisible(false);
     m_surface->setGradient(gradient);
     m_surface->axisY()->setMax(255);
     m_surface->setSurfaceGridEnabled(false);
     m_surface->setBackgroundVisible(false);
     m_surface->setSmoothSurfaceEnabled(false);
     m_surface->setActiveDataProxy(new QHeightMapSurfaceDataProxy());
-#endif
 
-    // Common initializers
-    m_surface->setSelectionMode(QDataVis::ModeNone);
-    m_surface->setGridVisible(false);
+    // Initialize scatter
+    m_scatter->setTheme(QDataVis::ThemeStoneMoss);
+    m_scatter->setSelectionMode(QDataVis::ModeNone);
+    m_scatter->setGridVisible(false);
+    m_scatter->setObjectType(QDataVis::Dots, false);
+    m_scatter->setShadowQuality(QDataVis::ShadowSoftLow);
+    m_scatter->setCameraPosition(0.0, 85.0, 110);
+    m_scatter->axisY()->setMin(-128);
+    m_scatter->axisY()->setMax(128);
+    m_scatter->axisX()->setMin(-m_resolution.width() / 2);
+    m_scatter->axisX()->setMax(m_resolution.width() / 2);
+    m_scatter->axisZ()->setMin(-m_resolution.height() / 2);
+    m_scatter->axisZ()->setMax(m_resolution.height() / 2);
+
+    // Initialize bars
+    m_bars->setTheme(QDataVis::ThemeQt);
+    m_bars->setSelectionMode(QDataVis::ModeNone);
+    m_bars->setGridVisible(false);
+    m_bars->setBarType(QDataVis::Bars, true);
+#if 1
+    m_bars->setShadowQuality(QDataVis::ShadowLow);
+#else
+    m_bars->setBarSpacing(QSizeF(0.0, 0.0));
+#endif
+    m_bars->setCameraPosition(0.0, 75.0);
+    m_bars->valueAxis()->setMax(255);
 
     // Hide scroll bar
     m_statusArea->verticalScrollBar()->setVisible(false);
@@ -115,11 +103,11 @@ void SurfaceData::updateData()
     QImage depthMap = m_kinect.getDepth();
     if (m_resize) // Resize for better performance
         depthMap = depthMap.scaled(m_resolution);
-#if defined(USE_SCATTER) || defined(USE_BARS)
-    setData(depthMap);
-#else
-    static_cast<QHeightMapSurfaceDataProxy *>(m_surface->activeDataProxy())->setHeightMap(depthMap);
-#endif
+    if (m_mode != Surface)
+        setData(depthMap);
+    else
+        static_cast<QHeightMapSurfaceDataProxy *>(m_surface->activeDataProxy())->setHeightMap(
+                depthMap);
 }
 
 void SurfaceData::updateStatus(QKinect::KinectStatus status)
@@ -170,6 +158,7 @@ void SurfaceData::setDistance(int distance)
 
 void SurfaceData::setResolution(int selection)
 {
+    m_resolutionLevel = selection;
     switch (selection) {
     case 0: {
         m_resize = true;
@@ -192,17 +181,17 @@ void SurfaceData::setResolution(int selection)
         break;
     }
     };
-#if defined(USE_SCATTER)
-    m_resize = true;
-    m_resolution /= 4;
-    m_surface->axisX()->setMin(-m_resolution.width() / 2);
-    m_surface->axisX()->setMax(m_resolution.width() / 2);
-    m_surface->axisZ()->setMin(-m_resolution.height() / 2);
-    m_surface->axisZ()->setMax(m_resolution.height() / 2);
-#elif defined(USE_BARS)
-    m_resize = true;
-    m_resolution /= 4;
-#endif
+    if (m_mode == Scatter) {
+        m_resize = true;
+        m_resolution /= 4;
+        m_scatter->axisX()->setMin(-m_resolution.width() / 2);
+        m_scatter->axisX()->setMax(m_resolution.width() / 2);
+        m_scatter->axisZ()->setMin(-m_resolution.height() / 2);
+        m_scatter->axisZ()->setMax(m_resolution.height() / 2);
+    } else if (m_mode == Bars) {
+        m_resize = true;
+        m_resolution /= 4;
+    }
 
     m_statusArea->append(QString(QStringLiteral("<b>Resolution:</b> %1 x %2")).arg(
                              m_resolution.width()).arg(m_resolution.height()));
@@ -216,7 +205,6 @@ void SurfaceData::scrollDown()
     scrollbar->setValue(scrollbar->maximum());
 }
 
-#if !defined(USE_SCATTER) && !defined(USE_BARS)
 void SurfaceData::useGradientOne()
 {
     m_surface->setTheme(QDataVis::ThemeIsabelle);
@@ -239,7 +227,7 @@ void SurfaceData::useGradientTwo()
     m_surface->setGradient(gradient);
     m_statusArea->append(QStringLiteral("<b>Colors:</b> Highlight foreground"));
 }
-#else
+
 void SurfaceData::setData(const QImage &image)
 {
     QImage heightImage = image;
@@ -251,38 +239,101 @@ void SurfaceData::setData(const QImage &image)
     int bitCount = imageWidth * 4 * (imageHeight - 1);
     int widthBits = imageWidth * 4;
 
-#if defined(USE_SCATTER)
-    QScatterDataArray *dataArray = new QScatterDataArray;
-    dataArray->resize(imageHeight * imageWidth);
-    QScatterDataItem *ptrToDataArray = &dataArray->first();
+    if (m_mode == Scatter) {
+        QScatterDataArray *dataArray = new QScatterDataArray;
+        dataArray->resize(imageHeight * imageWidth);
+        QScatterDataItem *ptrToDataArray = &dataArray->first();
 
-    int limitsX = imageWidth / 2;
-    int limitsZ = imageHeight / 2;
-    qreal height = 0;
+        int limitsX = imageWidth / 2;
+        int limitsZ = imageHeight / 2;
+        qreal height = 0;
 
-    for (int i = -limitsZ; i < limitsZ; i++, bitCount -= widthBits) {
-        for (int j = -limitsX; j < limitsX; j++) {
-            height = qreal(bits[bitCount + ((j + limitsX) * 4)]);
-            if (height > 0) {
-                ptrToDataArray->setPosition(QVector3D(qreal(j), height, qreal(i)));
-                ptrToDataArray++;
+        for (int i = -limitsZ; i < limitsZ; i++, bitCount -= widthBits) {
+            for (int j = -limitsX; j < limitsX; j++) {
+                height = qreal(bits[bitCount + ((j + limitsX) * 4)]) - 128.0;
+                if (height > -128) {
+                    ptrToDataArray->setPosition(QVector3D(qreal(j), height, qreal(i)));
+                    ptrToDataArray++;
+                }
             }
         }
+
+        static_cast<QScatterDataProxy *>(m_scatter->activeDataProxy())->resetArray(dataArray);
+    } else {
+        QBarDataArray *dataArray = new QBarDataArray;
+        dataArray->reserve(imageHeight);
+        for (int i = imageHeight; i > 0; i--, bitCount -= widthBits) {
+            QBarDataRow *newRow = new QBarDataRow(imageWidth);
+            for (int j = 0; j < imageWidth; j++)
+                (*newRow)[j] = qreal(bits[bitCount + (j * 4)]);
+            *dataArray << newRow;
+        }
+
+        static_cast<QBarDataProxy *>(m_bars->activeDataProxy())->resetArray(dataArray);
     }
+}
 
-    static_cast<QScatterDataProxy *>(m_surface->activeDataProxy())->resetArray(dataArray);
-#elif defined(USE_BARS)
-    QBarDataArray *dataArray = new QBarDataArray;
-    dataArray->reserve(imageHeight);
-    for (int i = imageHeight; i > 0; i--, bitCount -= widthBits) {
-        QBarDataRow *newRow = new QBarDataRow(imageWidth);
-        for (int j = 0; j < imageWidth; j++)
-            (*newRow)[j] = qreal(bits[bitCount + (j * 4)]);
-        *dataArray << newRow;
+void SurfaceData::changeMode(int mode)
+{
+    m_mode = VisualizationMode(mode);
+    switch (m_mode) {
+    case Surface: {
+        m_statusArea->append(QStringLiteral("<b>Visualization Type:</b> Surface"));
+        break;
     }
+    case Scatter: {
+        m_statusArea->append(QStringLiteral("<b>Visualization Type:</b> Scatter"));
+        break;
+    }
+    default: {
+        m_statusArea->append(QStringLiteral("<b>Visualization Type:</b> Bars"));
+        break;
+    }
+    }
+    // Reset resolution after mode change
+    setResolution(m_resolutionLevel);
+}
 
-    static_cast<QBarDataProxy *>(m_surface->activeDataProxy())->resetArray(dataArray);
-#endif
+ContainerChanger::ContainerChanger(QWidget *surface, QWidget *scatter, QWidget *bars,
+                                   QWidget *buttonOne, QWidget *buttonTwo)
+    : m_surface(surface),
+      m_scatter(scatter),
+      m_bars(bars),
+      m_button1(buttonOne),
+      m_button2(buttonTwo)
+{
 }
 
-#endif
+ContainerChanger::~ContainerChanger()
+{
+}
+
+void ContainerChanger::changeContainer(int container)
+{
+    switch (container) {
+    case 0: {
+        m_scatter->setVisible(false);
+        m_bars->setVisible(false);
+        m_surface->setVisible(true);
+        m_button1->setEnabled(true);
+        m_button2->setEnabled(true);
+        break;
+    }
+    case 1: {
+        m_surface->setVisible(false);
+        m_bars->setVisible(false);
+        m_scatter->setVisible(true);
+        m_button1->setEnabled(false);
+        m_button2->setEnabled(false);
+        break;
+    }
+    case 2: {
+        m_surface->setVisible(false);
+        m_scatter->setVisible(false);
+        m_bars->setVisible(true);
+        m_button1->setEnabled(false);
+        m_button2->setEnabled(false);
+        break;
+    }
+    }
+}
diff --git a/tests/kinectsurface/surfacedata.h b/tests/kinectsurface/surfacedata.h
index 707bf3c8..0b6b41f9 100644
--- a/tests/kinectsurface/surfacedata.h
+++ b/tests/kinectsurface/surfacedata.h
@@ -19,17 +19,10 @@
 #ifndef SURFACEDATA_H
 #define SURFACEDATA_H
 
-//#define USE_SCATTER
-//#define USE_BARS
-
 #include "QKinectWrapper.h"
-#if defined(USE_SCATTER)
 #include <QtDataVisualization/Q3DScatter>
-#elif defined(USE_BARS)
 #include <QtDataVisualization/Q3DBars>
-#else
 #include <QtDataVisualization/Q3DSurface>
-#endif
 #include <QTextEdit>
 
 using namespace QtDataVisualization;
@@ -39,13 +32,8 @@ class SurfaceData : public QObject
     Q_OBJECT
 
 public:
-#if defined(USE_SCATTER)
-    explicit SurfaceData(Q3DScatter *surface, QTextEdit *statusLabel);
-#elif defined(USE_BARS)
-    explicit SurfaceData(Q3DBars *surface, QTextEdit *statusLabel);
-#else
-    explicit SurfaceData(Q3DSurface *surface, QTextEdit *statusLabel);
-#endif
+    explicit SurfaceData(Q3DSurface *surface, Q3DScatter *scatter, Q3DBars *bars,
+                         QTextEdit *statusLabel);
     ~SurfaceData();
 
     void start();
@@ -56,28 +44,51 @@ public:
 
     void setDistance(int distance);
     void scrollDown();
-#if defined(USE_SCATTER) || defined(USE_BARS)
     void setData(const QImage &image);
-#else
     void useGradientOne();
     void useGradientTwo();
-#endif
+
+public:
+    enum VisualizationMode {
+        Surface = 0,
+        Scatter,
+        Bars
+    };
 
 public slots:
     void setResolution(int selection);
+    void changeMode(int mode);
 
 private:
-#if defined(USE_SCATTER)
-    Q3DScatter *m_surface;
-#elif defined(USE_BARS)
-    Q3DBars *m_surface;
-#else
     Q3DSurface *m_surface;
-#endif
+    Q3DScatter *m_scatter;
+    Q3DBars *m_bars;
     QTextEdit *m_statusArea;
     bool m_resize;
     QSize m_resolution;
+    int m_resolutionLevel;
+    VisualizationMode m_mode;
     QKinect::QKinectWrapper m_kinect;
 };
 
+class ContainerChanger : public QObject
+{
+    Q_OBJECT
+
+public:
+    explicit ContainerChanger(QWidget *surface, QWidget *scatter, QWidget *bars,
+                              QWidget *buttonOne, QWidget *buttonTwo);
+    ~ContainerChanger();
+
+public slots:
+    void changeContainer(int container);
+
+private:
+    QWidget *m_surface;
+    QWidget *m_scatter;
+    QWidget *m_bars;
+    QWidget *m_button1;
+    QWidget *m_button2;
+};
+
 #endif