diff options
Diffstat (limited to 'tests/manual/kinectsurface/surfacedata.cpp')
| -rw-r--r-- | tests/manual/kinectsurface/surfacedata.cpp | 403 |
1 files changed, 403 insertions, 0 deletions
diff --git a/tests/manual/kinectsurface/surfacedata.cpp b/tests/manual/kinectsurface/surfacedata.cpp new file mode 100644 index 00000000..89f696c4 --- /dev/null +++ b/tests/manual/kinectsurface/surfacedata.cpp @@ -0,0 +1,403 @@ +/****************************************************************************** +** +** Copyright (C) 2015 The Qt Company Ltd. +** Contact: http://www.qt.io/licensing/ +** +** This file is part of the Qt Data Visualization module. +** +** $QT_BEGIN_LICENSE:COMM$ +** +** Commercial License Usage +** Licensees holding valid commercial Qt licenses may use this file in +** accordance with the commercial license agreement provided with the +** Software or, alternatively, in accordance with the terms contained in +** a written agreement between you and The Qt Company. For licensing terms +** and conditions see http://www.qt.io/terms-conditions. For further +** information use the contact form at http://www.qt.io/contact-us. +** +** $QT_END_LICENSE$ +** +******************************************************************************/ + +#include "surfacedata.h" +#include "QKinectWrapper.h" +#include <QtDataVisualization/QValue3DAxis> +#include <QtDataVisualization/Q3DCamera> +#include <QtDataVisualization/QBar3DSeries> +#include <QtDataVisualization/QScatter3DSeries> +#include <QtDataVisualization/QSurface3DSeries> +#include <QtDataVisualization/Q3DTheme> +#include <QScrollBar> +#include <QSize> +#include <QImage> + +using namespace QtDataVisualization; + +//#define LOW_END_DEVICE // Uncomment for devices with limited processing/grpahics power + +#ifdef LOW_END_DEVICE +const QSize lowRes = QSize(160, 120); +const QSize medRes = QSize(192, 144); +const QSize hiRes = QSize(256, 192); +const QSize maxRes = QSize(320, 240); +#else +const QSize lowRes = QSize(320, 240); +const QSize medRes = QSize(384, 288); +const QSize hiRes = QSize(512, 384); +const QSize maxRes = QSize(640, 480); +#endif + +SurfaceData::SurfaceData(Q3DSurface *surface, Q3DScatter *scatter, Q3DBars *bars, + QTextEdit *statusArea) : + m_thermalTheme(new Q3DTheme(Q3DTheme::ThemeIsabelle)), + m_highlightTheme(new Q3DTheme(Q3DTheme::ThemeQt)), + m_surface(surface), + m_scatter(scatter), + m_bars(bars), + m_statusArea(statusArea), + m_resize(true), + m_resolution(lowRes), + m_resolutionLevel(0), + m_mode(Surface) +{ + // Initialize themes for surface + m_thermalTheme->setGridEnabled(false); + m_thermalTheme->setBackgroundEnabled(false); + m_thermalTheme->setColorStyle(Q3DTheme::ColorStyleRangeGradient); + QLinearGradient thermalGradient; + thermalGradient.setColorAt(0.0, Qt::black); + thermalGradient.setColorAt(0.33, Qt::blue); + thermalGradient.setColorAt(0.67, Qt::red); + thermalGradient.setColorAt(1.0, Qt::yellow); + QList<QLinearGradient> thermalGradients; + thermalGradients.append(thermalGradient); + m_thermalTheme->setBaseGradients(thermalGradients); + + m_highlightTheme->setGridEnabled(false); + m_highlightTheme->setBackgroundEnabled(false); + m_highlightTheme->setColorStyle(Q3DTheme::ColorStyleRangeGradient); + QLinearGradient highlightGradient; + highlightGradient.setColorAt(0.0, Qt::white); + highlightGradient.setColorAt(0.8, Qt::red); + highlightGradient.setColorAt(1.0, Qt::green); + QList<QLinearGradient> highlightGradients; + highlightGradients.append(highlightGradient); + m_highlightTheme->setBaseGradients(highlightGradients); + + // Initialize surface + m_surface->setActiveTheme(m_thermalTheme); + QLinearGradient gradient; + gradient.setColorAt(0.0, Qt::black); + gradient.setColorAt(0.33, Qt::blue); + gradient.setColorAt(0.67, Qt::red); + gradient.setColorAt(1.0, Qt::yellow); + m_surface->setSelectionMode(QAbstract3DGraph::SelectionNone); + m_surface->setShadowQuality(QAbstract3DGraph::ShadowQualityNone); + m_surface->scene()->activeCamera()->setCameraPosition(0.0, 90.0, 150.0); + m_surface->axisY()->setMax(255); + QSurface3DSeries *series1 = new QSurface3DSeries(new QHeightMapSurfaceDataProxy()); + series1->setFlatShadingEnabled(true); + series1->setDrawMode(QSurface3DSeries::DrawSurface); + m_surface->addSeries(series1); + + // Initialize scatter + m_scatter->activeTheme()->setType(Q3DTheme::ThemeStoneMoss); + m_scatter->setSelectionMode(QAbstract3DGraph::SelectionNone); + m_scatter->activeTheme()->setGridEnabled(false); + m_scatter->setShadowQuality(QAbstract3DGraph::ShadowQualityLow); + m_scatter->scene()->activeCamera()->setCameraPosition(0.0, 60.0, 150.0); + QScatter3DSeries *series2 = new QScatter3DSeries; +#ifdef LOW_END_DEVICE + series2->setMesh(QAbstract3DSeries::MeshPoint); +#else + m_scatter->activeTheme()->setColorStyle(Q3DTheme::ColorStyleRangeGradient); + series2->setMesh(QAbstract3DSeries::MeshMinimal); +#endif + m_scatter->addSeries(series2); + 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->activeTheme()->setType(Q3DTheme::ThemeArmyBlue); + QLinearGradient bargradient; + bargradient.setColorAt(0.48, Qt::white); + bargradient.setColorAt(0.481, Qt::red); + bargradient.setColorAt(0.52, Qt::red); + bargradient.setColorAt(0.521, Qt::black); + QList<QLinearGradient> bargradients; + bargradients.append(bargradient); + m_bars->activeTheme()->setBaseGradients(bargradients); + m_bars->activeTheme()->setColorStyle(Q3DTheme::ColorStyleRangeGradient); + m_bars->setSelectionMode(QAbstract3DGraph::SelectionNone); + m_bars->activeTheme()->setGridEnabled(false); + m_bars->activeTheme()->setBackgroundEnabled(false); +#ifdef LOW_END_DEVICE + m_bars->setShadowQuality(QAbstract3DGraph::ShadowQualityNone); +#else + m_bars->setShadowQuality(QAbstract3DGraph::ShadowQualityLow); +#endif + m_bars->setBarSpacing(QSizeF(0.0, 0.0)); + m_bars->scene()->activeCamera()->setCameraPosition(0.0, 65.0, 130.0); + QBar3DSeries *series3 = new QBar3DSeries; + series3->setMesh(QAbstract3DSeries::MeshBar); + m_bars->addSeries(series3); + m_bars->valueAxis()->setMax(255); + + // Hide scroll bar + m_statusArea->verticalScrollBar()->setVisible(false); + + // Connect Kinect signals + connect(&m_kinect, &QKinect::QKinectWrapper::dataNotification, this, &SurfaceData::updateData); + connect(&m_kinect, &QKinect::QKinectWrapper::statusNotification, this, + &SurfaceData::updateStatus); +} + +SurfaceData::~SurfaceData() +{ + delete m_bars; + delete m_surface; + delete m_scatter; +} + +void SurfaceData::updateData() +{ + QImage depthMap = m_kinect.getDepth(); + if (m_resize) // Resize for better performance + depthMap = depthMap.scaled(m_resolution); + if (m_mode != Surface) { + setData(depthMap); + } else { + static_cast<QHeightMapSurfaceDataProxy *>(m_surface->seriesList().at(0)->dataProxy())->setValueRanges( + 0, depthMap.width(), 0, depthMap.height()); + static_cast<QHeightMapSurfaceDataProxy *>(m_surface->seriesList().at(0)->dataProxy())->setHeightMap( + depthMap); + } +} + +void SurfaceData::updateStatus(QKinect::KinectStatus status) +{ + switch (status) { + case QKinect::Idle: { + m_statusArea->append(QStringLiteral("<b>Kinect:</b> Stopped")); + m_statusArea->append(QStringLiteral("<br><b>Ready</b><br>")); + break; + } + case QKinect::Initializing: { + m_statusArea->append(QStringLiteral("<b>Kinect:</b> Initializing")); + break; + } + case QKinect::OkRun: { + m_statusArea->append(QStringLiteral("<b>Kinect:</b> Running")); + m_statusArea->append(QString(QStringLiteral("<i> - resolution: %1 x %2</i>")).arg( + m_resolution.width()).arg(m_resolution.height())); + break; + } + default: { + m_statusArea->append(QStringLiteral("<b>Kinect:</b> Error")); + break; + } + }; +} + +void SurfaceData::start() +{ + if (m_kinect.isStopped()) + m_kinect.start(); + else + m_statusArea->append(QStringLiteral("<b>Kinect:</b> Already running")); +} + +void SurfaceData::stop() +{ + if (m_kinect.isRunning()) + m_kinect.stop(); + else + m_statusArea->append(QStringLiteral("<b>Kinect:</b> Already stopped")); +} + +void SurfaceData::setDistance(int distance) +{ + m_kinect.setMaxDist((unsigned int)(distance * 100)); +} + +void SurfaceData::setResolution(int selection) +{ + m_resolutionLevel = selection; + switch (selection) { + case 0: { + m_resize = true; + m_resolution = lowRes; + break; + } + case 1: { + m_resize = true; + m_resolution = medRes; + break; + } + case 2: { + m_resize = true; + m_resolution = hiRes; + break; + } + case 3: { + m_resize = false; + m_resolution = maxRes; + break; + } + }; + 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); + m_scatterDataArray = new QScatterDataArray; + m_scatterDataArray->resize(m_resolution.width() * m_resolution.height()); + } else if (m_mode == MeshBars) { + m_resize = true; + m_resolution /= 4; + m_barDataArray = new QBarDataArray; + m_barDataArray->reserve(m_resolution.height()); + for (int i = 0; i < m_resolution.height(); i++) { + QBarDataRow *newProxyRow = new QBarDataRow(m_resolution.width()); + m_barDataArray->append(newProxyRow); + } + } + + m_statusArea->append(QString(QStringLiteral("<b>Resolution:</b> %1 x %2")).arg( + m_resolution.width()).arg(m_resolution.height())); + if (m_kinect.isStopped()) + m_statusArea->append(QStringLiteral("<i> - change takes effect once Kinect is running</i>")); +} + +void SurfaceData::scrollDown() +{ + QScrollBar *scrollbar = m_statusArea->verticalScrollBar(); + scrollbar->setValue(scrollbar->maximum()); +} + +void SurfaceData::useGradientOne() +{ + m_surface->setActiveTheme(m_thermalTheme); + m_statusArea->append(QStringLiteral("<b>Colors:</b> Thermal image imitation")); +} + +void SurfaceData::useGradientTwo() +{ + m_surface->setActiveTheme(m_highlightTheme); + m_statusArea->append(QStringLiteral("<b>Colors:</b> Highlight foreground")); +} + +void SurfaceData::setData(const QImage &image) +{ + QImage heightImage = image; + + uchar *bits = heightImage.bits(); + + int imageHeight = heightImage.height(); + int imageWidth = heightImage.width(); + int bitCount = imageWidth * 4 * (imageHeight - 1); + int widthBits = imageWidth * 4; + + if (m_mode == Scatter) { + QScatterDataItem *ptrToDataArray = &m_scatterDataArray->first(); + + int limitsX = imageWidth / 2; + int limitsZ = imageHeight / 2; + qreal height = 0; + int count = 0; + + 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++; + count++; + } + } + } + + QScatterDataArray *dataArray = new QScatterDataArray(m_scatterDataArray->mid(0, count)); + m_scatter->seriesList().at(0)->dataProxy()->resetArray(dataArray); + } else { + QBarDataArray *dataArray = m_barDataArray; + for (int i = 0; i < imageHeight; i++, bitCount -= widthBits) { + QBarDataRow &newRow = *dataArray->at(i); + for (int j = 0; j < imageWidth; j++) + newRow[j] = qreal(bits[bitCount + (j * 4)]); + } + + m_bars->seriesList().at(0)->dataProxy()->resetArray(dataArray); + } +} + +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); +} + +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) +{ +} + +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; + } + } +} |