Decoding: do not allocate memory for the codestream of each codeblock
[openjpeg.git] / src / lib / openjp2 / tcd.h
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) 2001-2003, David Janssens
10  * Copyright (c) 2002-2003, Yannick Verschueren
11  * Copyright (c) 2003-2007, Francois-Olivier Devaux
12  * Copyright (c) 2003-2014, Antonin Descampe
13  * Copyright (c) 2005, Herve Drolon, FreeImage Team
14  * Copyright (c) 2008, 2011-2012, Centre National d'Etudes Spatiales (CNES), FR
15  * Copyright (c) 2012, CS Systemes d'Information, France
16  * All rights reserved.
17  *
18  * Redistribution and use in source and binary forms, with or without
19  * modification, are permitted provided that the following conditions
20  * are met:
21  * 1. Redistributions of source code must retain the above copyright
22  *    notice, this list of conditions and the following disclaimer.
23  * 2. Redistributions in binary form must reproduce the above copyright
24  *    notice, this list of conditions and the following disclaimer in the
25  *    documentation and/or other materials provided with the distribution.
26  *
27  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
28  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
29  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
30  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
31  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
32  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
33  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
34  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
35  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
36  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
37  * POSSIBILITY OF SUCH DAMAGE.
38  */
39 #ifndef OPJ_TCD_H
40 #define OPJ_TCD_H
41 /**
42 @file tcd.h
43 @brief Implementation of a tile coder/decoder (TCD)
44
45 The functions in TCD.C encode or decode each tile independently from
46 each other. The functions in TCD.C are used by other functions in J2K.C.
47 */
48
49 /** @defgroup TCD TCD - Implementation of a tile coder/decoder */
50 /*@{*/
51
52
53 /**
54 FIXME DOC
55 */
56 typedef struct opj_tcd_pass {
57     OPJ_UINT32 rate;
58     OPJ_FLOAT64 distortiondec;
59     OPJ_UINT32 len;
60     OPJ_BITFIELD term : 1;
61 } opj_tcd_pass_t;
62
63 /**
64 FIXME DOC
65 */
66 typedef struct opj_tcd_layer {
67     OPJ_UINT32 numpasses;       /* Number of passes in the layer */
68     OPJ_UINT32 len;             /* len of information */
69     OPJ_FLOAT64 disto;          /* add for index (Cfr. Marcela) */
70     OPJ_BYTE *data;             /* data */
71 } opj_tcd_layer_t;
72
73 /**
74 FIXME DOC
75 */
76 typedef struct opj_tcd_cblk_enc {
77     OPJ_BYTE* data;               /* Data */
78     opj_tcd_layer_t* layers;      /* layer information */
79     opj_tcd_pass_t* passes;       /* information about the passes */
80     OPJ_INT32 x0, y0, x1,
81               y1;     /* dimension of the code-blocks : left upper corner (x0, y0) right low corner (x1,y1) */
82     OPJ_UINT32 numbps;
83     OPJ_UINT32 numlenbits;
84     OPJ_UINT32 data_size;         /* Size of allocated data buffer */
85     OPJ_UINT32
86     numpasses;         /* number of pass already done for the code-blocks */
87     OPJ_UINT32 numpassesinlayers; /* number of passes in the layer */
88     OPJ_UINT32 totalpasses;       /* total number of passes */
89 } opj_tcd_cblk_enc_t;
90
91
92 /** Chunk of codestream data that is part of a T1 segment */
93 typedef struct opj_tcd_seg_data_chunk {
94     OPJ_BYTE *
95     data;                /* Point to tilepart buffer. We don't make a copy !
96                                        So the tilepart buffer must be kept alive
97                                        as long as we need to decode the codeblocks */
98     OPJ_UINT32 len;                 /* Usable length of data */
99 } opj_tcd_seg_data_chunk_t;
100
101 /** Segment of a code-block.
102  * A segment represent a number of consecutive coding passes, without termination
103  * of MQC or RAW between them. */
104 typedef struct opj_tcd_seg {
105     opj_tcd_seg_data_chunk_t* chunks; /* Array of chunks */
106     OPJ_UINT32 numchunks;           /* Number of valid chunks items */
107     OPJ_UINT32 numchunksalloc;      /* Number of chunks item allocated */
108     OPJ_UINT32
109     numpasses;           /* Number of passes decoded. Including those that we skip */
110     OPJ_UINT32
111     real_num_passes;     /* Number of passes actually to be decoded. To be used for code-block decoding */
112     OPJ_UINT32 maxpasses;           /* Maximum number of passes for this segment */
113     OPJ_UINT32
114     numnewpasses;        /* Number of new passes for current packed. Transitory value */
115     OPJ_UINT32
116     newlen;              /* Codestream length for this segment for current packed. Transitory value */
117 } opj_tcd_seg_t;
118
119 /* Code-block for decoding */
120 typedef struct opj_tcd_cblk_dec {
121     opj_tcd_seg_t* segs;            /* segments information */
122     OPJ_INT32 x0, y0, x1,
123               y1;       /* position of the code-blocks : left upper corner (x0, y0) right low corner (x1,y1) */
124     OPJ_UINT32 numbps;
125     OPJ_UINT32
126     numlenbits;          /* number of bits for len,, for the current packet. Transitory value */
127     OPJ_UINT32
128     numnewpasses;        /* number of pass added to the code-blocks, for the current packet. Transitory value */
129     OPJ_UINT32
130     numsegs;             /* number of segments, including those of packet we skip */
131     OPJ_UINT32
132     real_num_segs;       /* number of segments, to be used for code block decoding */
133     OPJ_UINT32 m_current_max_segs;  /* allocated number of segs[] items */
134 } opj_tcd_cblk_dec_t;
135
136 /**
137 FIXME DOC
138 */
139 typedef struct opj_tcd_precinct {
140     OPJ_INT32 x0, y0, x1,
141               y1;       /* dimension of the precinct : left upper corner (x0, y0) right low corner (x1,y1) */
142     OPJ_UINT32 cw, ch;              /* number of precinct in width and height */
143     union {                         /* code-blocks information */
144         opj_tcd_cblk_enc_t* enc;
145         opj_tcd_cblk_dec_t* dec;
146         void*               blocks;
147     } cblks;
148     OPJ_UINT32 block_size;          /* size taken by cblks (in bytes) */
149     opj_tgt_tree_t *incltree;       /* inclusion tree */
150     opj_tgt_tree_t *imsbtree;       /* IMSB tree */
151 } opj_tcd_precinct_t;
152
153 /**
154 FIXME DOC
155 */
156 typedef struct opj_tcd_band {
157     OPJ_INT32 x0, y0, x1,
158               y1;       /* dimension of the subband : left upper corner (x0, y0) right low corner (x1,y1) */
159     OPJ_UINT32 bandno;
160     opj_tcd_precinct_t *precincts;  /* precinct information */
161     OPJ_UINT32 precincts_data_size; /* size of data taken by precincts */
162     OPJ_INT32 numbps;
163     OPJ_FLOAT32 stepsize;
164 } opj_tcd_band_t;
165
166 /**
167 FIXME DOC
168 */
169 typedef struct opj_tcd_resolution {
170     OPJ_INT32 x0, y0, x1,
171               y1;       /* dimension of the resolution level : left upper corner (x0, y0) right low corner (x1,y1) */
172     OPJ_UINT32 pw, ph;
173     OPJ_UINT32 numbands;            /* number sub-band for the resolution level */
174     opj_tcd_band_t bands[3];        /* subband information */
175 } opj_tcd_resolution_t;
176
177 /**
178 FIXME DOC
179 */
180 typedef struct opj_tcd_tilecomp {
181     OPJ_INT32 x0, y0, x1,
182               y1;           /* dimension of component : left upper corner (x0, y0) right low corner (x1,y1) */
183     OPJ_UINT32 numresolutions;          /* number of resolutions level */
184     OPJ_UINT32
185     minimum_num_resolutions; /* number of resolutions level to decode (at max)*/
186     opj_tcd_resolution_t *resolutions;  /* resolutions information */
187     OPJ_UINT32
188     resolutions_size;        /* size of data for resolutions (in bytes) */
189     OPJ_INT32 *data;                    /* data of the component */
190     OPJ_BOOL  ownsData;                 /* if true, then need to free after usage, otherwise do not free */
191     OPJ_UINT32
192     data_size_needed;        /* we may either need to allocate this amount of data, or re-use image data and ignore this value */
193     OPJ_UINT32 data_size;               /* size of the data of the component */
194     OPJ_INT32 numpix;                   /* add fixed_quality */
195 } opj_tcd_tilecomp_t;
196
197
198 /**
199 FIXME DOC
200 */
201 typedef struct opj_tcd_tile {
202     OPJ_INT32 x0, y0, x1,
203               y1;       /* dimension of the tile : left upper corner (x0, y0) right low corner (x1,y1) */
204     OPJ_UINT32 numcomps;            /* number of components in tile */
205     opj_tcd_tilecomp_t *comps;  /* Components information */
206     OPJ_INT32 numpix;               /* add fixed_quality */
207     OPJ_FLOAT64 distotile;          /* add fixed_quality */
208     OPJ_FLOAT64 distolayer[100];    /* add fixed_quality */
209     OPJ_UINT32 packno;              /* packet number */
210 } opj_tcd_tile_t;
211
212 /**
213 FIXME DOC
214 */
215 typedef struct opj_tcd_image {
216     opj_tcd_tile_t *tiles;      /* Tiles information */
217 }
218 opj_tcd_image_t;
219
220
221 /**
222 Tile coder/decoder
223 */
224 typedef struct opj_tcd {
225     /** Position of the tilepart flag in Progression order*/
226     OPJ_INT32 tp_pos;
227     /** Tile part number*/
228     OPJ_UINT32 tp_num;
229     /** Current tile part number*/
230     OPJ_UINT32 cur_tp_num;
231     /** Total number of tileparts of the current tile*/
232     OPJ_UINT32 cur_totnum_tp;
233     /** Current Packet iterator number */
234     OPJ_UINT32 cur_pino;
235     /** info on each image tile */
236     opj_tcd_image_t *tcd_image;
237     /** image header */
238     opj_image_t *image;
239     /** coding parameters */
240     opj_cp_t *cp;
241     /** coding/decoding parameters common to all tiles */
242     opj_tcp_t *tcp;
243     /** current encoded/decoded tile */
244     OPJ_UINT32 tcd_tileno;
245     /** tell if the tcd is a decoder. */
246     OPJ_BITFIELD m_is_decoder : 1;
247     /** Thread pool */
248     opj_thread_pool_t* thread_pool;
249 } opj_tcd_t;
250
251 /** @name Exported functions */
252 /*@{*/
253 /* ----------------------------------------------------------------------- */
254
255 /**
256 Dump the content of a tcd structure
257 */
258 /*void tcd_dump(FILE *fd, opj_tcd_t *tcd, opj_tcd_image_t *img);*/ /* TODO MSD shoul use the new v2 structures */
259
260 /**
261 Create a new TCD handle
262 @param p_is_decoder FIXME DOC
263 @return Returns a new TCD handle if successful returns NULL otherwise
264 */
265 opj_tcd_t* opj_tcd_create(OPJ_BOOL p_is_decoder);
266
267 /**
268 Destroy a previously created TCD handle
269 @param tcd TCD handle to destroy
270 */
271 void opj_tcd_destroy(opj_tcd_t *tcd);
272
273 /**
274  * Initialize the tile coder and may reuse some memory.
275  * @param   p_tcd       TCD handle.
276  * @param   p_image     raw image.
277  * @param   p_cp        coding parameters.
278  * @param   p_tp        thread pool
279  *
280  * @return true if the encoding values could be set (false otherwise).
281 */
282 OPJ_BOOL opj_tcd_init(opj_tcd_t *p_tcd,
283                       opj_image_t * p_image,
284                       opj_cp_t * p_cp,
285                       opj_thread_pool_t* p_tp);
286
287 /**
288  * Allocates memory for decoding a specific tile.
289  *
290  * @param   p_tcd       the tile decoder.
291  * @param   p_tile_no   the index of the tile received in sequence. This not necessarily lead to the
292  * tile at index p_tile_no.
293  * @param p_manager the event manager.
294  *
295  * @return  true if the remaining data is sufficient.
296  */
297 OPJ_BOOL opj_tcd_init_decode_tile(opj_tcd_t *p_tcd, OPJ_UINT32 p_tile_no,
298                                   opj_event_mgr_t* p_manager);
299
300 void opj_tcd_makelayer_fixed(opj_tcd_t *tcd, OPJ_UINT32 layno,
301                              OPJ_UINT32 final);
302
303 void opj_tcd_rateallocate_fixed(opj_tcd_t *tcd);
304
305 void opj_tcd_makelayer(opj_tcd_t *tcd,
306                        OPJ_UINT32 layno,
307                        OPJ_FLOAT64 thresh,
308                        OPJ_UINT32 final);
309
310 OPJ_BOOL opj_tcd_rateallocate(opj_tcd_t *tcd,
311                               OPJ_BYTE *dest,
312                               OPJ_UINT32 * p_data_written,
313                               OPJ_UINT32 len,
314                               opj_codestream_info_t *cstr_info);
315
316 /**
317  * Gets the maximum tile size that will be taken by the tile once decoded.
318  */
319 OPJ_UINT32 opj_tcd_get_decoded_tile_size(opj_tcd_t *p_tcd);
320
321 /**
322  * Encodes a tile from the raw image into the given buffer.
323  * @param   p_tcd           Tile Coder handle
324  * @param   p_tile_no       Index of the tile to encode.
325  * @param   p_dest          Destination buffer
326  * @param   p_data_written  pointer to an int that is incremented by the number of bytes really written on p_dest
327  * @param   p_len           Maximum length of the destination buffer
328  * @param   p_cstr_info     Codestream information structure
329  * @return  true if the coding is successful.
330 */
331 OPJ_BOOL opj_tcd_encode_tile(opj_tcd_t *p_tcd,
332                              OPJ_UINT32 p_tile_no,
333                              OPJ_BYTE *p_dest,
334                              OPJ_UINT32 * p_data_written,
335                              OPJ_UINT32 p_len,
336                              struct opj_codestream_info *p_cstr_info);
337
338
339 /**
340 Decode a tile from a buffer into a raw image
341 @param tcd TCD handle
342 @param src Source buffer
343 @param len Length of source buffer
344 @param tileno Number that identifies one of the tiles to be decoded
345 @param cstr_info  FIXME DOC
346 @param manager the event manager.
347 */
348 OPJ_BOOL opj_tcd_decode_tile(opj_tcd_t *tcd,
349                              OPJ_BYTE *src,
350                              OPJ_UINT32 len,
351                              OPJ_UINT32 tileno,
352                              opj_codestream_index_t *cstr_info,
353                              opj_event_mgr_t *manager);
354
355
356 /**
357  * Copies tile data from the system onto the given memory block.
358  */
359 OPJ_BOOL opj_tcd_update_tile_data(opj_tcd_t *p_tcd,
360                                   OPJ_BYTE * p_dest,
361                                   OPJ_UINT32 p_dest_length);
362
363 /**
364  *
365  */
366 OPJ_UINT32 opj_tcd_get_encoded_tile_size(opj_tcd_t *p_tcd);
367
368 /**
369  * Initialize the tile coder and may reuse some meory.
370  *
371  * @param   p_tcd       TCD handle.
372  * @param   p_tile_no   current tile index to encode.
373  * @param p_manager the event manager.
374  *
375  * @return true if the encoding values could be set (false otherwise).
376 */
377 OPJ_BOOL opj_tcd_init_encode_tile(opj_tcd_t *p_tcd,
378                                   OPJ_UINT32 p_tile_no, opj_event_mgr_t* p_manager);
379
380 /**
381  * Copies tile data from the given memory block onto the system.
382  */
383 OPJ_BOOL opj_tcd_copy_tile_data(opj_tcd_t *p_tcd,
384                                 OPJ_BYTE * p_src,
385                                 OPJ_UINT32 p_src_length);
386
387 /**
388  * Allocates tile component data
389  *
390  *
391  */
392 OPJ_BOOL opj_alloc_tile_component_data(opj_tcd_tilecomp_t *l_tilec);
393
394 /** Returns whether a sub-band is empty (i.e. whether it has a null area)
395  * @param band Sub-band handle.
396  * @return OPJ_TRUE whether the sub-band is empty.
397  */
398 OPJ_BOOL opj_tcd_is_band_empty(opj_tcd_band_t* band);
399
400 /** Reinitialize a segment, without deallocating its chunks array */
401 void opj_tcd_reinit_segment(opj_tcd_seg_t* seg);
402
403 /* ----------------------------------------------------------------------- */
404 /*@}*/
405
406 /*@}*/
407
408 #endif /* OPJ_TCD_H */