src/bin/mj2/opj_mj2_decompress.c

   1 /*
   2 * Copyright (c) 2003-2004, Francois-Olivier Devaux
   3 * Copyright (c) 2002-2004,  Communications and remote sensing Laboratory, Universite catholique de Louvain, Belgium
   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 <stdlib.h>
  30 #include <string.h>
  31
  32 #include "opj_apps_config.h"
  33 #include "openjpeg.h"
  34 #include "j2k_lib.h"
  35 #include "cio.h"
  36 #include "j2k.h"
  37 #include "jp2.h"
  38 #include "mj2.h"
  39 #include "mj2_convert.h"
  40
  41 #ifdef OPJ_HAVE_LIBLCMS2
  42 #include <lcms2.h>
  43 #endif
  44 #ifdef OPJ_HAVE_LIBLCMS1
  45 #include <lcms.h>
  46 #endif
  47 #include "color.h"
  48 /* -------------------------------------------------------------------------- */
  49
  50 /**
  51 sample error callback expecting a FILE* client object
  52 */
  53 static void error_callback(const char *msg, void *client_data) {
  54         FILE *stream = (FILE*)client_data;
  55         fprintf(stream, "[ERROR] %s", msg);
  56 }
  57 /**
  58 sample warning callback expecting a FILE* client object
  59 */
  60 static void warning_callback(const char *msg, void *client_data) {
  61         FILE *stream = (FILE*)client_data;
  62         fprintf(stream, "[WARNING] %s", msg);
  63 }
  64 /**
  65 sample debug callback expecting a FILE* client object
  66 */
  67 static void info_callback(const char *msg, void *client_data) {
  68         FILE *stream = (FILE*)client_data;
  69         fprintf(stream, "[INFO] %s", msg);
  70 }
  71
  72 /* -------------------------------------------------------------------------- */
  73
  74
  75 int main(int argc, char *argv[]) {
  76         mj2_dparameters_t mj2_parameters;                       /* decompression parameters */
  77         opj_dinfo_t* dinfo;
  78         opj_event_mgr_t event_mgr;              /* event manager */
  79         opj_cio_t *cio = NULL;
  80   unsigned int tnum, snum;
  81   opj_mj2_t *movie;
  82   mj2_tk_t *track;
  83   mj2_sample_t *sample;
  84   unsigned char* frame_codestream;
  85   FILE *file, *outfile;
  86   char outfilename[50];
  87   opj_image_t *img = NULL;
  88         unsigned int max_codstrm_size = 0;
  89         double total_time = 0;
  90         unsigned int numframes = 0;
  91
  92   if (argc != 3) {
  93     printf("Usage: %s inputfile.mj2 outputfile.yuv\n",argv[0]);
  94     return 1;
  95   }
  96
  97   file = fopen(argv[1], "rb");
  98
  99   if (!file) {
 100     fprintf(stderr, "failed to open %s for reading\n", argv[1]);
 101     return 1;
 102   }
 103
 104   /* Checking output file */
 105   outfile = fopen(argv[2], "w");
 106   if (!file) {
 107     fprintf(stderr, "failed to open %s for writing\n", argv[2]);
 108     return 1;
 109   }
 110   fclose(outfile);
 111
 112         /*
 113         configure the event callbacks (not required)
 114         setting of each callback is optionnal
 115         */
 116         memset(&event_mgr, 0, sizeof(opj_event_mgr_t));
 117         event_mgr.error_handler = error_callback;
 118         event_mgr.warning_handler = warning_callback;
 119         event_mgr.info_handler = NULL;
 120
 121         /* get a MJ2 decompressor handle */
 122         dinfo = mj2_create_decompress();
 123         movie = (opj_mj2_t*)dinfo->mj2_handle;
 124
 125         /* catch events using our callbacks and give a local context */
 126         opj_set_event_mgr((opj_common_ptr)dinfo, &event_mgr, stderr);
 127
 128         memset(&mj2_parameters, 0, sizeof(mj2_dparameters_t));
 129         /* set J2K decoding parameters to default values */
 130         opj_set_default_decoder_parameters(&mj2_parameters.j2k_parameters);
 131
 132         /* setup the decoder decoding parameters using user parameters */
 133         mj2_setup_decoder(movie, &mj2_parameters);
 134
 135   if (mj2_read_struct(file, movie)) /* Creating the movie structure */
 136     return 1;
 137
 138   /* Decode first video track */
 139         for (tnum=0; tnum < (unsigned int)(movie->num_htk + movie->num_stk + movie->num_vtk); tnum++) {
 140                 if (movie->tk[tnum].track_type == 0)
 141                         break;
 142         }
 143
 144         if (movie->tk[tnum].track_type != 0) {
 145                 printf("Error. Movie does not contain any video track\n");
 146                 return 1;
 147         }
 148
 149   track = &movie->tk[tnum];
 150
 151   /* Output info on first video tracl */
 152   fprintf(stdout,"The first video track contains %d frames.\nWidth: %d, Height: %d \n\n",
 153     track->num_samples, track->w, track->h);
 154
 155         max_codstrm_size = track->sample[0].sample_size-8;
 156         frame_codestream = (unsigned char*) malloc(max_codstrm_size * sizeof(unsigned char));
 157
 158         numframes = track->num_samples;
 159
 160   for (snum=0; snum < numframes; snum++)
 161   {
 162                 double init_time = opj_clock();
 163                 double elapsed_time;
 164
 165     sample = &track->sample[snum];
 166                 if (sample->sample_size-8 > max_codstrm_size) {
 167                         max_codstrm_size =  sample->sample_size-8;
 168                         if ((frame_codestream = (unsigned char*)
 169                                 realloc(frame_codestream, max_codstrm_size)) == NULL) {
 170                                 printf("Error reallocation memory\n");
 171                                 return 1;
 172                         };
 173                 }
 174     fseek(file,sample->offset+8,SEEK_SET);
 175     fread(frame_codestream, sample->sample_size-8, 1, file);  /* Assuming that jp and ftyp markers size do */
 176
 177                 /* open a byte stream */
 178                 cio = opj_cio_open((opj_common_ptr)dinfo, frame_codestream, sample->sample_size-8);
 179
 180                 img = opj_decode(dinfo, cio); /* Decode J2K to image */
 181
 182 #ifdef WANT_SYCC_TO_RGB
 183         if(img->color_space == CLRSPC_SYCC)
 184   {
 185         color_sycc_to_rgb(img);
 186   }
 187 #endif
 188
 189         if(img->icc_profile_buf)
 190   {
 191 #if defined(OPJ_HAVE_LIBLCMS1) || defined(OPJ_HAVE_LIBLCMS2)
 192         color_apply_icc_profile(img);
 193 #endif
 194
 195         free(img->icc_profile_buf);
 196         img->icc_profile_buf = NULL; img->icc_profile_len = 0;
 197   }
 198
 199     if (((img->numcomps == 3) && (img->comps[0].dx == img->comps[1].dx / 2)
 200       && (img->comps[0].dx == img->comps[2].dx / 2 ) && (img->comps[0].dx == 1))
 201       || (img->numcomps == 1)) {
 202
 203       if (!imagetoyuv(img, argv[2]))    /* Convert image to YUV */
 204                                 return 1;
 205     }
 206     else if ((img->numcomps == 3) &&
 207       (img->comps[0].dx == 1) && (img->comps[1].dx == 1)&&
 208       (img->comps[2].dx == 1))/* If YUV 4:4:4 input --> to bmp */
 209     {
 210       fprintf(stdout,"The frames will be output in a bmp format (output_1.bmp, ...)\n");
 211       sprintf(outfilename,"output_%d.bmp",snum);
 212       if (imagetobmp(img, outfilename)) /* Convert image to BMP */
 213                                 return 1;
 214
 215     }
 216     else {
 217       fprintf(stdout,"Image component dimensions are unknown. Unable to output image\n");
 218       fprintf(stdout,"The frames will be output in a j2k file (output_1.j2k, ...)\n");
 219
 220       sprintf(outfilename,"output_%d.j2k",snum);
 221       outfile = fopen(outfilename, "wb");
 222       if (!outfile) {
 223                                 fprintf(stderr, "failed to open %s for writing\n",outfilename);
 224                                 return 1;
 225       }
 226       fwrite(frame_codestream,sample->sample_size-8,1,outfile);
 227       fclose(outfile);
 228     }
 229                 /* close the byte stream */
 230                 opj_cio_close(cio);
 231                 /* free image data structure */
 232                 opj_image_destroy(img);
 233                 elapsed_time = opj_clock()-init_time;
 234                 fprintf(stderr, "Frame number %d/%d decoded in %.2f mseconds\n", snum + 1, numframes, elapsed_time*1000);
 235                 total_time += elapsed_time;
 236
 237   }
 238
 239         free(frame_codestream);
 240   fclose(file);
 241
 242         /* free remaining structures */
 243         if(dinfo) {
 244                 mj2_destroy_decompress((opj_mj2_t*)dinfo->mj2_handle);
 245         }
 246         free(dinfo);
 247
 248         fprintf(stdout, "%d frame(s) correctly decompressed\n", snum);
 249         fprintf(stdout,"Total decoding time: %.2f seconds (%.1f fps)\n", total_time, (float)numframes/total_time);
 250
 251   return 0;
 252 }