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
|
"""
Demonstration of Tube
"""
import sys
from vispy import scene
from vispy.geometry.torusknot import TorusKnot
from colorsys import hsv_to_rgb
import numpy as np
canvas = scene.SceneCanvas(keys='interactive')
view = canvas.central_widget.add_view()
points1 = TorusKnot(5, 3).first_component[:-1]
points1[:, 0] -= 20.
points1[:, 2] -= 15.
points2 = points1.copy()
points2[:, 2] += 30.
points3 = points1.copy()
points3[:, 0] += 41.
points3[:, 2] += 30
points4 = points1.copy()
points4[:, 0] += 41.
points5 = points1.copy()
points5[:, 0] += 20.4
points5[:, 2] += 15
colors = np.linspace(0, 1, len(points1))
colors = np.array([hsv_to_rgb(c, 1, 1) for c in colors])
vertex_colors = np.random.random(8 * len(points1))
vertex_colors = np.array([hsv_to_rgb(c, 1, 1) for c in vertex_colors])
l1 = scene.visuals.Tube(points1,
shading='flat',
color=colors, # this is overridden by
# the vertex_colors argument
vertex_colors=vertex_colors,
tube_points=8)
l2 = scene.visuals.Tube(points2,
color=['red', 'green', 'blue'],
shading='smooth',
tube_points=8)
l3 = scene.visuals.Tube(points3,
color=colors,
shading='flat',
tube_points=8,
closed=True)
l4 = scene.visuals.Tube(points4,
color=colors,
shading='smooth',
tube_points=8,
mode='lines')
# generate sine wave radii
radii = np.sin(2 * np.pi * 440 * np.arange(points5.shape[0]) / 44000)
radii = (radii + 1.5) / 2
l5 = scene.visuals.Tube(points5,
radius=radii,
color='white',
shading='smooth',
closed=True,
tube_points=8)
view.add(l1)
view.add(l2)
view.add(l3)
view.add(l4)
view.add(l5)
view.camera = scene.TurntableCamera()
# tube does not expose its limits yet
view.camera.set_range((-20, 20), (-20, 20), (-20, 20))
canvas.show()
if __name__ == '__main__':
if sys.flags.interactive != 1:
canvas.app.run()
|
