123207f98b21c7e0d3aec7e63d6ab1db62ab2ba2
[openjpeg.git] / tests / j2k_random_tile_access.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) 2001-2003, David Janssens
5  * Copyright (c) 2002-2003, Yannick Verschueren
6  * Copyright (c) 2003-2007, Francois-Olivier Devaux and Antonin Descampe
7  * Copyright (c) 2005, Herve Drolon, FreeImage Team
8  * Copyright (c) 2006-2007, Parvatha Elangovan
9  * All rights reserved.
10  *
11  * Redistribution and use in source and binary forms, with or without
12  * modification, are permitted provided that the following conditions
13  * are met:
14  * 1. Redistributions of source code must retain the above copyright
15  *    notice, this list of conditions and the following disclaimer.
16  * 2. Redistributions in binary form must reproduce the above copyright
17  *    notice, this list of conditions and the following disclaimer in the
18  *    documentation and/or other materials provided with the distribution.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
21  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
24  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30  * POSSIBILITY OF SUCH DAMAGE.
31  */
32 #include "opj_config.h"
33
34 #include <stdio.h>
35 #include <string.h>
36 #include <stdlib.h>
37 #include <math.h>
38
39 #ifdef _WIN32
40 #include <windows.h>
41 #else
42 #include <strings.h>
43 #define _stricmp strcasecmp
44 #define _strnicmp strncasecmp
45 #endif /* _WIN32 */
46
47 #include "openjpeg.h"
48 #include "format_defs.h"
49
50 /* -------------------------------------------------------------------------- */
51 int get_file_format(const char *filename) {
52         unsigned int i;
53         static const char *extension[] = {"pgx", "pnm", "pgm", "ppm", "bmp","tif", "raw", "tga", "png", "j2k", "jp2", "jpt", "j2c", "jpc" };
54         static const int format[] = { PGX_DFMT, PXM_DFMT, PXM_DFMT, PXM_DFMT, BMP_DFMT, TIF_DFMT, RAW_DFMT, TGA_DFMT, PNG_DFMT, J2K_CFMT, JP2_CFMT, JPT_CFMT, J2K_CFMT, J2K_CFMT };
55         char * ext = strrchr(filename, '.');
56         if (ext == NULL)
57                 return -1;
58         ext++;
59         if(ext) {
60                 for(i = 0; i < sizeof(format)/sizeof(*format); i++) {
61                         if(_strnicmp(ext, extension[i], 3) == 0) {
62                                 return format[i];
63                         }
64                 }
65         }
66
67         return -1;
68 }
69
70 /* -------------------------------------------------------------------------- */
71
72 /**
73 sample error callback expecting a FILE* client object
74 */
75 void error_callback(const char *msg, void *client_data) {
76         FILE *stream = (FILE*)client_data;
77         fprintf(stream, "[ERROR] %s", msg);
78 }
79 /**
80 sample warning callback expecting a FILE* client object
81 */
82 void warning_callback(const char *msg, void *client_data) {
83         FILE *stream = (FILE*)client_data;
84         fprintf(stream, "[WARNING] %s", msg);
85 }
86 /**
87 sample debug callback expecting no client object
88 */
89 void info_callback(const char *msg, void *client_data) {
90         (void)client_data;
91         fprintf(stdout, "[INFO] %s", msg);
92 }
93
94
95 /* -------------------------------------------------------------------------- */
96 #define JP2_RFC3745_MAGIC "\x00\x00\x00\x0c\x6a\x50\x20\x20\x0d\x0a\x87\x0a"
97 #define JP2_MAGIC "\x0d\x0a\x87\x0a"
98 /* position 45: "\xff\x52" */
99 #define J2K_CODESTREAM_MAGIC "\xff\x4f\xff\x51"
100
101 static int infile_format(const char *fname)
102 {
103         FILE *reader;
104         const char *s, *magic_s;
105         int ext_format, magic_format;
106         unsigned char buf[12];
107         unsigned int l_nb_read;
108
109         reader = fopen(fname, "rb");
110
111         if (reader == NULL)
112                 return -1;
113
114         memset(buf, 0, 12);
115         l_nb_read = fread(buf, 1, 12, reader);
116         fclose(reader);
117         if (l_nb_read != 12)
118                 return -1;
119
120
121
122         ext_format = get_file_format(fname);
123
124         if (ext_format == JPT_CFMT)
125                 return JPT_CFMT;
126
127         if (memcmp(buf, JP2_RFC3745_MAGIC, 12) == 0 || memcmp(buf, JP2_MAGIC, 4) == 0) {
128                 magic_format = JP2_CFMT;
129                 magic_s = ".jp2";
130         }
131         else if (memcmp(buf, J2K_CODESTREAM_MAGIC, 4) == 0) {
132                 magic_format = J2K_CFMT;
133                 magic_s = ".j2k or .jpc or .j2c";
134         }
135         else
136                 return -1;
137
138         if (magic_format == ext_format)
139                 return ext_format;
140
141         s = fname + strlen(fname) - 4;
142
143         fputs("\n===========================================\n", stderr);
144         fprintf(stderr, "The extension of this file is incorrect.\n"
145                                         "FOUND %s. SHOULD BE %s\n", s, magic_s);
146         fputs("===========================================\n", stderr);
147
148         return magic_format;
149 }
150
151 /* -------------------------------------------------------------------------- */
152 /**
153  * J2K_RANDOM_TILE_ACCESS MAIN
154  */
155 /* -------------------------------------------------------------------------- */
156 int main(int argc, char **argv)
157 {
158         FILE *fsrc = NULL;
159
160         opj_dparameters_t parameters;                   /* decompression parameters */
161         opj_image_t* image = NULL;
162         opj_stream_t *l_stream = NULL;                          /* Stream */
163         opj_codec_t* l_codec = NULL;                            /* Handle to a decompressor */
164         opj_codestream_info_v2_t* cstr_info = NULL;
165
166         /* Index of corner tiles */
167         OPJ_UINT32 tile_ul = 0;
168         OPJ_UINT32 tile_ur = 0;
169         OPJ_UINT32 tile_lr = 0;
170         OPJ_UINT32 tile_ll = 0;
171
172         if (argc != 2) {
173                 fprintf(stderr, "Usage: %s <input_file>\n", argv[0]);
174                 return EXIT_FAILURE;
175         }
176
177         /* Set decoding parameters to default values */
178         opj_set_default_decoder_parameters(&parameters);
179
180         strncpy(parameters.infile, argv[1], OPJ_PATH_LEN - 1);
181
182         /* read the input file */
183         /* ------------------- */
184         fsrc = fopen(parameters.infile, "rb");
185         if (!fsrc) {
186                 fprintf(stderr, "ERROR -> failed to open %s for reading\n", parameters.infile);
187                 return EXIT_FAILURE;
188         }
189
190         /* decode the JPEG2000 stream */
191         /* -------------------------- */
192         parameters.decod_format = infile_format(parameters.infile);
193
194         switch(parameters.decod_format) {
195                 case J2K_CFMT:  /* JPEG-2000 codestream */
196                 {
197                         /* Get a decoder handle */
198                         l_codec = opj_create_decompress(CODEC_J2K);
199                         break;
200                 }
201                 case JP2_CFMT:  /* JPEG 2000 compressed image data */
202                 {
203                         /* Get a decoder handle */
204                         l_codec = opj_create_decompress(CODEC_JP2);
205                         break;
206                 }
207                 case JPT_CFMT:  /* JPEG 2000, JPIP */
208                 {
209                         /* Get a decoder handle */
210                         l_codec = opj_create_decompress(CODEC_JPT);
211                         break;
212                 }
213                 default:
214                         fprintf(stderr,
215                                 "Unrecognized format for input %s [accept only *.j2k, *.jp2, *.jpc or *.jpt]\n\n",
216                                 parameters.infile);
217                         return EXIT_FAILURE;
218         }
219
220         /* catch events using our callbacks and give a local context */         
221         opj_set_info_handler(l_codec, info_callback,00);
222         opj_set_warning_handler(l_codec, warning_callback,00);
223         opj_set_error_handler(l_codec, error_callback,00);
224
225         l_stream = opj_stream_create_default_file_stream(fsrc,1);
226         if (!l_stream){
227                 fclose(fsrc);
228                 fprintf(stderr, "ERROR -> failed to create the stream from the file\n");
229                 return EXIT_FAILURE;
230         }
231
232         /* Setup the decoder decoding parameters using user parameters */
233         if ( !opj_setup_decoder(l_codec, &parameters) ){
234                 fprintf(stderr, "ERROR -> j2k_dump: failed to setup the decoder\n");
235                 opj_stream_destroy(l_stream);
236                 fclose(fsrc);
237                 opj_destroy_codec(l_codec);
238                 return EXIT_FAILURE;
239         }
240
241         /* Read the main header of the codestream and if necessary the JP2 boxes*/
242         if(! opj_read_header(l_stream, l_codec, &image)){
243                 fprintf(stderr, "ERROR -> j2k_to_image: failed to read the header\n");
244                 opj_stream_destroy(l_stream);
245                 fclose(fsrc);
246                 opj_destroy_codec(l_codec);
247                 opj_image_destroy(image);
248                 return EXIT_FAILURE;
249         }
250
251         /* Extract some info from the code stream */
252         cstr_info = opj_get_cstr_info(l_codec);
253
254         fprintf(stdout, "The file contains %dx%d tiles\n", cstr_info->tw, cstr_info->th);
255
256         tile_ul = 0;
257         tile_ur = cstr_info->tw - 1;
258         tile_lr = cstr_info->tw * cstr_info->th - 1;
259         tile_ll = tile_lr - cstr_info->tw;
260
261 #define TEST_TILE( tile_index ) \
262         fprintf(stdout, "Decoding tile %d ...\n", tile_index); \
263         if(!opj_get_decoded_tile(l_codec, l_stream, image, tile_index )){ \
264                 fprintf(stderr, "ERROR -> j2k_to_image: failed to decode tile %d\n", tile_index); \
265                 opj_stream_destroy(l_stream); \
266                 opj_destroy_cstr_info_v2(&cstr_info); \
267                 opj_destroy_codec(l_codec); \
268                 opj_image_destroy(image); \
269                 fclose(fsrc); \
270                 return EXIT_FAILURE; \
271         } \
272         fprintf(stdout, "Tile %d is decoded successfully\n", tile_index);
273
274         TEST_TILE(tile_ul)
275         TEST_TILE(tile_lr)
276         TEST_TILE(tile_ul)
277         TEST_TILE(tile_ll)
278         TEST_TILE(tile_ur)
279         TEST_TILE(tile_lr)
280
281         /* Close the byte stream */
282         opj_stream_destroy(l_stream);
283
284         /* Destroy code stream info */
285         opj_destroy_cstr_info_v2(&cstr_info);
286
287         /* Free remaining structures */
288         opj_destroy_codec(l_codec);
289
290         /* Free image data structure */
291         opj_image_destroy(image);
292
293         /* Close the input file */
294         fclose(fsrc);
295
296         return EXIT_SUCCESS;
297 }
298 //end main
299
300
301
302