Create output file name from input when output option is not specified

ansilove · Jul 26, 2015 · 775fe09 · 775fe09
1 parent e8652b2
commit 775fe09
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/main.c b/src/main.c
@@ -121,7 +121,7 @@ int main(int argc, char *argv[])
     char *font;
     char *icecolors = NULL; // TODO : Re-enable icecolors
 
-    char *input, *output;
+    char *input, *output = NULL;
     char *retinaout = NULL;
 
     while ((getoptFlag = getopt(argc, argv, "b:c:ef:hi:o:rsv")) != -1) {
@@ -162,6 +162,13 @@ int main(int argc, char *argv[])
     argc -= optind; 
     argv += optind;
 
+    // create output file name if output is not specified
+    if (!output) {        
+        int outputLen = strlen(input) + 5;
+        output = malloc(outputLen);
+        snprintf(output, outputLen, "%s%s", input, ".png");
+    }
+
     // get file extension
     char *fext = strrchr(input, '.');
     fext = fext ? strtolower(fext) : "none";