From 5b3cb257506f3fbe8c5ad3ea8aedb87d51e4c87c Mon Sep 17 00:00:00 2001
From: Christian C <cc@localhost>
Date: Wed, 5 Mar 2025 19:58:33 -0800
Subject: Filter small labels

---
 src/main.c | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/main.c')

diff --git a/src/main.c b/src/main.c
index e199c43..8ee035f 100644
--- a/src/main.c
+++ b/src/main.c
@@ -180,6 +180,7 @@ int main(int argc, char** argv)
   CloseWindow();
 #else
   if (masks != NULL) {
+    // Get the area/ perimeter of each label
     TIME(ts_start);
     for (size_t y = 0; y < height; y++) {
       for (size_t x = 0; x < width; x++) {
@@ -198,6 +199,10 @@ int main(int argc, char** argv)
     print_label(root);
     printf("\n");
 #endif
+    // Get the smallest labels
+    struct AVLNode* small_label_tree = NULL;
+    small_label_tree = get_small_labels(NULL, root, 50, 50);
+    free_avl_tree(small_label_tree);
     free_avl_tree_nodes(root);
     root = NULL;
     write_array("../out.bin", masks, width*height*sizeof(uint16_t));
-- 
cgit v1.2.1