align datastructure objects in rhino

Author: Licini

src/compas/scene/volmeshobject.py

@@ -46,7 +46,7 @@ class VolMeshObject(SceneObject):
         Default is ``True``.
     show_faces : Union[bool, sequence[int]]
         Flag for showing or hiding the faces, or a list of keys for the faces to show.
-        Default is ``True``.
+        Default is ``False``.
     show_cells : bool
         Flag for showing or hiding the cells, or a list of keys for the cells to show.
         Default is ``True``.
@@ -76,7 +76,7 @@ def __init__(self, volmesh, **kwargs):
         self.edgewidth = kwargs.get("edgewidth", 1.0)
         self.show_vertices = kwargs.get("show_vertices", False)
         self.show_edges = kwargs.get("show_edges", True)
-        self.show_faces = kwargs.get("show_faces", True)
+        self.show_faces = kwargs.get("show_faces", False)
         self.show_cells = kwargs.get("show_cells", True)
 
     @property

src/compas_rhino/scene/graphobject.py

@@ -32,6 +32,12 @@ class GraphObject(RhinoSceneObject, BaseGraphObject):
 
     def __init__(self, graph, **kwargs):
         super(GraphObject, self).__init__(graph=graph, **kwargs)
+        self._guids_nodes = None
+        self._guids_edges = None
+        self._guids_nodelabels = None
+        self._guids_edgelabels = None
+        self._guids_spheres = None
+        self._guids_pipes = None
 
     # ==========================================================================
     # clear
@@ -45,8 +51,7 @@ def clear(self):
         None
 
         """
-        guids = compas_rhino.objects.get_objects(name="{}.*".format(self.graph.name))  # type: ignore
-        compas_rhino.objects.delete_objects(guids, purge=True)
+        compas_rhino.objects.delete_objects(self.guids, purge=True)
 
     def clear_nodes(self):
         """Delete all nodes drawn by this scene object.
@@ -56,8 +61,7 @@ def clear_nodes(self):
         None
 
         """
-        guids = compas_rhino.objects.get_objects(name="{}.node.*".format(self.graph.name))  # type: ignore
-        compas_rhino.objects.delete_objects(guids, purge=True)
+        compas_rhino.objects.delete_objects(self._guids_nodes, purge=True)
 
     def clear_edges(self):
         """Delete all edges drawn by this scene object.
@@ -67,44 +71,24 @@ def clear_edges(self):
         None
 
         """
-        guids = compas_rhino.objects.get_objects(name="{}.edge.*".format(self.graph.name))  # type: ignore
-        compas_rhino.objects.delete_objects(guids, purge=True)
+        compas_rhino.objects.delete_objects(self._guids_edges, purge=True)
 
     # ==========================================================================
     # draw
     # ==========================================================================
 
-    def draw(
-        self,
-        nodes=None,
-        edges=None,
-        nodecolor=None,
-        edgecolor=None,
-    ):
+    def draw(self):
         """Draw the graph using the chosen visualisation settings.
 
-        Parameters
-        ----------
-        nodes : list[int], optional
-            A list of nodes to draw.
-            Default is None, in which case all nodes are drawn.
-        edges : list[tuple[int, int]], optional
-            A list of edges to draw.
-            The default is None, in which case all edges are drawn.
-        nodecolor : :class:`compas.colors.Color` | dict[int, :class:`compas.colors.Color`], optional
-            The color of the nodes.
-        edgecolor : :class:`compas.colors.Color` | dict[tuple[int, int], :class:`compas.colors.Color`], optional
-            The color of the edges.
-
         Returns
         -------
         list[System.Guid]
             The GUIDs of the created Rhino objects.
 
         """
         self.clear()
-        guids = self.draw_nodes(nodes=nodes, color=nodecolor)
-        guids += self.draw_edges(edges=edges, color=edgecolor)
+        guids = self.draw_nodes(nodes=self.show_nodes, color=self.nodecolor, group=self.group)
+        guids += self.draw_edges(edges=self.show_edges, color=self.edgecolor, group=self.group)
         self._guids = guids
         return self.guids
 
@@ -129,6 +113,9 @@ def draw_nodes(self, nodes=None, color=None, group=None):
 
         self.nodecolor = color
 
+        if nodes is True:
+            nodes = list(self.graph.nodes())
+
         for node in nodes or self.graph.nodes():  # type: ignore
             name = "{}.node.{}".format(self.graph.name, node)  # type: ignore
             attr = attributes(name=name, color=self.nodecolor[node], layer=self.layer)  # type: ignore
@@ -141,6 +128,8 @@ def draw_nodes(self, nodes=None, color=None, group=None):
         if group:
             self.add_to_group(group, guids)
 
+        self._guids_nodes = guids
+
         return guids
 
     def draw_edges(self, edges=None, color=None, group=None, show_direction=False):
@@ -169,6 +158,9 @@ def draw_edges(self, edges=None, color=None, group=None, show_direction=False):
         arrow = "end" if show_direction else None
         self.edgecolor = color
 
+        if edges is True:
+            edges = list(self.graph.edges())
+
         for edge in edges or self.graph.edges():  # type: ignore
             u, v = edge
 
@@ -184,6 +176,8 @@ def draw_edges(self, edges=None, color=None, group=None, show_direction=False):
         if group:
             self.add_to_group(group, guids)
 
+        self._guids_edges = guids
+
         return guids
 
     # =============================================================================
@@ -232,6 +226,8 @@ def draw_nodelabels(self, text, color=None, group=None, fontheight=10, fontface=
         if group:
             self.add_to_group(group, guids)
 
+        self._guids_nodelabels = guids
+
         return guids
 
     def draw_edgelabels(self, text, color=None, group=None, fontheight=10, fontface="Arial Regular"):
@@ -280,6 +276,8 @@ def draw_edgelabels(self, text, color=None, group=None, fontheight=10, fontface=
         if group:
             self.add_to_group(group, guids)
 
+        self._guids_edgelabels = guids
+
         return guids
 
     # =============================================================================
@@ -322,6 +320,8 @@ def draw_spheres(self, radius, color=None, group=None):
         if group:
             self.add_to_group(group, guids)
 
+        self._guids_spheres = guids
+
         return guids
 
     def draw_pipes(self, radius, color=None, group=None):
@@ -361,4 +361,6 @@ def draw_pipes(self, radius, color=None, group=None):
         if group:
             self.add_to_group(group, guids)
 
+        self._guids_pipes = guids
+
         return guids

src/compas_rhino/scene/meshobject.py

@@ -42,16 +42,12 @@ class MeshObject(RhinoSceneObject, BaseMeshObject):
     disjoint : bool, optional
         Draw the faces of the mesh disjointed.
         Default is ``False``.
-    group : str, optional
-        The name of the group to add the mesh components. The group will be created if not already present.
-        Default is ``None``.
 
     """
 
-    def __init__(self, mesh, **kwargs):
+    def __init__(self, mesh, disjoint=False, **kwargs):
         super(MeshObject, self).__init__(mesh=mesh, **kwargs)
-        self.disjoint = kwargs.get("disjoint", False)
-        self.group = kwargs.get("group", None)
+        self.disjoint = disjoint
         self._guid_mesh = None
         self._guids_faces = None
         self._guids_edges = None

src/compas_rhino/scene/sceneobject.py

@@ -15,14 +15,18 @@ class RhinoSceneObject(SceneObject):
     ----------
     layer : str, optional
         A layer name.
+    group : str, optional
+        The name of the group to add the mesh components. The group will be created if not already present.
+        Default is ``None``.
     **kwargs : dict, optional
         Additional keyword arguments.
 
     """
 
-    def __init__(self, layer=None, **kwargs):
+    def __init__(self, layer=None, group=None, **kwargs):
         super(RhinoSceneObject, self).__init__(**kwargs)
         self.layer = layer
+        self.group = group
 
     def get_group(self, name):
         """Find the group with the given name, or create a new one.