gpujpegtool: dump input image in debug if not file

Debug dump "special" inputs like test patter (1x1.tst) or special devices
(/dev/random). File name is `n=$(basename $in_name); echo
${n%.*}.pnm`. Regular input files are not dumped.

Author: MartinPulec

libgpujpeg/gpujpeg_common.h

@@ -417,7 +417,8 @@ gpujpeg_image_load_from_file(const char* filename, uint8_t** image, size_t* imag
  * @return 0 if succeeds, otherwise nonzero
  */
 GPUJPEG_API int
-gpujpeg_image_save_to_file(char* filename, uint8_t* image, size_t image_size, const struct gpujpeg_image_parameters *param_image);
+gpujpeg_image_save_to_file(char* filename, const uint8_t* image, size_t image_size,
+                           const struct gpujpeg_image_parameters* param_image);
 
 /**
  * Reads/obtains properties from uncompressed file (PNM etc.)

src/gpujpeg_common.c

@@ -1211,8 +1211,8 @@ set_file_extension(char *filename,
 
 /* Documented at declaration */
 int
-gpujpeg_image_save_to_file(char *filename, uint8_t *image, size_t image_size,
-                           const struct gpujpeg_image_parameters *param_image)
+gpujpeg_image_save_to_file(char* filename, const uint8_t* image, size_t image_size,
+                           const struct gpujpeg_image_parameters* param_image)
 {
     if (strrchr(filename, '.') != NULL &&
         strcmp(strrchr(filename, '.'), ".XXX") == 0) {

src/main.c

@@ -227,6 +227,47 @@ parse_pixel_format(const char *arg)
     return ret;
 }
 
+/// dump input file if not a regular file (eg. test patern or  /dev/random)
+static void
+debug_dump_infile(const char* filename, const uint8_t* image_data, size_t image_size,
+                const struct gpujpeg_image_parameters* param_image)
+{
+    FILE* f = fopen(filename, "rb");
+    long file_sz = 0;
+    if ( f != NULL ) {
+        fseek(f, 0, SEEK_END);
+        file_sz = ftell(f);
+        fclose(f);
+    }
+    if ( file_sz > 0 ) { // regular file
+        return;
+    }
+    char raw_name[256];
+    const char* beg = strrchr(filename, '/');
+    if ( beg == NULL ) {
+        beg = strrchr(filename, '\\');
+    }
+    if ( beg == NULL ) {
+        beg = filename;
+    }
+    else {
+        beg += 1;
+    }
+    snprintf(raw_name, sizeof raw_name, "input-%s", beg);
+    char* end = strrchr(raw_name, '.');
+    if ( end == NULL ) {
+        snprintf(raw_name + strlen(raw_name), sizeof raw_name - strlen(raw_name), ".");
+        end = raw_name + strlen(raw_name);
+    }
+    else {
+        end += 1;
+    }
+    snprintf(end, sizeof raw_name - (end - raw_name), "XXX");
+    if ( gpujpeg_image_save_to_file(raw_name, image_data, image_size, param_image) == 0 ) {
+        printf("Input data saved to file %s.\n", raw_name);
+    }
+}
+
 #ifndef GIT_REV
 #define GIT_REV "unknown"
 #endif
@@ -557,6 +598,10 @@ main(int argc, char *argv[])
                 ret = EXIT_FAILURE; continue;
             }
 
+            if ( debug ) {
+                debug_dump_infile(input, image, image_size, &param_image);
+            }
+
             duration = gpujpeg_get_time() - duration;
             printf("Load Image:          %10.2f ms\n", duration * 1000.0);