applications/codec/j2k_dump.c

   1 /*
   2  * Copyright (c) 2010, Mathieu Malaterre, GDCM
   3  * Copyright (c) 2011, Mickael Savinaud, Communications & Systemes <mickael.savinaud@c-s.fr>
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
  16  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  19  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  20  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  21  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  22  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  23  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  24  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  25  * POSSIBILITY OF SUCH DAMAGE.
  26  */
  27 #include "opj_config.h"
  28
  29 #include <stdio.h>
  30 #include <string.h>
  31 #include <stdlib.h>
  32 #include <math.h>
  33
  34 #ifdef _WIN32
  35 #include "windirent.h"
  36 #else
  37 #include <dirent.h>
  38 #endif /* _WIN32 */
  39
  40 #ifdef _WIN32
  41 #include <windows.h>
  42 #else
  43 #include <strings.h>
  44 #define _stricmp strcasecmp
  45 #define _strnicmp strncasecmp
  46 #endif /* _WIN32 */
  47
  48 #include "openjpeg.h"
  49 #include "opj_getopt.h"
  50 #include "convert.h"
  51 #include "index.h"
  52
  53 #include "format_defs.h"
  54
  55 typedef struct dircnt{
  56         /** Buffer for holding images read from Directory*/
  57         char *filename_buf;
  58         /** Pointer to the buffer*/
  59         char **filename;
  60 }dircnt_t;
  61
  62
  63 typedef struct img_folder{
  64         /** The directory path of the folder containing input images*/
  65         char *imgdirpath;
  66         /** Output format*/
  67         const char *out_format;
  68         /** Enable option*/
  69         char set_imgdir;
  70         /** Enable Cod Format for output*/
  71         char set_out_format;
  72
  73 }img_fol_t;
  74
  75 /* -------------------------------------------------------------------------- */
  76 /* Declarations                                                               */
  77 int get_num_images(char *imgdirpath);
  78 int load_images(dircnt_t *dirptr, char *imgdirpath);
  79 int get_file_format(const char *filename);
  80 char get_next_file(int imageno,dircnt_t *dirptr,img_fol_t *img_fol, opj_dparameters_t *parameters);
  81 static int infile_format(const char *fname);
  82
  83 int parse_cmdline_decoder(int argc, char **argv, opj_dparameters_t *parameters,img_fol_t *img_fol);
  84 int parse_DA_values( char* inArg, unsigned int *DA_x0, unsigned int *DA_y0, unsigned int *DA_x1, unsigned int *DA_y1);
  85
  86 /* -------------------------------------------------------------------------- */
  87 void decode_help_display(void) {
  88         fprintf(stdout,"HELP for j2k_dump\n----\n\n");
  89         fprintf(stdout,"- the -h option displays this help information on screen\n\n");
  90
  91 /* UniPG>> */
  92         fprintf(stdout,"List of parameters for the JPEG 2000 "
  93 #ifdef USE_JPWL
  94                 "+ JPWL "
  95 #endif /* USE_JPWL */
  96                 "decoder:\n");
  97 /* <<UniPG */
  98         fprintf(stdout,"\n");
  99         fprintf(stdout,"\n");
 100         fprintf(stdout,"  -ImgDir \n");
 101         fprintf(stdout,"        Image file Directory path \n");
 102         fprintf(stdout,"  -i <compressed file>\n");
 103         fprintf(stdout,"    REQUIRED only if an Input image directory not specified\n");
 104         fprintf(stdout,"    Currently accepts J2K-files, JP2-files and JPT-files. The file type\n");
 105         fprintf(stdout,"    is identified based on its suffix.\n");
 106         fprintf(stdout,"  -o <output file>\n");
 107         fprintf(stdout,"    OPTIONAL\n");
 108         fprintf(stdout,"    Output file where file info will be dump.\n");
 109         fprintf(stdout,"    By default it will be in the stdout.\n");
 110         fprintf(stdout,"  -v "); /* FIXME WIP_MSD */
 111         fprintf(stdout,"    OPTIONAL\n");
 112         fprintf(stdout,"    Activate or not the verbose mode (display info and warning message)\n");
 113         fprintf(stdout,"    By default verbose mode is off.\n");
 114         fprintf(stdout,"\n");
 115 }
 116
 117 /* -------------------------------------------------------------------------- */
 118 int get_num_images(char *imgdirpath){
 119         DIR *dir;
 120         struct dirent* content;
 121         int num_images = 0;
 122
 123         /*Reading the input images from given input directory*/
 124
 125         dir= opendir(imgdirpath);
 126         if(!dir){
 127                 fprintf(stderr,"Could not open Folder %s\n",imgdirpath);
 128                 return 0;
 129         }
 130
 131         while((content=readdir(dir))!=NULL){
 132                 if(strcmp(".",content->d_name)==0 || strcmp("..",content->d_name)==0 )
 133                         continue;
 134                 num_images++;
 135         }
 136         return num_images;
 137 }
 138
 139 /* -------------------------------------------------------------------------- */
 140 int load_images(dircnt_t *dirptr, char *imgdirpath){
 141         DIR *dir;
 142         struct dirent* content;
 143         int i = 0;
 144
 145         /*Reading the input images from given input directory*/
 146
 147         dir= opendir(imgdirpath);
 148         if(!dir){
 149                 fprintf(stderr,"Could not open Folder %s\n",imgdirpath);
 150                 return 1;
 151         }else   {
 152                 fprintf(stderr,"Folder opened successfully\n");
 153         }
 154
 155         while((content=readdir(dir))!=NULL){
 156                 if(strcmp(".",content->d_name)==0 || strcmp("..",content->d_name)==0 )
 157                         continue;
 158
 159                 strcpy(dirptr->filename[i],content->d_name);
 160                 i++;
 161         }
 162         return 0;
 163 }
 164
 165 /* -------------------------------------------------------------------------- */
 166 int get_file_format(const char *filename) {
 167         unsigned int i;
 168         static const char *extension[] = {"pgx", "pnm", "pgm", "ppm", "bmp","tif", "raw", "tga", "png", "j2k", "jp2", "jpt", "j2c", "jpc"  };
 169         static const int format[] = { PGX_DFMT, PXM_DFMT, PXM_DFMT, PXM_DFMT, BMP_DFMT, TIF_DFMT, RAW_DFMT, TGA_DFMT, PNG_DFMT, J2K_CFMT, JP2_CFMT, JPT_CFMT, J2K_CFMT, J2K_CFMT };
 170         char * ext = strrchr(filename, '.');
 171         if (ext == NULL)
 172                 return -1;
 173         ext++;
 174         if(ext) {
 175                 for(i = 0; i < sizeof(format)/sizeof(*format); i++) {
 176                         if(_strnicmp(ext, extension[i], 3) == 0) {
 177                                 return format[i];
 178                         }
 179                 }
 180         }
 181
 182         return -1;
 183 }
 184
 185 /* -------------------------------------------------------------------------- */
 186 char get_next_file(int imageno,dircnt_t *dirptr,img_fol_t *img_fol, opj_dparameters_t *parameters){
 187         char image_filename[OPJ_PATH_LEN], infilename[OPJ_PATH_LEN],outfilename[OPJ_PATH_LEN],temp_ofname[OPJ_PATH_LEN];
 188         char *temp_p, temp1[OPJ_PATH_LEN]="";
 189
 190         strcpy(image_filename,dirptr->filename[imageno]);
 191         fprintf(stderr,"File Number %d \"%s\"\n",imageno,image_filename);
 192         parameters->decod_format = get_file_format(image_filename);
 193         if (parameters->decod_format == -1)
 194                 return 1;
 195         sprintf(infilename,"%s/%s",img_fol->imgdirpath,image_filename);
 196         strncpy(parameters->infile, infilename, sizeof(infilename));
 197
 198         /*Set output file*/
 199         strcpy(temp_ofname,strtok(image_filename,"."));
 200         while((temp_p = strtok(NULL,".")) != NULL){
 201                 strcat(temp_ofname,temp1);
 202                 sprintf(temp1,".%s",temp_p);
 203         }
 204         if(img_fol->set_out_format==1){
 205                 sprintf(outfilename,"%s/%s.%s",img_fol->imgdirpath,temp_ofname,img_fol->out_format);
 206                 strncpy(parameters->outfile, outfilename, sizeof(outfilename));
 207         }
 208         return 0;
 209 }
 210
 211 /* -------------------------------------------------------------------------- */
 212 #define JP2_RFC3745_MAGIC "\x00\x00\x00\x0c\x6a\x50\x20\x20\x0d\x0a\x87\x0a"
 213 #define JP2_MAGIC "\x0d\x0a\x87\x0a"
 214 /* position 45: "\xff\x52" */
 215 #define J2K_CODESTREAM_MAGIC "\xff\x4f\xff\x51"
 216
 217 static int infile_format(const char *fname)
 218 {
 219         FILE *reader;
 220         const char *s, *magic_s;
 221         int ext_format, magic_format;
 222         unsigned char buf[12];
 223         unsigned int l_nb_read;
 224
 225         reader = fopen(fname, "rb");
 226
 227         if (reader == NULL)
 228                 return -1;
 229
 230         memset(buf, 0, 12);
 231         l_nb_read = fread(buf, 1, 12, reader);
 232         fclose(reader);
 233         if (l_nb_read != 12)
 234                 return -1;
 235
 236
 237
 238         ext_format = get_file_format(fname);
 239
 240         if (ext_format == JPT_CFMT)
 241                 return JPT_CFMT;
 242
 243         if (memcmp(buf, JP2_RFC3745_MAGIC, 12) == 0 || memcmp(buf, JP2_MAGIC, 4) == 0) {
 244                 magic_format = JP2_CFMT;
 245                 magic_s = ".jp2";
 246         }
 247         else if (memcmp(buf, J2K_CODESTREAM_MAGIC, 4) == 0) {
 248                 magic_format = J2K_CFMT;
 249                 magic_s = ".j2k or .jpc or .j2c";
 250         }
 251         else
 252                 return -1;
 253
 254         if (magic_format == ext_format)
 255                 return ext_format;
 256
 257         s = fname + strlen(fname) - 4;
 258
 259         fputs("\n===========================================\n", stderr);
 260         fprintf(stderr, "The extension of this file is incorrect.\n"
 261                                         "FOUND %s. SHOULD BE %s\n", s, magic_s);
 262         fputs("===========================================\n", stderr);
 263
 264         return magic_format;
 265 }
 266 /* -------------------------------------------------------------------------- */
 267 /**
 268  * Parse the command line
 269  */
 270 /* -------------------------------------------------------------------------- */
 271 int parse_cmdline_decoder(int argc, char **argv, opj_dparameters_t *parameters,img_fol_t *img_fol) {
 272         int totlen, c;
 273         opj_option_t long_option[]={
 274                 {"ImgDir",REQ_ARG, NULL ,'y'},
 275         };
 276         const char optlist[] = "i:o:hv";
 277
 278         totlen=sizeof(long_option);
 279         img_fol->set_out_format = 0;
 280         do {
 281                 c = opj_getopt_long(argc, argv,optlist,long_option,totlen);
 282                 if (c == -1)
 283                         break;
 284                 switch (c) {
 285                         case 'i':                       /* input file */
 286                         {
 287                                 char *infile = opj_optarg;
 288                                 parameters->decod_format = infile_format(infile);
 289                                 switch(parameters->decod_format) {
 290                                         case J2K_CFMT:
 291                                                 break;
 292                                         case JP2_CFMT:
 293                                                 break;
 294                                         case JPT_CFMT:
 295                                                 break;
 296                                         default:
 297                                                 fprintf(stderr,
 298                                                         "!! Unrecognized format for infile : %s [accept only *.j2k, *.jp2, *.jpc or *.jpt] !!\n\n",
 299                                                         infile);
 300                                                 return 1;
 301                                 }
 302                                 strncpy(parameters->infile, infile, sizeof(parameters->infile)-1);
 303                         }
 304                         break;
 305
 306                                 /* ------------------------------------------------------ */
 307
 308                         case 'o':     /* output file */
 309                         {
 310                           char *outfile = opj_optarg;
 311                           strncpy(parameters->outfile, outfile, sizeof(parameters->outfile)-1);
 312                         }
 313                         break;
 314
 315                                 /* ----------------------------------------------------- */
 316
 317                         case 'h':                       /* display an help description */
 318                                 decode_help_display();
 319                                 return 1;
 320
 321                                 /* ------------------------------------------------------ */
 322
 323                         case 'y':                       /* Image Directory path */
 324                         {
 325                                 img_fol->imgdirpath = (char*)malloc(strlen(opj_optarg) + 1);
 326                                 strcpy(img_fol->imgdirpath,opj_optarg);
 327                                 img_fol->set_imgdir=1;
 328                         }
 329                         break;
 330
 331                         /* ----------------------------------------------------- */
 332
 333                         case 'v':               /* Verbose mode */
 334                         {
 335                                 parameters->m_verbose = 1;
 336                         }
 337                         break;
 338
 339                                 /* ----------------------------------------------------- */
 340                         default:
 341                                 fprintf(stderr,"WARNING -> this option is not valid \"-%c %s\"\n",c, opj_optarg);
 342                                 break;
 343                 }
 344         }while(c != -1);
 345
 346         /* check for possible errors */
 347         if(img_fol->set_imgdir==1){
 348                 if(!(parameters->infile[0]==0)){
 349                         fprintf(stderr, "Error: options -ImgDir and -i cannot be used together !!\n");
 350                         return 1;
 351                 }
 352                 if(img_fol->set_out_format == 0){
 353                         fprintf(stderr, "Error: When -ImgDir is used, -OutFor <FORMAT> must be used !!\n");
 354                         fprintf(stderr, "Only one format allowed! Valid format PGM, PPM, PNM, PGX, BMP, TIF, RAW and TGA!!\n");
 355                         return 1;
 356                 }
 357                 if(!((parameters->outfile[0] == 0))){
 358                         fprintf(stderr, "Error: options -ImgDir and -o cannot be used together !!\n");
 359                         return 1;
 360                 }
 361         }else{
 362                 if((parameters->infile[0] == 0) ) {
 363                         fprintf(stderr, "Example: %s -i image.j2k\n",argv[0]);
 364                         fprintf(stderr, "    Try: %s -h\n",argv[0]);
 365                         return 1;
 366                 }
 367         }
 368
 369         return 0;
 370 }
 371
 372 /* -------------------------------------------------------------------------- */
 373 /**
 374  * J2K_DUMP MAIN
 375  */
 376 /* -------------------------------------------------------------------------- */
 377 int main(int argc, char *argv[])
 378 {
 379         FILE *fsrc = NULL, *fout = NULL;
 380
 381         opj_dparameters_t parameters;                   /* Decompression parameters */
 382         opj_event_mgr_t event_mgr;                              /* Event manager */
 383         opj_image_t* image = NULL;                                      /* Image structure */
 384         opj_codec_t* dinfo = NULL;                              /* Handle to a decompressor */
 385         opj_stream_t *cio = NULL;                               /* Stream */
 386         opj_codestream_info_v2_t* cstr_info = NULL;
 387         opj_codestream_index_t* cstr_index = NULL;
 388
 389         OPJ_INT32 num_images, imageno;
 390         img_fol_t img_fol;
 391         dircnt_t *dirptr = NULL;
 392
 393 #ifdef MSD
 394         opj_bool l_go_on = OPJ_TRUE;
 395         OPJ_UINT32 l_max_data_size = 1000;
 396         OPJ_BYTE * l_data = (OPJ_BYTE *) malloc(1000);
 397 #endif
 398
 399         /* Set decoding parameters to default values */
 400         opj_set_default_decoder_parameters(&parameters);
 401
 402         /* Initialize img_fol */
 403         memset(&img_fol,0,sizeof(img_fol_t));
 404
 405         /* Parse input and get user encoding parameters */
 406         if(parse_cmdline_decoder(argc, argv, &parameters,&img_fol) == 1) {
 407                 return EXIT_FAILURE;
 408         }
 409
 410         /* Set default event mgr */
 411         opj_initialize_default_event_handler(&event_mgr, parameters.m_verbose);
 412
 413         /* Initialize reading of directory */
 414         if(img_fol.set_imgdir==1){
 415                 int it_image;
 416                 num_images=get_num_images(img_fol.imgdirpath);
 417
 418                 dirptr=(dircnt_t*)malloc(sizeof(dircnt_t));
 419                 if(dirptr){
 420                         dirptr->filename_buf = (char*)malloc(num_images*OPJ_PATH_LEN*sizeof(char));     /* Stores at max 10 image file names*/
 421                         dirptr->filename = (char**) malloc(num_images*sizeof(char*));
 422
 423                         if(!dirptr->filename_buf){
 424                                 return EXIT_FAILURE;
 425                         }
 426
 427                         for(it_image=0;it_image<num_images;it_image++){
 428                                 dirptr->filename[it_image] = dirptr->filename_buf + it_image*OPJ_PATH_LEN;
 429                         }
 430                 }
 431                 if(load_images(dirptr,img_fol.imgdirpath)==1){
 432                         return EXIT_FAILURE;
 433                 }
 434
 435                 if (num_images==0){
 436                         fprintf(stdout,"Folder is empty\n");
 437                         return EXIT_FAILURE;
 438                 }
 439         }else{
 440                 num_images=1;
 441         }
 442
 443         /* Try to open for writing the output file if necessary */
 444         if (parameters.outfile[0] != 0){
 445                 fout = fopen(parameters.outfile,"w");
 446                 if (!fout){
 447                         fprintf(stderr, "ERROR -> failed to open %s for writing\n", parameters.outfile);
 448                         return EXIT_FAILURE;
 449                 }
 450         }
 451         else
 452                 fout = stdout;
 453
 454         /* Read the header of each image one by one */
 455         for(imageno = 0; imageno < num_images ; imageno++){
 456
 457                 fprintf(stderr,"\n");
 458
 459                 if(img_fol.set_imgdir==1){
 460                         if (get_next_file(imageno, dirptr,&img_fol, &parameters)) {
 461                                 fprintf(stderr,"skipping file...\n");
 462                                 continue;
 463                         }
 464                 }
 465
 466                 /* Read the input file and put it in memory */
 467                 /* ---------------------------------------- */
 468                 fsrc = fopen(parameters.infile, "rb");
 469                 if (!fsrc) {
 470                         fprintf(stderr, "ERROR -> failed to open %s for reading\n", parameters.infile);
 471                         return EXIT_FAILURE;
 472                 }
 473
 474                 cio = opj_stream_create_default_file_stream(fsrc,1);
 475                 if (!cio){
 476                         fclose(fsrc);
 477                         fprintf(stderr, "ERROR -> failed to create the stream from the file\n");
 478                         return EXIT_FAILURE;
 479                 }
 480
 481                 /* Read the JPEG2000 stream */
 482                 /* ------------------------ */
 483
 484                 switch(parameters.decod_format) {
 485                         case J2K_CFMT:  /* JPEG-2000 codestream */
 486                         {
 487                                 /* Get a decoder handle */
 488                                 dinfo = opj_create_decompress_v2(CODEC_J2K);
 489                                 break;
 490                         }
 491                         case JP2_CFMT:  /* JPEG 2000 compressed image data */
 492                         {
 493                                 /* Get a decoder handle */
 494                                 dinfo = opj_create_decompress_v2(CODEC_JP2);
 495                                 break;
 496                         }
 497                         case JPT_CFMT:  /* JPEG 2000, JPIP */
 498                         {
 499                                 /* Get a decoder handle */
 500                                 dinfo = opj_create_decompress_v2(CODEC_JPT);
 501                                 break;
 502                         }
 503                         default:
 504                                 fprintf(stderr, "skipping file..\n");
 505                                 opj_stream_destroy(cio);
 506                                 continue;
 507                 }
 508
 509                 /* Setup the decoder decoding parameters using user parameters */
 510                 if ( !opj_setup_decoder_v2(dinfo, &parameters, &event_mgr) ){
 511                         fprintf(stderr, "ERROR -> j2k_dump: failed to setup the decoder\n");
 512                         opj_stream_destroy(cio);
 513                         fclose(fsrc);
 514                         opj_destroy_codec(dinfo);
 515                         fclose(fout);
 516                         return EXIT_FAILURE;
 517                 }
 518
 519                 /* Read the main header of the codestream and if necessary the JP2 boxes*/
 520                 if(! opj_read_header(cio, dinfo, &image)){
 521                         fprintf(stderr, "ERROR -> j2k_dump: failed to read the header\n");
 522                         opj_stream_destroy(cio);
 523                         fclose(fsrc);
 524                         opj_destroy_codec(dinfo);
 525                         opj_image_destroy(image);
 526                         fclose(fout);
 527                         return EXIT_FAILURE;
 528                 }
 529
 530                 opj_dump_codec(dinfo, OPJ_IMG_INFO | OPJ_J2K_MH_INFO | OPJ_J2K_MH_IND, fout );
 531
 532                 cstr_info = opj_get_cstr_info(dinfo);
 533
 534                 cstr_index = opj_get_cstr_index(dinfo);
 535
 536                 /* close the byte stream */
 537                 opj_stream_destroy(cio);
 538                 fclose(fsrc);
 539
 540                 /* free remaining structures */
 541                 if (dinfo) {
 542                         opj_destroy_codec(dinfo);
 543                 }
 544
 545                 /* destroy the image header */
 546                 opj_image_destroy(image);
 547
 548                 /* destroy the codestream index */
 549                 opj_destroy_cstr_index(&cstr_index);
 550
 551                 /* destroy the codestream info */
 552                 opj_destroy_cstr_info_v2(&cstr_info);
 553
 554         }
 555
 556         /* Close the output file */
 557         fclose(fout);
 558
 559   return EXIT_SUCCESS;
 560 }