devbox
凡人多烦事01
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

实战PyQt5：169-数据可视化之三维表面图交互演示_pyqt5显示三维

作者：凡人多烦事01 | 2024-02-23 00:44:25

赞

踩

pyqt5显示三维

表面图是数据可视化的另一种常见显示形式，本文演示如何使用交互方式控制三维表面图显示。

 

一、示例程序的基本功能

本文演示如何使用Q3Dsurface来创建一个简单的3D表面图，并使用一些控件来交互调节，以使3D达到合适的质量效果。这个实例程序演示了一下一些功能:

  • 如何设置QSurfaceDataProxy并为其设置合适的数据；
  • 如何使用QHeightMapSurfaceDataProxy显示一个3D高度图；
  • 3D图形的三种不同的选择模式；
  • 3D图形选定部分的轴范围的用法；
  • 更改主题；
  • 如何设置自定义曲面的渐变效果。

二、示例程序分析

演示程序包括两部分，我们将曲面的交互功能部分封装在类DemoWidget中， 首先调用QWidget::createWindowContainer()创建窗口容器。然后创建水平和垂直布局管理，将图形和垂直布局添加到一个水平布局中，垂直布局中用于添加交互部件。水平布局设置为DemoWidget的布局。

创建另一个类SurfaceGraph来处理数据添加以及与图形的其他交互。

现在简要分析SurfaceGraph类的功能：

设置代理(Proxy)和数据

首先，我们创建一个QSurfaceDataProxy实例,并将其附加到一个新创建的QSurface3DSeries上：

  1. self.sqrtSinProxy = QSurfaceDataProxy()
  2. self.sqrtSinSeries = QSurface3DSeries(self.sqrtSinProxy)

然后我们用简单的平方根和正弦波数据填充代理。这是通过创建一个新实例并向其中添加元素。

  1.     def fillSqrtSinProxy(self):
  2.         stepX = (sampleMax - sampleMin) / float(sampleCountX - 1)
  3.         stepZ = (sampleMax - sampleMin) / float(sampleCountZ - 1)
  4.  
  5.         dataArray = []
  6.         index = 0
  7.         for i in range(sampleCountZ):
  8.             z = min(sampleMax, i * stepZ + sampleMin)
  9.             x_arr = []
  10.             for j in range(sampleCountX):
  11.                 x = min(sampleMax, j * stepX + sampleMin)
  12.                 R = math.sqrt(z * z + x * x) + 0.01
  13.                 y = (math.sin(R) / R + 0.24) * 1.61
  14.                 x_arr.append(QSurfaceDataItem(QVector3D(x, y, z)))
  15.             dataArray.append(x_arr)
  16.  
  17.         self.sqrtSinProxy.resetArray(dataArray)

高度映射是通过实例化一个QHeightMapSurfaceDataProxy和一个包含高度数据的QImage来创建的。QHeightMapSurfaceDataProxy::setValueRanges()方法用于定义映射的值范围。在我们的例子中，地图的假想位置为34.0°N - 40.0°N和18.0°E - 24.0°E。这些值用于显示和定位地图到轴线。

  1.         heightMapImage = QImage(':/maps/mountain')
  2.         self.heightMapProxy = QHeightMapSurfaceDataProxy(heightMapImage)
  3.         self.heightMapSeries = QSurface3DSeries(self.heightMapProxy)
  4.         self.heightMapSeries.setItemLabelFormat('(@xLabel, @zLabel): @yLabel')
  5.         self.heightMapProxy.setValueRanges(34.0, 40.0, 18.0, 24.0)

为了演示不同的Proxy，这个示例有两个单选按钮，用户可以使用它们在Serial之间切换。当用户选择”Sqrt & Sin”单选按钮时，所选Serial将使用以下代码激活。

  1.         self.sqrtSinSeries.setDrawMode(QSurface3DSeries.DrawSurfaceAndWireframe)
  2.         self.sqrtSinSeries.setFlatShadingEnabled(True)
  3.  
  4.         self.graph.axisX().setLabelFormat('%.2f')
  5.         self.graph.axisZ().setLabelFormat('%.2f')
  6.         self.graph.axisX().setRange(sampleMin, sampleMax)
  7.         self.graph.axisY().setRange(0.0, 2.0)
  8.         self.graph.axisZ().setRange(sampleMin, sampleMax)
  9.         self.graph.axisX().setLabelAutoRotation(30)
  10.         self.graph.axisY().setLabelAutoRotation(90)
  11.         self.graph.axisZ().setLabelAutoRotation(30)
  12.  
  13.         self.graph.removeSeries(self.heightMapSeries)
  14.         self.graph.addSeries(self.sqrtSinSeries)

当用户选择”高度地图”单选按钮时，所选Serial将使用以下代码激活。

  1.         self.heightMapSeries.setDrawMode(QSurface3DSeries.DrawSurface)
  2.         self.heightMapSeries.setFlatShadingEnabled(False)
  3.  
  4.         self.graph.axisX().setLabelFormat('%.1f N')
  5.         self.graph.axisZ().setLabelFormat('%.1f E')
  6.         self.graph.axisX().setRange(34.0, 40.0)
  7.         self.graph.axisY().setAutoAdjustRange(True)
  8.         self.graph.axisZ().setRange(18.0, 24.0)
  9.         
  10.         self.graph.axisX().setTitle('纬度')
  11.         self.graph.axisY().setTitle('高度')
  12.         self.graph.axisZ().setTitle('经度')
  13.  
  14.         self.graph.removeSeries(self.sqrtSinSeries)
  15.         self.graph.addSeries(self.heightMapSeries)

选择模式

Q3Dsurface支持三种不同的选择模式。示例中使用单选按钮演示了这些功能，用户可以使用单选按钮激活合适的选择模式。

  1.     def toggleModeNone(self):
  2.         self.graph.setSelectionMode(QAbstract3DGraph.SelectionNone)
  3.  
  4.     def toggleModeItem(self):
  5.         self.graph.setSelectionMode(QAbstract3DGraph.SelectionItem)
  6.  
  7.     def toggleModeSliceRow(self):
  8.         self.graph.setSelectionMode(QAbstract3DGraph.SelectionItemAndRow | QAbstract3DGraph.SelectionSlice)
  10.     def toggleModeSliceColumn(self):
  11.         self.graph.setSelectionMode(QAbstract3DGraph.SelectionItemAndColumn | QAbstract3DGraph.SelectionSlice)

图形的坐标轴范围

示例有四个滑块控件，用于调整X和Z轴的最小值和最大值。

  1.         #为Sqrt&Sin 重新设置滑动条范围
  2.         self.rangeMinX = sampleMin
  3.         self.rangeMinZ = sampleMin
  4.         self.stepX = (sampleMax - sampleMin) / float(sampleCountX - 1)
  5.         self.stepZ = (sampleMax - sampleMin) / float(sampleCountZ - 1)
  6.         self.axisMinSliderX.setMaximum(sampleCountX - 2)
  7.         self.axisMinSliderX.setValue(0)
  8.         self.axisMaxSliderX.setMaximum(sampleCountX - 1)
  9.         self.axisMaxSliderX.setValue(sampleCountX - 1)
  10.         self.axisMinSliderZ.setMaximum(sampleCountZ - 2)
  11.         self.axisMinSliderZ.setValue(0)
  12.         self.axisMaxSliderZ.setMaximum(sampleCountZ - 1)
  13.         self.axisMaxSliderZ.setValue(sampleCountZ - 1)

主题

Q3DSurface支持所有Qt数据可视化的主题。本示例有一个用于选择主题的下拉菜单。下面的方法连接到菜单以激活所选主题。将主题类型更改为另一个预定义主题，该主题将所有主题属性覆盖为预定义值

  1.     def changeTheme(self, theme):
  2.         self.graph.activeTheme().setType(theme)

自定义表面渐变

示例程序演示了使用两个按钮的自定义表面渐变。渐变可以用QLinearGradient定义，其中所需的颜色被设置为位置。下面的代码展示了如何创建示例渐变并将其设置为系列。注意，还需要将颜色样式更改为Q3DTheme::ColorStyleRangeGradient来实际使用渐变。

  1.         gr = QLinearGradient()
  2.         gr.setColorAt(0.0, Qt.black)
  3.         gr.setColorAt(0.33, Qt.blue)
  4.         gr.setColorAt(0.67, Qt.red)
  5.         gr.setColorAt(1.0, Qt.yellow)
  6.  
  7.         self.graph.seriesList()[0].setBaseGradient(gr)
  8.         self.graph.seriesList()[0].setColorStyle(Q3DTheme.ColorStyleRangeGradient)

surfacegraph.py完整代码：

  1. import math
  2. from PyQt5.QtCore import Qt, QObject
  3. from PyQt5.QtGui import  QVector3D, QImage, QLinearGradient
  4. from PyQt5.QtDataVisualization import (Q3DSurface, QSurface3DSeries, QSurfaceDataProxy, QSurfaceDataItem,
  5.                                        QAbstract3DSeries, QValue3DAxis, QAbstract3DGraph, Q3DCamera, 
  6.                                        Q3DTheme, QHeightMapSurfaceDataProxy)
  7.  
  8. sampleCountX = 50
  9. sampleCountZ = 50
  10. heightMapGridStepX = 6
  11. heightMapGridStepZ = 6
  12. sampleMin = -8.0
  13. sampleMax = 8.0
  14.  
  15. class SurfaceGraph(QObject):   
  16.     def __init__(self, surface):
  17.         super(SurfaceGraph, self).__init__()
  18.  
  19.         self.graph = surface
  20.  
  21.         self.graph.setAxisX(QValue3DAxis())
  22.         self.graph.setAxisY(QValue3DAxis())
  23.         self.graph.setAxisZ(QValue3DAxis())
  24.  
  25.         self.sqrtSinProxy = QSurfaceDataProxy()
  26.         self.sqrtSinSeries = QSurface3DSeries(self.sqrtSinProxy)
  27.         self.fillSqrtSinProxy()
  28.  
  29.         heightMapImage = QImage(':/maps/mountain')
  30.         self.heightMapProxy = QHeightMapSurfaceDataProxy(heightMapImage)
  31.         self.heightMapSeries = QSurface3DSeries(self.heightMapProxy)
  32.         self.heightMapSeries.setItemLabelFormat('(@xLabel, @zLabel): @yLabel')
  33.         self.heightMapProxy.setValueRanges(34.0, 40.0, 18.0, 24.0)
  34.  
  35.         self.heightMapWidth = heightMapImage.width()
  36.         self.heightMapHeight = heightMapImage.height()
  37.  
  38.     def toggleModeNone(self):
  39.         self.graph.setSelectionMode(QAbstract3DGraph.SelectionNone)
  40.  
  41.     def toggleModeItem(self):
  42.         self.graph.setSelectionMode(QAbstract3DGraph.SelectionItem)
  43.  
  44.     def toggleModeSliceRow(self):
  45.         self.graph.setSelectionMode(QAbstract3DGraph.SelectionItemAndRow | QAbstract3DGraph.SelectionSlice)
  46.  
  47.     def toggleModeSliceColumn(self):
  48.         self.graph.setSelectionMode(QAbstract3DGraph.SelectionItemAndColumn | QAbstract3DGraph.SelectionSlice) 
  49.  
  50.     def setAxisMinSliderX(self, slider):
  51.         self.axisMinSliderX = slider
  52.  
  53.     def setAxisMaxSliderX(self, slider):
  54.         self.axisMaxSliderX = slider
  55.  
  56.     def setAxisMinSliderZ(self, slider):
  57.         self.axisMinSliderZ = slider
  58.  
  59.     def setAxisMaxSliderZ(self, slider):
  60.         self.axisMaxSliderZ = slider
  61.  
  62.     def fillSqrtSinProxy(self):
  63.         stepX = (sampleMax - sampleMin) / float(sampleCountX - 1)
  64.         stepZ = (sampleMax - sampleMin) / float(sampleCountZ - 1)
  65.  
  66.         dataArray = []
  67.         index = 0
  68.         for i in range(sampleCountZ):
  69.             z = min(sampleMax, i * stepZ + sampleMin)
  70.             x_arr = []
  71.             for j in range(sampleCountX):
  72.                 x = min(sampleMax, j * stepX + sampleMin)
  73.                 R = math.sqrt(z * z + x * x) + 0.01
  74.                 y = (math.sin(R) / R + 0.24) * 1.61
  75.                 x_arr.append(QSurfaceDataItem(QVector3D(x, y, z)))
  76.             dataArray.append(x_arr)
  77.  
  78.         self.sqrtSinProxy.resetArray(dataArray)
  79.  
  80.     def enableSqrtSinModel(self, enable):
  81.         if not enable:
  82.             return
  83.  
  84.         self.sqrtSinSeries.setDrawMode(QSurface3DSeries.DrawSurfaceAndWireframe)
  85.         self.sqrtSinSeries.setFlatShadingEnabled(True)
  86.  
  87.         self.graph.axisX().setLabelFormat('%.2f')
  88.         self.graph.axisZ().setLabelFormat('%.2f')
  89.         self.graph.axisX().setRange(sampleMin, sampleMax)
  90.         self.graph.axisY().setRange(0.0, 2.0)
  91.         self.graph.axisZ().setRange(sampleMin, sampleMax)
  92.         self.graph.axisX().setLabelAutoRotation(30)
  93.         self.graph.axisY().setLabelAutoRotation(90)
  94.         self.graph.axisZ().setLabelAutoRotation(30)
  95.  
  96.         self.graph.removeSeries(self.heightMapSeries)
  97.         self.graph.addSeries(self.sqrtSinSeries)
  98.  
  99.         #为Sqrt&Sin 重新设置滑动条范围
  100.         self.rangeMinX = sampleMin
  101.         self.rangeMinZ = sampleMin
  102.         self.stepX = (sampleMax - sampleMin) / float(sampleCountX - 1)
  103.         self.stepZ = (sampleMax - sampleMin) / float(sampleCountZ - 1)
  104.         self.axisMinSliderX.setMaximum(sampleCountX - 2)
  105.         self.axisMinSliderX.setValue(0)
  106.         self.axisMaxSliderX.setMaximum(sampleCountX - 1)
  107.         self.axisMaxSliderX.setValue(sampleCountX - 1)
  108.         self.axisMinSliderZ.setMaximum(sampleCountZ - 2)
  109.         self.axisMinSliderZ.setValue(0)
  110.         self.axisMaxSliderZ.setMaximum(sampleCountZ - 1)
  111.         self.axisMaxSliderZ.setValue(sampleCountZ - 1)
  112.  
  113.     def enableHeightMapModel(self, enable):
  114.         if not enable:
  115.             return
  116.  
  117.         self.heightMapSeries.setDrawMode(QSurface3DSeries.DrawSurface)
  118.         self.heightMapSeries.setFlatShadingEnabled(False)
  119.  
  120.         self.graph.axisX().setLabelFormat('%.1f N')
  121.         self.graph.axisZ().setLabelFormat('%.1f E')
  122.         self.graph.axisX().setRange(34.0, 40.0)
  123.         self.graph.axisY().setAutoAdjustRange(True)
  124.         self.graph.axisZ().setRange(18.0, 24.0)
  125.         
  126.         self.graph.axisX().setTitle('纬度')
  127.         self.graph.axisY().setTitle('高度')
  128.         self.graph.axisZ().setTitle('经度')
  129.  
  130.         self.graph.removeSeries(self.sqrtSinSeries)
  131.         self.graph.addSeries(self.heightMapSeries)
  132.  
  133.         #重新设置滚动条的范围
  134.         mapGridCountX = self.heightMapWidth / heightMapGridStepX
  135.         mapGridCountZ = self.heightMapHeight / heightMapGridStepZ
  136.         self.rangeMinX = 34.0
  137.         self.rangeMinZ = 18.0
  138.         self.stepX = 6.0 / float(mapGridCountX - 1)
  139.         self.stepZ = 6.0 / float(mapGridCountZ - 1)
  140.         self.axisMinSliderX.setMaximum(mapGridCountX - 2)
  141.         self.axisMinSliderX.setValue(0)
  142.         self.axisMaxSliderX.setMaximum(mapGridCountX - 1)
  143.         self.axisMaxSliderX.setValue(mapGridCountX - 1)
  144.         self.axisMinSliderZ.setMaximum(mapGridCountZ - 2)
  145.         self.axisMinSliderZ.setValue(0)
  146.         self.axisMaxSliderZ.setMaximum(mapGridCountZ - 1)
  147.         self.axisMaxSliderZ.setValue(mapGridCountZ - 1)
  148.  
  149.     def adjustXMin(self, minVal):
  150.         minX = self.stepX * minVal + self.rangeMinX
  151.  
  152.         maxVal = self.axisMaxSliderX.value()
  153.         if minVal >= maxVal:
  154.             maxVal = minVal + 1
  155.             self.axisMaxSliderX.setValue(maxVal)
  156.         maxX = self.stepX * maxVal + self.rangeMinX
  157.  
  158.         self.setAxisXRange(minX, maxX)
  159.  
  160.     def adjustXMax(self, maxVal):
  161.         maxX = self.stepX * maxVal + self.rangeMinX
  162.  
  163.         minVal = self.axisMinSliderX.value()
  164.         if maxVal <= minVal:
  165.             minVal = maxVal - 1
  166.             self.axisMinSliderX.setValue(minVal)
  167.         minX = self.stepX * minVal + self.rangeMinX
  168.  
  169.         self.setAxisXRange(minX, maxX)
  170.  
  171.     def adjustZMin(self, minVal):
  172.         minZ = self.stepZ * minVal + self.rangeMinZ
  173.  
  174.         maxVal = self.axisMaxSliderZ.value()
  175.         if minVal >= maxVal:
  176.             maxVal = minVal + 1
  177.             self.axisMaxSliderZ.setValue(maxVal)
  178.         maxZ = self.stepZ * maxVal + self.rangeMinZ
  179.  
  180.         self.setAxisZRange(minZ, maxZ)
  181.  
  182.     def adjustZMax(self, maxVal):
  183.         maxZ = self.stepZ * maxVal + self.rangeMinZ
  184.  
  185.         minVal = self.axisMinSliderZ.value()
  186.         if maxVal <= minVal:
  187.             minVal = maxVal - 1
  188.             self.axisMinSliderZ.setValue(minVal)
  189.         minZ = self.stepZ * minVal + self.rangeMinZ
  190.  
  191.         self.setAxisZRange(minZ, maxZ)
  192.  
  193.     def setAxisXRange(self, minVal, maxVal):
  194.         self.graph.axisX().setRange(minVal, maxVal)
  195.  
  196.     def setAxisZRange(self, minVal, maxVal):
  197.         self.graph.axisZ().setRange(minVal, maxVal)
  198.  
  199.     def changeTheme(self, theme):
  200.         self.graph.activeTheme().setType(theme)
  201.  
  202.     def setBlackToYellowGradient(self):
  203.         gr = QLinearGradient()
  204.         gr.setColorAt(0.0, Qt.black)
  205.         gr.setColorAt(0.33, Qt.blue)
  206.         gr.setColorAt(0.67, Qt.red)
  207.         gr.setColorAt(1.0, Qt.yellow)
  208.  
  209.         self.graph.seriesList()[0].setBaseGradient(gr)
  210.         self.graph.seriesList()[0].setColorStyle(Q3DTheme.ColorStyleRangeGradient)
  211.  
  212.     def setGreenToRedGradient(self):
  213.         gr = QLinearGradient()
  214.         gr.setColorAt(0.0, Qt.darkGreen)
  215.         gr.setColorAt(0.5, Qt.yellow)
  216.         gr.setColorAt(0.8, Qt.red)
  217.         gr.setColorAt(1.0, Qt.darkRed)
  218.  
  219.         self.graph.seriesList()[0].setBaseGradient(gr)
  220.         self.graph.seriesList()[0].setColorStyle(Q3DTheme.ColorStyleRangeGradient)

surfacedemo.py 完整代码:

  1. import sys
  2. from PyQt5 import QtCore, QtGui, QtWidgets
  3. from PyQt5.QtCore import Qt, QSize
  4. from PyQt5.QtGui import QLinearGradient, QPixmap, QPainter, QBrush, QIcon
  5. from PyQt5.QtWidgets import (QApplication, QWidget, QMessageBox, QSizePolicy,
  6.                             QHBoxLayout, QVBoxLayout, QComboBox, QPushButton,
  7.                             QCheckBox, QSlider, QGroupBox, QLabel, QRadioButton)
  8. from PyQt5.QtDataVisualization import (Q3DSurface, QAbstract3DSeries, QAbstract3DGraph)
  9.  
  10. from surfacegraph import SurfaceGraph
  11. import surface_rc 
  13. class DemoWidget(QWidget):
  14.     def __init__(self, parent=None):
  15.         super(DemoWidget, self).__init__(parent)   
  16.         
  17.          # 设置窗口标题
  18.         self.setWindowTitle('实战 Qt for Python: 三维表面图演示')      
  19.         # 设置窗口大小
  20.         self.resize(640, 480)
  21.       
  22.         self.initUi()
  23.         
  24.     def initUi(self):
  25.         
  26.         widgetgraph  = Q3DSurface()
  27.         container = QWidget.createWindowContainer(widgetgraph)
  29.         if not widgetgraph.hasContext():
  30.             msgBox = QMessageBox()
  31.             msgBox.setText('不能初始化OpenGL上下文')
  32.             msgBox.exec()
  33.             return;
  35.         screenSize = widgetgraph.screen().size()
  36.         container.setMinimumSize(QSize(int(screenSize.width() / 2.0), int(screenSize.height() / 1.5)))
  37.         container.setMaximumSize(screenSize)
  38.         container.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
  39.         container.setFocusPolicy(Qt.StrongFocus)
  41.         hLayout = QHBoxLayout()
  42.         vLayout = QVBoxLayout()
  43.         hLayout.addWidget(container, 1) #左边绘图部分
  44.         hLayout.addLayout(vLayout)      #右边控制部分
  45.         vLayout.setAlignment(Qt.AlignTop)
  47.         modelGroupBox = QGroupBox('模型')
  49.         sqrtSinModelRB = QRadioButton(self)
  50.         sqrtSinModelRB.setText('Sqrt && Sin')
  51.         sqrtSinModelRB.setChecked(False)
  53.         heightMapModelRB = QRadioButton(self)
  54.         heightMapModelRB.setText('高度地图')
  55.         heightMapModelRB.setChecked(False)
  57.         modelVBox = QVBoxLayout()
  58.         modelVBox.addWidget(sqrtSinModelRB)
  59.         modelVBox.addWidget(heightMapModelRB)
  60.         modelGroupBox.setLayout(modelVBox)
  62.         selectionGroupBox = QGroupBox('选择模式')
  64.         modeNoneRB = QRadioButton(self)
  65.         modeNoneRB.setText('无选择')
  66.         modeNoneRB.setChecked(False)
  68.         modeItemRB = QRadioButton(self)
  69.         modeItemRB.setText('条目')
  70.         modeItemRB.setChecked(False)
  72.         modeSliceRowRB = QRadioButton(self)
  73.         modeSliceRowRB.setText('Row Slice')
  74.         modeSliceRowRB.setChecked(False)
  76.         modeSliceColumnRB = QRadioButton(self)
  77.         modeSliceColumnRB.setText('Column Slice')
  78.         modeSliceColumnRB.setChecked(False)
  80.         selectionVBox = QVBoxLayout()
  81.         selectionVBox.addWidget(modeNoneRB)
  82.         selectionVBox.addWidget(modeItemRB)
  83.         selectionVBox.addWidget(modeSliceRowRB)
  84.         selectionVBox.addWidget(modeSliceColumnRB)
  85.         selectionGroupBox.setLayout(selectionVBox)
  87.         axisMinSliderX = QSlider(Qt.Horizontal, self)
  88.         axisMinSliderX.setMinimum(0)
  89.         axisMinSliderX.setTickInterval(1)
  90.         axisMinSliderX.setEnabled(True)
  91.         axisMaxSliderX = QSlider(Qt.Horizontal, self)
  92.         axisMaxSliderX.setMinimum(1)
  93.         axisMaxSliderX.setTickInterval(1)
  94.         axisMaxSliderX.setEnabled(True)
  95.         axisMinSliderZ = QSlider(Qt.Horizontal, self)
  96.         axisMinSliderZ.setMinimum(0)
  97.         axisMinSliderZ.setTickInterval(1)
  98.         axisMinSliderZ.setEnabled(True)
  99.         axisMaxSliderZ = QSlider(Qt.Horizontal, self)
  100.         axisMaxSliderZ.setMinimum(1)
  101.         axisMaxSliderZ.setTickInterval(1)
  102.         axisMaxSliderZ.setEnabled(True)
  104.         #图表主题控制
  105.         themeList = QComboBox(self)
  106.         themeList.addItem('Qt')
  107.         themeList.addItem('Primary Colors')
  108.         themeList.addItem('Digia')
  109.         themeList.addItem('Stone Moss')
  110.         themeList.addItem('Army Blue')
  111.         themeList.addItem('Retro')
  112.         themeList.addItem('Ebony')
  113.         themeList.addItem('Isabelle')
  114.         themeList.setCurrentIndex(6)
  116.         colorGroupBox = QGroupBox('自定义梯度')
  118.         grBtoY = QLinearGradient(0, 0, 1, 100)
  119.         grBtoY.setColorAt(1.0, Qt.black)
  120.         grBtoY.setColorAt(0.67, Qt.blue)
  121.         grBtoY.setColorAt(0.33, Qt.red)
  122.         grBtoY.setColorAt(0.0, Qt.yellow)
  124.         pm = QPixmap(24, 100)
  125.         pmp = QPainter(pm)
  126.         pmp.setBrush(QBrush(grBtoY))
  127.         pmp.setPen(Qt.NoPen)
  128.         pmp.drawRect(0, 0, 24, 100)
  129.         gradientBtoYPB = QPushButton(self)
  130.         gradientBtoYPB.setIcon(QIcon(pm))
  131.         gradientBtoYPB.setIconSize(QSize(24, 100))
  133.         grGtoR = QLinearGradient(0, 0, 1, 100)
  134.         grGtoR.setColorAt(1.0, Qt.darkGreen)
  135.         grGtoR.setColorAt(0.67, Qt.yellow)
  136.         grGtoR.setColorAt(0.33, Qt.red)
  137.         grGtoR.setColorAt(0.0, Qt.darkRed)
  139.         pmp.setBrush(QBrush(grGtoR))
  140.         pmp.setPen(Qt.NoPen)
  141.         pmp.drawRect(0, 0, 24, 100)
  142.         gradientGtoRPB = QPushButton(self)
  143.         gradientGtoRPB.setIcon(QIcon(pm))
  144.         gradientGtoRPB.setIconSize(QSize(24, 100))
  146.         #必须控制先后释放顺序
  147.         del pmp
  148.         del pm
  150.         colorHBox = QHBoxLayout()
  151.         colorHBox.addWidget(gradientBtoYPB)
  152.         colorHBox.addWidget(gradientGtoRPB)
  153.         colorGroupBox.setLayout(colorHBox)
  154.         
  155.         vLayout.addWidget(modelGroupBox)
  156.         vLayout.addWidget(selectionGroupBox)
  157.         vLayout.addWidget(QLabel('列范围'))
  158.         vLayout.addWidget(axisMinSliderX)
  159.         vLayout.addWidget(axisMaxSliderX)
  160.         vLayout.addWidget(QLabel('行范围'))
  161.         vLayout.addWidget(axisMinSliderZ)
  162.         vLayout.addWidget(axisMaxSliderZ)
  163.         vLayout.addWidget(QLabel('主题'))
  164.         vLayout.addWidget(themeList)
  165.         vLayout.addWidget(colorGroupBox)
  167.         self.modifier = SurfaceGraph(widgetgraph)
  169.         heightMapModelRB.toggled.connect(self.modifier.enableHeightMapModel)
  170.         sqrtSinModelRB.toggled.connect(self.modifier.enableSqrtSinModel)
  171.         modeNoneRB.toggled.connect(self.modifier.toggleModeNone)
  173.         modeItemRB.toggled.connect(self.modifier.toggleModeItem)
  174.         modeSliceRowRB.toggled.connect(self.modifier.toggleModeSliceRow)
  175.         modeSliceColumnRB.toggled.connect(self.modifier.toggleModeSliceColumn)
  176.         
  177.         axisMinSliderX.valueChanged.connect(self.modifier.adjustXMin)
  178.         axisMaxSliderX.valueChanged.connect(self.modifier.adjustXMax)
  179.         axisMinSliderZ.valueChanged.connect(self.modifier.adjustZMin)
  180.         axisMaxSliderZ.valueChanged.connect(self.modifier.adjustZMax)
  182.         themeList.currentIndexChanged.connect(self.modifier.changeTheme)
  184.         gradientBtoYPB.pressed.connect(self.modifier.setBlackToYellowGradient)
  185.         gradientGtoRPB.pressed.connect(self.modifier.setGreenToRedGradient)
  187.         self.modifier.setAxisMinSliderX(axisMinSliderX)
  188.         self.modifier.setAxisMaxSliderX(axisMaxSliderX)
  189.         self.modifier.setAxisMinSliderZ(axisMinSliderZ)
  190.         self.modifier.setAxisMaxSliderZ(axisMaxSliderZ)
  192.         sqrtSinModelRB.setChecked(True)
  193.         modeItemRB.setChecked(True)
  194.         themeList.setCurrentIndex(2)
  196.         self.setLayout(hLayout)
  198.     
  199. if __name__ == '__main__':
  200.     app = QApplication(sys.argv)
  201.     window = DemoWidget()
  202.     window.show()
  203.     sys.exit(app.exec())

运行结果如下图。

 3D表面图映射效果

3D表面图交互演示 

三、本文知识点

  • 三维表面图的不同模式(表面模式，表面加线模式等)；
  • 数据可视化中的三维表面图交互控制。

前一篇：实战PyQt5：168-数据可视化之三维散点图交互演示

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/凡人多烦事01/article/detail/132428
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号