diff options
Diffstat (limited to 'tests/manual/kinectsurface/surfacedata.cpp')
| -rw-r--r-- | tests/manual/kinectsurface/surfacedata.cpp | 403 |
1 files changed, 0 insertions, 403 deletions
diff --git a/tests/manual/kinectsurface/surfacedata.cpp b/tests/manual/kinectsurface/surfacedata.cpp deleted file mode 100644 index 89f696c4..00000000 --- a/tests/manual/kinectsurface/surfacedata.cpp +++ /dev/null @@ -1,403 +0,0 @@ -/****************************************************************************** -** -** 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; - } - } -} |