//
// Mendel90
//
// GNU GPL v2
// nop.head@gmail.com
// hydraraptor.blogspot.com
//
// Utilities
//
include <../utils/bom.scad>
include <../utils/polyholes.scad>
include <../utils/teardrops.scad>
include <../utils/cables.scad>
include <../utils/shields.scad>

module slot(h, r, l, center = true)
    linear_extrude(height = h, convexity = 6, center = center)
        hull() {
            translate([l/2,0,0])
                circle(r = r, center = true);
            translate([-l/2,0,0])
                circle(r = r, center = true);
        }

module hole_support(r, h, max_r = 999, closed = false) {
    n = sides(r);
    cr = corrected_radius(r, n);
    ir = min(cr, max_r - 2.25 * filament_width);
    or = ir + 2 * filament_width;
    difference() {
        cylinder(r = or, h = h, $fn = n);
        translate([0, 0, closed ? layer_height : -1])
            cylinder(r = ir, h = h + 2, $fn = n);
    }
}

module nut_trap_support(h, r, r2 = 0) {
    ir = r;
    ir2 = r2 ? r2 : r;
    or = ir + 2 * filament_width;
    or2 = ir2 + 2 * filament_width;
    difference() {
        cylinder(r2 = or, r1 = or2, h = h, $fn = 6);
        translate([0, 0, -1])
            cylinder(r2 = ir, r1 = or2, h = h + 2, $fn = 6);
    }
}

function nut_trap_radius(nut) = nut_radius(nut) + layer_height / 4;
function nut_trap_flat_radius(nut) = nut_trap_radius(nut) * cos(30);

module nut_trap(screw_r, nut_r, depth, horizontal = false, supported = false) {
    render(convexity = 5) union() {
        if(horizontal) {
            if(screw_r)
                teardrop_plus(r = screw_r, h = 200, center = true);
            cylinder(r = nut_r + layer_height / 4, h = depth * 2, center = true, $fn = 6);
        }
        else {
            difference() {
                union() {
                    if(screw_r)
                        poly_cylinder(r = screw_r, h = 200, center = true);
                    cylinder(r = nut_r, h = depth * 2, center = true, $fn = 6);
                }
                if(supported)
                    translate([0, 0, depth - eta])
                        cylinder(r = nut_r, h = layer_height, center = false);
            }
        }
    }
}

module part_screw_hole(screw, nut, horizontal = false, supported = false) {
    screw_r = screw_clearance_radius(screw);
    if(nut)
        translate([0, 0, nut_trap_depth(nut)])
            nut_trap(screw_r, nut_radius(nut), nut_trap_depth(nut), horizontal, supported);
    else
        if(horizontal)
            teardrop_plus(r = screw_r, h = 200, center = true);
        else
            poly_cylinder(r = screw_r, h = 200, center = true);
}

module fillet(r, h) {
    translate([r / 2, r / 2, 0])
        difference() {
            cube([r + eta, r + eta, h], center = true);
            translate([r/2, r/2, 0])
                cylinder(r = r, h = h + 1, center = true);
        }
}

module right_triangle(width, height, h, center = true) {
    linear_extrude(height = h, center = center)
        polygon(points = [[0,0], [width, 0], [0, height]]);
}

module rounded_square(w, h, r)
{
    union() {
        square([w - 2 * r, h], center = true);
        square([w, h - 2 * r], center = true);
        for(x = [-w/2 + r, w/2 - r])
            for(y = [-h/2 + r, h/2 - r])
                translate([x, y])
                    circle(r = r);
    }
}

module rounded_rectangle(size, r, center = true)
{
    w = size[0];
    h = size[1];
    linear_extrude(height = size[2], center = center)
        rounded_square(size[0], size[1], r);
}

//
// Cylinder with rounded ends
//
module rounded_cylinder(r, h, r2)
{
    rotate_extrude()
        union() {
            square([r - r2, h]);
            square([r, h - r2]);
            translate([r - r2, h - r2])
                circle(r = r2);
        }
}

module sector(r, a, h, , center = true) {
    linear_extrude(height = h, center = center)
        intersection() {
            circle(r = r, center = true);
                polygon(points = [
                    [0, 0],
                    [2 * r * cos(a / 2),  2 * r * sin(a / 2)],
                    [2 * r * cos(a / 2), -2 * r * sin(a / 2)],
                ]);
        }
}

module tube(or, ir, h, center = true) {
    linear_extrude(height = h, center = center, convexity = 5)
        difference() {
            circle(or);
            circle(ir);
        }
}

//
// Exploded view helper
//
module explode(v, offset = [0,0,0]) {
    if(exploded) {
        translate(v * exploded)
            child();
        render() hull() {
            sphere(0.2);
            translate(v * exploded + offset)
                sphere(0.2);
        }
    }
    else
        child();
}
//
// Restore the view point
//
module view(t,r,d = 1000)
    rotate([55, 0, 25])
        translate([0, 0, -d + 500])
            rotate([-r[0], 0, 0])
                rotate([0, -r[1], 0])
                    rotate([0, 0, -r[2]])
                        translate(-t)
                            child();