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
|
# -*- coding: utf-8 -*-
#
# Licensed under the terms of the PyQwt License
# Copyright (C) 2003-2009 Gerard Vermeulen, for the original PyQwt example
# Copyright (c) 2015 Pierre Raybaut, for the PyQt5/PySide port and further
# developments (e.g. ported to PythonQwt API)
# (see LICENSE file for more details)
SHOW = True # Show test in GUI-based test launcher
import os
import numpy as np
from qtpy.QtCore import Qt, QTimer
import qwt
from qwt.tests import utils
FNAMES = ("test_simple.svg", "test_simple.pdf", "test_simple.png")
class SimplePlot(qwt.QwtPlot):
NUM_POINTS = 100
TEST_EXPORT = True
def __init__(self):
qwt.QwtPlot.__init__(self)
self.setTitle("Really simple demo")
self.insertLegend(qwt.QwtLegend(), qwt.QwtPlot.RightLegend)
self.setAxisTitle(qwt.QwtPlot.xBottom, "X-axis")
self.setAxisTitle(qwt.QwtPlot.yLeft, "Y-axis")
self.enableAxis(self.xBottom)
self.setCanvasBackground(Qt.white)
qwt.QwtPlotGrid.make(self, color=Qt.lightGray, width=0, style=Qt.DotLine)
# insert a few curves
x = np.linspace(0.0, 10.0, self.NUM_POINTS)
qwt.QwtPlotCurve.make(x, np.sin(x), "y = sin(x)", self, linecolor="red")
qwt.QwtPlotCurve.make(x, np.cos(x), "y = cos(x)", self, linecolor="blue")
# insert a horizontal marker at y = 0
qwt.QwtPlotMarker.make(
label="y = 0",
align=Qt.AlignRight | Qt.AlignTop,
linestyle=qwt.QwtPlotMarker.HLine,
color="darkGreen",
plot=self,
)
# insert a vertical marker at x = 2 pi
qwt.QwtPlotMarker.make(
xvalue=2 * np.pi,
label="x = 2 pi",
align=Qt.AlignRight | Qt.AlignTop,
linestyle=qwt.QwtPlotMarker.VLine,
color="darkGreen",
plot=self,
)
if self.TEST_EXPORT and utils.TestEnvironment().unattended:
QTimer.singleShot(0, self.export_to_different_formats)
def export_to_different_formats(self):
for fname in FNAMES:
self.exportTo(fname)
def test_simple():
"""Simple plot example"""
utils.test_widget(SimplePlot, size=(600, 400))
for fname in FNAMES:
if os.path.isfile(fname):
os.remove(fname)
if __name__ == "__main__":
test_simple()
|
