Added SampleSphereStyles.py

rhinopython/SampleSphereStyles.py

@@ -0,0 +1,103 @@
+################################################################################
+# SampleSphereStyles.py
+# Copyright (c) 2019 Robert McNeel & Associates.
+# See License.md in the root of this repository for details.
+################################################################################
+import Rhino
+import scriptcontext as sc
+
+# Rhino.Input.Custom.GetPoint derived class that dynamically draws a sphere.   
+class SampleGetSphere(Rhino.Input.Custom.GetPoint):
+
+    # Class initializer
+    def __init__(self, plane):
+        self.m_plane = plane
+        self.m_sphere = Rhino.Geometry.Sphere(plane, 1.0)
+        self.m_draw = False
+        self.m_color = sc.doc.Layers.CurrentLayer.Color
+
+    # Calculates the sphere
+    def CalculateSphere(self, point):
+        rc = False
+        dir = point - self.m_plane.Origin
+        if (not dir.IsTiny()):
+            self.m_sphere.Radius = dir.Length
+            rc = True
+        return rc
+
+    # OnMouseMove override
+    def OnMouseMove(self, e):
+        self.m_draw = self.CalculateSphere(e.Point)
+        # Call base class...
+        Rhino.Input.Custom.GetPoint.OnMouseMove(self, e)
+
+    # OnDynamicDraw override
+    def OnDynamicDraw(self, e):
+        if self.m_draw:
+            e.Display.DrawSphere(self.m_sphere, self.m_color)
+        # Call base class...
+        Rhino.Input.Custom.GetPoint.OnDynamicDraw(self, e)
+
+    # Returns the calculated sphere
+    def Sphere(self):
+        return self.m_sphere
+
+# Main function.  
+def SampleSphereStyles():
+
+    styles = ["Baseball", "Quad", "UV", "MeshTriangle", "MeshQuad", "MeshUV"]
+    style = 2
+
+    # Select sphere style
+    go = Rhino.Input.Custom.GetOption()
+    go.SetCommandPrompt("Select sphere style")
+    go.AddOptionList("Style", styles, style)
+    go.AcceptNothing(True)
+    res = go.Get()
+    if res == Rhino.Input.GetResult.Option:
+        style = go.Option().CurrentListOptionIndex
+    elif res != Rhino.Input.GetResult.Nothing:
+        return Rhino.Commands.Result.Cancel
+
+    # Center of sphere
+    gp = Rhino.Input.Custom.GetPoint()
+    gp.SetCommandPrompt("Center of sphere")
+    gp.Get()
+    if gp.CommandResult() <> Rhino.Commands.Result.Success:
+        return gp.CommandResult()
+
+    # Calculate base plane
+    base_plane = gp.View().ActiveViewport.ConstructionPlane()
+    base_plane.Origin = gp.Point()
+
+    # Radius
+    gs = SampleGetSphere(base_plane)
+    gs.SetCommandPrompt("Radius")
+    gs.SetBasePoint(base_plane.Origin, True)
+    gs.DrawLineFromPoint(base_plane.Origin, True)
+    gs.Get()
+    if gs.CommandResult() <> Rhino.Commands.Result.Success:
+        return gs.CommandResult()
+
+    # One final calculation
+    if gs.CalculateSphere(gs.Point()):
+        sphere = gs.Sphere()
+        if style == 0:
+            sc.doc.Objects.AddBrep(Rhino.Geometry.Brep.CreateBaseballSphere(sphere.Center, sphere.Radius, sc.doc.ModelAbsoluteTolerance))
+        elif style == 1:
+            sc.doc.Objects.AddBrep(Rhino.Geometry.Brep.CreateQuadSphere(sphere))
+        elif style == 2:
+            sc.doc.Objects.AddSphere(gs.Sphere())
+        elif style == 3:
+            sc.doc.Objects.AddMesh(Rhino.Geometry.Mesh.CreateIcoSphere(sphere, 1))
+        elif style == 4:
+            sc.doc.Objects.AddMesh(Rhino.Geometry.Mesh.CreateQuadSphere(sphere, 2))
+        elif style == 5:
+            sc.doc.Objects.AddMesh(Rhino.Geometry.Mesh.CreateFromSphere(sphere, 10, 10))
+        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__":
+    SampleSphereStyles()