layout: Return an iterator over nodes instead of dots

It's more general. Needed because mesh will be also generated from fixed
and loose bends.

src/layout.rs

@@ -356,20 +356,25 @@ impl Layout {
         Ok(())
     }
 
-    pub fn dots(&self) -> impl Iterator<Item = FixedDotIndex> + '_ {
+    pub fn nodes(&self) -> impl Iterator<Item = Index> + '_ {
-        self.nodes()
+        self.node_indexes().map(|ni| {
-            .filter_map(|ni| ni.as_fixed_dot().map(|di| *di))
+            self.graph
+                .node_weight(ni.node_index())
+                .unwrap()
+                .retag(ni.node_index())
+        })
     }
 
     pub fn shapes(&self) -> impl Iterator<Item = Shape> + '_ {
-        self.nodes().map(|ni| ni.primitive(&self.graph).shape())
+        self.node_indexes()
+            .map(|ni| ni.primitive(&self.graph).shape())
     }
 
     pub fn node_count(&self) -> usize {
         self.graph.node_count()
     }
 
-    fn nodes(&self) -> impl Iterator<Item = Index> + '_ {
+    fn node_indexes(&self) -> impl Iterator<Item = Index> + '_ {
         self.rtree.iter().map(|wrapper| wrapper.data)
     }
 }

src/mesh.rs

@@ -8,7 +8,7 @@ use spade::{
 };
 
 use crate::{
-    graph::{DotIndex, FixedDotIndex, GetNodeIndex},
+    graph::{DotIndex, FixedDotIndex, GetNodeIndex, Index},
     layout::Layout,
 };
 use crate::{primitive::MakeShape, shape::ShapeTrait};
@@ -51,7 +51,8 @@ impl Mesh {
         self.dot_to_vertex = Vec::new();
         self.dot_to_vertex.resize(layout.graph.node_bound(), None);
 
-        for dot in layout.dots() {
+        for node in layout.nodes() {
+            if let Index::FixedDot(dot) = node {
                 let center = layout.primitive(dot).shape().center();
 
                 self.dot_to_vertex[dot.node_index().index()] = Some(VertexIndex {
@@ -62,6 +63,7 @@ impl Mesh {
                     })?,
                 });
             }
+        }
 
         Ok(())
     }