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=8.83

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=-3.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=6
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(15, 180-slant_degree)
base = base.close().extrude(20)

# Render
show_object(base)