Add capability to decode only a subset of all components of an image.
[openjpeg.git] / src / lib / openjp2 / openjpeg.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) 2005, Herve Drolon, FreeImage Team
8  * Copyright (c) 2008, 2011-2012, Centre National d'Etudes Spatiales (CNES), FR
9  * Copyright (c) 2012, CS Systemes d'Information, France
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 #ifdef _WIN32
35 #include <windows.h>
36 #endif /* _WIN32 */
37
38 #include "opj_includes.h"
39
40
41 /* ---------------------------------------------------------------------- */
42 /* Functions to set the message handlers */
43
44 OPJ_BOOL OPJ_CALLCONV opj_set_info_handler(opj_codec_t * p_codec,
45         opj_msg_callback p_callback,
46         void * p_user_data)
47 {
48     opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
49     if (! l_codec) {
50         return OPJ_FALSE;
51     }
52
53     l_codec->m_event_mgr.info_handler = p_callback;
54     l_codec->m_event_mgr.m_info_data = p_user_data;
55
56     return OPJ_TRUE;
57 }
58
59 OPJ_BOOL OPJ_CALLCONV opj_set_warning_handler(opj_codec_t * p_codec,
60         opj_msg_callback p_callback,
61         void * p_user_data)
62 {
63     opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
64     if (! l_codec) {
65         return OPJ_FALSE;
66     }
67
68     l_codec->m_event_mgr.warning_handler = p_callback;
69     l_codec->m_event_mgr.m_warning_data = p_user_data;
70
71     return OPJ_TRUE;
72 }
73
74 OPJ_BOOL OPJ_CALLCONV opj_set_error_handler(opj_codec_t * p_codec,
75         opj_msg_callback p_callback,
76         void * p_user_data)
77 {
78     opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
79     if (! l_codec) {
80         return OPJ_FALSE;
81     }
82
83     l_codec->m_event_mgr.error_handler = p_callback;
84     l_codec->m_event_mgr.m_error_data = p_user_data;
85
86     return OPJ_TRUE;
87 }
88
89 /* ---------------------------------------------------------------------- */
90
91 static OPJ_SIZE_T opj_read_from_file(void * p_buffer, OPJ_SIZE_T p_nb_bytes,
92                                      FILE * p_file)
93 {
94     OPJ_SIZE_T l_nb_read = fread(p_buffer, 1, p_nb_bytes, p_file);
95     return l_nb_read ? l_nb_read : (OPJ_SIZE_T) - 1;
96 }
97
98 static OPJ_UINT64 opj_get_data_length_from_file(FILE * p_file)
99 {
100     OPJ_OFF_T file_length = 0;
101
102     OPJ_FSEEK(p_file, 0, SEEK_END);
103     file_length = (OPJ_OFF_T)OPJ_FTELL(p_file);
104     OPJ_FSEEK(p_file, 0, SEEK_SET);
105
106     return (OPJ_UINT64)file_length;
107 }
108
109 static OPJ_SIZE_T opj_write_from_file(void * p_buffer, OPJ_SIZE_T p_nb_bytes,
110                                       FILE * p_file)
111 {
112     return fwrite(p_buffer, 1, p_nb_bytes, p_file);
113 }
114
115 static OPJ_OFF_T opj_skip_from_file(OPJ_OFF_T p_nb_bytes, FILE * p_user_data)
116 {
117     if (OPJ_FSEEK(p_user_data, p_nb_bytes, SEEK_CUR)) {
118         return -1;
119     }
120
121     return p_nb_bytes;
122 }
123
124 static OPJ_BOOL opj_seek_from_file(OPJ_OFF_T p_nb_bytes, FILE * p_user_data)
125 {
126     if (OPJ_FSEEK(p_user_data, p_nb_bytes, SEEK_SET)) {
127         return OPJ_FALSE;
128     }
129
130     return OPJ_TRUE;
131 }
132
133 /* ---------------------------------------------------------------------- */
134 #ifdef _WIN32
135 #ifndef OPJ_STATIC
136 BOOL APIENTRY
137 DllMain(HINSTANCE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
138 {
139
140     OPJ_ARG_NOT_USED(lpReserved);
141     OPJ_ARG_NOT_USED(hModule);
142
143     switch (ul_reason_for_call) {
144     case DLL_PROCESS_ATTACH :
145         break;
146     case DLL_PROCESS_DETACH :
147         break;
148     case DLL_THREAD_ATTACH :
149     case DLL_THREAD_DETACH :
150         break;
151     }
152
153     return TRUE;
154 }
155 #endif /* OPJ_STATIC */
156 #endif /* _WIN32 */
157
158 /* ---------------------------------------------------------------------- */
159
160 const char* OPJ_CALLCONV opj_version(void)
161 {
162     return OPJ_PACKAGE_VERSION;
163 }
164
165 /* ---------------------------------------------------------------------- */
166 /* DECOMPRESSION FUNCTIONS*/
167
168 opj_codec_t* OPJ_CALLCONV opj_create_decompress(OPJ_CODEC_FORMAT p_format)
169 {
170     opj_codec_private_t *l_codec = 00;
171
172     l_codec = (opj_codec_private_t*) opj_calloc(1, sizeof(opj_codec_private_t));
173     if (!l_codec) {
174         return 00;
175     }
176
177     l_codec->is_decompressor = 1;
178
179     switch (p_format) {
180     case OPJ_CODEC_J2K:
181         l_codec->opj_dump_codec = (void (*)(void*, OPJ_INT32, FILE*)) j2k_dump;
182
183         l_codec->opj_get_codec_info = (opj_codestream_info_v2_t* (*)(
184                                            void*)) j2k_get_cstr_info;
185
186         l_codec->opj_get_codec_index = (opj_codestream_index_t* (*)(
187                                             void*)) j2k_get_cstr_index;
188
189         l_codec->m_codec_data.m_decompression.opj_decode =
190             (OPJ_BOOL(*)(void *,
191                          struct opj_stream_private *,
192                          opj_image_t*, struct opj_event_mgr *)) opj_j2k_decode;
193
194         l_codec->m_codec_data.m_decompression.opj_end_decompress =
195             (OPJ_BOOL(*)(void *,
196                          struct opj_stream_private *,
197                          struct opj_event_mgr *)) opj_j2k_end_decompress;
198
199         l_codec->m_codec_data.m_decompression.opj_read_header =
200             (OPJ_BOOL(*)(struct opj_stream_private *,
201                          void *,
202                          opj_image_t **,
203                          struct opj_event_mgr *)) opj_j2k_read_header;
204
205         l_codec->m_codec_data.m_decompression.opj_destroy =
206             (void (*)(void *))opj_j2k_destroy;
207
208         l_codec->m_codec_data.m_decompression.opj_setup_decoder =
209             (void (*)(void *, opj_dparameters_t *)) opj_j2k_setup_decoder;
210
211         l_codec->m_codec_data.m_decompression.opj_read_tile_header =
212             (OPJ_BOOL(*)(void *,
213                          OPJ_UINT32*,
214                          OPJ_UINT32*,
215                          OPJ_INT32*, OPJ_INT32*,
216                          OPJ_INT32*, OPJ_INT32*,
217                          OPJ_UINT32*,
218                          OPJ_BOOL*,
219                          struct opj_stream_private *,
220                          struct opj_event_mgr *)) opj_j2k_read_tile_header;
221
222         l_codec->m_codec_data.m_decompression.opj_decode_tile_data =
223             (OPJ_BOOL(*)(void *,
224                          OPJ_UINT32,
225                          OPJ_BYTE*,
226                          OPJ_UINT32,
227                          struct opj_stream_private *,
228                          struct opj_event_mgr *)) opj_j2k_decode_tile;
229
230         l_codec->m_codec_data.m_decompression.opj_set_decode_area =
231             (OPJ_BOOL(*)(void *,
232                          opj_image_t*,
233                          OPJ_INT32, OPJ_INT32, OPJ_INT32, OPJ_INT32,
234                          struct opj_event_mgr *)) opj_j2k_set_decode_area;
235
236         l_codec->m_codec_data.m_decompression.opj_get_decoded_tile =
237             (OPJ_BOOL(*)(void *p_codec,
238                          opj_stream_private_t *p_cio,
239                          opj_image_t *p_image,
240                          struct opj_event_mgr * p_manager,
241                          OPJ_UINT32 tile_index)) opj_j2k_get_tile;
242
243         l_codec->m_codec_data.m_decompression.opj_set_decoded_resolution_factor =
244             (OPJ_BOOL(*)(void * p_codec,
245                          OPJ_UINT32 res_factor,
246                          struct opj_event_mgr * p_manager)) opj_j2k_set_decoded_resolution_factor;
247
248         l_codec->m_codec_data.m_decompression.opj_set_decoded_components =
249             (OPJ_BOOL(*)(void * p_codec,
250                          OPJ_UINT32 numcomps,
251                          const OPJ_UINT32 * comps_indices,
252                          struct opj_event_mgr * p_manager)) opj_j2k_set_decoded_components;
253
254         l_codec->opj_set_threads =
255             (OPJ_BOOL(*)(void * p_codec, OPJ_UINT32 num_threads)) opj_j2k_set_threads;
256
257         l_codec->m_codec = opj_j2k_create_decompress();
258
259         if (! l_codec->m_codec) {
260             opj_free(l_codec);
261             return NULL;
262         }
263
264         break;
265
266     case OPJ_CODEC_JP2:
267         /* get a JP2 decoder handle */
268         l_codec->opj_dump_codec = (void (*)(void*, OPJ_INT32, FILE*)) jp2_dump;
269
270         l_codec->opj_get_codec_info = (opj_codestream_info_v2_t* (*)(
271                                            void*)) jp2_get_cstr_info;
272
273         l_codec->opj_get_codec_index = (opj_codestream_index_t* (*)(
274                                             void*)) jp2_get_cstr_index;
275
276         l_codec->m_codec_data.m_decompression.opj_decode =
277             (OPJ_BOOL(*)(void *,
278                          struct opj_stream_private *,
279                          opj_image_t*,
280                          struct opj_event_mgr *)) opj_jp2_decode;
281
282         l_codec->m_codec_data.m_decompression.opj_end_decompress =
283             (OPJ_BOOL(*)(void *,
284                          struct opj_stream_private *,
285                          struct opj_event_mgr *)) opj_jp2_end_decompress;
286
287         l_codec->m_codec_data.m_decompression.opj_read_header =
288             (OPJ_BOOL(*)(struct opj_stream_private *,
289                          void *,
290                          opj_image_t **,
291                          struct opj_event_mgr *)) opj_jp2_read_header;
292
293         l_codec->m_codec_data.m_decompression.opj_read_tile_header =
294             (OPJ_BOOL(*)(void *,
295                          OPJ_UINT32*,
296                          OPJ_UINT32*,
297                          OPJ_INT32*,
298                          OPJ_INT32*,
299                          OPJ_INT32 *,
300                          OPJ_INT32 *,
301                          OPJ_UINT32 *,
302                          OPJ_BOOL *,
303                          struct opj_stream_private *,
304                          struct opj_event_mgr *)) opj_jp2_read_tile_header;
305
306         l_codec->m_codec_data.m_decompression.opj_decode_tile_data =
307             (OPJ_BOOL(*)(void *,
308                          OPJ_UINT32, OPJ_BYTE*, OPJ_UINT32,
309                          struct opj_stream_private *,
310                          struct opj_event_mgr *)) opj_jp2_decode_tile;
311
312         l_codec->m_codec_data.m_decompression.opj_destroy = (void (*)(
313                     void *))opj_jp2_destroy;
314
315         l_codec->m_codec_data.m_decompression.opj_setup_decoder =
316             (void (*)(void *, opj_dparameters_t *)) opj_jp2_setup_decoder;
317
318         l_codec->m_codec_data.m_decompression.opj_set_decode_area =
319             (OPJ_BOOL(*)(void *,
320                          opj_image_t*,
321                          OPJ_INT32, OPJ_INT32, OPJ_INT32, OPJ_INT32,
322                          struct opj_event_mgr *)) opj_jp2_set_decode_area;
323
324         l_codec->m_codec_data.m_decompression.opj_get_decoded_tile =
325             (OPJ_BOOL(*)(void *p_codec,
326                          opj_stream_private_t *p_cio,
327                          opj_image_t *p_image,
328                          struct opj_event_mgr * p_manager,
329                          OPJ_UINT32 tile_index)) opj_jp2_get_tile;
330
331         l_codec->m_codec_data.m_decompression.opj_set_decoded_resolution_factor =
332             (OPJ_BOOL(*)(void * p_codec,
333                          OPJ_UINT32 res_factor,
334                          opj_event_mgr_t * p_manager)) opj_jp2_set_decoded_resolution_factor;
335
336         l_codec->m_codec_data.m_decompression.opj_set_decoded_components =
337             (OPJ_BOOL(*)(void * p_codec,
338                          OPJ_UINT32 numcomps,
339                          const OPJ_UINT32 * comps_indices,
340                          struct opj_event_mgr * p_manager)) opj_jp2_set_decoded_components;
341
342         l_codec->opj_set_threads =
343             (OPJ_BOOL(*)(void * p_codec, OPJ_UINT32 num_threads)) opj_jp2_set_threads;
344
345         l_codec->m_codec = opj_jp2_create(OPJ_TRUE);
346
347         if (! l_codec->m_codec) {
348             opj_free(l_codec);
349             return 00;
350         }
351
352         break;
353     case OPJ_CODEC_UNKNOWN:
354     case OPJ_CODEC_JPT:
355     default:
356         opj_free(l_codec);
357         return 00;
358     }
359
360     opj_set_default_event_handler(&(l_codec->m_event_mgr));
361     return (opj_codec_t*) l_codec;
362 }
363
364 void OPJ_CALLCONV opj_set_default_decoder_parameters(opj_dparameters_t
365         *parameters)
366 {
367     if (parameters) {
368         memset(parameters, 0, sizeof(opj_dparameters_t));
369         /* default decoding parameters */
370         parameters->cp_layer = 0;
371         parameters->cp_reduce = 0;
372
373         parameters->decod_format = -1;
374         parameters->cod_format = -1;
375         parameters->flags = 0;
376         /* UniPG>> */
377 #ifdef USE_JPWL
378         parameters->jpwl_correct = OPJ_FALSE;
379         parameters->jpwl_exp_comps = JPWL_EXPECTED_COMPONENTS;
380         parameters->jpwl_max_tiles = JPWL_MAXIMUM_TILES;
381 #endif /* USE_JPWL */
382         /* <<UniPG */
383     }
384 }
385
386
387 OPJ_BOOL OPJ_CALLCONV opj_codec_set_threads(opj_codec_t *p_codec,
388         int num_threads)
389 {
390     if (p_codec && (num_threads >= 0)) {
391         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
392
393         return l_codec->opj_set_threads(l_codec->m_codec, (OPJ_UINT32)num_threads);
394     }
395     return OPJ_FALSE;
396 }
397
398 OPJ_BOOL OPJ_CALLCONV opj_setup_decoder(opj_codec_t *p_codec,
399                                         opj_dparameters_t *parameters
400                                        )
401 {
402     if (p_codec && parameters) {
403         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
404
405         if (! l_codec->is_decompressor) {
406             opj_event_msg(&(l_codec->m_event_mgr), EVT_ERROR,
407                           "Codec provided to the opj_setup_decoder function is not a decompressor handler.\n");
408             return OPJ_FALSE;
409         }
410
411         l_codec->m_codec_data.m_decompression.opj_setup_decoder(l_codec->m_codec,
412                 parameters);
413         return OPJ_TRUE;
414     }
415     return OPJ_FALSE;
416 }
417
418 OPJ_BOOL OPJ_CALLCONV opj_read_header(opj_stream_t *p_stream,
419                                       opj_codec_t *p_codec,
420                                       opj_image_t **p_image)
421 {
422     if (p_codec && p_stream) {
423         opj_codec_private_t* l_codec = (opj_codec_private_t*) p_codec;
424         opj_stream_private_t* l_stream = (opj_stream_private_t*) p_stream;
425
426         if (! l_codec->is_decompressor) {
427             opj_event_msg(&(l_codec->m_event_mgr), EVT_ERROR,
428                           "Codec provided to the opj_read_header function is not a decompressor handler.\n");
429             return OPJ_FALSE;
430         }
431
432         return l_codec->m_codec_data.m_decompression.opj_read_header(l_stream,
433                 l_codec->m_codec,
434                 p_image,
435                 &(l_codec->m_event_mgr));
436     }
437
438     return OPJ_FALSE;
439 }
440
441
442 OPJ_BOOL OPJ_CALLCONV opj_set_decoded_components(opj_codec_t *p_codec,
443         OPJ_UINT32 numcomps,
444         const OPJ_UINT32* comps_indices)
445 {
446     if (p_codec) {
447         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
448
449         if (! l_codec->is_decompressor) {
450             return OPJ_FALSE;
451         }
452
453         return  l_codec->m_codec_data.m_decompression.opj_set_decoded_components(
454                     l_codec->m_codec,
455                     numcomps,
456                     comps_indices,
457                     &(l_codec->m_event_mgr));
458     }
459     return OPJ_FALSE;
460 }
461
462 OPJ_BOOL OPJ_CALLCONV opj_decode(opj_codec_t *p_codec,
463                                  opj_stream_t *p_stream,
464                                  opj_image_t* p_image)
465 {
466     if (p_codec && p_stream) {
467         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
468         opj_stream_private_t * l_stream = (opj_stream_private_t *) p_stream;
469
470         if (! l_codec->is_decompressor) {
471             return OPJ_FALSE;
472         }
473
474         return l_codec->m_codec_data.m_decompression.opj_decode(l_codec->m_codec,
475                 l_stream,
476                 p_image,
477                 &(l_codec->m_event_mgr));
478     }
479
480     return OPJ_FALSE;
481 }
482
483 OPJ_BOOL OPJ_CALLCONV opj_set_decode_area(opj_codec_t *p_codec,
484         opj_image_t* p_image,
485         OPJ_INT32 p_start_x, OPJ_INT32 p_start_y,
486         OPJ_INT32 p_end_x, OPJ_INT32 p_end_y
487                                          )
488 {
489     if (p_codec) {
490         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
491
492         if (! l_codec->is_decompressor) {
493             return OPJ_FALSE;
494         }
495
496         return  l_codec->m_codec_data.m_decompression.opj_set_decode_area(
497                     l_codec->m_codec,
498                     p_image,
499                     p_start_x, p_start_y,
500                     p_end_x, p_end_y,
501                     &(l_codec->m_event_mgr));
502     }
503     return OPJ_FALSE;
504 }
505
506 OPJ_BOOL OPJ_CALLCONV opj_read_tile_header(opj_codec_t *p_codec,
507         opj_stream_t * p_stream,
508         OPJ_UINT32 * p_tile_index,
509         OPJ_UINT32 * p_data_size,
510         OPJ_INT32 * p_tile_x0, OPJ_INT32 * p_tile_y0,
511         OPJ_INT32 * p_tile_x1, OPJ_INT32 * p_tile_y1,
512         OPJ_UINT32 * p_nb_comps,
513         OPJ_BOOL * p_should_go_on)
514 {
515     if (p_codec && p_stream && p_data_size && p_tile_index) {
516         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
517         opj_stream_private_t * l_stream = (opj_stream_private_t *) p_stream;
518
519         if (! l_codec->is_decompressor) {
520             return OPJ_FALSE;
521         }
522
523         return l_codec->m_codec_data.m_decompression.opj_read_tile_header(
524                    l_codec->m_codec,
525                    p_tile_index,
526                    p_data_size,
527                    p_tile_x0, p_tile_y0,
528                    p_tile_x1, p_tile_y1,
529                    p_nb_comps,
530                    p_should_go_on,
531                    l_stream,
532                    &(l_codec->m_event_mgr));
533     }
534     return OPJ_FALSE;
535 }
536
537 OPJ_BOOL OPJ_CALLCONV opj_decode_tile_data(opj_codec_t *p_codec,
538         OPJ_UINT32 p_tile_index,
539         OPJ_BYTE * p_data,
540         OPJ_UINT32 p_data_size,
541         opj_stream_t *p_stream
542                                           )
543 {
544     if (p_codec && p_data && p_stream) {
545         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
546         opj_stream_private_t * l_stream = (opj_stream_private_t *) p_stream;
547
548         if (! l_codec->is_decompressor) {
549             return OPJ_FALSE;
550         }
551
552         return l_codec->m_codec_data.m_decompression.opj_decode_tile_data(
553                    l_codec->m_codec,
554                    p_tile_index,
555                    p_data,
556                    p_data_size,
557                    l_stream,
558                    &(l_codec->m_event_mgr));
559     }
560     return OPJ_FALSE;
561 }
562
563 OPJ_BOOL OPJ_CALLCONV opj_get_decoded_tile(opj_codec_t *p_codec,
564         opj_stream_t *p_stream,
565         opj_image_t *p_image,
566         OPJ_UINT32 tile_index)
567 {
568     if (p_codec && p_stream) {
569         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
570         opj_stream_private_t * l_stream = (opj_stream_private_t *) p_stream;
571
572         if (! l_codec->is_decompressor) {
573             return OPJ_FALSE;
574         }
575
576         return l_codec->m_codec_data.m_decompression.opj_get_decoded_tile(
577                    l_codec->m_codec,
578                    l_stream,
579                    p_image,
580                    &(l_codec->m_event_mgr),
581                    tile_index);
582     }
583
584     return OPJ_FALSE;
585 }
586
587 OPJ_BOOL OPJ_CALLCONV opj_set_decoded_resolution_factor(opj_codec_t *p_codec,
588         OPJ_UINT32 res_factor)
589 {
590     opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
591
592     if (!l_codec) {
593         return OPJ_FALSE;
594     }
595
596     return l_codec->m_codec_data.m_decompression.opj_set_decoded_resolution_factor(
597                l_codec->m_codec,
598                res_factor,
599                &(l_codec->m_event_mgr));
600 }
601
602 /* ---------------------------------------------------------------------- */
603 /* COMPRESSION FUNCTIONS*/
604
605 opj_codec_t* OPJ_CALLCONV opj_create_compress(OPJ_CODEC_FORMAT p_format)
606 {
607     opj_codec_private_t *l_codec = 00;
608
609     l_codec = (opj_codec_private_t*)opj_calloc(1, sizeof(opj_codec_private_t));
610     if (!l_codec) {
611         return 00;
612     }
613
614     l_codec->is_decompressor = 0;
615
616     switch (p_format) {
617     case OPJ_CODEC_J2K:
618         l_codec->m_codec_data.m_compression.opj_encode = (OPJ_BOOL(*)(void *,
619                 struct opj_stream_private *,
620                 struct opj_event_mgr *)) opj_j2k_encode;
621
622         l_codec->m_codec_data.m_compression.opj_end_compress = (OPJ_BOOL(*)(void *,
623                 struct opj_stream_private *,
624                 struct opj_event_mgr *)) opj_j2k_end_compress;
625
626         l_codec->m_codec_data.m_compression.opj_start_compress = (OPJ_BOOL(*)(void *,
627                 struct opj_stream_private *,
628                 struct opj_image *,
629                 struct opj_event_mgr *)) opj_j2k_start_compress;
630
631         l_codec->m_codec_data.m_compression.opj_write_tile = (OPJ_BOOL(*)(void *,
632                 OPJ_UINT32,
633                 OPJ_BYTE*,
634                 OPJ_UINT32,
635                 struct opj_stream_private *,
636                 struct opj_event_mgr *)) opj_j2k_write_tile;
637
638         l_codec->m_codec_data.m_compression.opj_destroy = (void (*)(
639                     void *)) opj_j2k_destroy;
640
641         l_codec->m_codec_data.m_compression.opj_setup_encoder = (OPJ_BOOL(*)(void *,
642                 opj_cparameters_t *,
643                 struct opj_image *,
644                 struct opj_event_mgr *)) opj_j2k_setup_encoder;
645
646         l_codec->m_codec = opj_j2k_create_compress();
647         if (! l_codec->m_codec) {
648             opj_free(l_codec);
649             return 00;
650         }
651
652         break;
653
654     case OPJ_CODEC_JP2:
655         /* get a JP2 decoder handle */
656         l_codec->m_codec_data.m_compression.opj_encode = (OPJ_BOOL(*)(void *,
657                 struct opj_stream_private *,
658                 struct opj_event_mgr *)) opj_jp2_encode;
659
660         l_codec->m_codec_data.m_compression.opj_end_compress = (OPJ_BOOL(*)(void *,
661                 struct opj_stream_private *,
662                 struct opj_event_mgr *)) opj_jp2_end_compress;
663
664         l_codec->m_codec_data.m_compression.opj_start_compress = (OPJ_BOOL(*)(void *,
665                 struct opj_stream_private *,
666                 struct opj_image *,
667                 struct opj_event_mgr *))  opj_jp2_start_compress;
668
669         l_codec->m_codec_data.m_compression.opj_write_tile = (OPJ_BOOL(*)(void *,
670                 OPJ_UINT32,
671                 OPJ_BYTE*,
672                 OPJ_UINT32,
673                 struct opj_stream_private *,
674                 struct opj_event_mgr *)) opj_jp2_write_tile;
675
676         l_codec->m_codec_data.m_compression.opj_destroy = (void (*)(
677                     void *)) opj_jp2_destroy;
678
679         l_codec->m_codec_data.m_compression.opj_setup_encoder = (OPJ_BOOL(*)(void *,
680                 opj_cparameters_t *,
681                 struct opj_image *,
682                 struct opj_event_mgr *)) opj_jp2_setup_encoder;
683
684         l_codec->m_codec = opj_jp2_create(OPJ_FALSE);
685         if (! l_codec->m_codec) {
686             opj_free(l_codec);
687             return 00;
688         }
689
690         break;
691
692     case OPJ_CODEC_UNKNOWN:
693     case OPJ_CODEC_JPT:
694     default:
695         opj_free(l_codec);
696         return 00;
697     }
698
699     opj_set_default_event_handler(&(l_codec->m_event_mgr));
700     return (opj_codec_t*) l_codec;
701 }
702
703 void OPJ_CALLCONV opj_set_default_encoder_parameters(opj_cparameters_t
704         *parameters)
705 {
706     if (parameters) {
707         memset(parameters, 0, sizeof(opj_cparameters_t));
708         /* default coding parameters */
709         parameters->cp_cinema = OPJ_OFF; /* DEPRECATED */
710         parameters->rsiz = OPJ_PROFILE_NONE;
711         parameters->max_comp_size = 0;
712         parameters->numresolution = 6;
713         parameters->cp_rsiz = OPJ_STD_RSIZ; /* DEPRECATED */
714         parameters->cblockw_init = 64;
715         parameters->cblockh_init = 64;
716         parameters->prog_order = OPJ_LRCP;
717         parameters->roi_compno = -1;        /* no ROI */
718         parameters->subsampling_dx = 1;
719         parameters->subsampling_dy = 1;
720         parameters->tp_on = 0;
721         parameters->decod_format = -1;
722         parameters->cod_format = -1;
723         parameters->tcp_rates[0] = 0;
724         parameters->tcp_numlayers = 0;
725         parameters->cp_disto_alloc = 0;
726         parameters->cp_fixed_alloc = 0;
727         parameters->cp_fixed_quality = 0;
728         parameters->jpip_on = OPJ_FALSE;
729         /* UniPG>> */
730 #ifdef USE_JPWL
731         parameters->jpwl_epc_on = OPJ_FALSE;
732         parameters->jpwl_hprot_MH = -1; /* -1 means unassigned */
733         {
734             int i;
735             for (i = 0; i < JPWL_MAX_NO_TILESPECS; i++) {
736                 parameters->jpwl_hprot_TPH_tileno[i] = -1; /* unassigned */
737                 parameters->jpwl_hprot_TPH[i] = 0; /* absent */
738             }
739         };
740         {
741             int i;
742             for (i = 0; i < JPWL_MAX_NO_PACKSPECS; i++) {
743                 parameters->jpwl_pprot_tileno[i] = -1; /* unassigned */
744                 parameters->jpwl_pprot_packno[i] = -1; /* unassigned */
745                 parameters->jpwl_pprot[i] = 0; /* absent */
746             }
747         };
748         parameters->jpwl_sens_size = 0; /* 0 means no ESD */
749         parameters->jpwl_sens_addr = 0; /* 0 means auto */
750         parameters->jpwl_sens_range = 0; /* 0 means packet */
751         parameters->jpwl_sens_MH = -1; /* -1 means unassigned */
752         {
753             int i;
754             for (i = 0; i < JPWL_MAX_NO_TILESPECS; i++) {
755                 parameters->jpwl_sens_TPH_tileno[i] = -1; /* unassigned */
756                 parameters->jpwl_sens_TPH[i] = -1; /* absent */
757             }
758         };
759 #endif /* USE_JPWL */
760         /* <<UniPG */
761     }
762 }
763
764 OPJ_BOOL OPJ_CALLCONV opj_setup_encoder(opj_codec_t *p_codec,
765                                         opj_cparameters_t *parameters,
766                                         opj_image_t *p_image)
767 {
768     if (p_codec && parameters && p_image) {
769         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
770
771         if (! l_codec->is_decompressor) {
772             return l_codec->m_codec_data.m_compression.opj_setup_encoder(l_codec->m_codec,
773                     parameters,
774                     p_image,
775                     &(l_codec->m_event_mgr));
776         }
777     }
778
779     return OPJ_FALSE;
780 }
781
782 OPJ_BOOL OPJ_CALLCONV opj_start_compress(opj_codec_t *p_codec,
783         opj_image_t * p_image,
784         opj_stream_t *p_stream)
785 {
786     if (p_codec && p_stream) {
787         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
788         opj_stream_private_t * l_stream = (opj_stream_private_t *) p_stream;
789
790         if (! l_codec->is_decompressor) {
791             return l_codec->m_codec_data.m_compression.opj_start_compress(l_codec->m_codec,
792                     l_stream,
793                     p_image,
794                     &(l_codec->m_event_mgr));
795         }
796     }
797
798     return OPJ_FALSE;
799 }
800
801 OPJ_BOOL OPJ_CALLCONV opj_encode(opj_codec_t *p_info, opj_stream_t *p_stream)
802 {
803     if (p_info && p_stream) {
804         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_info;
805         opj_stream_private_t * l_stream = (opj_stream_private_t *) p_stream;
806
807         if (! l_codec->is_decompressor) {
808             return l_codec->m_codec_data.m_compression.opj_encode(l_codec->m_codec,
809                     l_stream,
810                     &(l_codec->m_event_mgr));
811         }
812     }
813
814     return OPJ_FALSE;
815
816 }
817
818 OPJ_BOOL OPJ_CALLCONV opj_end_compress(opj_codec_t *p_codec,
819                                        opj_stream_t *p_stream)
820 {
821     if (p_codec && p_stream) {
822         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
823         opj_stream_private_t * l_stream = (opj_stream_private_t *) p_stream;
824
825         if (! l_codec->is_decompressor) {
826             return l_codec->m_codec_data.m_compression.opj_end_compress(l_codec->m_codec,
827                     l_stream,
828                     &(l_codec->m_event_mgr));
829         }
830     }
831     return OPJ_FALSE;
832
833 }
834
835 OPJ_BOOL OPJ_CALLCONV opj_end_decompress(opj_codec_t *p_codec,
836         opj_stream_t *p_stream)
837 {
838     if (p_codec && p_stream) {
839         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
840         opj_stream_private_t * l_stream = (opj_stream_private_t *) p_stream;
841
842         if (! l_codec->is_decompressor) {
843             return OPJ_FALSE;
844         }
845
846         return l_codec->m_codec_data.m_decompression.opj_end_decompress(
847                    l_codec->m_codec,
848                    l_stream,
849                    &(l_codec->m_event_mgr));
850     }
851
852     return OPJ_FALSE;
853 }
854
855 OPJ_BOOL OPJ_CALLCONV opj_set_MCT(opj_cparameters_t *parameters,
856                                   OPJ_FLOAT32 * pEncodingMatrix,
857                                   OPJ_INT32 * p_dc_shift, OPJ_UINT32 pNbComp)
858 {
859     OPJ_UINT32 l_matrix_size = pNbComp * pNbComp * (OPJ_UINT32)sizeof(OPJ_FLOAT32);
860     OPJ_UINT32 l_dc_shift_size = pNbComp * (OPJ_UINT32)sizeof(OPJ_INT32);
861     OPJ_UINT32 l_mct_total_size = l_matrix_size + l_dc_shift_size;
862
863     /* add MCT capability */
864     if (OPJ_IS_PART2(parameters->rsiz)) {
865         parameters->rsiz |= OPJ_EXTENSION_MCT;
866     } else {
867         parameters->rsiz = ((OPJ_PROFILE_PART2) | (OPJ_EXTENSION_MCT));
868     }
869     parameters->irreversible = 1;
870
871     /* use array based MCT */
872     parameters->tcp_mct = 2;
873     parameters->mct_data = opj_malloc(l_mct_total_size);
874     if (! parameters->mct_data) {
875         return OPJ_FALSE;
876     }
877
878     memcpy(parameters->mct_data, pEncodingMatrix, l_matrix_size);
879     memcpy(((OPJ_BYTE *) parameters->mct_data) +  l_matrix_size, p_dc_shift,
880            l_dc_shift_size);
881
882     return OPJ_TRUE;
883 }
884
885 OPJ_BOOL OPJ_CALLCONV opj_write_tile(opj_codec_t *p_codec,
886                                      OPJ_UINT32 p_tile_index,
887                                      OPJ_BYTE * p_data,
888                                      OPJ_UINT32 p_data_size,
889                                      opj_stream_t *p_stream)
890 {
891     if (p_codec && p_stream && p_data) {
892         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
893         opj_stream_private_t * l_stream = (opj_stream_private_t *) p_stream;
894
895         if (l_codec->is_decompressor) {
896             return OPJ_FALSE;
897         }
898
899         return l_codec->m_codec_data.m_compression.opj_write_tile(l_codec->m_codec,
900                 p_tile_index,
901                 p_data,
902                 p_data_size,
903                 l_stream,
904                 &(l_codec->m_event_mgr));
905     }
906
907     return OPJ_FALSE;
908 }
909
910 /* ---------------------------------------------------------------------- */
911
912 void OPJ_CALLCONV opj_destroy_codec(opj_codec_t *p_codec)
913 {
914     if (p_codec) {
915         opj_codec_private_t * l_codec = (opj_codec_private_t *) p_codec;
916
917         if (l_codec->is_decompressor) {
918             l_codec->m_codec_data.m_decompression.opj_destroy(l_codec->m_codec);
919         } else {
920             l_codec->m_codec_data.m_compression.opj_destroy(l_codec->m_codec);
921         }
922
923         l_codec->m_codec = 00;
924         opj_free(l_codec);
925     }
926 }
927
928 /* ---------------------------------------------------------------------- */
929
930 void OPJ_CALLCONV opj_dump_codec(opj_codec_t *p_codec,
931                                  OPJ_INT32 info_flag,
932                                  FILE* output_stream)
933 {
934     if (p_codec) {
935         opj_codec_private_t* l_codec = (opj_codec_private_t*) p_codec;
936
937         l_codec->opj_dump_codec(l_codec->m_codec, info_flag, output_stream);
938         return;
939     }
940
941     /* TODO return error */
942     /* fprintf(stderr, "[ERROR] Input parameter of the dump_codec function are incorrect.\n"); */
943     return;
944 }
945
946 opj_codestream_info_v2_t* OPJ_CALLCONV opj_get_cstr_info(opj_codec_t *p_codec)
947 {
948     if (p_codec) {
949         opj_codec_private_t* l_codec = (opj_codec_private_t*) p_codec;
950
951         return l_codec->opj_get_codec_info(l_codec->m_codec);
952     }
953
954     return NULL;
955 }
956
957 void OPJ_CALLCONV opj_destroy_cstr_info(opj_codestream_info_v2_t **cstr_info)
958 {
959     if (cstr_info) {
960
961         if ((*cstr_info)->m_default_tile_info.tccp_info) {
962             opj_free((*cstr_info)->m_default_tile_info.tccp_info);
963         }
964
965         if ((*cstr_info)->tile_info) {
966             /* FIXME not used for the moment*/
967         }
968
969         opj_free((*cstr_info));
970         (*cstr_info) = NULL;
971     }
972 }
973
974 opj_codestream_index_t * OPJ_CALLCONV opj_get_cstr_index(opj_codec_t *p_codec)
975 {
976     if (p_codec) {
977         opj_codec_private_t* l_codec = (opj_codec_private_t*) p_codec;
978
979         return l_codec->opj_get_codec_index(l_codec->m_codec);
980     }
981
982     return NULL;
983 }
984
985 void OPJ_CALLCONV opj_destroy_cstr_index(opj_codestream_index_t **p_cstr_index)
986 {
987     if (*p_cstr_index) {
988         j2k_destroy_cstr_index(*p_cstr_index);
989         (*p_cstr_index) = NULL;
990     }
991 }
992
993 opj_stream_t* OPJ_CALLCONV opj_stream_create_default_file_stream(
994     const char *fname, OPJ_BOOL p_is_read_stream)
995 {
996     return opj_stream_create_file_stream(fname, OPJ_J2K_STREAM_CHUNK_SIZE,
997                                          p_is_read_stream);
998 }
999
1000 opj_stream_t* OPJ_CALLCONV opj_stream_create_file_stream(
1001     const char *fname,
1002     OPJ_SIZE_T p_size,
1003     OPJ_BOOL p_is_read_stream)
1004 {
1005     opj_stream_t* l_stream = 00;
1006     FILE *p_file;
1007     const char *mode;
1008
1009     if (! fname) {
1010         return NULL;
1011     }
1012
1013     if (p_is_read_stream) {
1014         mode = "rb";
1015     } else {
1016         mode = "wb";
1017     }
1018
1019     p_file = fopen(fname, mode);
1020
1021     if (! p_file) {
1022         return NULL;
1023     }
1024
1025     l_stream = opj_stream_create(p_size, p_is_read_stream);
1026     if (! l_stream) {
1027         fclose(p_file);
1028         return NULL;
1029     }
1030
1031     opj_stream_set_user_data(l_stream, p_file,
1032                              (opj_stream_free_user_data_fn) fclose);
1033     opj_stream_set_user_data_length(l_stream,
1034                                     opj_get_data_length_from_file(p_file));
1035     opj_stream_set_read_function(l_stream, (opj_stream_read_fn) opj_read_from_file);
1036     opj_stream_set_write_function(l_stream,
1037                                   (opj_stream_write_fn) opj_write_from_file);
1038     opj_stream_set_skip_function(l_stream, (opj_stream_skip_fn) opj_skip_from_file);
1039     opj_stream_set_seek_function(l_stream, (opj_stream_seek_fn) opj_seek_from_file);
1040
1041     return l_stream;
1042 }
1043
1044
1045 void* OPJ_CALLCONV opj_image_data_alloc(OPJ_SIZE_T size)
1046 {
1047     void* ret = opj_aligned_malloc(size);
1048     /* printf("opj_image_data_alloc %p\n", ret); */
1049     return ret;
1050 }
1051
1052 void OPJ_CALLCONV opj_image_data_free(void* ptr)
1053 {
1054     /* printf("opj_image_data_free %p\n", ptr); */
1055     opj_aligned_free(ptr);
1056 }