diff options
Diffstat (limited to 'examples/charts/modeldata/modeldata.py')
| -rw-r--r-- | examples/charts/modeldata/modeldata.py | 67 |
1 files changed, 13 insertions, 54 deletions
diff --git a/examples/charts/modeldata/modeldata.py b/examples/charts/modeldata/modeldata.py index 1d10190f4..f12eb8999 100644 --- a/examples/charts/modeldata/modeldata.py +++ b/examples/charts/modeldata/modeldata.py @@ -1,43 +1,6 @@ - -############################################################################# -## -## 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$ -## -############################################################################# +# Copyright (C) 2022 The Qt Company Ltd. +# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause +from __future__ import annotations """PySide6 port of the Model Data example from Qt v5.x""" @@ -47,20 +10,20 @@ from random import randrange from PySide6.QtCore import QAbstractTableModel, QModelIndex, QRect, Qt from PySide6.QtGui import QColor, QPainter from PySide6.QtWidgets import (QApplication, QGridLayout, QHeaderView, - QTableView, QWidget) + QTableView, QWidget) from PySide6.QtCharts import QChart, QChartView, QLineSeries, QVXYModelMapper class CustomTableModel(QAbstractTableModel): def __init__(self): - QAbstractTableModel.__init__(self) + super().__init__() self.input_data = [] self.mapping = {} self.column_count = 4 self.row_count = 15 for i in range(self.row_count): - data_vec = [0]*self.column_count + data_vec = [0] * self.column_count for k in range(len(data_vec)): if k % 2 == 0: data_vec[k] = i * 50 + randrange(30) @@ -84,7 +47,7 @@ class CustomTableModel(QAbstractTableModel): else: return "y" else: - return "{}".format(section + 1) + return str(section + 1) def data(self, index, role=Qt.DisplayRole): if role == Qt.DisplayRole: @@ -116,10 +79,9 @@ class CustomTableModel(QAbstractTableModel): self.mapping = {} - class TableWidget(QWidget): def __init__(self): - QWidget.__init__(self) + super().__init__() self.model = CustomTableModel() @@ -140,12 +102,9 @@ class TableWidget(QWidget): self.mapper.setModel(self.model) self.chart.addSeries(self.series) - # for storing color hex from the series - seriesColorHex = "#000000" - # get the color of the series and use it for showing the mapped area - seriesColorHex = "{}".format(self.series.pen().color().name()) - self.model.add_mapping(seriesColorHex, QRect(0, 0, 2, self.model.rowCount())) + self.model.add_mapping(self.series.pen().color().name(), + QRect(0, 0, 2, self.model.rowCount())) # series 2 self.series = QLineSeries() @@ -159,8 +118,8 @@ class TableWidget(QWidget): self.chart.addSeries(self.series) # get the color of the series and use it for showing the mapped area - seriesColorHex = "{}".format(self.series.pen().color().name()) - self.model.add_mapping(seriesColorHex, QRect(2, 0, 2, self.model.rowCount())) + self.model.add_mapping(self.series.pen().color().name(), + QRect(2, 0, 2, self.model.rowCount())) self.chart.createDefaultAxes() self.chart_view = QChartView(self.chart) @@ -180,4 +139,4 @@ if __name__ == "__main__": app = QApplication(sys.argv) w = TableWidget() w.show() - sys.exit(app.exec_()) + sys.exit(app.exec()) |