- Few more drawing functions finished.

Author: andrewtliew

src/compas_blender/utilities/drawing.py

@@ -3,13 +3,13 @@
 from __future__ import division
 from __future__ import print_function
 
+from compas_blender.utilities import delete_object
 from compas_blender.utilities import set_objects_layer
 # from compas_blender.utilities import set_deselect
 # from compas_blender.utilities import set_objects_visible
-# from compas_blender.utilities import delete_object
 
 from compas.geometry import centroid_points
-# from compas.geometry import distance_point_point
+from compas.geometry import distance_point_point
 from compas.geometry import subtract_vectors
 
 try:
@@ -24,7 +24,7 @@
 __all__ = [
     'create_material',
     'xdraw_points',
-    # 'xdraw_lines',
+    'xdraw_lines',
     # 'xdraw_geodesics',
     # 'xdraw_breps',
     # 'xdraw_cylinders',
@@ -36,23 +36,25 @@
     # 'xdraw_faces',
     # 'xdraw_pointcloud',
     # 'xdraw_texts',
-    # 'draw_cylinder',
+    'draw_cylinder',
     # 'draw_plane',
     # 'draw_text',
     # 'draw_line',
 ]
 
 
-# def _link_objects(objects, copy=None):
+def _link_objects(objects, copy=None, layer=None):
 
-#     for object in objects:
-#         bpy.context.collection.objects.link(object)
+    for object in objects:
+        bpy.context.collection.objects.link(object)
 
-#     set_deselect()
-#     if copy:
-#         set_objects_visible(objects=[copy], visible=False)  # temp hack
+    if copy:
+        delete_object(object=copy)
 
-#     return objects
+    if layer:
+        set_objects_layer(objects=objects, layer=layer)
+
+    return objects
 
 
 def create_material(color, alpha=1):
@@ -68,26 +70,27 @@ def create_material(color, alpha=1):
         return bpy.data.materials[ckey]
 
 
-# def xdraw_points(points, **kwargs):
+def xdraw_points(points, layer=None):
 
-#     bpy.ops.object.empty_add(type='SPHERE', radius=1, location=[0, 0, 0])
-#     copy = bpy.context.object
+    bpy.ops.object.empty_add(type='SPHERE', radius=1, location=[0, 0, 0])
+    copy = bpy.context.object
 
-#     objects = [0] * len(points)
+    objects = [0] * len(points)
 
-#     for c, data in enumerate(points):
+    for c, data in enumerate(points):
 
-#         object          = copy.copy()
-#         object.scale   *= data.get('radius', 1)
-#         object.location = data.get('pos', [0, 0, 0])
-#         object.name     = data.get('name', 'point')
-#         # layer
-#         objects[c]      = object
+        object          = copy.copy()
+        object.scale   *= data.get('radius', 1)
+        object.location = data.get('pos', [0, 0, 0])
+        object.name     = data.get('name', 'point')
+        objects[c]      = object
+
+    return _link_objects(objects, copy, layer)
 
-#     return _link_objects(objects, copy)
 
+def xdraw_lines(lines, centroid=True, **kwargs):
 
-# def xdraw_lines(lines, centroid=True, **kwargs):
+    pass
 
 #     objects = [0] * len(lines)
 
@@ -281,23 +284,25 @@ def xdraw_mesh(vertices, edges=None, faces=None, name='mesh', color=[1, 1, 1], c
 #     return _link_objects(objects, copy)
 
 
-# def draw_cylinder(start, end, radius=1, color=[1, 1, 1], layer=None, div=10, **kwargs):
+def draw_cylinder(start, end, radius=1, color=[1, 1, 1], layer=None, div=10, name='cylinder'):
 
-#     bpy.ops.mesh.primitive_cylinder_add(radius=1, depth=1, vertices=div, location=[0, 0, 0])
-#     object = bpy.context.object
+    bpy.ops.mesh.primitive_cylinder_add(radius=1, depth=1, vertices=div, location=[0, 0, 0])
 
-#     L      = distance_point_point(start, end)
-#     pos    = centroid_points([start, end])
+    L   = distance_point_point(start, end)
+    pos = centroid_points([start, end])
+
+    object = bpy.context.object
+    object.name = name
+    object.rotation_euler[1] = acos((end[2] - start[2]) / L)
+    object.rotation_euler[2] = atan2(end[1] - start[1], end[0] - start[0])
+    object.location = pos
+    object.scale = ((radius, radius, L))
+    object.data.materials.append(create_material(color=color))
 
-#     object.name = 'cylinder'
-#     object.rotation_euler[1] = acos((end[2] - start[2]) / L)
-#     object.rotation_euler[2] = atan2(end[1] - start[1], end[0] - start[0])
-#     object.location = pos
-#     object.scale = ((radius, radius, L))
-#     object.data.materials.append(create_material(color=color))
-#     # layer
+    if layer:
+        set_objects_layer(objects=[object], layer=layer)
 
-#     return object
+    return object
 
 
 # def draw_plane(Lx=1, Ly=1, dx=0.5, dy=0.5, name='plane', layer=None, color=[1, 1, 1]):
@@ -341,45 +346,49 @@ def xdraw_mesh(vertices, edges=None, faces=None, name='mesh', color=[1, 1, 1], c
 #     return xdraw_mesh(name=name, vertices=vertices, faces=faces, layer=layer, color=color, centroid=False)
 
 
-# def draw_text(radius=1, pos=[0, 0, 0], text='text', layer=None, color=[1, 1, 1]):
+def draw_text(radius=1, pos=[0, 0, 0], text='text', layer=None, color=[1, 1, 1]):
+
+    bpy.ops.object.text_add(view_align=False)
 
-#     bpy.ops.object.text_add(view_align=False)
-#     object = bpy.context.object
+    object           = bpy.context.object
+    object.scale    *= radius
+    object.location  = pos
+    object.data.body = text
+    object.data.materials.append(create_material(color=color))
+
+    if layer:
+        set_objects_layer(objects=[object], layer=layer)
 
-#     object.scale    *= radius
-#     object.location  = pos
-#     object.data.body = text
-#     object.data.materials.append(create_material(color=color))
-#     # layer
+    return object
 
-#     return object
 
+def draw_line(start=[0, 0, 0], end=[1, 1, 1], width=0.05, centroid=True, name='line', color=[1, 1, 1], layer=None):
 
-# def draw_line(start=[0, 0, 0], end=[1, 1, 1], width=0.05, centroid=True, name='line', color=[1, 1, 1], **kwargs):
+    mp = centroid_points([start, end]) if centroid else [0, 0, 0]
 
-#     mp = centroid_points([start, end]) if centroid else [0, 0, 0]
+    curve = bpy.data.curves.new(name, type='CURVE')
+    curve.dimensions = '3D'
+    object = bpy.data.objects.new(name, curve)
+    object.location = mp
 
-#     curve = bpy.data.curves.new(name, type='CURVE')
-#     curve.dimensions = '3D'
-#     object = bpy.data.objects.new(name, curve)
-#     object.location = mp
+    spline = curve.splines.new('NURBS')
+    spline.points.add(2)
+    spline.points[0].co = list(subtract_vectors(start, mp)) + [1]
+    spline.points[1].co = list(subtract_vectors(end, mp)) + [1]
+    spline.order_u = 1
 
-#     spline = curve.splines.new('NURBS')
-#     spline.points.add(2)
-#     spline.points[0].co = list(subtract_vectors(start, mp)) + [1]
-#     spline.points[1].co = list(subtract_vectors(end, mp)) + [1]
-#     spline.order_u = 1
+    object.data.fill_mode = 'FULL'
+    object.data.bevel_depth = width
+    object.data.bevel_resolution = 0
+    object.data.resolution_u = 2
+    object.data.materials.append(create_material(color=color))
 
-#     object.data.fill_mode = 'FULL'
-#     object.data.bevel_depth = width
-#     object.data.bevel_resolution = 0
-#     object.data.resolution_u = 2
-#     object.data.materials.append(create_material(color=color))
-#     # layer
+    if layer:
+        set_objects_layer(objects=[object], layer=layer)
 
-#     bpy.context.collection.objects.link(object)
+    bpy.context.collection.objects.link(object)
 
-#     return object
+    return object
 
 
 # ==============================================================================
@@ -395,9 +404,9 @@ def xdraw_mesh(vertices, edges=None, faces=None, name='mesh', color=[1, 1, 1], c
 
     # clear_layer(layer='Collection 1')
 
-    # n = 5
+    n = 10
 
-    # points  = [{'pos': [0, 0, i], 'radius': 0.2, 'name': 'pt'} for i in range(n)]
+    points  = [{'pos': [0, 0, i], 'radius': 0.2, 'name': 'pt'} for i in range(n)]
     # lines   = [{'start': [1, 1, i], 'end': [1, 0, i], 'radius': 0.1, 'color': [1, 0, 1]} for i in range(n)]
     # cyls    = [{'start': [2, 1, i], 'end': [2, 0, i], 'radius': 0.1, 'color': [0, 0, 1]} for i in range(n)]
     # spheres = [{'pos': [3, 0, i], 'radius': 0.5, 'color': [0, 1, 0]} for i in range(n)]
@@ -406,7 +415,7 @@ def xdraw_mesh(vertices, edges=None, faces=None, name='mesh', color=[1, 1, 1], c
 
     # tic = time()
 
-    # xdraw_points(points=points)
+    xdraw_points(points=points)
     # xdraw_lines(lines=lines)
     # xdraw_cylinders(cylinders=cyls)
     # xdraw_spheres(spheres=spheres)
@@ -421,5 +430,9 @@ def xdraw_mesh(vertices, edges=None, faces=None, name='mesh', color=[1, 1, 1], c
     faces    = [[0, 1, 2], [2, 3, 0]]
     mesh     = xdraw_mesh(name='mesh', vertices=vertices, faces=faces, layer='Collection 2', color=[1, 0, 1])
 
+    draw_cylinder(start=[0, 0, 0], end=[1, 1, 1], radius=0.1, color=[1, 0, 1], layer='Collection 2')
+    draw_line(start=[2, 2, 2], end=[1, 1, 1], width=0.05, name='line', color=[1, 1, 0])
+    draw_text(radius=1, pos=[0, 0, 0], text='text', layer='Collection 2', color=[1, 0, 0])
+
     # objects = xdraw_pointcloud(points=points)
     # set_objects_show_names(objects=objects, show=True)