Merge pull request #1074 from compas-dev/brep_artist

Brep artist

Author: chenkasirer

CHANGELOG.md

@@ -20,6 +20,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 * Added `format` task using `black` formatter.
 * Added a `test_intersection_circle_circle_xy` in the `test_intersections`
 * Added split operation to `compas_rhino.geometry.Brep`.
+* Added a `RhinoArtist` in `compas_rhino`.
+* Added a `RhinoArtist` in `compas_ghpython`.
 
 ### Changed
 

src/compas_ghpython/artists/__init__.py

@@ -35,6 +35,7 @@
     SurfaceArtist
     TorusArtist
     VectorArtist
+    BrepArtist
 
 
 Datastructure Artists
@@ -91,6 +92,7 @@
 from compas.geometry import Surface
 from compas.geometry import Torus
 from compas.geometry import Vector
+from compas.geometry import Brep
 
 from compas.datastructures import Mesh
 from compas.datastructures import Network
@@ -119,6 +121,7 @@
 from .torusartist import TorusArtist
 from .vectorartist import VectorArtist
 from .volmeshartist import VolMeshArtist
+from .brepartist import BrepArtist
 
 
 ShapeArtist.default_color = (255, 255, 255)
@@ -160,6 +163,7 @@ def register_artists():
     Artist.register(Torus, TorusArtist, context="Grasshopper")
     Artist.register(Vector, VectorArtist, context="Grasshopper")
     Artist.register(VolMesh, VolMeshArtist, context="Grasshopper")
+    Artist.register(Brep, BrepArtist, context="Grasshopper")
     print("GH Artists registered.")
 
 

src/compas_ghpython/artists/brepartist.py

@@ -0,0 +1,28 @@
+from compas_ghpython.artists import GHArtist
+from compas_ghpython.utilities import draw_brep
+
+
+class BrepArtist(GHArtist):
+    """An artist for drawing a brep in Grasshopper.
+
+    Parameters
+    ==========
+    brep : :class:`~compas_rhino.geometry.RhinoBrep`
+        The brep to draw.
+
+    """
+
+    def __init__(self, brep):
+        super(BrepArtist, self).__init__()
+        self._brep = brep
+
+    def draw(self):
+        """Draw the brep as a Grasshopper geometry.
+
+        Returns
+        -------
+        :rhino:`Rhino.Geometry.Brep`
+            The Grasshopper geometry instance.
+
+        """
+        return draw_brep(self._brep)

src/compas_ghpython/utilities/__init__.py

@@ -23,6 +23,7 @@
     draw_spheres
     draw_mesh
     draw_network
+    draw_brep
 
 
 sets
@@ -69,6 +70,7 @@
     draw_mesh,
     draw_network,
     draw_circles,
+    draw_brep,
 )
 from .sets import list_to_ghtree, ghtree_to_list
 from .timer import update_component
@@ -87,6 +89,7 @@
     "draw_mesh",
     "draw_network",
     "draw_circles",
+    "draw_brep",
     "list_to_ghtree",
     "ghtree_to_list",
     "update_component",

src/compas_ghpython/utilities/drawing.py

@@ -463,3 +463,19 @@ def draw_circles(circles):
         radius = c["radius"]
         rg_circles.append(Circle(Plane(Point3d(*point), Vector3d(*normal)), radius))
     return rg_circles
+
+
+def draw_brep(brep):
+    """Draw a RhinoBrep in Grasshopper.
+
+    Parameters
+    ----------
+    brep : :class:`~compas.geometry.RhinoBrep`
+        The Brep to draw.
+
+    Returns
+    -------
+    :rhino:`Rhino.Geometry.Brep`
+
+    """
+    return brep.native_brep

src/compas_rhino/artists/__init__.py

@@ -21,6 +21,7 @@
     PolygonArtist
     PolylineArtist
     VectorArtist
+    BrepArtist
 
 
 Shape Artists
@@ -106,6 +107,7 @@
 
 from compas.geometry import Curve
 from compas.geometry import Surface
+from compas.geometry import Brep
 
 from compas.datastructures import Mesh
 from compas.datastructures import Network
@@ -137,6 +139,7 @@
 
 from .curveartist import CurveArtist
 from .surfaceartist import SurfaceArtist
+from .brepartist import BrepArtist
 
 BaseArtist = RhinoArtist
 
@@ -174,6 +177,7 @@ def register_artists():
     Artist.register(RobotModel, RobotModelArtist, context="Rhino")
     Artist.register(Curve, CurveArtist, context="Rhino")
     Artist.register(Surface, SurfaceArtist, context="Rhino")
+    Artist.register(Brep, BrepArtist, context="Rhino")
     print("Rhino Artists registered.")
 
 
@@ -201,4 +205,5 @@ def register_artists():
     "RobotModelArtist",
     "CurveArtist",
     "SurfaceArtist",
+    "BrepArtist",
 ]

src/compas_rhino/artists/brepartist.py

@@ -0,0 +1,28 @@
+import compas_rhino
+from .artist import RhinoArtist
+
+
+class BrepArtist(RhinoArtist):
+    """An artist for drawing a RhinoBrep.
+
+    Parameters
+    ==========
+    brep : :class:`~compas_rhino.geometry.RhinoBrep`
+        The Brep to draw.
+
+    """
+
+    def __init__(self, brep):
+        super(BrepArtist, self).__init__()
+        self._brep = brep
+
+    def draw(self, color=None):
+        """Bakes the Brep into the current document
+
+        Returns
+        -------
+        list(:rhino:`System.Guid`)
+            The guid of the baked Brep.
+
+        """
+        return [compas_rhino.draw_brep(self._brep, color)]

src/compas_rhino/utilities/__init__.py

@@ -198,6 +198,7 @@
     draw_circles,
     draw_curves,
     draw_surfaces,
+    draw_brep,
 )
 from .geometry import uv_points_from_surface
 from .constructors import volmesh_from_polysurfaces
@@ -297,6 +298,7 @@
     "draw_circles",
     "draw_curves",
     "draw_surfaces",
+    "draw_brep",
     "uv_points_from_surface",
     "volmesh_from_polysurfaces",
 ]

src/compas_rhino/utilities/drawing.py

@@ -82,6 +82,7 @@
     "draw_mesh",
     "draw_circles",
     "draw_surfaces",
+    "draw_brep",
 ]
 
 
@@ -1096,3 +1097,27 @@ def draw_surfaces(surfaces, **kwargs):
         obj.CommitChanges()
         guids.append(guid)
     return guids
+
+
+@wrap_drawfunc
+def draw_brep(brep, color=None, **kwargs):
+    """Draw a brep to the Rhino document.
+
+    Parameters
+    ----------
+    brep : :class:`~compas_rhino.geometry.RhinoBrep`
+        The brep to draw.
+    color : tuple[int, int, int] | tuple[float, float, float], optional
+        The color to draw the brep with.
+
+    Returns
+    -------
+    :rhino:`System.Guid`
+        The Rhino document GUID of the drawn Brep.
+
+    """
+    native_brep = brep.native_brep
+    if color:
+        for face in native_brep.Faces:
+            face.PerFaceColor = FromArgb(*color)
+    return add_brep(native_brep)