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, 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)
progress_monitor(j_common_ptr cinfo)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 (%d)\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)
start_progress_monitor(j_common_ptr cinfo,cd_progress_ptr progress)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)
end_progress_monitor(j_common_ptr cinfo)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)
keymatch(char * arg,const char * keyword,int minchars)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 *)
read_stdin(void)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 *)
write_stdout(void)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 }
157