From 9d36a8cb42d5ea4e25617e77099aadc909356269 Mon Sep 17 00:00:00 2001
From: Carlos Paniagua <cpaniaguam@gmail.com>
Date: Fri, 1 Nov 2024 19:02:10 -0400
Subject: [PATCH] feat: get_ice_masks with tiling

---
 src/tilingutils.jl | 86 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)

diff --git a/src/tilingutils.jl b/src/tilingutils.jl
index c0dfaf9f..2ac0a998 100644
--- a/src/tilingutils.jl
+++ b/src/tilingutils.jl
@@ -366,6 +366,92 @@ function imextendedmin(img, h)
     return imregionalmin(ImageSegmentation.hmin_transform(img, h))
 end
 
+# TODO: Add tests
+"""
+    get_ice_masks(
+    falsecolor_image,
+    morph_residue,
+    landmask::BitMatrix,
+    tiles,
+    binarize::Bool=true;
+    band_7_threshold::T=5,
+    band_2_threshold::T=230,
+    band_1_threshold::T=240,
+    band_7_threshold_relaxed::T=10,
+    band_1_threshold_relaxed::T=190,
+    possible_ice_threshold::T=75,
+    factor::T=255,
+)
+
+Get the ice masks from the falsecolor image and morphological residue given a particualr tiling configuration.
+
+# Arguments
+- `falsecolor_image`: The falsecolor image.
+- `morph_residue`: The morphological residue image.
+- `landmask`: The landmask.
+- `tiles`: The tiles.
+- `binarize::Bool=true`: Whether to binarize the tiling.
+- `band_7_threshold=5`: The threshold for band 7.
+- `band_2_threshold=230`: The threshold for band 2.
+- `band_1_threshold=240`: The threshold for band 1.
+- `band_7_threshold_relaxed=10`: The relaxed threshold for band 7.
+- `band_1_threshold_relaxed=190`: The relaxed threshold for band 1.
+- `possible_ice_threshold=75`: The threshold for possible ice.
+- `factor=255`: The factor to apply to the local maxima mask.
+
+# Returns
+- A named tuple `(icemask, bin)` where:
+  - `icemask`: The ice mask.
+  - `bin`: The binarized tiling.
+"""
+function get_ice_masks(
+    falsecolor_image,
+    morph_residue,
+    landmask::BitMatrix,
+    tiles,
+    binarize::Bool=true;
+    band_7_threshold::T=5,
+    band_2_threshold::T=230,
+    band_1_threshold::T=240,
+    band_7_threshold_relaxed::T=10,
+    band_1_threshold_relaxed::T=190,
+    possible_ice_threshold::T=75,
+    factor::T=255,
+) where {T<:Integer}
+
+    # Make canvases
+    ice_mask = BitMatrix(zeros(Bool, size(falsecolor_image)))
+    if binarize
+        binarized_tiling = zeros(Int, size(falsecolor_image))
+    end
+
+    fc_landmasked = apply_landmask(falsecolor_image, landmask)
+
+    for tile in tiles
+        #  Conditionally update binarized_tiling as its not used in some workflows
+        if binarize
+            binarized_tiling[tile...] .= imbinarize(morph_residue[tile...])
+        end
+
+        morph_residue_seglabels = kmeans_segmentation(Gray.(morph_residue[tile...] / 255))
+
+        floes_label = get_nlabel(
+            fc_landmasked[tile...],
+            morph_residue_seglabels,
+            factor;
+            band_7_threshold=band_7_threshold,
+            band_2_threshold=band_2_threshold,
+            band_1_threshold=band_1_threshold,
+            band_7_threshold_relaxed=band_7_threshold_relaxed,
+            band_1_threshold_relaxed=band_1_threshold_relaxed,
+            possible_ice_threshold=possible_ice_threshold,
+        )
+
+        ice_mask[tile...] .= (morph_residue_seglabels .== floes_label)
+    end
+    return (icemask=ice_mask, bin=binarized_tiling)
+end
+
 function imregionalmin(A, conn=2)
     return ImageMorphology.local_minima(A; connectivity=conn) .> 0
 end