Files
OneCAD-Draft/src/scene/preview.rs
T

470 lines
18 KiB
Rust

// Auto-split from scene/mod.rs. Pure text-move; behaviour unchanged.
use super::*;
use crate::command::{AreaPreviewRegion, AreaPreviewSource};
use crate::scene::model::hatch_model::HatchPattern;
impl Scene {
// ── Preview wire ──────────────────────────────────────────────────────
pub fn set_preview_wires(&mut self, wires: Vec<WireModel>) {
// Preview wires are an overlay appended to the cached base wire set in
// `build_primitive`; they are NOT part of the tessellation cache. So a
// preview update must NOT bump `geometry_epoch` — that would re-
// tessellate the whole model on every rubber-band frame. The overlay
// forces a GPU wire re-upload on its own (the `has_overlay` content-id
// path), and iced redraws after the message that set the preview.
self.preview_wires = wires;
}
/// Publish all edited hatches as one live fill overlay.
pub fn set_preview_hatches(&mut self, handles: &[Handle]) {
let mut models = Vec::new();
for &handle in handles {
// Keep the existing direct Hatch/Solid preview path. INSERT has no
// entry in `self.hatches`, so this is a no-op for it and the full
// block expansion below supplies its fills.
self.append_preview_hatch(handle, &mut models);
}
models.extend(self.preview_insert_hatch_models(handles));
self.preview_hatches = std::sync::Arc::new(models);
}
pub fn set_command_preview_hatches(&mut self, models: Vec<HatchModel>) {
self.preview_hatches = std::sync::Arc::new(models);
}
pub fn set_area_preview_regions(&mut self, regions: &[AreaPreviewRegion]) {
let mut models = Vec::new();
for region in regions {
match &region.source {
AreaPreviewSource::Handles(handles) => {
for &handle in handles {
self.append_area_preview_handle(handle, region.subtract, &mut models);
}
}
AreaPreviewSource::Boundary(boundary) => {
if let Some(model) = Self::area_preview_hatch(
std::slice::from_ref(boundary),
region.subtract,
) {
models.push(model);
}
}
}
}
self.preview_hatches = std::sync::Arc::new(models);
}
fn append_area_preview_handle(
&self,
handle: Handle,
subtract: bool,
models: &mut Vec<HatchModel>,
) {
if let Some(mut model) = self.hatches.get(&handle).cloned() {
Self::style_area_preview_hatch(&mut model, subtract);
models.push(model);
return;
}
let direct_boundary = crate::scene::project::clip_boundary_polygon_for_document(
&self.document,
handle,
0.0,
);
let mut rings = if direct_boundary.len() >= 3 {
vec![direct_boundary
.into_iter()
.map(|point| [point[0] as f64, point[1] as f64])
.collect()]
} else {
Vec::new()
};
if !rings.is_empty() {
if let Some(model) = Self::area_preview_hatch(&rings, subtract) {
models.push(model);
}
return;
}
for wire in self.wire_models_for(&[handle]) {
let mut ring = Vec::new();
for (index, point) in wire.points.iter().enumerate() {
let low = wire.points_low.get(index).copied().unwrap_or([0.0; 3]);
let x = point[0] as f64 + low[0] as f64;
let y = point[1] as f64 + low[1] as f64;
if x.is_finite() && y.is_finite() {
let candidate = [x, y];
if ring.last().is_none_or(|last| *last != candidate) {
ring.push(candidate);
}
} else {
Self::push_area_preview_ring(&mut rings, &mut ring);
}
}
Self::push_area_preview_ring(&mut rings, &mut ring);
}
if let Some(model) = Self::area_preview_hatch(&rings, subtract) {
models.push(model);
}
}
fn push_area_preview_ring(rings: &mut Vec<Vec<[f64; 2]>>, ring: &mut Vec<[f64; 2]>) {
if ring.len() >= 3 {
let min_x = ring.iter().map(|point| point[0]).fold(f64::INFINITY, f64::min);
let max_x = ring
.iter()
.map(|point| point[0])
.fold(f64::NEG_INFINITY, f64::max);
let min_y = ring.iter().map(|point| point[1]).fold(f64::INFINITY, f64::min);
let max_y = ring
.iter()
.map(|point| point[1])
.fold(f64::NEG_INFINITY, f64::max);
let diagonal_sq = (max_x - min_x).powi(2) + (max_y - min_y).powi(2);
let twice_area = ring
.iter()
.zip(ring.iter().cycle().skip(1))
.take(ring.len())
.map(|(a, b)| a[0] * b[1] - b[0] * a[1])
.sum::<f64>()
.abs();
if twice_area > diagonal_sq * 1e-12 {
rings.push(std::mem::take(ring));
return;
}
}
ring.clear();
}
fn area_preview_hatch(rings: &[Vec<[f64; 2]>], subtract: bool) -> Option<HatchModel> {
let origin = rings.iter().find_map(|ring| ring.first()).copied()?;
let mut boundary = Vec::new();
let mut first = true;
for ring in rings.iter().filter(|ring| ring.len() >= 3) {
if !first {
boundary.push([f32::NAN, f32::NAN]);
}
first = false;
boundary.extend(
ring.iter()
.map(|point| [(point[0] - origin[0]) as f32, (point[1] - origin[1]) as f32]),
);
}
if boundary.len() < 3 {
return None;
}
let mut model = HatchModel {
render_instance: None,
world_origin: origin,
boundary: std::sync::Arc::new(boundary),
boundary_wcs: None,
boundary_exterior: None,
boundary_sources: None,
pattern: HatchPattern::Solid,
name: "AREA_PREVIEW".into(),
color: [0.0; 4],
aci: 0,
line_weight_px: 1.0,
angle_offset: 0.0,
scale: 1.0,
draw_depth: 0.0,
};
Self::style_area_preview_hatch(&mut model, subtract);
Some(model)
}
fn style_area_preview_hatch(model: &mut HatchModel, subtract: bool) {
model.pattern = HatchPattern::Solid;
model.name = "AREA_PREVIEW".into();
model.color = if subtract {
[1.0, 0.28, 0.18, 0.16]
} else {
[0.15, 0.55, 1.0, 0.12]
};
model.aci = 0;
model.line_weight_px = 1.0;
model.angle_offset = 0.0;
model.scale = 1.0;
model.draw_depth = 0.0;
}
fn append_preview_hatch(&self, handle: Handle, models: &mut Vec<HatchModel>) {
let Some(mut model) = self.hatches.get(&handle).cloned() else {
return;
};
let Some(entity) = self.document.get_entity(handle) else {
return;
};
let style = self.render_style(entity);
model.aci = style.4;
model.line_weight_px = style.3;
if !matches!(
model.pattern,
crate::scene::model::hatch_model::HatchPattern::Gradient { .. }
) {
model.color = style.0;
}
if self.selected.contains(&handle) {
model.color = [0.15, 0.55, 1.00, model.color[3]];
}
model.draw_depth = self
.draw_depth_map()
.get(&handle.value())
.map_or(0.0, |depth| depth[0]);
if let EntityType::Hatch(hatch) = entity {
if let Some(background) = crate::entities::hatch::background_color(hatch) {
let mut backdrop = model.clone();
backdrop.pattern =
crate::scene::model::hatch_model::HatchPattern::Solid;
let (background_color, background_aci) = match background {
acadrust::types::Color::ByLayer => {
let layer = self.document.layers.get(&hatch.common.layer);
let aci = layer
.and_then(|layer| match &layer.color {
acadrust::types::Color::Index(index) => Some(*index),
_ => None,
})
.unwrap_or(0);
(
crate::scene::view::render::layer_render_style(
&self.document,
&hatch.common.layer,
)
.color,
aci,
)
}
acadrust::types::Color::ByBlock => (style.0, style.4),
acadrust::types::Color::Index(index) => (
crate::scene::convert::tess_util::aci_to_rgba(
&acadrust::types::Color::Index(index),
),
index,
),
other => (
crate::scene::convert::tess_util::aci_to_rgba(&other),
0,
),
};
backdrop.color = background_color;
backdrop.aci = background_aci;
backdrop.name = "SOLID".into();
models.push(backdrop);
}
}
models.push(model);
}
pub fn set_preview_text(&mut self, verts: Vec<crate::scene::pipeline::text_gpu::TextVertex>) {
// Overlay glyphs — same reasoning as `set_preview_wires`: no geometry
// bump. Uploaded to a dedicated per-frame text buffer in `prepare`.
self.preview_text = verts;
}
pub fn clear_preview_wire(&mut self) {
// No geometry bump — see `set_preview_wires`. Dropping the overlay
// flips the wire content id back to the base tessellation id, which
// re-uploads the base wires (without the preview) on the next frame.
self.preview_wires = vec![];
if !self.preview_hatches.is_empty() {
self.preview_hatches = std::sync::Arc::new(Vec::new());
}
self.preview_text = vec![];
self.interim_wire = None;
// Drop any point-picked window marquee (STRETCH) so it doesn't linger
// after the command ends before the next mouse move. (#291)
self.selection.borrow_mut().preview_box = None;
}
pub fn wire_models_for(&self, handles: &[acadrust::Handle]) -> Vec<WireModel> {
handles
.iter()
.flat_map(|h| {
match self.document.get_entity(*h) {
// Hatches carry no outline in the normal wire set, but an
// edit preview (move / copy / array / grip-drag) needs to
// show the shape following the cursor. Build a live boundary
// from the current HatchModel — `apply_grip` keeps it in
// step, so the preview tracks a dragged grip in real time.
Some(EntityType::Hatch(_)) => {
self.hatch_outline_wire(*h).into_iter().collect()
}
Some(e) => self.tessellate_one(e),
None => Vec::new(),
}
})
.collect()
}
/// Split a MIRROR selection into plain ghost wires (reflected wholesale) and
/// text ghosts paired with their bounding-box centre. Lets the preview match
/// the commit for TEXT: MIRRTEXT on → true glyph mirror (full reflection, same
/// as plain geometry); MIRRTEXT off → right-reading at the mirror-symmetric
/// position (reflect the centre, translate) instead of hugging the axis.
pub fn mirror_preview_parts(
&self,
handles: &[Handle],
) -> (Vec<WireModel>, Vec<(WireModel, glam::DVec3)>) {
let mut plain: Vec<WireModel> = Vec::new();
let mut texts: Vec<(WireModel, glam::DVec3)> = Vec::new();
for h in handles {
let Some(e) = self.document.get_entity(*h) else {
continue;
};
if matches!(e, EntityType::Text(_)) {
let wires = self.tessellate_one(e);
let mut lo = [f64::INFINITY; 2];
let mut hi = [f64::NEG_INFINITY; 2];
for w in &wires {
for (i, p) in w.points.iter().enumerate() {
if !p[0].is_finite() || !p[1].is_finite() {
continue;
}
// Reconstruct f64 world from the double-single pair (the
// low residual may be absent on some wires).
let l = w.points_low.get(i).copied().unwrap_or([0.0; 3]);
let (x, y) = (p[0] as f64 + l[0] as f64, p[1] as f64 + l[1] as f64);
lo[0] = lo[0].min(x);
lo[1] = lo[1].min(y);
hi[0] = hi[0].max(x);
hi[1] = hi[1].max(y);
}
}
let center = if lo[0] <= hi[0] {
glam::DVec3::new((lo[0] + hi[0]) * 0.5, (lo[1] + hi[1]) * 0.5, 0.0)
} else {
glam::DVec3::ZERO
};
for w in wires {
texts.push((w, center));
}
} else {
plain.extend(self.tessellate_one(e));
}
}
(plain, texts)
}
/// Boundary outline wire for a hatch, reconstructed from its cached
/// `HatchModel` (offsets from `world_origin`). Used only for edit previews —
/// the normal render shows the fill, not this outline.
fn hatch_outline_wire(&self, handle: Handle) -> Option<WireModel> {
let m = self.hatches.get(&handle)?;
Self::hatch_model_outline_wire(handle, m)
}
fn hatch_model_outline_wire(handle: Handle, m: &HatchModel) -> Option<WireModel> {
let (wx, wy) = (m.world_origin[0], m.world_origin[1]);
let pts: Vec<[f64; 3]> = m
.boundary
.iter()
.map(|&[x, y]| {
if x.is_finite() && y.is_finite() {
[wx + x as f64, wy + y as f64, 0.0]
} else {
[f64::NAN; 3]
}
})
.collect();
if pts.len() < 2 {
return None;
}
Some(WireModel::solid_f64(
handle.value().to_string(),
pts,
m.color,
false,
))
}
/// Fill-only geometry has no resident wire for the normal rollover xray.
/// Supply an orange boundary overlay while an entity-pick command previews
/// a top-level hatch or an Insert whose visible children contain hatches.
pub fn fill_hover_preview_wires(&self, handle: Handle) -> Vec<WireModel> {
let mut wires: Vec<WireModel> = match self.document.get_entity(handle) {
Some(EntityType::Hatch(_)) => self.hatch_outline_wire(handle).into_iter().collect(),
Some(EntityType::Insert(_)) => self
.insert_hatches_for_click()
.get(&handle)
.into_iter()
.flatten()
.filter_map(|model| Self::hatch_model_outline_wire(handle, model))
.collect(),
_ => Vec::new(),
};
for wire in &mut wires {
wire.color = WireModel::HOVER;
wire.line_weight_px = wire.line_weight_px.max(2.0);
}
wires
}
/// Build wire models for an arbitrary slice of entities (e.g. clipboard contents).
/// Entities need not be in the document — they are tessellated directly.
pub fn wires_for_entities(&self, entities: &[acadrust::EntityType]) -> Vec<WireModel> {
entities
.iter()
.flat_map(|e| self.tessellate_one(e))
.collect()
}
pub fn set_interim_wire(&mut self, w: WireModel) {
// Overlay wire — same reasoning as `set_preview_wires`: no geometry
// bump, so the model isn't re-tessellated on every interim update.
self.interim_wire = Some(w);
}
/// Tessellate one block definition's entities into block-local wire models
/// (insertion base at origin) for the block-palette thumbnail. Nested INSERTs
/// expand through the block cache. Unknown / empty block → `vec![]`.
pub(crate) fn block_preview_wires(&self, name: &str) -> Vec<WireModel> {
use acadrust::EntityType;
let Some(br) = self.document.block_records.get(name) else {
return vec![];
};
let mut out = Vec::new();
for &eh in &br.entity_handles {
let Some(e) = self.document.get_entity(eh) else {
continue;
};
if matches!(e, EntityType::Block(_) | EntityType::BlockEnd(_)) {
continue;
}
out.extend(self.tessellate_one(e));
}
out
}
}
#[cfg(test)]
mod tests {
use super::*;
use acadrust::entities::Line;
use acadrust::types::Vector3;
use acadrust::EntityType;
#[test]
fn block_preview_wires_tessellates_block_entities() {
let mut s = Scene::new();
let mut line = Line::new();
line.start = Vector3::new(0.0, 0.0, 0.0);
line.end = Vector3::new(10.0, 5.0, 0.0);
s.define_block_from_owned_entities(
vec![EntityType::Line(line)],
"Widget",
glam::DVec3::ZERO,
)
.unwrap();
let wires = s.block_preview_wires("Widget");
assert!(!wires.is_empty(), "a LINE block must tessellate to wires");
}
#[test]
fn block_preview_wires_empty_for_unknown_block() {
let s = Scene::new();
assert!(s.block_preview_wires("Nope").is_empty());
}
}