use acadrust::entities::Arc;
use crate::t;

use crate::command::EntityTransform;
use crate::entities::common::{
    center_grip, edit_angle_prop as edit_angle, edit_prop as edit, format_angle, format_area,
    format_length, parse_f64, ro_prop as ro, square_grip,
};
use crate::entities::traits::RenderConvertible;
use crate::scene::convert::acad_to_render::{extrusion_wall_tris, RenderEntity, RenderObject};
use crate::scene::model::object::{GripApply, GripDef, PropSection};
use crate::scene::model::wire_model::TangentGeom;

const TAU: f64 = std::f64::consts::TAU;

fn to_render(arc: &Arc) -> RenderEntity {
    let cx = arc.center.x;
    let cy = arc.center.y;
    let cz = arc.center.z;
    let r = arc.radius;
    let sa = arc.start_angle;
    let ea = arc.end_angle;
    let normal = (arc.normal.x, arc.normal.y, arc.normal.z);

    let (ax, ay) = crate::scene::view::transform::ocs_axes(normal);
    let (cwx, cwy, cwz) = crate::scene::view::transform::ocs_point_to_wcs((cx, cy, cz), normal);
    let arc_pt = |a: f64| {
        let (c, s) = (a.cos(), a.sin());
        [
            cwx + r * c * ax.0 + r * s * ay.0,
            cwy + r * c * ax.1 + r * s * ay.1,
            cwz + r * c * ax.2 + r * s * ay.2,
        ]
    };

    // Centre, ends, arc-length midpoint and the quadrants the sweep actually
    // covers, all from the entity's own curve. Circles and ellipses (closed
    // curves) deliberately emit no midpoint; see #34.
    let curve = crate::entities::curve::arc_curve(arc);
    let snap = crate::entities::curve::snap_from(&curve);
    let tangent = TangentGeom::Arc {
        center: [cwx, cwy, cwz],
        axis_x: [ax.0, ax.1, ax.2],
        axis_y: [ay.0, ay.1, ay.2],
        radius: r,
        start_angle: sa,
        end_angle: ea,
    };

    if arc.thickness.abs() > 1e-10 {
        let t = arc.thickness;
        let (nx, ny, nz) = normal;
        let n = 32usize;
        let ccw_end = if ea >= sa { ea } else { ea + TAU };
        let (start_a, end_a) = (sa, ccw_end);
        let base: Vec<[f64; 3]> = (0..=n)
            .map(|i| {
                let p = arc_pt(start_a + (end_a - start_a) * (i as f64 / n as f64));
                [p[0], p[1], p[2]]
            })
            .collect();
        let mut pts: Vec<[f64; 3]> = Vec::with_capacity((n + 1) * 2 + 8);
        pts.extend_from_slice(&base);
        pts.push([f64::NAN; 3]);
        for &[x, y, z] in &base {
            pts.push([x + t * nx, y + t * ny, z + t * nz]);
        }
        pts.push([f64::NAN; 3]);
        for end in [arc_pt(sa), arc_pt(ea)] {
            pts.push(end);
            pts.push([end[0] + t * nx, end[1] + t * ny, end[2] + t * nz]);
            pts.push([f64::NAN; 3]);
        }
        pts.pop();
        return RenderEntity {
            pick_tris: extrusion_wall_tris(&base, [t * nx, t * ny, t * nz]),
            object: RenderObject::Lines(pts),
            snap_pts: snap.snap_pts.clone(),
            tangent_geoms: vec![tangent],
            key_vertices: vec![],
            fill_tris: vec![],
        };
    }

    // Sampled through the arc's own curve, which is the one definition of
    // its geometry. EXTRUDE, REVOLVE and SWEEP read that definition directly
    // rather than this point list, so nothing is lost by drawing from it.
    let points = crate::entities::curve::curve_points(&crate::entities::curve::arc_curve(arc));
    RenderEntity {
        pick_tris: Vec::new(),
        object: RenderObject::Lines(points),
        snap_pts: snap.snap_pts.clone(),
        tangent_geoms: vec![tangent],
        key_vertices: vec![],
        fill_tris: vec![],
    }
}

fn control_points(arc: &Arc) -> [glam::DVec3; 3] {
    let curve = crate::entities::curve::arc_curve(arc);
    [
        glam::DVec3::from_array(curve.point_at(0.0)),
        glam::DVec3::from_array(curve.point_at(0.5)),
        glam::DVec3::from_array(curve.point_at(1.0)),
    ]
}

pub(crate) fn refit_grips(
    arc: &mut Arc,
    original: &Arc,
    edits: &[(usize, glam::DVec3)],
) -> bool {
    let mut points = control_points(original);
    let mut changed = false;
    for &(grip_id, point) in edits {
        let index = match grip_id {
            1 => 0,
            2 => 2,
            3 => 1,
            _ => continue,
        };
        points[index] = point;
        changed = true;
    }
    if !changed {
        return false;
    }

    let plane = crate::entities::curve::arc_curve(original).plane;
    let Some(a) = plane.project(points[0].to_array()) else {
        return false;
    };
    let Some(b) = plane.project(points[1].to_array()) else {
        return false;
    };
    let Some(c) = plane.project(points[2].to_array()) else {
        return false;
    };
    let Some(fit) = cadkernel::geom2d::arc_through_points(a, b, c) else {
        return false;
    };

    arc.center.x = fit.centre[0];
    arc.center.y = fit.centre[1];
    arc.center.z = original.center.z;
    arc.radius = fit.radius;
    arc.start_angle = fit.start_angle;
    arc.end_angle = fit.end_angle;
    true
}

fn grips(arc: &Arc) -> Vec<GripDef> {
    let (x, y, z) = crate::scene::view::transform::ocs_point_to_wcs(
        (arc.center.x, arc.center.y, arc.center.z),
        (arc.normal.x, arc.normal.y, arc.normal.z),
    );
    let ctr = glam::DVec3::new(x, y, z);
    let [start, middle, end] = control_points(arc);
    vec![
        center_grip(0, ctr),
        square_grip(1, start),
        square_grip(2, end),
        square_grip(3, middle),
    ]
}

fn properties(arc: &Arc) -> Vec<PropSection> {
    let r = arc.radius;
    let sa = arc.start_angle;
    let ea = arc.end_angle;
    let sweep = (ea - sa).rem_euclid(TAU);
    let total_angle = sweep.to_degrees();
    let arc_length = r * sweep;
    let area = crate::entities::curve::arc_curve(arc)
        .curve
        .chord_closed_area()
        .unwrap_or(0.0)
        .abs();

    let normal = (arc.normal.x, arc.normal.y, arc.normal.z);
    let (ax, ay) = crate::scene::view::transform::ocs_axes(normal);
    let (cwx, cwy, cwz) = crate::scene::view::transform::ocs_point_to_wcs(
        (arc.center.x, arc.center.y, arc.center.z),
        normal,
    );
    let arc_pt = |a: f64| {
        let (c, s) = (a.cos(), a.sin());
        (
            cwx + r * c * ax.0 + r * s * ay.0,
            cwy + r * c * ax.1 + r * s * ay.1,
            cwz + r * c * ax.2 + r * s * ay.2,
        )
    };
    let (sx, sy, sz) = arc_pt(sa);
    let (ex, ey, ez) = arc_pt(ea);

    vec![PropSection {
        title: t!("Geometry").into_owned(),
        props: vec![
            ro(t!("Start X").as_ref(), "start_x", format_length(sx)),
            ro(t!("Start Y").as_ref(), "start_y", format_length(sy)),
            ro(t!("Start Z").as_ref(), "start_z", format_length(sz)),
            edit(t!("Center X").as_ref(), "center_x", cwx),
            edit(t!("Center Y").as_ref(), "center_y", cwy),
            edit(t!("Center Z").as_ref(), "center_z", cwz),
            ro(t!("End X").as_ref(), "end_x", format_length(ex)),
            ro(t!("End Y").as_ref(), "end_y", format_length(ey)),
            ro(t!("End Z").as_ref(), "end_z", format_length(ez)),
            edit(t!("Radius").as_ref(), "radius", arc.radius),
            edit_angle(t!("Start angle").as_ref(), "start_angle", sa.to_degrees()),
            edit_angle(t!("End angle").as_ref(), "end_angle", ea.to_degrees()),
            ro(t!("Total angle").as_ref(), "total_angle", format_angle(total_angle.to_radians())),
            ro(t!("Arc length").as_ref(), "arc_length", format_length(arc_length)),
            ro(t!("Area").as_ref(), "area", format_area(area)),
            edit(t!("Normal X").as_ref(), "normal_x", arc.normal.x),
            edit(t!("Normal Y").as_ref(), "normal_y", arc.normal.y),
            edit(t!("Normal Z").as_ref(), "normal_z", arc.normal.z),
        ],
    }]
}

fn apply_geom_prop(arc: &mut Arc, field: &str, value: &str) {
    let Some(v) = parse_f64(value) else {
        return;
    };
    match field {
        "center_x" | "center_y" | "center_z" => {
            let normal = (arc.normal.x, arc.normal.y, arc.normal.z);
            let (mut x, mut y, mut z) = crate::scene::view::transform::ocs_point_to_wcs(
                (arc.center.x, arc.center.y, arc.center.z),
                normal,
            );
            match field {
                "center_x" => x = v,
                "center_y" => y = v,
                "center_z" => z = v,
                _ => {}
            }
            let (ox, oy, oz) = crate::scene::view::transform::wcs_point_to_ocs((x, y, z), normal);
            arc.center.x = ox;
            arc.center.y = oy;
            arc.center.z = oz;
        }
        "radius" if v > 0.0 => arc.radius = v,
        "start_angle" => arc.start_angle = v.to_radians(),
        "end_angle" => arc.end_angle = v.to_radians(),
        "normal_x" | "normal_y" | "normal_z" => {
            let center = arc.center_wcs();
            let mut normal = cadkernel::space::Vec3::new(arc.normal.x, arc.normal.y, arc.normal.z);
            match field {
                "normal_x" => normal.x = v,
                "normal_y" => normal.y = v,
                "normal_z" => normal.z = v,
                _ => {}
            }
            if let Some(normal) = normal.normalize() {
                arc.normal = acadrust::types::Vector3::new(normal.x, normal.y, normal.z);
                let (x, y, z) = crate::scene::view::transform::wcs_point_to_ocs(
                    (center.x, center.y, center.z),
                    (normal.x, normal.y, normal.z),
                );
                arc.center = acadrust::types::Vector3::new(x, y, z);
            }
        }
        _ => {}
    }
}

fn apply_grip(arc: &mut Arc, grip_id: usize, apply: GripApply) {
    match (grip_id, apply) {
        (0, GripApply::Translate(d)) => {
            let (x, y, z) = crate::scene::view::transform::wcs_point_to_ocs(
                (d.x, d.y, d.z),
                (arc.normal.x, arc.normal.y, arc.normal.z),
            );
            arc.center.x += x;
            arc.center.y += y;
            arc.center.z += z;
        }
        (0, GripApply::Absolute(p)) => {
            let (x, y, z) = crate::scene::view::transform::wcs_point_to_ocs(
                (p[0], p[1], p[2]),
                (arc.normal.x, arc.normal.y, arc.normal.z),
            );
            arc.center.x = x;
            arc.center.y = y;
            arc.center.z = z;
        }
        (1..=3, GripApply::Absolute(p)) => {
            let original = arc.clone();
            let _ = refit_grips(arc, &original, &[(grip_id, p)]);
        }
        _ => {}
    }
}

fn apply_transform(arc: &mut Arc, t: &EntityTransform) {
    crate::scene::view::transform::apply_standard_entity_transform(arc, t, |entity, p1, p2| {
        crate::scene::view::transform::reflect_xy_point(
            &mut entity.center.x,
            &mut entity.center.y,
            p1,
            p2,
        );
        let dx = (p2.x - p1.x) as f64;
        let dy = (p2.y - p1.y) as f64;
        let line_angle = dy.atan2(dx);
        let tmp = entity.start_angle;
        entity.start_angle = 2.0 * line_angle - entity.end_angle;
        entity.end_angle = 2.0 * line_angle - tmp;
    });
}

impl RenderConvertible for Arc {
    fn to_render(&self, _document: &acadrust::CadDocument) -> Option<RenderEntity> {
        Some(to_render(self))
    }
}

impl crate::entities::traits::Grippable for Arc {
    fn grips(&self) -> Vec<GripDef> {
        grips(self)
    }
    fn apply_grip(&mut self, grip_id: usize, apply: GripApply) {
        apply_grip(self, grip_id, apply);
    }
    fn grip_menu(&self, grip_id: usize) -> Vec<crate::scene::model::object::GripMenuItem> {
        use crate::scene::model::object::{GripMenuAction, GripMenuItem};
        match grip_id {
            0 => vec![GripMenuItem {
                label: "Stretch",
                action: GripMenuAction::Stretch,
            }],
            3 => vec![
                GripMenuItem {
                    label: "Stretch",
                    action: GripMenuAction::Stretch,
                },
                GripMenuItem {
                    label: "Radius",
                    action: GripMenuAction::Radius,
                },
                GripMenuItem {
                    label: "Arc Length",
                    action: GripMenuAction::ArcLength,
                },
            ],
            _ => vec![
                GripMenuItem {
                    label: "Stretch",
                    action: GripMenuAction::Stretch,
                },
                GripMenuItem {
                    label: "Lengthen",
                    action: GripMenuAction::Lengthen,
                },
            ],
        }
    }
    fn apply_grip_menu(&mut self, _grip_id: usize, _action: crate::scene::model::object::GripMenuAction) {
        // Radius / Arc Length / Lengthen all need a follow-up prompt;
        // the actual edit happens in `apply_grip_menu_value`.
    }

    fn grip_menu_value_prompt(
        &self,
        _grip_id: usize,
        action: crate::scene::model::object::GripMenuAction,
    ) -> Option<&'static str> {
        use crate::scene::model::object::GripMenuAction as A;
        match action {
            A::Radius => Some("New radius"),
            A::ArcLength => Some("New arc length"),
            A::Lengthen => Some("Distance"),
            _ => None,
        }
    }

    fn grip_menu_point_value(
        &self,
        grip_id: usize,
        action: crate::scene::model::object::GripMenuAction,
        point: glam::DVec3,
    ) -> Option<f64> {
        use cadkernel::geom2d::{Circle as KernelCircle, Curve as KernelCurve, Vec2};
        use crate::scene::model::object::GripMenuAction as A;
        if self.radius <= 1.0e-9 {
            return None;
        }
        let curve = crate::entities::curve::arc_curve(self);
        let point = curve.plane.project(point.to_array())?;
        let KernelCurve::Arc(arc) = &curve.curve else {
            unreachable!("arc entity must produce an arc curve")
        };
        if matches!(action, A::Radius) && grip_id == 3 {
            let radius = Vec2::from(point).distance(Vec2::from(arc.centre));
            return (radius > 1.0e-9).then_some(radius);
        }
        if matches!(action, A::ArcLength) && grip_id == 3 {
            let middle = Vec2::from(curve.curve.point_at(0.5));
            let tangent = Vec2::from(curve.curve.tangent_at(0.5)).normalize()?;
            let length = curve.length() + (Vec2::from(point) - middle).dot(tangent);
            let circumference = KernelCircle {
                centre: arc.centre,
                radius: arc.radius,
            }
            .length();
            return (length > 1.0e-9 && length < circumference - 1.0e-9).then_some(length);
        }
        if !matches!(action, A::Lengthen) {
            return None;
        }
        let cursor_angle = KernelCurve::Circle(KernelCircle {
            centre: arc.centre,
            radius: arc.radius,
        })
        .parameter_at(point)
            * TAU;
        let current_sweep = arc.sweep();
        let desired_sweep = match grip_id {
            1 => (arc.end_angle - cursor_angle).rem_euclid(TAU),
            2 => (cursor_angle - arc.start_angle).rem_euclid(TAU),
            _ => return None,
        };
        if desired_sweep <= 1.0e-9 {
            return None;
        }
        Some((desired_sweep - current_sweep) * self.radius)
    }

    fn apply_grip_menu_value(
        &mut self,
        grip_id: usize,
        action: crate::scene::model::object::GripMenuAction,
        value: f64,
    ) {
        use crate::scene::model::object::GripMenuAction as A;
        match action {
            A::Radius if value > 0.0 => self.radius = value,
            A::ArcLength if value > 0.0 && self.radius > 1e-9 => {
                // Hold start_angle, derive new end_angle from arc length
                // = r * Δθ.
                let new_span = value / self.radius;
                if new_span < TAU - 1.0e-9 {
                    self.end_angle = self.start_angle + new_span;
                }
            }
            A::Lengthen => {
                // Extend either end by `value` arc-length units along
                // the arc. Positive `value` lengthens; negative
                // shortens. Grip 1 = start endpoint, grip 2 = end endpoint.
                if self.radius < 1e-9 {
                    return;
                }
                let dtheta = value / self.radius;
                let current = (self.end_angle - self.start_angle).rem_euclid(TAU);
                let next = current + dtheta;
                if next <= 1.0e-9 || next >= TAU - 1.0e-9 {
                    return;
                }
                match grip_id {
                    1 => self.start_angle -= dtheta,
                    2 => self.end_angle += dtheta,
                    _ => {}
                }
            }
            _ => {}
        }
    }
}

impl crate::entities::traits::PropertyEditable for Arc {
    fn geometry_properties(&self, _text_style_names: &[String]) -> Vec<PropSection> {
        properties(self)
    }
    fn apply_geom_prop(&mut self, field: &str, value: &str) {
        apply_geom_prop(self, field, value);
    }
}

impl crate::entities::traits::Transformable for Arc {
    fn apply_transform(&mut self, t: &EntityTransform) {
        apply_transform(self, t);
    }
}

impl crate::entities::traits::MassPropsCalc for acadrust::entities::Arc {
    fn mass_props(&self) -> crate::entities::traits::MassProps {
        let curve = crate::entities::curve::arc_curve(self);
        let area = curve.curve.chord_closed_area().unwrap_or(0.0).abs();
        let centroid = curve
            .curve
            .chord_closed_centroid()
            .map(|point| curve.plane.point_at(point))
            .unwrap_or_else(|| curve.point_at(0.5));
        crate::entities::traits::MassProps {
            area,
            perimeter: curve.length(),
            cx: centroid[0],
            cy: centroid[1],
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::entities::traits::Grippable;
    use crate::scene::model::object::GripMenuAction;

    #[test]
    fn midpoint_grip_drives_radius_and_arc_length_in_entity_plane() {
        let mut arc = Arc::default();
        arc.center = acadrust::types::Vector3::new(1.0, 2.0, 3.0);
        arc.normal = acadrust::types::Vector3::new(0.0, 1.0, 0.0);
        arc.radius = 2.0;
        arc.start_angle = 0.0;
        arc.end_angle = std::f64::consts::FRAC_PI_2;

        let curve = crate::entities::curve::arc_curve(&arc);
        let middle_angle = std::f64::consts::FRAC_PI_4;
        let radius_point = curve.plane.point_at([
            arc.center.x + 5.0 * middle_angle.cos(),
            arc.center.y + 5.0 * middle_angle.sin(),
        ]);
        let radius = arc
            .grip_menu_point_value(3, GripMenuAction::Radius, radius_point.into())
            .expect("radius value");
        assert!((radius - 5.0).abs() < 1.0e-9);

        let middle = glam::DVec3::from_array(curve.point_at(0.5));
        let tangent = glam::DVec3::from_array(curve.tangent_at(0.5)).normalize();
        let wanted_length = curve.length() + 1.25;
        let arc_length = arc
            .grip_menu_point_value(
                3,
                GripMenuAction::ArcLength,
                middle + tangent * 1.25,
            )
            .expect("arc length value");
        assert!((arc_length - wanted_length).abs() < 1.0e-9);

        arc.apply_grip_menu_value(3, GripMenuAction::ArcLength, arc_length);
        assert!(
            (crate::entities::curve::arc_curve(&arc).length() - wanted_length).abs() < 1.0e-9
        );
    }
}