diff options
Diffstat (limited to 'examples/datavisualization/surface/surfacegraph.py')
| -rw-r--r-- | examples/datavisualization/surface/surfacegraph.py | 275 |
1 files changed, 275 insertions, 0 deletions
diff --git a/examples/datavisualization/surface/surfacegraph.py b/examples/datavisualization/surface/surfacegraph.py new file mode 100644 index 000000000..8af18b550 --- /dev/null +++ b/examples/datavisualization/surface/surfacegraph.py @@ -0,0 +1,275 @@ +############################################################################# +## +## Copyright (C) 2021 The Qt Company Ltd. +## Contact: http://www.qt.io/licensing/ +## +## This file is part of the Qt for Python examples of the Qt Toolkit. +## +## $QT_BEGIN_LICENSE:BSD$ +## You may use this file under the terms of the BSD license as follows: +## +## "Redistribution and use in source and binary forms, with or without +## modification, are permitted provided that the following conditions are +## met: +## * Redistributions of source code must retain the above copyright +## notice, this list of conditions and the following disclaimer. +## * Redistributions in binary form must reproduce the above copyright +## notice, this list of conditions and the following disclaimer in +## the documentation and/or other materials provided with the +## distribution. +## * Neither the name of The Qt Company Ltd nor the names of its +## contributors may be used to endorse or promote products derived +## from this software without specific prior written permission. +## +## +## THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +## "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +## LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +## A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT +## OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, +## SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +## LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +## DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +## THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +## (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +## OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE." +## +## $QT_END_LICENSE$ +## +############################################################################# + +import math + +from PySide6.QtCore import QObject, Qt, Slot +from PySide6.QtDataVisualization import (Q3DTheme, QAbstract3DGraph, + QHeightMapSurfaceDataProxy, + QSurface3DSeries, QSurfaceDataItem, + QSurfaceDataProxy, QValue3DAxis) +from PySide6.QtGui import QImage, QLinearGradient, QVector3D +from PySide6.QtWidgets import QSlider + +sampleCountX = 50 +sampleCountZ = 50 +heightMapGridStepX = 6 +heightMapGridStepZ = 6 +sampleMin = -8.0 +sampleMax = 8.0 + + +class SurfaceGraph(QObject): + def __init__(self, surface, parent=None): + QObject.__init__(self, parent) + + self.m_graph = surface + self.m_graph.setAxisX(QValue3DAxis()) + self.m_graph.setAxisY(QValue3DAxis()) + self.m_graph.setAxisZ(QValue3DAxis()) + + self.m_sqrtSinProxy = QSurfaceDataProxy() + self.m_sqrtSinSeries = QSurface3DSeries(self.m_sqrtSinProxy) + self.fillSqrtSinProxy() + + heightMapImage = QImage("mountain.png") + self.m_heightMapProxy = QHeightMapSurfaceDataProxy(heightMapImage) + self.m_heightMapSeries = QSurface3DSeries(self.m_heightMapProxy) + self.m_heightMapSeries.setItemLabelFormat("(@xLabel, @zLabel): @yLabel") + self.m_heightMapProxy.setValueRanges(34.0, 40.0, 18.0, 24.0) + + self.m_heightMapWidth = heightMapImage.width() + self.m_heightMapHeight = heightMapImage.height() + + self.m_axisMinSliderX = QSlider() + self.m_axisMaxSliderX = QSlider() + self.m_axisMinSliderZ = QSlider() + self.m_axisMaxSliderZ = QSlider() + self.m_rangeMinX = 0.0 + self.m_rangeMinZ = 0.0 + self.m_stepX = 0.0 + self.m_stepZ = 0.0 + + def fillSqrtSinProxy(self): + stepX = (sampleMax - sampleMin) / float(sampleCountX - 1) + stepZ = (sampleMax - sampleMin) / float(sampleCountZ - 1) + + dataArray = [] + for i in range(sampleCountZ): + newRow = [] + # Keep values within range bounds, since just adding step can cause + # minor drift due to the rounding errors. + z = min(sampleMax, (i * stepZ + sampleMin)) + for j in range(sampleCountX): + x = min(sampleMax, (j * stepX + sampleMin)) + R = math.sqrt(z * z + x * x) + 0.01 + y = (math.sin(R) / R + 0.24) * 1.61 + newRow.append(QSurfaceDataItem(QVector3D(x, y, z))) + dataArray.append(newRow) + + self.m_sqrtSinProxy.resetArray(dataArray) + + def enableSqrtSinModel(self, enable): + if enable: + self.m_sqrtSinSeries.setDrawMode(QSurface3DSeries.DrawSurfaceAndWireframe) + self.m_sqrtSinSeries.setFlatShadingEnabled(True) + + self.m_graph.axisX().setLabelFormat("%.2f") + self.m_graph.axisZ().setLabelFormat("%.2f") + self.m_graph.axisX().setRange(sampleMin, sampleMax) + self.m_graph.axisY().setRange(0.0, 2.0) + self.m_graph.axisZ().setRange(sampleMin, sampleMax) + self.m_graph.axisX().setLabelAutoRotation(30) + self.m_graph.axisY().setLabelAutoRotation(90) + self.m_graph.axisZ().setLabelAutoRotation(30) + + self.m_graph.removeSeries(self.m_heightMapSeries) + self.m_graph.addSeries(self.m_sqrtSinSeries) + + # Reset range sliders for Sqrt&Sin + self.m_rangeMinX = sampleMin + self.m_rangeMinZ = sampleMin + self.m_stepX = (sampleMax - sampleMin) / float(sampleCountX - 1) + self.m_stepZ = (sampleMax - sampleMin) / float(sampleCountZ - 1) + self.m_axisMinSliderX.setMaximum(sampleCountX - 2) + self.m_axisMinSliderX.setValue(0) + self.m_axisMaxSliderX.setMaximum(sampleCountX - 1) + self.m_axisMaxSliderX.setValue(sampleCountX - 1) + self.m_axisMinSliderZ.setMaximum(sampleCountZ - 2) + self.m_axisMinSliderZ.setValue(0) + self.m_axisMaxSliderZ.setMaximum(sampleCountZ - 1) + self.m_axisMaxSliderZ.setValue(sampleCountZ - 1) + + def enableHeightMapModel(self, enable): + if enable: + self.m_heightMapSeries.setDrawMode(QSurface3DSeries.DrawSurface) + self.m_heightMapSeries.setFlatShadingEnabled(False) + + self.m_graph.axisX().setLabelFormat("%.1f N") + self.m_graph.axisZ().setLabelFormat("%.1f E") + self.m_graph.axisX().setRange(34.0, 40.0) + self.m_graph.axisY().setAutoAdjustRange(True) + self.m_graph.axisZ().setRange(18.0, 24.0) + + self.m_graph.axisX().setTitle("Latitude") + self.m_graph.axisY().setTitle("Height") + self.m_graph.axisZ().setTitle("Longitude") + + self.m_graph.removeSeries(self.m_sqrtSinSeries) + self.m_graph.addSeries(self.m_heightMapSeries) + + # Reset range sliders for height map + mapGridCountX = self.m_heightMapWidth / heightMapGridStepX + mapGridCountZ = self.m_heightMapHeight / heightMapGridStepZ + self.m_rangeMinX = 34.0 + self.m_rangeMinZ = 18.0 + self.m_stepX = 6.0 / float(mapGridCountX - 1) + self.m_stepZ = 6.0 / float(mapGridCountZ - 1) + self.m_axisMinSliderX.setMaximum(mapGridCountX - 2) + self.m_axisMinSliderX.setValue(0) + self.m_axisMaxSliderX.setMaximum(mapGridCountX - 1) + self.m_axisMaxSliderX.setValue(mapGridCountX - 1) + self.m_axisMinSliderZ.setMaximum(mapGridCountZ - 2) + self.m_axisMinSliderZ.setValue(0) + self.m_axisMaxSliderZ.setMaximum(mapGridCountZ - 1) + self.m_axisMaxSliderZ.setValue(mapGridCountZ - 1) + + def adjustXMin(self, minimum): + minX = self.m_stepX * float(minimum) + self.m_rangeMinX + + maximum = self.m_axisMaxSliderX.value() + if minimum >= maximum: + maximum = minimum + 1 + self.m_axisMaxSliderX.setValue(maximum) + maxX = self.m_stepX * maximum + self.m_rangeMinX + + self.setAxisXRange(minX, maxX) + + def adjustXMax(self, maximum): + maxX = self.m_stepX * float(maximum) + self.m_rangeMinX + + minimum = self.m_axisMinSliderX.value() + if maximum <= minimum: + minimum = maximum - 1 + self.m_axisMinSliderX.setValue(minimum) + minX = self.m_stepX * minimum + self.m_rangeMinX + + self.setAxisXRange(minX, maxX) + + def adjustZMin(self, minimum): + minZ = self.m_stepZ * float(minimum) + self.m_rangeMinZ + + maximum = self.m_axisMaxSliderZ.value() + if minimum >= maximum: + maximum = minimum + 1 + self.m_axisMaxSliderZ.setValue(maximum) + maxZ = self.m_stepZ * maximum + self.m_rangeMinZ + + self.setAxisZRange(minZ, maxZ) + + def adjustZMax(self, maximum): + maxX = self.m_stepZ * float(maximum) + self.m_rangeMinZ + + minimum = self.m_axisMinSliderZ.value() + if maximum <= minimum: + minimum = maximum - 1 + self.m_axisMinSliderZ.setValue(minimum) + minX = self.m_stepZ * minimum + self.m_rangeMinZ + + self.setAxisZRange(minX, maxX) + + def setAxisXRange(self, minimum, maximum): + self.m_graph.axisX().setRange(minimum, maximum) + + def setAxisZRange(self, minimum, maximum): + self.m_graph.axisZ().setRange(minimum, maximum) + + @Slot() + def changeTheme(self, theme): + self.m_graph.activeTheme().setType(Q3DTheme.Theme(theme)) + + def setBlackToYellowGradient(self): + gr = QLinearGradient() + gr.setColorAt(0.0, Qt.black) + gr.setColorAt(0.33, Qt.blue) + gr.setColorAt(0.67, Qt.red) + gr.setColorAt(1.0, Qt.yellow) + + self.m_graph.seriesList()[0].setBaseGradient(gr) + self.m_graph.seriesList()[0].setColorStyle(Q3DTheme.ColorStyleRangeGradient) + + def setGreenToRedGradient(self): + gr = QLinearGradient() + gr.setColorAt(0.0, Qt.darkGreen) + gr.setColorAt(0.5, Qt.yellow) + gr.setColorAt(0.8, Qt.red) + gr.setColorAt(1.0, Qt.darkRed) + + series = self.m_graph.seriesList()[0] + series.setBaseGradient(gr) + series.setColorStyle(Q3DTheme.ColorStyleRangeGradient) + + def toggleModeNone(self): + self.m_graph.setSelectionMode(QAbstract3DGraph.SelectionNone) + + def toggleModeItem(self): + self.m_graph.setSelectionMode(QAbstract3DGraph.SelectionItem) + + def toggleModeSliceRow(self): + self.m_graph.setSelectionMode( + QAbstract3DGraph.SelectionItemAndRow | QAbstract3DGraph.SelectionSlice + ) + + def toggleModeSliceColumn(self): + self.m_graph.setSelectionMode( + QAbstract3DGraph.SelectionItemAndColumn | QAbstract3DGraph.SelectionSlice + ) + + def setAxisMinSliderX(self, slider): + self.m_axisMinSliderX = slider + + def setAxisMaxSliderX(self, slider): + self.m_axisMaxSliderX = slider + + def setAxisMinSliderZ(self, slider): + self.m_axisMinSliderZ = slider + + def setAxisMaxSliderZ(self, slider): + self.m_axisMaxSliderZ = slider |