surfacegraph.cpp Example File

surface/surfacegraph.cpp
/**************************************************************************** ** ** Copyright (C) 2016 The Qt Company Ltd. ** Contact: https://www.qt.io/licensing/ ** ** This file is part of the Qt Data Visualization module of the Qt Toolkit. ** ** $QT_BEGIN_LICENSE:GPL$ ** 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 https://www.qt.io/terms-conditions. For further ** information use the contact form at https://www.qt.io/contact-us. ** ** GNU General Public License Usage ** Alternatively, this file may be used under the terms of the GNU ** General Public License version 3 or (at your option) any later version ** approved by the KDE Free Qt Foundation. The licenses are as published by ** the Free Software Foundation and appearing in the file LICENSE.GPL3 ** included in the packaging of this file. Please review the following ** information to ensure the GNU General Public License requirements will ** be met: https://www.gnu.org/licenses/gpl-3.0.html. ** ** $QT_END_LICENSE$ ** ****************************************************************************/
#include "surfacegraph.h" #include <QtDataVisualization/QValue3DAxis> #include <QtDataVisualization/Q3DTheme> #include <QtGui/QImage> #include <QtCore/qmath.h> using namespace QtDataVisualization; const int sampleCountX = 50; const int sampleCountZ = 50; const int heightMapGridStepX = 6; const int heightMapGridStepZ = 6; const float sampleMin = -8.0f; const float sampleMax = 8.0f; SurfaceGraph::SurfaceGraph(Q3DSurface *surface) : m_graph(surface) { m_graph->setAxisX(new QValue3DAxis); m_graph->setAxisY(new QValue3DAxis); m_graph->setAxisZ(new QValue3DAxis); m_sqrtSinProxy = new QSurfaceDataProxy(); m_sqrtSinSeries = new QSurface3DSeries(m_sqrtSinProxy); fillSqrtSinProxy(); QImage heightMapImage(":/maps/mountain"); m_heightMapProxy = new QHeightMapSurfaceDataProxy(heightMapImage); m_heightMapSeries = new QSurface3DSeries(m_heightMapProxy); m_heightMapSeries->setItemLabelFormat(QStringLiteral("(@xLabel, @zLabel): @yLabel")); m_heightMapProxy->setValueRanges(34.0f, 40.0f, 18.0f, 24.0f); m_heightMapWidth = heightMapImage.width(); m_heightMapHeight = heightMapImage.height(); } SurfaceGraph::~SurfaceGraph() { delete m_graph; } void SurfaceGraph::fillSqrtSinProxy() { float stepX = (sampleMax - sampleMin) / float(sampleCountX - 1); float stepZ = (sampleMax - sampleMin) / float(sampleCountZ - 1); QSurfaceDataArray *dataArray = new QSurfaceDataArray; dataArray->reserve(sampleCountZ); for (int i = 0 ; i < sampleCountZ ; i++) { QSurfaceDataRow *newRow = new QSurfaceDataRow(sampleCountX); // Keep values within range bounds, since just adding step can cause minor drift due // to the rounding errors. float z = qMin(sampleMax, (i * stepZ + sampleMin)); int index = 0; for (int j = 0; j < sampleCountX; j++) { float x = qMin(sampleMax, (j * stepX + sampleMin)); float R = qSqrt(z * z + x * x) + 0.01f; float y = (qSin(R) / R + 0.24f) * 1.61f; (*newRow)[index++].setPosition(QVector3D(x, y, z)); } *dataArray << newRow; } m_sqrtSinProxy->resetArray(dataArray); } void SurfaceGraph::enableSqrtSinModel(bool enable) { if (enable) { m_sqrtSinSeries->setDrawMode(QSurface3DSeries::DrawSurfaceAndWireframe); m_sqrtSinSeries->setFlatShadingEnabled(true); m_graph->axisX()->setLabelFormat("%.2f"); m_graph->axisZ()->setLabelFormat("%.2f"); m_graph->axisX()->setRange(sampleMin, sampleMax); m_graph->axisY()->setRange(0.0f, 2.0f); m_graph->axisZ()->setRange(sampleMin, sampleMax); m_graph->axisX()->setLabelAutoRotation(30); m_graph->axisY()->setLabelAutoRotation(90); m_graph->axisZ()->setLabelAutoRotation(30); m_graph->removeSeries(m_heightMapSeries); m_graph->addSeries(m_sqrtSinSeries); // Reset range sliders for Sqrt&Sin m_rangeMinX = sampleMin; m_rangeMinZ = sampleMin; m_stepX = (sampleMax - sampleMin) / float(sampleCountX - 1); m_stepZ = (sampleMax - sampleMin) / float(sampleCountZ - 1); m_axisMinSliderX->setMaximum(sampleCountX - 2); m_axisMinSliderX->setValue(0); m_axisMaxSliderX->setMaximum(sampleCountX - 1); m_axisMaxSliderX->setValue(sampleCountX - 1); m_axisMinSliderZ->setMaximum(sampleCountZ - 2); m_axisMinSliderZ->setValue(0); m_axisMaxSliderZ->setMaximum(sampleCountZ - 1); m_axisMaxSliderZ->setValue(sampleCountZ - 1); } } void SurfaceGraph::enableHeightMapModel(bool enable) { if (enable) { m_heightMapSeries->setDrawMode(QSurface3DSeries::DrawSurface); m_heightMapSeries->setFlatShadingEnabled(false); m_graph->axisX()->setLabelFormat("%.1f N"); m_graph->axisZ()->setLabelFormat("%.1f E"); m_graph->axisX()->setRange(34.0f, 40.0f); m_graph->axisY()->setAutoAdjustRange(true); m_graph->axisZ()->setRange(18.0f, 24.0f); m_graph->axisX()->setTitle(QStringLiteral("Latitude")); m_graph->axisY()->setTitle(QStringLiteral("Height")); m_graph->axisZ()->setTitle(QStringLiteral("Longitude")); m_graph->removeSeries(m_sqrtSinSeries); m_graph->addSeries(m_heightMapSeries); // Reset range sliders for height map int mapGridCountX = m_heightMapWidth / heightMapGridStepX; int mapGridCountZ = m_heightMapHeight / heightMapGridStepZ; m_rangeMinX = 34.0f; m_rangeMinZ = 18.0f; m_stepX = 6.0f / float(mapGridCountX - 1); m_stepZ = 6.0f / float(mapGridCountZ - 1); m_axisMinSliderX->setMaximum(mapGridCountX - 2); m_axisMinSliderX->setValue(0); m_axisMaxSliderX->setMaximum(mapGridCountX - 1); m_axisMaxSliderX->setValue(mapGridCountX - 1); m_axisMinSliderZ->setMaximum(mapGridCountZ - 2); m_axisMinSliderZ->setValue(0); m_axisMaxSliderZ->setMaximum(mapGridCountZ - 1); m_axisMaxSliderZ->setValue(mapGridCountZ - 1); } } void SurfaceGraph::adjustXMin(int min) { float minX = m_stepX * float(min) + m_rangeMinX; int max = m_axisMaxSliderX->value(); if (min >= max) { max = min + 1; m_axisMaxSliderX->setValue(max); } float maxX = m_stepX * max + m_rangeMinX; setAxisXRange(minX, maxX); } void SurfaceGraph::adjustXMax(int max) { float maxX = m_stepX * float(max) + m_rangeMinX; int min = m_axisMinSliderX->value(); if (max <= min) { min = max - 1; m_axisMinSliderX->setValue(min); } float minX = m_stepX * min + m_rangeMinX; setAxisXRange(minX, maxX); } void SurfaceGraph::adjustZMin(int min) { float minZ = m_stepZ * float(min) + m_rangeMinZ; int max = m_axisMaxSliderZ->value(); if (min >= max) { max = min + 1; m_axisMaxSliderZ->setValue(max); } float maxZ = m_stepZ * max + m_rangeMinZ; setAxisZRange(minZ, maxZ); } void SurfaceGraph::adjustZMax(int max) { float maxX = m_stepZ * float(max) + m_rangeMinZ; int min = m_axisMinSliderZ->value(); if (max <= min) { min = max - 1; m_axisMinSliderZ->setValue(min); } float minX = m_stepZ * min + m_rangeMinZ; setAxisZRange(minX, maxX); } void SurfaceGraph::setAxisXRange(float min, float max) { m_graph->axisX()->setRange(min, max); } void SurfaceGraph::setAxisZRange(float min, float max) { m_graph->axisZ()->setRange(min, max); } void SurfaceGraph::changeTheme(int theme) { m_graph->activeTheme()->setType(Q3DTheme::Theme(theme)); } void SurfaceGraph::setBlackToYellowGradient() { QLinearGradient gr; gr.setColorAt(0.0, Qt::black); gr.setColorAt(0.33, Qt::blue); gr.setColorAt(0.67, Qt::red); gr.setColorAt(1.0, Qt::yellow); m_graph->seriesList().at(0)->setBaseGradient(gr); m_graph->seriesList().at(0)->setColorStyle(Q3DTheme::ColorStyleRangeGradient); } void SurfaceGraph::setGreenToRedGradient() { QLinearGradient gr; gr.setColorAt(0.0, Qt::darkGreen); gr.setColorAt(0.5, Qt::yellow); gr.setColorAt(0.8, Qt::red); gr.setColorAt(1.0, Qt::darkRed); m_graph->seriesList().at(0)->setBaseGradient(gr); m_graph->seriesList().at(0)->setColorStyle(Q3DTheme::ColorStyleRangeGradient); }