1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
# Copyright (C) 2023 The Qt Company Ltd.
# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
from PySide6.QtCore import QObject, QSize, Qt
from PySide6.QtWidgets import (QCheckBox, QComboBox, QCommandLinkButton,
QLabel, QHBoxLayout, QSizePolicy,
QVBoxLayout, QWidget, )
from PySide6.QtDataVisualization import (QAbstract3DSeries, Q3DScatter)
from scatterdatamodifier import ScatterDataModifier
class ScatterGraph(QObject):
def __init__(self):
super().__init__()
self._scatterGraph = Q3DScatter()
self._container = None
self._scatterWidget = None
def initialize(self, minimum_graph_size, maximum_graph_size):
if not self._scatterGraph.hasContext():
return -1
self._scatterWidget = QWidget()
hLayout = QHBoxLayout(self._scatterWidget)
self._container = QWidget.createWindowContainer(self._scatterGraph, self._scatterWidget)
self._container.setMinimumSize(minimum_graph_size)
self._container.setMaximumSize(maximum_graph_size)
self._container.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self._container.setFocusPolicy(Qt.StrongFocus)
hLayout.addWidget(self._container, 1)
vLayout = QVBoxLayout()
hLayout.addLayout(vLayout)
cameraButton = QCommandLinkButton(self._scatterWidget)
cameraButton.setText("Change camera preset")
cameraButton.setDescription("Switch between a number of preset camera positions")
cameraButton.setIconSize(QSize(0, 0))
itemCountButton = QCommandLinkButton(self._scatterWidget)
itemCountButton.setText("Toggle item count")
itemCountButton.setDescription("Switch between 900 and 10000 data points")
itemCountButton.setIconSize(QSize(0, 0))
rangeButton = QCommandLinkButton(self._scatterWidget)
rangeButton.setText("Toggle axis ranges")
rangeButton.setDescription("Switch between automatic axis ranges and preset ranges")
rangeButton.setIconSize(QSize(0, 0))
backgroundCheckBox = QCheckBox(self._scatterWidget)
backgroundCheckBox.setText("Show background")
backgroundCheckBox.setChecked(True)
gridCheckBox = QCheckBox(self._scatterWidget)
gridCheckBox.setText("Show grid")
gridCheckBox.setChecked(True)
smoothCheckBox = QCheckBox(self._scatterWidget)
smoothCheckBox.setText("Smooth dots")
smoothCheckBox.setChecked(True)
itemStyleList = QComboBox(self._scatterWidget)
itemStyleList.addItem("Sphere", QAbstract3DSeries.MeshSphere)
itemStyleList.addItem("Cube", QAbstract3DSeries.MeshCube)
itemStyleList.addItem("Minimal", QAbstract3DSeries.MeshMinimal)
itemStyleList.addItem("Point", QAbstract3DSeries.MeshPoint)
itemStyleList.setCurrentIndex(0)
themeList = QComboBox(self._scatterWidget)
themeList.addItem("Qt")
themeList.addItem("Primary Colors")
themeList.addItem("Digia")
themeList.addItem("Stone Moss")
themeList.addItem("Army Blue")
themeList.addItem("Retro")
themeList.addItem("Ebony")
themeList.addItem("Isabelle")
themeList.setCurrentIndex(3)
shadowQuality = QComboBox(self._scatterWidget)
shadowQuality.addItem("None")
shadowQuality.addItem("Low")
shadowQuality.addItem("Medium")
shadowQuality.addItem("High")
shadowQuality.addItem("Low Soft")
shadowQuality.addItem("Medium Soft")
shadowQuality.addItem("High Soft")
shadowQuality.setCurrentIndex(6)
vLayout.addWidget(cameraButton)
vLayout.addWidget(itemCountButton)
vLayout.addWidget(rangeButton)
vLayout.addWidget(backgroundCheckBox)
vLayout.addWidget(gridCheckBox)
vLayout.addWidget(smoothCheckBox)
vLayout.addWidget(QLabel("Change dot style"))
vLayout.addWidget(itemStyleList)
vLayout.addWidget(QLabel("Change theme"))
vLayout.addWidget(themeList)
vLayout.addWidget(QLabel("Adjust shadow quality"))
vLayout.addWidget(shadowQuality, 1, Qt.AlignTop)
self._modifier = ScatterDataModifier(self._scatterGraph, self)
cameraButton.clicked.connect(self._modifier.changePresetCamera)
itemCountButton.clicked.connect(self._modifier.toggleItemCount)
rangeButton.clicked.connect(self._modifier.toggleRanges)
backgroundCheckBox.stateChanged.connect(self._modifier.setBackgroundEnabled)
gridCheckBox.stateChanged.connect(self._modifier.setGridEnabled)
smoothCheckBox.stateChanged.connect(self._modifier.setSmoothDots)
self._modifier.backgroundEnabledChanged.connect(backgroundCheckBox.setChecked)
self._modifier.gridEnabledChanged.connect(gridCheckBox.setChecked)
itemStyleList.currentIndexChanged.connect(self._modifier.changeStyle)
themeList.currentIndexChanged.connect(self._modifier.changeTheme)
shadowQuality.currentIndexChanged.connect(self._modifier.changeShadowQuality)
self._modifier.shadowQualityChanged.connect(shadowQuality.setCurrentIndex)
self._scatterGraph.shadowQualityChanged.connect(self._modifier.shadowQualityUpdatedByVisual)
return True
def scatterWidget(self):
return self._scatterWidget
|