Added counter to draw loop to show the amount of nodes computed

1 files changed, 18 insertions, 0 deletions

diff --git a/src/main.cpp b/src/main.cpp
index 91610db..6f81d5b 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -111,6 +111,13 @@ int main(void)
   Fraction best_frac{1, 2};
   word_t root = allocator.alloc({best_frac});
   to_iterate.push(root);
+
+  word_t count      = 1;
+  word_t prev_count = 0;
+  std::stringstream count_format;
+  std::string count_str;
+  word_t count_str_width = 0;
+
   InitWindow(WIDTH, HEIGHT, "Calkin-Wilf Trees");
   while (!WindowShouldClose())
   {
@@ -119,9 +126,20 @@ int main(void)
       iterate(to_iterate, allocator);
       count += 2;
     }
+    if (prev_count != count)
+    {
+      prev_count = count;
+      count_format << "Count=" << count;
+      count_str = count_format.str();
+      count_format.str("");
+      count_str_width = MeasureText(count_str.c_str(), FONT_SIZE * 2);
+    }
+
     ClearBackground(BLACK);
     BeginDrawing();
     draw_number_line(allocator);
+    DrawText(count_str.c_str(), WIDTH / 2 - count_str_width / 2,
+             LINE_TOP - HEIGHT / 8, FONT_SIZE * 2, WHITE);
     EndDrawing();
   }
   CloseWindow();