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