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
|
# -*- coding: utf-8 -*-
# Copyright (c) Vispy Development Team. All Rights Reserved.
# Distributed under the (new) BSD License. See LICENSE.txt for more info.
"""
* Demonstrates the usage of SurfacePlot() using meshes created in spherical
and cylindrical coordinates to generate surfaces with rotational symmetry.
* Meshes can be generated in xy, yz or zx axes. See create_cylinder().
* Also this example demonstrates the creation of surfaces with discontinuity
See create_circular_hole()
"""
import sys
import numpy as np
from vispy import scene
from vispy.scene.visuals import SurfacePlot
from vispy.scene.visuals import XYZAxis
def create_cylinder(radius, length, center=(0.0, 0.0, 0.0)):
""" Creates the data of a cylinder oriented along z axis whose center, radius and length are given as inputs
Based on the example given at: https://stackoverflow.com/a/49311446/2602319
"""
z = np.linspace(0, length, 100)
theta = np.linspace(0, 2 * np.pi, 100)
theta_grid, z_grid = np.meshgrid(theta, z)
x_grid = radius * np.cos(theta_grid) + center[0]
y_grid = radius * np.sin(theta_grid) + center[1]
z_grid = z_grid + center[2]
return x_grid, y_grid, z_grid
def create_paraboloid(a, b, c, radius=1.0, center=(0.0, 0.0, 0.0)):
"""
Creates the data of a paraboloid whose center, radius and a, b, c parameters are given as inputs
"""
# Generate the grid in cylindrical coordinates
r = np.linspace(0, radius, 100)
theta = np.linspace(0, 2 * np.pi, 100)
R, THETA = np.meshgrid(r, theta)
# Convert to rectangular coordinates
x_grid, y_grid = R * np.cos(THETA), R * np.sin(THETA)
z_grid = c * ((x_grid / a) ** 2 + (y_grid / b) ** 2) # Elliptic paraboloid
return x_grid + center[0], y_grid + center[1], z_grid + center[2]
def create_sphere(radius=1.0, center=(0.0, 0.0, 0.0)):
"""
Creates the data of a sphere whose center, and radius are given as inputs
"""
# Generate the grid in spherical coordinates
# Names of the spherical coordinate axes according to ISO convention
theta = np.linspace(0, np.pi, 50)
phi = np.linspace(0, 2 * np.pi, 50)
PHI, THETA = np.meshgrid(phi, theta)
RHO = radius # Size of the sphere
# Convert to cartesian coordinates
x_grid = (RHO * np.sin(THETA) * np.cos(PHI)) + center[0]
y_grid = (RHO * np.sin(THETA) * np.sin(PHI)) + center[1]
z_grid = (RHO * np.cos(THETA)) + center[2]
return x_grid, y_grid, z_grid
def create_circular_hole(x_grid, y_grid, hole_radius=0.5, center=(0.0, 0.0)):
X = np.where((x_grid - center[0]) ** 2 + (y_grid - center[1]) ** 2 <= hole_radius ** 2, np.NAN, x_grid)
Y = np.where((x_grid - center[0]) ** 2 + (y_grid - center[1]) ** 2 <= hole_radius ** 2, np.NAN, y_grid)
return X, Y
# Prepare canvas
canvas = scene.SceneCanvas(keys='interactive', size=(800, 600), show=True)
view = canvas.central_widget.add_view()
camera = scene.cameras.TurntableCamera(fov=60, elevation=30, azimuth=50, parent=view.scene)
camera.set_range((-4, 4), (-4, 4), (-4, 4))
view.camera = camera
# Add a 3D axis
XYZAxis(parent=view.scene)
x_grid, y_grid, z_grid = create_paraboloid(6, 3, 3, radius=2, center=(0, 0, 3))
x_grid, y_grid = create_circular_hole(x_grid, y_grid, hole_radius=0.5, center=(0.0, 0.0))
SurfacePlot(x_grid, y_grid, z_grid, color="aquamarine", parent=view.scene)
x_grid, y_grid, z_grid = create_cylinder(1, 3, center=(0, 0, 0))
SurfacePlot(x_grid, y_grid, z_grid, color="lightgreen", parent=view.scene)
x_grid, y_grid, z_grid = create_sphere(0.5, center=(0, 0, 4))
SurfacePlot(x_grid, y_grid, z_grid, color='lightseagreen', parent=view.scene)
if __name__ == '__main__' and sys.flags.interactive == 0:
canvas.app.run()
|
