ref: f9e3d8813674fe45ec12a3b75f5ab9024747a41d
dir: /jbig2dec.c/
/* jbig2dec Copyright (C) 2001-2002 artofcode LLC. This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. $Id: jbig2dec.c,v 1.31 2002/07/13 00:32:13 giles Exp $ */ #ifdef HAVE_CONFIG_H #include "config.h" #include "config_types.h" #elif _WIN32 #include "config_win32.h" #endif #ifdef HAVE_STDINT_H #include <stdint.h> #endif #include <stdio.h> # include <stdlib.h> # include <stddef.h> # include <string.h> #ifdef HAVE_GETOPT_H # include <getopt.h> #else # include "getopt.h" #endif #include "jbig2.h" #include "jbig2_image.h" typedef enum { usage,dump,render } jbig2dec_mode; typedef struct { jbig2dec_mode mode; int verbose; char *output_file; } jbig2dec_params_t; static int print_usage (void) { fprintf(stderr, "Usage: jbig2dec [options] <file.jbig2>\n" " or jbig2dec [options] <global_stream> <page_stream>\n" "\n" " When invoked with a single file, it attempts to parse it as\n" " a normal jbig2 file. Invoked with two files, it treats the\n" " first as the global segments, and the second as the segment\n" " stream for a particular page. This is useful for examining\n" " embedded streams.\n" "\n" " available options:\n" " -h --help this usage summary\n" " -q --quiet suppress diagnostic output\n" " -d --dump print the structure of the jbig2 file\n" " rather than explicitly decoding\n" " -o <file> send decoded output to <file>\n" " Defaults to the the input with a different\n" " extension. Pass '-' for stdout.\n" "\n" ); return 1; } static int parse_options(int argc, char *argv[], jbig2dec_params_t *params) { static struct option long_options[] = { {"quiet", 0, NULL, 'q'}, {"help", 0, NULL, 'h'}, {"dump", 0, NULL, 'd'}, {"output", 1, NULL, 'o'}, {NULL, 0, NULL, 0} }; int option_idx = 1; int option; while (1) { option = getopt_long(argc, argv, "qhdo:", long_options, &option_idx); if (option == -1) break; fprintf(stderr, "option '%c' value '%s'\n", option, optarg); switch (option) { case 0: // unknown long option if (!params->verbose) fprintf(stdout, "unrecognized option: --%s\n", long_options[option_idx].name); break; case 'q': params->verbose = 0; break; case 'h': case '?': params->mode = usage; break; case 'd': params->mode=dump; break; case 'o': params->output_file = strdup(optarg); break; default: if (!params->verbose) fprintf(stdout, "unrecognized option: -%c\n", option); break; } } fprintf(stderr, "final option index %d out of %d\n", optind, argc); return (optind); } static int error_callback(void *error_callback_data, const char *buf, Jbig2Severity severity, int32_t seg_idx) { char *type; char segment[22]; switch (severity) { case JBIG2_SEVERITY_DEBUG: type = "DEBUG"; break;; case JBIG2_SEVERITY_INFO: type = "info"; break;; case JBIG2_SEVERITY_WARNING: type = "WARNING"; break;; case JBIG2_SEVERITY_FATAL: type = "FATAL ERROR"; break;; default: type = "unknown message"; break;; } if (seg_idx == -1) segment[0] = '\0'; else snprintf(segment, sizeof(segment), "(segment 0x%02x)", seg_idx); fprintf(stderr, "jbig2dec %s %s %s\n", type, buf, segment); return 0; } static char * make_output_filename(const char *input_filename, const char *extension) { char *output_filename; const char *c, *e; int len; if (extension == NULL) { fprintf(stderr, "make_output_filename called with no extension!\n"); exit (1); } if (input_filename == NULL) c = "out"; else { /* strip any leading path */ c = strrchr(input_filename, '/'); /* *nix */ if (c == NULL) c = strrchr(input_filename, '\\'); /* win32/dos */ if (c != NULL) c++; /* skip the path separator */ else c = input_filename; /* no leading path */ } /* make sure we haven't just stripped the last character */ if (*c == '\0') c = "out"; /* strip the extension */ len = strlen(c); e = strrchr(c, '.'); if (e != NULL) len -= strlen(e); /* allocate enough space for the base + ext */ output_filename = malloc(len + strlen(extension) + 1); if (output_filename == NULL) { fprintf(stderr, "couldn't allocate memory for output_filename\n"); exit (1); } strncpy(output_filename, c, len); strncpy(output_filename + len, extension, strlen(extension)); *(output_filename + len + strlen(extension)) = '\0'; /* return the new string */ return (output_filename); } static int write_page_image(jbig2dec_params_t *params, Jbig2Image *image) { if (!strcmp(params->output_file, "-")) { fprintf(stderr, "writing decoded page to stdout\n"); #ifdef HAVE_LIBPNG jbig2_image_write_png(image, stdout); #else jbig2_image_write_pbm(image, stdout); #endif } else { fprintf(stderr, "saving decoded page as '%s'\n", params->output_file); #ifdef HAVE_LIBPNG jbig2_image_write_png_file(image, params->output_file); #else jbig2_image_write_pbm_file(image, params->output_file); #endif } return 0; } int main (int argc, char **argv) { FILE *f = NULL, *f_page = NULL; Jbig2Ctx *ctx; uint8_t buf[4096]; jbig2dec_params_t params = {render,1,NULL}; int filearg; filearg = parse_options(argc, argv, ¶ms); switch (params.mode) { case usage: print_usage(); exit (0); break; case dump: fprintf(stderr, "Sorry, segment dump not yet implemented\n"); break; case render: if ((argc - filearg) == 1) // only one argument--open as a jbig2 file { char *fn = argv[filearg]; f = fopen(fn, "rb"); if (f == NULL) { fprintf(stderr, "error opening %s\n", fn); return 1; } } else if ((argc - filearg) == 2) // two arguments open as separate global and page streams { char *fn = argv[filearg]; char *fn_page = argv[filearg+1]; f = fopen(fn, "rb"); if (f == NULL) { fprintf(stderr, "error opening %s\n", fn); return 1; } f_page = fopen(fn_page, "rb"); if (f_page == NULL) { fprintf(stderr, "error opening %s\n", fn_page); return 1; } } else // any other number of arguments return print_usage(); ctx = jbig2_ctx_new(NULL, f_page != NULL ? JBIG2_OPTIONS_EMBEDDED : 0, NULL, error_callback, NULL); // pull the whole file/global stream into memory for (;;) { int n_bytes = fread(buf, 1, sizeof(buf), f); if (n_bytes <= 0) break; jbig2_data_in(ctx, buf, n_bytes); } fclose(f); // if there's a local page stream read that in its entirety if (f_page != NULL) { Jbig2GlobalCtx *global_ctx = jbig2_make_global_ctx(ctx); ctx = jbig2_ctx_new(NULL, JBIG2_OPTIONS_EMBEDDED, global_ctx, error_callback, NULL); for (;;) { int n_bytes = fread(buf, 1, sizeof(buf), f_page); if (n_bytes <= 0) break; jbig2_data_in(ctx, buf, n_bytes); } fclose(f_page); jbig2_global_ctx_free(global_ctx); } // retrieve and output the returned pages { Jbig2Image *image; /* work around broken CVision embedded streams */ if (f_page != NULL) jbig2_complete_page(ctx); if (params.output_file == NULL) { #ifdef HAVE_LIBPNG params.output_file = make_output_filename(argv[filearg], ".png"); #else params.output_file = make_output_filename(argv[filearg], ".pbm"); #endif } /* retrieve and write out all the completed pages */ while ((image = jbig2_page_out(ctx)) != NULL) { write_page_image(¶ms, image); jbig2_release_page(ctx, image); } } jbig2_ctx_free(ctx); } /* end params.mode switch */ // fin return 0; }