Revision control

++ // Copyright (C) 2023 The Qt Company Ltd.
++ // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
++
++ #include "surfacegraphmodifier.h"
++ #include "highlightseries.h"
++ #include "topographicseries.h"
++
++ #include <QtCore/qmath.h>
++ #include <QtGraphs/qgraphstheme.h>
++ #include <QtGraphs/qvalue3daxis.h>
++
++ #include <QtGui/qimage.h>
++ #include <QtGui/qquaternion.h>
++
++ using namespace Qt::StringLiterals;
++
++ const int sampleCountX = 150;
++ const int sampleCountZ = 150;
++ const int heightMapGridStepX = 6;
++ const int heightMapGridStepZ = 6;
++ const float sampleMin = -8.f;
++ const float sampleMax = 8.f;
++
++ const float areaWidth = 8000.f;
++ const float areaHeight = 8000.f;
++ const float aspectRatio = 0.1389f;
++ const float minRange = areaWidth * 0.49f;
++
++ SurfaceGraphModifier::SurfaceGraphModifier(Q3DSurfaceWidgetItem *surface, QLabel *label, QObject *parent)
++ : QObject(parent)
++ , m_graph(surface)
++ , m_textField(label)
++ , m_aspectRatio(aspectRatio)
++ {
++ m_graph->setCameraZoomLevel(85.f);
++ m_graph->setCameraPreset(QtGraphs3D::CameraPreset::IsometricRight);
++ m_graph->activeTheme()->setTheme(QGraphsTheme::Theme::MixSeries);
++ m_graph->activeTheme()->setLabelBackgroundVisible(false);
++ m_graph->activeTheme()->setLabelBorderVisible(false);
++
++ m_graph->setAxisX(new QValue3DAxis);
++ m_graph->setAxisY(new QValue3DAxis);
++ m_graph->setAxisZ(new QValue3DAxis);
++
++ //
++ // Sqrt Sin
++ //
++ //! [0]
++ m_sqrtSinProxy = new QSurfaceDataProxy();
++ m_sqrtSinSeries = new QSurface3DSeries(m_sqrtSinProxy);
++ //! [0]
++ fillSqrtSinProxy();
++
++ //
++ // Multisurface heightmap
++ //
++ //! [2]
++ // Create the first surface layer
++ QImage heightMapImageOne(":/data/layer_1.png");
++ m_heightMapProxyOne = new QHeightMapSurfaceDataProxy(heightMapImageOne);
++ m_heightMapSeriesOne = new QSurface3DSeries(m_heightMapProxyOne);
++ m_heightMapSeriesOne->setItemLabelFormat(u"(@xLabel, @zLabel): @yLabel"_s);
++ m_heightMapProxyOne->setValueRanges(34.f, 40.f, 18.f, 24.f);
++ //! [2]
++
++ // Create the other 2 surface layers
++ QImage heightMapImageTwo(":/data/layer_2.png");
++ m_heightMapProxyTwo = new QHeightMapSurfaceDataProxy(heightMapImageTwo);
++ m_heightMapSeriesTwo = new QSurface3DSeries(m_heightMapProxyTwo);
++ m_heightMapSeriesTwo->setItemLabelFormat(u"(@xLabel, @zLabel): @yLabel"_s);
++ m_heightMapProxyTwo->setValueRanges(34.f, 40.f, 18.f, 24.f);
++
++ QImage heightMapImageThree(":/data/layer_3.png");
++ m_heightMapProxyThree = new QHeightMapSurfaceDataProxy(heightMapImageThree);
++ m_heightMapSeriesThree = new QSurface3DSeries(m_heightMapProxyThree);
++ m_heightMapSeriesThree->setItemLabelFormat(u"(@xLabel, @zLabel): @yLabel"_s);
++ m_heightMapProxyThree->setValueRanges(34.f, 40.f, 18.f, 24.f);
++
++ // The images are the same size, so it's enough to get the dimensions from one
++ m_heightMapWidth = heightMapImageOne.width();
++ m_heightMapHeight = heightMapImageOne.height();
++
++ // Set the gradients for multi-surface layers
++ QLinearGradient grOne;
++ grOne.setColorAt(0., Qt::black);
++ grOne.setColorAt(0.38, Qt::darkYellow);
++ grOne.setColorAt(0.39, Qt::darkGreen);
++ grOne.setColorAt(0.5, Qt::darkGray);
++ grOne.setColorAt(1., Qt::gray);
++ m_heightMapSeriesOne->setBaseGradient(grOne);
++ m_heightMapSeriesOne->setColorStyle(QGraphsTheme::ColorStyle::RangeGradient);
++
++ QLinearGradient grTwo;
++ grTwo.setColorAt(0.39, Qt::blue);
++ grTwo.setColorAt(0.4, Qt::white);
++ m_heightMapSeriesTwo->setBaseGradient(grTwo);
++ m_heightMapSeriesTwo->setColorStyle(QGraphsTheme::ColorStyle::RangeGradient);
++
++ QLinearGradient grThree;
++ grThree.setColorAt(0., Qt::white);
++ grThree.setColorAt(0.05, Qt::black);
++ m_heightMapSeriesThree->setBaseGradient(grThree);
++ m_heightMapSeriesThree->setColorStyle(QGraphsTheme::ColorStyle::RangeGradient);
++
++ // Custom items and label
++ connect(m_graph,
++ &Q3DGraphsWidgetItem::selectedElementChanged,
++ this,
++ &SurfaceGraphModifier::handleElementSelected);
++
++ m_selectionAnimation = new QPropertyAnimation(this);
++ m_selectionAnimation->setPropertyName("scaling");
++ m_selectionAnimation->setDuration(500);
++ m_selectionAnimation->setLoopCount(-1);
++
++ QFont titleFont = QFont("Century Gothic", 30);
++ titleFont.setBold(true);
++ m_titleLabel = new QCustom3DLabel("Oil Rigs on Imaginary Sea",
++ titleFont,
++ QVector3D(0.f, 1.2f, 0.f),
++ QVector3D(1.f, 1.f, 0.f),
++ QQuaternion());
++ m_titleLabel->setPositionAbsolute(true);
++ m_titleLabel->setFacingCamera(true);
++ m_titleLabel->setBackgroundColor(QColor(0x66cdaa));
++ m_graph->addCustomItem(m_titleLabel);
++ m_titleLabel->setVisible(false);
++
++ // Make two of the custom object visible
++ toggleItemOne(true);
++ toggleItemTwo(true);
++
++ //
++ // Topographic map
++ //
++ m_topography = new TopographicSeries();
++ m_topography->setTopographyFile(":/data/topography.png", areaWidth, areaHeight);
++ m_topography->setItemLabelFormat(u"@yLabel m"_s);
++
++ m_highlight = new HighlightSeries();
++ m_highlight->setTopographicSeries(m_topography);
++ m_highlight->setMinHeight(minRange * aspectRatio);
++ m_highlight->handleGradientChange(areaWidth * aspectRatio);
++ //! [16]
++ QObject::connect(m_graph->axisY(),
++ &QValue3DAxis::maxChanged,
++ m_highlight,
++ &HighlightSeries::handleGradientChange);
++ //! [16]
++
++ m_areaMinValue = 0.f;
++ m_areaMaxValue = areaWidth;
++ m_axisXMinValue = m_areaMinValue;
++ m_axisXMaxValue = m_areaMaxValue;
++ m_axisZMinValue = m_areaMinValue;
++ m_axisZMaxValue = m_areaMaxValue;
++ m_axisXMinRange = minRange;
++ m_axisZMinRange = minRange;
++
++ QObject::connect(m_graph,
++ &Q3DGraphsWidgetItem::dragged,
++ this,
++ &SurfaceGraphModifier::handleAxisDragging);
++
++ QObject::connect(m_graph, &Q3DGraphsWidgetItem::wheel, this, &SurfaceGraphModifier::onWheel);
++ }
++
++ SurfaceGraphModifier::~SurfaceGraphModifier() = default;
++
++ void SurfaceGraphModifier::fillSqrtSinProxy()
++ {
++ float stepX = (sampleMax - sampleMin) / float(sampleCountX - 1);
++ float stepZ = (sampleMax - sampleMin) / float(sampleCountZ - 1);
++
++ //! [1]
++ QSurfaceDataArray dataArray;
++ dataArray.reserve(sampleCountZ);
++ for (int i = 0; i < sampleCountZ; ++i) {
++ QSurfaceDataRow newRow;
++ newRow.reserve(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));
++ 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.append(QSurfaceDataItem(x, y, z));
++ }
++ dataArray.append(newRow);
++ }
++
++ m_sqrtSinProxy->resetArray(dataArray);
++ //! [1]
++ }
++
++ void SurfaceGraphModifier::enableSqrtSinModel(bool enable)
++ {
++ if (enable) {
++ //! [3]
++ m_sqrtSinSeries->setDrawMode(QSurface3DSeries::DrawSurfaceAndWireframe);
++ m_sqrtSinSeries->setShading(QSurface3DSeries::Shading::Flat);
++
++ m_graph->axisX()->setLabelFormat("%.2f");
++ m_graph->axisZ()->setLabelFormat("%.2f");
++ m_graph->axisX()->setRange(sampleMin, sampleMax);
++ m_graph->axisY()->setRange(0.f, 2.f);
++ m_graph->axisZ()->setRange(sampleMin, sampleMax);
++ m_graph->axisX()->setLabelAutoAngle(30.f);
++ m_graph->axisY()->setLabelAutoAngle(90.f);
++ m_graph->axisZ()->setLabelAutoAngle(30.f);
++
++ m_graph->removeSeries(m_heightMapSeriesOne);
++ m_graph->removeSeries(m_heightMapSeriesTwo);
++ m_graph->removeSeries(m_heightMapSeriesThree);
++ m_graph->removeSeries(m_topography);
++ m_graph->removeSeries(m_highlight);
++
++ m_graph->addSeries(m_sqrtSinSeries);
++ //! [3]
++
++ m_titleLabel->setVisible(false);
++ m_graph->axisX()->setTitleVisible(false);
++ m_graph->axisY()->setTitleVisible(false);
++ m_graph->axisZ()->setTitleVisible(false);
++
++ m_graph->axisX()->setTitle({});
++ m_graph->axisY()->setTitle({});
++ m_graph->axisZ()->setTitle({});
++
++ QObject::disconnect(m_graph,
++ &Q3DGraphsWidgetItem::dragged,
++ this,
++ &SurfaceGraphModifier::handleAxisDragging);
++
++ QObject::disconnect(m_graph,
++ &Q3DGraphsWidgetItem::wheel,
++ this,
++ &SurfaceGraphModifier::onWheel);
++ m_graph->setDefaultInputHandler();
++ m_graph->setZoomEnabled(true);
++
++ //! [6]
++ // 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->setMinimum(0);
++ m_axisMinSliderX->setMaximum(sampleCountX - 2);
++ m_axisMinSliderX->setValue(0);
++ m_axisMaxSliderX->setMinimum(1);
++ m_axisMaxSliderX->setMaximum(sampleCountX - 1);
++ m_axisMaxSliderX->setValue(sampleCountX - 1);
++ m_axisMinSliderZ->setMinimum(0);
++ m_axisMinSliderZ->setMaximum(sampleCountZ - 2);
++ m_axisMinSliderZ->setValue(0);
++ m_axisMaxSliderZ->setMinimum(1);
++ m_axisMaxSliderZ->setMaximum(sampleCountZ - 1);
++ m_axisMaxSliderZ->setValue(sampleCountZ - 1);
++ //! [6]
++ }
++ }
++
++ void SurfaceGraphModifier::enableHeightMapModel(bool enable)
++ {
++ if (enable) {
++ m_heightMapSeriesOne->setDrawMode(QSurface3DSeries::DrawSurface);
++ m_heightMapSeriesOne->setShading(QSurface3DSeries::Shading::Smooth);
++ m_heightMapSeriesTwo->setDrawMode(QSurface3DSeries::DrawSurface);
++ m_heightMapSeriesTwo->setShading(QSurface3DSeries::Shading::Smooth);
++ m_heightMapSeriesThree->setDrawMode(QSurface3DSeries::DrawSurface);
++ m_heightMapSeriesThree->setShading(QSurface3DSeries::Shading::Smooth);
++
++ m_graph->axisX()->setLabelFormat("%.1f N");
++ m_graph->axisZ()->setLabelFormat("%.1f E");
++ m_graph->axisX()->setRange(34.f, 40.f);
++ //! [4]
++ m_graph->axisY()->setAutoAdjustRange(true);
++ //! [4]
++ m_graph->axisZ()->setRange(18.f, 24.f);
++
++ m_graph->axisX()->setTitle(u"Latitude"_s);
++ m_graph->axisY()->setTitle(u"Height"_s);
++ m_graph->axisZ()->setTitle(u"Longitude"_s);
++
++ m_graph->removeSeries(m_sqrtSinSeries);
++ m_graph->removeSeries(m_topography);
++ m_graph->removeSeries(m_highlight);
++ m_graph->addSeries(m_heightMapSeriesOne);
++ m_graph->addSeries(m_heightMapSeriesTwo);
++ m_graph->addSeries(m_heightMapSeriesThree);
++
++ QObject::disconnect(m_graph,
++ &Q3DGraphsWidgetItem::dragged,
++ this,
++ &SurfaceGraphModifier::handleAxisDragging);
++
++ QObject::disconnect(m_graph,
++ &Q3DGraphsWidgetItem::wheel,
++ this,
++ &SurfaceGraphModifier::onWheel);
++ m_graph->setDefaultInputHandler();
++ m_graph->setZoomEnabled(true);
++
++ m_titleLabel->setVisible(true);
++ m_graph->axisX()->setTitleVisible(true);
++ m_graph->axisY()->setTitleVisible(true);
++ m_graph->axisZ()->setTitleVisible(true);
++
++ // Reset range sliders for height map
++ int mapGridCountX = m_heightMapWidth / heightMapGridStepX;
++ int mapGridCountZ = m_heightMapHeight / heightMapGridStepZ;
++ m_rangeMinX = 34.f;
++ m_rangeMinZ = 18.f;
++ m_stepX = 6.f / float(mapGridCountX - 1);
++ m_stepZ = 6.f / float(mapGridCountZ - 1);
++ m_axisMinSliderX->setMinimum(0);
++ m_axisMinSliderX->setMaximum(mapGridCountX - 2);
++ m_axisMinSliderX->setValue(0);
++ m_axisMaxSliderX->setMinimum(1);
++ m_axisMaxSliderX->setMaximum(mapGridCountX - 1);
++ m_axisMaxSliderX->setValue(mapGridCountX - 1);
++ m_axisMinSliderZ->setMinimum(0);
++ m_axisMinSliderZ->setMaximum(mapGridCountZ - 2);
++ m_axisMinSliderZ->setValue(0);
++ m_axisMaxSliderZ->setMinimum(1);
++ m_axisMaxSliderZ->setMaximum(mapGridCountZ - 1);
++ m_axisMaxSliderZ->setValue(mapGridCountZ - 1);
++ }
++ }
++
++ void SurfaceGraphModifier::enableTopographyModel(bool enable)
++ {
++ if (enable) {
++ m_graph->axisX()->setLabelFormat("%i");
++ m_graph->axisZ()->setLabelFormat("%i");
++ m_graph->axisX()->setRange(0.f, areaWidth);
++ m_graph->axisY()->setRange(100.f, areaWidth * aspectRatio);
++ m_graph->axisZ()->setRange(0.f, areaHeight);
++ m_graph->axisX()->setLabelAutoAngle(30.f);
++ m_graph->axisY()->setLabelAutoAngle(90.f);
++ m_graph->axisZ()->setLabelAutoAngle(30.f);
++
++ m_graph->removeSeries(m_sqrtSinSeries);
++ m_graph->removeSeries(m_heightMapSeriesOne);
++ m_graph->removeSeries(m_heightMapSeriesTwo);
++ m_graph->removeSeries(m_heightMapSeriesThree);
++ m_graph->addSeries(m_topography);
++ m_graph->addSeries(m_highlight);
++
++ m_titleLabel->setVisible(false);
++ m_graph->axisX()->setTitleVisible(false);
++ m_graph->axisY()->setTitleVisible(false);
++ m_graph->axisZ()->setTitleVisible(false);
++
++ m_graph->axisX()->setTitle({});
++ m_graph->axisY()->setTitle({});
++ m_graph->axisZ()->setTitle({});
++
++ //! [5]
++ m_graph->setDragButton(Qt::LeftButton);
++ QObject::connect(m_graph,
++ &Q3DGraphsWidgetItem::dragged,
++ this,
++ &SurfaceGraphModifier::handleAxisDragging);
++
++ QObject::connect(m_graph,
++ &Q3DGraphsWidgetItem::wheel,
++ this,
++ &SurfaceGraphModifier::onWheel);
++ m_graph->setZoomEnabled(false);
++ //! [5]
++
++ // Reset range sliders for topography map
++ m_rangeMinX = 0.f;
++ m_rangeMinZ = 0.f;
++ m_stepX = 1.f;
++ m_stepZ = 1.f;
++ m_axisMinSliderX->setMinimum(0);
++ m_axisMinSliderX->setMaximum(areaWidth - 200);
++ m_axisMinSliderX->setValue(0);
++ m_axisMaxSliderX->setMinimum(200);
++ m_axisMaxSliderX->setMaximum(areaWidth);
++ m_axisMaxSliderX->setValue(areaWidth);
++ m_axisMinSliderZ->setMinimum(0);
++ m_axisMinSliderZ->setMaximum(areaHeight - 200);
++ m_axisMinSliderZ->setValue(0);
++ m_axisMaxSliderZ->setMinimum(200);
++ m_axisMaxSliderZ->setMaximum(areaHeight);
++ m_axisMaxSliderZ->setValue(areaHeight);
++ }
++ }
++
++ void SurfaceGraphModifier::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 SurfaceGraphModifier::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 SurfaceGraphModifier::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 SurfaceGraphModifier::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);
++ }
++
++ //! [7]
++ void SurfaceGraphModifier::setAxisXRange(float min, float max)
++ {
++ m_graph->axisX()->setRange(min, max);
++ }
++ //! [7]
++
++ void SurfaceGraphModifier::setAxisZRange(float min, float max)
++ {
++ m_graph->axisZ()->setRange(min, max);
++ }
++
++ void SurfaceGraphModifier::setBlackToYellowGradient()
++ {
++ //! [8]
++ QLinearGradient gr;
++ gr.setColorAt(0.f, Qt::black);
++ gr.setColorAt(0.33f, Qt::blue);
++ gr.setColorAt(0.67f, Qt::red);
++ gr.setColorAt(1.f, Qt::yellow);
++
++ m_sqrtSinSeries->setBaseGradient(gr);
++ m_sqrtSinSeries->setColorStyle(QGraphsTheme::ColorStyle::RangeGradient);
++ //! [8]
++ }
++
++ void SurfaceGraphModifier::setGreenToRedGradient()
++ {
++ QLinearGradient gr;
++ gr.setColorAt(0.f, Qt::darkGreen);
++ gr.setColorAt(0.5f, Qt::yellow);
++ gr.setColorAt(0.8f, Qt::red);
++ gr.setColorAt(1.f, Qt::darkRed);
++
++ m_sqrtSinSeries->setBaseGradient(gr);
++ m_sqrtSinSeries->setColorStyle(QGraphsTheme::ColorStyle::RangeGradient);
++ }
++
++ void SurfaceGraphModifier::toggleItemOne(bool show)
++ {
++ //! [10]
++ QVector3D positionOne = QVector3D(39.f, 77.f, 19.2f);
++ //! [10]
++ QVector3D positionOnePipe = QVector3D(39.f, 45.f, 19.2f);
++ QVector3D positionOneLabel = QVector3D(39.f, 107.f, 19.2f);
++ if (show) {
++ //! [9]
++ QImage color = QImage(2, 2, QImage::Format_RGB32);
++ color.fill(Qt::red);
++ //! [9]
++ //! [11]
++ auto *item = new QCustom3DItem(":/data/oilrig.mesh",
++ positionOne,
++ QVector3D(0.0125f, 0.0125f, 0.0125f),
++ QQuaternion::fromAxisAndAngle(0.f, 1.f, 0.f, 45.f),
++ color);
++ //! [11]
++ //! [12]
++ m_graph->addCustomItem(item);
++ //! [12]
++ item = new QCustom3DItem(":/data/pipe.mesh",
++ positionOnePipe,
++ QVector3D(0.0025f, 0.25f, 0.0025f),
++ QQuaternion(),
++ color);
++ item->setShadowCasting(false);
++ m_graph->addCustomItem(item);
++
++ //! [13]
++ auto *label = new QCustom3DLabel();
++ label->setText("Oil Rig One");
++ label->setPosition(positionOneLabel);
++ label->setScaling(QVector3D(1.f, 1.f, 1.f));
++ m_graph->addCustomItem(label);
++ //! [13]
++ } else {
++ resetSelection();
++ //! [14]
++ m_graph->removeCustomItemAt(positionOne);
++ //! [14]
++ m_graph->removeCustomItemAt(positionOnePipe);
++ m_graph->removeCustomItemAt(positionOneLabel);
++ }
++ }
++
++ void SurfaceGraphModifier::toggleItemTwo(bool show)
++ {
++ QVector3D positionTwo = QVector3D(34.5f, 77.f, 23.4f);
++ QVector3D positionTwoPipe = QVector3D(34.5f, 45.f, 23.4f);
++ QVector3D positionTwoLabel = QVector3D(34.5f, 107.f, 23.4f);
++ if (show) {
++ QImage color = QImage(2, 2, QImage::Format_RGB32);
++ color.fill(Qt::red);
++ auto *item = new QCustom3DItem();
++ item->setMeshFile(":/data/oilrig.mesh");
++ item->setPosition(positionTwo);
++ item->setScaling(QVector3D(0.0125f, 0.0125f, 0.0125f));
++ item->setRotation(QQuaternion::fromAxisAndAngle(0.f, 1.f, 0.f, 25.f));
++ item->setTextureImage(color);
++ m_graph->addCustomItem(item);
++ item = new QCustom3DItem(":/data/pipe.mesh",
++ positionTwoPipe,
++ QVector3D(0.0025f, 0.25f, 0.0025f),
++ QQuaternion(),
++ color);
++ item->setShadowCasting(false);
++ m_graph->addCustomItem(item);
++
++ auto *label = new QCustom3DLabel();
++ label->setText("Oil Rig Two");
++ label->setPosition(positionTwoLabel);
++ label->setScaling(QVector3D(1.f, 1.f, 1.f));
++ m_graph->addCustomItem(label);
++ } else {
++ resetSelection();
++ m_graph->removeCustomItemAt(positionTwo);
++ m_graph->removeCustomItemAt(positionTwoPipe);
++ m_graph->removeCustomItemAt(positionTwoLabel);
++ }
++ }
++
++ void SurfaceGraphModifier::toggleItemThree(bool show)
++ {
++ QVector3D positionThree = QVector3D(34.5f, 86.f, 19.1f);
++ QVector3D positionThreeLabel = QVector3D(34.5f, 116.f, 19.1f);
++ if (show) {
++ QImage color = QImage(2, 2, QImage::Format_RGB32);
++ color.fill(Qt::darkMagenta);
++ auto *item = new QCustom3DItem();
++ item->setMeshFile(":/data/refinery.mesh");
++ item->setPosition(positionThree);
++ item->setScaling(QVector3D(0.02f, 0.02f, 0.02f));
++ item->setRotation(QQuaternion::fromAxisAndAngle(0.f, 1.f, 0.f, 75.f));
++ item->setTextureImage(color);
++ m_graph->addCustomItem(item);
++
++ auto *label = new QCustom3DLabel();
++ label->setText("Refinery");
++ label->setPosition(positionThreeLabel);
++ label->setScaling(QVector3D(1.f, 1.f, 1.f));
++ m_graph->addCustomItem(label);
++ } else {
++ resetSelection();
++ m_graph->removeCustomItemAt(positionThree);
++ m_graph->removeCustomItemAt(positionThreeLabel);
++ }
++ }
++
++ void SurfaceGraphModifier::toggleSeeThrough(bool seethrough)
++ {
++ if (seethrough) {
++ m_graph->seriesList().at(0)->setDrawMode(QSurface3DSeries::DrawWireframe);
++ m_graph->seriesList().at(1)->setDrawMode(QSurface3DSeries::DrawWireframe);
++ } else {
++ m_graph->seriesList().at(0)->setDrawMode(QSurface3DSeries::DrawSurface);
++ m_graph->seriesList().at(1)->setDrawMode(QSurface3DSeries::DrawSurface);
++ }
++ }
++
++ void SurfaceGraphModifier::toggleOilHighlight(bool highlight)
++ {
++ if (highlight) {
++ QLinearGradient grThree;
++ grThree.setColorAt(0.0, Qt::black);
++ grThree.setColorAt(0.05, Qt::red);
++ m_graph->seriesList().at(2)->setBaseGradient(grThree);
++ } else {
++ QLinearGradient grThree;
++ grThree.setColorAt(0.0, Qt::white);
++ grThree.setColorAt(0.05, Qt::black);
++ m_graph->seriesList().at(2)->setBaseGradient(grThree);
++ }
++ }
++
++ void SurfaceGraphModifier::toggleShadows(bool shadows)
++ {
++ if (shadows)
++ m_graph->setShadowQuality(QtGraphs3D::ShadowQuality::Medium);
++ else
++ m_graph->setShadowQuality(QtGraphs3D::ShadowQuality::None);
++ }
++
++ //! [15]
++ void SurfaceGraphModifier::toggleSurfaceTexture(bool enable)
++ {
++ if (enable)
++ m_topography->setTextureFile(":/data/maptexture.jpg");
++ else
++ m_topography->setTextureFile("");
++ }
++ //! [15]
++
++ void SurfaceGraphModifier::handleElementSelected(QtGraphs3D::ElementType type)
++ {
++ resetSelection();
++ if (type == QtGraphs3D::ElementType::CustomItem) {
++ QCustom3DItem *item = m_graph->selectedCustomItem();
++ QString text;
++ if (qobject_cast<QCustom3DLabel *>(item) != 0) {
++ text.append("Custom label: ");
++ } else {
++ QStringList split = item->meshFile().split("/");
++ text.append(split.last());
++ text.append(": ");
++ }
++ int index = m_graph->selectedCustomItemIndex();
++ text.append(QString::number(index));
++ m_textField->setText(text);
++ m_previouslyAnimatedItem = item;
++ m_previousScaling = item->scaling();
++ m_selectionAnimation->setTargetObject(item);
++ m_selectionAnimation->setStartValue(item->scaling());
++ m_selectionAnimation->setEndValue(item->scaling() * 1.5f);
++ m_selectionAnimation->start();
++ } else if (type == QtGraphs3D::ElementType::Series) {
++ QString text = "Surface (";
++ QSurface3DSeries *series = m_graph->selectedSeries();
++ if (series) {
++ QPoint point = series->selectedPoint();
++ QString posStr;
++ posStr.setNum(point.x());
++ text.append(posStr);
++ text.append(", ");
++ posStr.setNum(point.y());
++ text.append(posStr);
++ }
++ text.append(")");
++ m_textField->setText(text);
++ } else if (type > QtGraphs3D::ElementType::Series
++ && type < QtGraphs3D::ElementType::CustomItem) {
++ int index = m_graph->selectedLabelIndex();
++ QString text;
++ if (type == QtGraphs3D::ElementType::AxisXLabel) {
++ text.append("Axis X label: ");
++ m_state = StateDraggingX;
++ } else if (type == QtGraphs3D::ElementType::AxisYLabel) {
++ text.append("Axis Y label: ");
++ m_state = StateDraggingY;
++ } else {
++ text.append("Axis Z label: ");
++ m_state = StateDraggingZ;
++ }
++ text.append(QString::number(index));
++ m_textField->setText(text);
++ } else {
++ m_textField->setText("Nothing");
++ m_state = StateNormal;
++ }
++ }
++
++ void SurfaceGraphModifier::resetSelection()
++ {
++ m_selectionAnimation->stop();
++ if (m_previouslyAnimatedItem)
++ m_previouslyAnimatedItem->setScaling(m_previousScaling);
++ m_previouslyAnimatedItem = nullptr;
++ }
++
++ void SurfaceGraphModifier::checkConstraints()
++ {
++ //! [19]
++ if (m_axisXMinValue < m_areaMinValue)
++ m_axisXMinValue = m_areaMinValue;
++ if (m_axisXMaxValue > m_areaMaxValue)
++ m_axisXMaxValue = m_areaMaxValue;
++ // Don't allow too much zoom in
++ if ((m_axisXMaxValue - m_axisXMinValue) < m_axisXMinRange) {
++ float adjust = (m_axisXMinRange - (m_axisXMaxValue - m_axisXMinValue)) / 2.f;
++ m_axisXMinValue -= adjust;
++ m_axisXMaxValue += adjust;
++ }
++ //! [19]
++
++ if (m_axisZMinValue < m_areaMinValue)
++ m_axisZMinValue = m_areaMinValue;
++ if (m_axisZMaxValue > m_areaMaxValue)
++ m_axisZMaxValue = m_areaMaxValue;
++ // Don't allow too much zoom in
++ if ((m_axisZMaxValue - m_axisZMinValue) < m_axisZMinRange) {
++ float adjust = (m_axisZMinRange - (m_axisZMaxValue - m_axisZMinValue)) / 2.f;
++ m_axisZMinValue -= adjust;
++ m_axisZMaxValue += adjust;
++ }
++ }
++
++ void SurfaceGraphModifier::handleAxisDragging(QVector2D delta)
++ {
++ float distance = 0.f;
++
++ // Get scene orientation from active camera
++ float xRotation = m_graph->cameraXRotation();
++
++ // Calculate directional drag multipliers based on rotation
++ float xMulX = qCos(qDegreesToRadians(xRotation));
++ float xMulY = qSin(qDegreesToRadians(xRotation));
++ float zMulX = qSin(qDegreesToRadians(xRotation));
++ float zMulY = qCos(qDegreesToRadians(xRotation));
++
++ // Get the drag amount
++ QPoint move = delta.toPoint();
++
++ // Adjust axes
++ switch (m_state) {
++ //! [17]
++ case StateDraggingX:
++ distance = (move.x() * xMulX - move.y() * xMulY) * m_speedModifier;
++ m_axisXMinValue -= distance;
++ m_axisXMaxValue -= distance;
++ if (m_axisXMinValue < m_areaMinValue) {
++ float dist = m_axisXMaxValue - m_axisXMinValue;
++ m_axisXMinValue = m_areaMinValue;
++ m_axisXMaxValue = m_axisXMinValue + dist;
++ }
++ if (m_axisXMaxValue > m_areaMaxValue) {
++ float dist = m_axisXMaxValue - m_axisXMinValue;
++ m_axisXMaxValue = m_areaMaxValue;
++ m_axisXMinValue = m_axisXMaxValue - dist;
++ }
++ m_graph->axisX()->setRange(m_axisXMinValue, m_axisXMaxValue);
++ break;
++ //! [17]
++ case StateDraggingZ:
++ distance = (move.x() * zMulX + move.y() * zMulY) * m_speedModifier;
++ m_axisZMinValue += distance;
++ m_axisZMaxValue += distance;
++ if (m_axisZMinValue < m_areaMinValue) {
++ float dist = m_axisZMaxValue - m_axisZMinValue;
++ m_axisZMinValue = m_areaMinValue;
++ m_axisZMaxValue = m_axisZMinValue + dist;
++ }
++ if (m_axisZMaxValue > m_areaMaxValue) {
++ float dist = m_axisZMaxValue - m_axisZMinValue;
++ m_axisZMaxValue = m_areaMaxValue;
++ m_axisZMinValue = m_axisZMaxValue - dist;
++ }
++ m_graph->axisZ()->setRange(m_axisZMinValue, m_axisZMaxValue);
++ break;
++ default:
++ break;
++ }
++ }
++
++ //! [18]
++ void SurfaceGraphModifier::onWheel(QWheelEvent *event)
++ {
++ float delta = float(event->angleDelta().y());
++
++ m_axisXMinValue += delta;
++ m_axisXMaxValue -= delta;
++ m_axisZMinValue += delta;
++ m_axisZMaxValue -= delta;
++ checkConstraints();
++
++ float y = (m_axisXMaxValue - m_axisXMinValue) * m_aspectRatio;
++
++ m_graph->axisX()->setRange(m_axisXMinValue, m_axisXMaxValue);
++ m_graph->axisY()->setRange(100.f, y);
++ m_graph->axisZ()->setRange(m_axisZMinValue, m_axisZMaxValue);
++ }
++ //! [18]