import cadquery as cq

import math

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

slant_degree=15.0
thickness=20
rod_thickness=15.2
loop_thickness=1.5
stepper_motor_size=42
offset=9

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)))
base = cq.Workplane('XY').lineTo(0,offset).polarLine(stepper_motor_size, 90-slant_degree)
end_x=gap+stepper_motor_size-5
end_y=-4.5
angle=40
outer_ring_diameter=rod_thickness+2*loop_thickness
center_x=end_x-math.cos(math.radians(angle))*outer_ring_diameter/2
center_y=end_y-math.sin(math.radians(angle))*outer_ring_diameter/2

def findRodInter(n_x, n_y):
    p = Point(center_x, center_y)
    c = p.buffer(rod_thickness/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


base = base.hLine(width).polarLine(stepper_motor_size-5,270-slant_degree).tangentArcPoint((end_x, end_y), relative=False)
d=20
t=7
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(d, 180-slant_degree).vLine(t)

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+2)

base = base.lineTo(coor_x, coor_y)
base = base.threePointArc((center_x+rod_thickness/2, center_y), (last_x, last_y))
base = base.polarLine(d, 180-slant_degree)
base = base.close().extrude(20)

# Fillets
outer_fillets=2.5
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)

# Shell
base = base.faces('+Z').shell(-1.6)

# Stepper motor mount points
stepper_hole_width=31.0
stepper_hole_radius=stepper_hole_width/2*math.sqrt(2)
stepper_center = cq.Workplane('XY').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(-1.6)

# Render
show_object(base, name='Follow Focus Base')