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
|
# -*- coding: utf-8 -*-
# -----------------------------------------------------------------------------
# Copyright (c) Vispy Development Team. All Rights Reserved.
# Distributed under the (new) BSD License. See LICENSE.txt for more info.
# -----------------------------------------------------------------------------
"""
Demonstration of the Klein bottle rendering using Mesh.
"""
import sys
from vispy import app, scene
from vispy.geometry.parametric import surface
def klein(u, v):
from math import pi, cos, sin
if u < pi:
x = 3 * cos(u) * (1 + sin(u)) + \
(2 * (1 - cos(u) / 2)) * cos(u) * cos(v)
z = -8 * sin(u) - 2 * (1 - cos(u) / 2) * sin(u) * cos(v)
else:
x = 3 * cos(u) * (1 + sin(u)) + (2 * (1 - cos(u) / 2)) * cos(v + pi)
z = -8 * sin(u)
y = -2 * (1 - cos(u) / 2) * sin(v)
return x/5, y/5, z/5
# Prepare canvas
canvas = scene.SceneCanvas(keys='interactive', size=(800, 600), show=True)
# Set up a viewbox to display the image with interactive pan/zoom
view = canvas.central_widget.add_view()
# Add camera
view.camera = scene.cameras.ArcballCamera(parent=view.scene)
# Add mesh
vertices, indices = surface(klein, urepeat=3)
indices = indices.reshape(len(indices)//3, 3)
mesh = scene.visuals.Mesh(vertices=vertices['position'], faces=indices,
color='white', parent=view.scene, shading='smooth')
if __name__ == '__main__' and sys.flags.interactive == 0:
app.run()
|
