#!/usr/bin/env python3
# type: ignore

#**************************************************************************
#   Copyright (C) 2025, Paul Lutus                                        *
#                                                                         *
#   This program is free software; you can redistribute it and/or modify  *
#   it under the terms of the GNU General Public License as published by  *
#   the Free Software Foundation; either version 2 of the License, or     *
#   (at your option) any later version.                                   *
#                                                                         *
#   This program is distributed in the hope that it will be useful,       *
#   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
#   GNU General Public License for more details.                          *
#                                                                         *
#   You should have received a copy of the GNU General Public License     *
#   along with this program; if not, write to the                         *
#   Free Software Foundation, Inc.,                                       *
#   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
#**************************************************************************

import cadquery as cq

import math, sys

# classic interpolation algorithm


def ntrp(x, xa, xb, ya, yb):
    return (x - xa) * (yb - ya) / (xb - xa) + ya


def save_part(data, name):
    path = f"{name}.stl"
    cq.exporters.export(data, path)
    print(f'Saved "{path}"')


def build_container(W, H, D, ax, scale, corner_radius, wall_thickness, cover=False):
    box = (
        cq.Workplane("XY")
        .rect(W * scale, H * scale / 2)
        .extrude(D * scale)
        .translate((0, -H * scale / 4, 0))
    )
    box = box.faces("<Y").edges("|Z").fillet(corner_radius)
    top_arc = cq.Workplane("XY").circle(H * scale / 2).extrude(D * scale)
    if not cover:
        axle = (
            cq.Workplane("XY")
            .circle(ax / scale**4)
            .extrude(D * scale)
            .translate((0, 0, 0))
        )
        return box + top_arc - axle
    else:
        return box + top_arc


def build_subassembly(
    W, H, D, ax, corner_radius, filament_opening_radius, wall_thickness, cover=False
):
    body = build_container(W, H, D, ax, 1.0, corner_radius, wall_thickness, cover)
    comp_w = (W - wall_thickness) / W
    interior = build_container(
        W, H, D, ax, comp_w, corner_radius, wall_thickness, cover
    )
    body -= interior.translate((0, 0, wall_thickness))
    if not cover:
        wedge = (
            cq.Workplane("XZ")
            .rect(W - corner_radius * 2, H / 16.0)
            .extrude(-wall_thickness / 2)
            .edges("|Y")
            .fillet(H / 128.0)
            .translate((0, -((H / 2) + wall_thickness / 2), H / 32))
        )
        body += wedge
        port = (
            cq.Workplane("XZ")
            .circle(filament_opening_radius)
            .extrude(64)
            .translate((0, 32, 0))
            .rotate((0, 0, 0), (0, 0, 1), -45)
            .translate((0, H / 2, D / 2))
        )
        body -= port
    return body - interior.translate((0, 0, wall_thickness))


def main():

    # box data, all dimensions mm

    name = "filament_reel_holder"

    # ALl dimensions in mm

    wall_thickness = 3.0

    W = 210  # width (X)
    H = 210  # height (Y)
    D = 80  # depth (Z)
    ax = 25  # axle radius
    filament_opening_radius = 2.0

    # curvature at corners
    corner_radius = 10

    # this determines the tightness of fit between body and cover
    body_cover_gap = 0.05

    cover_edge_height = 20

    body = build_subassembly(
        W, H, D, ax, corner_radius, filament_opening_radius, wall_thickness
    )

    save_part(body, f"{name}_body")

    cover = build_subassembly(
        W + body_cover_gap + wall_thickness,
        H + body_cover_gap + wall_thickness,
        cover_edge_height,
        ax,
        corner_radius,
        filament_opening_radius,
        wall_thickness,
        True,
    )

    save_part(cover, f"{name}_cover")


if __name__ == "__main__":
    main()