Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Clean up approximation code #245

Merged
merged 8 commits into from
Feb 23, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ bytemuck = "1.7.3"
decorum = "0.3.1"
futures = "0.3.21"
libloading = "0.7.2"
map-macro = "0.2.0"
nalgebra = "0.30.0"
notify = "5.0.0-pre.13"
num-traits = "0.2.14"
Expand Down
191 changes: 26 additions & 165 deletions src/kernel/approximation.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use std::{cmp::Ordering, collections::HashSet};
use std::collections::HashSet;

use crate::math::{Point, Scalar, Segment};

Expand All @@ -11,7 +11,7 @@ pub struct Approximation {
///
/// These could be actual vertices from the model, points that approximate
/// an edge, or points that approximate a face.
pub points: Vec<Point<3>>,
pub points: HashSet<Point<3>>,

/// Segments that approximate edges
///
Expand All @@ -21,7 +21,7 @@ pub struct Approximation {
/// All the points of these segments will also be available in the `points`
/// field of this struct. This can be verified by calling
/// [`Approximation::validate`].
pub segments: Vec<Segment<3>>,
pub segments: HashSet<Segment<3>>,
}

impl Approximation {
Expand Down Expand Up @@ -57,24 +57,27 @@ impl Approximation {
}
}

let mut segments = Vec::new();
let mut segments = HashSet::new();
for segment in segment_points.windows(2) {
let p0 = segment[0];
let p1 = segment[1];

segments.push(Segment::from([p0, p1]));
segments.insert(Segment::from([p0, p1]));
}

Self { points, segments }
Self {
points: points.into_iter().collect(),
segments,
}
}

/// Compute an approximation for a cycle
///
/// `tolerance` defines how far the approximation is allowed to deviate from
/// the actual cycle.
pub fn for_cycle(cycle: &Cycle, tolerance: Scalar) -> Self {
let mut points = Vec::new();
let mut segments = Vec::new();
let mut points = HashSet::new();
let mut segments = HashSet::new();

for edge in &cycle.edges {
let approx = Self::for_edge(edge, tolerance);
Expand All @@ -83,32 +86,6 @@ impl Approximation {
segments.extend(approx.segments);
}

// As this is a cycle, neighboring edges are going to share vertices.
// Let's remove all those duplicates.
points.sort_by(|a, b| {
if a.x < b.x {
return Ordering::Less;
}
if a.x > b.x {
return Ordering::Greater;
}
if a.y < b.y {
return Ordering::Less;
}
if a.y > b.y {
return Ordering::Greater;
}
if a.z < b.z {
return Ordering::Less;
}
if a.z > b.z {
return Ordering::Greater;
}

Ordering::Equal
});
points.dedup();

Self { points, segments }
}

Expand All @@ -117,8 +94,8 @@ impl Approximation {
/// `tolerance` defines how far the approximation is allowed to deviate from
/// the actual edges.
pub fn for_edges(edges: &Edges, tolerance: Scalar) -> Self {
let mut points = Vec::new();
let mut segments = Vec::new();
let mut points = HashSet::new();
let mut segments = HashSet::new();

for cycle in &edges.cycles {
let approx = Self::for_cycle(cycle, tolerance);
Expand All @@ -129,87 +106,14 @@ impl Approximation {

Self { points, segments }
}

/// Validate the approximation
///
/// Returns an `Err(ValidationError)`, if the validation is not valid. See
/// [`ValidationError`] for the ways that the approximation can be invalid.
pub fn validate(&self) -> Result<(), ValidationError> {
let mut duplicate_points = Vec::new();
let mut duplicate_segments = Vec::new();
let mut invalid_segments = Vec::new();
let mut segments_with_invalid_points = Vec::new();

// Verify that there are no duplicate points
let mut points = HashSet::new();
for &point in &self.points {
if points.contains(&point) {
duplicate_points.push(point);
}

points.insert(point);
}

let mut segments = HashSet::new();
for &segment in &self.segments {
let [a, b] = segment.points();
// Verify that there are no duplicate segments
let ab = [a, b];
let ba = [b, a];
if segments.contains(&ab) {
duplicate_segments.push(segment);
}
segments.insert(ab);
segments.insert(ba);

// Verify that segments are actually segments
if a == b {
invalid_segments.push(segment);
}

// Verify that segments refer to valid points
if !(self.points.contains(&a) && self.points.contains(&b)) {
segments_with_invalid_points.push(segment);
}
}

if !(duplicate_points.is_empty()
&& duplicate_segments.is_empty()
&& invalid_segments.is_empty()
&& segments_with_invalid_points.is_empty())
{
return Err(ValidationError {
duplicate_points,
duplicate_segments,
invalid_segments,
segments_with_invalid_points,
});
}

Ok(())
}
}

/// Error returned by [`Approximation::validate`]
#[derive(Debug)]
pub struct ValidationError {
/// Points that are duplicated
pub duplicate_points: Vec<Point<3>>,

/// Segments that are duplicated
pub duplicate_segments: Vec<Segment<3>>,

/// Segments that have two equal points
pub invalid_segments: Vec<Segment<3>>,

/// Segments that do not refer to points from the approximation
pub segments_with_invalid_points: Vec<Segment<3>>,
}

#[cfg(test)]
mod tests {
use std::cell::RefCell;

use map_macro::set;

use crate::{
kernel::{
geometry::Curve,
Expand All @@ -224,7 +128,7 @@ mod tests {
use super::Approximation;

#[test]
fn test_for_edge() {
fn for_edge() {
let tolerance = Scalar::ONE;

let a = Point::from([1., 2., 3.]);
Expand All @@ -244,8 +148,8 @@ mod tests {
assert_eq!(
Approximation::for_edge(&edge_regular, tolerance),
Approximation {
points: vec![a, b, c, d],
segments: vec![
points: set![a, b, c, d],
segments: set![
Segment::from([a, b]),
Segment::from([b, c]),
Segment::from([c, d]),
Expand All @@ -257,14 +161,14 @@ mod tests {
assert_eq!(
Approximation::for_edge(&edge_self_connected, tolerance),
Approximation {
points: vec![b, c],
segments: vec![Segment::from([b, c]), Segment::from([c, b])],
points: set![b, c],
segments: set![Segment::from([b, c]), Segment::from([c, b])],
}
);
}

#[test]
fn test_for_cycle() {
fn for_cycle() {
let tolerance = Scalar::ONE;

let a = Point::from([1., 2., 3.]);
Expand All @@ -291,8 +195,8 @@ mod tests {
assert_eq!(
Approximation::for_cycle(&cycle, tolerance),
Approximation {
points: vec![a, b, c],
segments: vec![
points: set![a, b, c],
segments: set![
Segment::from([a, b]),
Segment::from([b, c]),
Segment::from([c, a]),
Expand All @@ -302,7 +206,7 @@ mod tests {
}

#[test]
fn test_for_edges() {
fn for_edges() {
let tolerance = Scalar::ONE;

let a = Point::from([1., 2., 3.]);
Expand Down Expand Up @@ -339,8 +243,8 @@ mod tests {
assert_eq!(
Approximation::for_edges(&edges, tolerance),
Approximation {
points: vec![a, b, c, d],
segments: vec![
points: set![a, b, c, d],
segments: set![
Segment::from([a, b]),
Segment::from([b, a]),
Segment::from([c, d]),
Expand All @@ -349,47 +253,4 @@ mod tests {
}
);
}

#[test]
fn test_validate() {
let a = Point::from([0., 1., 2.]);
let b = Point::from([1., 2., 3.]);
let c = Point::from([3., 5., 8.]);

let valid = Approximation {
points: vec![a, b, c],
segments: vec![Segment::from([a, b])],
};
assert!(valid.validate().is_ok());

let duplicate_points = Approximation {
points: vec![a, b, c, b],
segments: vec![Segment::from([a, b])],
};
assert!(duplicate_points.validate().is_err());

let duplicate_segments = Approximation {
points: vec![a, b, c],
segments: vec![Segment::from([a, b]), Segment::from([a, b])],
};
assert!(duplicate_segments.validate().is_err());

let duplicate_segments_inverted = Approximation {
points: vec![a, b, c],
segments: vec![Segment::from([a, b]), Segment::from([b, a])],
};
assert!(duplicate_segments_inverted.validate().is_err());

let invalid_segment = Approximation {
points: vec![a, b, c],
segments: vec![Segment::from([a, a])],
};
assert!(invalid_segment.validate().is_err());

let segment_with_invalid_point = Approximation {
points: vec![a, c],
segments: vec![Segment::from([a, b])],
};
assert!(segment_with_invalid_point.validate().is_err());
}
}
1 change: 0 additions & 1 deletion src/kernel/topology/faces.rs
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,6 @@ impl Face {
match self {
Self::Face { edges, surface } => {
let approx = Approximation::for_edges(edges, tolerance);
approx.validate().expect("Invalid approximation");

let points: Vec<_> = approx
.points
Expand Down
6 changes: 6 additions & 0 deletions src/math/segment.rs
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,12 @@ impl Segment<3> {

impl<const D: usize> From<[Point<D>; 2]> for Segment<D> {
fn from(points: [Point<D>; 2]) -> Self {
let [a, b] = points;

if a == b {
panic!("Invalid segment; both points are identical {a:?}");
}

Self {
a: points[0],
b: points[1],
Expand Down