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
  28 #include <stdio.h>
  29 #include <string.h>
  30 #include <stdlib.h>
  31 #include <math.h>
  32
  33 #ifdef _WIN32
  34 #include "windirent.h"
  35 #else
  36 #include <dirent.h>
  37 #endif /* _WIN32 */
  38
  39 #ifdef _WIN32
  40 #include <windows.h>
  41 #else
  42 #include <strings.h>
  43 #define _stricmp strcasecmp
  44 #define _strnicmp strncasecmp
  45 #endif /* _WIN32 */
  46
  47 #include "opj_config.h"
  48 #include "openjpeg.h"
  49 #include "j2k.h"
  50 #include "jp2.h"
  51 #include "opj_getopt.h"
  52 #include "convert.h"
  53 #include "index.h"
  54
  55 #include "format_defs.h"
  56
  57 typedef struct dircnt{
  58         /** Buffer for holding images read from Directory*/
  59         char *filename_buf;
  60         /** Pointer to the buffer*/
  61         char **filename;
  62 }dircnt_t;
  63
  64
  65 typedef struct img_folder{
  66         /** The directory path of the folder containing input images*/
  67         char *imgdirpath;
  68         /** Output format*/
  69         const char *out_format;
  70         /** Enable option*/
  71         char set_imgdir;
  72         /** Enable Cod Format for output*/
  73         char set_out_format;
  74
  75 }img_fol_t;
  76
  77 void decode_help_display(void) {
  78         fprintf(stdout,"HELP for j2k_dump\n----\n\n");
  79         fprintf(stdout,"- the -h option displays this help information on screen\n\n");
  80
  81 /* UniPG>> */
  82         fprintf(stdout,"List of parameters for the JPEG 2000 "
  83 #ifdef USE_JPWL
  84                 "+ JPWL "
  85 #endif /* USE_JPWL */
  86                 "decoder:\n");
  87 /* <<UniPG */
  88         fprintf(stdout,"\n");
  89         fprintf(stdout,"\n");
  90         fprintf(stdout,"  -ImgDir \n");
  91         fprintf(stdout,"        Image file Directory path \n");
  92         fprintf(stdout,"  -i <compressed file>\n");
  93         fprintf(stdout,"    REQUIRED only if an Input image directory not specified\n");
  94         fprintf(stdout,"    Currently accepts J2K-files, JP2-files and JPT-files. The file type\n");
  95         fprintf(stdout,"    is identified based on its suffix.\n");
  96         fprintf(stdout,"  -o <output file>\n");
  97         fprintf(stdout,"    OPTIONAL\n");
  98         fprintf(stdout,"    Output file where file info will be dump.\n");
  99         fprintf(stdout,"    By default it will be in the stdout.\n");
 100         fprintf(stdout,"\n");
 101 }
 102
 103 /* -------------------------------------------------------------------------- */
 104 static void j2k_dump_image(FILE *fd, opj_image_header_t * img);
 105 static void j2k_dump_cp(FILE *fd, opj_image_t * img, opj_cp_v2_t * cp);
 106
 107 int get_num_images(char *imgdirpath){
 108         DIR *dir;
 109         struct dirent* content;
 110         int num_images = 0;
 111
 112         /*Reading the input images from given input directory*/
 113
 114         dir= opendir(imgdirpath);
 115         if(!dir){
 116                 fprintf(stderr,"Could not open Folder %s\n",imgdirpath);
 117                 return 0;
 118         }
 119
 120         while((content=readdir(dir))!=NULL){
 121                 if(strcmp(".",content->d_name)==0 || strcmp("..",content->d_name)==0 )
 122                         continue;
 123                 num_images++;
 124         }
 125         return num_images;
 126 }
 127
 128 int load_images(dircnt_t *dirptr, char *imgdirpath){
 129         DIR *dir;
 130         struct dirent* content;
 131         int i = 0;
 132
 133         /*Reading the input images from given input directory*/
 134
 135         dir= opendir(imgdirpath);
 136         if(!dir){
 137                 fprintf(stderr,"Could not open Folder %s\n",imgdirpath);
 138                 return 1;
 139         }else   {
 140                 fprintf(stderr,"Folder opened successfully\n");
 141         }
 142
 143         while((content=readdir(dir))!=NULL){
 144                 if(strcmp(".",content->d_name)==0 || strcmp("..",content->d_name)==0 )
 145                         continue;
 146
 147                 strcpy(dirptr->filename[i],content->d_name);
 148                 i++;
 149         }
 150         return 0;
 151 }
 152
 153 int get_file_format(char *filename) {
 154         unsigned int i;
 155         static const char *extension[] = {"pgx", "pnm", "pgm", "ppm", "bmp","tif", "raw", "tga", "png", "j2k", "jp2", "jpt", "j2c", "jpc"  };
 156         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 };
 157         char * ext = strrchr(filename, '.');
 158         if (ext == NULL)
 159                 return -1;
 160         ext++;
 161         if(ext) {
 162                 for(i = 0; i < sizeof(format)/sizeof(*format); i++) {
 163                         if(_strnicmp(ext, extension[i], 3) == 0) {
 164                                 return format[i];
 165                         }
 166                 }
 167         }
 168
 169         return -1;
 170 }
 171
 172 char get_next_file(int imageno,dircnt_t *dirptr,img_fol_t *img_fol, opj_dparameters_t *parameters){
 173         char image_filename[OPJ_PATH_LEN], infilename[OPJ_PATH_LEN],outfilename[OPJ_PATH_LEN],temp_ofname[OPJ_PATH_LEN];
 174         char *temp_p, temp1[OPJ_PATH_LEN]="";
 175
 176         strcpy(image_filename,dirptr->filename[imageno]);
 177         fprintf(stderr,"File Number %d \"%s\"\n",imageno,image_filename);
 178         parameters->decod_format = get_file_format(image_filename);
 179         if (parameters->decod_format == -1)
 180                 return 1;
 181         sprintf(infilename,"%s/%s",img_fol->imgdirpath,image_filename);
 182         strncpy(parameters->infile, infilename, sizeof(infilename));
 183
 184         /* Set output file */
 185         strcpy(temp_ofname,strtok(image_filename,"."));
 186         while((temp_p = strtok(NULL,".")) != NULL){
 187                 strcat(temp_ofname,temp1);
 188                 sprintf(temp1,".%s",temp_p);
 189         }
 190         if(img_fol->set_out_format==1){
 191                 sprintf(outfilename,"%s/%s.%s",img_fol->imgdirpath,temp_ofname,img_fol->out_format);
 192                 strncpy(parameters->outfile, outfilename, sizeof(outfilename));
 193         }
 194         return 0;
 195 }
 196
 197 /* -------------------------------------------------------------------------- */
 198 int parse_cmdline_decoder(int argc, char **argv, opj_dparameters_t *parameters,img_fol_t *img_fol) {
 199         /* parse the command line */
 200         int totlen, c;
 201         opj_option_t long_option[]={
 202                 {"ImgDir",REQ_ARG, NULL ,'y'},
 203         };
 204         const char optlist[] = "i:o:h";
 205
 206         totlen=sizeof(long_option);
 207         img_fol->set_out_format = 0;
 208         do {
 209                 c = opj_getopt_long(argc, argv,optlist,long_option,totlen);
 210                 if (c == -1)
 211                         break;
 212                 switch (c) {
 213                         case 'i':                       /* input file */
 214                         {
 215                                 char *infile = opj_optarg;
 216                                 parameters->decod_format = get_file_format(infile);
 217                                 switch(parameters->decod_format) {
 218                                         case J2K_CFMT:
 219                                                 break;
 220                                         case JP2_CFMT:
 221                                                 break;
 222                                         case JPT_CFMT:
 223                                                 break;
 224                                         default:
 225                                                 fprintf(stderr,
 226                                                         "!! Unrecognized format for infile : %s [accept only *.j2k, *.jp2, *.jpc or *.jpt] !!\n\n",
 227                                                         infile);
 228                                                 return 1;
 229                                 }
 230                                 strncpy(parameters->infile, infile, sizeof(parameters->infile)-1);
 231                         }
 232                         break;
 233
 234                         /* ------------------------------------------------------ */
 235
 236                         case 'o':     /* output file */
 237                         {
 238                           char *outfile = opj_optarg;
 239                           strncpy(parameters->outfile, outfile, sizeof(parameters->outfile)-1);
 240                         }
 241                         break;
 242
 243                                 /* ----------------------------------------------------- */
 244
 245                         case 'h':                       /* display an help description */
 246                                 decode_help_display();
 247                                 return 1;
 248
 249                                 /* ------------------------------------------------------ */
 250
 251                         case 'y':                       /* Image Directory path */
 252                                 {
 253                                         img_fol->imgdirpath = (char*)malloc(strlen(opj_optarg) + 1);
 254                                         strcpy(img_fol->imgdirpath,opj_optarg);
 255                                         img_fol->set_imgdir=1;
 256                                 }
 257                                 break;
 258
 259                                 /* ----------------------------------------------------- */
 260
 261                         default:
 262                                 fprintf(stderr,"WARNING -> this option is not valid \"-%c %s\"\n",c, opj_optarg);
 263                                 break;
 264                 }
 265         }while(c != -1);
 266
 267         /* check for possible errors */
 268         if(img_fol->set_imgdir==1){
 269                 if(!(parameters->infile[0]==0)){
 270                         fprintf(stderr, "Error: options -ImgDir and -i cannot be used together !!\n");
 271                         return 1;
 272                 }
 273                 if(img_fol->set_out_format == 0){
 274                         fprintf(stderr, "Error: When -ImgDir is used, -OutFor <FORMAT> must be used !!\n");
 275                         fprintf(stderr, "Only one format allowed! Valid format PGM, PPM, PNM, PGX, BMP, TIF, RAW and TGA!!\n");
 276                         return 1;
 277                 }
 278                 if(!((parameters->outfile[0] == 0))){
 279                         fprintf(stderr, "Error: options -ImgDir and -o cannot be used together !!\n");
 280                         return 1;
 281                 }
 282         }else{
 283                 if((parameters->infile[0] == 0) ) {
 284                         fprintf(stderr, "Example: %s -i image.j2k\n",argv[0]);
 285                         fprintf(stderr, "    Try: %s -h\n",argv[0]);
 286                         return 1;
 287                 }
 288         }
 289
 290         return 0;
 291 }
 292
 293 /* -------------------------------------------------------------------------- */
 294
 295 /**
 296 sample error callback expecting a FILE* client object
 297 */
 298 void error_callback(const char *msg, void *client_data) {
 299         FILE *stream = (FILE*)client_data;
 300         fprintf(stream, "[ERROR] %s", msg);
 301 }
 302 /**
 303 sample warning callback expecting a FILE* client object
 304 */
 305 void warning_callback(const char *msg, void *client_data) {
 306         FILE *stream = (FILE*)client_data;
 307         fprintf(stream, "[WARNING] %s", msg);
 308 }
 309 /**
 310 sample debug callback expecting no client object
 311 */
 312 void info_callback(const char *msg, void *client_data) {
 313         (void)client_data;
 314         fprintf(stdout, "[INFO] %s", msg);
 315 }
 316
 317 /* -------------------------------------------------------------------------- */
 318
 319 int main(int argc, char *argv[])
 320 {
 321         int ret;
 322         opj_dparameters_t parameters;   /* decompression parameters */
 323         img_fol_t img_fol;
 324         opj_event_mgr_t event_mgr;              /* event manager */
 325         opj_image_header_t* image = NULL;
 326         FILE *fsrc = NULL, *fout = NULL;
 327         int num_images;
 328         int i,imageno;
 329         dircnt_t *dirptr = NULL;
 330         opj_codec_t* dinfo = NULL;      /* handle to a decompressor */
 331         opj_stream_t *cio = NULL;
 332         opj_codestream_info_t* cstr_info =NULL;  /* Codestream information structure */
 333         /* OPJ_INT32 l_tile_x0,l_tile_y0; */
 334         /* OPJ_UINT32 l_tile_width,l_tile_height,l_nb_tiles_x,l_nb_tiles_y; */
 335
 336
 337         /* FIXME configure the event callbacks (not required) */
 338         memset(&event_mgr, 0, sizeof(opj_event_mgr_t));
 339         event_mgr.error_handler = error_callback;
 340         event_mgr.warning_handler = warning_callback;
 341         event_mgr.info_handler = info_callback;
 342         event_mgr.client_data = stderr;
 343
 344         /* set decoding parameters to default values */
 345         opj_set_default_decoder_parameters(&parameters);
 346
 347         /* Initialize img_fol */
 348         memset(&img_fol,0,sizeof(img_fol_t));
 349
 350         /* parse input and get user encoding parameters */
 351         if(parse_cmdline_decoder(argc, argv, &parameters,&img_fol) == 1) {
 352                 return EXIT_FAILURE;
 353         }
 354
 355         /* Initialize reading of directory */
 356         if(img_fol.set_imgdir==1){
 357                 num_images=get_num_images(img_fol.imgdirpath);
 358
 359                 dirptr=(dircnt_t*)malloc(sizeof(dircnt_t));
 360                 if(dirptr){
 361                         dirptr->filename_buf = (char*)malloc(num_images*OPJ_PATH_LEN*sizeof(char));     /* Stores at max 10 image file names */
 362                         dirptr->filename = (char**) malloc(num_images*sizeof(char*));
 363
 364                         if(!dirptr->filename_buf){
 365                                 return EXIT_FAILURE;
 366                         }
 367
 368                         for(i=0;i<num_images;i++){
 369                                 dirptr->filename[i] = dirptr->filename_buf + i*OPJ_PATH_LEN;
 370                         }
 371                 }
 372                 if(load_images(dirptr,img_fol.imgdirpath)==1){
 373                         return EXIT_FAILURE;
 374                 }
 375
 376                 if (num_images==0){
 377                         fprintf(stdout,"Folder is empty\n");
 378                         return EXIT_FAILURE;
 379                 }
 380         }else{
 381                 num_images=1;
 382         }
 383
 384         /* Try to open for writing the output file if necessary */
 385         if (parameters.outfile[0] != 0){
 386                 fout = fopen(parameters.outfile,"w");
 387                 if (!fout){
 388                         fprintf(stderr, "ERROR -> failed to open %s for writing\n", parameters.outfile);
 389                         return EXIT_FAILURE;
 390                 }
 391         }
 392         else
 393                 fout = stdout;
 394
 395         /* Read the header of each image one by one */
 396         for(imageno = 0; imageno < num_images ; imageno++){
 397                 image = NULL;
 398                 fprintf(stderr,"\n");
 399
 400                 if(img_fol.set_imgdir==1){
 401                         if (get_next_file(imageno, dirptr,&img_fol, &parameters)) {
 402                                 fprintf(stderr,"skipping file...\n");
 403                                 continue;
 404                         }
 405                 }
 406
 407                 /*NEW V2 STYLE*/
 408                 /* read the input file and put it in memory */
 409                 /* ---------------------------------------- */
 410                 fsrc = fopen(parameters.infile, "rb");
 411                 if (!fsrc) {
 412                         fprintf(stderr, "ERROR -> failed to open %s for reading\n", parameters.infile);
 413                         return EXIT_FAILURE;
 414                 }
 415
 416                 cio = opj_stream_create_default_file_stream(fsrc,1);
 417
 418                 /* decode the code-stream */
 419                 /* ---------------------- */
 420
 421                 switch(parameters.decod_format) {
 422                         case J2K_CFMT:
 423                         {       /* JPEG-2000 codestream */
 424
 425                                 /* get a decoder handle */
 426                                 dinfo = opj_create_decompress_v2(CODEC_J2K);
 427                                 break;
 428                         }
 429                         case JP2_CFMT:
 430                         {       /* JPEG 2000 compressed image data */
 431
 432                                 /* get a decoder handle */
 433                                 dinfo = opj_create_decompress_v2(CODEC_JP2);
 434                                 break;
 435                         }
 436                         case JPT_CFMT:
 437                         {       /* JPEG 2000, JPIP */
 438
 439                                 /* get a decoder handle */
 440                                 dinfo = opj_create_decompress_v2(CODEC_JPT);
 441                                 break;
 442                         }
 443                         default:
 444                                 fprintf(stderr, "skipping file..\n");
 445                                 opj_stream_destroy(cio);
 446                                 continue;
 447                 }
 448
 449                 /* setup the decoder decoding parameters using user parameters */
 450                 opj_setup_decoder_v2(dinfo, &parameters, &event_mgr);
 451
 452                 if(! opj_read_header(cio, dinfo, &image, &cstr_info)){
 453                         fprintf(stderr, "ERROR -> j2k_dump: failed to read the header\n");
 454                         opj_stream_destroy(cio);
 455                         fclose(fsrc);
 456                         opj_destroy_codec(dinfo);
 457                         return EXIT_FAILURE;
 458                 }
 459
 460                 /* Dump file informations from header */
 461
 462
 463                 /* close the byte stream */
 464                 opj_stream_destroy(cio);
 465                 fclose(fsrc);
 466
 467                 /* free remaining structures */
 468                 if (dinfo) {
 469                         opj_destroy_codec(dinfo);
 470                 }
 471
 472
 473                 opj_image_header_destroy(image);
 474
 475         }
 476 /*NEW V2 STYLE*/
 477
 478         /* Close the output file */
 479         fclose(fout);
 480
 481   return EXIT_SUCCESS;
 482 }
 483
 484
 485 static void j2k_dump_image(FILE *fd, opj_image_header_t * img) {
 486         int compno;
 487         fprintf(fd, "image {\n");
 488         fprintf(fd, "  x0=%d, y0=%d, x1=%d, y1=%d\n", img->x0, img->y0, img->x1, img->y1);
 489         fprintf(fd, "  numcomps=%d\n", img->numcomps);
 490         for (compno = 0; compno < img->numcomps; compno++) {
 491                 opj_image_comp_header_t *comp = &img->comps[compno];
 492                 fprintf(fd, "  comp %d {\n", compno);
 493                 fprintf(fd, "    dx=%d, dy=%d\n", comp->dx, comp->dy);
 494                 fprintf(fd, "    prec=%d\n", comp->prec);
 495                 /* fprintf(fd, "    bpp=%d\n", comp->bpp); */
 496                 fprintf(fd, "    sgnd=%d\n", comp->sgnd);
 497                 fprintf(fd, "  }\n");
 498         }
 499         fprintf(fd, "  XTOsiz=%d, YTOsiz=%d, XTsiz=%d, YTsiz=%d\n", img->tile_x0, img->tile_y0, img->tile_width, img->tile_height);
 500         fprintf(fd, "  Nb of tiles in x direction=%d, Nb of tiles in y direction=%d\n", img->nb_tiles_x, img->nb_tiles_y);
 501         fprintf(fd, "}\n");
 502 }
 503
 504 static void j2k_dump_cp(FILE *fd, opj_image_t * img, opj_cp_v2_t * cp) {
 505         int tileno, compno, layno, bandno, resno, numbands;
 506         fprintf(fd, "coding parameters {\n");
 507         fprintf(fd, "  tx0=%d, ty0=%d\n", cp->tx0, cp->ty0);
 508         fprintf(fd, "  tdx=%d, tdy=%d\n", cp->tdx, cp->tdy);
 509         fprintf(fd, "  tw=%d, th=%d\n", cp->tw, cp->th);
 510         for (tileno = 0; tileno < cp->tw * cp->th; tileno++) {
 511                 opj_tcp_v2_t *tcp = &cp->tcps[tileno];
 512                 fprintf(fd, "  tile %d {\n", tileno);
 513                 fprintf(fd, "    csty=%x\n", tcp->csty);
 514                 fprintf(fd, "    prg=%d\n", tcp->prg);
 515                 fprintf(fd, "    numlayers=%d\n", tcp->numlayers);
 516                 fprintf(fd, "    mct=%d\n", tcp->mct);
 517                 fprintf(fd, "    rates=");
 518                 for (layno = 0; layno < tcp->numlayers; layno++) {
 519                         fprintf(fd, "%.1f ", tcp->rates[layno]);
 520                 }
 521                 fprintf(fd, "\n");
 522                 for (compno = 0; compno < img->numcomps; compno++) {
 523                         opj_tccp_t *tccp = &tcp->tccps[compno];
 524                         fprintf(fd, "    comp %d {\n", compno);
 525                         fprintf(fd, "      csty=%x\n", tccp->csty);
 526                         fprintf(fd, "      numresolutions=%d\n", tccp->numresolutions);
 527                         fprintf(fd, "      cblkw=%d\n", tccp->cblkw);
 528                         fprintf(fd, "      cblkh=%d\n", tccp->cblkh);
 529                         fprintf(fd, "      cblksty=%x\n", tccp->cblksty);
 530                         fprintf(fd, "      qmfbid=%d\n", tccp->qmfbid);
 531                         fprintf(fd, "      qntsty=%d\n", tccp->qntsty);
 532                         fprintf(fd, "      numgbits=%d\n", tccp->numgbits);
 533                         fprintf(fd, "      roishift=%d\n", tccp->roishift);
 534                         fprintf(fd, "      stepsizes=");
 535                         numbands = tccp->qntsty == J2K_CCP_QNTSTY_SIQNT ? 1 : tccp->numresolutions * 3 - 2;
 536                         for (bandno = 0; bandno < numbands; bandno++) {
 537                                 fprintf(fd, "(%d,%d) ", tccp->stepsizes[bandno].mant,
 538                                         tccp->stepsizes[bandno].expn);
 539                         }
 540                         fprintf(fd, "\n");
 541
 542                         if (tccp->csty & J2K_CCP_CSTY_PRT) {
 543                                 fprintf(fd, "      prcw=");
 544                                 for (resno = 0; resno < tccp->numresolutions; resno++) {
 545                                         fprintf(fd, "%d ", tccp->prcw[resno]);
 546                                 }
 547                                 fprintf(fd, "\n");
 548                                 fprintf(fd, "      prch=");
 549                                 for (resno = 0; resno < tccp->numresolutions; resno++) {
 550                                         fprintf(fd, "%d ", tccp->prch[resno]);
 551                                 }
 552                                 fprintf(fd, "\n");
 553                         }
 554                         fprintf(fd, "    }\n");
 555                 } /*end of component*/
 556                 fprintf(fd, "  }\n");
 557         } /*end of tile */
 558         fprintf(fd, "}\n");
 559 }
 560