feat(_extend_ladybug_geometry): extend polyline2d

ladybug-tools · Jan 18, 2022 · bd46f52 · bd46f52
1 parent 06fabec
commit bd46f52
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ladybug_vtk/_extend_ladybug_geometry.py b/ladybug_vtk/_extend_ladybug_geometry.py
@@ -1,14 +1,18 @@
 """Add capability to turn a Ladybug geometry object into a VTK polydata object."""
 
-from ladybug_geometry.geometry2d import Point2D, LineSegment2D
+from ladybug_geometry.geometry2d import Point2D, LineSegment2D, Polyline2D
 from ladybug_geometry.geometry3d import Point3D, LineSegment3D, Polyline3D, Arc3D, \
     Mesh3D, Face3D, Polyface3D, Cone, Sphere, Cylinder
 from .fromgeometry import from_line2d, from_point2d, from_point3d, from_line3d, \
     from_polyline3d, from_arc3d, from_mesh3d, from_face3d, from_polyface3d, from_cone,\
-    from_sphere, from_cylinder
+    from_sphere, from_cylinder, from_polyline2d
 
+# 2d goemetry
 Point2D.to_polydata = from_point2d
 LineSegment2D.to_polydata = from_line2d
+Polyline2D.to_polydata = from_polyline2d
+
+# 3d geometry
 Point3D.to_polydata = from_point3d
 LineSegment3D.to_polydata = from_line3d
 Polyline3D.to_polydata = from_polyline3d