From 879bb46cc292579f6a564d5e3d10952fd8c57c75 Mon Sep 17 00:00:00 2001
From: Weird Constructor <weirdconstructor@gmail.com>
Date: Sat, 23 Jul 2022 13:39:52 +0200
Subject: [PATCH] Changed get_connections API

---
 src/matrix.rs | 24 +++++++++++++++---------
 1 file changed, 15 insertions(+), 9 deletions(-)

diff --git a/src/matrix.rs b/src/matrix.rs
index 84fa965..d5a2280 100644
--- a/src/matrix.rs
+++ b/src/matrix.rs
@@ -979,18 +979,24 @@ impl Matrix {
     }
 
     /// Retrieves the immediate connections to adjacent cells and returns a list.
+    /// Returns none if there is no cell at the given position.
     ///
     /// Returns a vector with pairs of this content:
     ///
     ///    (
-    ///        (this_cell_connection_dir, this_cell_node_io_index),
-    ///        (other_cell_connection_dir, other_cell_node_io_index, (other_cell_x, other_cell_y))
+    ///        (center_cell, center_connection_dir, center_node_io_index),
+    ///        (
+    ///            other_cell,
+    ///            other_connection_dir,
+    ///            other__node_io_index,
+    ///            (other_cell_x, other_cell_y)
+    ///        )
     ///    )
     pub fn get_connections(
         &self,
         x: usize,
         y: usize,
-    ) -> Option<Vec<((CellDir, u8), (CellDir, u8, (usize, usize)))>> {
+    ) -> Option<Vec<((Cell, CellDir, u8), (Cell, CellDir, u8, (usize, usize)))>> {
         let this_cell = self.get(x, y)?;
 
         let mut ret = vec![];
@@ -1007,8 +1013,8 @@ impl Matrix {
                     if let Some(other_cell) = self.get(nx, ny) {
                         if let Some(other_node_io_idx) = other_cell.local_port_idx(dir.flip()) {
                             ret.push((
-                                (dir, node_io_idx),
-                                (dir.flip(), other_node_io_idx, (nx, ny)),
+                                (*this_cell, dir, node_io_idx),
+                                (*other_cell, dir.flip(), other_node_io_idx, (nx, ny)),
                             ));
                         }
                     }
@@ -1407,8 +1413,8 @@ mod tests {
         let res = matrix.get_connections(1, 0);
         let res = res.expect("Found connected cells");
 
-        let (src_dir, src_io_idx) = res[0].0;
-        let (dst_dir, dst_io_idx, (nx, ny)) = res[0].1;
+        let (_src_cell, src_dir, src_io_idx) = res[0].0;
+        let (_dst_cell, dst_dir, dst_io_idx, (nx, ny)) = res[0].1;
 
         assert_eq!(src_dir, CellDir::B, "Found first connection at bottom");
         assert_eq!(src_io_idx, 0, "Correct output port");
@@ -1416,8 +1422,8 @@ mod tests {
         assert_eq!(dst_io_idx, 0, "Correct output port");
         assert_eq!((nx, ny), (1, 1), "Correct other position");
 
-        let (src_dir, src_io_idx) = res[1].0;
-        let (dst_dir, dst_io_idx, (nx, ny)) = res[1].1;
+        let (_src_cell, src_dir, src_io_idx) = res[1].0;
+        let (_dst_cell, dst_dir, dst_io_idx, (nx, ny)) = res[1].1;
 
         assert_eq!(src_dir, CellDir::TL, "Found first connection at bottom");
         assert_eq!(src_io_idx, 0, "Correct output port");