Merge pull request #1196 from Ramox81/codex/parity-divide-20260911

This commit is contained in:
Hakan Seven committed 2026-09-11 15:37:37 +03:00
commit 7018a9c915
5 files changed
+212 -77

No files matched your search

+6 -4
View File
@@ -3897,23 +3897,25 @@ impl OpenCADStudio {
self.tabs[i].scene.clear_preview_wire();
self.restore_pre_cmd_tangent();
}
CmdResult::DivideEntity { handle, n } => {
CmdResult::DivideEntity { handle, n, marker } => {
use crate::modules::draw::inquiry::divide::divide_entity;
let pts = self.tabs[i]
.scene
.document
.get_entity(handle)
.map(|e| divide_entity(e, n))
.map(|e| divide_entity(e, n, marker.as_ref()))
.unwrap_or_default();
let count = pts.len();
if count > 0 {
self.push_undo_snapshot(i, "DIVIDE");
for p in pts {
let layer = self.tabs[i].active_layer.clone();
for mut p in pts {
p.as_entity_mut().set_layer(layer.clone());
self.tabs[i].scene.add_entity(p);
}
self.tabs[i].dirty = true;
self.command_line
.push_output(crate::tf!("DIVIDE: {count} point(s) placed.").as_ref());
.push_output(crate::tf!("DIVIDE: {count} marker(s) placed.").as_ref());
} else {
self.command_line
.push_error(crate::t!("DIVIDE: entity type not supported or N < 2.").as_ref());
+1 -1
View File
@@ -989,7 +989,7 @@ impl OpenCADStudio {
"DIVIDE" => {
use crate::modules::draw::inquiry::divide::DivideCommand;
let cmd = DivideCommand::new();
let cmd = DivideCommand::new().with_blocks(self.tabs[i].scene.custom_block_names());
self.command_line.push_info(&cmd.prompt());
self.tabs[i].active_cmd = Some(Box::new(cmd));
}
+9 -1
View File
@@ -1546,7 +1546,7 @@ pub enum CmdResult {
op: crate::modules::draw::modify::pedit::PeditOp,
},
/// Place Point entities at N equal intervals along the entity.
DivideEntity { handle: Handle, n: usize },
DivideEntity { handle: Handle, n: usize, marker: Option<CurveMarker> },
/// Place Point entities at `segment_length` intervals along the entity.
MeasureEntity { handle: Handle, segment_length: f64 },
/// Extend/trim a Line or Arc by the given mode; end command.
@@ -2004,6 +2004,14 @@ impl InputKind {
}
}
#[derive(Clone, Debug)]
pub struct CurveMarker {
pub block: String,
pub align: bool,
pub plane: WorkingPlane,
}
/// Current screen projection and configured aperture for point-feature picking.
#[derive(Clone, Copy)]
pub struct PointPickContext {
+45
View File
@@ -107,6 +107,51 @@ pub fn entity_curve(entity: &EntityType) -> Option<PlanarCurve> {
}
}
/// Exact spatial source geometry for commands that traverse nonplanar curves.
/// Curve construction and arc-length calculations remain in the kernel.
pub fn entity_spatial_measurement(entity: &EntityType) -> Option<cadkernel::space::ArcLengthCurve3> {
use cadkernel::space::{ArcLengthCurve3, NurbsCurve3, Parameterization};
let curve = match entity {
EntityType::Polyline3D(polyline) => {
if polyline.flags.spline_fit {
let degree = match polyline.smooth_type as i16 { 5 => 2, 6 => 3, _ => return None };
let controls: Vec<_> = polyline.vertices.iter().filter(|vertex| vertex.flags & 16 != 0)
.map(|vertex| [vertex.position.x, vertex.position.y, vertex.position.z]).collect();
let curve = NurbsCurve3::from_control_polygon(degree, &controls, polyline.is_closed())?;
return ArcLengthCurve3::from_nurbs(curve);
}
let points: Vec<_> = polyline.vertices.iter().map(|vertex| {
[vertex.position.x, vertex.position.y, vertex.position.z]
}).collect();
return ArcLengthCurve3::from_polyline(&points, polyline.is_closed());
}
EntityType::Spline(spline) => {
if crate::entities::spline::uses_fit_method(spline) {
let points: Vec<_> = spline.fit_points.iter().copied().map(xyz).collect();
let parameterization = match spline.knot_parameterization {
1 => Parameterization::Centripetal, 2 => Parameterization::Uniform,
_ => Parameterization::Chord,
};
if spline.flags.periodic {
NurbsCurve3::interpolate_periodic(&points, parameterization)?
} else {
if spline.flags.closed { return None; }
NurbsCurve3::interpolate_fit(&points, Some(xyz(spline.begin_tangent)),
Some(xyz(spline.end_tangent)), parameterization)?
}
} else {
let controls: Vec<_> = spline.control_points.iter().copied().map(xyz).collect();
let weights = if spline.weights.is_empty() { vec![1.0; controls.len()] }
else { spline.weights.clone() };
NurbsCurve3::new_strict(spline.degree as usize, controls, spline.knots.clone(), weights)?
.with_periodicity(spline.flags.periodic)
}
}
_ => return None,
};
ArcLengthCurve3::from_nurbs(curve)
}
/// The plane an OCS-stored entity's coordinates are read in.
///
/// `elevation` is the entity's third stored coordinate — the distance along
+151 -71
View File
@@ -1,86 +1,125 @@
// DIVIDE command — place Point entities at N equal intervals along an entity.
// MEASURE command — place Point entities at fixed-distance intervals along an entity.
use acadrust::entities::Point as PointEnt;
use cadkernel::space::PlanarCurve;
// DIVIDE and MEASURE place points or block references along a curve.
use acadrust::entities::{Insert, Point as PointEnt};
use acadrust::types::Vector3;
use acadrust::{EntityType, Handle};
use acadrust::{Entity, EntityType, Handle};
use cadkernel::space::PlanarCurve;
use glam::DVec3;
use crate::entities::curve::entity_curve;
use crate::command::{CadCommand, CmdOption, CmdResult, CurveMarker, WorkingPlane};
use crate::entities::curve::{entity_curve, entity_spatial_measurement};
use crate::t;
const MEASURE: bool = false;
use crate::command::{CadCommand, CmdResult};
#[derive(Clone, Copy, PartialEq)]
enum Step { Pick, Amount, BlockName, Align }
// ── DIVIDE ─────────────────────────────────────────────────────────────────
pub struct DivideCommand {
pub struct DivisionMarkerCommand {
target: Option<Handle>,
waiting_for_n: bool,
pick_point: DVec3,
step: Step,
blocks: Vec<String>,
block: Option<String>,
align: bool,
plane: WorkingPlane,
valid_pick: bool,
}
impl DivideCommand {
pub type DivideCommand = DivisionMarkerCommand;
impl DivisionMarkerCommand {
pub fn new() -> Self {
Self {
target: None,
waiting_for_n: false,
}
Self { target: None, pick_point: DVec3::ZERO, step: Step::Pick,
blocks: Vec::new(), block: None, align: true,
plane: WorkingPlane::default(), valid_pick: false }
}
pub fn with_blocks(mut self, blocks: Vec<String>) -> Self {
self.blocks = blocks;
self
}
fn marker(&self) -> Option<CurveMarker> {
self.block.as_ref().map(|name| CurveMarker {
block: name.clone(), align: self.align, plane: self.plane,
})
}
}
impl CadCommand for DivideCommand {
fn name(&self) -> &'static str {
"DIVIDE"
}
impl CadCommand for DivisionMarkerCommand {
fn name(&self) -> &'static str { if MEASURE { "MEASURE" } else { "DIVIDE" } }
fn prompt(&self) -> String {
if self.target.is_none() {
t!("DIVIDE Select object to divide:").into_owned()
} else {
t!("DIVIDE Enter number of segments:").into_owned()
let prompt = match self.step {
Step::Pick if MEASURE => "Select object to measure:",
Step::Pick => "Select object to divide:",
Step::BlockName => "Enter name of block to insert:",
Step::Align => "Align block with object? [Yes/No] <Yes>:",
Step::Amount if MEASURE && self.block.is_none() => "Specify length of segment or [Block]:",
Step::Amount if MEASURE => "Specify length of segment:",
Step::Amount if self.block.is_none() => "Enter number of segments or [Block]:",
Step::Amount => "Enter number of segments:",
};
format!("{} {}", self.name(), prompt)
}
fn options(&self) -> Vec<CmdOption> {
match self.step {
Step::Amount if self.block.is_none() => vec![CmdOption::new("Block", "B")],
Step::Align => vec![CmdOption::new("Yes", "Y"), CmdOption::new("No", "N")],
_ => vec![],
}
}
fn needs_entity_pick(&self) -> bool {
self.target.is_none()
fn set_working_plane(&mut self, plane: WorkingPlane) { self.plane = plane; }
fn needs_entity_pick(&self) -> bool { self.step == Step::Pick }
fn inject_before_entity_pick(&self) -> bool { true }
fn inject_picked_entity(&mut self, entity: EntityType) {
self.valid_pick = measurable(&entity).is_some();
}
fn on_entity_pick(&mut self, handle: Handle, _pt: DVec3) -> CmdResult {
if handle.is_null() {
return CmdResult::NeedPoint;
}
fn on_entity_pick(&mut self, handle: Handle, pt: DVec3) -> CmdResult {
if handle.is_null() || !self.valid_pick { return CmdResult::NeedPoint; }
self.target = Some(handle);
self.waiting_for_n = true;
self.pick_point = pt;
self.step = Step::Amount;
CmdResult::NeedPoint
}
fn wants_text_input(&self) -> bool {
self.waiting_for_n
}
fn wants_text_input(&self) -> bool { self.step != Step::Pick }
fn dyn_field(&self) -> crate::command::DynField {
if self.waiting_for_n {
crate::command::DynField::Scalar
} else {
crate::command::DynField::Point
}
if self.step == Step::Amount { crate::command::DynField::Scalar }
else { crate::command::DynField::Point }
}
fn dyn_commit_as_text(&self) -> bool { self.step == Step::Amount }
fn on_text_input(&mut self, text: &str) -> Option<CmdResult> {
let n: usize = text.trim().parse().ok().filter(|&n| n >= 2)?;
let handle = self.target?;
self.waiting_for_n = false;
Some(CmdResult::DivideEntity { handle, n })
}
fn on_point(&mut self, _pt: DVec3) -> CmdResult {
CmdResult::NeedPoint
let text = text.trim();
match self.step {
Step::BlockName => {
self.block = Some(self.blocks.iter().find(|name| name.eq_ignore_ascii_case(text))?.clone());
self.step = Step::Align;
}
Step::Align => {
self.align = match text.to_ascii_uppercase().as_str() {
"Y" | "YES" => true,
"N" | "NO" => false,
_ => return None,
};
self.step = Step::Amount;
}
Step::Amount => {
if self.block.is_none() && matches!(text.to_ascii_uppercase().as_str(), "B" | "BLOCK") {
self.step = Step::BlockName;
} else {
let n = text.parse::<usize>().ok().filter(|n| (2..=32767).contains(n))?;
return Some(CmdResult::DivideEntity { handle: self.target?, n, marker: self.marker() });
}
}
Step::Pick => return None,
}
Some(CmdResult::NeedPoint)
}
fn on_point(&mut self, _pt: DVec3) -> CmdResult { CmdResult::NeedPoint }
fn on_enter(&mut self) -> CmdResult {
CmdResult::Cancel
if self.step == Step::Align {
self.align = true;
self.step = Step::Amount;
CmdResult::NeedPoint
} else { CmdResult::Cancel }
}
}
// ── MEASURE ────────────────────────────────────────────────────────────────
pub struct MeasureCommand {
target: Option<Handle>,
@@ -157,30 +196,30 @@ impl CadCommand for MeasureCommand {
}
}
// ── Geometry ───────────────────────────────────────────────────────────────
/// Compute N-1 equally spaced points along the entity (DIVIDE).
pub fn divide_entity(entity: &EntityType, n: usize) -> Vec<EntityType> {
if n < 2 {
pub fn divide_entity(entity: &EntityType, n: usize, marker: Option<&CurveMarker>) -> Vec<EntityType> {
if !(2..=32767).contains(&n) {
return vec![];
}
let Some((curve, total)) = measurable(entity) else {
return vec![];
};
let step = total / n as f64;
(1..n)
.map(|k| make_point(curve.point_at_distance(step * k as f64)))
let last = if curve.is_closed() { n } else { n - 1 };
(1..=last)
.map(|k| make_marker(&curve, step * k as f64, marker))
.collect()
}
/// Compute points at fixed `segment_length` intervals along the entity (MEASURE).
pub fn measure_entity(entity: &EntityType, segment_length: f64) -> Vec<EntityType> {
if segment_length <= 0.0 {
return vec![];
}
let Some((curve, total)) = measurable(entity) else {
let Some(curve) = entity_curve(entity) else {
return vec![];
};
let total = curve.length();
if !total.is_finite() || total <= 1e-10 { return vec![]; }
let mut pts = Vec::new();
let mut walked = segment_length;
while walked < total - 1e-6 {
@@ -190,6 +229,24 @@ pub fn measure_entity(entity: &EntityType, segment_length: f64) -> Vec<EntityTyp
pts
}
fn make_marker(curve: &MeasuredCurve, distance: f64, marker: Option<&CurveMarker>) -> EntityType {
let pos = curve.point_at_distance(distance);
let Some(marker) = marker else {
let mut point = PointEnt::new();
point.location = Vector3::new(pos[0], pos[1], pos[2]);
return EntityType::Point(point);
};
let local = marker.plane.to_local(DVec3::from_array(pos));
let mut insert = Insert::new(marker.block.clone(), Vector3::new(local.x, local.y, local.z));
if marker.align {
let tangent = DVec3::from_array(curve.tangent_at(curve.parameter_at_distance(distance)));
let tangent = marker.plane.vector_to_local(tangent);
insert.rotation = tangent.y.atan2(tangent.x);
}
insert.apply_transform(&marker.plane.to_world_transform());
EntityType::Insert(insert)
}
fn make_point(pos: [f64; 3]) -> EntityType {
let mut p = PointEnt::new();
p.location = Vector3::new(pos[0], pos[1], pos[2]);
@@ -198,13 +255,36 @@ fn make_point(pos: [f64; 3]) -> EntityType {
/// The entity's curve and its length, or `None` for anything that cannot be
/// walked along — a hatch, a block, an unbounded ray.
fn measurable(entity: &EntityType) -> Option<(PlanarCurve, f64)> {
let curve = entity_curve(entity)?;
let total = curve.curve.length();
enum MeasuredCurve {
Planar(PlanarCurve),
Spatial(cadkernel::space::ArcLengthCurve3),
}
impl MeasuredCurve {
fn is_closed(&self) -> bool {
match self { Self::Planar(curve) => curve.is_closed(), Self::Spatial(curve) => curve.is_closed() }
}
fn point_at_distance(&self, distance: f64) -> [f64; 3] {
match self { Self::Planar(curve) => curve.point_at_distance(distance), Self::Spatial(curve) => curve.point_at_distance(distance) }
}
fn parameter_at_distance(&self, distance: f64) -> f64 {
match self { Self::Planar(curve) => curve.parameter_at_distance(distance), Self::Spatial(curve) => curve.parameter_at_distance(distance) }
}
fn tangent_at(&self, parameter: f64) -> [f64; 3] {
match self { Self::Planar(curve) => curve.tangent_at(parameter), Self::Spatial(curve) => curve.tangent_at(parameter) }
}
}
fn measurable(entity: &EntityType) -> Option<(MeasuredCurve, f64)> {
let (curve, total) = if let Some(curve) = entity_curve(entity) {
let total = curve.length();
(MeasuredCurve::Planar(curve), total)
} else {
let curve = entity_spatial_measurement(entity)?;
let total = curve.length();
(MeasuredCurve::Spatial(curve), total)
};
(total.is_finite() && total > 1e-10).then_some((curve, total))
}
// ── Autocomplete registry ─────────────────────────────────
inventory::submit!(crate::command::CommandRegistration { names: &["DIVIDE"] }); // DivideCommand
inventory::submit!(crate::command::CommandRegistration { names: &["MEASURE"] }); // MeasureCommand
inventory::submit!(crate::command::CommandRegistration { names: &["DIVIDE"] });
inventory::submit!(crate::command::CommandRegistration { names: &["MEASURE"] });