import cadquery as cq
from cq_gears import SpurGear

import math

from shapely.geometry import LineString
from shapely.geometry import Point

# M3 screws for attaching lid
m3_tap=2.5
m3_head_diameter=6
m3_head_depth=3
m3_passthrough=3.5

# Fastener dimensions
fastener_width = 11
fastener_height = 3.5
fastener_length = 20

# NEMA 17 stepper motor
stepper_motor_size=42.0
stepper_hole_width=31.0

# Case
part_thinkness=20
outer_fillets=2.5
slant_degree=15.0
wall_width=1.6
loop_width=wall_width
loop_angle=40
rod_diameter=15.2
offset=9
tab_height=7
tab_width=25

# Gears
mod=0.8
gear_gap=0.2
gear_offset=0.5
large_gear_diameter=8.5
gear_thinkness=10

# Other
mount_gap=2

# Calculated values
lid_thicknesss = (part_thinkness-gear_thinkness)/2
gap = stepper_motor_size * math.sin(math.radians(slant_degree)) / math.sin(math.radians(90-slant_degree))
width = math.sqrt(gap**2 + stepper_motor_size**2 - 2*gap*stepper_motor_size*math.cos(math.radians(90)))
end_x=gap+stepper_motor_size-5
end_y=-4.3
outer_ring_diameter=rod_diameter+2*loop_width
center_x=end_x-math.cos(math.radians(loop_angle))*outer_ring_diameter/2
center_y=end_y-math.sin(math.radians(loop_angle))*outer_ring_diameter/2

def findRodInter(n_x, n_y):
    p = Point(center_x, center_y)
    c = p.buffer(rod_diameter/2).boundary
    l = LineString([(n_x, n_y), (n_x+100*math.cos(math.radians(-slant_degree)), n_y+100*math.sin(math.radians(-slant_degree)))])
    i = c.intersection(l)

    if len(i.geoms) > 1 and i.geoms[0].coords[0][0] > i.geoms[1].coords[0][0]:
        x = i.geoms[1].coords[0][0]
        y = i.geoms[1].coords[0][1]
    else:
        x = i.geoms[0].coords[0][0]
        y = i.geoms[0].coords[0][1]
    return x, y

def createOutline(ex_amount):
    base = cq.Workplane('XY').lineTo(0,offset).polarLine(stepper_motor_size, 90-slant_degree)
    base = base.hLine(width).polarLine(stepper_motor_size-5,270-slant_degree).tangentArcPoint((end_x, end_y), relative=False)

    outer_x=center_x+math.cos(math.radians(270-slant_degree))*outer_ring_diameter/2
    outer_y=center_y+math.sin(math.radians(270-slant_degree))*outer_ring_diameter/2
    base = base.threePointArc((center_x+outer_ring_diameter/2 ,center_y), (outer_x, outer_y))\
               .polarLine(tab_width, 180-slant_degree).vLine(tab_height)

    n_x=base.val().endPoint().toTuple()[0]
    n_y=base.val().endPoint().toTuple()[1]
    coor_x, coor_y = findRodInter(n_x, n_y)
    last_x, last_y = findRodInter(n_x, n_y+mount_gap)

    base = base.lineTo(coor_x, coor_y)
    base = base.threePointArc((center_x+rod_diameter/2, center_y), (last_x, last_y))
    base = base.polarLine(tab_width, 180-slant_degree).close().extrude(ex_amount)

    # Fillets
    base = base.edges('|Z and >Y').fillet(outer_fillets)
    base = base.edges('|Z and <X').fillet(outer_fillets)
    base = base.edges('|Z').edges('<<Y[-2]').fillet(5)
    return base

lid = createOutline(-lid_thicknesss).translate((0,0,part_thinkness))
base = createOutline(part_thinkness)

# Shell
copy = base.faces('>>Z[0]').workplane().center(0, -offset-loop_width)\
                                       .polarLine(stepper_motor_size, slant_degree)\
                                       .polarLine(-2*stepper_motor_size, 90+slant_degree)\
                                       .hLine(-2*stepper_motor_size).close().cutThruAll()
copy = copy.faces('+Z').workplane().center(0, -offset-loop_width)\
                                   .polarLine(2*stepper_motor_size,-slant_degree)\
                                   .vLine(stepper_motor_size)\
                                   .hLine(-2*stepper_motor_size)\
                                   .close().cutBlind(-lid_thicknesss)
base = base.faces('+Z').shell(-wall_width)
base = base.add(copy)
base = base.faces('<<Z[0]').workplane().center(0, offset+loop_width)\
                                       .polarLine(stepper_motor_size, -slant_degree)\
                                       .polarLine(stepper_motor_size, -slant_degree-3.75)\
                                       .vLine(2*stepper_motor_size)\
                                       .hLine(-2*stepper_motor_size)\
                                       .close().cutBlind(-lid_thicknesss)

# Stepper motor mount points
stepper_hole_radius=stepper_hole_width/2*math.sqrt(2)
stepper_center = cq.Workplane('XY').center(0, -offset-loop_width)\
                                   .lineTo(0, offset)\
                                   .polarLine(stepper_motor_size/2, 90-slant_degree)\
                                   .polarLine(stepper_motor_size/2, -slant_degree)\
                                   .val().endPoint().toTuple()
base = base.faces('>>Z[0]').workplane().center(stepper_center[0], -stepper_center[1])\
                                       .circle(22.5/2)\
                                       .pushPoints([(stepper_hole_radius*math.cos(math.radians(45+slant_degree)),\
                                                     stepper_hole_radius*math.sin(math.radians(45+slant_degree))),\
                                                    (stepper_hole_radius*math.cos(math.radians(135+slant_degree)),\
                                                     stepper_hole_radius*math.sin(math.radians(135+slant_degree))),\
                                                    (stepper_hole_radius*math.cos(math.radians(225+slant_degree)),\
                                                     stepper_hole_radius*math.sin(math.radians(225+slant_degree))),\
                                                    (stepper_hole_radius*math.cos(math.radians(315+slant_degree)),\
                                                     stepper_hole_radius*math.sin(math.radians(315+slant_degree)))])\
                                       .circle(3/2).cutBlind(-wall_width)

lid = lid.cut(base)
copy = lid.faces('>>Z[0]').workplane().center(0, -offset-loop_width)\
                                      .polarLine(stepper_motor_size*2, slant_degree)\
                                      .polarLine(-2*stepper_motor_size, 90+slant_degree)\
                                      .hLine(-3*stepper_motor_size).close().cutThruAll()
lid = lid.faces('>Z[0]').shell(.000001-wall_width)
lid = lid.add(copy)

lid = lid.faces('>Z[1]').workplane().center(stepper_center[0], -stepper_center[1]-offset-loop_width)\
                                    .circle(5).extrude(lid_thicknesss-wall_width)
lid = lid.faces('>Z[0]').workplane().circle(5/2).cutThruAll()

base = base.faces('<Z[1]').workplane().center(21,20.15).circle(26.5).cutBlind(-gear_thinkness-gear_gap)
t=11/2
base = base.faces('<Z[2]').workplane().hLine(t, forConstruction=True)\
                                      .threePointArc((-t,-t/2),(0,t))\
                                      .polarLine(t, -slant_degree)\
                                      .close().extrude(-lid_thicknesss+wall_width)
base = base.faces('<Z[2]').workplane().circle(large_gear_diameter/2-gear_offset).extrude(gear_thinkness/2)
base = base.faces('<Z[2]').workplane().circle(m3_tap/2).cutBlind(-part_thinkness/2+wall_width)

# Threaded insert
base = base.faces('<<Y[-1]').workplane().center(-5,0).circle(6.5/2).cutBlind(-t-mount_gap*2-6)
base = base.faces('<<Y[-1]').workplane().circle(8.5/2).cutBlind(-t-mount_gap)

# Fastener
base = base.faces('<<Y[-1]').workplane().circle((fastener_width+1)/2).cutBlind(-4)
fastener = base.faces('<<Y[-4]').workplane().circle(fastener_width/2).extrude(4.5+fastener_height, combine=False)
fastener = fastener.faces('<<Y[-1]').workplane().center(-fastener_length,0)\
                                                .line(20,-fastener_width/2)\
                                                .vLine(fastener_width)\
                                                .close().extrude(-fastener_height)
fastener = fastener.faces('<<Y[-1]').workplane().circle(fastener_width/6).extrude(-fastener_height)
fastener = fastener.faces('<<Y[-1]').center(fastener_length,0)\
                                    .sketch().regularPolygon(8.5/2,6,tag='outer')\
                                             .finalize().cutBlind(-4)
fastener = fastener.faces('<<Y[-6]').workplane().circle(5/2).cutThruAll()

# Lid large gear mount point
lid = lid.faces('>Z[0]').workplane().center(21,-20.15)\
                                    .hLine(t, forConstruction=True)\
                                    .threePointArc((-t,t/2),(0,-t))\
                                    .polarLine(t, slant_degree)\
                                    .close().extrude(-lid_thicknesss+wall_width)
lid = lid.faces('>Z[0]').workplane().circle(large_gear_diameter/2-gear_offset).extrude(gear_thinkness/2)
lid = lid.faces('>Z[0]').workplane().circle(2).cutThruAll()
lid = lid.faces('<Z[0]').workplane().circle(m3_head_diameter/2).cutBlind(-m3_head_depth)

# Lid Holes
lid = lid.faces('<Z[0]').workplane().center(-stepper_motor_size+wall_width,-stepper_motor_size+offset/2-loop_width)\
                                    .circle(m3_passthrough/2).cutThruAll()
drill = lid.faces('<Z[0]').workplane().circle(m3_tap/2)\
                                      .extrude(-part_thinkness+wall_width, combine=False)
base = base.cut(drill)
lid = lid.faces('<Z[0]').workplane().circle(m3_head_diameter/2).cutBlind(-m3_head_depth)

# Gears
l_gear = cq.Workplane('XY').gear(SpurGear(module=mod,
                                          teeth_number=63,
                                          width=gear_thinkness,
                                          bore_d=large_gear_diameter))
l_gear = l_gear.translate((21+stepper_center[0],20.15+stepper_center[1]+offset+loop_width,(gear_thinkness+gear_offset)/2))
s_gear = cq.Workplane('XY').gear(SpurGear(module=mod,
                                          teeth_number=11,
                                          width=gear_thinkness))


# Stepper motor mount point
tab_distance = 2
shaft_diameter = 5
copy = s_gear.faces('<Z[0]').workplane().vLine(tab_distance, forConstruction=True)\
                                        .hLine(10).vLine(-10).hLine(-20).vLine(10)\
                                                                        .close().cutThruAll()
s_gear = s_gear.faces('<Z[0]').sketch().circle(shaft_diameter/2).finalize().cutThruAll()
s_gear = s_gear.add(copy)
s_gear = s_gear.translate((stepper_center[0],stepper_center[1]+offset+loop_width,(gear_thinkness+gear_offset)/2))

# Render
show_object(base, name='Follow Focus Base')
show_object(lid, name='Follow Focus Lid')
show_object(fastener, name='Follow Focus Fastener')
show_object(l_gear, name='Follow Focus Large Gear')
show_object(s_gear, name='Follow Focus Small Gear')