cdjpeg.c

   1 /*
   2  * cdjpeg.c
   3  *
   4  * This file was part of the Independent JPEG Group's software:
   5  * Copyright (C) 1991-1997, Thomas G. Lane.
   6  * libjpeg-turbo Modifications:
   7  * Copyright (C) 2019, 2022, D. R. Commander.
   8  * For conditions of distribution and use, see the accompanying README.ijg
   9  * file.
  10  *
  11  * This file contains common support routines used by the IJG application
  12  * programs (cjpeg, djpeg, jpegtran).
  13  */
  14
  15 #include "cdjpeg.h"             /* Common decls for cjpeg/djpeg applications */
  16 #include <ctype.h>              /* to declare isupper(), tolower() */
  17 #ifdef USE_SETMODE
  18 #include <fcntl.h>              /* to declare setmode()'s parameter macros */
  19 /* If you have setmode() but not <io.h>, just delete this line: */
  20 #include <io.h>                 /* to declare setmode() */
  21 #endif
  22
  23
  24 /*
  25  * Optional progress monitor: display a percent-done figure on stderr.
  26  */
  27
  28 METHODDEF(void)
  29 progress_monitor(j_common_ptr cinfo)
  30 {
  31   cd_progress_ptr prog = (cd_progress_ptr)cinfo->progress;
  32
  33   if (prog->max_scans != 0 && cinfo->is_decompressor) {
  34     int scan_no = ((j_decompress_ptr)cinfo)->input_scan_number;
  35
  36     if (scan_no > (int)prog->max_scans) {
  37       fprintf(stderr, "Scan number %d exceeds maximum scans (%u)\n", scan_no,
  38               prog->max_scans);
  39       exit(EXIT_FAILURE);
  40     }
  41   }
  42
  43   if (prog->report) {
  44     int total_passes = prog->pub.total_passes + prog->total_extra_passes;
  45     int percent_done =
  46       (int)(prog->pub.pass_counter * 100L / prog->pub.pass_limit);
  47
  48     if (percent_done != prog->percent_done) {
  49       prog->percent_done = percent_done;
  50       if (total_passes > 1) {
  51         fprintf(stderr, "\rPass %d/%d: %3d%% ",
  52                 prog->pub.completed_passes + prog->completed_extra_passes + 1,
  53                 total_passes, percent_done);
  54       } else {
  55         fprintf(stderr, "\r %3d%% ", percent_done);
  56       }
  57       fflush(stderr);
  58     }
  59   }
  60 }
  61
  62
  63 GLOBAL(void)
  64 start_progress_monitor(j_common_ptr cinfo, cd_progress_ptr progress)
  65 {
  66   /* Enable progress display, unless trace output is on */
  67   if (cinfo->err->trace_level == 0) {
  68     progress->pub.progress_monitor = progress_monitor;
  69     progress->completed_extra_passes = 0;
  70     progress->total_extra_passes = 0;
  71     progress->max_scans = 0;
  72     progress->report = FALSE;
  73     progress->percent_done = -1;
  74     cinfo->progress = &progress->pub;
  75   }
  76 }
  77
  78
  79 GLOBAL(void)
  80 end_progress_monitor(j_common_ptr cinfo)
  81 {
  82   /* Clear away progress display */
  83   if (cinfo->err->trace_level == 0) {
  84     fprintf(stderr, "\r                \r");
  85     fflush(stderr);
  86   }
  87 }
  88
  89
  90 /*
  91  * Case-insensitive matching of possibly-abbreviated keyword switches.
  92  * keyword is the constant keyword (must be lower case already),
  93  * minchars is length of minimum legal abbreviation.
  94  */
  95
  96 GLOBAL(boolean)
  97 keymatch(char *arg, const char *keyword, int minchars)
  98 {
  99   register int ca, ck;
 100   register int nmatched = 0;
 101
 102   while ((ca = *arg++) != '\0') {
 103     if ((ck = *keyword++) == '\0')
 104       return FALSE;             /* arg longer than keyword, no good */
 105     if (isupper(ca))            /* force arg to lcase (assume ck is already) */
 106       ca = tolower(ca);
 107     if (ca != ck)
 108       return FALSE;             /* no good */
 109     nmatched++;                 /* count matched characters */
 110   }
 111   /* reached end of argument; fail if it's too short for unique abbrev */
 112   if (nmatched < minchars)
 113     return FALSE;
 114   return TRUE;                  /* A-OK */
 115 }
 116
 117
 118 /*
 119  * Routines to establish binary I/O mode for stdin and stdout.
 120  * Non-Unix systems often require some hacking to get out of text mode.
 121  */
 122
 123 GLOBAL(FILE *)
 124 read_stdin(void)
 125 {
 126   FILE *input_file = stdin;
 127
 128 #ifdef USE_SETMODE              /* need to hack file mode? */
 129   setmode(fileno(stdin), O_BINARY);
 130 #endif
 131 #ifdef USE_FDOPEN               /* need to re-open in binary mode? */
 132   if ((input_file = fdopen(fileno(stdin), READ_BINARY)) == NULL) {
 133     fprintf(stderr, "Cannot reopen stdin\n");
 134     exit(EXIT_FAILURE);
 135   }
 136 #endif
 137   return input_file;
 138 }
 139
 140
 141 GLOBAL(FILE *)
 142 write_stdout(void)
 143 {
 144   FILE *output_file = stdout;
 145
 146 #ifdef USE_SETMODE              /* need to hack file mode? */
 147   setmode(fileno(stdout), O_BINARY);
 148 #endif
 149 #ifdef USE_FDOPEN               /* need to re-open in binary mode? */
 150   if ((output_file = fdopen(fileno(stdout), WRITE_BINARY)) == NULL) {
 151     fprintf(stderr, "Cannot reopen stdout\n");
 152     exit(EXIT_FAILURE);
 153   }
 154 #endif
 155   return output_file;
 156 }