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
|
#!/usr/bin/env python
from numarray import *
import sys
from PyQt5.Qwt3D import *
from PyQt5.Qt import *
# enable all tracing options of the SIP generated bindings (requires -r option)
if False:
import sip
sip.settracemask(0x3f)
def matrix2d(nx, ny, minx, maxx, miny, maxy, function):
"""Return a data matrix to test the interface to the C++ member function
bool SurfacePlot::loadFromData(
double **, unsigned int, unsigned int, double, double, double, double);
"""
# columns
xs = multiply.outer(
minx + ((maxx-minx)/(nx-1))*arange(nx), ones(ny, Float))
# rows
ys = multiply.outer(
ones((nx,), Float), miny+((maxy-miny)/(ny-1))*arange(ny))
return function(xs, ys)
# matrix2d()
def matrix3d(nx, ny, minx, maxx, miny, maxy, function):
"""Return a data matrix to test the interface to the C++ member function
bool SurfacePlot::loadFromData(
Triple **, unsigned int, unsigned int, bool = false, bool = false);
"""
xyzs = zeros((nx, ny, 3), Float)
# columns
xyzs[:,:,0] = multiply.outer(
minx + ((maxx-minx)/(nx-1))*arange(nx), ones(ny, Float))
# rows
xyzs[:,:,1] = multiply.outer(
ones((nx,), Float), miny+((maxy-miny)/(ny-1))*arange(ny))
# result
xyzs[:,:,2] = function(xyzs[:,:,0], xyzs[:,:,1])
return xyzs
# matrix3d()
def saddle(x, y):
return x*y
# saddle()
class Plot(SurfacePlot):
def __init__(self, *args):
SurfacePlot.__init__(self, *args)
self.setBackgroundColor(RGBA(1.0, 1.0, 0.6))
self.setRotation(30, 0, 15)
self.setScale(1.0, 1.0, 1.0)
nx, ny, minx, maxx, miny, maxy = 3, 5, -1.0, 1.0, -1.0, 1.0
if True:
zs = matrix2d(nx, ny, minx, maxx, miny, maxy, saddle)
print type(zs)
print zs
self.loadFromData(zs, minx, maxx, miny, maxy)
else:
xyzs = matrix3d(nx, ny, minx, maxx, miny, maxy, saddle)
print type(zs)
print xyzs
self.loadFromData(xyzs)
axes = self.coordinates().axes # alias
for axis, label in ((X1, "x"), (Y1, "y"), (Z1, "z")):
axes[axis].setAutoScale(False)
axes[axis].setMajors(5) # 6 major ticks
axes[axis].setMinors(3) # 2 minor ticks
axes[axis].setLabelString(label)
self.setCoordinateStyle(BOX)
self.coordinates().setGridLines(True, True)
self.coordinates().setLineSmooth(True)
self.updateData()
self.updateGL()
# __init__()
# class Plot
def make():
demo = Plot()
demo.show()
# Matrox cards on Linux work better with a resize() after show()
demo.resize(600, 400)
return demo
# make()
def main(args):
app = QApplication(args)
demo = make()
app.exec_()
# main()
# Admire
if __name__ == '__main__':
main(sys.argv)
# Local Variables: ***
# mode: python ***
# End: ***
|
