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