From 39e2c5348fca784e9725ae99dcb406045d26cb37 Mon Sep 17 00:00:00 2001
From: Dale Fugier <dale@mcneel.com>
Date: Thu, 7 Feb 2019 07:55:35 -0800
Subject: [PATCH] Added SampleDrawExtrudedCurves.py

---
 rhinopython/SampleDrawExtrudedCurves.py | 84 +++++++++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 rhinopython/SampleDrawExtrudedCurves.py

diff --git a/rhinopython/SampleDrawExtrudedCurves.py b/rhinopython/SampleDrawExtrudedCurves.py
new file mode 100644
index 00000000..85d3c8ba
--- /dev/null
+++ b/rhinopython/SampleDrawExtrudedCurves.py
@@ -0,0 +1,84 @@
+################################################################################
+# SampleDrawExtrudedCurves.py
+# Copyright (c) 2019 Robert McNeel & Associates.
+# See License.md in the root of this repository for details.
+################################################################################
+import scriptcontext as sc
+import Rhino
+import System
+
+# GetObject-derived class that limits object picking to planar curves
+class GetPlanarCurve(Rhino.Input.Custom.GetObject):
+    def CustomGeometryFilter(self, rh_object, geometry, component_index):
+        if not rh_object: 
+            return False
+        if not geometry: 
+            return False
+        if not isinstance(geometry, Rhino.Geometry.Curve): 
+            return False
+        tol = rh_object.Document.ModelAbsoluteTolerance
+        rc, plane = geometry.TryGetPlane(tol)
+        return rc
+
+# DisplayConduit-derived class that draws Breps
+class DrawBrepConduit(Rhino.Display.DisplayConduit):
+    def __init__(self, breps):
+        self.breps = breps
+        
+    def CalculateBoundingBox(self, e):
+        for brep in self.breps:
+            e.IncludeBoundingBox(brep.GetBoundingBox(False))
+            
+    def PreDrawObjects(self, e):
+        att = e.Display.DisplayPipelineAttributes
+        if att.ShadingEnabled:
+            material = Rhino.Display.DisplayMaterial(System.Drawing.Color.DarkGray)
+            for brep in self.breps:
+                e.Display.DrawBrepShaded(brep, material)
+        for brep in self.breps:
+            e.Display.DrawBrepWires(brep, System.Drawing.Color.Black)
+
+# Sample function that select planar curve, extrudes them in some fixed
+# direction, and then draws them in a custom display conduit.
+def SampleDrawExtrudedCurves():
+    go = GetPlanarCurve()
+    go.SetCommandPrompt("Select planar curves")
+    go.SubObjectSelect = False
+    go.GetMultiple(1, 0)
+    if go.CommandResult() != Rhino.Commands.Result.Success:
+        return
+    
+    tol = sc.doc.ModelAbsoluteTolerance
+    breps = []
+    
+    for objref in go.Objects():
+        crv = objref.Curve();
+        rc, plane = crv.TryGetPlane(tol)
+        dir = plane.ZAxis * 3
+        srf = Rhino.Geometry.Surface.CreateExtrusion(crv, dir)
+        brep = srf.ToBrep()
+        brep.Faces.SplitKinkyFaces(Rhino.RhinoMath.DefaultAngleTolerance, True)
+        if crv.IsClosed:
+            solid = brep.CapPlanarHoles(tol)
+            if Rhino.Geometry.BrepSolidOrientation.Inward == solid.SolidOrientation:
+                solid.Flip();
+            breps.append(solid)
+        else:
+            breps.append(brep)
+            
+        objref.Object().Select(False)
+    
+    conduit = DrawBrepConduit(breps)
+    conduit.Enabled = True
+    sc.doc.Views.Redraw()
+
+    rs.GetString('Press <Enter> to continue')
+
+    conduit.Enabled = False
+    sc.doc.Views.Redraw()
+
+# Check to see if this file is being executed as the "main" python
+# script instead of being used as a module by some other python script
+# This allows us to use the module which ever way we want.
+if __name__=="__main__":
+    SampleDrawExtrudedCurves()