use crate::{
Angle, Bounds, Distance, HashablePt2D, InfiniteLine, Line, Polygon, Pt2D, Ring, EPSILON_DIST,
};
use serde::{Deserialize, Serialize};
use std::collections::HashSet;
use std::fmt;
const MITER_THRESHOLD: f64 = 500.0;
pub enum ArrowCap {
Triangle,
}
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
pub struct PolyLine {
pts: Vec<Pt2D>,
length: Distance,
}
impl PolyLine {
pub fn new(pts: Vec<Pt2D>) -> Result<PolyLine, String> {
if pts.len() < 2 {
return Err(format!("Need at least two points for a PolyLine"));
}
let length = pts.windows(2).fold(Distance::ZERO, |so_far, pair| {
so_far + pair[0].dist_to(pair[1])
});
if pts.windows(2).any(|pair| pair[0] == pair[1]) {
return Err(format!(
"PL with total length {} and {} pts has ~dupe adjacent pts",
length,
pts.len(),
));
}
let result = PolyLine { pts, length };
let (_, dupes) = to_set(result.points());
if !dupes.is_empty() {
return Err(format!(
"PL with total length {} and {} pts has dupe non-adjacent pts",
result.length,
result.pts.len(),
));
}
Ok(result)
}
pub fn must_new(pts: Vec<Pt2D>) -> PolyLine {
PolyLine::new(pts).unwrap()
}
pub fn unchecked_new(pts: Vec<Pt2D>) -> PolyLine {
assert!(pts.len() >= 2);
let length = pts.windows(2).fold(Distance::ZERO, |so_far, pair| {
so_far + pair[0].dist_to(pair[1])
});
PolyLine { pts, length }
}
pub fn deduping_new(mut pts: Vec<Pt2D>) -> Result<PolyLine, String> {
pts.dedup();
PolyLine::new(pts)
}
pub fn to_thick_boundary(
&self,
self_width: Distance,
boundary_width: Distance,
) -> Option<Polygon> {
if self_width <= boundary_width || self.length() <= boundary_width + EPSILON_DIST {
return None;
}
let slice = self.exact_slice(boundary_width / 2.0, self.length() - boundary_width / 2.0);
let mut side1 =
slice.shift_with_sharp_angles((self_width - boundary_width) / 2.0, MITER_THRESHOLD);
let mut side2 =
slice.shift_with_sharp_angles(-(self_width - boundary_width) / 2.0, MITER_THRESHOLD);
side2.reverse();
side1.extend(side2);
side1.push(side1[0]);
side1.dedup();
Some(Ring::must_new(side1).to_outline(boundary_width))
}
pub fn reversed(&self) -> PolyLine {
let mut pts = self.pts.clone();
pts.reverse();
PolyLine::must_new(pts)
}
pub fn extend(self, other: PolyLine) -> Result<PolyLine, String> {
if *self.pts.last().unwrap() != other.pts[0] {
return Err(format!(
"can't extend PL; last and first points don't match"
));
}
let mut self_pts = self.pts;
let mut other_pts = other.pts;
loop {
let (pl1, _) = to_set(&self_pts);
let (pl2, _) = to_set(&other_pts[1..]);
if pl1.intersection(&pl2).next().is_some() {
for (other_rev_idx, pt) in other_pts.iter().rev().enumerate() {
if pl1.contains(&pt.to_hashable()) {
while self_pts.last().unwrap() != pt {
self_pts.pop();
}
other_pts = other_pts[other_pts.len() - 1 - other_rev_idx..].to_vec();
break;
}
}
assert_eq!(*self_pts.last().unwrap(), other_pts[0]);
} else {
break;
}
}
if other_pts.len() >= 2 {
let same_line = self_pts[self_pts.len() - 2]
.angle_to(self_pts[self_pts.len() - 1])
.approx_eq(other_pts[0].angle_to(other_pts[1]), 0.1);
if same_line {
self_pts.pop();
}
}
self_pts.extend(other_pts.iter().skip(1));
PolyLine::new(self_pts)
}
pub fn must_extend(self, other: PolyLine) -> PolyLine {
self.extend(other).unwrap()
}
pub fn must_push(self, pt: Pt2D) -> PolyLine {
let new = PolyLine::must_new(vec![self.last_pt(), pt]);
self.must_extend(new)
}
pub fn append(first: Vec<Pt2D>, second: Vec<Pt2D>) -> Result<Vec<Pt2D>, String> {
if second.is_empty() {
return Ok(first);
}
if first.is_empty() {
return Ok(second);
}
Ok(PolyLine::new(first)?
.extend(PolyLine::new(second)?)?
.into_points())
}
pub fn points(&self) -> &Vec<Pt2D> {
&self.pts
}
pub fn into_points(self) -> Vec<Pt2D> {
self.pts
}
pub fn lines(&self) -> Vec<Line> {
self.pts
.windows(2)
.map(|pair| Line::must_new(pair[0], pair[1]))
.collect()
}
pub fn length(&self) -> Distance {
self.length
}
pub fn slice(&self, start: Distance, end: Distance) -> Result<(PolyLine, Distance), String> {
if start > end || start < Distance::ZERO || end < Distance::ZERO {
return Err(format!("Can't get a polyline slice [{}, {}]", start, end));
}
if start > self.length() {
return Err(format!(
"Can't get a polyline slice [{}, {}] on something of length {}",
start,
end,
self.length()
));
}
if end - start < EPSILON_DIST {
return Err(format!(
"Can't get a polyline slice [{}, {}] -- too small",
start, end
));
}
let mut result: Vec<Pt2D> = Vec::new();
let mut dist_so_far = Distance::ZERO;
for line in self.lines().iter() {
let length = line.length();
if result.is_empty() && dist_so_far + length >= start {
result.push(line.must_dist_along(start - dist_so_far));
}
if dist_so_far + length >= end {
let last_pt = line.must_dist_along(end - dist_so_far);
if *result.last().unwrap() == last_pt {
result.pop();
}
result.push(last_pt);
if result.len() == 1 {
return Err(format!(
"slice({}, {}) on {} did something weird",
start, end, self
));
}
return Ok((PolyLine::new(result)?, Distance::ZERO));
}
if !result.is_empty() && *result.last().unwrap() != line.pt2() {
result.push(line.pt2());
}
dist_so_far += length;
}
if result.is_empty() {
return Err(format!(
"Slice [{}, {}] has a start too big for polyline of length {}",
start,
end,
self.length()
));
}
if result.len() == 1 {
return Err(format!(
"Slice [{}, {}] on {} wound up a single point",
start, end, self
));
}
Ok((PolyLine::new(result)?, end - dist_so_far))
}
pub fn exact_slice(&self, start: Distance, end: Distance) -> PolyLine {
self.maybe_exact_slice(start, end).unwrap()
}
pub fn maybe_exact_slice(&self, start: Distance, end: Distance) -> Result<PolyLine, String> {
let (pl, leftover) = self
.slice(start, end)
.map_err(|_| format!("exact_slice({}, {}) yielded empty slice", start, end))?;
if leftover > EPSILON_DIST {
return Err(format!(
"exact_slice({}, {}) on a PL of length {} yielded leftover distance of {}",
start,
end,
self.length(),
leftover
));
}
Ok(pl)
}
pub fn first_half(&self) -> PolyLine {
self.exact_slice(Distance::ZERO, self.length() / 2.0)
}
pub fn second_half(&self) -> PolyLine {
self.exact_slice(self.length() / 2.0, self.length())
}
pub fn dist_along(&self, dist_along: Distance) -> Result<(Pt2D, Angle), String> {
if dist_along < Distance::ZERO {
return Err(format!("dist_along {} is negative", dist_along));
}
if dist_along > self.length() {
return Err(format!(
"dist_along {} is longer than {}",
dist_along,
self.length()
));
}
let mut dist_left = dist_along;
let mut length_remeasured = Distance::ZERO;
for (idx, l) in self.lines().iter().enumerate() {
let length = l.length();
length_remeasured += length;
let epsilon = if idx == self.pts.len() - 2 {
EPSILON_DIST
} else {
Distance::ZERO
};
if dist_left <= length + epsilon {
return Ok((l.must_dist_along(dist_left), l.angle()));
}
dist_left -= length;
}
panic!(
"PolyLine dist_along of {} broke on length {} (recalculated length {}): {}",
dist_along,
self.length(),
length_remeasured,
self
);
}
pub fn must_dist_along(&self, dist_along: Distance) -> (Pt2D, Angle) {
self.dist_along(dist_along).unwrap()
}
pub fn middle(&self) -> Pt2D {
match self.dist_along(self.length() / 2.0) {
Ok((pt, _)) => pt,
Err(err) => {
println!(
"Guessing middle of PL with length {}: {}",
self.length(),
err
);
self.first_pt()
}
}
}
pub fn first_pt(&self) -> Pt2D {
self.pts[0]
}
pub fn last_pt(&self) -> Pt2D {
*self.pts.last().unwrap()
}
pub fn first_line(&self) -> Line {
Line::must_new(self.pts[0], self.pts[1])
}
pub fn last_line(&self) -> Line {
Line::must_new(self.pts[self.pts.len() - 2], self.pts[self.pts.len() - 1])
}
pub fn shift_right(&self, width: Distance) -> Result<PolyLine, String> {
self.shift_with_corrections(width)
}
pub fn must_shift_right(&self, width: Distance) -> PolyLine {
self.shift_right(width).unwrap()
}
pub fn shift_left(&self, width: Distance) -> Result<PolyLine, String> {
self.shift_with_corrections(-width)
}
pub fn must_shift_left(&self, width: Distance) -> PolyLine {
self.shift_left(width).unwrap()
}
fn shift_with_corrections(&self, width: Distance) -> Result<PolyLine, String> {
let raw = self.shift_with_sharp_angles(width, MITER_THRESHOLD);
let result = PolyLine::deduping_new(raw)?;
if result.pts.len() == self.pts.len() {
fix_angles(self, result)
} else {
Ok(result)
}
}
fn shift_with_sharp_angles(&self, width: Distance, miter_threshold: f64) -> Vec<Pt2D> {
if self.pts.len() == 2 {
let l = Line::must_new(self.pts[0], self.pts[1]).shift_either_direction(width);
return vec![l.pt1(), l.pt2()];
}
let mut result: Vec<Pt2D> = Vec::new();
let mut pt3_idx = 2;
let mut pt1_raw = self.pts[0];
let mut pt2_raw = self.pts[1];
loop {
let pt3_raw = self.pts[pt3_idx];
let l1 = Line::must_new(pt1_raw, pt2_raw).shift_either_direction(width);
let l2 = Line::must_new(pt2_raw, pt3_raw).shift_either_direction(width);
if pt3_idx == 2 {
result.push(l1.pt1());
}
if let Some(pt2_shift) = l1.infinite().intersection(&l2.infinite()) {
let dist_away = l1.pt1().raw_dist_to(pt2_shift);
if dist_away < miter_threshold {
result.push(pt2_shift);
} else {
result.push(l1.pt2());
}
} else {
result.push(l1.pt2());
}
if pt3_idx == self.pts.len() - 1 {
result.push(l2.pt2());
break;
}
pt1_raw = pt2_raw;
pt2_raw = pt3_raw;
pt3_idx += 1;
}
assert!(result.len() == self.pts.len());
result
}
pub fn make_polygons(&self, width: Distance) -> Polygon {
self.make_polygons_with_miter_threshold(width, MITER_THRESHOLD)
}
pub fn make_polygons_with_miter_threshold(
&self,
width: Distance,
miter_threshold: f64,
) -> Polygon {
let side1 = self.shift_with_sharp_angles(width / 2.0, miter_threshold);
let side2 = self.shift_with_sharp_angles(-width / 2.0, miter_threshold);
assert_eq!(side1.len(), side2.len());
let side2_offset = side1.len();
let mut points = side1;
points.extend(side2);
let mut indices = Vec::new();
for high_idx in 1..self.pts.len() {
indices.extend(vec![high_idx, high_idx - 1, side2_offset + high_idx - 1]);
indices.extend(vec![
side2_offset + high_idx,
side2_offset + high_idx - 1,
high_idx,
]);
}
Polygon::precomputed(points, indices)
}
pub fn exact_dashed_polygons(
&self,
width: Distance,
dash_len: Distance,
dash_separation: Distance,
) -> Vec<Polygon> {
let mut polygons: Vec<Polygon> = Vec::new();
let total_length = self.length();
let mut start = Distance::ZERO;
loop {
if start + dash_len >= total_length {
break;
}
polygons.push(
self.exact_slice(start, start + dash_len)
.make_polygons(width),
);
start += dash_len + dash_separation;
}
polygons
}
pub fn dashed_lines(
&self,
width: Distance,
dash_len: Distance,
dash_separation: Distance,
) -> Vec<Polygon> {
if self.length() <= dash_separation * 2.0 + EPSILON_DIST {
return vec![self.make_polygons(width)];
}
self.exact_slice(dash_separation, self.length() - dash_separation)
.exact_dashed_polygons(width, dash_len, dash_separation)
}
pub fn make_arrow(&self, thickness: Distance, cap: ArrowCap) -> Polygon {
let head_size = thickness * 2.0;
let triangle_height = head_size / 2.0_f64.sqrt();
if self.length() < triangle_height + EPSILON_DIST {
return self.make_polygons(thickness);
}
let slice = self.exact_slice(Distance::ZERO, self.length() - triangle_height);
let angle = slice.last_pt().angle_to(self.last_pt());
let corner1 = self
.last_pt()
.project_away(head_size, angle.rotate_degs(-135.0));
let corner2 = self
.last_pt()
.project_away(head_size, angle.rotate_degs(135.0));
let mut pts = slice.shift_with_sharp_angles(thickness / 2.0, MITER_THRESHOLD);
match cap {
ArrowCap::Triangle => {
pts.push(corner2);
pts.push(self.last_pt());
pts.push(corner1);
}
}
let mut side2 = slice.shift_with_sharp_angles(-thickness / 2.0, MITER_THRESHOLD);
side2.reverse();
pts.extend(side2);
pts.push(pts[0]);
pts.dedup();
Ring::must_new(pts).to_polygon()
}
pub fn make_double_arrow(&self, thickness: Distance, cap: ArrowCap) -> Polygon {
let head_size = thickness * 2.0;
let triangle_height = head_size / 2.0_f64.sqrt();
if self.length() < triangle_height * 2.0 + EPSILON_DIST {
return self.make_polygons(thickness);
}
let slice = self.exact_slice(triangle_height, self.length() - triangle_height);
let angle = slice.last_pt().angle_to(self.last_pt());
let corner1 = self
.last_pt()
.project_away(head_size, angle.rotate_degs(-135.0));
let corner2 = self
.last_pt()
.project_away(head_size, angle.rotate_degs(135.0));
let mut pts = slice.shift_with_sharp_angles(thickness / 2.0, MITER_THRESHOLD);
match cap {
ArrowCap::Triangle => {
pts.push(corner2);
pts.push(self.last_pt());
pts.push(corner1);
}
}
let mut side2 = slice.shift_with_sharp_angles(-thickness / 2.0, MITER_THRESHOLD);
side2.reverse();
pts.extend(side2);
let angle = self.first_pt().angle_to(slice.first_pt());
let corner3 = self
.first_pt()
.project_away(head_size, angle.rotate_degs(-45.0));
let corner4 = self
.first_pt()
.project_away(head_size, angle.rotate_degs(45.0));
match cap {
ArrowCap::Triangle => {
pts.push(corner3);
pts.push(self.first_pt());
pts.push(corner4);
}
}
pts.push(pts[0]);
pts.dedup();
Ring::must_new(pts).to_polygon()
}
pub fn dashed_arrow(
&self,
width: Distance,
dash_len: Distance,
dash_separation: Distance,
cap: ArrowCap,
) -> Vec<Polygon> {
let mut polygons = self.exact_dashed_polygons(width, dash_len, dash_separation);
let last_line = self.last_line();
let last_len = last_line.length();
let arrow_line = if last_len <= dash_len {
last_line
} else {
Line::must_new(
last_line.must_dist_along(last_len - dash_len),
last_line.pt2(),
)
};
polygons.push(arrow_line.to_polyline().make_arrow(width, cap));
polygons
}
pub fn intersection(&self, other: &PolyLine) -> Option<(Pt2D, Angle)> {
assert_ne!(self, other);
for l1 in self.lines() {
let mut hits: Vec<(Pt2D, Angle)> = Vec::new();
for l2 in other.lines() {
if let Some(pt) = l1.intersection(&l2) {
hits.push((pt, l1.angle()));
}
}
if let Some(hit) = hits.into_iter().min_by_key(|(pt, _)| {
self.get_slice_ending_at(*pt)
.map(|pl| pl.length())
.unwrap_or(Distance::ZERO)
}) {
return Some(hit);
}
}
None
}
pub fn intersection_infinite(&self, other: &InfiniteLine) -> Option<Pt2D> {
for l in self.lines() {
if let Some(hit) = l.intersection_infinite(other) {
return Some(hit);
}
}
None
}
pub fn get_slice_ending_at(&self, pt: Pt2D) -> Option<PolyLine> {
if self.first_pt() == pt {
return None;
}
if let Some(idx) = self.lines().iter().position(|l| l.contains_pt(pt)) {
let mut pts = self.pts.clone();
pts.truncate(idx + 1);
if *pts.last().unwrap() == pt {
pts.pop();
}
pts.push(pt);
if pts.len() == 1 {
return None;
}
Some(PolyLine::must_new(pts))
} else {
panic!("Can't get_slice_ending_at: {} doesn't contain {}", self, pt);
}
}
pub fn get_slice_starting_at(&self, pt: Pt2D) -> Option<PolyLine> {
if self.last_pt() == pt {
return None;
}
if let Some(idx) = self.lines().iter().position(|l| l.contains_pt(pt)) {
let mut pts = self.pts.clone();
pts = pts.split_off(idx + 1);
if pt != pts[0] {
pts.insert(0, pt);
}
Some(PolyLine::must_new(pts))
} else {
panic!(
"Can't get_slice_starting_at: {} doesn't contain {}",
self, pt
);
}
}
pub fn dist_along_of_point(&self, pt: Pt2D) -> Option<(Distance, Angle)> {
let mut dist_along = Distance::ZERO;
for l in self.lines() {
if let Some(dist) = l.dist_along_of_point(pt) {
return Some((dist_along + dist, l.angle()));
} else {
dist_along += l.length();
}
}
None
}
pub fn trim_to_endpts(&self, pt1: Pt2D, pt2: Pt2D) -> PolyLine {
assert!(pt1 != pt2);
let mut dist1 = self.dist_along_of_point(pt1).unwrap().0;
let mut dist2 = self.dist_along_of_point(pt2).unwrap().0;
if dist1 > dist2 {
std::mem::swap(&mut dist1, &mut dist2);
}
self.exact_slice(dist1, dist2)
}
pub fn get_bounds(&self) -> Bounds {
Bounds::from(&self.pts)
}
pub fn extend_to_length(&self, min_len: Distance) -> PolyLine {
let need_len = min_len - self.length();
if need_len <= Distance::ZERO {
return self.clone();
}
let line = self.last_line();
if let Ok(extension) = PolyLine::new(vec![
line.pt2(),
line.pt2().project_away(need_len, line.angle()),
]) {
self.clone().must_extend(extension)
} else {
let mut pts = self.clone().into_points();
pts.pop();
pts.push(line.pt2().project_away(need_len, line.angle()));
PolyLine::must_new(pts)
}
}
}
impl fmt::Display for PolyLine {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
writeln!(f, "PolyLine::new(vec![ // length {}", self.length)?;
for (idx, pt) in self.pts.iter().enumerate() {
write!(f, " Pt2D::new({}, {}),", pt.x(), pt.y())?;
if idx > 0 {
let line = Line::must_new(self.pts[idx - 1], *pt);
write!(
f,
" // {}, {} (+ {} @ {})",
pt.x() - self.pts[idx - 1].x(),
pt.y() - self.pts[idx - 1].y(),
line.length(),
line.angle(),
)?;
}
writeln!(f)?;
}
write!(f, "])")
}
}
fn fix_angles(orig: &PolyLine, result: PolyLine) -> Result<PolyLine, String> {
let mut pts = result.pts.clone();
for (idx, (orig_l, shifted_l)) in orig.lines().iter().zip(result.lines().iter()).enumerate() {
let orig_angle = orig_l.angle();
let shifted_angle = shifted_l.angle();
if !orig_angle.approx_eq(shifted_angle, 1.0) {
pts.swap(idx, idx + 1);
}
}
PolyLine::new(pts)
}
fn to_set(pts: &[Pt2D]) -> (HashSet<HashablePt2D>, HashSet<HashablePt2D>) {
let mut deduped = HashSet::new();
let mut dupes = HashSet::new();
for pt in pts {
let pt = pt.to_hashable();
if deduped.contains(&pt) {
dupes.insert(pt);
} else {
deduped.insert(pt);
}
}
(deduped, dupes)
}