#!/usr/bin/env python # -*- coding: utf-8 -*- """ Created on 2019/10/2 @author: Irony @site: https://pyqt.site , https://github.com/PyQt5 @email: 892768447@qq.com @file: ChartThemes @description: 图表主题动画等 """ ############################################################################# ## ## Copyright (C) 2013 Riverbank Computing Limited ## Copyright (C) 2012 Digia Plc ## All rights reserved. ## ## This file is part of the PyQtChart examples. ## ## $QT_BEGIN_LICENSE$ ## Licensees holding valid Qt Commercial licenses may use this file in ## accordance with the Qt Commercial License Agreement provided with the ## Software or, alternatively, in accordance with the terms contained in ## a written agreement between you and Digia. ## $QT_END_LICENSE$ ## ############################################################################# import random try: from PyQt5.QtChart import (QAreaSeries, QBarSet, QChart, QChartView, QLineSeries, QPieSeries, QScatterSeries, QSplineSeries, QStackedBarSeries) from PyQt5.QtCore import pyqtSlot, QPointF, Qt from PyQt5.QtGui import QColor, QPainter, QPalette from PyQt5.QtWidgets import QApplication, QMainWindow, QCheckBox, QComboBox, QGridLayout, QHBoxLayout, \ QLabel, QSizePolicy, QWidget except ImportError: from PySide2.QtCore import Slot as pyqtSlot, QPointF, Qt from PySide2.QtGui import QColor, QPainter, QPalette from PySide2.QtWidgets import QApplication, QMainWindow, QCheckBox, QComboBox, QGridLayout, QHBoxLayout, \ QLabel, QSizePolicy, QWidget from PySide2.QtCharts import QtCharts QChartView = QtCharts.QChartView QChart = QtCharts.QChart QAreaSeries = QtCharts.QAreaSeries QBarSet = QtCharts.QBarSet QLineSeries = QtCharts.QLineSeries QPieSeries = QtCharts.QPieSeries QScatterSeries = QtCharts.QScatterSeries QSplineSeries = QtCharts.QSplineSeries QStackedBarSeries = QtCharts.QStackedBarSeries class ThemeWidget(QWidget): def __init__(self, parent=None): super(ThemeWidget, self).__init__(parent) self.m_charts = [] self.m_listCount = 3 self.m_valueMax = 10 self.m_valueCount = 7 self.m_dataTable = self.generateRandomData(self.m_listCount, self.m_valueMax, self.m_valueCount) self.m_themeComboBox = self.createThemeBox() self.m_antialiasCheckBox = QCheckBox("Anti-aliasing") self.m_animatedComboBox = self.createAnimationBox() self.m_legendComboBox = self.createLegendBox() self.connectSignals() # Create the layout. baseLayout = QGridLayout() settingsLayout = QHBoxLayout() settingsLayout.addWidget(QLabel("Theme:")) settingsLayout.addWidget(self.m_themeComboBox) settingsLayout.addWidget(QLabel("Animation:")) settingsLayout.addWidget(self.m_animatedComboBox) settingsLayout.addWidget(QLabel("Legend:")) settingsLayout.addWidget(self.m_legendComboBox) settingsLayout.addWidget(self.m_antialiasCheckBox) settingsLayout.addStretch() baseLayout.addLayout(settingsLayout, 0, 0, 1, 3) # Create the charts. chartView = QChartView(self.createAreaChart()) baseLayout.addWidget(chartView, 1, 0) self.m_charts.append(chartView) chartView = QChartView(self.createBarChart(self.m_valueCount)) baseLayout.addWidget(chartView, 1, 1) self.m_charts.append(chartView) chartView = QChartView(self.createLineChart()) baseLayout.addWidget(chartView, 1, 2) self.m_charts.append(chartView) chartView = QChartView(self.createPieChart()) # Funny things happen if the pie slice labels no not fit the screen... chartView.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored) baseLayout.addWidget(chartView, 2, 0) self.m_charts.append(chartView) chartView = QChartView(self.createSplineChart()) baseLayout.addWidget(chartView, 2, 1) self.m_charts.append(chartView) chartView = QChartView(self.createScatterChart()) baseLayout.addWidget(chartView, 2, 2) self.m_charts.append(chartView) self.setLayout(baseLayout) # Set the defaults. self.m_antialiasCheckBox.setChecked(True) self.updateUI() def connectSignals(self): self.m_themeComboBox.currentIndexChanged.connect(self.updateUI) self.m_antialiasCheckBox.toggled.connect(self.updateUI) self.m_animatedComboBox.currentIndexChanged.connect(self.updateUI) self.m_legendComboBox.currentIndexChanged.connect(self.updateUI) def generateRandomData(self, listCount, valueMax, valueCount): random.seed() dataTable = [] for i in range(listCount): dataList = [] yValue = 0.0 f_valueCount = float(valueCount) for j in range(valueCount): yValue += random.uniform(0, valueMax) / f_valueCount value = QPointF( j + random.random() * self.m_valueMax / f_valueCount, yValue) label = "Slice " + str(i) + ":" + str(j) dataList.append((value, label)) dataTable.append(dataList) return dataTable def createThemeBox(self): themeComboBox = QComboBox() themeComboBox.addItem("Light", QChart.ChartThemeLight) themeComboBox.addItem("Blue Cerulean", QChart.ChartThemeBlueCerulean) themeComboBox.addItem("Dark", QChart.ChartThemeDark) themeComboBox.addItem("Brown Sand", QChart.ChartThemeBrownSand) themeComboBox.addItem("Blue NCS", QChart.ChartThemeBlueNcs) themeComboBox.addItem("High Contrast", QChart.ChartThemeHighContrast) themeComboBox.addItem("Blue Icy", QChart.ChartThemeBlueIcy) return themeComboBox def createAnimationBox(self): animationComboBox = QComboBox() animationComboBox.addItem("No Animations", QChart.NoAnimation) animationComboBox.addItem("GridAxis Animations", QChart.GridAxisAnimations) animationComboBox.addItem("Series Animations", QChart.SeriesAnimations) animationComboBox.addItem("All Animations", QChart.AllAnimations) return animationComboBox def createLegendBox(self): legendComboBox = QComboBox() legendComboBox.addItem("No Legend ", 0) legendComboBox.addItem("Legend Top", Qt.AlignTop) legendComboBox.addItem("Legend Bottom", Qt.AlignBottom) legendComboBox.addItem("Legend Left", Qt.AlignLeft) legendComboBox.addItem("Legend Right", Qt.AlignRight) return legendComboBox def createAreaChart(self): chart = QChart() chart.setTitle("Area chart") # The lower series is initialized to zero values. lowerSeries = None y_points = [] for i, data_list in enumerate(self.m_dataTable): upperSeries = QLineSeries(chart) for j, (value, _) in enumerate(data_list): y = value.y() if lowerSeries is None: upperSeries.append(QPointF(j, y)) y_points.append(y) else: new_y = y_points[i] + y upperSeries.append(QPointF(j, new_y)) y_points[j] += new_y area = QAreaSeries(upperSeries, lowerSeries) area.setName("Series " + str(i)) chart.addSeries(area) lowerSeries = upperSeries chart.createDefaultAxes() return chart def createBarChart(self, valueCount): chart = QChart() chart.setTitle("Bar chart") series = QStackedBarSeries(chart) for i, data_list in enumerate(self.m_dataTable): set = QBarSet("Bar set " + str(i)) for value, _ in data_list: set << value.y() series.append(set) chart.addSeries(series) chart.createDefaultAxes() return chart def createLineChart(self): chart = QChart() chart.setTitle("Line chart") for i, data_list in enumerate(self.m_dataTable): series = QLineSeries(chart) for value, _ in data_list: series.append(value) series.setName("Series " + str(i)) chart.addSeries(series) chart.createDefaultAxes() return chart def createPieChart(self): chart = QChart() chart.setTitle("Pie chart") pieSize = 1.0 / len(self.m_dataTable) for i, data_list in enumerate(self.m_dataTable): series = QPieSeries(chart) for value, label in data_list: slice = series.append(label, value.y()) if series.count() == 1: slice.setLabelVisible() slice.setExploded() hPos = (pieSize / 2) + (i / float(len(self.m_dataTable))) series.setPieSize(pieSize) series.setHorizontalPosition(hPos) series.setVerticalPosition(0.5) chart.addSeries(series) return chart def createSplineChart(self): chart = QChart() chart.setTitle("Spline chart") for i, data_list in enumerate(self.m_dataTable): series = QSplineSeries(chart) for value, _ in data_list: series.append(value) series.setName("Series " + str(i)) chart.addSeries(series) chart.createDefaultAxes() return chart def createScatterChart(self): chart = QChart() chart.setTitle("Scatter chart") for i, data_list in enumerate(self.m_dataTable): series = QScatterSeries(chart) for value, _ in data_list: series.append(value) series.setName("Series " + str(i)) chart.addSeries(series) chart.createDefaultAxes() return chart @pyqtSlot() def updateUI(self): theme = self.m_themeComboBox.itemData( self.m_themeComboBox.currentIndex()) if self.m_charts[0].chart().theme() != theme: for chartView in self.m_charts: chartView.chart().setTheme(QChart.ChartTheme(theme)) pal = self.window().palette() if theme == QChart.ChartThemeLight: pal.setColor(QPalette.Window, QColor(0xf0f0f0)) pal.setColor(QPalette.WindowText, QColor(0x404044)) elif theme == QChart.ChartThemeDark: pal.setColor(QPalette.Window, QColor(0x121218)) pal.setColor(QPalette.WindowText, QColor(0xd6d6d6)) elif theme == QChart.ChartThemeBlueCerulean: pal.setColor(QPalette.Window, QColor(0x40434a)) pal.setColor(QPalette.WindowText, QColor(0xd6d6d6)) elif theme == QChart.ChartThemeBrownSand: pal.setColor(QPalette.Window, QColor(0x9e8965)) pal.setColor(QPalette.WindowText, QColor(0x404044)) elif theme == QChart.ChartThemeBlueNcs: pal.setColor(QPalette.Window, QColor(0x018bba)) pal.setColor(QPalette.WindowText, QColor(0x404044)) elif theme == QChart.ChartThemeHighContrast: pal.setColor(QPalette.Window, QColor(0xffab03)) pal.setColor(QPalette.WindowText, QColor(0x181818)) elif theme == QChart.ChartThemeBlueIcy: pal.setColor(QPalette.Window, QColor(0xcee7f0)) pal.setColor(QPalette.WindowText, QColor(0x404044)) else: pal.setColor(QPalette.Window, QColor(0xf0f0f0)) pal.setColor(QPalette.WindowText, QColor(0x404044)) self.window().setPalette(pal) checked = self.m_antialiasCheckBox.isChecked() for chartView in self.m_charts: chartView.setRenderHint(QPainter.Antialiasing, checked) options = QChart.AnimationOptions( self.m_animatedComboBox.itemData( self.m_animatedComboBox.currentIndex())) if self.m_charts[0].chart().animationOptions() != options: for chartView in self.m_charts: chartView.chart().setAnimationOptions(options) alignment = self.m_legendComboBox.itemData( self.m_legendComboBox.currentIndex()) for chartView in self.m_charts: legend = chartView.chart().legend() if alignment == 0: legend.hide() else: legend.setAlignment(Qt.Alignment(alignment)) legend.show() if __name__ == '__main__': import sys app = QApplication(sys.argv) window = QMainWindow() widget = ThemeWidget() window.setCentralWidget(widget) window.resize(900, 600) window.show() sys.exit(app.exec_())