drawing: create `BandIndex` object to pass around bands

src/drawing/band.rs

@@ -0,0 +1,6 @@
+use crate::drawing::seg::{LoneLooseSegIndex, SeqLooseSegIndex};
+
+pub enum BandIndex {
+    Straight(LoneLooseSegIndex),
+    Bended(SeqLooseSegIndex),
+}

src/drawing/drawing.rs

@@ -5,45 +5,36 @@ use geo::Point;
 use rstar::{RTree, RTreeObject, AABB};
 use thiserror::Error;
 
-use super::graph::GetLayer;
-use super::loose::{GetNextLoose, Loose, LooseIndex};
-use super::rules::RulesTrait;
-use super::segbend::Segbend;
-use crate::drawing::bend::BendIndex;
-use crate::drawing::collect::Collect;
-use crate::drawing::dot::DotWeight;
-use crate::drawing::graph::GetMaybeNet;
-use crate::drawing::guide::Guide;
-use crate::drawing::primitive::GetLimbs;
-use crate::drawing::rules::GetConditions;
-use crate::drawing::wraparoundable::{GetWraparound, Wraparoundable, WraparoundableIndex};
 use crate::drawing::{
-    bend::{FixedBendIndex, LooseBendIndex, LooseBendWeight},
-    dot::{DotIndex, FixedDotIndex, FixedDotWeight, LooseDotIndex, LooseDotWeight},
-    graph::{MakePrimitive, PrimitiveIndex, PrimitiveWeight},
+    band::BandIndex,
+    bend::{BendIndex, BendWeight, FixedBendIndex, LooseBendIndex, LooseBendWeight},
+    collect::Collect,
+    dot::{DotIndex, DotWeight, FixedDotIndex, FixedDotWeight, LooseDotIndex, LooseDotWeight},
+    graph::{GetLayer, GetMaybeNet, MakePrimitive, PrimitiveIndex, PrimitiveWeight},
+    guide::Guide,
+    loose::{GetNextLoose, Loose, LooseIndex},
     primitive::{
-        GenericPrimitive, GetCore, GetInnerOuter, GetJoints, GetOtherJoint, MakePrimitiveShape,
+        GenericPrimitive, GetCore, GetInnerOuter, GetJoints, GetLimbs, GetOtherJoint,
+        MakePrimitiveShape,
     },
+    rules::{GetConditions, RulesTrait},
     seg::{
-        FixedSegIndex, FixedSegWeight, LoneLooseSegIndex, LoneLooseSegWeight, SegIndex,
+        FixedSegIndex, FixedSegWeight, LoneLooseSegIndex, LoneLooseSegWeight, SegIndex, SegWeight,
         SeqLooseSegIndex, SeqLooseSegWeight,
     },
+    segbend::Segbend,
+    wraparoundable::{GetWraparound, Wraparoundable, WraparoundableIndex},
 };
-use crate::geometry::compound::CompoundManagerTrait;
-use crate::geometry::with_rtree::BboxedIndex;
-use crate::geometry::GenericNode;
 use crate::geometry::{
+    compound::CompoundManagerTrait,
     primitive::{PrimitiveShape, PrimitiveShapeTrait},
-    with_rtree::GeometryWithRtree,
-    BendWeightTrait, DotWeightTrait, Geometry, GeometryLabel, GetOffset, GetPos, GetWidth,
-    SegWeightTrait,
+    with_rtree::{BboxedIndex, GeometryWithRtree},
+    BendWeightTrait, DotWeightTrait, GenericNode, Geometry, GeometryLabel, GetOffset, GetPos,
+    GetWidth, SegWeightTrait,
 };
 use crate::graph::{GenericIndex, GetNodeIndex};
 use crate::math::NoTangents;
 
-use super::bend::BendWeight;
-use super::seg::SegWeight;
-
 #[enum_dispatch]
 #[derive(Error, Debug, Clone, Copy)]
 pub enum LayoutException {
@@ -94,13 +85,18 @@ impl<CW: Copy, R: RulesTrait> Drawing<CW, R> {
         }
     }
 
-    pub fn remove_band(&mut self, first_loose: SeqLooseSegIndex) {
+    pub fn remove_band(&mut self, band: BandIndex) {
+        match band {
+            BandIndex::Straight(seg) => {
+                self.geometry_with_rtree.remove_seg(seg.into());
+            }
+            BandIndex::Bended(first_loose_seg) => {
                 let mut dots = vec![];
                 let mut segs = vec![];
                 let mut bends = vec![];
                 let mut outers = vec![];
 
-        let mut maybe_loose = Some(first_loose.into());
+                let mut maybe_loose = Some(first_loose_seg.into());
                 let mut prev = None;
 
                 while let Some(loose) = maybe_loose {
@@ -149,6 +145,8 @@ impl<CW: Copy, R: RulesTrait> Drawing<CW, R> {
                     self.update_this_and_outward_bows(outer).unwrap(); // Must never fail.
                 }
             }
+        }
+    }
 
     #[debug_ensures(ret.is_ok() -> self.geometry_with_rtree.graph().node_count() == old(self.geometry_with_rtree.graph().node_count() + 1))]
     #[debug_ensures(ret.is_err() -> self.geometry_with_rtree.graph().node_count() == old(self.geometry_with_rtree.graph().node_count()))]

src/drawing/mod.rs

@@ -1,5 +1,6 @@
 #[macro_use]
 pub mod graph;
+pub mod band;
 pub mod bend;
 pub mod collect;
 pub mod dot;