Skip to main content

brep_render/engine_state/
feature_dims.rs

1use brep_kernel::first_reference_name;
2use super::*;
3use super::sketch_edit_ops::is_plain_number_literal;
4use brep_gizmos::hit_region::{point_region, segment_region, HitShape};
5
6/// The world-space overlay group carrying the FD leaders + arrowheads.
7const FEATURE_DIM_OVERLAY: &str = "feature-dim-leaders";
8
9/// How clearly a dimension handle must win its own point before it is published
10/// as a widget (px, screen). See [`EngineState::dimension_handle_points`].
11const GRAB_MARGIN_PX: f64 = 2.0;
12
13/// The role a dimension-gizmo hit region plays. An `Arrow` is a value-drag handle
14/// (a linear leader or the angular arc handle, keyed by its `field_key`); `Origin`
15/// is the shared origin/center BALL (the mode toggle), keyed by the `field_key` of
16/// the FIRST dim anchored there — several dims share one ball (a cube's three
17/// leaders), so the name is a stable handle for it rather than a claim that it
18/// belongs to that one dim. Both share ONE region list so the pick + the drawn
19/// outline can never drift.
20#[derive(Clone, Debug)]
21enum DimRole {
22    Arrow(String),
23    Origin(String),
24}
25
26impl EngineState {
27    /// The armed ◎ gizmo mode: `"none"`, `"transform"`, or `"dimension"`. Drives
28    /// the ◎ highlight + the app's dimension-overlay draw / input routing.
29    pub fn gizmo_mode(&self) -> &'static str {
30        match self.transform_gizmo.mode {
31            GizmoMode::None => "none",
32            GizmoMode::Transform => "transform",
33            GizmoMode::Dimension => "dimension",
34        }
35    }
36
37    /// Whether the DIMENSION gizmo is armed for THIS feature (drives the ◎
38    /// dimension-mode highlight).
39    pub fn dimension_armed_for(&self, feature_id: &str) -> bool {
40        matches!(self.transform_gizmo.mode, GizmoMode::Dimension)
41            && self.transform_gizmo.feature_id.as_deref() == Some(feature_id)
42    }
43
44    /// The dimension-armed feature id (empty unless in dimension mode).
45    pub fn dimension_armed_feature(&self) -> String {
46        if matches!(self.transform_gizmo.mode, GizmoMode::Dimension) {
47            self.transform_gizmo.feature_id.clone().unwrap_or_default()
48        } else {
49            String::new()
50        }
51    }
52
53    /// Arm the DIMENSION gizmo for `feature_id`: hide the transform widget, show
54    /// the annotation overlay. Re-arming a different feature moves it.
55    pub fn arm_dimension(&mut self, feature_id: &str) {
56        // The widget slot is shared — an armed component Move gizmo yields.
57        self.component_move_reset();
58        self.transform_gizmo.feature_id = Some(feature_id.to_string());
59        self.transform_gizmo.mode = GizmoMode::Dimension;
60        self.transform_gizmo.drag = None;
61        // The transform widget and the dimension overlay are mutually exclusive.
62        let _ = self.widgets.set_transform_json("null");
63        self.refresh_feature_dimension_overlay();
64        self.dirty = true;
65    }
66
67    // --- The orange center-sphere ◎ TOGGLE (dimension ↔ transform) ---------
68    //
69    // A single orange sphere sits at the gizmo center in BOTH modes: the
70    // transform gizmo's `HANDLE_CENTER` sphere and the dimension arrows' shared
71    // origin sphere project to the same point. Clicking it flips the two modes,
72    // mirroring the old app's `CombinedTransformControls` center-handle toggle
73    // (pointer-down on `HANDLE_CENTER` calls
74    // `toggleDisplayMode`). The viewport routes a bare CLICK here; a DRAG on the
75    // center still free-moves via `transform_press` (unchanged).
76
77    /// Whether a screen-px pick in TRANSFORM mode lands on the orange CENTER
78    /// free-move sphere (`HANDLE_CENTER`). The viewport uses this to make a bare
79    /// click on the center TOGGLE to the dimension arrows (via
80    /// [`toggle_to_dimension`](Self::toggle_to_dimension)) instead of swallowing
81    /// it as a generic handle click. False in any other gizmo mode.
82    pub fn transform_center_pick(&self, x: f64, y: f64) -> bool {
83        matches!(self.transform_gizmo.mode, GizmoMode::Transform)
84            && self.transform_pick(x, y) == brep_gizmos::transform::HANDLE_CENTER
85    }
86
87    /// Whether a screen-px pick in DIMENSION mode lands on an orange ORIGIN
88    /// sphere of the armed feature's dimension arrows. Each distinct annotation
89    /// draws such a sphere — a LINEAR dim at its `point_a` (a cube's three axis
90    /// dims share one, a cone/pyramid draw two), an ANGULAR dim at its arc
91    /// `center` (the vertex; its sweep-END sphere is the angle DRAG handle, not a
92    /// toggle) — so every one is projected via the camera and hit-tested against
93    /// the screen-constant sphere radius. The viewport uses this to TOGGLE back to
94    /// the transform gizmo (via [`toggle_to_transform`](Self::toggle_to_transform)),
95    /// which is the ONLY way an angular-only feature (a revolve) reaches transform.
96    /// False in any other gizmo mode. The hit radius mirrors the gizmo center's own
97    /// tolerance (`PX_CENTER_RAD + 2.0`, transform.rs) so the two toggle targets match.
98    pub fn dimension_origin_pick(&self, x: f64, y: f64) -> bool {
99        // The origin sphere (LINEAR at `point_a`, ANGULAR at the arc `center`) is an
100        // `Origin`-role region. 2D-test the cursor against the SAME screen-space
101        // regions the outline draws — a point-in-circle test, so what is outlined
102        // is exactly what toggles.
103        let p = [x as f32, y as f32];
104        self.dimension_hit_regions()
105            .into_iter()
106            .any(|(role, shape)| matches!(role, DimRole::Origin(_)) && shape.contains(p))
107    }
108
109    /// Whether a screen-px pick in DIMENSION mode lands on a dimension ARROWHEAD
110    /// (a linear leader's orange cone TIP at `point_b`, or an angular arc's orange
111    /// sweep-END handle sphere). Returns the grabbed annotation's `field_key` — the
112    /// viewport routes a DRAG that starts here to [`feature_dimension_drag`](Self::feature_dimension_drag), editing that param live (Fix 4). `None` in any other
113    /// gizmo mode / when no arrowhead is under the pointer. Distinct from
114    /// [`dimension_origin_pick`](Self::dimension_origin_pick): that grabs the SHARED
115    /// origin sphere (a mode toggle), this grabs an arrowHEAD (a value edit). The
116    /// nearest arrowhead within the screen-constant hit radius wins.
117    pub fn dimension_arrow_pick(&self, x: f64, y: f64) -> Option<String> {
118        // The NEAREST `Arrow`-role region containing the cursor wins. LINEAR: the
119        // WHOLE leader CAPSULE (`point_a → point_b`) is grabbable, so a cursor
120        // anywhere on the visible shaft grabs it — even when `point_b` crosses
121        // BEHIND the eye (the reported sizeY failure), because in ortho the whole
122        // leader still projects (and in perspective the region is front-clipped to
123        // its visible part). ANGULAR: the arc sweep-END handle CIRCLE. These are
124        // the SAME screen-space regions the outline draws, so what is outlined is
125        // exactly what grabs.
126        let p = [x as f32, y as f32];
127        let mut best: Option<(f32, String)> = None;
128        for (role, shape) in self.dimension_hit_regions() {
129            if let DimRole::Arrow(key) = role {
130                let d = shape.spine_distance(p);
131                if d <= shape.radius() && best.as_ref().map(|(bd, _)| d < *bd).unwrap_or(true) {
132                    best = Some((d, key));
133                }
134            }
135        }
136        best.map(|(_, key)| key)
137    }
138
139    /// The authoritative SCREEN-space (viewport-local px) pickable regions of the
140    /// armed feature's dimension gizmo, each paired with its [`DimRole`]. The
141    /// SINGLE source `dimension_arrow_pick` (its `Arrow` regions), `dimension_origin_pick`
142    /// (its `Origin` regions), and `dimension_hit_areas_json` (draws them ALL) all
143    /// consume — so the grabbable area is exactly the drawn outline. Projection +
144    /// the perspective front-clip happen ONCE in [`brep_gizmos::hit_region`].
145    ///   * LINEAR → a leader CAPSULE (`point_a → point_b`, `ARROW_HANDLE_HIT_RAD_PX`)
146    ///     with role `Arrow` + an origin CIRCLE (`point_a`, `ORIGIN_SPHERE_RAD_PX + 2`)
147    ///     with role `Origin`.
148    ///   * ANGULAR → an arc-handle CIRCLE (`arrow_handle_point`,
149    ///     `ARROW_HANDLE_HIT_RAD_PX`) with role `Arrow` + an arc-center origin
150    ///     CIRCLE (`center`, `ORIGIN_SPHERE_RAD_PX + 2`) with role `Origin`.
151    /// Origin balls are deduped by world position (a torus's linear origin + its
152    /// angular center coincide) so they match the single drawn sphere. `[]` unless
153    /// the DIMENSION gizmo is armed for a feature.
154    fn dimension_hit_regions(&self) -> Vec<(DimRole, HitShape)> {
155        if !matches!(self.transform_gizmo.mode, GizmoMode::Dimension) {
156            return Vec::new();
157        }
158        let feature = self.dimension_armed_feature();
159        if feature.is_empty() {
160            return Vec::new();
161        }
162        let wpp = self.camera.world_per_pixel();
163        let arrow_px = crate::feature_dimensions::ARROW_HANDLE_HIT_RAD_PX as f32;
164        let origin_px = (crate::feature_dimensions::ORIGIN_SPHERE_RAD_PX + 2.0) as f32;
165        let cam = &self.camera;
166        let mut out: Vec<(DimRole, HitShape)> = Vec::new();
167        let mut origins: Vec<[f64; 3]> = Vec::new();
168        for ann in self.feature_dimension_annotations(&feature) {
169            match ann.kind {
170                crate::feature_dimensions::FeatureDimKind::Linear => {
171                    if let Some(shape) = segment_region(cam, ann.point_a, ann.point_b, arrow_px) {
172                        out.push((DimRole::Arrow(ann.field_key.clone()), shape));
173                    }
174                    push_origin_region(&mut out, &mut origins, cam, ann.point_a, origin_px, &ann.field_key);
175                }
176                crate::feature_dimensions::FeatureDimKind::Angular => {
177                    let handle = crate::feature_dimensions::arrow_handle_point(&ann, wpp);
178                    if let Some(shape) = point_region(cam, handle, arrow_px) {
179                        out.push((DimRole::Arrow(ann.field_key.clone()), shape));
180                    }
181                    push_origin_region(&mut out, &mut origins, cam, ann.center, origin_px, &ann.field_key);
182                }
183            }
184        }
185        out
186    }
187
188    /// The grabbable ARROWHEAD of every dimension the armed ◎ gizmo draws, as
189    /// `(field_key, (x, y))` in viewport-local px — what a host presses to drag
190    /// that param's value.
191    ///
192    /// The point is the arrowHEAD end of the `Arrow` region
193    /// (`dimension_hit_regions`), not its middle:
194    /// a LINEAR leader's capsule runs `point_a → point_b` and the whole shaft
195    /// grabs, but the cone TIP at `b` is the thing a reader sees and a
196    /// walkthrough should be filmed pulling; an ANGULAR dim's region is already
197    /// the sweep-end handle circle, so its center is the point.
198    ///
199    /// Published ONLY where the gizmo's own pick answers with that field, AND
200    /// still answers it `GRAB_MARGIN_PX` away in each direction — the same
201    /// rule [`transform_handle_points`](Self::transform_handle_points) follows,
202    /// with a margin the transform widget does not need.
203    ///
204    /// The margin is not caution, it is a bug this found. A pick is decided by
205    /// `spine_distance`, nearest wins and a tie goes to the first annotation, so
206    /// where two handles COINCIDE the winner turns on the last bit of an `f32`.
207    /// A full-circle torus is exactly that: `arc`'s sweep-end ball sits at 360°,
208    /// which is where the `majorRadius` leader points, so the two are the same
209    /// screen point. Publishing `dim:arc` there passed an equality test taken in
210    /// `f64` and then lost the press, because a rect goes out through JSON and
211    /// comes back a few ulps away — and the drag silently resized the torus
212    /// instead of sweeping it. Requiring the field to win by a couple of pixels
213    /// drops such a handle rather than publishing one that grabs its neighbour.
214    ///
215    /// Empty in any mode but DIMENSION.
216    pub fn dimension_handle_points(&self) -> Vec<(String, (f64, f64))> {
217        use brep_gizmos::hit_region::HitShape;
218        self.dimension_hit_regions()
219            .into_iter()
220            .filter_map(|(role, shape)| {
221                let DimRole::Arrow(key) = role else { return None };
222                let [x, y] = match shape {
223                    HitShape::Capsule { b, .. } => b,
224                    HitShape::Circle { c, .. } => c,
225                };
226                let point = (f64::from(x), f64::from(y));
227                let grabs = |dx: f64, dy: f64| {
228                    self.dimension_arrow_pick(point.0 + dx, point.1 + dy).as_deref()
229                        == Some(key.as_str())
230                };
231                const M: f64 = GRAB_MARGIN_PX;
232                let robust = grabs(0.0, 0.0)
233                    && grabs(M, 0.0)
234                    && grabs(-M, 0.0)
235                    && grabs(0.0, M)
236                    && grabs(0.0, -M);
237                robust.then_some((key, point))
238            })
239            .collect()
240    }
241
242    /// The ORIGIN BALLS the armed ◎ gizmo draws, as `(field_key, (x, y))` in
243    /// viewport-local px — the orange spheres a CLICK toggles the gizmo between
244    /// its dimension and transform forms at.
245    ///
246    /// A ball is keyed by the first dim anchored on it, because several share
247    /// one: a cube's three leaders all start at its min corner and draw a single
248    /// sphere (`radius`/`sizeX`), while a cone or a pyramid draws two — one at
249    /// the base centre and one at the top (`radiusBottom` and `radiusTop`). The
250    /// name is a stable handle for the ball, not a claim that it belongs to that
251    /// dim alone.
252    ///
253    /// Published under the same rule as the arrowheads: only where the gizmo's
254    /// own [`dimension_origin_pick`](Self::dimension_origin_pick) answers at the
255    /// point, so a ball hidden behind an arrowhead that outranks it is dropped
256    /// rather than published somewhere a click would do something else.
257    ///
258    /// Empty in any mode but DIMENSION.
259    pub fn dimension_origin_points(&self) -> Vec<(String, (f64, f64))> {
260        use brep_gizmos::hit_region::HitShape;
261        self.dimension_hit_regions()
262            .into_iter()
263            .filter_map(|(role, shape)| {
264                let DimRole::Origin(key) = role else { return None };
265                let HitShape::Circle { c, .. } = shape else { return None };
266                let point = (f64::from(c[0]), f64::from(c[1]));
267                self.dimension_origin_pick(point.0, point.1).then_some((key, point))
268            })
269            .collect()
270    }
271
272    /// DEBUG overlay: the EXACT SCREEN-space (viewport-local px) pickable regions
273    /// of the armed feature's dimension gizmo — the SAME regions
274    /// `dimension_arrow_pick` + `dimension_origin_pick` 2D-test the cursor against
275    /// (`dimension_hit_regions`) — so the red
276    /// outline can NEVER drift from the grabbable area. Each item is a
277    /// `{ kind:"capsule", a:[x,y], b:[x,y], r }` (linear leaders) or
278    /// `{ kind:"circle", c:[x,y], r }` (origin / arc-handle spheres); the app only
279    /// offsets by `rect.min`. `[]` unless the DIMENSION gizmo is armed.
280    pub fn dimension_hit_areas_json(&self) -> String {
281        let regions = self.dimension_hit_regions();
282        super::camera_widgets::hit_shapes_json(regions.iter().map(|(_, shape)| shape))
283    }
284
285    /// Toggle the armed ◎ gizmo from TRANSFORM to DIMENSION for the currently
286    /// transform-armed feature (the orange center-sphere click). No-op unless a
287    /// feature is transform-armed.
288    pub fn toggle_to_dimension(&mut self) {
289        let feature = self.transform_armed_feature();
290        if !feature.is_empty() && self.feature_dimension_annotations_json(&feature) != "[]" {
291            self.arm_dimension(&feature);
292        }
293    }
294
295    /// Toggle the armed ◎ gizmo from DIMENSION to TRANSFORM for the currently
296    /// dimension-armed feature (the orange origin-sphere click). No-op unless a
297    /// feature is dimension-armed.
298    pub fn toggle_to_transform(&mut self) {
299        let feature = self.dimension_armed_feature();
300        if feature.is_empty() {
301            return;
302        }
303        // A PLANE has NO transform — its placement is fully the orientation +
304        // `offset_distance` the offset dimension gizmo drives — so it never gets a
305        // transform gizmo: the origin-sphere toggle stays in dimension mode. (Other
306        // dim features, e.g. a revolve, still toggle to their transform gizmo.)
307        if let Some(index) = self.history.index_of(&feature) {
308            if self.history.feature_type(index).as_deref() == Some("P") {
309                return;
310            }
311        }
312        self.arm_transform(&feature);
313    }
314
315    /// The linear dimension annotations for `feature_id` (resolving expression
316    /// params against the live history env first). `[]` for a feature type with
317    /// no FD-1 builder / a missing feature.
318    fn feature_dimension_annotations(
319        &self,
320        feature_id: &str,
321    ) -> Vec<crate::feature_dimensions::FeatureDimAnnotation> {
322        let Some(index) = self.history.index_of(feature_id) else {
323            return Vec::new();
324        };
325        let Some(feature_type) = self.history.feature_type(index) else {
326            return Vec::new();
327        };
328        let Some(params) = self.history.feature_params(index) else {
329            return Vec::new();
330        };
331        let resolved = self.resolve_param_expressions(&params);
332        // Resolve any scene references the builder needs (extrude profile plane,
333        // revolve axis line) from the run report's profiles/axes — keyed off the
334        // ORIGINAL params so reference-name strings are read verbatim.
335        let refs = self.feature_dimension_refs(&feature_type, &params);
336        crate::feature_dimensions::build_annotations_with_refs(&feature_type, &resolved, &refs)
337    }
338
339    /// Resolve the scene references a feature-dimension builder needs beyond its
340    /// pure params: the extrude/revolve profile PLANE (center + normal) and the
341    /// revolve AXIS line. Sourced from the run report the engine already holds —
342    /// `sketch_profiles` (the sketch's world profile, which survives being
343    /// consumed by the extrude/revolve since only solids honor `removed`) and
344    /// `sketch_axes` (a sketch's published axis lines) — with resident-scene
345    /// fallbacks for a profile that is a solid FACE (its display-mesh plane, see
346    /// [`Self::lookup_profile_plane`]) and an axis that is a solid EDGE (its
347    /// polyline). Empty for any other feature type; missing pieces stay `None`
348    /// so the builder degrades to `[]` gracefully.
349    fn feature_dimension_refs(
350        &self,
351        feature_type: &str,
352        params: &serde_json::Value,
353    ) -> crate::feature_dimensions::ResolvedRefs {
354        let mut refs = crate::feature_dimensions::ResolvedRefs::default();
355        match feature_type {
356            "E" => {
357                if let Some((center, normal)) = self.lookup_profile_plane(params.get("profile")) {
358                    refs.profile_center = Some(center);
359                    refs.profile_normal = Some(normal);
360                }
361            }
362            "R" => {
363                if let Some((center, normal)) = self.lookup_profile_plane(params.get("profile")) {
364                    refs.profile_center = Some(center);
365                    refs.profile_normal = Some(normal);
366                }
367                if let Some((point, dir)) = self.lookup_axis_line(params.get("axis")) {
368                    refs.axis_point = Some(point);
369                    refs.axis_dir = Some(dir);
370                }
371            }
372            "P" => {
373                // The plane feature registers ONE scene frame under its own id; the
374                // offset dim hangs off that resolved plane (origin + z-axis normal).
375                if let Some(id) = params.get("id").and_then(|v| v.as_str()) {
376                    if let Some((_, frame)) =
377                        self.construction_frames.iter().find(|(name, _)| name == id)
378                    {
379                        refs.plane_origin = Some(vec3_to_arr(frame.origin));
380                        refs.plane_normal = Some(fd_normalize3(vec3_to_arr(frame.z_axis)));
381                        // Screen-constant handle stub for the offset ≈ 0 case (~48 px).
382                        refs.plane_dim_length = Some(self.camera.world_per_pixel() * 48.0);
383                    }
384                }
385            }
386            _ => {}
387        }
388        refs
389    }
390
391    /// Resolve a `profile` reference param to the world plane the extrude/revolve
392    /// gizmo hangs off, as `(center, unit normal)`. A SKETCH profile first (its
393    /// outer-loop centroid + `z_axis`), else a resident solid FACE named by the
394    /// reference (the area-weighted centroid + outward normal of its display
395    /// mesh) — the same two sources the kernel's extrude/revolve accept for
396    /// `profile` (`scene.resolve_profile`, then `resolve_face` →
397    /// `face_profile`, whose `z_axis` is likewise the OUTWARD face normal, so the
398    /// arc's `orient_revolve_axis` sign matches the kernel's sweep direction).
399    /// `None` when neither resolves — a face whose owning solid was later
400    /// consumed (a boolean target) is no longer in the scene, and the builder
401    /// then degrades to `[]` as before.
402    fn lookup_profile_plane(
403        &self,
404        profile_param: Option<&serde_json::Value>,
405    ) -> Option<([f64; 3], [f64; 3])> {
406        if let Some(profile) = self.lookup_sketch_profile(profile_param) {
407            return Some((sketch_profile_centroid(profile), vec3_to_arr(profile.z_axis)));
408        }
409        let name = first_reference_name(profile_param)?;
410        self.scene.face_plane_world(&name)
411    }
412
413    /// Resolve a `profile` reference param to the sketch profile the engine holds
414    /// (exact name, or the `:PROFILE`-suffixed form — mirrors `SceneMap::resolve_profile`).
415    fn lookup_sketch_profile(
416        &self,
417        profile_param: Option<&serde_json::Value>,
418    ) -> Option<&brep_kernel::SketchProfile> {
419        let name = first_reference_name(profile_param)?;
420        // `{sketch}:FACE` (the display-sheet pick) and `{sketch}:PROFILE` alias the
421        // base sketch id the run report keys `sketch_profiles` by: the kernel's ONE
422        // sketch-name rule, so the extrude/revolve gizmo resolves the profile the
423        // kernel does.
424        let base = brep_kernel::feature_pipeline::canonical_sketch_name(&name);
425        self.sketch_profiles
426            .iter()
427            .find(|(id, _)| id == &name || id == base)
428            .map(|(_, profile)| profile)
429    }
430
431    /// Resolve an `axis` reference param to a world line `(point, unit direction)`:
432    /// a published sketch axis first (`sketch_axes`), else a resident solid EDGE's
433    /// polyline endpoints (`scene.edge_polyline_world`). `None` if neither resolves.
434    fn lookup_axis_line(
435        &self,
436        axis_param: Option<&serde_json::Value>,
437    ) -> Option<([f64; 3], [f64; 3])> {
438        let name = first_reference_name(axis_param)?;
439        if let Some((_, axis)) = self.sketch_axes.iter().find(|(id, _)| id == &name) {
440            let dir = fd_normalize3(vec3_to_arr(axis.direction));
441            return Some((vec3_to_arr(axis.point), dir));
442        }
443        // Fallback: a resident edge used as an axis — take its polyline endpoints.
444        let poly = self.scene.edge_polyline_world(&name)?;
445        let first = *poly.first()?;
446        let last = *poly.last()?;
447        let dir = fd_sub3(last, first);
448        if fd_norm3(dir) < 1e-9 {
449            return None;
450        }
451        Some((first, fd_normalize3(dir)))
452    }
453
454    /// A copy of `params` with each top-level STRING field evaluated against the
455    /// history's `expressions` + `configurator` (the kernel `eval_expression`) and
456    /// replaced by its finite numeric result — so an expression-valued param
457    /// (e.g. `sizeX: "a + b"`) places its dimension at the resolved length.
458    /// Non-numeric strings (ids, enum options) fail to eval and stay verbatim.
459    fn resolve_param_expressions(&self, params: &serde_json::Value) -> serde_json::Value {
460        let expressions = self.history.expressions();
461        let configurator = self.history.configurator();
462        let mut out = params.clone();
463        if let Some(object) = out.as_object_mut() {
464            for value in object.values_mut() {
465                if let Some(source) = value.as_str() {
466                    if let Ok(number) =
467                        brep_kernel::eval_expression(&expressions, &configurator, source)
468                    {
469                        if number.is_finite() {
470                            *value = serde_json::json!(number);
471                        }
472                    }
473                }
474            }
475        }
476        out
477    }
478
479    /// The dimension annotations for `feature_id` as JSON:
480    /// `[{ fieldKey, pointA, pointB, value, label, mid }]` (world-space points;
481    /// `mid` is the leader midpoint the app anchors the label at). `[]` when the
482    /// feature type has no FD-1 builder.
483    pub fn feature_dimension_annotations_json(&self, feature_id: &str) -> String {
484        use crate::feature_dimensions::FeatureDimKind;
485        let annotations = self.feature_dimension_annotations(feature_id);
486        let wpp = self.camera.world_per_pixel();
487        let out: Vec<serde_json::Value> = annotations
488            .iter()
489            .map(|a| {
490                // The chip anchor: a linear leader's midpoint, or an angular arc's
491                // mid-sweep point at the screen-constant radius (camera-dependent,
492                // so computed here with the live `world_per_pixel`). `kind` lets the
493                // app format the chip (`A 234°` for an angular value in DEGREES).
494                let (kind, mid) = match a.kind {
495                    FeatureDimKind::Linear => ("linear", a.midpoint()),
496                    FeatureDimKind::Angular => {
497                        ("angular", crate::feature_dimensions::angular_chip_anchor(a, wpp))
498                    }
499                };
500                serde_json::json!({
501                    "fieldKey": a.field_key,
502                    "pointA": a.point_a,
503                    "pointB": a.point_b,
504                    "value": a.value,
505                    "label": a.label,
506                    "mid": mid,
507                    "kind": kind,
508                })
509            })
510            .collect();
511        serde_json::to_string(&out).unwrap_or_else(|_| "[]".to_string())
512    }
513
514    /// The `{ mode, feature, annotations }` snapshot the headless verifier reads
515    /// (published as `__brepFeatureDim`).
516    pub fn feature_dimension_state_json(&self) -> String {
517        let feature = self.dimension_armed_feature();
518        let annotations: serde_json::Value = if feature.is_empty() {
519            serde_json::json!([])
520        } else {
521            serde_json::from_str(&self.feature_dimension_annotations_json(&feature))
522                .unwrap_or_else(|_| serde_json::json!([]))
523        };
524        serde_json::json!({
525            "mode": self.gizmo_mode(),
526            "feature": feature,
527            "annotations": annotations,
528        })
529        .to_string()
530    }
531
532    /// The `set_overlay` JSON for the `feature-dim-leaders` group — the annotation
533    /// leaders + arrowheads for the dimension-armed feature (empty when not in
534    /// dimension mode, so a stale group is cleared).
535    fn feature_dimension_overlay_json(&self) -> String {
536        let feature = self.dimension_armed_feature();
537        let annotations = if feature.is_empty() {
538            Vec::new()
539        } else {
540            self.feature_dimension_annotations(&feature)
541        };
542        let (positions, colors) = crate::feature_dimensions::leaders_buffers(
543            &annotations,
544            self.camera.world_per_pixel(),
545        );
546        serde_json::json!({
547            "groups": [
548                {
549                    "name": FEATURE_DIM_OVERLAY,
550                    "renderOrder": 10003,
551                    "tris": { "positions": positions, "colors": colors },
552                }
553            ]
554        })
555        .to_string()
556    }
557
558    /// (Re)project the dimension leaders onto the current geometry. Called on arm
559    /// + after every param change (drag / value edit / rerun in dimension mode)
560    /// + on a material ZOOM (`Self::ensure_feature_dimension_overlay_current`).
561    /// Remembers the `world_per_pixel` it baked at, which is what lets that
562    /// per-frame ensure fire on change ONLY.
563    pub fn refresh_feature_dimension_overlay(&mut self) {
564        let json = self.feature_dimension_overlay_json();
565        let _ = self.set_overlay_json(&json);
566        let wpp = self.camera.world_per_pixel();
567        self.feature_dim_overlay_wpp = if wpp > 0.0 { wpp } else { f64::MIN_POSITIVE };
568    }
569
570    /// Clear the dimension overlay (an empty group), e.g. when disarming or
571    /// switching to transform mode.
572    pub(super) fn clear_feature_dimension_overlay(&mut self) {
573        let _ = self.set_overlay_json(&serde_json::json!({
574            "groups": [ { "name": FEATURE_DIM_OVERLAY } ]
575        }).to_string());
576        self.feature_dim_overlay_wpp = 0.0;
577    }
578
579    /// Per-frame upkeep for the DIMENSION gizmo (driven by
580    /// [`Self::ensure_overlays_current`]).
581    ///
582    /// The group is baked into pre-expanded vertices at feed time, and its
583    /// rod/cone/origin-sphere sizing — plus the angular arc's entire world
584    /// RADIUS (`ANGLE_ARC_RAD_PX × world_per_pixel`) — is screen-constant. So a
585    /// zoom that is not followed by a re-bake leaves the handles drawn at the old
586    /// pixel size, and the angular sweep handle drawn at the old world position
587    /// while [`Self::dimension_hit_regions`] (which projects against the LIVE
588    /// camera) grabs at the new one: the outline and the drawn handle drift apart.
589    ///
590    /// Re-bakes ONLY on a material `world_per_pixel` change
591    /// ([`overlay_wpp_stale`](super::overlay_wpp_stale)) — a quiet frame does not
592    /// touch the overlay, so there is no per-frame dirty loop. Nothing armed →
593    /// nothing baked, and the remembered zoom is dropped so re-arming re-bakes.
594    pub(super) fn ensure_feature_dimension_overlay_current(&mut self) {
595        if !matches!(self.transform_gizmo.mode, GizmoMode::Dimension)
596            || self.dimension_armed_feature().is_empty()
597        {
598            self.feature_dim_overlay_wpp = 0.0;
599            return;
600        }
601        let wpp = self.camera.world_per_pixel();
602        if super::overlay_wpp_stale(self.feature_dim_overlay_wpp, wpp) {
603            self.refresh_feature_dimension_overlay();
604        }
605    }
606
607    /// Finalize a dimension drag with a fresh run of the committed parameters.
608    /// Abandon background preview work first so releasing the arrow does not wait
609    /// behind an expensive intermediate value. The final run forces this feature
610    /// and its consumers to rebuild, rather than replaying the cached handle.
611    pub fn feature_dimension_release(&mut self, feature_id: &str) {
612        if self.history.index_of(feature_id).is_none() {
613            return;
614        }
615        self.cancel_run_with_notice(false);
616        self.rerun_history_forcing(Some(feature_id));
617    }
618
619    /// Drag a dimension handle: project the pointer pixel `(x, y)` onto the
620    /// annotation's world axis (`pointA → pointB`), take the distance along the
621    /// axis from `pointA` as the new value (correcting for any transform scale so
622    /// the PARAM — not the scaled world length — is what changes), set the param,
623    /// and re-run the history live. Degenerate projections (parallel ray / zero
624    /// axis) no-op.
625    pub fn feature_dimension_drag(&mut self, feature_id: &str, field_key: &str, x: f64, y: f64) {
626        let annotations = self.feature_dimension_annotations(feature_id);
627        let Some(annotation) = annotations.iter().find(|a| a.field_key == field_key) else {
628            return;
629        };
630        if annotation.kind == crate::feature_dimensions::FeatureDimKind::Angular {
631            if let Some(degrees) = self.angular_drag_degrees(annotation, x, y) {
632                self.write_feature_dimension_param(feature_id, field_key, serde_json::json!(degrees));
633                // Live-follow: re-bake the world-space leaders onto the rebuilt
634                // geometry so the arc tracks the pointer this frame (Fix 3).
635                self.refresh_feature_dimension_overlay();
636            }
637            return;
638        }
639        let a = annotation.point_a;
640        let b = annotation.point_b;
641        let axis = fd_sub3(b, a);
642        let len = fd_norm3(axis);
643        if len < 1e-9 {
644            return;
645        }
646        let dir = [axis[0] / len, axis[1] / len, axis[2] / len];
647        let ray = self.camera.pick_ray(x, y);
648        let ray_dir = fd_normalize3(ray.dir);
649        let Some(t_world) = closest_t_on_axis(a, dir, ray.origin, ray_dir) else {
650            return;
651        };
652        // World distance → param value: correct for the local axis scale via the
653        // CURRENT ratio (world length / current param). Under unit scale this is
654        // the identity; when the param is ~0 there is no ratio, so use the world
655        // distance directly (unit-scale assumption).
656        let scale_recip = if annotation.value.abs() > 1e-9 && len > 1e-9 {
657            annotation.value / len
658        } else {
659            1.0
660        };
661        // Preserve SIGN so a linear dim can be dragged through the origin to the
662        // negative side (a directional dim — cube size, height — then extends the
663        // other way; the kernel takes |value| for magnitude dims). A small dead-zone
664        // keeps it off an exact 0 (a degenerate extent the builders reject).
665        let raw = t_world * scale_recip;
666        let new_value = if raw >= 0.0 { raw.max(1e-4) } else { raw.min(-1e-4) };
667        self.write_feature_dimension_param(feature_id, field_key, serde_json::json!(new_value));
668        // Live-follow: re-bake the world-space leaders onto the rebuilt geometry so
669        // the arrow tracks the pointer this frame (Fix 3).
670        self.refresh_feature_dimension_overlay();
671    }
672
673    /// Map a pointer pixel to a swept angle (DEGREES) for an ANGULAR annotation:
674    /// search the sweep for the degree whose arc-end projects nearest the pointer
675    /// (a coarse 2° pass, then a ±2° refine at 0.25°), snap to 1°, clamp to
676    /// `[-360, 360]`. Ported from the overlay `angle` drag. The magnitude is
677    /// floored off exactly 0 so a torus `arc` drag never lands on 0 — which the
678    /// kernel's `|| 360` falsy fallback would flip to a FULL torus mid-drag.
679    /// `None` if the arc never projects in front of the camera.
680    ///
681    /// `pub(super)`: the assembly-constraint angle-arc drag
682    /// (`assembly_overlay.rs`) maps its pointer through this SAME search so the
683    /// two angle gizmos share one drag feel.
684    pub(super) fn angular_drag_degrees(
685        &self,
686        ann: &crate::feature_dimensions::FeatureDimAnnotation,
687        x: f64,
688        y: f64,
689    ) -> Option<f64> {
690        let radius = crate::feature_dimensions::ANGLE_ARC_RAD_PX * self.camera.world_per_pixel();
691        if radius <= 1e-9 {
692            return None;
693        }
694        // A sweep of `deg` and `deg - 360` share the SAME arc-end world point, so
695        // the screen-nearest search alone can't tell them apart at the wrap. Break
696        // the tie toward the CURRENT value (angle-unwrap
697        // continuity) with a tiny bias `~1e-6·Δ°²` — decisive only when screen
698        // errors are essentially equal, negligible against any real pointer move.
699        let current = ann.value;
700        let combined = |screen_err: f64, deg: f64| -> f64 {
701            let d = deg - current;
702            screen_err + 1e-6 * d * d
703        };
704        let mut best_deg = current;
705        let mut best_err = f64::INFINITY;
706        // Coarse sweep over the full range.
707        let mut deg = -360.0;
708        while deg <= 360.0 {
709            if let Some(err) = self.angle_arc_end_err(ann, radius, deg, x, y) {
710                let err = combined(err, deg);
711                if err < best_err {
712                    best_err = err;
713                    best_deg = deg;
714                }
715            }
716            deg += 2.0;
717        }
718        if !best_err.is_finite() {
719            return None;
720        }
721        // Refine around the coarse best.
722        let center = best_deg;
723        let mut deg = center - 2.0;
724        while deg <= center + 2.0 {
725            if (-360.0..=360.0).contains(&deg) {
726                if let Some(err) = self.angle_arc_end_err(ann, radius, deg, x, y) {
727                    let err = combined(err, deg);
728                    if err < best_err {
729                        best_err = err;
730                        best_deg = deg;
731                    }
732                }
733            }
734            deg += 0.25;
735        }
736        let clamped = best_deg.round().clamp(-360.0, 360.0);
737        let floored = if clamped.abs() < 0.1 {
738            if clamped < 0.0 { -0.1 } else { 0.1 }
739        } else {
740            clamped
741        };
742        Some(floored)
743    }
744
745    /// Squared screen-pixel distance from `(x, y)` to the arc-end at `deg` for an
746    /// angular annotation (`center + rotate(ref_dir, axis, deg) * radius`), or
747    /// `None` when that point is behind the camera.
748    fn angle_arc_end_err(
749        &self,
750        ann: &crate::feature_dimensions::FeatureDimAnnotation,
751        radius: f64,
752        deg: f64,
753        x: f64,
754        y: f64,
755    ) -> Option<f64> {
756        let dir = fd_normalize3(crate::feature_dimensions::rotate_about_axis(
757            ann.ref_dir,
758            ann.axis,
759            deg.to_radians(),
760        ));
761        let p = [
762            ann.center[0] + dir[0] * radius,
763            ann.center[1] + dir[1] * radius,
764            ann.center[2] + dir[2] * radius,
765        ];
766        let (sx, sy, depth) = self.camera.project(p);
767        if depth <= 0.0 {
768            return None;
769        }
770        Some((sx - x) * (sx - x) + (sy - y) * (sy - y))
771    }
772
773    /// Edit a dimension value from a label field: a plain numeric literal sets the
774    /// param to that number; otherwise the input is treated as an EXPRESSION —
775    /// evaluated LIVE against the history's `expressions` + `configurator` (the
776    /// kernel `eval_expression`) and, on success, STORED as the expression string
777    /// (the kernel re-evaluates it via `ctx.number`, so it stays live). A blank /
778    /// bad-expression input no-ops (never corrupts the feature). Re-runs live.
779    pub fn feature_dimension_set_value(&mut self, feature_id: &str, field_key: &str, input: &str) {
780        let trimmed = input.trim();
781        if trimmed.is_empty() {
782            return;
783        }
784        if is_plain_number_literal(trimmed) {
785            let Ok(number) = trimmed.parse::<f64>() else {
786                return;
787            };
788            if !number.is_finite() {
789                return;
790            }
791            self.write_feature_dimension_param(feature_id, field_key, serde_json::json!(number));
792        } else {
793            // Validate the expression before storing it (a bad expression no-ops).
794            let expressions = self.history.expressions();
795            let configurator = self.history.configurator();
796            match brep_kernel::eval_expression(&expressions, &configurator, trimmed) {
797                Ok(number) if number.is_finite() => {
798                    self.write_feature_dimension_param(
799                        feature_id,
800                        field_key,
801                        serde_json::Value::String(trimmed.to_string()),
802                    );
803                }
804                _ => {}
805            }
806        }
807    }
808
809    /// Set one `inputParams` field of `feature_id` (a number or an expression
810    /// string) and re-run the history (which re-projects the leaders in dimension
811    /// mode). No-op when the feature is absent.
812    fn write_feature_dimension_param(
813        &mut self,
814        feature_id: &str,
815        field_key: &str,
816        value: serde_json::Value,
817    ) {
818        let Some(index) = self.history.index_of(feature_id) else {
819            return;
820        };
821        let mut params = self
822            .history
823            .feature_params(index)
824            .unwrap_or_else(|| serde_json::json!({}));
825        let Some(object) = params.as_object_mut() else {
826            return;
827        };
828        // A drag frame whose pointer has not moved (egui reports `dragged()`
829        // on every frame the button is held, and the frame loop keeps running
830        // while a run is in flight) resolves to the value already written. No
831        // run for it: each submit is a whole-history rebuild on the runner.
832        if object.get(field_key) == Some(&value) {
833            return;
834        }
835        object.insert(field_key.to_string(), value);
836        let _ = self.update_feature_params(feature_id, &params.to_string());
837    }
838}
839
840// --- FD-1 geometry helpers (self-contained, `fd_` prefixed to avoid clashes) ---
841
842fn fd_sub3(a: [f64; 3], b: [f64; 3]) -> [f64; 3] {
843    [a[0] - b[0], a[1] - b[1], a[2] - b[2]]
844}
845
846fn fd_dot3(a: [f64; 3], b: [f64; 3]) -> f64 {
847    a[0] * b[0] + a[1] * b[1] + a[2] * b[2]
848}
849
850fn fd_norm3(v: [f64; 3]) -> f64 {
851    fd_dot3(v, v).sqrt()
852}
853
854fn fd_normalize3(v: [f64; 3]) -> [f64; 3] {
855    let n = fd_norm3(v);
856    if n < 1e-12 {
857        [0.0, 0.0, 1.0]
858    } else {
859        [v[0] / n, v[1] / n, v[2] / n]
860    }
861}
862
863/// Push an origin-ball region at world `p` unless a ball already sits there (dedup
864/// by world position, matching `leaders_buffers`' single drawn sphere). The shared
865/// point-region builder projects it to a screen circle; a behind-eye (perspective)
866/// origin is omitted — invisible, so not grabbable — matching the outline.
867fn push_origin_region(
868    out: &mut Vec<(DimRole, HitShape)>,
869    seen: &mut Vec<[f64; 3]>,
870    cam: &crate::view::ViewCamera,
871    p: [f64; 3],
872    px: f32,
873    field_key: &str,
874) {
875    if seen.iter().any(|o| fd_norm3(fd_sub3(*o, p)) < 1e-6) {
876        return;
877    }
878    seen.push(p);
879    if let Some(shape) = point_region(cam, p, px) {
880        out.push((DimRole::Origin(field_key.to_string()), shape));
881    }
882}
883
884/// The parameter `t` of the point on the axis line `a + t*dir` (dir UNIT) closest
885/// to the ray `ray_o + s*ray_d` (ray_d UNIT). `None` when the two are parallel
886/// (no well-defined projection). `t` is the signed world distance along `dir`
887/// from `a`. (`pub(super)`: shared with the constraint distance-arrow drag.)
888pub(super) fn closest_t_on_axis(
889    a: [f64; 3],
890    dir: [f64; 3],
891    ray_o: [f64; 3],
892    ray_d: [f64; 3],
893) -> Option<f64> {
894    let w0 = fd_sub3(a, ray_o);
895    let b = fd_dot3(dir, ray_d);
896    let d = fd_dot3(dir, w0);
897    let e = fd_dot3(ray_d, w0);
898    let denom = 1.0 - b * b;
899    if denom.abs() < 1e-9 {
900        return None;
901    }
902    Some((b * e - d) / denom)
903}
904
905fn vec3_to_arr(v: brep_kernel::Vec3) -> [f64; 3] {
906    [v.x, v.y, v.z]
907}
908
909/// The world CENTROID of a sketch profile — the average of its outer-loop curve
910/// start points (the profile-polygon vertices), approximating the previous
911/// face-average-center computation. Falls back to the sketch plane origin when
912/// no outer loop is available. Used to anchor the extrude/revolve gizmos on the
913/// geometry rather than at a possibly-far sketch-plane origin.
914fn sketch_profile_centroid(profile: &brep_kernel::SketchProfile) -> [f64; 3] {
915    let mut sum = [0.0f64; 3];
916    let mut count = 0usize;
917    if let Some(outer) = profile.regions.first().and_then(|region| region.first()) {
918        for curve in &outer.curves {
919            if let Ok(domain) = curve.domain() {
920                if let Ok(point) = curve.evaluate(domain[0]) {
921                    sum[0] += point.x;
922                    sum[1] += point.y;
923                    sum[2] += point.z;
924                    count += 1;
925                }
926            }
927        }
928    }
929    if count > 0 {
930        [sum[0] / count as f64, sum[1] / count as f64, sum[2] / count as f64]
931    } else {
932        vec3_to_arr(profile.origin)
933    }
934}
935