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 {
  51     FILE *stream = (FILE*)client_data;
  52     fprintf(stream, "[ERROR] %s", msg);
  53 }
  54 /**
  55 sample warning callback expecting a FILE* client object
  56 */
  57 void warning_callback(const char *msg, void *client_data)
  58 {
  59     FILE *stream = (FILE*)client_data;
  60     fprintf(stream, "[WARNING] %s", msg);
  61 }
  62 /**
  63 sample debug callback expecting a FILE* client object
  64 */
  65 void info_callback(const char *msg, void *client_data)
  66 {
  67     FILE *stream = (FILE*)client_data;
  68     fprintf(stream, "[INFO] %s", msg);
  69 }
  70
  71 /* -------------------------------------------------------------------------- */
  72
  73
  74 int main(int argc, char *argv[])
  75 {
  76     opj_dinfo_t* dinfo;
  77     opj_event_mgr_t event_mgr;      /* event manager */
  78     int tnum;
  79     unsigned int snum;
  80     opj_mj2_t *movie;
  81     mj2_tk_t *track;
  82     mj2_sample_t *sample;
  83     unsigned char* frame_codestream;
  84     FILE *file, *outfile;
  85     char outfilename[50];
  86     mj2_dparameters_t parameters;
  87
  88     if (argc != 3) {
  89         printf("Usage: %s mj2filename output_prefix\n", argv[0]);
  90         printf("Example: %s foreman.mj2 output/foreman\n", argv[0]);
  91         return 1;
  92     }
  93
  94     file = fopen(argv[1], "rb");
  95
  96     if (!file) {
  97         fprintf(stderr, "failed to open %s for reading\n", argv[1]);
  98         return 1;
  99     }
 100
 101     /*
 102     configure the event callbacks (not required)
 103     setting of each callback is optional
 104     */
 105     memset(&event_mgr, 0, sizeof(opj_event_mgr_t));
 106     event_mgr.error_handler = error_callback;
 107     event_mgr.warning_handler = warning_callback;
 108     event_mgr.info_handler = info_callback;
 109
 110     /* get a MJ2 decompressor handle */
 111     dinfo = mj2_create_decompress();
 112
 113     /* catch events using our callbacks and give a local context */
 114     opj_set_event_mgr((opj_common_ptr)dinfo, &event_mgr, stderr);
 115
 116     /* setup the decoder decoding parameters using user parameters */
 117     memset(&parameters, 0, sizeof(mj2_dparameters_t));
 118     movie = (opj_mj2_t*) dinfo->mj2_handle;
 119     mj2_setup_decoder(movie, &parameters);
 120
 121     if (mj2_read_struct(file, movie)) { /* Creating the movie structure*/
 122         return 1;
 123     }
 124
 125     /* Decode first video track */
 126     tnum = 0;
 127     while (movie->tk[tnum].track_type != 0) {
 128         tnum ++;
 129     }
 130
 131     track = &movie->tk[tnum];
 132
 133     fprintf(stdout, "Extracting %d frames from file...\n", track->num_samples);
 134
 135     for (snum = 0; snum < track->num_samples; snum++) {
 136         sample = &track->sample[snum];
 137         frame_codestream = (unsigned char*) malloc(sample->sample_size -
 138                            8); /* Skipping JP2C marker*/
 139         fseek(file, sample->offset + 8, SEEK_SET);
 140         fread(frame_codestream, sample->sample_size - 8, 1,
 141               file); /* Assuming that jp and ftyp markers size do*/
 142
 143         sprintf(outfilename, "%s_%05d.j2k", argv[2], snum);
 144         outfile = fopen(outfilename, "wb");
 145         if (!outfile) {
 146             fprintf(stderr, "failed to open %s for writing\n", outfilename);
 147             return 1;
 148         }
 149         fwrite(frame_codestream, sample->sample_size - 8, 1, outfile);
 150         fclose(outfile);
 151         free(frame_codestream);
 152     }
 153     fclose(file);
 154     fprintf(stdout, "%d frames correctly extracted\n", snum);
 155
 156     /* free remaining structures */
 157     if (dinfo) {
 158         mj2_destroy_decompress((opj_mj2_t*)dinfo->mj2_handle);
 159     }
 160
 161     return 0;
 162 }