1/* 2 * jpegtran.c 3 * 4 * This file was part of the Independent JPEG Group's software: 5 * Copyright (C) 1995-2010, Thomas G. Lane, Guido Vollbeding. 6 * libjpeg-turbo Modifications: 7 * Copyright (C) 2010, D. R. Commander. 8 * For conditions of distribution and use, see the accompanying README file. 9 * 10 * This file contains a command-line user interface for JPEG transcoding. 11 * It is very similar to cjpeg.c, and partly to djpeg.c, but provides 12 * lossless transcoding between different JPEG file formats. It also 13 * provides some lossless and sort-of-lossless transformations of JPEG data. 14 */ 15 16#include "cdjpeg.h" /* Common decls for cjpeg/djpeg applications */ 17#include "transupp.h" /* Support routines for jpegtran */ 18#include "jversion.h" /* for version message */ 19#include "config.h" 20 21#ifdef USE_CCOMMAND /* command-line reader for Macintosh */ 22#ifdef __MWERKS__ 23#include <SIOUX.h> /* Metrowerks needs this */ 24#include <console.h> /* ... and this */ 25#endif 26#ifdef THINK_C 27#include <console.h> /* Think declares it here */ 28#endif 29#endif 30 31 32/* 33 * Argument-parsing code. 34 * The switch parser is designed to be useful with DOS-style command line 35 * syntax, ie, intermixed switches and file names, where only the switches 36 * to the left of a given file name affect processing of that file. 37 * The main program in this file doesn't actually use this capability... 38 */ 39 40 41static const char * progname; /* program name for error messages */ 42static char * outfilename; /* for -outfile switch */ 43static JCOPY_OPTION copyoption; /* -copy switch */ 44static jpeg_transform_info transformoption; /* image transformation options */ 45 46 47LOCAL(void) 48usage (void) 49/* complain about bad command line */ 50{ 51 fprintf(stderr, "usage: %s [switches] ", progname); 52#ifdef TWO_FILE_COMMANDLINE 53 fprintf(stderr, "inputfile outputfile\n"); 54#else 55 fprintf(stderr, "[inputfile]\n"); 56#endif 57 58 fprintf(stderr, "Switches (names may be abbreviated):\n"); 59 fprintf(stderr, " -copy none Copy no extra markers from source file\n"); 60 fprintf(stderr, " -copy comments Copy only comment markers (default)\n"); 61 fprintf(stderr, " -copy all Copy all extra markers\n"); 62#ifdef ENTROPY_OPT_SUPPORTED 63 fprintf(stderr, " -optimize Optimize Huffman table (smaller file, but slow compression)\n"); 64#endif 65#ifdef C_PROGRESSIVE_SUPPORTED 66 fprintf(stderr, " -progressive Create progressive JPEG file\n"); 67#endif 68 fprintf(stderr, "Switches for modifying the image:\n"); 69#if TRANSFORMS_SUPPORTED 70 fprintf(stderr, " -crop WxH+X+Y Crop to a rectangular subarea\n"); 71 fprintf(stderr, " -grayscale Reduce to grayscale (omit color data)\n"); 72 fprintf(stderr, " -flip [horizontal|vertical] Mirror image (left-right or top-bottom)\n"); 73 fprintf(stderr, " -perfect Fail if there is non-transformable edge blocks\n"); 74 fprintf(stderr, " -rotate [90|180|270] Rotate image (degrees clockwise)\n"); 75#endif 76#if TRANSFORMS_SUPPORTED 77 fprintf(stderr, " -transpose Transpose image\n"); 78 fprintf(stderr, " -transverse Transverse transpose image\n"); 79 fprintf(stderr, " -trim Drop non-transformable edge blocks\n"); 80#endif 81 fprintf(stderr, "Switches for advanced users:\n"); 82#ifdef C_ARITH_CODING_SUPPORTED 83 fprintf(stderr, " -arithmetic Use arithmetic coding\n"); 84#endif 85 fprintf(stderr, " -restart N Set restart interval in rows, or in blocks with B\n"); 86 fprintf(stderr, " -maxmemory N Maximum memory to use (in kbytes)\n"); 87 fprintf(stderr, " -outfile name Specify name for output file\n"); 88 fprintf(stderr, " -verbose or -debug Emit debug output\n"); 89 fprintf(stderr, "Switches for wizards:\n"); 90#ifdef C_MULTISCAN_FILES_SUPPORTED 91 fprintf(stderr, " -scans file Create multi-scan JPEG per script file\n"); 92#endif 93 exit(EXIT_FAILURE); 94} 95 96 97LOCAL(void) 98select_transform (JXFORM_CODE transform) 99/* Silly little routine to detect multiple transform options, 100 * which we can't handle. 101 */ 102{ 103#if TRANSFORMS_SUPPORTED 104 if (transformoption.transform == JXFORM_NONE || 105 transformoption.transform == transform) { 106 transformoption.transform = transform; 107 } else { 108 fprintf(stderr, "%s: can only do one image transformation at a time\n", 109 progname); 110 usage(); 111 } 112#else 113 fprintf(stderr, "%s: sorry, image transformation was not compiled\n", 114 progname); 115 exit(EXIT_FAILURE); 116#endif 117} 118 119 120LOCAL(int) 121parse_switches (j_compress_ptr cinfo, int argc, char **argv, 122 int last_file_arg_seen, boolean for_real) 123/* Parse optional switches. 124 * Returns argv[] index of first file-name argument (== argc if none). 125 * Any file names with indexes <= last_file_arg_seen are ignored; 126 * they have presumably been processed in a previous iteration. 127 * (Pass 0 for last_file_arg_seen on the first or only iteration.) 128 * for_real is FALSE on the first (dummy) pass; we may skip any expensive 129 * processing. 130 */ 131{ 132 int argn; 133 char * arg; 134 boolean simple_progressive; 135 char * scansarg = NULL; /* saves -scans parm if any */ 136 137 /* Set up default JPEG parameters. */ 138 simple_progressive = FALSE; 139 outfilename = NULL; 140 copyoption = JCOPYOPT_DEFAULT; 141 transformoption.transform = JXFORM_NONE; 142 transformoption.perfect = FALSE; 143 transformoption.trim = FALSE; 144 transformoption.force_grayscale = FALSE; 145 transformoption.crop = FALSE; 146 transformoption.slow_hflip = FALSE; 147 cinfo->err->trace_level = 0; 148 149 /* Scan command line options, adjust parameters */ 150 151 for (argn = 1; argn < argc; argn++) { 152 arg = argv[argn]; 153 if (*arg != '-') { 154 /* Not a switch, must be a file name argument */ 155 if (argn <= last_file_arg_seen) { 156 outfilename = NULL; /* -outfile applies to just one input file */ 157 continue; /* ignore this name if previously processed */ 158 } 159 break; /* else done parsing switches */ 160 } 161 arg++; /* advance past switch marker character */ 162 163 if (keymatch(arg, "arithmetic", 1)) { 164 /* Use arithmetic coding. */ 165#ifdef C_ARITH_CODING_SUPPORTED 166 cinfo->arith_code = TRUE; 167#else 168 fprintf(stderr, "%s: sorry, arithmetic coding not supported\n", 169 progname); 170 exit(EXIT_FAILURE); 171#endif 172 173 } else if (keymatch(arg, "copy", 2)) { 174 /* Select which extra markers to copy. */ 175 if (++argn >= argc) /* advance to next argument */ 176 usage(); 177 if (keymatch(argv[argn], "none", 1)) { 178 copyoption = JCOPYOPT_NONE; 179 } else if (keymatch(argv[argn], "comments", 1)) { 180 copyoption = JCOPYOPT_COMMENTS; 181 } else if (keymatch(argv[argn], "all", 1)) { 182 copyoption = JCOPYOPT_ALL; 183 } else 184 usage(); 185 186 } else if (keymatch(arg, "crop", 2)) { 187 /* Perform lossless cropping. */ 188#if TRANSFORMS_SUPPORTED 189 if (++argn >= argc) /* advance to next argument */ 190 usage(); 191 if (! jtransform_parse_crop_spec(&transformoption, argv[argn])) { 192 fprintf(stderr, "%s: bogus -crop argument '%s'\n", 193 progname, argv[argn]); 194 exit(EXIT_FAILURE); 195 } 196#else 197 select_transform(JXFORM_NONE); /* force an error */ 198#endif 199 200 } else if (keymatch(arg, "debug", 1) || keymatch(arg, "verbose", 1)) { 201 /* Enable debug printouts. */ 202 /* On first -d, print version identification */ 203 static boolean printed_version = FALSE; 204 205 if (! printed_version) { 206 fprintf(stderr, "%s version %s (build %s)\n", 207 PACKAGE_NAME, VERSION, BUILD); 208 fprintf(stderr, "%s\n\n", JCOPYRIGHT); 209 fprintf(stderr, "Emulating The Independent JPEG Group's software, version %s\n\n", 210 JVERSION); 211 printed_version = TRUE; 212 } 213 cinfo->err->trace_level++; 214 215 } else if (keymatch(arg, "flip", 1)) { 216 /* Mirror left-right or top-bottom. */ 217 if (++argn >= argc) /* advance to next argument */ 218 usage(); 219 if (keymatch(argv[argn], "horizontal", 1)) 220 select_transform(JXFORM_FLIP_H); 221 else if (keymatch(argv[argn], "vertical", 1)) 222 select_transform(JXFORM_FLIP_V); 223 else 224 usage(); 225 226 } else if (keymatch(arg, "grayscale", 1) || keymatch(arg, "greyscale",1)) { 227 /* Force to grayscale. */ 228#if TRANSFORMS_SUPPORTED 229 transformoption.force_grayscale = TRUE; 230#else 231 select_transform(JXFORM_NONE); /* force an error */ 232#endif 233 234 } else if (keymatch(arg, "maxmemory", 3)) { 235 /* Maximum memory in Kb (or Mb with 'm'). */ 236 long lval; 237 char ch = 'x'; 238 239 if (++argn >= argc) /* advance to next argument */ 240 usage(); 241 if (sscanf(argv[argn], "%ld%c", &lval, &ch) < 1) 242 usage(); 243 if (ch == 'm' || ch == 'M') 244 lval *= 1000L; 245 cinfo->mem->max_memory_to_use = lval * 1000L; 246 247 } else if (keymatch(arg, "optimize", 1) || keymatch(arg, "optimise", 1)) { 248 /* Enable entropy parm optimization. */ 249#ifdef ENTROPY_OPT_SUPPORTED 250 cinfo->optimize_coding = TRUE; 251#else 252 fprintf(stderr, "%s: sorry, entropy optimization was not compiled\n", 253 progname); 254 exit(EXIT_FAILURE); 255#endif 256 257 } else if (keymatch(arg, "outfile", 4)) { 258 /* Set output file name. */ 259 if (++argn >= argc) /* advance to next argument */ 260 usage(); 261 outfilename = argv[argn]; /* save it away for later use */ 262 263 } else if (keymatch(arg, "perfect", 2)) { 264 /* Fail if there is any partial edge MCUs that the transform can't 265 * handle. */ 266 transformoption.perfect = TRUE; 267 268 } else if (keymatch(arg, "progressive", 2)) { 269 /* Select simple progressive mode. */ 270#ifdef C_PROGRESSIVE_SUPPORTED 271 simple_progressive = TRUE; 272 /* We must postpone execution until num_components is known. */ 273#else 274 fprintf(stderr, "%s: sorry, progressive output was not compiled\n", 275 progname); 276 exit(EXIT_FAILURE); 277#endif 278 279 } else if (keymatch(arg, "restart", 1)) { 280 /* Restart interval in MCU rows (or in MCUs with 'b'). */ 281 long lval; 282 char ch = 'x'; 283 284 if (++argn >= argc) /* advance to next argument */ 285 usage(); 286 if (sscanf(argv[argn], "%ld%c", &lval, &ch) < 1) 287 usage(); 288 if (lval < 0 || lval > 65535L) 289 usage(); 290 if (ch == 'b' || ch == 'B') { 291 cinfo->restart_interval = (unsigned int) lval; 292 cinfo->restart_in_rows = 0; /* else prior '-restart n' overrides me */ 293 } else { 294 cinfo->restart_in_rows = (int) lval; 295 /* restart_interval will be computed during startup */ 296 } 297 298 } else if (keymatch(arg, "rotate", 2)) { 299 /* Rotate 90, 180, or 270 degrees (measured clockwise). */ 300 if (++argn >= argc) /* advance to next argument */ 301 usage(); 302 if (keymatch(argv[argn], "90", 2)) 303 select_transform(JXFORM_ROT_90); 304 else if (keymatch(argv[argn], "180", 3)) 305 select_transform(JXFORM_ROT_180); 306 else if (keymatch(argv[argn], "270", 3)) 307 select_transform(JXFORM_ROT_270); 308 else 309 usage(); 310 311 } else if (keymatch(arg, "scans", 1)) { 312 /* Set scan script. */ 313#ifdef C_MULTISCAN_FILES_SUPPORTED 314 if (++argn >= argc) /* advance to next argument */ 315 usage(); 316 scansarg = argv[argn]; 317 /* We must postpone reading the file in case -progressive appears. */ 318#else 319 fprintf(stderr, "%s: sorry, multi-scan output was not compiled\n", 320 progname); 321 exit(EXIT_FAILURE); 322#endif 323 324 } else if (keymatch(arg, "transpose", 1)) { 325 /* Transpose (across UL-to-LR axis). */ 326 select_transform(JXFORM_TRANSPOSE); 327 328 } else if (keymatch(arg, "transverse", 6)) { 329 /* Transverse transpose (across UR-to-LL axis). */ 330 select_transform(JXFORM_TRANSVERSE); 331 332 } else if (keymatch(arg, "trim", 3)) { 333 /* Trim off any partial edge MCUs that the transform can't handle. */ 334 transformoption.trim = TRUE; 335 336 } else { 337 usage(); /* bogus switch */ 338 } 339 } 340 341 /* Post-switch-scanning cleanup */ 342 343 if (for_real) { 344 345#ifdef C_PROGRESSIVE_SUPPORTED 346 if (simple_progressive) /* process -progressive; -scans can override */ 347 jpeg_simple_progression(cinfo); 348#endif 349 350#ifdef C_MULTISCAN_FILES_SUPPORTED 351 if (scansarg != NULL) /* process -scans if it was present */ 352 if (! read_scan_script(cinfo, scansarg)) 353 usage(); 354#endif 355 } 356 357 return argn; /* return index of next arg (file name) */ 358} 359 360 361/* 362 * The main program. 363 */ 364 365int 366main (int argc, char **argv) 367{ 368 struct jpeg_decompress_struct srcinfo; 369 struct jpeg_compress_struct dstinfo; 370 struct jpeg_error_mgr jsrcerr, jdsterr; 371#ifdef PROGRESS_REPORT 372 struct cdjpeg_progress_mgr progress; 373#endif 374 jvirt_barray_ptr * src_coef_arrays; 375 jvirt_barray_ptr * dst_coef_arrays; 376 int file_index; 377 /* We assume all-in-memory processing and can therefore use only a 378 * single file pointer for sequential input and output operation. 379 */ 380 FILE * fp; 381 382 /* On Mac, fetch a command line. */ 383#ifdef USE_CCOMMAND 384 argc = ccommand(&argv); 385#endif 386 387 progname = argv[0]; 388 if (progname == NULL || progname[0] == 0) 389 progname = "jpegtran"; /* in case C library doesn't provide it */ 390 391 /* Initialize the JPEG decompression object with default error handling. */ 392 srcinfo.err = jpeg_std_error(&jsrcerr); 393 jpeg_create_decompress(&srcinfo); 394 /* Initialize the JPEG compression object with default error handling. */ 395 dstinfo.err = jpeg_std_error(&jdsterr); 396 jpeg_create_compress(&dstinfo); 397 398 /* Now safe to enable signal catcher. 399 * Note: we assume only the decompression object will have virtual arrays. 400 */ 401#ifdef NEED_SIGNAL_CATCHER 402 enable_signal_catcher((j_common_ptr) &srcinfo); 403#endif 404 405 /* Scan command line to find file names. 406 * It is convenient to use just one switch-parsing routine, but the switch 407 * values read here are mostly ignored; we will rescan the switches after 408 * opening the input file. Also note that most of the switches affect the 409 * destination JPEG object, so we parse into that and then copy over what 410 * needs to affects the source too. 411 */ 412 413 file_index = parse_switches(&dstinfo, argc, argv, 0, FALSE); 414 jsrcerr.trace_level = jdsterr.trace_level; 415 srcinfo.mem->max_memory_to_use = dstinfo.mem->max_memory_to_use; 416 417#ifdef TWO_FILE_COMMANDLINE 418 /* Must have either -outfile switch or explicit output file name */ 419 if (outfilename == NULL) { 420 if (file_index != argc-2) { 421 fprintf(stderr, "%s: must name one input and one output file\n", 422 progname); 423 usage(); 424 } 425 outfilename = argv[file_index+1]; 426 } else { 427 if (file_index != argc-1) { 428 fprintf(stderr, "%s: must name one input and one output file\n", 429 progname); 430 usage(); 431 } 432 } 433#else 434 /* Unix style: expect zero or one file name */ 435 if (file_index < argc-1) { 436 fprintf(stderr, "%s: only one input file\n", progname); 437 usage(); 438 } 439#endif /* TWO_FILE_COMMANDLINE */ 440 441 /* Open the input file. */ 442 if (file_index < argc) { 443 if ((fp = fopen(argv[file_index], READ_BINARY)) == NULL) { 444 fprintf(stderr, "%s: can't open %s for reading\n", progname, argv[file_index]); 445 exit(EXIT_FAILURE); 446 } 447 } else { 448 /* default input file is stdin */ 449 fp = read_stdin(); 450 } 451 452#ifdef PROGRESS_REPORT 453 start_progress_monitor((j_common_ptr) &dstinfo, &progress); 454#endif 455 456 /* Specify data source for decompression */ 457 jpeg_stdio_src(&srcinfo, fp); 458 459 /* Enable saving of extra markers that we want to copy */ 460 jcopy_markers_setup(&srcinfo, copyoption); 461 462 /* Read file header */ 463 (void) jpeg_read_header(&srcinfo, TRUE); 464 465 /* Any space needed by a transform option must be requested before 466 * jpeg_read_coefficients so that memory allocation will be done right. 467 */ 468#if TRANSFORMS_SUPPORTED 469 /* Fail right away if -perfect is given and transformation is not perfect. 470 */ 471 if (!jtransform_request_workspace(&srcinfo, &transformoption)) { 472 fprintf(stderr, "%s: transformation is not perfect\n", progname); 473 exit(EXIT_FAILURE); 474 } 475#endif 476 477 /* Read source file as DCT coefficients */ 478 src_coef_arrays = jpeg_read_coefficients(&srcinfo); 479 480 /* Initialize destination compression parameters from source values */ 481 jpeg_copy_critical_parameters(&srcinfo, &dstinfo); 482 483 /* Adjust destination parameters if required by transform options; 484 * also find out which set of coefficient arrays will hold the output. 485 */ 486#if TRANSFORMS_SUPPORTED 487 dst_coef_arrays = jtransform_adjust_parameters(&srcinfo, &dstinfo, 488 src_coef_arrays, 489 &transformoption); 490#else 491 dst_coef_arrays = src_coef_arrays; 492#endif 493 494 /* Close input file, if we opened it. 495 * Note: we assume that jpeg_read_coefficients consumed all input 496 * until JPEG_REACHED_EOI, and that jpeg_finish_decompress will 497 * only consume more while (! cinfo->inputctl->eoi_reached). 498 * We cannot call jpeg_finish_decompress here since we still need the 499 * virtual arrays allocated from the source object for processing. 500 */ 501 if (fp != stdin) 502 fclose(fp); 503 504 /* Open the output file. */ 505 if (outfilename != NULL) { 506 if ((fp = fopen(outfilename, WRITE_BINARY)) == NULL) { 507 fprintf(stderr, "%s: can't open %s for writing\n", progname, outfilename); 508 exit(EXIT_FAILURE); 509 } 510 } else { 511 /* default output file is stdout */ 512 fp = write_stdout(); 513 } 514 515 /* Adjust default compression parameters by re-parsing the options */ 516 file_index = parse_switches(&dstinfo, argc, argv, 0, TRUE); 517 518 /* Specify data destination for compression */ 519 jpeg_stdio_dest(&dstinfo, fp); 520 521 /* Start compressor (note no image data is actually written here) */ 522 jpeg_write_coefficients(&dstinfo, dst_coef_arrays); 523 524 /* Copy to the output file any extra markers that we want to preserve */ 525 jcopy_markers_execute(&srcinfo, &dstinfo, copyoption); 526 527 /* Execute image transformation, if any */ 528#if TRANSFORMS_SUPPORTED 529 jtransform_execute_transformation(&srcinfo, &dstinfo, 530 src_coef_arrays, 531 &transformoption); 532#endif 533 534 /* Finish compression and release memory */ 535 jpeg_finish_compress(&dstinfo); 536 jpeg_destroy_compress(&dstinfo); 537 (void) jpeg_finish_decompress(&srcinfo); 538 jpeg_destroy_decompress(&srcinfo); 539 540 /* Close output file, if we opened it */ 541 if (fp != stdout) 542 fclose(fp); 543 544#ifdef PROGRESS_REPORT 545 end_progress_monitor((j_common_ptr) &dstinfo); 546#endif 547 548 /* All done. */ 549 exit(jsrcerr.num_warnings + jdsterr.num_warnings ?EXIT_WARNING:EXIT_SUCCESS); 550 return 0; /* suppress no-return-value warnings */ 551} 552