src/bin/mj2/opj_mj2_extract.c

   1 /*
   2  * The copyright in this software is being made available under the 2-clauses
   3  * BSD License, included below. This software may be subject to other third
   4  * party and contributor rights, including patent rights, and no such rights
   5  * are granted under this license.
   6  *
   7  * Copyright (c) 2002-2014, Universite catholique de Louvain (UCL), Belgium
   8  * Copyright (c) 2002-2014, Professor Benoit Macq
   9  * Copyright (c) 2003-2007, Francois-Olivier Devaux
  10  * All rights reserved.
  11  *
  12  * Redistribution and use in source and binary forms, with or without
  13  * modification, are permitted provided that the following conditions
  14  * are met:
  15  * 1. Redistributions of source code must retain the above copyright
  16  *    notice, this list of conditions and the following disclaimer.
  17  * 2. Redistributions in binary form must reproduce the above copyright
  18  *    notice, this list of conditions and the following disclaimer in the
  19  *    documentation and/or other materials provided with the distribution.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
  22  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  25  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  26  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  27  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  28  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  29  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  31  * POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 #include <stdio.h>
  35 #include <stdlib.h>
  36 #include <string.h>
  37
  38 #include "openjpeg.h"
  39 #include "cio.h"
  40 #include "j2k.h"
  41 #include "jp2.h"
  42 #include "mj2.h"
  43
  44 /* -------------------------------------------------------------------------- */
  45
  46 /**
  47 sample error callback expecting a FILE* client object
  48 */
  49 void error_callback(const char *msg, void *client_data) {
  50         FILE *stream = (FILE*)client_data;
  51         fprintf(stream, "[ERROR] %s", msg);
  52 }
  53 /**
  54 sample warning callback expecting a FILE* client object
  55 */
  56 void warning_callback(const char *msg, void *client_data) {
  57         FILE *stream = (FILE*)client_data;
  58         fprintf(stream, "[WARNING] %s", msg);
  59 }
  60 /**
  61 sample debug callback expecting a FILE* client object
  62 */
  63 void info_callback(const char *msg, void *client_data) {
  64         FILE *stream = (FILE*)client_data;
  65         fprintf(stream, "[INFO] %s", msg);
  66 }
  67
  68 /* -------------------------------------------------------------------------- */
  69
  70
  71 int main(int argc, char *argv[]) {
  72         opj_dinfo_t* dinfo;
  73         opj_event_mgr_t event_mgr;              /* event manager */
  74   int tnum;
  75   unsigned int snum;
  76   opj_mj2_t *movie;
  77   mj2_tk_t *track;
  78   mj2_sample_t *sample;
  79   unsigned char* frame_codestream;
  80   FILE *file, *outfile;
  81   char outfilename[50];
  82         mj2_dparameters_t parameters;
  83
  84   if (argc != 3) {
  85     printf("Usage: %s mj2filename output_location\n",argv[0]);
  86     printf("Example: %s foreman.mj2 output/foreman\n",argv[0]);
  87     return 1;
  88   }
  89
  90   file = fopen(argv[1], "rb");
  91
  92   if (!file) {
  93     fprintf(stderr, "failed to open %s for reading\n", argv[1]);
  94     return 1;
  95   }
  96
  97         /*
  98         configure the event callbacks (not required)
  99         setting of each callback is optionnal
 100         */
 101         memset(&event_mgr, 0, sizeof(opj_event_mgr_t));
 102         event_mgr.error_handler = error_callback;
 103         event_mgr.warning_handler = warning_callback;
 104         event_mgr.info_handler = info_callback;
 105
 106         /* get a MJ2 decompressor handle */
 107         dinfo = mj2_create_decompress();
 108
 109         /* catch events using our callbacks and give a local context */
 110         opj_set_event_mgr((opj_common_ptr)dinfo, &event_mgr, stderr);
 111
 112         /* setup the decoder decoding parameters using user parameters */
 113         memset(&parameters, 0, sizeof(mj2_dparameters_t));
 114         movie = (opj_mj2_t*) dinfo->mj2_handle;
 115         mj2_setup_decoder(movie, &parameters);
 116
 117   if (mj2_read_struct(file, movie)) /* Creating the movie structure*/
 118     return 1;
 119
 120   /* Decode first video track */
 121   tnum = 0;
 122   while (movie->tk[tnum].track_type != 0)
 123     tnum ++;
 124
 125   track = &movie->tk[tnum];
 126
 127   fprintf(stdout,"Extracting %d frames from file...\n",track->num_samples);
 128
 129   for (snum=0; snum < track->num_samples; snum++)
 130   {
 131     sample = &track->sample[snum];
 132     frame_codestream = (unsigned char*) malloc (sample->sample_size-8); /* Skipping JP2C marker*/
 133     fseek(file,sample->offset+8,SEEK_SET);
 134     fread(frame_codestream,sample->sample_size-8,1, file);  /* Assuming that jp and ftyp markers size do*/
 135
 136     sprintf(outfilename,"%s_%05d.j2k",argv[2],snum);
 137     outfile = fopen(outfilename, "wb");
 138     if (!outfile) {
 139       fprintf(stderr, "failed to open %s for writing\n",outfilename);
 140       return 1;
 141     }
 142     fwrite(frame_codestream,sample->sample_size-8,1,outfile);
 143     fclose(outfile);
 144     free(frame_codestream);
 145     }
 146   fclose(file);
 147   fprintf(stdout, "%d frames correctly extracted\n", snum);
 148
 149         /* free remaining structures */
 150         if(dinfo) {
 151                 mj2_destroy_decompress((opj_mj2_t*)dinfo->mj2_handle);
 152         }
 153
 154   return 0;
 155 }