use acadrust::entities::{Polyline, Polyline2D, Polyline3D}; use truck_modeling::{builder, Edge, Point3, Wire}; use crate::command::EntityTransform; use crate::entities::common::{edit_prop as edit, parse_f64, ro_prop as ro, square_grip}; use crate::entities::traits::{Grippable, PropertyEditable, Transformable, TruckConvertible}; use crate::scene::convert::acad_to_truck::{TruckEntity, TruckObject}; use crate::scene::model::object::{GripApply, GripDef, PropSection, PropValue, Property}; use crate::scene::model::wire_model::TangentGeom; // ── Polyline (old-style 3D heavy polyline) ──────────────────────────────────── fn tessellate_polyline(pl: &Polyline) -> TruckEntity { let pts: Vec<[f64; 3]> = pl .vertices .iter() .map(|v| [v.location.x, v.location.y, v.location.z]) .collect(); let mut points = pts.clone(); if pl.flags.is_closed() && pts.len() >= 2 { points.push(pts[0]); } let key_verts = pts.clone(); TruckEntity { object: TruckObject::Lines(points), snap_pts: vec![], tangent_geoms: vec![], key_vertices: key_verts, fill_tris: vec![], } } impl TruckConvertible for Polyline { fn to_truck(&self, _document: &acadrust::CadDocument) -> Option { Some(tessellate_polyline(self)) } } impl Grippable for Polyline { fn grips(&self) -> Vec { self.vertices .iter() .enumerate() .map(|(i, v)| { square_grip( i, glam::DVec3::new(v.location.x, v.location.y, v.location.z), ) }) .collect() } fn apply_grip(&mut self, grip_id: usize, apply: GripApply) { if let Some(v) = self.vertices.get_mut(grip_id) { match apply { GripApply::Translate(d) => { v.location.x += d.x as f64; v.location.y += d.y as f64; v.location.z += d.z as f64; } GripApply::Absolute(p) => { v.location.x = p.x as f64; v.location.y = p.y as f64; v.location.z = p.z as f64; } } } } fn grip_menu(&self, _grip_id: usize) -> Vec { use crate::scene::model::object::{GripMenuAction, GripMenuItem}; vec![ GripMenuItem { label: "Stretch", action: GripMenuAction::Stretch, }, GripMenuItem { label: "Add Vertex", action: GripMenuAction::AddVertex, }, GripMenuItem { label: "Remove Vertex", action: GripMenuAction::RemoveVertex, }, ] } fn apply_grip_menu(&mut self, grip_id: usize, action: crate::scene::model::object::GripMenuAction) { use crate::scene::model::object::GripMenuAction as A; let n = self.vertices.len(); match action { A::AddVertex if grip_id < n => { let i1 = (grip_id + 1) % n; if i1 == 0 && grip_id + 1 == n { return; } let v0 = &self.vertices[grip_id]; let v1 = &self.vertices[i1]; let mx = (v0.location.x + v1.location.x) * 0.5; let my = (v0.location.y + v1.location.y) * 0.5; let mz = (v0.location.z + v1.location.z) * 0.5; let mut new_v = v0.clone(); new_v.location.x = mx; new_v.location.y = my; new_v.location.z = mz; let insert_at = (grip_id + 1).min(self.vertices.len()); self.vertices.insert(insert_at, new_v); } A::RemoveVertex if grip_id < n && n > 2 => { self.vertices.remove(grip_id); } _ => {} } } } impl PropertyEditable for Polyline { fn geometry_properties(&self, _text_style_names: &[String]) -> Vec { vec![PropSection { title: "Geometry".into(), props: vec![ ro("Vertices", "vertices", self.vertices.len().to_string()), Property { label: "Closed".into(), field: "pl_closed", value: PropValue::BoolToggle { field: "pl_closed", value: self.flags.is_closed(), }, }, ], }] } fn apply_geom_prop(&mut self, field: &str, value: &str) { if field == "pl_closed" { let closed = if value == "toggle" { !self.flags.is_closed() } else { value == "true" }; self.flags.set_closed(closed); } } } impl Transformable for Polyline { fn apply_transform(&mut self, t: &EntityTransform) { crate::scene::view::transform::apply_standard_entity_transform(self, t, |entity, p1, p2| { for v in &mut entity.vertices { crate::scene::view::transform::reflect_xy_point( &mut v.location.x, &mut v.location.y, p1, p2, ); } }); } } // ── Polyline2D (heavy 2D polyline with bulge) ───────────────────────────────── fn tessellate_polyline2d(pl: &Polyline2D) -> TruckEntity { let verts = &pl.vertices; if verts.is_empty() { return TruckEntity { object: TruckObject::Lines(vec![]), snap_pts: vec![], tangent_geoms: vec![], key_vertices: vec![], fill_tris: vec![], }; } let elev = pl.elevation; let normal = (pl.normal.x, pl.normal.y, pl.normal.z); let count = verts.len(); let seg_count = if pl.is_closed() { count } else { count - 1 }; let mut edges: Vec = Vec::new(); let mut tangents: Vec = Vec::new(); let mut key_verts: Vec<[f64; 3]> = Vec::new(); let to_wcs = |x: f64, y: f64| -> (f64, f64, f64) { crate::scene::view::transform::ocs_point_to_wcs((x, y, elev), normal) }; let to_pt = |v: &acadrust::entities::Vertex2D| -> Point3 { let (wx, wy, wz) = to_wcs(v.location.x, v.location.y); Point3::new(wx, wy, wz) }; if pl.thickness.abs() > 1e-10 { let (nx, ny, nz) = normal; let t = pl.thickness; let off = |p: [f64; 3]| -> [f64; 3] { [p[0] + t * nx, p[1] + t * ny, p[2] + t * nz] }; let to_f32 = |p: [f64; 3]| -> [f32; 3] { [p[0] as f32, p[1] as f32, p[2] as f32] }; let mut path: Vec<[f64; 3]> = Vec::new(); let mut kv: Vec<[f64; 3]> = Vec::new(); let mut tgs: Vec = Vec::new(); let (w0x, w0y, w0z) = to_wcs(verts[0].location.x, verts[0].location.y); path.push([w0x, w0y, w0z]); kv.push([w0x, w0y, w0z]); for i in 0..seg_count { let va = &verts[i]; let vb = &verts[(i + 1) % count]; let (ox0, oy0) = (va.location.x, va.location.y); let (ox1, oy1) = (vb.location.x, vb.location.y); let bulge = va.bulge; if bulge.abs() < 1e-9 { let (wx, wy, wz) = to_wcs(ox1, oy1); path.push([wx, wy, wz]); let p1_pt = path[path.len() - 2]; let p2_pt = *path.last().unwrap(); tgs.push(TangentGeom::Line { p1: to_f32(p1_pt), p2: to_f32(p2_pt), }); } else if let Some(arc) = crate::entities::common::BulgeArc::from_bulge([ox0, oy0], [ox1, oy1], bulge) { let (wcx, wcy, wcz) = to_wcs(arc.center[0], arc.center[1]); tgs.push(TangentGeom::Circle { center: [wcx as f32, wcy as f32, wcz as f32], radius: arc.radius as f32, }); for j in 1..=16usize { let s = arc.sample(j as f64 / 16.0); let (wx, wy, wz) = to_wcs(s[0], s[1]); path.push([wx, wy, wz]); } } let (wbx, wby, wbz) = to_wcs(ox1, oy1); kv.push([wbx, wby, wbz]); } let mut pts: Vec<[f64; 3]> = Vec::with_capacity(path.len() * 2 + kv.len() * 3 + 4); pts.extend_from_slice(&path); pts.push([f64::NAN; 3]); for &p in &path { pts.push(off(p)); } if !kv.is_empty() { pts.push([f64::NAN; 3]); for (i, &pb) in kv.iter().enumerate() { pts.push(pb); pts.push(off(pb)); if i + 1 < kv.len() { pts.push([f64::NAN; 3]); } } } return TruckEntity { object: TruckObject::Lines(pts), snap_pts: vec![], tangent_geoms: tgs, key_vertices: kv, fill_tris: vec![], }; } for i in 0..seg_count { let v0 = &verts[i]; let v1 = &verts[(i + 1) % count]; let p0 = to_pt(v0); let p1 = to_pt(v1); let bulge = v0.bulge; if bulge.abs() < 1e-9 { let tv0 = builder::vertex(p0); let tv1 = builder::vertex(p1); edges.push(builder::line(&tv0, &tv1)); tangents.push(TangentGeom::Line { p1: [p0.x as f32, p0.y as f32, p0.z as f32], p2: [p1.x as f32, p1.y as f32, p1.z as f32], }); } else if let Some(arc) = crate::entities::common::BulgeArc::from_bulge( [v0.location.x, v0.location.y], [v1.location.x, v1.location.y], bulge, ) { let mid_s = arc.sample(0.5); let (mid_wx, mid_wy, mid_wz) = to_wcs(mid_s[0], mid_s[1]); let p_mid = Point3::new(mid_wx, mid_wy, mid_wz); let tv0 = builder::vertex(p0); let tv1 = builder::vertex(p1); edges.push(builder::circle_arc(&tv0, &tv1, p_mid)); let (wcx, wcy, wcz) = to_wcs(arc.center[0], arc.center[1]); tangents.push(TangentGeom::Circle { center: [wcx as f32, wcy as f32, wcz as f32], radius: arc.radius as f32, }); } if i == 0 { key_verts.push([p0.x, p0.y, p0.z]); } key_verts.push([p1.x, p1.y, p1.z]); } TruckEntity { object: TruckObject::Contour(edges.into_iter().collect::()), snap_pts: vec![], tangent_geoms: tangents, key_vertices: key_verts, fill_tris: vec![], } } impl TruckConvertible for Polyline2D { fn to_truck(&self, _document: &acadrust::CadDocument) -> Option { Some(tessellate_polyline2d(self)) } } impl Grippable for Polyline2D { fn grips(&self) -> Vec { let elev = self.elevation; self.vertices .iter() .enumerate() .map(|(i, v)| square_grip(i, glam::DVec3::new(v.location.x, v.location.y, elev))) .collect() } fn apply_grip(&mut self, grip_id: usize, apply: GripApply) { if let Some(v) = self.vertices.get_mut(grip_id) { match apply { GripApply::Translate(d) => { v.location.x += d.x as f64; v.location.y += d.y as f64; } GripApply::Absolute(p) => { v.location.x = p.x as f64; v.location.y = p.y as f64; } } } } fn grip_menu(&self, _grip_id: usize) -> Vec { use crate::scene::model::object::{GripMenuAction, GripMenuItem}; vec![ GripMenuItem { label: "Stretch", action: GripMenuAction::Stretch, }, GripMenuItem { label: "Add Vertex", action: GripMenuAction::AddVertex, }, GripMenuItem { label: "Remove Vertex", action: GripMenuAction::RemoveVertex, }, ] } fn apply_grip_menu(&mut self, grip_id: usize, action: crate::scene::model::object::GripMenuAction) { use crate::scene::model::object::GripMenuAction as A; let n = self.vertices.len(); let elev = self.elevation; match action { A::AddVertex if grip_id < n => { let i1 = (grip_id + 1) % n; if i1 == 0 && grip_id + 1 == n { return; } let v0 = &self.vertices[grip_id]; let v1 = &self.vertices[i1]; let mx = (v0.location.x + v1.location.x) * 0.5; let my = (v0.location.y + v1.location.y) * 0.5; let mut new_v = v0.clone(); new_v.location.x = mx; new_v.location.y = my; new_v.location.z = elev; let insert_at = (grip_id + 1).min(self.vertices.len()); self.vertices.insert(insert_at, new_v); } A::RemoveVertex if grip_id < n && n > 2 => { self.vertices.remove(grip_id); } _ => {} } } } impl PropertyEditable for Polyline2D { fn geometry_properties(&self, _text_style_names: &[String]) -> Vec { let n = self.vertices.len(); let mut area = 0.0; let mut length = 0.0; let seg_count = if self.is_closed() { n } else { n.saturating_sub(1) }; for i in 0..seg_count { let a = &self.vertices[i].location; let b = &self.vertices[(i + 1) % n].location; area += a.x * b.y - b.x * a.y; length += ((b.x - a.x).powi(2) + (b.y - a.y).powi(2)).sqrt(); } area = (area * 0.5).abs(); let v0 = self.vertices.first(); let vertex_x = v0.map(|v| format!("{:.4}", v.location.x)).unwrap_or_default(); let vertex_y = v0.map(|v| format!("{:.4}", v.location.y)).unwrap_or_default(); let seg_start_w = v0.map(|v| format!("{:.4}", v.start_width)).unwrap_or_default(); let seg_end_w = v0.map(|v| format!("{:.4}", v.end_width)).unwrap_or_default(); vec![ PropSection { title: "Geometry".into(), props: vec![ ro("Current Vertex", "pl2_current_vertex", if n > 0 { "1" } else { "" }), ro("Vertex X", "pl2_vertex_x", vertex_x), ro("Vertex Y", "pl2_vertex_y", vertex_y), ro("Start segment width", "pl2_seg_start_w", seg_start_w), ro("End segment width", "pl2_seg_end_w", seg_end_w), edit("Global width", "pl2_start_w", self.start_width), edit("Elevation", "pl2_elevation", self.elevation), ro("Area", "pl2_area", format!("{area:.4}")), ro("Length", "pl2_length", format!("{length:.4}")), ], }, PropSection { title: "Misc".into(), props: vec![ Property { label: "Closed".into(), field: "pl2_closed", value: PropValue::BoolToggle { field: "pl2_closed", value: self.is_closed(), }, }, ro( "Linetype generation", "pl2_ltype_gen", if self.flags.bits() & 128 != 0 { "Enabled" } else { "Disabled" }, ), ], }, ] } fn apply_geom_prop(&mut self, field: &str, value: &str) { match field { "pl2_closed" => { let closed = if value == "toggle" { !self.is_closed() } else { value == "true" }; if closed { self.close(); } else { self.flags.set_closed(false); } } "pl2_elevation" => { if let Some(v) = parse_f64(value) { self.elevation = v; } } "pl2_start_w" => { if let Some(v) = parse_f64(value) { if v >= 0.0 { self.start_width = v; self.end_width = v; } } } _ => {} } } } impl Transformable for Polyline2D { fn apply_transform(&mut self, t: &EntityTransform) { crate::scene::view::transform::apply_standard_entity_transform(self, t, |entity, p1, p2| { for v in &mut entity.vertices { crate::scene::view::transform::reflect_xy_point( &mut v.location.x, &mut v.location.y, p1, p2, ); // Bulge encodes which side the arc bows to; a reflection // reverses it or every curved segment flips to the wrong side. v.bulge = -v.bulge; } }); } } // ── Polyline3D ──────────────────────────────────────────────────────────────── fn tessellate_polyline3d(pl: &Polyline3D) -> TruckEntity { let to_pt = |v: &acadrust::entities::Vertex3DPolyline| -> [f64; 3] { [v.position.x, v.position.y, v.position.z] }; // DXF vertex flags: 8 = spline-fit curve point, 16 = spline frame control point. // When spline-fit vertices are present use them for the wire and control points for snap; // otherwise treat all vertices uniformly. let spline_curve: Vec<_> = pl.vertices.iter().filter(|v| v.flags & 8 != 0).collect(); let ctrl_pts: Vec<_> = pl.vertices.iter().filter(|v| v.flags & 16 != 0).collect(); let (wire_pts, key_verts) = if !spline_curve.is_empty() { let wire: Vec<[f64; 3]> = spline_curve.iter().map(|v| to_pt(v)).collect(); let ctrl: Vec<[f64; 3]> = ctrl_pts.iter().map(|v| to_pt(v)).collect(); (wire, ctrl) } else { let pts: Vec<[f64; 3]> = pl.vertices.iter().map(to_pt).collect(); (pts.clone(), pts) }; let mut points = wire_pts.clone(); if pl.is_closed() && wire_pts.len() >= 2 { points.push(wire_pts[0]); } TruckEntity { object: TruckObject::Lines(points), snap_pts: vec![], tangent_geoms: vec![], key_vertices: key_verts, fill_tris: vec![], } } impl TruckConvertible for Polyline3D { fn to_truck(&self, _document: &acadrust::CadDocument) -> Option { Some(tessellate_polyline3d(self)) } } impl Grippable for Polyline3D { fn grips(&self) -> Vec { self.vertices .iter() .enumerate() .map(|(i, v)| { square_grip( i, glam::DVec3::new(v.position.x, v.position.y, v.position.z), ) }) .collect() } fn apply_grip(&mut self, grip_id: usize, apply: GripApply) { if let Some(v) = self.vertices.get_mut(grip_id) { match apply { GripApply::Translate(d) => { v.position.x += d.x as f64; v.position.y += d.y as f64; v.position.z += d.z as f64; } GripApply::Absolute(p) => { v.position.x = p.x as f64; v.position.y = p.y as f64; v.position.z = p.z as f64; } } } } fn grip_menu(&self, _grip_id: usize) -> Vec { use crate::scene::model::object::{GripMenuAction, GripMenuItem}; vec![ GripMenuItem { label: "Stretch", action: GripMenuAction::Stretch, }, GripMenuItem { label: "Add Vertex", action: GripMenuAction::AddVertex, }, GripMenuItem { label: "Remove Vertex", action: GripMenuAction::RemoveVertex, }, ] } fn apply_grip_menu(&mut self, grip_id: usize, action: crate::scene::model::object::GripMenuAction) { use crate::scene::model::object::GripMenuAction as A; let n = self.vertices.len(); match action { A::AddVertex if grip_id < n => { let i1 = (grip_id + 1) % n; if i1 == 0 && grip_id + 1 == n { return; } let v0 = &self.vertices[grip_id]; let v1 = &self.vertices[i1]; let mx = (v0.position.x + v1.position.x) * 0.5; let my = (v0.position.y + v1.position.y) * 0.5; let mz = (v0.position.z + v1.position.z) * 0.5; let mut new_v = v0.clone(); new_v.position.x = mx; new_v.position.y = my; new_v.position.z = mz; let insert_at = (grip_id + 1).min(self.vertices.len()); self.vertices.insert(insert_at, new_v); } A::RemoveVertex if grip_id < n && n > 2 => { self.vertices.remove(grip_id); } _ => {} } } } impl PropertyEditable for Polyline3D { fn geometry_properties(&self, _text_style_names: &[String]) -> Vec { use acadrust::entities::polyline3d::SmoothSurfaceType as SST; let n = self.vertices.len(); let v0 = self.vertices.first(); let vertex_x = v0.map(|v| format!("{:.4}", v.position.x)).unwrap_or_default(); let vertex_y = v0.map(|v| format!("{:.4}", v.position.y)).unwrap_or_default(); let vertex_z = v0.map(|v| format!("{:.4}", v.position.z)).unwrap_or_default(); let fit_smooth = match self.smooth_type { SST::None => "None", SST::QuadraticBSpline => "Quadratic", SST::CubicBSpline => "Cubic", SST::Bezier => "Bezier", }; vec![ PropSection { title: "Geometry".into(), props: vec![ ro("Vertex", "pl3_vertex", if n > 0 { "1" } else { "" }), ro("Vertex X", "pl3_vertex_x", vertex_x), ro("Vertex Y", "pl3_vertex_y", vertex_y), ro("Vertex Z", "pl3_vertex_z", vertex_z), ], }, PropSection { title: "Misc".into(), props: vec![ Property { label: "Closed".into(), field: "pl3_closed", value: PropValue::BoolToggle { field: "pl3_closed", value: self.is_closed(), }, }, ro("Fit/Smooth", "pl3_smooth", fit_smooth), ], }, ] } fn apply_geom_prop(&mut self, field: &str, value: &str) { if field == "pl3_closed" { let closed = if value == "toggle" { !self.is_closed() } else { value == "true" }; if closed { self.close(); } else { self.open(); } } } } impl Transformable for Polyline3D { fn apply_transform(&mut self, t: &EntityTransform) { crate::scene::view::transform::apply_standard_entity_transform(self, t, |entity, p1, p2| { for v in &mut entity.vertices { crate::scene::view::transform::reflect_xy_point( &mut v.position.x, &mut v.position.y, p1, p2, ); } }); } } /// Generate solid-fill boundary polygons for each wide segment of a Polyline2D. /// Solid-fill bands for a wide Polyline2D, plus the `world_origin` they are /// relative to (the first vertex). See `lwpolyline::wide_fills` — offsets are /// f32 from `origin` so the band stays precise at UTM-scale coordinates. pub(crate) fn wide_fills(pl: &acadrust::entities::Polyline2D) -> ([f64; 2], Vec>) { let hw_default = (pl.start_width.max(pl.end_width) / 2.0) as f32; let verts = &pl.vertices; let n = verts.len(); if n < 2 { return ([0.0; 2], vec![]); } let origin = [verts[0].location.x, verts[0].location.y]; let seg_count = if pl.is_closed() { n } else { n - 1 }; let mut out = Vec::new(); for i in 0..seg_count { let v0 = &verts[i]; let v1 = &verts[(i + 1) % n]; let hw0 = if v0.start_width > 1e-9 { v0.start_width as f32 / 2.0 } else { hw_default }; let hw1 = if v0.end_width > 1e-9 { v0.end_width as f32 / 2.0 } else { hw_default }; if hw0 < 1e-6 && hw1 < 1e-6 { continue; } let p0 = [ (v0.location.x - origin[0]) as f32, (v0.location.y - origin[1]) as f32, ]; let p1 = [ (v1.location.x - origin[0]) as f32, (v1.location.y - origin[1]) as f32, ]; if let Some(poly) = crate::entities::common::polyline_segment_fill(p0, p1, hw0, hw1, v0.bulge as f32) { out.push(poly); } } (origin, out) }