#!/usr/bin/python3.11 ############################################################################# ## ## Copyright (C) 2014 Riverbank Computing Limited. ## Copyright (C) 2014 Digia Plc ## All rights reserved. ## For any questions to Digia, please use contact form at http://qt.digia.com ## ## This file is part of the QtDataVisualization module. ## ## Licensees holding valid Qt Enterprise licenses may use this file in ## accordance with the Qt Enterprise License Agreement provided with the ## Software or, alternatively, in accordance with the terms contained in ## a written agreement between you and Digia. ## ## If you have questions regarding the use of this file, please use ## contact form at http://qt.digia.com ## ############################################################################# import math from PyQt5.QtCore import QFileInfo, QObject, QSize, Qt, QTimer from PyQt5.QtDataVisualization import (Q3DCamera, Q3DScatter, Q3DTheme, QAbstract3DGraph, QAbstract3DSeries, QScatter3DSeries, QScatterDataItem) from PyQt5.QtGui import QColor, QLinearGradient, QQuaternion, QVector3D from PyQt5.QtWidgets import (QApplication, QHBoxLayout, QLabel, QPushButton, QSlider, QSizePolicy, QVBoxLayout, QWidget) class ScatterDataModifier(QObject): verticalRange = 8.0 horizontalRange = verticalRange ellipse_a = horizontalRange / 3.0 ellipse_b = verticalRange doublePi = math.pi * 2.0 radiansToDegrees = 360.0 / doublePi animationFrames = 30.0 def __init__(self, scatter): super(ScatterDataModifier, self).__init__() mesh_dir = QFileInfo(__file__).absolutePath() + '/mesh' self.m_graph = scatter self.m_rotationTimer = QTimer() self.m_fieldLines = 12 self.m_arrowsPerLine = 16 self.m_magneticField = QScatter3DSeries() self.m_sun = QScatter3DSeries() self.m_angleOffset = 0.0 self.m_angleStep = self.doublePi / self.m_arrowsPerLine / self.animationFrames self.m_graph.setShadowQuality(QAbstract3DGraph.ShadowQualityNone) self.m_graph.scene().activeCamera().setCameraPreset( Q3DCamera.CameraPresetFront) # Magnetic field lines use custom narrow arrow. self.m_magneticField.setItemSize(0.2) self.m_magneticField.setMesh(QAbstract3DSeries.MeshUserDefined) self.m_magneticField.setUserDefinedMesh(mesh_dir + '/narrowarrow.obj') fieldGradient = QLinearGradient(0, 0, 16, 1024) fieldGradient.setColorAt(0.0, Qt.black) fieldGradient.setColorAt(1.0, Qt.white) self.m_magneticField.setBaseGradient(fieldGradient) self.m_magneticField.setColorStyle(Q3DTheme.ColorStyleRangeGradient) # For 'sun' we use a custom large sphere. self.m_sun.setItemSize(0.2) self.m_sun.setName("Sun") self.m_sun.setItemLabelFormat("@seriesName") self.m_sun.setMesh(QAbstract3DSeries.MeshUserDefined) self.m_sun.setUserDefinedMesh(mesh_dir + '/largesphere.obj') self.m_sun.setBaseColor(QColor(0xff, 0xbb, 0x00)) self.m_sun.dataProxy().addItem(QScatterDataItem(QVector3D())) self.m_graph.addSeries(self.m_magneticField) self.m_graph.addSeries(self.m_sun) # Configure the axes according to the data. self.m_graph.axisX().setRange(-self.horizontalRange, self.horizontalRange) self.m_graph.axisY().setRange(-self.verticalRange, self.verticalRange) self.m_graph.axisZ().setRange(-self.horizontalRange, self.horizontalRange) self.m_graph.axisX().setSegmentCount(self.horizontalRange) self.m_graph.axisZ().setSegmentCount(self.horizontalRange) self.m_rotationTimer.timeout.connect(self.triggerRotation) self.toggleRotation() self.generateData() def generateData(self): magneticFieldArray = [] for i in range(self.m_fieldLines): horizontalAngle = (self.doublePi * i) / self.m_fieldLines xCenter = self.ellipse_a * math.cos(horizontalAngle) zCenter = self.ellipse_a * math.sin(horizontalAngle) # Rotate - arrow is always tangential to the origin. yRotation = QQuaternion.fromAxisAndAngle(0.0, 1.0, 0.0, horizontalAngle * self.radiansToDegrees) for j in range(self.m_arrowsPerLine): # Calculate the point on the ellipse centered on the origin and # parallel to the x-axis. verticalAngle = ((self.doublePi * j) / self.m_arrowsPerLine) + self.m_angleOffset xUnrotated = self.ellipse_a * math.cos(verticalAngle) y = self.ellipse_b * math.sin(verticalAngle) # Rotate the ellipse around the y-axis. xRotated = xUnrotated * math.cos(horizontalAngle) zRotated = xUnrotated * math.sin(horizontalAngle) # Add the offset. x = xCenter + xRotated z = zCenter + zRotated zRotation = QQuaternion.fromAxisAndAngle(0.0, 0.0, 1.0, verticalAngle * self.radiansToDegrees) totalRotation = yRotation * zRotation itm = QScatterDataItem(QVector3D(x, y, z), totalRotation) magneticFieldArray.append(itm) if self.m_graph.selectedSeries() is self.m_magneticField: self.m_graph.clearSelection() self.m_magneticField.dataProxy().resetArray(magneticFieldArray) def setFieldLines(self, lines): self.m_fieldLines = lines self.generateData() def setArrowsPerLine(self, arrows): self.m_arrowsPerLine = arrows self.m_angleOffset = 0.0 self.m_angleStep = self.doublePi / self.m_arrowsPerLine / self.animationFrames self.generateData() def triggerRotation(self): self.m_angleOffset += self.m_angleStep self.generateData() def toggleSun(self): self.m_sun.setVisible(not self.m_graph.seriesList()[1].isVisible()) def toggleRotation(self): if self.m_rotationTimer.isActive(): self.m_rotationTimer.stop() else: self.m_rotationTimer.start(15) if __name__ == '__main__': import sys app = QApplication(sys.argv) graph = Q3DScatter() container = QWidget.createWindowContainer(graph) screenSize = graph.screen().size() container.setMinimumSize( QSize(screenSize.width() / 2, screenSize.height() / 1.5)) container.setMaximumSize(screenSize) container.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) container.setFocusPolicy(Qt.StrongFocus) widget = QWidget() hLayout = QHBoxLayout(widget) vLayout = QVBoxLayout() hLayout.addWidget(container, 1) hLayout.addLayout(vLayout) widget.setWindowTitle("Item rotations example - Magnetic field of the sun") toggleRotationButton = QPushButton("Toggle animation") toggleSunButton = QPushButton("Toggle Sun") fieldLinesSlider = QSlider(Qt.Horizontal) fieldLinesSlider.setTickInterval(1) fieldLinesSlider.setMinimum(1) fieldLinesSlider.setValue(12) fieldLinesSlider.setMaximum(128) arrowsSlider = QSlider(Qt.Horizontal) arrowsSlider.setTickInterval(1) arrowsSlider.setMinimum(8) arrowsSlider.setValue(16) arrowsSlider.setMaximum(32) vLayout.addWidget(toggleRotationButton) vLayout.addWidget(toggleSunButton) vLayout.addWidget(QLabel("Field Lines (1 - 128):")) vLayout.addWidget(fieldLinesSlider) vLayout.addWidget(QLabel("Arrows per line (8 - 32):")) vLayout.addWidget(arrowsSlider, 1, Qt.AlignTop) modifier = ScatterDataModifier(graph) toggleRotationButton.clicked.connect(modifier.toggleRotation) toggleSunButton.clicked.connect(modifier.toggleSun) fieldLinesSlider.valueChanged.connect(modifier.setFieldLines) arrowsSlider.valueChanged.connect(modifier.setArrowsPerLine) widget.show() sys.exit(app.exec_())