put back

Author: tomvanmele

src/compas_rhino/artists/__init__.py

@@ -86,7 +86,7 @@
 
 from compas.plugins import plugin
 from compas.artists import Artist
-
+from compas.artists import ShapeArtist
 from compas.geometry import Circle
 from compas.geometry import Frame
 from compas.geometry import Line
@@ -141,64 +141,81 @@
 BaseArtist = RhinoArtist
 
 
-@plugin(category='drawing-utils', pluggable_name='clear', requires=['Rhino'])
+ShapeArtist.default_color = (255, 255, 255)
+
+# MeshArtist.default_color = (0, 0, 0)
+# MeshArtist.default_vertexcolor = (255, 255, 255)
+# MeshArtist.default_edgecolor = (0, 0, 0)
+# MeshArtist.default_facecolor = (255, 255, 255)
+
+NetworkArtist.default_nodecolor = (255, 255, 255)
+NetworkArtist.default_edgecolor = (0, 0, 0)
+
+VolMeshArtist.default_color = (0, 0, 0)
+VolMeshArtist.default_vertexcolor = (255, 255, 255)
+VolMeshArtist.default_edgecolor = (0, 0, 0)
+VolMeshArtist.default_facecolor = (255, 255, 255)
+VolMeshArtist.default_cellcolor = (255, 0, 0)
+
+
+@plugin(category="drawing-utils", pluggable_name="clear", requires=["Rhino"])
 def clear_rhino():
     compas_rhino.clear()
 
 
-@plugin(category='drawing-utils', pluggable_name='redraw', requires=['Rhino'])
+@plugin(category="drawing-utils", pluggable_name="redraw", requires=["Rhino"])
 def redraw_rhino():
     compas_rhino.redraw()
 
 
-@plugin(category='factories', requires=['Rhino'])
+@plugin(category="factories", requires=["Rhino"])
 def register_artists():
-    Artist.register(Circle, CircleArtist, context='Rhino')
-    Artist.register(Frame, FrameArtist, context='Rhino')
-    Artist.register(Line, LineArtist, context='Rhino')
-    Artist.register(Plane, PlaneArtist, context='Rhino')
-    Artist.register(Point, PointArtist, context='Rhino')
-    Artist.register(Polygon, PolygonArtist, context='Rhino')
-    Artist.register(Polyline, PolylineArtist, context='Rhino')
-    Artist.register(Vector, VectorArtist, context='Rhino')
-    Artist.register(Box, BoxArtist, context='Rhino')
-    Artist.register(Capsule, CapsuleArtist, context='Rhino')
-    Artist.register(Cone, ConeArtist, context='Rhino')
-    Artist.register(Cylinder, CylinderArtist, context='Rhino')
-    Artist.register(Polyhedron, PolyhedronArtist, context='Rhino')
-    Artist.register(Sphere, SphereArtist, context='Rhino')
-    Artist.register(Torus, TorusArtist, context='Rhino')
-    Artist.register(Mesh, MeshArtist, context='Rhino')
-    Artist.register(Network, NetworkArtist, context='Rhino')
-    Artist.register(VolMesh, VolMeshArtist, context='Rhino')
-    Artist.register(RobotModel, RobotModelArtist, context='Rhino')
-    Artist.register(Curve, CurveArtist, context='Rhino')
-    Artist.register(Surface, SurfaceArtist, context='Rhino')
-    print('Rhino Artists registered.')
+    Artist.register(Circle, CircleArtist, context="Rhino")
+    Artist.register(Frame, FrameArtist, context="Rhino")
+    Artist.register(Line, LineArtist, context="Rhino")
+    Artist.register(Plane, PlaneArtist, context="Rhino")
+    Artist.register(Point, PointArtist, context="Rhino")
+    Artist.register(Polygon, PolygonArtist, context="Rhino")
+    Artist.register(Polyline, PolylineArtist, context="Rhino")
+    Artist.register(Vector, VectorArtist, context="Rhino")
+    Artist.register(Box, BoxArtist, context="Rhino")
+    Artist.register(Capsule, CapsuleArtist, context="Rhino")
+    Artist.register(Cone, ConeArtist, context="Rhino")
+    Artist.register(Cylinder, CylinderArtist, context="Rhino")
+    Artist.register(Polyhedron, PolyhedronArtist, context="Rhino")
+    Artist.register(Sphere, SphereArtist, context="Rhino")
+    Artist.register(Torus, TorusArtist, context="Rhino")
+    Artist.register(Mesh, MeshArtist, context="Rhino")
+    Artist.register(Network, NetworkArtist, context="Rhino")
+    Artist.register(VolMesh, VolMeshArtist, context="Rhino")
+    Artist.register(RobotModel, RobotModelArtist, context="Rhino")
+    Artist.register(Curve, CurveArtist, context="Rhino")
+    Artist.register(Surface, SurfaceArtist, context="Rhino")
+    print("Rhino Artists registered.")
 
 
 __all__ = [
-    'BaseArtist',
-    'RhinoArtist',
-    'CircleArtist',
-    'FrameArtist',
-    'LineArtist',
-    'PlaneArtist',
-    'PointArtist',
-    'PolygonArtist',
-    'PolylineArtist',
-    'VectorArtist',
-    'BoxArtist',
-    'CapsuleArtist',
-    'ConeArtist',
-    'CylinderArtist',
-    'PolyhedronArtist',
-    'SphereArtist',
-    'TorusArtist',
-    'MeshArtist',
-    'NetworkArtist',
-    'VolMeshArtist',
-    'RobotModelArtist',
-    'CurveArtist',
-    'SurfaceArtist',
+    "BaseArtist",
+    "RhinoArtist",
+    "CircleArtist",
+    "FrameArtist",
+    "LineArtist",
+    "PlaneArtist",
+    "PointArtist",
+    "PolygonArtist",
+    "PolylineArtist",
+    "VectorArtist",
+    "BoxArtist",
+    "CapsuleArtist",
+    "ConeArtist",
+    "CylinderArtist",
+    "PolyhedronArtist",
+    "SphereArtist",
+    "TorusArtist",
+    "MeshArtist",
+    "NetworkArtist",
+    "VolMeshArtist",
+    "RobotModelArtist",
+    "CurveArtist",
+    "SurfaceArtist",
 ]