Timing macro

1 files changed, 6 insertions, 1 deletions

diff --git a/src/main.c b/src/main.c
index 02f5520..abfc57f 100644
--- a/src/main.c
+++ b/src/main.c
@@ -13,6 +13,8 @@
 
 #define OFFSET 16
 
+#define N_DILATIONS 3
+
 int main(int argc, char** argv)
 {
   //-----------------------------------------------
@@ -54,11 +56,14 @@ int main(int argc, char** argv)
   }
   printf("%u labels found\n", starting_label-1);
   printf("Mask dimensions: %u %u\n", width, height);
-  for (uint16_t count = 0; count < 3; count++) {
+  TIME(ts_start);
+  for (uint16_t count = 0; count < N_DILATIONS; count++) {
     uint16_t *new_labels = dilate(masks, width, height);
     free(masks);
     masks = new_labels;
   }
+  TIME(ts_end);
+  printf("Dilation took %f ms\n", 1000*diff_time(&ts_end, &ts_start));
   //free(masks);
   //-----------------------------------------------