PyCAD

from Object import Object
import geom
import wx
import cad
import math
from Object import PyProperty
from Object import PyPropertyLength
from Object import PyPropertyDoubleScaled

type = 0

class Gear(Object):
    def __init__(self, mod = 1.0, num_teeth = 12, num_teeth2 = 52):
        Object.__init__(self, 0)
        self.tm = geom.Matrix()
        self.solid = None # created to represent gear1
        self.solid2 = None # created to represent gear1
        self.solids1 = [] # id numbers of solids to show transformed with gear1
        self.rackSolid = None
        self.rackSolid2 = None
        self.solids2 = [] # id numbers of solids to show transformed with gear2
        self.start_grip_point = None
        self.numTeeth = num_teeth
        self.numTeeth2 = num_teeth2
        self.module = mod
        self.xOffset = 0.0   # profile shift coefficient (x)
        self.addendumOffset = 0.0
        self.addendumMultiplier = 1.0
        self.dedendumMultiplier = 1.0
        self.pressureAngle = 0.34906585039886 # 20 degrees
        self.tipRelief = 0.05
        self.rootClearance = 0.1
        self.rootRoundness = 0.25
        self.numInvoluteFacets = 10
        self.thickness = 5.0
        self.gripper_point = None
        self.showRack = False
        self.showRack2 = False
        self.color = cad.Color(128, 128, 128)
        self.xOffset2 = 0.0   # profile shift coefficient (x)
        self.addendumOffset2 = 0.0
        self.addendumMultiplier2 = 1.0
        self.dedendumMultiplier2 = 1.0
        self.tipRelief2 = 0.05
        self.rootClearance2 = 0.1
        self.rootRoundness2 = 0.25
        self.thickness2 = 5.0
        self.zOffset2 = 0.0 # move gear2 up by this amount
        self.centreDist = 32.0
        self.color2 = cad.Color(128, 128, 128)
        
    def GetIconFilePath(self):
        return wx.GetApp().cad_dir + '/icons/gear.png'

    def GetTitle(self):
        teeth_str = str(self.numTeeth) + (' tooth' if self.numTeeth == 1 else ' teeth')
        return "Gear - M" + str(self.module) + ' ' + teeth_str

    def GetType(self):
        return type

    def GetTypeString(self):
        return "Gear"

    def GetColor(self):
        return self.color
        
    def SetColor(self, col):
        self.color = col
        
    def KillGLLists(self):
        self.solid = None
        self.rackSolid = None
        self.solid2 = None
        self.rackSolid2 = None
        
    def OnRecalculate(self):
        self.KillGLLists()
        wx.GetApp().Repaint()
        
    def OnRecalculate1(self):
        self.solid = None
        self.rackSolid = None
        wx.GetApp().Repaint()
        
    def OnRecalculate2(self):
        self.solid2 = None
        self.rackSolid2 = None
        wx.GetApp().Repaint()
        
    def OnGlCommands(self, select, marked, no_color):
        cad.DrawPushMatrix()
        cad.DrawMultMatrix(self.tm)
        
        if self.showRack2:
            if self.rackSolid2 == None:
                self.rackSolid2 = self.MakeRackSolid(True)
            rackm = geom.Matrix()
            rackm.Translate(geom.Point3D(self.centreDist, self.gripper_point.y + self.module * math.pi * 0.5, 0))
            cad.DrawPushMatrix()
            cad.DrawMultMatrix(rackm)  
            rot = geom.Matrix()
            radius = self.module * self.numTeeth2 * 0.5
            # for gear2 rotate it 180 degrees
            rot.Rotate(math.pi)
            cad.DrawMultMatrix(rot)
            self.rackSolid2.OnGlCommands(False, marked, no_color)
            cad.DrawPopMatrix()
        else:
            if self.solid == None:
                self.solid = self.MakeSolid()
            rot = geom.Matrix()
            radius = self.module * self.numTeeth * 0.5
            rot.Rotate(self.gripper_point.y / radius) # y coordinate is rotation in radians
            cad.DrawPushMatrix()
            cad.DrawMultMatrix(rot)  
            self.solid.OnGlCommands(False, marked, no_color)
            cad.DrawPopMatrix()
        
        if self.showRack:
            if self.rackSolid == None:
                self.rackSolid = self.MakeRackSolid()
            rackm = geom.Matrix()
            rackm.Translate(geom.Point3D(0, self.gripper_point.y, 0))
            cad.DrawPushMatrix()
            cad.DrawMultMatrix(rackm)  
            self.rackSolid.OnGlCommands(False, marked, no_color)
            cad.DrawPopMatrix()
        else:
            if self.solid2 == None:
                self.solid2 = self.MakeSolid(True)
            cad.DrawPushMatrix()
            trans = geom.Matrix()
            trans.Translate(geom.Point3D(self.centreDist, 0, self.zOffset2))
            cad.DrawMultMatrix(trans)  
            rot = geom.Matrix()
            radius = self.module * self.numTeeth2 * 0.5
            # for gear2 rotate it 180 degrees + half a tooth
            rot.Rotate(-self.gripper_point.y / radius + math.pi / self.numTeeth2 + math.pi) # y coordinate is rotation in radians
            cad.DrawMultMatrix(rot)  
            self.solid2.OnGlCommands(False, marked, no_color)
            cad.DrawPopMatrix()

        cad.DrawPopMatrix()
        
    def Transform(self, mat):
        self.tm.Multiply(mat)
            
    def WriteXml(self):
        cad.SetXmlMatrix('tm', self.tm)
        cad.SetXmlValue('numTeeth', self.numTeeth)
        cad.SetXmlValue('numTeeth2', self.numTeeth2)
        cad.SetXmlValue('module', self.module)
        cad.SetXmlValue('xOffset', self.xOffset)
        cad.SetXmlValue('addendumOffset', self.addendumOffset)
        cad.SetXmlValue('addendumMultiplier', self.addendumMultiplier)
        cad.SetXmlValue('dedendumMultiplier', self.dedendumMultiplier)
        cad.SetXmlValue('pressureAngle', self.pressureAngle)
        cad.SetXmlValue('tipRelief', self.tipRelief)
        cad.SetXmlValue('rootClearance', self.rootClearance)
        cad.SetXmlValue('rootRoundness', self.rootRoundness)        
        cad.SetXmlValue('numInvoluteFacets', self.numInvoluteFacets)
        cad.SetXmlValue('thickness', self.thickness)
        cad.SetXmlValue('showRack', self.showRack)        
        cad.SetXmlValue('showRack2', self.showRack2)   
        cad.SetXmlValue('xOffset2', self.xOffset2)
        cad.SetXmlValue('addendumOffset2', self.addendumOffset2)
        cad.SetXmlValue('addendumMultiplier2', self.addendumMultiplier2)
        cad.SetXmlValue('dedendumMultiplier2', self.dedendumMultiplier2)
        cad.SetXmlValue('tipRelief2', self.tipRelief2)
        cad.SetXmlValue('rootClearance2', self.rootClearance2)
        cad.SetXmlValue('rootRoundness2', self.rootRoundness2)
        cad.SetXmlValue('thickness2', self.thickness2)
        cad.SetXmlValue('zOffset2', self.zOffset2)
        cad.SetXmlValue('centreDist', self.centreDist)
             
        Object.WriteXml(self)

    def ReadXml(self):
        self.tm = cad.GetXmlMatrix('tm')
        self.numTeeth = cad.GetXmlInt('numTeeth', self.numTeeth)
        self.module = cad.GetXmlFloat('module', self.module)
        self.xOffset = cad.GetXmlFloat('xOffset', self.xOffset)
        self.addendumOffset = cad.GetXmlFloat('addendumOffset', self.addendumOffset)
        self.addendumMultiplier = cad.GetXmlFloat('addendumMultiplier', self.addendumMultiplier)
        self.dedendumMultiplier = cad.GetXmlFloat('dedendumMultiplier', self.dedendumMultiplier)
        self.pressureAngle = cad.GetXmlFloat('pressureAngle', self.pressureAngle)
        self.tipRelief = cad.GetXmlFloat('tipRelief', self.tipRelief)
        self.rootClearance = cad.GetXmlFloat('rootClearance', self.rootClearance)
        self.rootRoundness = cad.GetXmlFloat('rootRoundness', self.rootRoundness)
        self.numInvoluteFacets = cad.GetXmlInt('numInvoluteFacets', self.numInvoluteFacets)
        self.thickness = cad.GetXmlFloat('thickness', self.thickness)
        self.showRack = cad.GetXmlBool('showRack', self.showRack)
        self.showRack2 = cad.GetXmlBool('showRack2', self.showRack2)
        self.xOffset2 = cad.GetXmlFloat('xOffset2', self.xOffset2)
        self.addendumOffset2 = cad.GetXmlFloat('addendumOffset2', self.addendumOffset2)
        self.addendumMultiplier2 = cad.GetXmlFloat('addendumMultiplier2', self.addendumMultiplier2)
        self.dedendumMultiplier2 = cad.GetXmlFloat('dedendumMultiplier2', self.dedendumMultiplier2)
        self.tipRelief2 = cad.GetXmlFloat('tipRelief2', self.tipRelief2)
        self.rootClearance2 = cad.GetXmlFloat('rootClearance2', self.rootClearance2)
        self.rootRoundness2 = cad.GetXmlFloat('rootRoundness2', self.rootRoundness2)
        self.thickness2 = cad.GetXmlFloat('thickness2', self.thickness2)
        self.zOffset2 = cad.GetXmlFloat('zOffset2', self.zOffset2)
        self.centreDist = cad.GetXmlFloat('centreDist', self.centreDist)

        Object.ReadXml(self)
                
    def MakeACopy(self):
        copy = Gear()
        copy.CopyFrom(self)
        return copy
    
    def CopyFrom(self, o):
        self.tm = o.tm
        Object.CopyFrom(self, o)
        self.solids1 = []
        self.solids1 += object.solids1
        self.solids2 = []
        self.solids2 += object.solids2
        self.numTeeth = object.num_teeth
        self.numTeeth2 = object.numTeeth2
        self.module = object.module
        self.xOffset = object.xOffset
        self.addendumOffset = object.addendumOffset
        self.addendumMultiplier = object.addendumMultiplier
        self.dedendumMultiplier = object.dedendumMultiplier
        self.pressureAngle = object.pressureAngle
        self.tipRelief = object.tipRelief
        self.rootClearance = object.rootClearance
        self.rootRoundness = object.rootRoundness
        self.numInvoluteFacets = object.numInvoluteFacets
        self.thickness = object.thickness
        self.showRack = object.showRack
        self.color = object.color
        self.xOffset2 = object.xOffset2
        self.addendumOffset2 = object.addendumOffset2
        self.addendumMultiplier2 = object.addendumMultiplier2
        self.dedendumMultiplier2 = object.dedendumMultiplier2
        self.tipRelief2 = object.tipRelief2
        self.rootClearance2 = object.rootClearance2
        self.rootRoundness2 = object.rootRoundness2
        self.zOffset2 = object.zOffset2
        self.centreDist = object.centreDist
        self.color2 = object.color2
    
    def GetBox(self):
        box = geom.Box3D()
        pitch_radius = float(self.module) * self.numTeeth * 0.5
        outside_radius = pitch_radius + (self.addendumMultiplier*self.module + self.addendumOffset)
        steps = 20
        angle_step = 2.0 * math.pi / steps
        angle = 0.0
        for i in range(0, 20):
            p = geom.Point3D(outside_radius * math.cos(angle), outside_radius * math.sin(angle), 0.0)
            pz = geom.Point3D(p.x, p.y, self.thickness)
            p.Transform(self.tm)
            pz.Transform(self.tm)
            box.InsertPoint(p.x, p.y, p.z)
            box.InsertPoint(pz.x, pz.y, pz.z)
            angle += angle_step
        return box
     
    def GetProperties(self):
        properties = []
        properties.append(PyProperty("num teeth", 'numTeeth', self, self.OnRecalculate1))
        properties.append(PyProperty("module", 'module', self, self.OnRecalculate))
        properties.append(PyPropertyLength("x offset", 'xOffset', self, recalculate=self.OnRecalculate1))
        properties.append(PyPropertyLength("addendum offset", 'addendumOffset', self, recalculate = self.OnRecalculate1))
        properties.append(PyPropertyLength("addendum multiplier", 'addendumMultiplier', self, recalculate = self.OnRecalculate1))
        properties.append(PyPropertyLength("dedendum multiplier", 'dedendumMultiplier', self, recalculate = self.OnRecalculate1))
        properties.append(PyPropertyDoubleScaled("pressure angle", 'pressureAngle', self, 180.0 / math.pi, self.OnRecalculate))
        properties.append(PyProperty("tip relief", 'tipRelief', self, self.OnRecalculate1))
        properties.append(PyProperty("root clearance", 'rootClearance', self, self.OnRecalculate1))
        properties.append(PyProperty("root roundness", 'rootRoundness', self, self.OnRecalculate1))
        properties.append(PyProperty("num involute facets", 'numInvoluteFacets', self, self.OnRecalculate))
        properties.append(PyPropertyLength("thickness", 'thickness', self, recalculate = self.OnRecalculate1))
        properties.append(PyProperty("show rack", 'showRack', self, wx.GetApp().Repaint))
        properties.append(PyProperty("show rack 2", 'showRack2', self, wx.GetApp().Repaint))
        properties.append(PyPropertyLength("x offset 2", 'xOffset2', self, recalculate=self.OnRecalculate2))
        properties.append(PyPropertyLength("addendum offset 2", 'addendumOffset2', self, recalculate = self.OnRecalculate2))
        properties.append(PyPropertyLength("addendum multiplier 2", 'addendumMultiplier2', self, recalculate = self.OnRecalculate2))
        properties.append(PyPropertyLength("dedendum multiplier 2", 'dedendumMultiplier2', self, recalculate = self.OnRecalculate2))
        properties.append(PyProperty("tip relief 2", 'tipRelief2', self, self.OnRecalculate2))
        properties.append(PyProperty("root clearance 2", 'rootClearance2', self, self.OnRecalculate2))
        properties.append(PyProperty("root roundness 2", 'rootRoundness2', self, self.OnRecalculate2))
        properties.append(PyPropertyLength("thickness2", 'thickness2', self, recalculate = self.OnRecalculate2))
        properties.append(PyPropertyLength("z offset 2", 'zOffset2', self, recalculate = wx.GetApp().Repaint))
        properties.append(PyPropertyLength("centre distance", 'centreDist', self, recalculate = wx.GetApp().Repaint))

                                             
        properties += Object.GetProperties(self)

        return properties
    
    def AddSketch(self):
        cad.AddUndoably(self.MakeSketch())
        
    def MakeSketch(self, gear2 = False):
        import step
        sketch = cad.NewSketch()
        
        numTeeth = self.numTeeth2 if gear2 else self.numTeeth
        addendumMultiplier = self.addendumMultiplier2 if gear2 else self.addendumMultiplier
        dedendumMultiplier = self.dedendumMultiplier2 if gear2 else self.dedendumMultiplier
        addendumOffset = self.addendumOffset2 if gear2 else self.addendumOffset
        xOffset = self.xOffset2 if gear2 else self.xOffset
        tipRelief = self.tipRelief2 if gear2 else self.tipRelief
        rootClearance = self.rootClearance2 if gear2 else self.rootClearance
             
        pitch_radius = float(self.module) * numTeeth * 0.5
        base_radius = pitch_radius * math.cos(self.pressureAngle)
        outside_radius = pitch_radius + addendumMultiplier * self.module + addendumOffset + xOffset * self.module * self.module * math.tan(self.pressureAngle)
        inside_radius = pitch_radius - dedendumMultiplier * self.module

        if inside_radius < base_radius:
           inside_radius = base_radius
        
        inside_phi_and_angle = involute_intersect(inside_radius, base_radius)
        outside_phi_and_angle = involute_intersect(outside_radius, base_radius)
        tip_relief_phi_and_angle = involute_intersect(outside_radius - tipRelief, base_radius)
        middle_phi_and_angle = involute_intersect(pitch_radius, base_radius)

        for i in range(0, numTeeth):
            tooth_angle = math.pi*2*i/numTeeth
            next_tooth_angle = math.pi*2*(i+1)/numTeeth
            
            relief_vector = geom.Point(math.cos(tooth_angle), math.sin(tooth_angle))
            root_vector = ~relief_vector
            
            # incremental_angle - to space the middle point at a quarter of a cycle
            incremental_angle = 0.5*math.pi/numTeeth - middle_phi_and_angle[1]
            
            # angle between tooth center and one flank rotated to take into account xOffset
            incremental_angle -= (2.0 * xOffset * self.module * math.tan(self.pressureAngle)) / (2.0 * pitch_radius)

            # get the previous tooth's involute points
            prev_involute_points = []
            involute(prev_involute_points, tooth_angle - incremental_angle, True, inside_phi_and_angle, tip_relief_phi_and_angle, base_radius, self.numInvoluteFacets)

            # get the up hill involute points
            involute_points = []
            involute(involute_points, tooth_angle + incremental_angle, False, inside_phi_and_angle, tip_relief_phi_and_angle, base_radius, self.numInvoluteFacets)

            # get the downhill involute points
            involute_points2 = []
            involute(involute_points2, next_tooth_angle - incremental_angle, True, inside_phi_and_angle, tip_relief_phi_and_angle, base_radius, self.numInvoluteFacets)
            
            # make involute splines
            three_d_pts = []
            for p in prev_involute_points:
                three_d_pts.append( geom.Point3D(p.x, p.y, 0) )
            previous_spline = step.NewSplineFromPoints(three_d_pts)

            three_d_pts = []
            for p in involute_points:
                three_d_pts.append( geom.Point3D(p.x, p.y, 0) )
            uphill_spline = step.NewSplineFromPoints(three_d_pts)
           
            three_d_pts = []
            for p in involute_points2:
                three_d_pts.append( geom.Point3D(p.x, p.y, 0) )
            downhill_spline = step.NewSplineFromPoints(three_d_pts)
            
            # root profile
            points = []
            # start with end of involute
            points.append(prev_involute_points[-1])
            # add a mid point
            points.append(geom.Point(math.cos(tooth_angle) * inside_radius, math.sin(tooth_angle) * inside_radius) + relief_vector * (-rootClearance))
            # end with the start of the next involute
            points.append(involute_points[0])
            
            # add a spline
            three_d_pts = []
            for p in points:
                three_d_pts.append( geom.Point3D(p.x, p.y, 0) )
            sketch.Add( step.NewSplineFromPoints(three_d_pts, previous_spline.GetEndTangent(), uphill_spline.GetStartTangent()) )
            
            # up hill involute            
            sketch.Add( uphill_spline )
            
            # tip relief
            points = []
            points.append(involute_points[-1])
            if math.fabs(tipRelief) > 0.00000000001:
                angle3 = tooth_angle + (outside_phi_and_angle[1] + incremental_angle)
                angle4 = next_tooth_angle - (outside_phi_and_angle[1] + incremental_angle)
                points.append(point_at_rad_and_angle(outside_radius, angle3 + (tipRelief * 0.5)/outside_radius))
                points.append(point_at_rad_and_angle(outside_radius, angle4 - (tipRelief * 0.5)/outside_radius))
            points.append(involute_points2[0])

            # add lines
            prev_point = None
            for p in points:
                p3d = geom.Point3D(p.x, p.y, 0)
                if prev_point != None:
                    sketch.Add(cad.NewLine(prev_point, p3d))
                prev_point = p3d

            # downhill involute
            sketch.Add( downhill_spline )

        return sketch
    
    def MakeRackSketch(self, gear2 = False):
        sketch = cad.NewSketch()
        
        num_rack_teeth = 6 # on each side, so twice this
        
        numTeeth = self.numTeeth2 if gear2 else self.numTeeth
        xOffset = self.xOffset2 if gear2 else self.xOffset

        pitch_radius = float(self.module) * numTeeth * 0.5
        
        # spacing between teeth
        pitch = math.pi * float(self.module)
        
        left_x = pitch_radius - self.module + xOffset * self.module
        right_x = left_x + self.module * 2
        rack_side_x = right_x + self.module
        pressureAngle_y = math.tan(self.pressureAngle) * 2.0 * self.module
        tip_or_root_y = (pitch - (2 * pressureAngle_y)) * 0.5
        
        points = []
        
        for i in range(num_rack_teeth, -num_rack_teeth - 1, -1):
            points.append(geom.Point(right_x, tip_or_root_y * 0.5 + pressureAngle_y + pitch * i))
            points.append(geom.Point(left_x, tip_or_root_y * 0.5 + pitch * i))
            points.append(geom.Point(left_x, -tip_or_root_y * 0.5 + pitch * i))
            points.append(geom.Point(right_x, -tip_or_root_y * 0.5 - pressureAngle_y + pitch * i))
            points.append(geom.Point(right_x, -tip_or_root_y * 1.5 - pressureAngle_y + pitch * i))
            
        # finish off the shape
        points.append(geom.Point(rack_side_x, points[-1].y))
        points.append(geom.Point(rack_side_x, points[0].y))
        points.append(points[0])

        # add lines
        prev_point = None
        for p in points:
            p3d = geom.Point3D(p.x, p.y, 0)
            if prev_point != None:
                sketch.Add(cad.NewLine(prev_point, p3d))
            prev_point = p3d
        return sketch

    def AddSolid(self):
        cad.AddUndoably(self.MakeSolid())
        
    def MakeSolid(self, gear2 = False):
        import step
        objects = [self.MakeSketch(gear2)]   
        thickness = self.thickness2 if gear2 else self.thickness
        new_solids = step.CreateExtrusion(objects, thickness, True, False, 0.0, cad.Color(128, 128, 128))
        s = new_solids[0]
        if s != None:
            # subtract cylinder in the middle
            cyl = step.NewCyl()
            cyl.radius = 1
            cyl.height = 20
            cyl.OnApplyProperties()
            objects = [s, cyl]
            s = step.CutShapes(objects)
            
            # calculate the gripper point
            if not gear2:
                box = s.GetBox()
                self.gripper_point = geom.Point3D(self.numTeeth * self.module * 0.5, 0, self.thickness)
        return s
    
    def MakeRackSolid(self, gear2 = False):
        import step
        objects = [self.MakeRackSketch(gear2)]
        thickness = self.thickness2 if gear2 else self.thickness 
        new_solids = step.CreateExtrusion(objects, thickness, True, False, 0.0, cad.Color(128, 128, 128))
        s = new_solids[0]
        return s
    
    def GetGrippers(self, just_for_endof):
        if self.gripper_point != None:
            cad.AddGripper(cad.GripData(self.gripper_point, cad.GripperType.Stretch, 0))
        
    def Stretch(self):
        self.start_grip_point = None
        
    def StretchTemporary(self):
        if self.start_grip_point == None:
            self.start_grip_point = geom.Point3D(self.gripper_point)
        shift = cad.GetStretchShift()
        self.gripper_point.y = self.start_grip_point.y + shift.y


def point_at_phi(phi, base_radius):
    x = base_radius * phi
    sx = math.cos(phi) * base_radius
    sy = math.sin(phi) * base_radius
    return geom.Point(sx + math.sin(phi) * x, sy - math.cos(phi) * x)

def point_at_rad_and_angle(r, angle):
    return geom.Point(r * math.cos(angle), r * math.sin(angle))

def involute_intersect1(r, base_radius, phi, phi_step):
    while True:
        p = point_at_phi(phi, base_radius)
        if p.Length() > r:
            if phi_step > 0.0000000001:
                return involute_intersect1(r, base_radius, phi-phi_step, phi_step * 0.5)
            return (phi, math.atan2(p.y, p.x))
        phi = phi + phi_step
            
def involute_intersect(r, base_radius):
    return involute_intersect1(r, base_radius, 1.0, 1.0)
    
    
def involute(points, tooth_angle, do_reverse, inside_phi_and_angle, tip_relief_phi_and_angle, base_radius, steps):
    first = True
    
    start = steps if do_reverse else 0
    stop = -1 if do_reverse else (steps + 1)
    step = -1 if do_reverse else 1
    for i in range(start, stop, step):
        phi = inside_phi_and_angle[0] + (tip_relief_phi_and_angle[0] - inside_phi_and_angle[0])*i/steps

        # calculate point on the first tooth
        p = point_at_phi(phi, base_radius)
        if do_reverse:
            p.y = -p.y # mirror for reverse

        # rotate by tooth angle
        x = p.x * math.cos(tooth_angle) -p.y*math.sin(tooth_angle)
        y = p.y * math.cos(tooth_angle) + p.x * math.sin(tooth_angle)
        points.append( geom.Point(x, y))