Merge pull request #750 from mayeut/issue-729-2
[openjpeg.git] / src / bin / jp2 / opj_decompress.c
index cfc04a9eda7a9f3a81ec64338acd48277cc13611..f3b1cd5c9a3569ef39b4ed5d18f195f60a674396 100644 (file)
@@ -75,6 +75,7 @@
 #include "color.h"
 
 #include "format_defs.h"
+#include "opj_string.h"
 
 typedef struct dircnt{
        /** Buffer for holding images read from Directory*/
@@ -121,6 +122,8 @@ typedef struct opj_decompress_params
        int decod_format;
        /** output file format 0: PGX, 1: PxM, 2: BMP */
        int cod_format;
+       /** index file name */
+       char indexfilename[OPJ_PATH_LEN];
        
        /** Decoding area left boundary */
        OPJ_UINT32 DA_x0;
@@ -145,6 +148,8 @@ typedef struct opj_decompress_params
        int force_rgb;
        /* upsample components according to their dx/dy values */
        int upsample;
+       /* split output components to different files */
+       int split_pnm;
 }opj_decompress_parameters;
 
 /* -------------------------------------------------------------------------- */
@@ -155,7 +160,7 @@ int get_file_format(const char *filename);
 char get_next_file(int imageno,dircnt_t *dirptr,img_fol_t *img_fol, opj_decompress_parameters *parameters);
 static int infile_format(const char *fname);
 
-int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *parameters,img_fol_t *img_fol, char *indexfilename);
+int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *parameters,img_fol_t *img_fol);
 int parse_DA_values( char* inArg, unsigned int *DA_x0, unsigned int *DA_y0, unsigned int *DA_x1, unsigned int *DA_y1);
 
 static opj_image_t* convert_gray_to_rgb(opj_image_t* original);
@@ -173,12 +178,12 @@ static void decode_help_display(void) {
                       "        Image file Directory path \n"
                       "  -OutFor <PBM|PGM|PPM|PNM|PAM|PGX|PNG|BMP|TIF|RAW|RAWL|TGA>\n"
                       "    REQUIRED only if -ImgDir is used\n"
-                      "        Output format for decompressed images.\n"
-                      "  -i <compressed file>\n"
+                      "        Output format for decompressed images.\n");
+       fprintf(stdout,"  -i <compressed file>\n"
                       "    REQUIRED only if an Input image directory is not specified\n"
                       "    Currently accepts J2K-files, JP2-files and JPT-files. The file type\n"
-                      "    is identified based on its suffix.\n"
-                      "  -o <decompressed file>\n"
+                      "    is identified based on its suffix.\n");
+       fprintf(stdout,"  -o <decompressed file>\n"
                       "    REQUIRED\n"
                       "    Currently accepts formats specified above (see OutFor option)\n"
                       "    Binary data is written to the file (not ascii). If a PGX\n"
@@ -186,8 +191,8 @@ static void decode_help_display(void) {
                       "    components: an indice starting from 0 will then be appended to the\n"
                       "    output filename, just before the \"pgx\" extension. If a PGM filename\n"
                       "    is given and there are more than one component, only the first component\n"
-                      "    will be written to the file.\n"
-                      "  -r <reduce factor>\n"
+                      "    will be written to the file.\n");
+       fprintf(stdout,"  -r <reduce factor>\n"
                       "    Set the number of highest resolution levels to be discarded. The\n"
                       "    image resolution is effectively divided by 2 to the power of the\n"
                       "    number of discarded levels. The reduce factor is limited by the\n"
@@ -195,8 +200,8 @@ static void decode_help_display(void) {
                       "  -l <number of quality layers to decode>\n"
                       "    Set the maximum number of quality layers to decode. If there are\n"
                       "    less quality layers than the specified number, all the quality layers\n"
-                      "    are decoded.\n"
-                      "  -x  \n" 
+                      "    are decoded.\n");
+       fprintf(stdout,"  -x  \n"
                       "    Create an index file *.Idx (-x index_name.Idx) \n"
                       "  -d <x0,y0,x1,y1>\n"
                       "    OPTIONAL\n"
@@ -205,19 +210,21 @@ static void decode_help_display(void) {
                       "  -t <tile_number>\n"
                       "    OPTIONAL\n"
                       "    Set the tile number of the decoded tile. Follow the JPEG2000 convention from left-up to bottom-up\n"
-                      "    By default all tiles are decoded.\n"
-                      "  -p <comp 0 precision>[C|S][,<comp 1 precision>[C|S][,...]]\n"
+                      "    By default all tiles are decoded.\n");
+       fprintf(stdout,"  -p <comp 0 precision>[C|S][,<comp 1 precision>[C|S][,...]]\n"
                       "    OPTIONAL\n"
-                      "    Force the precision (bit depth) of components.\n"
-                      "    There shall be at least 1 value. Theres no limit on the number of values (comma separated, last values ignored if too much values).\n"
+                      "    Force the precision (bit depth) of components.\n");
+       fprintf(stdout,"    There shall be at least 1 value. Theres no limit on the number of values (comma separated, last values ignored if too much values).\n"
                       "    If there are less values than components, the last value is used for remaining components.\n"
                       "    If 'C' is specified (default), values are clipped.\n"
                       "    If 'S' is specified, values are scaled.\n"
-                      "    A 0 value can be specified (meaning original bit depth).\n"
-                      "  -force-rgb\n"
+                      "    A 0 value can be specified (meaning original bit depth).\n");
+       fprintf(stdout,"  -force-rgb\n"
                       "    Force output image colorspace to RGB\n"
                       "  -upsample\n"
                       "    Downsampled components will be upsampled to image size\n"
+                      "  -split-pnm\n"
+                      "    Split output components to different files when writing to PNM\n"
                       "\n");
 /* UniPG>> */
 #ifdef USE_JPWL
@@ -248,7 +255,7 @@ static OPJ_BOOL parse_precision(const char* option, opj_decompress_parameters* p
        
        for(;;)
        {
-               OPJ_UINT32 prec;
+               int prec;
                char mode;
                char comma;
                int count;
@@ -266,7 +273,7 @@ static OPJ_BOOL parse_precision(const char* option, opj_decompress_parameters* p
                        count = 3;
                }
                if (count == 3) {
-                       if (prec > 32U) {
+                       if ((prec < 1) || (prec > 32)) {
                                fprintf(stderr,"Invalid precision %d in precision option %s\n", prec, option);
                                l_result = OPJ_FALSE;
                                break;
@@ -284,7 +291,7 @@ static OPJ_BOOL parse_precision(const char* option, opj_decompress_parameters* p
                        
                        if (parameters->precision == NULL) {
                                /* first one */
-                               parameters->precision = malloc(sizeof(opj_precision));
+                               parameters->precision = (opj_precision *)malloc(sizeof(opj_precision));
                                if (parameters->precision == NULL) {
                                        fprintf(stderr,"Could not allocate memory for precision option\n");
                                        l_result = OPJ_FALSE;
@@ -300,7 +307,7 @@ static OPJ_BOOL parse_precision(const char* option, opj_decompress_parameters* p
                                        break;
                                }
                                
-                               l_new = realloc(parameters->precision, l_new_size * sizeof(opj_precision));
+                               l_new = (opj_precision *)realloc(parameters->precision, l_new_size * sizeof(opj_precision));
                                if (l_new == NULL) {
                                        fprintf(stderr,"Could not allocate memory for precision option\n");
                                        l_result = OPJ_FALSE;
@@ -309,7 +316,7 @@ static OPJ_BOOL parse_precision(const char* option, opj_decompress_parameters* p
                                parameters->precision = l_new;
                        }
                        
-                       parameters->precision[parameters->nb_precision].prec = prec;
+                       parameters->precision[parameters->nb_precision].prec = (OPJ_UINT32)prec;
                        switch (mode) {
                                case 'C':
                                        parameters->precision[parameters->nb_precision].mode = OPJ_PREC_MODE_CLIP;
@@ -357,6 +364,7 @@ int get_num_images(char *imgdirpath){
                        continue;
                num_images++;
        }
+       closedir(dir);
        return num_images;
 }
 
@@ -383,6 +391,7 @@ int load_images(dircnt_t *dirptr, char *imgdirpath){
                strcpy(dirptr->filename[i],content->d_name);
                i++;
        }
+       closedir(dir);
        return 0;       
 }
 
@@ -391,7 +400,7 @@ int get_file_format(const char *filename) {
        unsigned int i;
        static const char *extension[] = {"pgx", "pnm", "pgm", "ppm", "bmp","tif", "raw", "rawl", "tga", "png", "j2k", "jp2", "jpt", "j2c", "jpc" };
        static const int format[] = { PGX_DFMT, PXM_DFMT, PXM_DFMT, PXM_DFMT, BMP_DFMT, TIF_DFMT, RAW_DFMT, RAWL_DFMT, TGA_DFMT, PNG_DFMT, J2K_CFMT, JP2_CFMT, JPT_CFMT, J2K_CFMT, J2K_CFMT };
-       char * ext = strrchr(filename, '.');
+       const char * ext = strrchr(filename, '.');
        if (ext == NULL)
                return -1;
        ext++;
@@ -423,7 +432,9 @@ char get_next_file(int imageno,dircnt_t *dirptr,img_fol_t *img_fol, opj_decompre
        parameters->decod_format = infile_format(infilename);
        if (parameters->decod_format == -1)
                return 1;
-       strncpy(parameters->infile, infilename, sizeof(infilename));
+       if (opj_strcpy_s(parameters->infile, sizeof(parameters->infile), infilename) != 0) {
+               return 1;
+       }
 
        /*Set output file*/
        strcpy(temp_ofname,strtok(image_filename,"."));
@@ -433,7 +444,9 @@ char get_next_file(int imageno,dircnt_t *dirptr,img_fol_t *img_fol, opj_decompre
        }
        if(img_fol->set_out_format==1){
                sprintf(outfilename,"%s/%s.%s",img_fol->imgdirpath,temp_ofname,img_fol->out_format);
-               strncpy(parameters->outfile, outfilename, sizeof(outfilename));
+               if (opj_strcpy_s(parameters->outfile, sizeof(parameters->outfile), outfilename) != 0) {
+                       return 1;
+               }
        }
        return 0;
 }
@@ -499,14 +512,15 @@ static int infile_format(const char *fname)
  * Parse the command line
  */
 /* -------------------------------------------------------------------------- */
-int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *parameters,img_fol_t *img_fol, char *indexfilename) {
+int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *parameters,img_fol_t *img_fol) {
        /* parse the command line */
        int totlen, c;
        opj_option_t long_option[]={
-               {"ImgDir",    REQ_ARG, NULL ,'y'},
-               {"OutFor",    REQ_ARG, NULL ,'O'},
-               {"force-rgb", NO_ARG,  &(parameters->force_rgb), 1},
-               {"upsample",  NO_ARG,  &(parameters->upsample),  1}
+               {"ImgDir",    REQ_ARG, NULL,'y'},
+               {"OutFor",    REQ_ARG, NULL,'O'},
+               {"force-rgb", NO_ARG,  NULL, 1},
+               {"upsample",  NO_ARG,  NULL, 1},
+               {"split-pnm", NO_ARG,  NULL, 1}
        };
 
        const char optlist[] = "i:o:r:l:x:d:t:p:"
@@ -517,6 +531,10 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
 #endif /* USE_JPWL */
 /* <<UniPG */
             "h"                ;
+
+       long_option[2].flag = &(parameters->force_rgb);
+       long_option[3].flag = &(parameters->upsample);
+       long_option[4].flag = &(parameters->split_pnm);
        totlen=sizeof(long_option);
        opj_reset_options_reading();
        img_fol->set_out_format = 0;
@@ -550,7 +568,10 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
                                                        infile);
                                                return 1;
                                }
-                               strncpy(parameters->infile, infile, sizeof(parameters->infile)-1);
+                               if (opj_strcpy_s(parameters->infile, sizeof(parameters->infile), infile) != 0) {
+                                       fprintf(stderr, "[ERROR] Path is too long\n");
+                                       return 1;
+                               }
                        }
                        break;
                                
@@ -578,10 +599,13 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
                                        case PNG_DFMT:
                                                break;
                                        default:
-                                               fprintf(stderr, "Unknown output format image %s [only *.pnm, *.pgm, *.ppm, *.pgx, *.bmp, *.tif, *.raw or *.tga]!! \n", outfile);
+                                               fprintf(stderr, "Unknown output format image %s [only *.png, *.pnm, *.pgm, *.ppm, *.pgx, *.bmp, *.tif, *.raw or *.tga]!!\n", outfile);
                                                return 1;
                                }
-                               strncpy(parameters->outfile, outfile, sizeof(parameters->outfile)-1);
+                               if (opj_strcpy_s(parameters->outfile, sizeof(parameters->outfile), outfile) != 0) {
+                                       fprintf(stderr, "[ERROR] Path is too long\n");
+                                       return 1;
+                               }
                        }
                        break;
                        
@@ -620,7 +644,7 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
                                                img_fol->out_format = "png";
                                                break;
                                        default:
-                                               fprintf(stderr, "Unknown output format image %s [only *.pnm, *.pgm, *.ppm, *.pgx, *.bmp, *.tif, *.raw or *.tga]!! \n", outformat);
+                                               fprintf(stderr, "Unknown output format image %s [only *.png, *.pnm, *.pgm, *.ppm, *.pgx, *.bmp, *.tif, *.raw or *.tga]!!\n", outformat);
                                                return 1;
                                                break;
                                }
@@ -665,11 +689,14 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
 
                        case 'd':               /* Input decode ROI */
                        {
-                               int size_optarg = (int)strlen(opj_optarg) + 1;
-                               char *ROI_values = (char*) malloc((size_t)size_optarg);
+                               size_t size_optarg = (size_t)strlen(opj_optarg) + 1U;
+                               char *ROI_values = (char*) malloc(size_optarg);
+                               if (ROI_values == NULL) {
+                                       fprintf(stderr, "[ERROR] Couldn't allocate memory\n");
+                                       return 1;
+                               }
                                ROI_values[0] = '\0';
-                               strncpy(ROI_values, opj_optarg, strlen(opj_optarg));
-                               ROI_values[strlen(opj_optarg)] = '\0';
+                               memcpy(ROI_values, opj_optarg, size_optarg);
                                /*printf("ROI_values = %s [%d / %d]\n", ROI_values, strlen(ROI_values), size_optarg ); */
                                parse_DA_values( ROI_values, &parameters->DA_x0, &parameters->DA_y0, &parameters->DA_x1, &parameters->DA_y1);
 
@@ -690,8 +717,10 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
 
                        case 'x':                       /* Creation of index file */
                                {
-                                       char *index = opj_optarg;
-                                       strncpy(indexfilename, index, OPJ_PATH_LEN);
+                                       if (opj_strcpy_s(parameters->indexfilename, sizeof(parameters->indexfilename), opj_optarg) != 0) {
+                                               fprintf(stderr, "[ERROR] Path is too long\n");
+                                               return 1;
+                                       }
                                }
                                break;
                                
@@ -852,7 +881,7 @@ OPJ_FLOAT64 opj_clock(void) {
        /* cout << "freq = " << ((double) freq.QuadPart) << endl; */
     /* t is the high resolution performance counter (see MSDN) */
     QueryPerformanceCounter ( & t ) ;
-    return ( t.QuadPart /(OPJ_FLOAT64) freq.QuadPart ) ;
+       return freq.QuadPart ? (t.QuadPart / (OPJ_FLOAT64)freq.QuadPart) : 0;
 #else
        /* Unix or Linux: use resource usage */
     struct rusage t;
@@ -1079,7 +1108,7 @@ static opj_image_t* upsample_image_components(opj_image_t* original)
                                l_dst += l_new_cmp->w;
                        }
                        
-                       if(l_new_cmp->h > (l_org_cmp->dy - 1U)) { /* check substraction overflow for really small images */
+                       if(l_new_cmp->h > (l_org_cmp->dy - 1U)) { /* check subtraction overflow for really small images */
                                for (; y < l_new_cmp->h - (l_org_cmp->dy - 1U); y += l_org_cmp->dy) {
                                        OPJ_UINT32 x, dy;
                                        OPJ_UINT32 xorg;
@@ -1088,7 +1117,7 @@ static opj_image_t* upsample_image_components(opj_image_t* original)
                                        for (x = 0U; x < xoff; ++x) {
                                                l_dst[x] = 0;
                                        }
-                                       if (l_new_cmp->w > (l_org_cmp->dx - 1U)) { /* check substraction overflow for really small images */
+                                       if (l_new_cmp->w > (l_org_cmp->dx - 1U)) { /* check subtraction overflow for really small images */
                                                for (; x < l_new_cmp->w - (l_org_cmp->dx - 1U); x += l_org_cmp->dx, ++xorg) {
                                                        OPJ_UINT32 dx;
                                                        for (dx = 0U; dx < l_org_cmp->dx; ++dx) {
@@ -1116,7 +1145,7 @@ static opj_image_t* upsample_image_components(opj_image_t* original)
                                for (x = 0U; x < xoff; ++x) {
                                        l_dst[x] = 0;
                                }
-                               if (l_new_cmp->w > (l_org_cmp->dx - 1U)) { /* check substraction overflow for really small images */
+                               if (l_new_cmp->w > (l_org_cmp->dx - 1U)) { /* check subtraction overflow for really small images */
                                        for (; x < l_new_cmp->w - (l_org_cmp->dx - 1U); x += l_org_cmp->dx, ++xorg) {
                                                OPJ_UINT32 dx;
                                                for (dx = 0U; dx < l_org_cmp->dx; ++dx) {
@@ -1156,8 +1185,6 @@ int main(int argc, char **argv)
        opj_codec_t* l_codec = NULL;                            /* Handle to a decompressor */
        opj_codestream_index_t* cstr_index = NULL;
 
-       char indexfilename[OPJ_PATH_LEN];       /* index file name */
-
        OPJ_INT32 num_images, imageno;
        img_fol_t img_fol;
        dircnt_t *dirptr = NULL;
@@ -1168,14 +1195,11 @@ int main(int argc, char **argv)
        /* set decoding parameters to default values */
        set_default_parameters(&parameters);
 
-       /* FIXME Initialize indexfilename and img_fol */
-       *indexfilename = 0;
-
        /* Initialize img_fol */
        memset(&img_fol,0,sizeof(img_fol_t));
 
        /* parse input and get user encoding parameters */
-       if(parse_cmdline_decoder(argc, argv, &parameters,&img_fol, indexfilename) == 1) {
+       if(parse_cmdline_decoder(argc, argv, &parameters,&img_fol) == 1) {
                destroy_parameters(&parameters);
                return EXIT_FAILURE;
        }
@@ -1340,10 +1364,6 @@ int main(int argc, char **argv)
                /* Close the byte stream */
                opj_stream_destroy(l_stream);
 
-               if(image->color_space == OPJ_CLRSPC_SYCC){
-                       color_sycc_to_rgb(image); /* FIXME */
-               }
-               
                if( image->color_space != OPJ_CLRSPC_SYCC 
                        && image->numcomps == 3 && image->comps[0].dx == image->comps[0].dy
                        && image->comps[1].dx != 1 )
@@ -1351,9 +1371,22 @@ int main(int argc, char **argv)
                else if (image->numcomps <= 2)
                        image->color_space = OPJ_CLRSPC_GRAY;
 
+               if(image->color_space == OPJ_CLRSPC_SYCC){
+                       color_sycc_to_rgb(image);
+               }
+               else if((image->color_space == OPJ_CLRSPC_CMYK) && (parameters.cod_format != TIF_DFMT)){
+                       color_cmyk_to_rgb(image);
+               }
+               else if(image->color_space == OPJ_CLRSPC_EYCC){
+                       color_esycc_to_rgb(image);
+               }
+               
                if(image->icc_profile_buf) {
 #if defined(OPJ_HAVE_LIBLCMS1) || defined(OPJ_HAVE_LIBLCMS2)
-                       color_apply_icc_profile(image); /* FIXME */
+                       if(image->icc_profile_len)
+                        color_apply_icc_profile(image);
+                       else
+                        color_cielab_to_rgb(image);
 #endif
                        free(image->icc_profile_buf);
                        image->icc_profile_buf = NULL; image->icc_profile_len = 0;
@@ -1433,7 +1466,7 @@ int main(int argc, char **argv)
                /* ------------------- */
                switch (parameters.cod_format) {
                case PXM_DFMT:                  /* PNM PGM PPM */
-                       if (imagetopnm(image, parameters.outfile)) {
+                       if (imagetopnm(image, parameters.outfile, parameters.split_pnm)) {
                 fprintf(stderr,"[ERROR] Outfile %s not generated\n",parameters.outfile);
         failed = 1;
                        }
@@ -1532,11 +1565,12 @@ int main(int argc, char **argv)
                /* destroy the codestream index */
                opj_destroy_cstr_index(&cstr_index);
 
-               if(failed) remove(parameters.outfile);
+               if(failed) (void)remove(parameters.outfile); /* ignore return value */
        }
        destroy_parameters(&parameters);
-       fprintf(stdout, "decode time: %d ms \n", (int)( (tCumulative * 1000) / numDecompressedImages));
-       scanf("%d");
+       if (numDecompressedImages) {
+               fprintf(stdout, "decode time: %d ms\n", (int)( (tCumulative * 1000.0) / (OPJ_FLOAT64)numDecompressedImages));
+       }
        return failed ? EXIT_FAILURE : EXIT_SUCCESS;
 }
 /*end main*/