Migration to Mask

1 files changed, 4 insertions, 1 deletions

diff --git a/lib/seg/util.c b/lib/seg/util.c
index bc2f837..7b8d317 100644
--- a/lib/seg/util.c
+++ b/lib/seg/util.c
@@ -283,8 +283,11 @@ Mask *tif_to_labels(char *tif_file_name, MaskData_t *starting_label_p)
 }
 
 // Convert mask to bitmap
-Bitmap* image_mask_data_to_bitmap(MaskData_t* buffer, uint32_t width, uint32_t height)
+Bitmap* image_mask_data_to_bitmap(const Mask* mask)
 {
+  MaskData_t *buffer = mask->image[0];
+  uint32_t width = mask->width;
+  uint32_t height = mask->height;
   Pixel* out_buffer = (Pixel*)calloc(width*height, sizeof(Pixel));
   if (out_buffer == NULL) {
     return NULL;