-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathvisualization_bezier.py
30 lines (28 loc) · 1.04 KB
/
visualization_bezier.py
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
# Import NURBS-Python and the visualization component
from geomdl import BSpline
from geomdl import utilities
# from geomdl.visualization import VisPlotly as vis
from geomdl.visualization import VisMPL as vis
# Create a BSpline surface instance
surf = BSpline.Surface()
# Set evaluation delta
surf.delta = 0.05
# Set degrees
surf.degree_u = 3
surf.degree_v = 3
# Set control points
control_points = [[0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 0, 1.5], [1, 0, -0.5], [0, 1, 0], [0, 1, 0], [-1, 0, -0.5], [0, 0, 1.5], [0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 1, 0]]
surf.set_ctrlpts(control_points, 4, 4)
# Auto generate knot vectors
surf.knotvector_u = utilities.generate_knot_vector(surf.degree_u, 4)
surf.knotvector_v = utilities.generate_knot_vector(surf.degree_v, 4)
# Evaluate surface
surf.evaluate()
# Visualization config
vis_config = vis.VisConfig(figure_size=[8,8])
# Visualization component
vis_comp = vis.VisSurface(vis_config)
# Set visualization component of the surface
surf.vis = vis_comp
# Render the surface
surf.render()