Two more arguments

1 files changed, 12 insertions, 2 deletions

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