SampleSphereStyles.py

################################################################################
# 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()