forked from 1b-agent/opencadstudio-upstream
Merge PR #1179: Add tolerant OVERKILL cleanup options
# Conflicts: # src/app/commands/styleprops.rs
This commit is contained in:
commit
32952e3eda
5 files changed
+299
-61
No files matched your search
@@ -15,6 +15,7 @@ pub mod mirror;
|
||||
pub mod mledit;
|
||||
pub mod ncopy;
|
||||
pub mod offset;
|
||||
pub mod overkill;
|
||||
pub mod pedit;
|
||||
pub mod block_edit;
|
||||
pub mod refedit;
|
||||
|
||||
@@ -0,0 +1,178 @@
|
||||
use acadrust::{EntityType, Handle};
|
||||
use glam::DVec3;
|
||||
use crate::command::{CadCommand, CmdOption, CmdResult};
|
||||
|
||||
#[derive(Clone, Copy)]
|
||||
pub struct Settings {
|
||||
pub tolerance: f64,
|
||||
pub ignore: u16,
|
||||
pub optimize: bool,
|
||||
pub overlap: bool,
|
||||
pub end_to_end: bool,
|
||||
pub associativity: bool,
|
||||
}
|
||||
impl Default for Settings {
|
||||
fn default() -> Self { Self { tolerance: 1e-6, ignore: 0, optimize: true,
|
||||
overlap: true, end_to_end: true, associativity: true } }
|
||||
}
|
||||
fn remembered() -> &'static std::sync::Mutex<Settings> {
|
||||
static SETTINGS: std::sync::OnceLock<std::sync::Mutex<Settings>> = std::sync::OnceLock::new();
|
||||
SETTINGS.get_or_init(|| std::sync::Mutex::new(Settings::default()))
|
||||
}
|
||||
enum Step { Selection, Options, Ignore, Tolerance, Optimize, Overlap, EndToEnd, Associativity }
|
||||
pub struct OverkillCommand { selected: Vec<Handle>, step: Step, settings: Settings }
|
||||
impl OverkillCommand {
|
||||
pub fn new(selected: Vec<Handle>) -> Self {
|
||||
let step = if selected.is_empty() { Step::Selection } else { Step::Options };
|
||||
Self { selected, step, settings: *remembered().lock().unwrap_or_else(|e|e.into_inner()) }
|
||||
}
|
||||
fn finish(&self) -> CmdResult {
|
||||
let s = self.settings;
|
||||
*remembered().lock().unwrap_or_else(|e|e.into_inner())=s;
|
||||
CmdResult::Relaunch(format!("OVERKILL_APPLY {} {} {} {} {} {}", s.tolerance,
|
||||
s.ignore, u8::from(s.optimize), u8::from(s.overlap), u8::from(s.end_to_end),
|
||||
u8::from(s.associativity)), self.selected.clone())
|
||||
}
|
||||
}
|
||||
impl CadCommand for OverkillCommand {
|
||||
fn name(&self) -> &'static str { "OVERKILL" }
|
||||
fn prompt(&self) -> String {
|
||||
match self.step {
|
||||
Step::Selection => "Select objects:".into(),
|
||||
Step::Options => format!("Tolerance={} Ignore={} Optimize={} Overlap={} End-to-end={}\nEnter an option [Done/Ignore/tOlerance/optimize Plines/combine parTial overlap/combine Endtoend/Associativity] <Done>:", self.settings.tolerance, self.settings.ignore, self.settings.optimize, self.settings.overlap, self.settings.end_to_end),
|
||||
Step::Ignore => "Properties to ignore [None/All/Color/LAyer/Ltype/ltScale/LWeight/Thickness/TRansparency/plotSTyle/Material]:".into(),
|
||||
Step::Tolerance => format!("Specify tolerance <{}>:", self.settings.tolerance),
|
||||
Step::Optimize => format!("Optimize segments within polylines [Yes/No] <{}>:",if self.settings.optimize{"Yes"}else{"No"}),
|
||||
Step::Overlap => format!("Combine partially overlapping objects [Yes/No] <{}>:",if self.settings.overlap{"Yes"}else{"No"}),
|
||||
Step::EndToEnd => format!("Combine end-to-end objects [Yes/No] <{}>:",if self.settings.end_to_end{"Yes"}else{"No"}),
|
||||
Step::Associativity => format!("Maintain associative objects [Yes/No] <{}>:",if self.settings.associativity{"Yes"}else{"No"}),
|
||||
}
|
||||
}
|
||||
fn options(&self) -> Vec<CmdOption> {
|
||||
match self.step {
|
||||
Step::Options => [("Done","D"),("Ignore","I"),("Tolerance","O"),("Optimize polylines","P"),("Combine overlap","T"),("Combine end-to-end","E"),("Associativity","A")].into_iter().map(|(l,k)|CmdOption::new(l,k)).collect(),
|
||||
Step::Optimize|Step::Overlap|Step::EndToEnd|Step::Associativity => vec![CmdOption::new("Yes","Y"),CmdOption::new("No","N")],
|
||||
_ => Vec::new(),
|
||||
}
|
||||
}
|
||||
fn is_selection_gathering(&self) -> bool { matches!(self.step,Step::Selection) }
|
||||
fn on_selection_complete(&mut self, handles: Vec<Handle>) -> CmdResult { self.selected=handles; CmdResult::NeedPoint }
|
||||
fn wants_text_input(&self) -> bool { !self.is_selection_gathering() }
|
||||
fn on_point(&mut self, _:DVec3) -> CmdResult { CmdResult::NeedPoint }
|
||||
fn on_enter(&mut self) -> CmdResult {
|
||||
match self.step {
|
||||
Step::Selection if self.selected.is_empty() => CmdResult::Cancel,
|
||||
Step::Options => self.finish(),
|
||||
_ => { self.step=Step::Options; CmdResult::NeedPoint }
|
||||
}
|
||||
}
|
||||
fn on_text_input(&mut self, text:&str) -> Option<CmdResult> {
|
||||
let k=text.trim().to_ascii_uppercase();
|
||||
match self.step {
|
||||
Step::Options => match k.as_str() {
|
||||
"D"|"DONE" => return Some(self.finish()),
|
||||
"I"|"IGNORE" => self.step=Step::Ignore,
|
||||
"O"|"TOLERANCE" => self.step=Step::Tolerance,
|
||||
"P"|"PLINES" => self.step=Step::Optimize,
|
||||
"T"|"PARTIAL" => self.step=Step::Overlap,
|
||||
"E"|"ENDTOEND" => self.step=Step::EndToEnd,
|
||||
"A"|"ASSOCIATIVITY" => self.step=Step::Associativity,
|
||||
_=>{},
|
||||
},
|
||||
Step::Ignore => {
|
||||
let flag=match k.as_str() { "N"|"NONE"=>{self.settings.ignore=0;0}, "A"|"ALL"=>511,
|
||||
"C"|"COLOR"=>1,"LA"|"LAYER"=>2,"L"|"LTYPE"=>4,"S"|"LTSCALE"=>8,
|
||||
"LW"|"LWEIGHT"=>16,"T"|"THICKNESS"=>32,"TR"|"TRANSPARENCY"=>64,
|
||||
"ST"|"PLOTSTYLE"=>128,"M"|"MATERIAL"=>256,_=>return Some(CmdResult::NeedPoint)};
|
||||
self.settings.ignore|=flag; self.step=Step::Options;
|
||||
}
|
||||
Step::Tolerance => if let Ok(v)=k.parse::<f64>() { if v.is_finite()&&v>=0.0 {
|
||||
self.settings.tolerance=v; self.step=Step::Options;
|
||||
} },
|
||||
Step::Optimize|Step::Overlap|Step::EndToEnd|Step::Associativity => {
|
||||
let yes=match k.as_str(){"Y"|"YES"=>true,"N"|"NO"=>false,_=>return Some(CmdResult::NeedPoint)};
|
||||
match self.step {Step::Optimize=>self.settings.optimize=yes,Step::Overlap=>self.settings.overlap=yes,
|
||||
Step::EndToEnd=>self.settings.end_to_end=yes,Step::Associativity=>self.settings.associativity=yes,_=>{}}
|
||||
self.step=Step::Options;
|
||||
}
|
||||
_=>{},
|
||||
}
|
||||
Some(CmdResult::NeedPoint)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn normalized(entity:&EntityType, ignore:u16) -> EntityType {
|
||||
let mut e=entity.clone();
|
||||
let common=e.common_mut();
|
||||
let default=acadrust::entities::EntityCommon::new();
|
||||
common.handle=Handle::NULL;
|
||||
if ignore&1!=0 {common.color=default.color;common.color_name=None;common.color_book_handle=None;}
|
||||
if ignore&2!=0 {common.layer.clear();}
|
||||
if ignore&4!=0 {common.linetype.clear();common.linetype_handle=None;}
|
||||
if ignore&8!=0 {common.linetype_scale=1.0;}
|
||||
if ignore&16!=0 {common.line_weight=default.line_weight;}
|
||||
if ignore&64!=0 {common.transparency=default.transparency;}
|
||||
if ignore&128!=0 {common.plotstyle_flags=0;common.plotstyle_handle=None;}
|
||||
if ignore&256!=0 {common.material_flags=0;common.material_handle=None;}
|
||||
if ignore&32!=0 {match &mut e {EntityType::Line(v)=>v.thickness=0.0,
|
||||
EntityType::Circle(v)=>v.thickness=0.0,EntityType::Arc(v)=>v.thickness=0.0,
|
||||
EntityType::LwPolyline(v)=>v.thickness=0.0,_=>{}}}
|
||||
e
|
||||
}
|
||||
|
||||
pub fn optimize(entity:&mut EntityType, tolerance:f64) {
|
||||
if let EntityType::LwPolyline(p)=entity {
|
||||
// Width changes and bulged spans carry geometry that must be retained.
|
||||
if p.vertices.iter().any(|v|v.bulge!=0.0||v.start_width!=0.0||v.end_width!=0.0) {return;}
|
||||
let points:Vec<_>=p.vertices.iter().map(|v|[v.location.x,v.location.y,p.elevation]).collect();
|
||||
let indices=cadkernel::space::simplify_linear_chain(&points,tolerance);
|
||||
if indices.len()>=if p.is_closed{3}else{2} {
|
||||
p.vertices=indices.into_iter().map(|i|p.vertices[i].clone()).collect();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use acadrust::entities::{Line, LwPolyline};
|
||||
use acadrust::types::{Color, Vector2};
|
||||
|
||||
#[test]
|
||||
fn normalization_ignores_only_the_requested_properties() {
|
||||
let mut first = Line::from_coords(0.0, 0.0, 0.0, 2.0, 0.0, 0.0);
|
||||
first.common.handle = Handle::new(1);
|
||||
first.common.layer = "FIRST".into();
|
||||
first.common.color = Color::from_index(1);
|
||||
let mut second = first.clone();
|
||||
second.common.handle = Handle::new(2);
|
||||
second.common.layer = "SECOND".into();
|
||||
second.common.color = Color::from_index(2);
|
||||
|
||||
assert_ne!(normalized(&EntityType::Line(first.clone()), 0), normalized(&EntityType::Line(second.clone()), 0));
|
||||
assert_eq!(normalized(&EntityType::Line(first), 1 | 2), normalized(&EntityType::Line(second), 1 | 2));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn polyline_optimization_removes_only_plain_collinear_vertices() {
|
||||
let mut plain = EntityType::LwPolyline(LwPolyline::from_points(vec![
|
||||
Vector2::new(0.0, 0.0),
|
||||
Vector2::new(1.0, 0.0),
|
||||
Vector2::new(2.0, 0.0),
|
||||
]));
|
||||
optimize(&mut plain, 1e-9);
|
||||
let EntityType::LwPolyline(plain) = plain else { unreachable!() };
|
||||
assert_eq!(plain.vertices.len(), 2);
|
||||
|
||||
let mut styled = LwPolyline::from_points(vec![
|
||||
Vector2::new(0.0, 0.0),
|
||||
Vector2::new(1.0, 0.0),
|
||||
Vector2::new(2.0, 0.0),
|
||||
]);
|
||||
styled.vertices[1].start_width = 1.0;
|
||||
let mut styled = EntityType::LwPolyline(styled);
|
||||
optimize(&mut styled, 1e-9);
|
||||
let EntityType::LwPolyline(styled) = styled else { unreachable!() };
|
||||
assert_eq!(styled.vertices.len(), 3);
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user