1 files changed, 1 insertions, 30 deletions
diff --git a/main.c b/main.c
index c93660e..dd8a1ac 100644
--- a/main.c
+++ b/main.c
@@ -13,8 +13,6 @@
 #include <raylib.h>
 #include <raymath.h>
 
-#include <stb/stb_image_write.h>
-
 #include "./lib.h"
 
 struct StepArg
@@ -150,34 +148,7 @@ int main(void)
   }
 
   CloseWindow();
-
-  unsigned char *image_data =
-      calloc(3 * state.dwidth * state.dwidth, sizeof(*image_data));
-
-  size_t image_ptr = 0;
-
-  for (size_t i = 0; i < state.dwidth; ++i)
-    for (size_t j = 0; j < state.dwidth; ++j, image_ptr += 3)
-    {
-      Color c           = {0};
-      uint64_t sandpile = state.data[(i * state.dwidth) + j];
-      if (sandpile == 0)
-        c = BLACK;
-      else if (sandpile == 1)
-        c = MAGENTA;
-      else if (sandpile == 2)
-        c = RED;
-      else if (sandpile == 3)
-        c = BLUE;
-
-      image_data[image_ptr]     = c.r;
-      image_data[image_ptr + 1] = c.g;
-      image_data[image_ptr + 2] = c.b;
-    }
-
-  stbi_write_png("data.png", state.dwidth, state.dwidth, 3, image_data,
-                 3 * state.dwidth);
-  free(image_data);
+  write_to_png(&state, "data.png");
   free(state.data);
   return 0;
 }