From 34cbe3e143096e8d6890011a0674e54142b224f6 Mon Sep 17 00:00:00 2001
From: "Christopher L. Crutchfield" <ccrutchf@ucsd.edu>
Date: Wed, 19 Jun 2024 15:39:36 -0700
Subject: [PATCH] fix: borrow the mask in fish head tail detector

---
 src/fish/fish_head_tail_detector.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/fish/fish_head_tail_detector.rs b/src/fish/fish_head_tail_detector.rs
index ff033f5..0b74995 100644
--- a/src/fish/fish_head_tail_detector.rs
+++ b/src/fish/fish_head_tail_detector.rs
@@ -34,7 +34,7 @@ pub struct FishHeadTailDetector {
 }
 
 impl FishHeadTailDetector {
-    pub fn find_head_tail(mask: Array2<u8>) -> Result<(Array1<usize>, Array1<usize>), HeadTailError> {
+    pub fn find_head_tail(mask: &Array2<u8>) -> Result<(Array1<usize>, Array1<usize>), HeadTailError> {
         let nonzero: Array1<(i32, i32)> = mask
                 .indexed_iter()
                 .filter_map(|(index, &item)| if item != 0 {Some((index.0 as i32, index.1 as i32))} else { None })
@@ -244,7 +244,7 @@ mod tests {
     fn test() {
         let rust_img = image::io::Reader::open("./data/segmentations.png").unwrap().decode().unwrap().as_luma8().unwrap().clone();
         let mask: ArrayBase<OwnedRepr<u8>, Dim<[usize; 2]>> = Array2::from_shape_vec((rust_img.height() as usize, rust_img.width() as usize), rust_img.as_raw().clone()).unwrap();
-        let res = FishHeadTailDetector::find_head_tail(mask).unwrap();
+        let res = FishHeadTailDetector::find_head_tail(&mask).unwrap();
         assert_eq!(res, (array![1073, 1114], array![2317,1054]));
     }
 }