import cadquery as cq
from jupyter_cadquery.cadquery import show
from jupyter_cadquery import set_sidecar

set_sidecar("CadQuery")

# Build list of (x,y) points from xs and ys
def permute(data):
    out = []
    for x in data['xs']:
        for y in data['ys']:
            out.append((x,y))
    return out

# Center Option is specified so that origin of work plane is consistent.
# The default method is to use center of mass to center the origin, 
# which causes geometry to walk around if operations are re-ordered.    
def top(obj):
    return obj.faces(">Z").workplane(centerOption='ProjectedOrigin', origin=cq.Vector(0,0,1))


# Definition of a stock end plate with 
# corner radius and mounting holes.    
stock = dict(
    height = 29.5, width = 108.5,
    thickness = 2, radius = 3,
    holes = dict(
   	 	diameter = 3.5,
		xs = [-96.5/2, 96.5/2],
    	ys = [11.7, -9.8]
	)
)

# Make the rectangle
stock_plate = cq.Workplane("XY").rect(stock['width'],    stock['height']).extrude(stock['thickness'])

# Cut the mounting holes and radius the corners
    stock_plate = top(stock_plate) \
    .pushPoints(permute(stock['holes'])).hole(stock['holes']['diameter']) \
    .edges("|Z").fillet(stock['radius'])

# Create the geometry and save to a STEP file
stock_plate.val().exportStep("stock.step")

# Show in the WebGL view
show(stock_plate, axes=False, grid=True, ortho=True, axes0=True)

# XS and YS are measured from bottom left corner of panel
usb = dict(
    height = 14.75, width = 8.00, radius = 1.75,
    xs = [18.25, 32.25, 46.25, 60.25],
    ys = [16.12]
)

rgb = dict(
    diameter = 2.75,
    xs = [68.57 + idx*4.57 for idx in [0,1,2,3,4,5]],
    ys = [13.00, 17.13]
)


# Transform positions measured from bottom-left corner to offsets from center (left-right & up-down) of plate
def corner(points):
    return [(x-stock['width']/2,y-stock['height']/2) for x,y in points]

# This operation could be done with rectangular cut + selection 
# of vertical edges & radius operation.  But, limiting selection
# to just the newly created edges is difficult in some designs.
def round_rect(plane, center, opts):

    w = opts['width']/2
    h = opts['height']/2
    r = opts['radius']

    # Move to bottom left corner, top of that arc
    # Left edge and top left arc
    # Top edge and top right arc
    # Right edge and bottom right arc
    # Bottom edge and bottom left arc

    return plane.center(*center) \
        .moveTo(-w, -h+r) \
        .vLineTo( h-r).radiusArc((-w+r,   h), r) \
        .hLineTo( w-r).radiusArc((   w, h-r), r) \
        .vLineTo(-h+r).radiusArc(( w-r,  -h), r) \
        .hLineTo(-w+r).radiusArc((  -w,-h+r), r) \
        .close()


# Start with the stock plate geometry
front_plate = stock_plate

# Cut holes for USB connectors
for xy in corner(permute(usb)):
    front_plate = round_rect(top(front_plate), xy, usb).cutThruAll()

# Cut holes for RGB status LEDs
front_plate = top(front_plate).pushPoints(corner(permute(rgb))).hole(rgb['diameter'])  

# Save STEP file
front_plate.val().exportStep("front-plate.step")