From c2a41f2c2e3593ae028f1a440d4066077b5fa261 Mon Sep 17 00:00:00 2001
From: Christian C <cc@localhost>
Date: Wed, 5 Mar 2025 22:52:40 -0800
Subject: Two more arguments

---
 src/main.c | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

(limited to 'src/main.c')

diff --git a/src/main.c b/src/main.c
index 41f9b85..5bdcadb 100644
--- a/src/main.c
+++ b/src/main.c
@@ -286,11 +286,21 @@ int main(int argc, char** argv)
 #else
   if (masks != NULL) {
     struct bitmap_t* bitmap = uint16_to_bitmap(masks, width, height);
+    char* png_name = "../out.png";
+    char* bin_name = "../out.bin";
     if (bitmap != NULL) {
-      save_png(bitmap, "../out.png");
+      if (argc > 3) {
+	save_png(bitmap, argv[3]);
+      } else {
+	save_png(bitmap, png_name);
+      }
       free(bitmap);
     }
-    write_array("../out.bin", masks, width*height*sizeof(uint16_t));
+    if (argc > 2) {
+      write_array(argv[2], masks, width*height*sizeof(uint16_t));
+    } else {
+      write_array(bin_name, masks, width*height*sizeof(uint16_t));
+    }
     free(masks);
   }
 #endif
-- 
cgit v1.2.1