diff --git a/src/app/doc_api.rs b/src/app/doc_api.rs index 360daa23a..d749e4a6c 100644 --- a/src/app/doc_api.rs +++ b/src/app/doc_api.rs @@ -999,14 +999,6 @@ fn kernel_placement(p: &PlacementSpec) -> cadkernel::brep::Placement { } } -/// Signed volume and centroid of a closed triangle mesh via the divergence -/// theorem: V = Σ v0·(v1×v2)/6, C = Σ (v0+v1+v2)·tetra_vol / (4V). Single source -/// of truth for both (used by the cold-cache `volume`/`centroid` fallback). -fn mesh_volume_centroid(mesh: &cadkernel::brep::Mesh) -> (f64, [f64; 3]) { - // Delegate to the crate's single source of truth (divergence theorem). - ocs_doc_api::geom::mesh_volume_centroid(mesh) -} - #[cfg(test)] mod tests { use super::*; @@ -3235,7 +3227,7 @@ mod tests { let body = crate::scene::convert::solid3d_tess::kernel_body(solid) .expect("re-lift reloaded solid failed"); let mesh = cadkernel::brep::mesh_body(&body, 0.5, 1e-3); - let vol = mesh_volume_centroid(&mesh).0; + let vol = ocs_doc_api::geom::mesh_volume_centroid(&mesh).0; assert!( (vol - 125.0).abs() < 1.0, "reloaded intersection volume {vol}" diff --git a/src/entities/circle.rs b/src/entities/circle.rs index 34d579453..bc037abfb 100644 --- a/src/entities/circle.rs +++ b/src/entities/circle.rs @@ -10,7 +10,39 @@ use crate::scene::convert::acad_to_render::{extrusion_wall_tris, RenderEntity, R use crate::scene::model::object::{GripApply, GripDef, PropSection}; use crate::scene::model::wire_model::TangentGeom; -fn to_render(circle: &Circle) -> RenderEntity { +/// The number of segments to tessellate a circle into, based on its radius and the +/// current viewport's world units per pixel (`wpp`). +/// +/// Targets a maximum chord sagitta of ~0.25 pixels on screen so circles stay visually +/// round at any zoom level, while bounded between a minimum of 48 (preventing +/// coarse polygons when small or far away) and a maximum of 4096 (capping buffer size +/// on extreme zoom). +pub fn circle_segments(radius: f64, wpp: Option) -> usize { + const MIN_SEGMENTS: usize = 48; + const MAX_SEGMENTS: usize = 4096; + let Some(w) = wpp.filter(|&w| w.is_finite() && w > 0.0) else { + return MIN_SEGMENTS; + }; + let r = radius.abs(); + if !r.is_finite() || r <= 0.0 { + return MIN_SEGMENTS; + } + // Target chord sagitta of ~0.25 px on screen: + // sag = r * (1 - cos(theta / 2)) <= tolerance + let tolerance = (w as f64) * 0.25; + if r <= tolerance { + return MIN_SEGMENTS; + } + let cos_half = (1.0 - tolerance / r).clamp(-1.0, 1.0); + let step = 2.0 * cos_half.acos(); + if step <= 0.0 { + return MAX_SEGMENTS; + } + let n = (std::f64::consts::TAU / step).ceil() as usize; + n.clamp(MIN_SEGMENTS, MAX_SEGMENTS) +} + +pub fn to_render_with_wpp(circle: &Circle, wpp: Option) -> RenderEntity { let cx = circle.center.x; let cy = circle.center.y; let cz = circle.center.z; @@ -31,10 +63,12 @@ fn to_render(circle: &Circle) -> RenderEntity { radius: r, }; + let n = circle_segments(r, wpp); + if circle.thickness.abs() > 1e-10 { let t = circle.thickness; let (nx, ny, nz) = normal; - let n = 64usize; + let n = n.max(64); let tau = std::f64::consts::TAU; let circ_pt = |a: f64| -> (f64, f64, f64) { let (c, s) = (a.cos(), a.sin()); @@ -76,7 +110,8 @@ fn to_render(circle: &Circle) -> RenderEntity { } // Keep tessellation on the entity curve for large-coordinate precision. - let pts = crate::entities::curve::curve_points(&curve); + let max_angle = std::f64::consts::TAU / (n as f64); + let pts = curve.tessellate_angle(max_angle); RenderEntity { pick_tris: Vec::new(), @@ -88,6 +123,22 @@ fn to_render(circle: &Circle) -> RenderEntity { } } +fn to_render(circle: &Circle) -> RenderEntity { + to_render_with_wpp(circle, None) +} + +/// Zoom-aware render converter for EntityType::Circle. +pub fn relative_render( + entity: &acadrust::EntityType, + _document: &acadrust::CadDocument, + wpp: Option, +) -> Option { + let acadrust::EntityType::Circle(circle) = entity else { + return None; + }; + Some(to_render_with_wpp(circle, wpp)) +} + fn grips(circle: &Circle) -> Vec { let center = circle.center_wcs(); let (axis_x, axis_y) = circle.axes_wcs(); @@ -220,3 +271,94 @@ impl crate::entities::traits::MassPropsCalc for Circle { } } } + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn test_circle_segments_scales_with_zoom() { + let r = 10.0; + // Default / invalid cases return baseline floor of 48. + assert_eq!(circle_segments(r, None), 48); + assert_eq!(circle_segments(r, Some(-1.0)), 48); + assert_eq!(circle_segments(r, Some(0.0)), 48); + assert_eq!(circle_segments(r, Some(f32::NAN)), 48); + assert_eq!(circle_segments(r, Some(f32::INFINITY)), 48); + + // Zoomed far out: screen radius is tiny, capped at floor 48. + assert_eq!(circle_segments(r, Some(10.0)), 48); + assert_eq!(circle_segments(r, Some(2.0)), 48); + + // Zooming closer increases segments monotonically. + let seg_0_1 = circle_segments(r, Some(0.1)); + let seg_0_01 = circle_segments(r, Some(0.01)); + let seg_0_001 = circle_segments(r, Some(0.001)); + let seg_0_0001 = circle_segments(r, Some(0.0001)); + + assert!(seg_0_1 >= 48, "seg_0_1 was {seg_0_1}"); + assert!(seg_0_01 > seg_0_1, "seg_0_01 ({seg_0_01}) <= seg_0_1 ({seg_0_1})"); + assert!(seg_0_001 > seg_0_01, "seg_0_001 ({seg_0_001}) <= seg_0_01 ({seg_0_01})"); + assert!(seg_0_0001 > seg_0_001, "seg_0_0001 ({seg_0_0001}) <= seg_0_001 ({seg_0_001})"); + + // Extreme zoom in hits maximum cap of 4096. + assert_eq!(circle_segments(r, Some(1e-9)), 4096); + } + + #[test] + fn test_circle_segments_scales_with_radius() { + let wpp = Some(0.1); + let seg_small = circle_segments(1.0, wpp); + let seg_med = circle_segments(20.0, wpp); + let seg_large = circle_segments(100.0, wpp); + let seg_huge = circle_segments(500.0, wpp); + + assert_eq!(seg_small, 48); + assert!(seg_med > seg_small, "{seg_med} <= {seg_small}"); + assert!(seg_large > seg_med, "{seg_large} <= {seg_med}"); + assert!(seg_huge > seg_large, "{seg_huge} <= {seg_large}"); + } + + #[test] + fn test_circle_to_render_points_increase_with_zoom() { + let mut circle = Circle::default(); + circle.radius = 10.0; + + let far_render = to_render_with_wpp(&circle, Some(1.0)); + let close_render = to_render_with_wpp(&circle, Some(0.01)); + + let far_count = match far_render.object { + RenderObject::Lines(pts) => pts.len(), + _ => panic!("Expected RenderObject::Lines"), + }; + let close_count = match close_render.object { + RenderObject::Lines(pts) => pts.len(), + _ => panic!("Expected RenderObject::Lines"), + }; + + assert_eq!(far_count, 49); // 48 segments + 1 closing vertex + assert!(close_count > far_count, "close_count was {close_count} vs far_count {far_count}"); + } + + #[test] + fn test_thick_circle_to_render_points_increase_with_zoom() { + let mut circle = Circle::default(); + circle.radius = 10.0; + circle.thickness = 2.0; + + let far_render = to_render_with_wpp(&circle, Some(1.0)); + let close_render = to_render_with_wpp(&circle, Some(0.01)); + + let far_count = match far_render.object { + RenderObject::Lines(pts) => pts.len(), + _ => panic!("Expected RenderObject::Lines"), + }; + let close_count = match close_render.object { + RenderObject::Lines(pts) => pts.len(), + _ => panic!("Expected RenderObject::Lines"), + }; + + assert!(close_count > far_count, "{close_count} <= {far_count}"); + } +} + diff --git a/src/entities/dimension.rs b/src/entities/dimension.rs index a219b0de3..667990bd2 100644 --- a/src/entities/dimension.rs +++ b/src/entities/dimension.rs @@ -5408,6 +5408,21 @@ fn map_wire_ocs_to_wcs(wire: &mut WireModel, normal: Vector3) { *axis_x = map_vector(*axis_x); *axis_y = map_vector(*axis_y); } + TangentGeom::PlanarEllipse { + center, + major_axis, + normal, + .. + } => { + let mapped = map(center[0], center[1], center[2]); + let map_vector = |axis: [f64; 3]| { + let mapped = map(axis[0], axis[1], axis[2]); + [mapped.0, mapped.1, mapped.2] + }; + *center = [mapped.0, mapped.1, mapped.2]; + *major_axis = map_vector(*major_axis); + *normal = map_vector(*normal); + } } } wire.aabb = WireModel::UNBOUNDED_AABB; diff --git a/src/entities/ellipse.rs b/src/entities/ellipse.rs index 297f140b6..6530f9e38 100644 --- a/src/entities/ellipse.rs +++ b/src/entities/ellipse.rs @@ -15,6 +15,7 @@ use crate::entities::common::{ use crate::entities::traits::RenderConvertible; use crate::scene::convert::acad_to_render::{RenderEntity, RenderObject}; use crate::scene::model::object::{GripApply, GripDef, PropSection}; +use crate::scene::model::wire_model::TangentGeom; fn to_render(ell: &Ellipse) -> RenderEntity { // ELLIPSE is one of the few WCS entities in DXF: `center` (code 10) and @@ -29,6 +30,15 @@ fn to_render(ell: &Ellipse) -> RenderEntity { .map(crate::entities::curve::snap_from) .unwrap_or_default(); + let tangent = TangentGeom::PlanarEllipse { + center: [ell.center.x, ell.center.y, ell.center.z], + major_axis: [ell.major_axis.x, ell.major_axis.y, ell.major_axis.z], + normal: [ell.normal.x, ell.normal.y, ell.normal.z], + minor_axis_ratio: ell.minor_axis_ratio, + start_param: ell.start_parameter, + end_param: ell.end_parameter, + }; + // The points come from the entity's own kernel curve and angular policy. // // What does not change is the shape of the object. EXTRUDE, REVOLVE and @@ -39,7 +49,7 @@ fn to_render(ell: &Ellipse) -> RenderEntity { pick_tris: Vec::new(), object: RenderObject::Lines(Vec::new()), snap_pts: snap.snap_pts, - tangent_geoms: vec![], + tangent_geoms: vec![tangent.clone()], key_vertices: vec![], fill_tris: vec![], }; @@ -57,7 +67,7 @@ fn to_render(ell: &Ellipse) -> RenderEntity { pick_tris: Vec::new(), object, snap_pts: snap.snap_pts, - tangent_geoms: vec![], + tangent_geoms: vec![tangent], key_vertices: vec![], fill_tris: vec![], } @@ -284,16 +294,16 @@ fn apply_axis_grip(ell: &mut Ellipse, grip_id: usize, point: glam::DVec3) { fn apply_grip(ell: &mut Ellipse, grip_id: usize, apply: GripApply) { match (grip_id, apply) { (0, GripApply::Translate(d)) => { - ell.center.x += d.x as f64; - ell.center.y += d.y as f64; - ell.center.z += d.z as f64; + ell.center.x += d.x; + ell.center.y += d.y; + ell.center.z += d.z; } (0, GripApply::Absolute(p)) => { - ell.center.x = p.x as f64; - ell.center.y = p.y as f64; - ell.center.z = p.z as f64; + ell.center.x = p.x; + ell.center.y = p.y; + ell.center.z = p.z; } - (1 | 2 | 3 | 4, GripApply::Absolute(p)) => apply_axis_grip(ell, grip_id, p), + (1..=4, GripApply::Absolute(p)) => apply_axis_grip(ell, grip_id, p), (5 | 6, GripApply::Absolute(p)) => { let Some(parameter) = parameter_at_point(ell, p) else { return; diff --git a/src/scene/cache/block_cache.rs b/src/scene/cache/block_cache.rs index aa8acca81..9416faae6 100644 --- a/src/scene/cache/block_cache.rs +++ b/src/scene/cache/block_cache.rs @@ -1268,7 +1268,8 @@ fn translated_prototype_wire( } } TangentGeom::PlanarCircle { center, .. } - | TangentGeom::Arc { center, .. } => { + | TangentGeom::Arc { center, .. } + | TangentGeom::PlanarEllipse { center, .. } => { for axis in 0..3 { center[axis] += delta[axis]; } @@ -2727,6 +2728,31 @@ fn transform_tangent( radius: radius * ((sx + sy) * 0.5), }) } + TangentGeom::PlanarEllipse { + center, + major_axis, + normal, + minor_axis_ratio, + start_param, + end_param, + } => { + let c = t.apply(Vector3::new(center[0], center[1], center[2])); + let m = t.apply_rotation(Vector3::new(major_axis[0], major_axis[1], major_axis[2])); + let n = t.apply_rotation(Vector3::new(normal[0], normal[1], normal[2])); + let n_len = n.length(); + if n_len <= 1.0e-12 { + return None; + } + let n = n / n_len; + Some(TangentGeom::PlanarEllipse { + center: [c.x, c.y, c.z], + major_axis: [m.x, m.y, m.z], + normal: [n.x, n.y, n.z], + minor_axis_ratio: *minor_axis_ratio, + start_param: *start_param, + end_param: *end_param, + }) + } } } diff --git a/src/scene/convert/tessellate.rs b/src/scene/convert/tessellate.rs index 5d0044060..d933002e1 100644 --- a/src/scene/convert/tessellate.rs +++ b/src/scene/convert/tessellate.rs @@ -707,9 +707,11 @@ pub fn tessellate( } // ── Try the kernel path first ─────────────────────────────────────────── + // Zoom-adaptive circles scale their segment count from the current zoom level and radius. // Relative-PDSIZE points size their glyph from the current zoom so they // stay a roughly constant on-screen size; otherwise the header-driven path. - let te = crate::entities::point::relative_render(entity, document, world_per_pixel) + let te = crate::entities::circle::relative_render(entity, document, world_per_pixel) + .or_else(|| crate::entities::point::relative_render(entity, document, world_per_pixel)) .or_else(|| crate::entities::light::relative_render(entity, document, world_per_pixel)) .or_else(|| match entity { EntityType::Text(text) => Some(crate::entities::text::to_render_at_scale( diff --git a/src/scene/mod.rs b/src/scene/mod.rs index fb48c0cef..1aa4c7269 100644 --- a/src/scene/mod.rs +++ b/src/scene/mod.rs @@ -2232,6 +2232,7 @@ impl Scene { self.annotation_all_visible(), None, None, + None, ); let gen = WIRE_CONTENT_GEN.fetch_add(1, Ordering::Relaxed); self.last_model_wire_gen.set(gen); @@ -5161,9 +5162,9 @@ impl Scene { pub(super) fn model_tile_wires_arc( &self, _tile_idx: usize, - _cam: &Camera, + cam: &Camera, _cam_aspect: f32, - _tile_pixel_height: f32, + tile_pixel_height: f32, ) -> Arc> { // In a BEDIT block editor the resident set is the edited block's own // (block-local) entities; otherwise model space. (#261) @@ -5174,7 +5175,12 @@ impl Scene { &self.document, &self.document.header.current_annotation_scale, ); - self.resident_wires_for(block, None, scale, None, None) + let wpp = if tile_pixel_height > 0.0 { + Some((2.0 * cam.ortho_size()) / tile_pixel_height) + } else { + self.world_per_pixel() + }; + self.resident_wires_for(block, None, scale, None, None, wpp) } /// Unified static-hold wire builder — the ONE tessellation path every @@ -5223,6 +5229,7 @@ impl Scene { annotation_scale_handle: Option, frozen_layers: Option<&HashSet>, style_viewport: Option, + wpp: Option, ) -> Arc> { // Normalize an inert anno override away so distinct viewport scales // share one resident set when annotation can't change the wires. @@ -5239,6 +5246,7 @@ impl Scene { self.paper_bg_color }; let all_visible = self.annotation_all_visible(); + let quantized_wpp = Self::quantize_wpp(wpp); let key = self.resident_wire_key( block, bg, @@ -5247,6 +5255,7 @@ impl Scene { all_visible, frozen_layers, style_viewport, + quantized_wpp, ); { let sets = self.resident_wire_sets.borrow(); @@ -5271,17 +5280,18 @@ impl Scene { all_visible, frozen_layers, style_viewport, + quantized_wpp, ) { return arc; } - // Build once: full tessellation, no cull (region = None), no zoom LOD - // (wpp = None) — for every space, exactly like the Model static-hold. + // Build once: full tessellation, no cull (region = None), zoom LOD + // bounded by quantized_wpp for every space. let t_tess = iced::time::Instant::now(); let mut wires = self.wires_for_block_culled( block, None, - None, + quantized_wpp, frozen_layers, anno_scale_override, annotation_scale_handle, @@ -5327,6 +5337,9 @@ impl Scene { // spaces or re-scaling a viewport can't accumulate dead full sets. let cur_epoch = self.geometry_epoch; sets.retain(|_, set| set.epoch == cur_epoch); + if sets.len() > 16 { + sets.clear(); + } sets.insert( key, ResidentWireSet { @@ -5348,6 +5361,7 @@ impl Scene { all_visible: bool, frozen_layers: Option<&HashSet>, style_viewport: Option, + quantized_wpp: Option, ) -> u64 { let mut key: u64 = 0xcbf2_9ce4_8422_2325; let mut mix = @@ -5362,6 +5376,9 @@ impl Scene { mix(annotation_scale_handle.map(|handle| handle.value()).unwrap_or(0)); mix(all_visible as u64); mix(self.viewport_style_key(style_viewport)); + mix(quantized_wpp + .map(|w| w.to_bits() as u64) + .unwrap_or(u64::MAX)); match frozen_layers { Some(frozen) => { let mut signature = 0u64; @@ -5376,6 +5393,18 @@ impl Scene { key } + /// Quantize world units per pixel (`wpp`) into half-octave bands (~1.414x zoom ratio). + /// + /// This provides discrete, stable zoom bands so that panning (where wpp is constant) + /// and small subpixel viewport drifts or mouse-wheel motions within a band do not + /// cause re-tessellation or GPU buffer re-uploads, while zooming past a band threshold + /// smoothly scales circle tessellation detail. + pub fn quantize_wpp(wpp: Option) -> Option { + let w = wpp.filter(|&w| w.is_finite() && w > 0.0)?; + let step = (w.log2() * 2.0).floor() * 0.5; + Some(2.0f32.powf(step)) + } + /// The GPU wire-arena handoff for a viewport whose content id is `gen`: /// `(prev_gen, changed handles)` when the Model set reached `gen` via an /// incremental resident patch, else `None`. Read (not consumed) so every @@ -5462,6 +5491,7 @@ impl Scene { all_visible: bool, frozen_layers: Option<&HashSet>, style_viewport: Option, + quantized_wpp: Option, ) -> Option>> { if self.viewport_style_key(style_viewport) != 0 { return None; @@ -5545,7 +5575,7 @@ impl Scene { e, Some(&blk), None, - None, + quantized_wpp, anno_scale_override.is_some(), ); memo_updates.push((*h, Arc::new(raw.clone()))); @@ -5827,7 +5857,7 @@ impl Scene { } let layout_block = self.current_layout_block_handle(); let scale = self.paper_annotation_scale_handle(); - let base = self.resident_wires_for(layout_block, None, scale, None, None); + let base = self.resident_wires_for(layout_block, None, scale, None, None, None); let mut wires = (*base).clone(); // The overall "sheet" viewport now IS the paper view itself, so its own // border rectangle must not be drawn as an entity on the sheet. @@ -9061,7 +9091,7 @@ impl Scene { static EN: std::sync::OnceLock = std::sync::OnceLock::new(); *EN.get_or_init(|| std::env::var("OCS_NO_RESIDENT_MEMO").is_err()) } - let resident = base_ok && view_aabb.is_none() && wpp.is_none() && resident_memo_enabled(); + let resident = base_ok && view_aabb.is_none() && resident_memo_enabled(); let memo_active = base_ok && (view_aabb.is_some() || resident); // Reported in one line below. `tess_ms` / `memo_misses` are assigned by // both branches; the rest exist only on the memo path and stay zero. @@ -10200,6 +10230,7 @@ guard={:016x} guard_stale={} avp={} anno={:.4} anno_h={} all_vis={} sdf_gen={}", scale, None, None, + None, ); let wire_points = wires.iter().flat_map(|wire| wire.key_vertices.iter().copied()); let mesh_points = self.meshes.iter().filter_map(|(&handle, set)| { @@ -11213,6 +11244,7 @@ mod layout_cache_tests { scale, None, None, + None, ); let generation = s.last_model_wire_gen.get(); let bounds = s.model_space_extents().unwrap(); @@ -11226,4 +11258,40 @@ mod layout_cache_tests { assert!(!source.is_empty()); } } + + #[test] + fn circle_wires_scale_with_zoom_level() { + let mut s = Scene::new(); + let mut circle = acadrust::entities::Circle::default(); + circle.radius = 100.0; + let handle = s.add_entity(EntityType::Circle(circle)); + + // Far camera: distance is large -> wpp is large + let mut cam_far = Camera::default(); + cam_far.distance = 1000.0; + + // Close camera: distance is small -> wpp is small + let mut cam_close = Camera::default(); + cam_close.distance = 1.0; + + let wires_far = s.model_tile_wires_arc(0, &cam_far, 1.0, 1000.0); + let circle_wire_far = wires_far + .iter() + .find(|w| w.name == handle.value().to_string()) + .unwrap(); + let far_pts = circle_wire_far.points.len(); + + let wires_close = s.model_tile_wires_arc(0, &cam_close, 1.0, 1000.0); + let circle_wire_close = wires_close + .iter() + .find(|w| w.name == handle.value().to_string()) + .unwrap(); + let close_pts = circle_wire_close.points.len(); + + assert!( + close_pts > far_pts, + "Close camera points ({close_pts}) should exceed far camera points ({far_pts})" + ); + assert!(close_pts > 500, "Close points ({close_pts}) should be > 500"); + } } diff --git a/src/scene/model/wire_model.rs b/src/scene/model/wire_model.rs index a7938bfa3..1f76e7129 100644 --- a/src/scene/model/wire_model.rs +++ b/src/scene/model/wire_model.rs @@ -47,6 +47,15 @@ pub enum TangentGeom { start_angle: f64, end_angle: f64, }, + /// Ellipse or elliptical arc in an arbitrary world-space plane. + PlanarEllipse { + center: [f64; 3], + major_axis: [f64; 3], + normal: [f64; 3], + minor_axis_ratio: f64, + start_param: f64, + end_param: f64, + }, } /// Viewport-relative display transform for a point marker. diff --git a/src/scene/paper.rs b/src/scene/paper.rs index 450074122..ed6e75e38 100644 --- a/src/scene/paper.rs +++ b/src/scene/paper.rs @@ -819,7 +819,7 @@ impl Scene { fn model_wires_for_viewport( &self, vp_handle: Handle, - _screen_height_px: f32, + screen_height_px: f32, ) -> Arc> { use rustc_hash::FxHashSet as HSet; @@ -829,12 +829,17 @@ impl Scene { // wheel tick whenever the drawing contained one annotative object. // Explicit viewport annotation-scale changes still rebuild the resident // set; PSLTSCALE is a viewport GPU uniform. - let frozen = match self.document.get_entity(vp_handle) { + let (frozen, wpp) = match self.document.get_entity(vp_handle) { Some(EntityType::Viewport(vp)) => { let f: HSet = vp.frozen_layers.iter().cloned().collect(); - f + let w = if screen_height_px > 0.0 && vp.view_height > 0.0 { + Some((vp.view_height as f32) / screen_height_px) + } else { + None + }; + (f, w) } - _ => HSet::default(), + _ => (HSet::default(), None), }; let scale_handle = self.viewport_scale_handle(vp_handle); @@ -844,6 +849,7 @@ impl Scene { scale_handle, Some(&frozen), Some(vp_handle), + wpp, ) } diff --git a/src/scene/pipeline/circle_gpu.rs b/src/scene/pipeline/circle_gpu.rs new file mode 100644 index 000000000..8c32369a0 --- /dev/null +++ b/src/scene/pipeline/circle_gpu.rs @@ -0,0 +1,374 @@ +//! GPU-instanced analytical circle buffer & pipeline layout. +//! +//! Renders 2D/planar circles as analytical screen/plane-aligned quads with +//! screen-space derivative anti-aliasing. Each circle is a single instance +//! (6 vertices for a 2-triangle quad), eliminating CPU re-tessellation on zoom +//! and providing infinite visual smoothness. + +use iced::wgpu; + +#[repr(C)] +#[derive(Copy, Clone, Debug, bytemuck::Pod, bytemuck::Zeroable)] +pub struct CircleInstance { + pub center_high: [f32; 3], + pub _pad0: f32, + pub center_low: [f32; 3], + pub radius: f32, + pub axis_x: [f32; 3], + pub start_angle: f32, + pub axis_y: [f32; 3], + pub end_angle: f32, + pub color: [f32; 4], + /// `[half_width, pattern_length, draw_depth, padding]` + pub params: [f32; 4], + pub pat0: [f32; 4], + pub pat1: [f32; 4], +} + +impl CircleInstance { + pub fn layout<'a>() -> wgpu::VertexBufferLayout<'a> { + const ATTRS: &[wgpu::VertexAttribute] = &[ + wgpu::VertexAttribute { + offset: std::mem::offset_of!(CircleInstance, center_high) as u64, + shader_location: 0, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(CircleInstance, center_low) as u64, + shader_location: 1, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(CircleInstance, axis_x) as u64, + shader_location: 2, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(CircleInstance, axis_y) as u64, + shader_location: 3, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(CircleInstance, color) as u64, + shader_location: 4, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(CircleInstance, params) as u64, + shader_location: 5, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(CircleInstance, pat0) as u64, + shader_location: 6, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(CircleInstance, pat1) as u64, + shader_location: 7, + format: wgpu::VertexFormat::Float32x4, + }, + ]; + wgpu::VertexBufferLayout { + array_stride: std::mem::size_of::() as u64, + step_mode: wgpu::VertexStepMode::Instance, + attributes: ATTRS, + } + } +} + +pub struct CircleGpu { + pub instance_buffer: wgpu::Buffer, + pub instance_count: u32, +} + +impl CircleGpu { + pub fn from_instances( + device: &wgpu::Device, + queue: &wgpu::Queue, + label: &str, + instances: &[CircleInstance], + ) -> Self { + let instance_buffer = super::gpu_upload::upload_buffer( + device, + queue, + label, + instances, + wgpu::BufferUsages::VERTEX, + ); + Self { + instance_buffer, + instance_count: instances.len() as u32, + } + } +} + +/// Build the base and xray circle pipelines. +pub fn create_pipelines( + device: &wgpu::Device, + frame_bgl: &wgpu::BindGroupLayout, + color_format: wgpu::TextureFormat, + sample_count: u32, + content_stencil: &wgpu::StencilState, +) -> (wgpu::RenderPipeline, wgpu::RenderPipeline) { + let shader = device.create_shader_module(wgpu::ShaderModuleDescriptor { + label: Some("circle.wgsl"), + source: wgpu::ShaderSource::Wgsl(include_str!("../../shaders/circle.wgsl").into()), + }); + let layout = device.create_pipeline_layout(&wgpu::PipelineLayoutDescriptor { + label: Some("circle.pipeline.layout"), + bind_group_layouts: &[Some(frame_bgl)], + immediate_size: 0, + }); + let circle_pipeline = device.create_render_pipeline(&wgpu::RenderPipelineDescriptor { + label: Some("circle.pipeline"), + layout: Some(&layout), + vertex: wgpu::VertexState { + module: &shader, + entry_point: Some("vs_main"), + buffers: &[CircleInstance::layout()], + compilation_options: Default::default(), + }, + fragment: Some(wgpu::FragmentState { + module: &shader, + entry_point: Some("fs_main"), + targets: &[Some(wgpu::ColorTargetState { + format: color_format, + blend: Some(wgpu::BlendState::ALPHA_BLENDING), + write_mask: wgpu::ColorWrites::ALL, + })], + compilation_options: Default::default(), + }), + primitive: wgpu::PrimitiveState { + topology: wgpu::PrimitiveTopology::TriangleList, + ..Default::default() + }, + depth_stencil: Some(wgpu::DepthStencilState { + format: wgpu::TextureFormat::Depth24PlusStencil8, + depth_write_enabled: Some(true), + depth_compare: Some(wgpu::CompareFunction::LessEqual), + stencil: content_stencil.clone(), + bias: wgpu::DepthBiasState::default(), + }), + multisample: wgpu::MultisampleState { + count: sample_count, + mask: !0, + alpha_to_coverage_enabled: false, + }, + multiview_mask: None, + cache: None, + }); + let circle_xray_pipeline = device.create_render_pipeline(&wgpu::RenderPipelineDescriptor { + label: Some("circle_xray.pipeline"), + layout: Some(&layout), + vertex: wgpu::VertexState { + module: &shader, + entry_point: Some("vs_main"), + buffers: &[CircleInstance::layout()], + compilation_options: Default::default(), + }, + fragment: Some(wgpu::FragmentState { + module: &shader, + entry_point: Some("fs_main"), + targets: &[Some(wgpu::ColorTargetState { + format: color_format, + blend: Some(wgpu::BlendState::ALPHA_BLENDING), + write_mask: wgpu::ColorWrites::ALL, + })], + compilation_options: Default::default(), + }), + primitive: wgpu::PrimitiveState { + topology: wgpu::PrimitiveTopology::TriangleList, + ..Default::default() + }, + depth_stencil: Some(wgpu::DepthStencilState { + format: wgpu::TextureFormat::Depth24PlusStencil8, + depth_write_enabled: Some(false), + depth_compare: Some(wgpu::CompareFunction::Always), + stencil: content_stencil.clone(), + bias: wgpu::DepthBiasState::default(), + }), + multisample: wgpu::MultisampleState { + count: sample_count, + mask: !0, + alpha_to_coverage_enabled: false, + }, + multiview_mask: None, + cache: None, + }); + (circle_pipeline, circle_xray_pipeline) +} + +/// Helper to extract an analytical circle or circular arc instance from a `WireModel`. +pub fn extract_circle_instance( + wire: &crate::scene::WireModel, + draw_depth: f32, +) -> Option { + // Only thin 2D planar circles or circular arcs have exactly 1 PlanarCircle / Arc tangent geometry + // and no 3D mesh fills or text quads. + if wire.tangent_geoms.len() != 1 + || !wire.fill_tris.is_empty() + || !wire.pick_tris.is_empty() + || !wire.text_verts.is_empty() + || wire.render_instance.is_some() + { + return None; + } + let (center, axis_x, axis_y, radius, start_angle, end_angle) = match wire.tangent_geoms[0] { + crate::scene::model::wire_model::TangentGeom::PlanarCircle { + center, + axis_x, + axis_y, + radius, + } => (center, axis_x, axis_y, radius, 0.0f32, std::f32::consts::TAU), + crate::scene::model::wire_model::TangentGeom::Arc { + center, + axis_x, + axis_y, + radius, + start_angle, + end_angle, + } => ( + center, + axis_x, + axis_y, + radius, + start_angle as f32, + end_angle as f32, + ), + _ => return None, + }; + if radius <= 0.0 || !radius.is_finite() || !start_angle.is_finite() || !end_angle.is_finite() { + return None; + } + + let (ch, cl) = split_ds_xyz(center[0], center[1], center[2]); + let hw = wire.line_weight_px * 0.5; + let pat0 = [wire.pattern[0], wire.pattern[1], wire.pattern[2], wire.pattern[3]]; + let pat1 = [wire.pattern[4], wire.pattern[5], wire.pattern[6], wire.pattern[7]]; + + Some(CircleInstance { + center_high: ch, + _pad0: 0.0, + center_low: cl, + radius: radius as f32, + axis_x: [axis_x[0] as f32, axis_x[1] as f32, axis_x[2] as f32], + start_angle, + axis_y: [axis_y[0] as f32, axis_y[1] as f32, axis_y[2] as f32], + end_angle, + color: wire.color, + params: [hw, wire.pattern_length, draw_depth, 0.0], + pat0, + pat1, + }) +} + +fn split_ds_xyz(x: f64, y: f64, z: f64) -> ([f32; 3], [f32; 3]) { + let hx = x as f32; + let lx = (x - hx as f64) as f32; + let hy = y as f32; + let ly = (y - hy as f64) as f32; + let hz = z as f32; + let lz = (z - hz as f64) as f32; + ([hx, hy, hz], [lx, ly, lz]) +} + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn circle_instance_size_matches_stride() { + assert_eq!(std::mem::size_of::(), 128); + } + + #[test] + fn circle_shader_validates_with_naga() { + let source = include_str!("../../shaders/circle.wgsl"); + let module = naga::front::wgsl::parse_str(source).expect("circle.wgsl parses cleanly"); + let mut validator = naga::valid::Validator::new( + naga::valid::ValidationFlags::all(), + naga::valid::Capabilities::all(), + ); + validator + .validate(&module) + .expect("circle.wgsl validates cleanly with naga"); + } + + #[test] + fn extract_circle_instance_planar_success() { + let mut wire = crate::scene::WireModel::default(); + wire.tangent_geoms.push(crate::scene::model::wire_model::TangentGeom::PlanarCircle { + center: [100.0, 200.0, 300.0], + axis_x: [1.0, 0.0, 0.0], + axis_y: [0.0, 1.0, 0.0], + radius: 50.0, + }); + wire.line_weight_px = 2.0; + wire.color = [1.0, 0.0, 0.0, 1.0]; + + let inst = extract_circle_instance(&wire, 1.5).expect("should extract planar circle"); + assert_eq!(inst.radius, 50.0); + assert_eq!(inst.start_angle, 0.0); + assert_eq!(inst.end_angle, std::f32::consts::TAU); + assert_eq!(inst.center_high, [100.0, 200.0, 300.0]); + assert_eq!(inst.color, [1.0, 0.0, 0.0, 1.0]); + assert_eq!(inst.params[0], 1.0); // half lineweight + assert_eq!(inst.params[2], 1.5); // draw depth + } + + #[test] + fn extract_circle_instance_arc_success() { + let mut wire = crate::scene::WireModel::default(); + wire.tangent_geoms.push(crate::scene::model::wire_model::TangentGeom::Arc { + center: [10.0, 20.0, 30.0], + axis_x: [1.0, 0.0, 0.0], + axis_y: [0.0, 1.0, 0.0], + radius: 25.0, + start_angle: 0.5, + end_angle: 2.5, + }); + wire.line_weight_px = 3.0; + wire.color = [0.0, 1.0, 0.0, 1.0]; + + let inst = extract_circle_instance(&wire, 0.5).expect("should extract circular arc"); + assert_eq!(inst.radius, 25.0); + assert_eq!(inst.start_angle, 0.5); + assert_eq!(inst.end_angle, 2.5); + assert_eq!(inst.center_high, [10.0, 20.0, 30.0]); + assert_eq!(inst.color, [0.0, 1.0, 0.0, 1.0]); + assert_eq!(inst.params[0], 1.5); + } + + #[test] + fn extract_circle_instance_rejects_non_planar_or_complex() { + let mut wire = crate::scene::WireModel::default(); + wire.tangent_geoms.push(crate::scene::model::wire_model::TangentGeom::PlanarCircle { + center: [0.0, 0.0, 0.0], + axis_x: [1.0, 0.0, 0.0], + axis_y: [0.0, 1.0, 0.0], + radius: 10.0, + }); + // 1. With fill_tris (e.g. 3D solid or hatch) + wire.fill_tris = vec![[0.0; 3]]; + assert!(extract_circle_instance(&wire, 0.0).is_none()); + + // 2. Clear fill_tris, add render_instance (block instance) + wire.fill_tris.clear(); + wire.render_instance = Some(crate::scene::model::instance_model::RenderInstance { + source_id: 1, + translation: [0.0; 3], + }); + assert!(extract_circle_instance(&wire, 0.0).is_none()); + + // 3. Invalid radius + wire.render_instance = None; + wire.tangent_geoms[0] = crate::scene::model::wire_model::TangentGeom::PlanarCircle { + center: [0.0, 0.0, 0.0], + axis_x: [1.0, 0.0, 0.0], + axis_y: [0.0, 1.0, 0.0], + radius: -5.0, + }; + assert!(extract_circle_instance(&wire, 0.0).is_none()); + } +} diff --git a/src/scene/pipeline/ellipse_gpu.rs b/src/scene/pipeline/ellipse_gpu.rs new file mode 100644 index 000000000..1c6b0765c --- /dev/null +++ b/src/scene/pipeline/ellipse_gpu.rs @@ -0,0 +1,420 @@ +//! GPU-instanced analytical ellipse buffer & pipeline layout. +//! +//! Renders 2D/planar ellipses and elliptical arcs as analytical screen/plane-aligned quads +//! with sub-pixel screen-space derivative anti-aliasing. Each ellipse is a single instance +//! (6 vertices for a 2-triangle quad), eliminating CPU re-tessellation on zoom and providing +//! mathematically exact curvature at any magnification. + +use iced::wgpu; + +#[repr(C)] +#[derive(Copy, Clone, Debug, bytemuck::Pod, bytemuck::Zeroable)] +pub struct EllipseInstance { + pub center_high: [f32; 3], + pub _pad0: f32, + pub center_low: [f32; 3], + pub minor_axis_ratio: f32, + pub major_axis: [f32; 3], + pub start_param: f32, + pub normal: [f32; 3], + pub end_param: f32, + pub color: [f32; 4], + /// `[half_width, pattern_length, draw_depth, padding]` + pub params: [f32; 4], + pub pat0: [f32; 4], + pub pat1: [f32; 4], +} + +impl EllipseInstance { + pub fn layout<'a>() -> wgpu::VertexBufferLayout<'a> { + const ATTRS: &[wgpu::VertexAttribute] = &[ + wgpu::VertexAttribute { + offset: std::mem::offset_of!(EllipseInstance, center_high) as u64, + shader_location: 0, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(EllipseInstance, center_low) as u64, + shader_location: 1, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(EllipseInstance, major_axis) as u64, + shader_location: 2, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(EllipseInstance, normal) as u64, + shader_location: 3, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(EllipseInstance, color) as u64, + shader_location: 4, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(EllipseInstance, params) as u64, + shader_location: 5, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(EllipseInstance, pat0) as u64, + shader_location: 6, + format: wgpu::VertexFormat::Float32x4, + }, + wgpu::VertexAttribute { + offset: std::mem::offset_of!(EllipseInstance, pat1) as u64, + shader_location: 7, + format: wgpu::VertexFormat::Float32x4, + }, + ]; + wgpu::VertexBufferLayout { + array_stride: std::mem::size_of::() as u64, + step_mode: wgpu::VertexStepMode::Instance, + attributes: ATTRS, + } + } +} + +pub struct EllipseGpu { + pub instance_buffer: wgpu::Buffer, + pub instance_count: u32, +} + +impl EllipseGpu { + pub fn from_instances( + device: &wgpu::Device, + queue: &wgpu::Queue, + label: &str, + instances: &[EllipseInstance], + ) -> Self { + let instance_buffer = super::gpu_upload::upload_buffer( + device, + queue, + label, + instances, + wgpu::BufferUsages::VERTEX, + ); + Self { + instance_buffer, + instance_count: instances.len() as u32, + } + } +} + +/// Build the base and xray ellipse pipelines. +pub fn create_pipelines( + device: &wgpu::Device, + frame_bgl: &wgpu::BindGroupLayout, + color_format: wgpu::TextureFormat, + sample_count: u32, + content_stencil: &wgpu::StencilState, +) -> (wgpu::RenderPipeline, wgpu::RenderPipeline) { + let shader = device.create_shader_module(wgpu::ShaderModuleDescriptor { + label: Some("ellipse.wgsl"), + source: wgpu::ShaderSource::Wgsl(include_str!("../../shaders/ellipse.wgsl").into()), + }); + let layout = device.create_pipeline_layout(&wgpu::PipelineLayoutDescriptor { + label: Some("ellipse.pipeline.layout"), + bind_group_layouts: &[Some(frame_bgl)], + immediate_size: 0, + }); + let ellipse_pipeline = device.create_render_pipeline(&wgpu::RenderPipelineDescriptor { + label: Some("ellipse.pipeline"), + layout: Some(&layout), + vertex: wgpu::VertexState { + module: &shader, + entry_point: Some("vs_main"), + buffers: &[EllipseInstance::layout()], + compilation_options: Default::default(), + }, + fragment: Some(wgpu::FragmentState { + module: &shader, + entry_point: Some("fs_main"), + targets: &[Some(wgpu::ColorTargetState { + format: color_format, + blend: Some(wgpu::BlendState::ALPHA_BLENDING), + write_mask: wgpu::ColorWrites::ALL, + })], + compilation_options: Default::default(), + }), + primitive: wgpu::PrimitiveState { + topology: wgpu::PrimitiveTopology::TriangleList, + ..Default::default() + }, + depth_stencil: Some(wgpu::DepthStencilState { + format: wgpu::TextureFormat::Depth24PlusStencil8, + depth_write_enabled: Some(true), + depth_compare: Some(wgpu::CompareFunction::LessEqual), + stencil: content_stencil.clone(), + bias: wgpu::DepthBiasState::default(), + }), + multisample: wgpu::MultisampleState { + count: sample_count, + mask: !0, + alpha_to_coverage_enabled: false, + }, + multiview_mask: None, + cache: None, + }); + let ellipse_xray_pipeline = device.create_render_pipeline(&wgpu::RenderPipelineDescriptor { + label: Some("ellipse_xray.pipeline"), + layout: Some(&layout), + vertex: wgpu::VertexState { + module: &shader, + entry_point: Some("vs_main"), + buffers: &[EllipseInstance::layout()], + compilation_options: Default::default(), + }, + fragment: Some(wgpu::FragmentState { + module: &shader, + entry_point: Some("fs_main"), + targets: &[Some(wgpu::ColorTargetState { + format: color_format, + blend: Some(wgpu::BlendState::ALPHA_BLENDING), + write_mask: wgpu::ColorWrites::ALL, + })], + compilation_options: Default::default(), + }), + primitive: wgpu::PrimitiveState { + topology: wgpu::PrimitiveTopology::TriangleList, + ..Default::default() + }, + depth_stencil: Some(wgpu::DepthStencilState { + format: wgpu::TextureFormat::Depth24PlusStencil8, + depth_write_enabled: Some(false), + depth_compare: Some(wgpu::CompareFunction::Always), + stencil: content_stencil.clone(), + bias: wgpu::DepthBiasState::default(), + }), + multisample: wgpu::MultisampleState { + count: sample_count, + mask: !0, + alpha_to_coverage_enabled: false, + }, + multiview_mask: None, + cache: None, + }); + (ellipse_pipeline, ellipse_xray_pipeline) +} + +/// Helper to extract an analytical ellipse instance from a `WireModel`. +pub fn extract_ellipse_instance( + wire: &crate::scene::WireModel, + draw_depth: f32, +) -> Option { + if wire.tangent_geoms.len() != 1 + || !wire.fill_tris.is_empty() + || !wire.pick_tris.is_empty() + || !wire.text_verts.is_empty() + || wire.render_instance.is_some() + { + return None; + } + let crate::scene::model::wire_model::TangentGeom::PlanarEllipse { + center, + major_axis, + normal, + minor_axis_ratio, + start_param, + end_param, + } = wire.tangent_geoms[0] else { + return None; + }; + + let major_len_sq = major_axis[0] * major_axis[0] + major_axis[1] * major_axis[1] + major_axis[2] * major_axis[2]; + let norm_len_sq = normal[0] * normal[0] + normal[1] * normal[1] + normal[2] * normal[2]; + if major_len_sq <= 1e-12 || !major_len_sq.is_finite() + || norm_len_sq <= 1e-12 || !norm_len_sq.is_finite() + || minor_axis_ratio <= 1e-6 || !minor_axis_ratio.is_finite() + || !start_param.is_finite() || !end_param.is_finite() + { + return None; + } + + let (ch, cl) = split_ds_xyz(center[0], center[1], center[2]); + let hw = wire.line_weight_px * 0.5; + let pat0 = [wire.pattern[0], wire.pattern[1], wire.pattern[2], wire.pattern[3]]; + let pat1 = [wire.pattern[4], wire.pattern[5], wire.pattern[6], wire.pattern[7]]; + + Some(EllipseInstance { + center_high: ch, + _pad0: 0.0, + center_low: cl, + minor_axis_ratio: minor_axis_ratio.min(1.0) as f32, + major_axis: [major_axis[0] as f32, major_axis[1] as f32, major_axis[2] as f32], + start_param: start_param as f32, + normal: [normal[0] as f32, normal[1] as f32, normal[2] as f32], + end_param: end_param as f32, + color: wire.color, + params: [hw, wire.pattern_length, draw_depth, 0.0], + pat0, + pat1, + }) +} + +fn split_ds_xyz(x: f64, y: f64, z: f64) -> ([f32; 3], [f32; 3]) { + let hx = x as f32; + let lx = (x - hx as f64) as f32; + let hy = y as f32; + let ly = (y - hy as f64) as f32; + let hz = z as f32; + let lz = (z - hz as f64) as f32; + ([hx, hy, hz], [lx, ly, lz]) +} + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn ellipse_instance_size_matches_stride() { + assert_eq!(std::mem::size_of::(), 128); + } + + #[test] + fn ellipse_shader_validates_with_naga() { + let source = include_str!("../../shaders/ellipse.wgsl"); + let module = naga::front::wgsl::parse_str(source).expect("ellipse.wgsl parses cleanly"); + let mut validator = naga::valid::Validator::new( + naga::valid::ValidationFlags::all(), + naga::valid::Capabilities::all(), + ); + validator + .validate(&module) + .expect("ellipse.wgsl validates cleanly with naga"); + } + + #[test] + fn extract_ellipse_instance_success() { + let mut wire = crate::scene::WireModel::default(); + wire.tangent_geoms.push(crate::scene::model::wire_model::TangentGeom::PlanarEllipse { + center: [100.0, 200.0, 300.0], + major_axis: [50.0, 0.0, 0.0], + normal: [0.0, 0.0, 1.0], + minor_axis_ratio: 0.5, + start_param: 0.0, + end_param: std::f64::consts::TAU, + }); + wire.line_weight_px = 2.0; + wire.color = [1.0, 0.5, 0.0, 1.0]; + + let inst = extract_ellipse_instance(&wire, 1.0).expect("should extract ellipse"); + assert_eq!(inst.center_high, [100.0, 200.0, 300.0]); + assert_eq!(inst.major_axis, [50.0, 0.0, 0.0]); + assert_eq!(inst.minor_axis_ratio, 0.5); + assert_eq!(inst.start_param, 0.0); + assert_eq!(inst.end_param, std::f32::consts::TAU); + assert_eq!(inst.color, [1.0, 0.5, 0.0, 1.0]); + assert_eq!(inst.params[0], 1.0); + assert_eq!(inst.params[2], 1.0); + } + + #[test] + fn extract_elliptical_arc_instance_success() { + let mut wire = crate::scene::WireModel::default(); + wire.tangent_geoms.push(crate::scene::model::wire_model::TangentGeom::PlanarEllipse { + center: [0.0, 0.0, 0.0], + major_axis: [10.0, 0.0, 0.0], + normal: [0.0, 0.0, 1.0], + minor_axis_ratio: 0.8, + start_param: 0.5, + end_param: 2.5, + }); + wire.line_weight_px = 1.0; + wire.color = [0.0, 1.0, 0.0, 1.0]; + + let inst = extract_ellipse_instance(&wire, 0.5).expect("should extract elliptical arc"); + assert_eq!(inst.minor_axis_ratio, 0.8); + assert_eq!(inst.start_param, 0.5); + assert_eq!(inst.end_param, 2.5); + assert_eq!(inst.params[0], 0.5); // half-width + assert_eq!(inst.params[2], 0.5); // draw-depth + } + + fn rust_closest_point_ellipse(p_in: [f32; 2], ab_in: [f32; 2]) -> [f32; 2] { + let px = p_in[0].abs(); + let py = p_in[1].abs(); + let a = ab_in[0]; + let b = ab_in[1]; + if (a - b).abs() < 1e-6 { + let len = (px * px + py * py).sqrt(); + let q = if len > 1e-6 { + [px / len * a, py / len * a] + } else { + [a, 0.0] + }; + return [ + if p_in[0] >= 0.0 { q[0] } else { -q[0] }, + if p_in[1] >= 0.0 { q[1] } else { -q[1] }, + ]; + } + + // Initial estimate from eccentric anomaly of radial direction + let mut t = (py * a).atan2(px * b); + let k = a * a - b * b; + + // 3 iterations of Newton-Raphson + for _ in 0..3 { + let (st, ct) = t.sin_cos(); + let f = k * st * ct - px * a * st + py * b * ct; + let f_prime = k * (ct * ct - st * st) - px * a * ct - py * b * st; + if f_prime.abs() > 1e-7 { + t -= f / f_prime; + } + } + t = t.clamp(0.0, std::f32::consts::FRAC_PI_2); + let q = [a * t.cos(), b * t.sin()]; + [ + if p_in[0] >= 0.0 { q[0] } else { -q[0] }, + if p_in[1] >= 0.0 { q[1] } else { -q[1] }, + ] + } + + #[test] + fn closest_point_ellipse_accuracy() { + let a = 100.0f32; + let b = 40.0f32; + // Test points directly on the ellipse: closest point should be itself! + for i in 0..36 { + let angle = (i as f32) * std::f32::consts::TAU / 36.0; + let px = a * angle.cos(); + let py = b * angle.sin(); + let q = rust_closest_point_ellipse([px, py], [a, b]); + let dist = ((q[0] - px).powi(2) + (q[1] - py).powi(2)).sqrt(); + assert!( + dist < 0.2, + "Point on ellipse at angle {angle} failed: px={px}, py={py}, q={:?}, dist={dist}", + q + ); + // Verify q is on the ellipse (x/a)^2 + (y/b)^2 == 1 + let eq = (q[0] / a).powi(2) + (q[1] / b).powi(2); + assert!((eq - 1.0).abs() < 0.01, "eq={eq} for angle {angle}"); + } + + // Test points offset from ellipse (inside and outside) + for i in 0..36 { + let angle = (i as f32) * std::f32::consts::TAU / 36.0; + for offset in [-5.0f32, -1.0, 1.0, 5.0] { + let norm = [angle.cos() / a, angle.sin() / b]; + let norm_len = (norm[0] * norm[0] + norm[1] * norm[1]).sqrt(); + let un = [norm[0] / norm_len, norm[1] / norm_len]; + let px = a * angle.cos() + offset * un[0]; + let py = b * angle.sin() + offset * un[1]; + let q = rust_closest_point_ellipse([px, py], [a, b]); + let eq = (q[0] / a).powi(2) + (q[1] / b).powi(2); + assert!( + (eq - 1.0).abs() < 0.05, + "Offset point at angle {angle}, offset {offset}: q={:?}, eq={eq}", + q + ); + let measured_dist = ((q[0] - px).powi(2) + (q[1] - py).powi(2)).sqrt(); + assert!( + (measured_dist - offset.abs()).abs() < 0.2, + "Distance error at angle {angle}, offset {offset}: measured={measured_dist}" + ); + } + } + } +} diff --git a/src/scene/pipeline/mod.rs b/src/scene/pipeline/mod.rs index 908fece79..f6896df0b 100644 --- a/src/scene/pipeline/mod.rs +++ b/src/scene/pipeline/mod.rs @@ -1,6 +1,8 @@ #[cfg(test)] mod gpu_tests; mod device_capabilities; +pub mod circle_gpu; +pub mod ellipse_gpu; pub mod face3d_gpu; pub mod gpu_budget; pub mod gpu_upload; @@ -20,6 +22,8 @@ pub mod wire_gpu; use iced::wgpu; use iced::{Rectangle, Size}; +pub use circle_gpu::{CircleGpu, CircleInstance}; +pub use ellipse_gpu::{EllipseGpu, EllipseInstance}; pub use face3d_gpu::Face3DGpu; pub use wipeout_gpu::WipeoutGpu; pub use image_gpu::ImageGpu; @@ -79,6 +83,8 @@ pub struct Pipeline { shadow_pipeline: wgpu::RenderPipeline, shadow_plain_pipeline: wgpu::RenderPipeline, wire_pipeline: wgpu::RenderPipeline, + circle_pipeline: wgpu::RenderPipeline, + ellipse_pipeline: wgpu::RenderPipeline, block_wire_pipeline: wgpu::RenderPipeline, /// Stamps clip-boundary polygons into the stencil buffer (viewports + XCLIP). clip_mask_pipeline: wgpu::RenderPipeline, @@ -89,6 +95,8 @@ pub struct Pipeline { /// Same shader as wire_pipeline but depth_compare=Greater, depth_write_enabled=false. /// Used to draw ghost copies of selected wires through occluding geometry. wire_xray_pipeline: wgpu::RenderPipeline, + circle_xray_pipeline: wgpu::RenderPipeline, + ellipse_xray_pipeline: wgpu::RenderPipeline, block_wire_xray_pipeline: wgpu::RenderPipeline, /// Layout for the per-wire `WireConst` storage buffer (group 1 of the wire / /// xray pipelines). `Some` on any storage-capable device; `None` in packed @@ -225,6 +233,8 @@ pub struct Pipeline { /// replace this thin draw-range list on camera changes without touching the /// shared resident buffer. pub(crate) gpu_wires: std::sync::Arc>, + pub(crate) gpu_circles: std::sync::Arc>, + pub(crate) gpu_ellipses: std::sync::Arc>, pub(crate) gpu_block_wires: std::sync::Arc>, /// Persistent per-entity wire instance arena (capability-selected format). /// When active, `gpu_wires` is a thin wrapper over this arena's buffers and an @@ -256,6 +266,8 @@ pub struct Pipeline { clip_boundary: Option<(wgpu::Buffer, u32)>, /// Ghost copies (25% alpha) of selected wires for the X-ray depth pass. gpu_selected_wires: Vec, + gpu_selected_circles: Vec, + gpu_selected_ellipses: Vec, gpu_selected_block_wires: Vec, /// Command-preview / interim / grip-drag overlay wires. Re-uploaded every /// frame they are present (small), drawn on top of the base wire pass — so @@ -2083,6 +2095,22 @@ impl Pipeline { &content_stencil, ); + let (circle_pipeline, circle_xray_pipeline) = circle_gpu::create_pipelines( + device, + &frame_bgl, + format, + MSAA_SAMPLES, + &content_stencil, + ); + + let (ellipse_pipeline, ellipse_xray_pipeline) = ellipse_gpu::create_pipelines( + device, + &frame_bgl, + format, + MSAA_SAMPLES, + &content_stencil, + ); + let viewcube = ViewCubePipeline::new(device, queue, format); let init_size = Size::new(1, 1); @@ -2223,6 +2251,10 @@ impl Pipeline { wire_black_pipeline, block_wire_black_pipeline, wire_xray_pipeline, + circle_pipeline, + circle_xray_pipeline, + ellipse_pipeline, + ellipse_xray_pipeline, block_wire_xray_pipeline, wire_const_bgl, block_wire_const_bgl, @@ -2295,6 +2327,8 @@ impl Pipeline { blit_uniform_buffer, surface_format: format, gpu_wires: std::sync::Arc::new(vec![]), + gpu_circles: std::sync::Arc::new(vec![]), + gpu_ellipses: std::sync::Arc::new(vec![]), gpu_block_wires: std::sync::Arc::new(vec![]), wire_arena: None, wire_arena_mesh: None, @@ -2308,6 +2342,8 @@ impl Pipeline { silhouette_key: (usize::MAX, u64::MAX, [u32::MAX; 3], false), clip_boundary: None, gpu_selected_wires: vec![], + gpu_selected_circles: vec![], + gpu_selected_ellipses: vec![], gpu_selected_block_wires: vec![], gpu_preview_wires: vec![], gpu_wipeouts: vec![], @@ -2371,6 +2407,66 @@ impl Pipeline { ) } + /// Upload GPU-instanced analytical circles. + pub fn upload_circles( + &mut self, + device: &wgpu::Device, + queue: &wgpu::Queue, + wires: &[WireModel], + depth_map: &rustc_hash::FxHashMap, + ) -> Vec { + let mut instances: Vec = Vec::new(); + for wire in wires { + if !wire.display_visible { + continue; + } + let depth = wire_gpu::wire_draw_depth(wire, depth_map); + if let Some(inst) = circle_gpu::extract_circle_instance(wire, depth) { + instances.push(inst); + } + } + if instances.is_empty() { + vec![] + } else { + vec![CircleGpu::from_instances( + device, + queue, + "viewer.circles", + &instances, + )] + } + } + + /// Upload GPU-instanced analytical ellipses. + pub fn upload_ellipses( + &mut self, + device: &wgpu::Device, + queue: &wgpu::Queue, + wires: &[WireModel], + depth_map: &rustc_hash::FxHashMap, + ) -> Vec { + let mut instances: Vec = Vec::new(); + for wire in wires { + if !wire.display_visible { + continue; + } + let depth = wire_gpu::wire_draw_depth(wire, depth_map); + if let Some(inst) = ellipse_gpu::extract_ellipse_instance(wire, depth) { + instances.push(inst); + } + } + if instances.is_empty() { + vec![] + } else { + vec![EllipseGpu::from_instances( + device, + queue, + "viewer.ellipses", + &instances, + )] + } + } + /// Build resident batches and a handle index shared across viewport slots. pub fn build_wire_buffers( &mut self, @@ -2383,6 +2479,8 @@ impl Pipeline { std::sync::Arc>, std::sync::Arc>, std::sync::Arc>>, + std::sync::Arc>, + std::sync::Arc>, ) { // Batch the wire pass: instead of one GPU buffer + one draw call per // WireModel (tens of thousands on a large drawing), merge maximal runs @@ -2407,11 +2505,19 @@ impl Pipeline { i += 1; continue; } + if circle_gpu::extract_circle_instance(&wires[i], 0.0).is_some() + || ellipse_gpu::extract_ellipse_instance(&wires[i], 0.0).is_some() + { + i += 1; + continue; + } let mesh_edge = is_mesh_edge(&wires[i]); let mut j = i + 1; while j < wires.len() && wires[j].display_visible && wires[j].render_instance.is_none() + && circle_gpu::extract_circle_instance(&wires[j], 0.0).is_none() + && ellipse_gpu::extract_ellipse_instance(&wires[j], 0.0).is_none() && is_mesh_edge(&wires[j]) == mesh_edge { j += 1; @@ -2462,10 +2568,14 @@ impl Pipeline { wires.len(), ); } + let circle_batches = self.upload_circles(device, queue, wires, depth_map); + let ellipse_batches = self.upload_ellipses(device, queue, wires, depth_map); ( std::sync::Arc::new(batches), std::sync::Arc::new(block_batches), std::sync::Arc::new(index), + std::sync::Arc::new(circle_batches), + std::sync::Arc::new(ellipse_batches), ) } @@ -2492,6 +2602,8 @@ impl Pipeline { if selected.is_empty() && hovered.is_empty() && annotation_context_wires.is_empty() { self.gpu_selected_wires = vec![]; self.gpu_selected_block_wires = vec![]; + self.gpu_selected_circles = vec![]; + self.gpu_selected_ellipses = vec![]; return; } // Gather borrowed highlighted wires via the prebuilt index — @@ -2527,15 +2639,79 @@ impl Pipeline { hover_wires.push(wire); } } + let mut selected_circles: Vec = Vec::new(); + for &wire in &selected_wires { + let depth = wire_gpu::wire_draw_depth(wire, depth_map); + if let Some(mut inst) = circle_gpu::extract_circle_instance(wire, depth) { + if let Some(tint) = selected_tint { + inst.color = tint; + } + selected_circles.push(inst); + } + } + for &wire in &hover_wires { + let depth = wire_gpu::wire_draw_depth(wire, depth_map); + if let Some(mut inst) = circle_gpu::extract_circle_instance(wire, depth) { + inst.color = WireModel::HOVER; + selected_circles.push(inst); + } + } + self.gpu_selected_circles = if selected_circles.is_empty() { + vec![] + } else { + vec![CircleGpu::from_instances( + device, + queue, + "viewer.selected_circles", + &selected_circles, + )] + }; + + let mut selected_ellipses: Vec = Vec::new(); + for &wire in &selected_wires { + let depth = wire_gpu::wire_draw_depth(wire, depth_map); + if let Some(mut inst) = ellipse_gpu::extract_ellipse_instance(wire, depth) { + if let Some(tint) = selected_tint { + inst.color = tint; + } + selected_ellipses.push(inst); + } + } + for &wire in &hover_wires { + let depth = wire_gpu::wire_draw_depth(wire, depth_map); + if let Some(mut inst) = ellipse_gpu::extract_ellipse_instance(wire, depth) { + inst.color = WireModel::HOVER; + selected_ellipses.push(inst); + } + } + self.gpu_selected_ellipses = if selected_ellipses.is_empty() { + vec![] + } else { + vec![EllipseGpu::from_instances( + device, + queue, + "viewer.selected_ellipses", + &selected_ellipses, + )] + }; + let selected_regular: Vec<&WireModel> = selected_wires .iter() .copied() - .filter(|wire| wire.render_instance.is_none()) + .filter(|wire| { + wire.render_instance.is_none() + && circle_gpu::extract_circle_instance(wire, 0.0).is_none() + && ellipse_gpu::extract_ellipse_instance(wire, 0.0).is_none() + }) .collect(); let hover_regular: Vec<&WireModel> = hover_wires .iter() .copied() - .filter(|wire| wire.render_instance.is_none()) + .filter(|wire| { + wire.render_instance.is_none() + && circle_gpu::extract_circle_instance(wire, 0.0).is_none() + && ellipse_gpu::extract_ellipse_instance(wire, 0.0).is_none() + }) .collect(); let selected_blocks: Vec<&WireModel> = selected_wires .iter() @@ -4373,6 +4549,26 @@ impl Pipeline { } bind_and_draw_block_wire(&mut pass, wire); } + // Analytical GPU circles (instanced screen-space quads) + if self.gpu_circles.iter().any(|cg| cg.instance_count > 0) { + pass.set_pipeline(&self.circle_pipeline); + for cg in self.gpu_circles.iter() { + if cg.instance_count > 0 { + pass.set_vertex_buffer(0, cg.instance_buffer.slice(..)); + pass.draw(0..6, 0..cg.instance_count); + } + } + } + // Analytical GPU ellipses (instanced screen-space quads) + if self.gpu_ellipses.iter().any(|eg| eg.instance_count > 0) { + pass.set_pipeline(&self.ellipse_pipeline); + for eg in self.gpu_ellipses.iter() { + if eg.instance_count > 0 { + pass.set_vertex_buffer(0, eg.instance_buffer.slice(..)); + pass.draw(0..6, 0..eg.instance_count); + } + } + } // Live overlay wires (command preview / interim / grip drag) always // on top: the xray pipeline (depth_compare=Always, no depth write) // keeps them visible through any occluding geometry — a 3D solid, or @@ -4563,6 +4759,32 @@ impl Pipeline { bind_and_draw_block_wire(&mut pass, wire); } } + if !self.gpu_selected_circles.is_empty() { + pass.set_pipeline(if hidden_line { + &self.circle_pipeline + } else { + &self.circle_xray_pipeline + }); + for cg in &self.gpu_selected_circles { + if cg.instance_count > 0 { + pass.set_vertex_buffer(0, cg.instance_buffer.slice(..)); + pass.draw(0..6, 0..cg.instance_count); + } + } + } + if !self.gpu_selected_ellipses.is_empty() { + pass.set_pipeline(if hidden_line { + &self.ellipse_pipeline + } else { + &self.ellipse_xray_pipeline + }); + for eg in &self.gpu_selected_ellipses { + if eg.instance_count > 0 { + pass.set_vertex_buffer(0, eg.instance_buffer.slice(..)); + pass.draw(0..6, 0..eg.instance_count); + } + } + } if let Some(atlas) = &self.text_atlas_gpu { if !self.text_highlight_gpu.is_empty() { pass.set_pipeline(&self.text_highlight_pipeline); @@ -4711,6 +4933,8 @@ impl Pipeline { self.gpu_wires = std::sync::Arc::new(Vec::new()); self.gpu_block_wires = std::sync::Arc::new(Vec::new()); + self.gpu_circles = std::sync::Arc::new(Vec::new()); + self.gpu_ellipses = std::sync::Arc::new(Vec::new()); self.wire_handle_index = std::sync::Arc::new(rustc_hash::FxHashMap::default()); self.wire_arena = None; self.wire_arena_mesh = None; @@ -4721,6 +4945,8 @@ impl Pipeline { self.gpu_selected_wires.clear(); self.gpu_selected_block_wires.clear(); + self.gpu_selected_circles.clear(); + self.gpu_selected_ellipses.clear(); self.gpu_preview_wires.clear(); self.hatch_gpu.clear(); self.gpu_wipeouts.clear(); @@ -4990,6 +5216,7 @@ impl MultiPipeline { self.wire_buffer_cache.retain(|_, entry| { std::sync::Arc::strong_count(&entry.0) > 1 || std::sync::Arc::strong_count(&entry.1) > 1 + || std::sync::Arc::strong_count(&entry.3) > 1 }); self.block_geometry.retain(|_, chunks| { chunks.iter().any(|chunk| std::sync::Arc::strong_count(&chunk.bind_group) > 1) @@ -5140,6 +5367,8 @@ pub struct MultiPipeline { std::sync::Arc>, std::sync::Arc>, std::sync::Arc>>, + std::sync::Arc>, + std::sync::Arc>, ), >, } diff --git a/src/scene/pipeline/wire_arena.rs b/src/scene/pipeline/wire_arena.rs index 4fe85fab0..8af96ba2f 100644 --- a/src/scene/pipeline/wire_arena.rs +++ b/src/scene/pipeline/wire_arena.rs @@ -126,7 +126,11 @@ pub fn split_wires(wires: &[WireModel]) -> (Vec<&WireModel>, Vec<&WireModel>) { wires .iter() .filter(|wire| { - wire.display_visible && !wire.points.is_empty() && wire.render_instance.is_none() + wire.display_visible + && !wire.points.is_empty() + && wire.render_instance.is_none() + && super::circle_gpu::extract_circle_instance(wire, 0.0).is_none() + && super::ellipse_gpu::extract_ellipse_instance(wire, 0.0).is_none() }) .partition(|wire| !wire.fill_is_3d) } diff --git a/src/scene/view/render.rs b/src/scene/view/render.rs index 72fd2b92d..0c734c86f 100644 --- a/src/scene/view/render.rs +++ b/src/scene/view/render.rs @@ -853,6 +853,22 @@ impl shader::Primitive for Primitive { &mut pipeline.block_geometry, ), ); + inner.gpu_circles = std::sync::Arc::new( + inner.upload_circles( + device, + queue, + &vp_wires, + &draw_depths, + ), + ); + inner.gpu_ellipses = std::sync::Arc::new( + inner.upload_ellipses( + device, + queue, + &vp_wires, + &draw_depths, + ), + ); if _patched { wire_arena::patch_handle_index( &mut inner.wire_handle_index, @@ -911,10 +927,14 @@ impl shader::Primitive for Primitive { // Other panes retain their own references to shared geometry. inner.gpu_wires = std::sync::Arc::new(Vec::new()); inner.gpu_block_wires = std::sync::Arc::new(Vec::new()); + inner.gpu_circles = std::sync::Arc::new(Vec::new()); + inner.gpu_ellipses = std::sync::Arc::new(Vec::new()); let held_before = pipeline.wire_buffer_cache.len(); - pipeline.wire_buffer_cache.retain(|_, (w, b, _)| { + pipeline.wire_buffer_cache.retain(|_, (w, b, _, c, e)| { std::sync::Arc::strong_count(w) > 1 || std::sync::Arc::strong_count(b) > 1 + || std::sync::Arc::strong_count(c) > 1 + || std::sync::Arc::strong_count(e) > 1 }); if _perf { crate::perf_record!( @@ -959,6 +979,8 @@ impl shader::Primitive for Primitive { inner.gpu_wires = built.0; inner.gpu_block_wires = built.1; inner.wire_handle_index = built.2; + inner.gpu_circles = built.3; + inner.gpu_ellipses = built.4; } // end !arena_served inner.cached_wire_id = vp.wire_content_id; if _perf { diff --git a/src/shaders/circle.wgsl b/src/shaders/circle.wgsl new file mode 100644 index 000000000..3e88c88aa --- /dev/null +++ b/src/shaders/circle.wgsl @@ -0,0 +1,224 @@ +// GPU-instanced analytical circle shader — renders 2D/planar circles as screen/plane-aligned quads. +// Topology: TriangleList, 6 vertices drawn per INSTANCE. +// +// One instance = one complete circle. The six vertex IDs map to the corners of a +// two-triangle quad covering the circle's bounding box (+ lineweight margin). +// +// The fragment shader analytically computes the exact distance to the circle perimeter: +// d_world = abs(length(local_xy) - radius) +// and uses screen-space derivatives (fwidth) to achieve perfect sub-pixel anti-aliasing +// at any zoom level, without any polygon faceting or CPU re-tessellation. + +struct Uniforms { + viewport_size: vec2, + world_per_pixel: f32, + lwdisplay_enable: f32, + flat_shade: f32, + transparency_enable: f32, + linetype_scale: f32, + lineweight_scale: f32, + // Relative-to-eye (double-single) + view_rot: mat4x4, + eye_high: vec3, + _pad_eh: f32, + eye_low: vec3, + _pad_el: f32, +} +@group(0) @binding(0) var u: Uniforms; + +const DRAW_ORDER_BIAS: f32 = 0.001; +const MODEL_LINEWEIGHT_BOOST: f32 = 2.0; +const MODEL_LINEWEIGHT_MAX_PX: f32 = 10.0; +const TAU: f32 = 6.283185307179586; + +struct InstanceIn { + @location(0) center_high: vec4, // xyz = center_high, w = unused + @location(1) center_low: vec4, // xyz = center_low, w = radius + @location(2) axis_x: vec4, // xyz = axis_x, w = start_angle + @location(3) axis_y: vec4, // xyz = axis_y, w = end_angle + @location(4) color: vec4, // rgba + @location(5) params: vec4, // half_width_px, pattern_length, draw_depth, unused + @location(6) pat0: vec4, + @location(7) pat1: vec4, +} + +struct VertexOut { + @builtin(position) clip_pos: vec4, + @location(0) local_pos: vec2, + @location(1) radius: f32, + @location(2) hw_px: f32, + @location(3) color: vec4, + @location(4) pattern_length: f32, + @location(5) pat0: vec4, + @location(6) pat1: vec4, + @location(7) @interpolate(flat) min_elem: f32, + @location(8) @interpolate(flat) start_angle: f32, + @location(9) @interpolate(flat) end_angle: f32, +} + +fn resolve_hw(px_hw: f32) -> f32 { + var display_hw = max(px_hw * u.lineweight_scale, 0.5); + if u.lineweight_scale < 0.0 { + let scale = -u.lineweight_scale; + let base_hw = select( + min(px_hw * MODEL_LINEWEIGHT_BOOST, MODEL_LINEWEIGHT_MAX_PX * 0.5), + 0.5, + px_hw <= 0.5, + ); + display_hw = max(base_hw * scale, 0.5); + } + return select(0.5, display_hw, u.lwdisplay_enable > 0.5); +} + +@vertex fn vs_main(@builtin(vertex_index) vid: u32, in: InstanceIn) -> VertexOut { + // Two-triangle unit quad corner table: + // vid 0,1,2 = (-1,-1) ( 1,-1) ( 1, 1) + // vid 3,4,5 = (-1,-1) ( 1, 1) (-1, 1) + let u_arr = array(-1.0, 1.0, 1.0, -1.0, 1.0, -1.0); + let v_arr = array(-1.0, -1.0, 1.0, -1.0, 1.0, 1.0); + let u_val = u_arr[vid]; + let v_val = v_arr[vid]; + + let center_high = in.center_high.xyz; + let center_low = in.center_low.xyz; + let radius = in.center_low.w; + let axis_x = in.axis_x.xyz; + let axis_y = in.axis_y.xyz; + let color = in.color; + let hw_px = resolve_hw(in.params.x); + let pattern_length = in.params.y; + let draw_depth = in.params.z; + + // Expand bounding quad so anti-aliasing margin and lineweight never clip + let margin_world = (hw_px + 2.0) * max(u.world_per_pixel, 1e-6); + let ext = radius + margin_world; + + let center_rel = (center_high - u.eye_high) + (center_low - u.eye_low); + let world_pos_rel = center_rel + (u_val * axis_x + v_val * axis_y) * ext; + + var clip_pos = u.view_rot * vec4(world_pos_rel, 1.0); + clip_pos.z = clip_pos.z - draw_depth * DRAW_ORDER_BIAS * clip_pos.w; + + var out: VertexOut; + out.clip_pos = clip_pos; + out.local_pos = vec2(u_val * ext, v_val * ext); + out.radius = radius; + out.hw_px = hw_px; + out.color = color; + out.pattern_length = pattern_length; + out.pat0 = in.pat0; + out.pat1 = in.pat1; + + // Compute smallest non-zero pattern element for LOD short-circuiting + let elems = array(in.pat0.x, in.pat0.y, in.pat0.z, in.pat0.w, in.pat1.x, in.pat1.y, in.pat1.z, in.pat1.w); + var min_elem = 1e9; + for (var i = 0u; i < 8u; i++) { + let e = abs(elems[i]); + if e > 0.0 && e < min_elem { + min_elem = e; + } + } + out.min_elem = select(0.0, min_elem, min_elem < 1e8); + + out.start_angle = in.axis_x.w; + out.end_angle = in.axis_y.w; + + return out; +} + +fn mod_tau(val: f32) -> f32 { + let m = val % TAU; + return select(m, m + TAU, m < 0.0); +} + +fn in_dash(dist: f32, pat_len: f32, p0: vec4, p1: vec4) -> bool { + let elems = array(p0.x, p0.y, p0.z, p0.w, p1.x, p1.y, p1.z, p1.w); + var count = 0u; + for (var i = 0u; i < 8u; i++) { + if elems[i] != 0.0 { count = i + 1u; } + } + if count == 0u { + return true; + } + + let d = ((dist % pat_len) + pat_len) % pat_len; + var pos = 0.0f; + let dot_half = u.world_per_pixel * 0.75; + for (var i = 0u; i < count; i++) { + let elem = elems[i]; + if elem == 0.0 { + let dd = abs(d - pos); + if min(dd, pat_len - dd) <= dot_half { return true; } + } else if elem > 0.0 { + if d >= pos && d < pos + elem { return true; } + pos += elem; + } else { + pos += -elem; + } + } + return false; +} + +@fragment fn fs_main(in: VertexOut) -> @location(0) vec4 { + let r = length(in.local_pos); + + // Screen-space derivative anti-aliasing: + // Use the gradient of the monotonic radial distance `r` instead of `abs(r - radius)`. + // Taking derivatives across the cusp of abs() cancels out at scanline crossings, + // which previously collapsed fw to near-zero and punched periodic 1-2px holes (false dashes). + let grad = vec2(dpdx(r), dpdy(r)); + let grad_len = length(grad); + let fw = max(select(u.world_per_pixel, grad_len, grad_len > 1e-6), 1e-6); + + let sa = in.start_angle; + let ea = in.end_angle; + var sweep = mod_tau(ea - sa); + if sweep <= 1e-5 && abs(ea - sa) > 1e-5 { + sweep = TAU; + } + + let angle = atan2(in.local_pos.y, in.local_pos.x); + let theta = select(angle, angle + TAU, angle < 0.0); + let d_theta = mod_tau(theta - sa); + + var d_world: f32; + var arc_dist: f32; + + if sweep >= TAU - 1e-5 || d_theta <= sweep { + // Inside arc sweep or full circle + d_world = abs(r - in.radius); + arc_dist = in.radius * d_theta; + } else { + // Outside sweep: measure distance to start and end endpoints for round end caps + let p_start = in.radius * vec2(cos(sa), sin(sa)); + let p_end = in.radius * vec2(cos(ea), sin(ea)); + let d_start = length(in.local_pos - p_start); + let d_end = length(in.local_pos - p_end); + if d_start < d_end { + d_world = d_start; + arc_dist = 0.0; + } else { + d_world = d_end; + arc_dist = in.radius * sweep; + } + } + + let d_px = d_world / fw; + let alpha_cov = clamp(0.5 + in.hw_px - d_px, 0.0, 1.0); + if alpha_cov <= 0.0 { + discard; + } + + // Linetype dash pattern test + if in.pattern_length > 0.0 { + let pat_len = in.pattern_length * u.linetype_scale; + if in.min_elem * u.linetype_scale >= u.world_per_pixel && pat_len > 0.0 { + if !in_dash(arc_dist, pat_len, in.pat0, in.pat1) { + discard; + } + } + } + + let alpha = select(1.0, in.color.a, u.transparency_enable > 0.5) * alpha_cov; + return vec4(in.color.rgb, alpha); +} diff --git a/src/shaders/ellipse.wgsl b/src/shaders/ellipse.wgsl new file mode 100644 index 000000000..b6c35ba62 --- /dev/null +++ b/src/shaders/ellipse.wgsl @@ -0,0 +1,275 @@ +// GPU-instanced analytical ellipse shader — renders 2D/planar ellipses and elliptical arcs as quads. +// Topology: TriangleList, 6 vertices drawn per INSTANCE. +// +// Evaluates Inigo Quilez's exact 2D Ellipse Signed Distance Function (quartic closest-point solver) +// with sub-pixel anti-aliasing and parametric arc endpoint clamping for round end caps. + +struct Uniforms { + viewport_size: vec2, + world_per_pixel: f32, + lwdisplay_enable: f32, + flat_shade: f32, + transparency_enable: f32, + linetype_scale: f32, + lineweight_scale: f32, + // Relative-to-eye (double-single) + view_rot: mat4x4, + eye_high: vec3, + _pad_eh: f32, + eye_low: vec3, + _pad_el: f32, +} +@group(0) @binding(0) var u: Uniforms; + +const DRAW_ORDER_BIAS: f32 = 0.001; +const MODEL_LINEWEIGHT_BOOST: f32 = 2.0; +const MODEL_LINEWEIGHT_MAX_PX: f32 = 10.0; +const TAU: f32 = 6.283185307179586; + +struct InstanceIn { + @location(0) center_high: vec4, // xyz = center_high, w = unused + @location(1) center_low: vec4, // xyz = center_low, w = minor_axis_ratio + @location(2) major_axis: vec4, // xyz = major_axis (semi-major vector), w = start_param + @location(3) normal: vec4, // xyz = normal, w = end_param + @location(4) color: vec4, // rgba + @location(5) params: vec4, // half_width_px, pattern_length, draw_depth, unused + @location(6) pat0: vec4, + @location(7) pat1: vec4, +} + +struct VertexOut { + @builtin(position) clip_pos: vec4, + @location(0) local_pos: vec2, + @location(1) semi_axes: vec2, + @location(2) hw_px: f32, + @location(3) color: vec4, + @location(4) pattern_length: f32, + @location(5) pat0: vec4, + @location(6) pat1: vec4, + @location(7) @interpolate(flat) min_elem: f32, + @location(8) @interpolate(flat) start_param: f32, + @location(9) @interpolate(flat) end_param: f32, +} + +fn resolve_hw(px_hw: f32) -> f32 { + var display_hw = max(px_hw * u.lineweight_scale, 0.5); + if u.lineweight_scale < 0.0 { + let scale = -u.lineweight_scale; + let base_hw = select( + min(px_hw * MODEL_LINEWEIGHT_BOOST, MODEL_LINEWEIGHT_MAX_PX * 0.5), + 0.5, + px_hw <= 0.5, + ); + display_hw = max(base_hw * scale, 0.5); + } + return select(0.5, display_hw, u.lwdisplay_enable > 0.5); +} + +@vertex fn vs_main(@builtin(vertex_index) vid: u32, in: InstanceIn) -> VertexOut { + let u_arr = array(-1.0, 1.0, 1.0, -1.0, 1.0, -1.0); + let v_arr = array(-1.0, -1.0, 1.0, -1.0, 1.0, 1.0); + let u_val = u_arr[vid]; + let v_val = v_arr[vid]; + + let center_high = in.center_high.xyz; + let center_low = in.center_low.xyz; + let major_vec = in.major_axis.xyz; + let norm_vec = in.normal.xyz; + + let a = length(major_vec); + let u_axis = select(vec3(1.0, 0.0, 0.0), major_vec / a, a > 1e-6); + let n_len = length(norm_vec); + let n_axis = select(vec3(0.0, 0.0, 1.0), norm_vec / n_len, n_len > 1e-6); + let v_axis = normalize(cross(n_axis, u_axis)); + + let ratio = clamp(in.center_low.w, 1e-6, 1.0); + let b = a * ratio; + + let color = in.color; + let hw_px = resolve_hw(in.params.x); + let pattern_length = in.params.y; + let draw_depth = in.params.z; + + let margin_world = (hw_px + 2.0) * max(u.world_per_pixel, 1e-6); + let ext_u = a + margin_world; + let ext_v = b + margin_world; + + let center_rel = (center_high - u.eye_high) + (center_low - u.eye_low); + let world_pos_rel = center_rel + (u_val * ext_u) * u_axis + (v_val * ext_v) * v_axis; + + var clip_pos = u.view_rot * vec4(world_pos_rel, 1.0); + clip_pos.z = clip_pos.z - draw_depth * DRAW_ORDER_BIAS * clip_pos.w; + + var out: VertexOut; + out.clip_pos = clip_pos; + out.local_pos = vec2(u_val * ext_u, v_val * ext_v); + out.semi_axes = vec2(a, b); + out.hw_px = hw_px; + out.color = color; + out.pattern_length = pattern_length; + out.pat0 = in.pat0; + out.pat1 = in.pat1; + out.start_param = in.major_axis.w; + out.end_param = in.normal.w; + + let elems = array(in.pat0.x, in.pat0.y, in.pat0.z, in.pat0.w, in.pat1.x, in.pat1.y, in.pat1.z, in.pat1.w); + var min_elem = 1e9; + for (var i = 0u; i < 8u; i++) { + let e = abs(elems[i]); + if e > 0.0 && e < min_elem { + min_elem = e; + } + } + out.min_elem = select(0.0, min_elem, min_elem < 1e8); + + return out; +} + +fn mod_tau(val: f32) -> f32 { + let m = val % TAU; + return select(m, m + TAU, m < 0.0); +} + +struct ClosestResult { + q: vec2, + t: f32, +} + +fn closest_point_ellipse(p_in: vec2, ab: vec2) -> ClosestResult { + let px = abs(p_in.x); + let py = abs(p_in.y); + let a = ab.x; + let b = ab.y; + if abs(a - b) < 1e-5 { + let len = length(p_in); + let q_circ = select(p_in, normalize(p_in) * a, len > 1e-6); + let ang = atan2(q_circ.y, q_circ.x); + let full_t = select(ang, ang + TAU, ang < 0.0); + return ClosestResult(q_circ, full_t); + } + + // Initial estimate from eccentric anomaly of radial projection + var t = atan2(py * a, px * b); + let k = a * a - b * b; + + // 3 iterations of Newton-Raphson for machine precision + for (var i = 0u; i < 3u; i++) { + let st = sin(t); + let ct = cos(t); + let f = k * st * ct - px * a * st + py * b * ct; + let f_prime = k * (ct * ct - st * st) - px * a * ct - py * b * st; + if abs(f_prime) > 1e-7 { + t -= f / f_prime; + } + } + t = clamp(t, 0.0, 1.5707963267948966); + + let q_quad = vec2(a * cos(t), b * sin(t)); + let q = vec2( + select(-q_quad.x, q_quad.x, p_in.x >= 0.0), + select(-q_quad.y, q_quad.y, p_in.y >= 0.0) + ); + + var full_t = t; + if (p_in.x < 0.0) { + full_t = 3.141592653589793 - t; + } + if (p_in.y < 0.0) { + full_t = select(TAU - t, 3.141592653589793 + t, p_in.x < 0.0); + } + + return ClosestResult(q, full_t); +} + +fn in_dash(dist: f32, pat_len: f32, p0: vec4, p1: vec4) -> bool { + let elems = array(p0.x, p0.y, p0.z, p0.w, p1.x, p1.y, p1.z, p1.w); + var count = 0u; + for (var i = 0u; i < 8u; i++) { + if elems[i] != 0.0 { count = i + 1u; } + } + if count == 0u { + return true; + } + + let d = ((dist % pat_len) + pat_len) % pat_len; + var pos = 0.0f; + let dot_half = u.world_per_pixel * 0.75; + for (var i = 0u; i < count; i++) { + let elem = elems[i]; + if elem == 0.0 { + let dd = abs(d - pos); + if min(dd, pat_len - dd) <= dot_half { return true; } + } else if elem > 0.0 { + if d >= pos && d < pos + elem { return true; } + pos += elem; + } else { + pos += -elem; + } + } + return false; +} + +@fragment fn fs_main(in: VertexOut) -> @location(0) vec4 { + let ab = in.semi_axes; + let res = closest_point_ellipse(in.local_pos, ab); + let q = res.q; + let t = res.t; + + let sa = in.start_param; + let ea = in.end_param; + var sweep = mod_tau(ea - sa); + if sweep <= 1e-5 && abs(ea - sa) > 1e-5 { + sweep = TAU; + } + + let d_t = mod_tau(t - sa); + + var d_world: f32; + var arc_dist: f32; + + // Ramanujan approximation of ellipse perimeter for dash scaling + let h_val = ((ab.x - ab.y) * (ab.x - ab.y)) / max((ab.x + ab.y) * (ab.x + ab.y), 1e-6); + let perimeter = 3.141592653589793 * (ab.x + ab.y) * (1.0 + (3.0 * h_val) / max(10.0 + sqrt(max(4.0 - 3.0 * h_val, 0.0)), 1e-6)); + + if sweep >= TAU - 1e-5 || d_t <= sweep { + d_world = length(in.local_pos - q); + arc_dist = perimeter * (d_t / TAU); + } else { + let p_start = vec2(ab.x * cos(sa), ab.y * sin(sa)); + let p_end = vec2(ab.x * cos(ea), ab.y * sin(ea)); + let d_start = length(in.local_pos - p_start); + let d_end = length(in.local_pos - p_end); + if d_start < d_end { + d_world = d_start; + arc_dist = 0.0; + } else { + d_world = d_end; + arc_dist = perimeter * (sweep / TAU); + } + } + + // Isotropic quad screen-space filter width: + // Deriving from interpolated linear coordinates eliminates scanline derivative cancellations + let dx = dpdx(in.local_pos); + let dy = dpdy(in.local_pos); + let fw_len = sqrt(0.5 * (dot(dx, dx) + dot(dy, dy))); + let fw = max(select(u.world_per_pixel, fw_len, fw_len > 1e-6), 1e-6); + + let d_px = d_world / fw; + let alpha_cov = clamp(0.5 + in.hw_px - d_px, 0.0, 1.0); + if alpha_cov <= 0.0 { + discard; + } + + if in.pattern_length > 0.0 { + let pat_len = in.pattern_length * u.linetype_scale; + if in.min_elem * u.linetype_scale >= u.world_per_pixel && pat_len > 0.0 { + if !in_dash(arc_dist, pat_len, in.pat0, in.pat1) { + discard; + } + } + } + + let alpha = select(1.0, in.color.a, u.transparency_enable > 0.5) * alpha_cov; + return vec4(in.color.rgb, alpha); +} diff --git a/src/snap.rs b/src/snap.rs index 950b29615..515ee4278 100644 --- a/src/snap.rs +++ b/src/snap.rs @@ -1778,6 +1778,7 @@ impl Snapper { .fold(f32::INFINITY, f32::min); (world, edge_d2) } + TangentGeom::PlanarEllipse { .. } => continue, }; let (tier, sub) = ( snap_tier(SnapType::Tangent), @@ -1814,6 +1815,7 @@ impl Snapper { radius: *radius, } } + TangentGeom::PlanarEllipse { .. } => unreachable!(), }; best = Some(SnapResult { world: world_pt,