Fix closeup logic

1 files changed, 5 insertions, 8 deletions

diff --git a/src/lib/seg/util.c b/src/lib/seg/util.c
index 8a5f5e5..677e8f5 100644
--- a/src/lib/seg/util.c
+++ b/src/lib/seg/util.c
@@ -115,39 +115,36 @@ void dilate(uint16_t** mask, uint32_t width, uint32_t height)
 uint16_t* _erode(uint16_t* mask, uint32_t width, uint32_t height)
 {
   uint16_t *new_mask = (uint16_t*)calloc(width*height,sizeof(uint16_t));
+  memcpy(new_mask, mask, width*height*sizeof(uint16_t));
   for (size_t y = 0; y < height; y++) {
     for (size_t x = 0; x < width; x++) {
       size_t current_position = xy_to_coord(x, y, width, height);
-      if (mask[current_position] != 0) {
-	new_mask[current_position] = mask[current_position];
-	continue;
-      }
       size_t proposed_position;
       if (x != 0) {
 	proposed_position = xy_to_coord(x-1, y, width, height);
 	if (mask[proposed_position] == 0) {
-	  new_mask[current_position] = mask[proposed_position];
+	  new_mask[current_position] = 0;
 	  continue;
 	}
       }
       if ((x+1) != width) {
 	proposed_position = xy_to_coord(x+1, y, width, height);
 	if (mask[proposed_position] == 0) {
-	  new_mask[current_position] = mask[proposed_position];
+	  new_mask[current_position] = 0;
 	  continue;
 	}
       }
       if (y != 0) {
 	proposed_position = xy_to_coord(x, y-1, width, height);
 	if (mask[proposed_position] == 0) {
-	  new_mask[current_position] = mask[proposed_position];
+	  new_mask[current_position] = 0;
 	  continue;
 	}
       }
       if ((y+1) != height) {
 	proposed_position = xy_to_coord(x, y+1, width, height);
 	if (mask[proposed_position] == 0) {
-	  new_mask[current_position] = mask[proposed_position];
+	  new_mask[current_position] = 0;
 	  continue;
 	}
       }