from compas.colors import Color
from compas.geometry import NurbsSurface
from compas.geometry import Point
from compas.geometry import Polyline
from compas_viewer import Viewer
points = [
[
Point(0, 0, 0),
Point(1, 0, +0),
Point(2, 0, +0),
Point(3, 0, +0),
Point(4, 0, +0),
Point(5, 0, 0),
],
[
Point(0, 1, 0),
Point(1, 1, -1),
Point(2, 1, -1),
Point(3, 1, -1),
Point(4, 1, -1),
Point(5, 1, 0),
],
[
Point(0, 2, 0),
Point(1, 2, -1),
Point(2, 2, +2),
Point(3, 2, +2),
Point(4, 2, -1),
Point(5, 2, 0),
],
[
Point(0, 3, 0),
Point(1, 3, -1),
Point(2, 3, +2),
Point(3, 3, +2),
Point(4, 3, -1),
Point(5, 3, 0),
],
[
Point(0, 4, 0),
Point(1, 4, -1),
Point(2, 4, -1),
Point(3, 4, -1),
Point(4, 4, -1),
Point(5, 4, 0),
],
[
Point(0, 5, 0),
Point(1, 5, +0),
Point(2, 5, +0),
Point(3, 5, +0),
Point(4, 5, +0),
Point(5, 5, 0),
],
]
weights = [
[1.0, 1.0, 1.0, 1.0, 1.0, 1.0],
[1.0, 1.0, 1.0, 1.0, 1.0, 1.0],
[1.0, 1.0, 1.0, 1.0, 1.0, 1.0],
[1.0, 1.0, 1.0, 1.0, 1.0, 1.0],
[1.0, 1.0, 1.0, 1.0, 1.0, 1.0],
[1.0, 1.0, 1.0, 1.0, 1.0, 1.0],
]
surface = NurbsSurface.from_parameters(
points=points,
weights=weights,
knots_u=[
1.0,
1 + 1 / 9,
1 + 2 / 9,
1 + 3 / 9,
1 + 4 / 9,
1 + 5 / 9,
1 + 6 / 9,
1 + 7 / 9,
1 + 8 / 9,
2.0,
],
knots_v=[0.0, 1 / 9, 2 / 9, 3 / 9, 4 / 9, 5 / 9, 6 / 9, 7 / 9, 8 / 9, 1.0],
mults_u=[1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
mults_v=[1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
degree_u=3,
degree_v=3,
)
# ==============================================================================
# Visualisation
# ==============================================================================
viewer = Viewer()
viewer.scene.add(surface)
# control polygon
points = list(surface.points)
viewer.scene.add([Polyline(row) for row in points], linewidth=1, linecolor=Color(0.3, 0.3, 0.3))
viewer.scene.add([Polyline(col) for col in zip(*points)], linewidth=1, linecolor=Color(0.3, 0.3, 0.3))
viewer.scene.add(points, pointsize=10)
# isocurves
u_curves = []
for u in surface.space_u(7): # type: ignore
u_curves.append(surface.isocurve_u(u).to_polyline())
v_curves = []
for v in surface.space_v(7): # type: ignore
v_curves.append(surface.isocurve_v(v).to_polyline())
viewer.scene.add(u_curves, linecolor=Color(0.8, 0.8, 0.8), linewidth=3)
viewer.scene.add(v_curves, linecolor=Color(0.8, 0.8, 0.8), linewidth=3)
viewer.show()