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
33 #include "opj_includes.h"
34
35 /** @defgroup PI PI - Implementation of a packet iterator */
36 /*@{*/
37
38 /** @name Local static functions */
39 /*@{*/
40
41 /**
42 Get next packet in layer-resolution-component-precinct order.
43 @param pi packet iterator to modify
44 @return returns false if pi pointed to the last packet or else returns true
45 */
46 static bool pi_next_lrcp(opj_pi_iterator_t * pi);
47 /**
48 Get next packet in resolution-layer-component-precinct order.
49 @param pi packet iterator to modify
50 @return returns false if pi pointed to the last packet or else returns true
51 */
52 static bool pi_next_rlcp(opj_pi_iterator_t * pi);
53 /**
54 Get next packet in resolution-precinct-component-layer order.
55 @param pi packet iterator to modify
56 @return returns false if pi pointed to the last packet or else returns true
57 */
58 static bool pi_next_rpcl(opj_pi_iterator_t * pi);
59 /**
60 Get next packet in precinct-component-resolution-layer order.
61 @param pi packet iterator to modify
62 @return returns false if pi pointed to the last packet or else returns true
63 */
64 static bool pi_next_pcrl(opj_pi_iterator_t * pi);
65 /**
66 Get next packet in component-precinct-resolution-layer order.
67 @param pi packet iterator to modify
68 @return returns false if pi pointed to the last packet or else returns true
69 */
70 static bool pi_next_cprl(opj_pi_iterator_t * pi);
71
72 /*@}*/
73
74 /*@}*/
75
76 /*
77 ==========================================================
78 local functions
79 ==========================================================
80 */
81
pi_next_lrcp(opj_pi_iterator_t * pi)82 static bool pi_next_lrcp(opj_pi_iterator_t * pi) {
83 opj_pi_comp_t *comp = NULL;
84 opj_pi_resolution_t *res = NULL;
85 long index = 0;
86
87 if (!pi->first) {
88 comp = &pi->comps[pi->compno];
89 res = &comp->resolutions[pi->resno];
90 goto LABEL_SKIP;
91 } else {
92 pi->first = 0;
93 }
94
95 for (pi->layno = pi->poc.layno0; pi->layno < pi->poc.layno1; pi->layno++) {
96 for (pi->resno = pi->poc.resno0; pi->resno < pi->poc.resno1;
97 pi->resno++) {
98 for (pi->compno = pi->poc.compno0; pi->compno < pi->poc.compno1; pi->compno++) {
99 comp = &pi->comps[pi->compno];
100 if (pi->resno >= comp->numresolutions) {
101 continue;
102 }
103 res = &comp->resolutions[pi->resno];
104 if (!pi->tp_on){
105 pi->poc.precno1 = res->pw * res->ph;
106 }
107 for (pi->precno = pi->poc.precno0; pi->precno < pi->poc.precno1; pi->precno++) {
108 index = pi->layno * pi->step_l + pi->resno * pi->step_r + pi->compno * pi->step_c + pi->precno * pi->step_p;
109 if (!pi->include[index]) {
110 pi->include[index] = 1;
111 return true;
112 }
113 LABEL_SKIP:;
114 }
115 }
116 }
117 }
118
119 return false;
120 }
121
pi_next_rlcp(opj_pi_iterator_t * pi)122 static bool pi_next_rlcp(opj_pi_iterator_t * pi) {
123 opj_pi_comp_t *comp = NULL;
124 opj_pi_resolution_t *res = NULL;
125 long index = 0;
126
127 if (!pi->first) {
128 comp = &pi->comps[pi->compno];
129 res = &comp->resolutions[pi->resno];
130 goto LABEL_SKIP;
131 } else {
132 pi->first = 0;
133 }
134
135 for (pi->resno = pi->poc.resno0; pi->resno < pi->poc.resno1; pi->resno++) {
136 for (pi->layno = pi->poc.layno0; pi->layno < pi->poc.layno1; pi->layno++) {
137 for (pi->compno = pi->poc.compno0; pi->compno < pi->poc.compno1; pi->compno++) {
138 comp = &pi->comps[pi->compno];
139 if (pi->resno >= comp->numresolutions) {
140 continue;
141 }
142 res = &comp->resolutions[pi->resno];
143 if(!pi->tp_on){
144 pi->poc.precno1 = res->pw * res->ph;
145 }
146 for (pi->precno = pi->poc.precno0; pi->precno < pi->poc.precno1; pi->precno++) {
147 index = pi->layno * pi->step_l + pi->resno * pi->step_r + pi->compno * pi->step_c + pi->precno * pi->step_p;
148 if (!pi->include[index]) {
149 pi->include[index] = 1;
150 return true;
151 }
152 LABEL_SKIP:;
153 }
154 }
155 }
156 }
157
158 return false;
159 }
160
pi_next_rpcl(opj_pi_iterator_t * pi)161 static bool pi_next_rpcl(opj_pi_iterator_t * pi) {
162 opj_pi_comp_t *comp = NULL;
163 opj_pi_resolution_t *res = NULL;
164 long index = 0;
165
166 if (!pi->first) {
167 goto LABEL_SKIP;
168 } else {
169 int compno, resno;
170 pi->first = 0;
171 pi->dx = 0;
172 pi->dy = 0;
173 for (compno = 0; compno < pi->numcomps; compno++) {
174 comp = &pi->comps[compno];
175 for (resno = 0; resno < comp->numresolutions; resno++) {
176 int dx, dy;
177 res = &comp->resolutions[resno];
178 dx = comp->dx * (1 << (res->pdx + comp->numresolutions - 1 - resno));
179 dy = comp->dy * (1 << (res->pdy + comp->numresolutions - 1 - resno));
180 pi->dx = !pi->dx ? dx : int_min(pi->dx, dx);
181 pi->dy = !pi->dy ? dy : int_min(pi->dy, dy);
182 }
183 }
184 }
185 if (!pi->tp_on){
186 pi->poc.ty0 = pi->ty0;
187 pi->poc.tx0 = pi->tx0;
188 pi->poc.ty1 = pi->ty1;
189 pi->poc.tx1 = pi->tx1;
190 }
191 for (pi->resno = pi->poc.resno0; pi->resno < pi->poc.resno1; pi->resno++) {
192 for (pi->y = pi->poc.ty0; pi->y < pi->poc.ty1; pi->y += pi->dy - (pi->y % pi->dy)) {
193 for (pi->x = pi->poc.tx0; pi->x < pi->poc.tx1; pi->x += pi->dx - (pi->x % pi->dx)) {
194 for (pi->compno = pi->poc.compno0; pi->compno < pi->poc.compno1; pi->compno++) {
195 int levelno;
196 int trx0, try0;
197 int trx1, try1;
198 int rpx, rpy;
199 int prci, prcj;
200 comp = &pi->comps[pi->compno];
201 if (pi->resno >= comp->numresolutions) {
202 continue;
203 }
204 res = &comp->resolutions[pi->resno];
205 levelno = comp->numresolutions - 1 - pi->resno;
206 trx0 = int_ceildiv(pi->tx0, comp->dx << levelno);
207 try0 = int_ceildiv(pi->ty0, comp->dy << levelno);
208 trx1 = int_ceildiv(pi->tx1, comp->dx << levelno);
209 try1 = int_ceildiv(pi->ty1, comp->dy << levelno);
210 rpx = res->pdx + levelno;
211 rpy = res->pdy + levelno;
212 if (!((pi->y % (comp->dy << rpy) == 0) || ((pi->y == pi->ty0) && ((try0 << levelno) % (1 << rpy))))){
213 continue;
214 }
215 if (!((pi->x % (comp->dx << rpx) == 0) || ((pi->x == pi->tx0) && ((trx0 << levelno) % (1 << rpx))))){
216 continue;
217 }
218
219 if ((res->pw==0)||(res->ph==0)) continue;
220
221 if ((trx0==trx1)||(try0==try1)) continue;
222
223 prci = int_floordivpow2(int_ceildiv(pi->x, comp->dx << levelno), res->pdx)
224 - int_floordivpow2(trx0, res->pdx);
225 prcj = int_floordivpow2(int_ceildiv(pi->y, comp->dy << levelno), res->pdy)
226 - int_floordivpow2(try0, res->pdy);
227 pi->precno = prci + prcj * res->pw;
228 for (pi->layno = pi->poc.layno0; pi->layno < pi->poc.layno1; pi->layno++) {
229 index = pi->layno * pi->step_l + pi->resno * pi->step_r + pi->compno * pi->step_c + pi->precno * pi->step_p;
230 if (!pi->include[index]) {
231 pi->include[index] = 1;
232 return true;
233 }
234 LABEL_SKIP:;
235 }
236 }
237 }
238 }
239 }
240
241 return false;
242 }
243
pi_next_pcrl(opj_pi_iterator_t * pi)244 static bool pi_next_pcrl(opj_pi_iterator_t * pi) {
245 opj_pi_comp_t *comp = NULL;
246 opj_pi_resolution_t *res = NULL;
247 long index = 0;
248
249 if (!pi->first) {
250 comp = &pi->comps[pi->compno];
251 goto LABEL_SKIP;
252 } else {
253 int compno, resno;
254 pi->first = 0;
255 pi->dx = 0;
256 pi->dy = 0;
257 for (compno = 0; compno < pi->numcomps; compno++) {
258 comp = &pi->comps[compno];
259 for (resno = 0; resno < comp->numresolutions; resno++) {
260 int dx, dy;
261 res = &comp->resolutions[resno];
262 dx = comp->dx * (1 << (res->pdx + comp->numresolutions - 1 - resno));
263 dy = comp->dy * (1 << (res->pdy + comp->numresolutions - 1 - resno));
264 pi->dx = !pi->dx ? dx : int_min(pi->dx, dx);
265 pi->dy = !pi->dy ? dy : int_min(pi->dy, dy);
266 }
267 }
268 }
269 if (!pi->tp_on){
270 pi->poc.ty0 = pi->ty0;
271 pi->poc.tx0 = pi->tx0;
272 pi->poc.ty1 = pi->ty1;
273 pi->poc.tx1 = pi->tx1;
274 }
275 for (pi->y = pi->poc.ty0; pi->y < pi->poc.ty1; pi->y += pi->dy - (pi->y % pi->dy)) {
276 for (pi->x = pi->poc.tx0; pi->x < pi->poc.tx1; pi->x += pi->dx - (pi->x % pi->dx)) {
277 for (pi->compno = pi->poc.compno0; pi->compno < pi->poc.compno1; pi->compno++) {
278 comp = &pi->comps[pi->compno];
279 for (pi->resno = pi->poc.resno0; pi->resno < int_min(pi->poc.resno1, comp->numresolutions); pi->resno++) {
280 int levelno;
281 int trx0, try0;
282 int trx1, try1;
283 int rpx, rpy;
284 int prci, prcj;
285 res = &comp->resolutions[pi->resno];
286 levelno = comp->numresolutions - 1 - pi->resno;
287 trx0 = int_ceildiv(pi->tx0, comp->dx << levelno);
288 try0 = int_ceildiv(pi->ty0, comp->dy << levelno);
289 trx1 = int_ceildiv(pi->tx1, comp->dx << levelno);
290 try1 = int_ceildiv(pi->ty1, comp->dy << levelno);
291 rpx = res->pdx + levelno;
292 rpy = res->pdy + levelno;
293 if (!((pi->y % (comp->dy << rpy) == 0) || ((pi->y == pi->ty0) && ((try0 << levelno) % (1 << rpy))))){
294 continue;
295 }
296 if (!((pi->x % (comp->dx << rpx) == 0) || ((pi->x == pi->tx0) && ((trx0 << levelno) % (1 << rpx))))){
297 continue;
298 }
299
300 if ((res->pw==0)||(res->ph==0)) continue;
301
302 if ((trx0==trx1)||(try0==try1)) continue;
303
304 prci = int_floordivpow2(int_ceildiv(pi->x, comp->dx << levelno), res->pdx)
305 - int_floordivpow2(trx0, res->pdx);
306 prcj = int_floordivpow2(int_ceildiv(pi->y, comp->dy << levelno), res->pdy)
307 - int_floordivpow2(try0, res->pdy);
308 pi->precno = prci + prcj * res->pw;
309 for (pi->layno = pi->poc.layno0; pi->layno < pi->poc.layno1; pi->layno++) {
310 index = pi->layno * pi->step_l + pi->resno * pi->step_r + pi->compno * pi->step_c + pi->precno * pi->step_p;
311 if (!pi->include[index]) {
312 pi->include[index] = 1;
313 return true;
314 }
315 LABEL_SKIP:;
316 }
317 }
318 }
319 }
320 }
321
322 return false;
323 }
324
pi_next_cprl(opj_pi_iterator_t * pi)325 static bool pi_next_cprl(opj_pi_iterator_t * pi) {
326 opj_pi_comp_t *comp = NULL;
327 opj_pi_resolution_t *res = NULL;
328 long index = 0;
329
330 if (!pi->first) {
331 comp = &pi->comps[pi->compno];
332 goto LABEL_SKIP;
333 } else {
334 pi->first = 0;
335 }
336
337 for (pi->compno = pi->poc.compno0; pi->compno < pi->poc.compno1; pi->compno++) {
338 int resno;
339 comp = &pi->comps[pi->compno];
340 pi->dx = 0;
341 pi->dy = 0;
342 for (resno = 0; resno < comp->numresolutions; resno++) {
343 int dx, dy;
344 res = &comp->resolutions[resno];
345 dx = comp->dx * (1 << (res->pdx + comp->numresolutions - 1 - resno));
346 dy = comp->dy * (1 << (res->pdy + comp->numresolutions - 1 - resno));
347 pi->dx = !pi->dx ? dx : int_min(pi->dx, dx);
348 pi->dy = !pi->dy ? dy : int_min(pi->dy, dy);
349 }
350 if (!pi->tp_on){
351 pi->poc.ty0 = pi->ty0;
352 pi->poc.tx0 = pi->tx0;
353 pi->poc.ty1 = pi->ty1;
354 pi->poc.tx1 = pi->tx1;
355 }
356 for (pi->y = pi->poc.ty0; pi->y < pi->poc.ty1; pi->y += pi->dy - (pi->y % pi->dy)) {
357 for (pi->x = pi->poc.tx0; pi->x < pi->poc.tx1; pi->x += pi->dx - (pi->x % pi->dx)) {
358 for (pi->resno = pi->poc.resno0; pi->resno < int_min(pi->poc.resno1, comp->numresolutions); pi->resno++) {
359 int levelno;
360 int trx0, try0;
361 int trx1, try1;
362 int rpx, rpy;
363 int prci, prcj;
364 res = &comp->resolutions[pi->resno];
365 levelno = comp->numresolutions - 1 - pi->resno;
366 trx0 = int_ceildiv(pi->tx0, comp->dx << levelno);
367 try0 = int_ceildiv(pi->ty0, comp->dy << levelno);
368 trx1 = int_ceildiv(pi->tx1, comp->dx << levelno);
369 try1 = int_ceildiv(pi->ty1, comp->dy << levelno);
370 rpx = res->pdx + levelno;
371 rpy = res->pdy + levelno;
372 if (!((pi->y % (comp->dy << rpy) == 0) || ((pi->y == pi->ty0) && ((try0 << levelno) % (1 << rpy))))){
373 continue;
374 }
375 if (!((pi->x % (comp->dx << rpx) == 0) || ((pi->x == pi->tx0) && ((trx0 << levelno) % (1 << rpx))))){
376 continue;
377 }
378
379 if ((res->pw==0)||(res->ph==0)) continue;
380
381 if ((trx0==trx1)||(try0==try1)) continue;
382
383 prci = int_floordivpow2(int_ceildiv(pi->x, comp->dx << levelno), res->pdx)
384 - int_floordivpow2(trx0, res->pdx);
385 prcj = int_floordivpow2(int_ceildiv(pi->y, comp->dy << levelno), res->pdy)
386 - int_floordivpow2(try0, res->pdy);
387 pi->precno = prci + prcj * res->pw;
388 for (pi->layno = pi->poc.layno0; pi->layno < pi->poc.layno1; pi->layno++) {
389 index = pi->layno * pi->step_l + pi->resno * pi->step_r + pi->compno * pi->step_c + pi->precno * pi->step_p;
390 if (!pi->include[index]) {
391 pi->include[index] = 1;
392 return true;
393 }
394 LABEL_SKIP:;
395 }
396 }
397 }
398 }
399 }
400
401 return false;
402 }
403
404 /*
405 ==========================================================
406 Packet iterator interface
407 ==========================================================
408 */
409
pi_create_decode(opj_image_t * image,opj_cp_t * cp,int tileno)410 opj_pi_iterator_t *pi_create_decode(opj_image_t *image, opj_cp_t *cp, int tileno) {
411 int p, q;
412 int compno, resno, pino;
413 opj_pi_iterator_t *pi = NULL;
414 opj_tcp_t *tcp = NULL;
415 opj_tccp_t *tccp = NULL;
416
417 tcp = &cp->tcps[tileno];
418
419 pi = (opj_pi_iterator_t*) opj_calloc((tcp->numpocs + 1), sizeof(opj_pi_iterator_t));
420 if(!pi) {
421 /* TODO: throw an error */
422 return NULL;
423 }
424
425 for (pino = 0; pino < tcp->numpocs + 1; pino++) { /* change */
426 int maxres = 0;
427 int maxprec = 0;
428 p = tileno % cp->tw;
429 q = tileno / cp->tw;
430
431 pi[pino].tx0 = int_max(cp->tx0 + p * cp->tdx, image->x0);
432 pi[pino].ty0 = int_max(cp->ty0 + q * cp->tdy, image->y0);
433 pi[pino].tx1 = int_min(cp->tx0 + (p + 1) * cp->tdx, image->x1);
434 pi[pino].ty1 = int_min(cp->ty0 + (q + 1) * cp->tdy, image->y1);
435 pi[pino].numcomps = image->numcomps;
436
437 pi[pino].comps = (opj_pi_comp_t*) opj_calloc(image->numcomps, sizeof(opj_pi_comp_t));
438 if(!pi[pino].comps) {
439 /* TODO: throw an error */
440 pi_destroy(pi, cp, tileno);
441 return NULL;
442 }
443
444 for (compno = 0; compno < pi->numcomps; compno++) {
445 int tcx0, tcy0, tcx1, tcy1;
446 opj_pi_comp_t *comp = &pi[pino].comps[compno];
447 tccp = &tcp->tccps[compno];
448 comp->dx = image->comps[compno].dx;
449 comp->dy = image->comps[compno].dy;
450 comp->numresolutions = tccp->numresolutions;
451
452 comp->resolutions = (opj_pi_resolution_t*) opj_calloc(comp->numresolutions, sizeof(opj_pi_resolution_t));
453 if(!comp->resolutions) {
454 /* TODO: throw an error */
455 pi_destroy(pi, cp, tileno);
456 return NULL;
457 }
458
459 tcx0 = int_ceildiv(pi->tx0, comp->dx);
460 tcy0 = int_ceildiv(pi->ty0, comp->dy);
461 tcx1 = int_ceildiv(pi->tx1, comp->dx);
462 tcy1 = int_ceildiv(pi->ty1, comp->dy);
463 if (comp->numresolutions > maxres) {
464 maxres = comp->numresolutions;
465 }
466
467 for (resno = 0; resno < comp->numresolutions; resno++) {
468 int levelno;
469 int rx0, ry0, rx1, ry1;
470 int px0, py0, px1, py1;
471 opj_pi_resolution_t *res = &comp->resolutions[resno];
472 if (tccp->csty & J2K_CCP_CSTY_PRT) {
473 res->pdx = tccp->prcw[resno];
474 res->pdy = tccp->prch[resno];
475 } else {
476 res->pdx = 15;
477 res->pdy = 15;
478 }
479 levelno = comp->numresolutions - 1 - resno;
480 rx0 = int_ceildivpow2(tcx0, levelno);
481 ry0 = int_ceildivpow2(tcy0, levelno);
482 rx1 = int_ceildivpow2(tcx1, levelno);
483 ry1 = int_ceildivpow2(tcy1, levelno);
484 px0 = int_floordivpow2(rx0, res->pdx) << res->pdx;
485 py0 = int_floordivpow2(ry0, res->pdy) << res->pdy;
486 px1 = int_ceildivpow2(rx1, res->pdx) << res->pdx;
487 py1 = int_ceildivpow2(ry1, res->pdy) << res->pdy;
488 res->pw = (rx0==rx1)?0:((px1 - px0) >> res->pdx);
489 res->ph = (ry0==ry1)?0:((py1 - py0) >> res->pdy);
490
491 if (res->pw*res->ph > maxprec) {
492 maxprec = res->pw*res->ph;
493 }
494
495 }
496 }
497
498 tccp = &tcp->tccps[0];
499 pi[pino].step_p = 1;
500 pi[pino].step_c = maxprec * pi[pino].step_p;
501 pi[pino].step_r = image->numcomps * pi[pino].step_c;
502 pi[pino].step_l = maxres * pi[pino].step_r;
503
504 if (pino == 0) {
505 pi[pino].include = (short int*) opj_calloc(image->numcomps * maxres * tcp->numlayers * maxprec, sizeof(short int));
506 if(!pi[pino].include) {
507 /* TODO: throw an error */
508 pi_destroy(pi, cp, tileno);
509 return NULL;
510 }
511 }
512 else {
513 pi[pino].include = pi[pino - 1].include;
514 }
515
516 if (tcp->POC == 0) {
517 pi[pino].first = 1;
518 pi[pino].poc.resno0 = 0;
519 pi[pino].poc.compno0 = 0;
520 pi[pino].poc.layno1 = tcp->numlayers;
521 pi[pino].poc.resno1 = maxres;
522 pi[pino].poc.compno1 = image->numcomps;
523 pi[pino].poc.prg = tcp->prg;
524 } else {
525 pi[pino].first = 1;
526 pi[pino].poc.resno0 = tcp->pocs[pino].resno0;
527 pi[pino].poc.compno0 = tcp->pocs[pino].compno0;
528 pi[pino].poc.layno1 = tcp->pocs[pino].layno1;
529 pi[pino].poc.resno1 = tcp->pocs[pino].resno1;
530 pi[pino].poc.compno1 = tcp->pocs[pino].compno1;
531 pi[pino].poc.prg = tcp->pocs[pino].prg;
532 }
533 pi[pino].poc.layno0 = 0;
534 pi[pino].poc.precno0 = 0;
535 pi[pino].poc.precno1 = maxprec;
536
537 }
538
539 return pi;
540 }
541
542
pi_initialise_encode(opj_image_t * image,opj_cp_t * cp,int tileno,J2K_T2_MODE t2_mode)543 opj_pi_iterator_t *pi_initialise_encode(opj_image_t *image, opj_cp_t *cp, int tileno, J2K_T2_MODE t2_mode){
544 int p, q, pino;
545 int compno, resno;
546 int maxres = 0;
547 int maxprec = 0;
548 opj_pi_iterator_t *pi = NULL;
549 opj_tcp_t *tcp = NULL;
550 opj_tccp_t *tccp = NULL;
551
552 tcp = &cp->tcps[tileno];
553
554 pi = (opj_pi_iterator_t*) opj_calloc((tcp->numpocs + 1), sizeof(opj_pi_iterator_t));
555 if(!pi) { return NULL;}
556 pi->tp_on = cp->tp_on;
557
558 for(pino = 0;pino < tcp->numpocs+1 ; pino ++){
559 p = tileno % cp->tw;
560 q = tileno / cp->tw;
561
562 pi[pino].tx0 = int_max(cp->tx0 + p * cp->tdx, image->x0);
563 pi[pino].ty0 = int_max(cp->ty0 + q * cp->tdy, image->y0);
564 pi[pino].tx1 = int_min(cp->tx0 + (p + 1) * cp->tdx, image->x1);
565 pi[pino].ty1 = int_min(cp->ty0 + (q + 1) * cp->tdy, image->y1);
566 pi[pino].numcomps = image->numcomps;
567
568 pi[pino].comps = (opj_pi_comp_t*) opj_calloc(image->numcomps, sizeof(opj_pi_comp_t));
569 if(!pi[pino].comps) {
570 pi_destroy(pi, cp, tileno);
571 return NULL;
572 }
573
574 for (compno = 0; compno < pi[pino].numcomps; compno++) {
575 int tcx0, tcy0, tcx1, tcy1;
576 opj_pi_comp_t *comp = &pi[pino].comps[compno];
577 tccp = &tcp->tccps[compno];
578 comp->dx = image->comps[compno].dx;
579 comp->dy = image->comps[compno].dy;
580 comp->numresolutions = tccp->numresolutions;
581
582 comp->resolutions = (opj_pi_resolution_t*) opj_malloc(comp->numresolutions * sizeof(opj_pi_resolution_t));
583 if(!comp->resolutions) {
584 pi_destroy(pi, cp, tileno);
585 return NULL;
586 }
587
588 tcx0 = int_ceildiv(pi[pino].tx0, comp->dx);
589 tcy0 = int_ceildiv(pi[pino].ty0, comp->dy);
590 tcx1 = int_ceildiv(pi[pino].tx1, comp->dx);
591 tcy1 = int_ceildiv(pi[pino].ty1, comp->dy);
592 if (comp->numresolutions > maxres) {
593 maxres = comp->numresolutions;
594 }
595
596 for (resno = 0; resno < comp->numresolutions; resno++) {
597 int levelno;
598 int rx0, ry0, rx1, ry1;
599 int px0, py0, px1, py1;
600 opj_pi_resolution_t *res = &comp->resolutions[resno];
601 if (tccp->csty & J2K_CCP_CSTY_PRT) {
602 res->pdx = tccp->prcw[resno];
603 res->pdy = tccp->prch[resno];
604 } else {
605 res->pdx = 15;
606 res->pdy = 15;
607 }
608 levelno = comp->numresolutions - 1 - resno;
609 rx0 = int_ceildivpow2(tcx0, levelno);
610 ry0 = int_ceildivpow2(tcy0, levelno);
611 rx1 = int_ceildivpow2(tcx1, levelno);
612 ry1 = int_ceildivpow2(tcy1, levelno);
613 px0 = int_floordivpow2(rx0, res->pdx) << res->pdx;
614 py0 = int_floordivpow2(ry0, res->pdy) << res->pdy;
615 px1 = int_ceildivpow2(rx1, res->pdx) << res->pdx;
616 py1 = int_ceildivpow2(ry1, res->pdy) << res->pdy;
617 res->pw = (rx0==rx1)?0:((px1 - px0) >> res->pdx);
618 res->ph = (ry0==ry1)?0:((py1 - py0) >> res->pdy);
619
620 if (res->pw*res->ph > maxprec) {
621 maxprec = res->pw * res->ph;
622 }
623 }
624 }
625
626 tccp = &tcp->tccps[0];
627 pi[pino].step_p = 1;
628 pi[pino].step_c = maxprec * pi[pino].step_p;
629 pi[pino].step_r = image->numcomps * pi[pino].step_c;
630 pi[pino].step_l = maxres * pi[pino].step_r;
631
632 for (compno = 0; compno < pi->numcomps; compno++) {
633 opj_pi_comp_t *comp = &pi->comps[compno];
634 for (resno = 0; resno < comp->numresolutions; resno++) {
635 int dx, dy;
636 opj_pi_resolution_t *res = &comp->resolutions[resno];
637 dx = comp->dx * (1 << (res->pdx + comp->numresolutions - 1 - resno));
638 dy = comp->dy * (1 << (res->pdy + comp->numresolutions - 1 - resno));
639 pi[pino].dx = !pi->dx ? dx : int_min(pi->dx, dx);
640 pi[pino].dy = !pi->dy ? dy : int_min(pi->dy, dy);
641 }
642 }
643
644 if (pino == 0) {
645 pi[pino].include = (short int*) opj_calloc(tcp->numlayers * pi[pino].step_l, sizeof(short int));
646 if(!pi[pino].include) {
647 pi_destroy(pi, cp, tileno);
648 return NULL;
649 }
650 }
651 else {
652 pi[pino].include = pi[pino - 1].include;
653 }
654
655 /* Generation of boundaries for each prog flag*/
656 if(tcp->POC && ( cp->cinema || ((!cp->cinema) && (t2_mode == FINAL_PASS)))){
657 tcp->pocs[pino].compS= tcp->pocs[pino].compno0;
658 tcp->pocs[pino].compE= tcp->pocs[pino].compno1;
659 tcp->pocs[pino].resS = tcp->pocs[pino].resno0;
660 tcp->pocs[pino].resE = tcp->pocs[pino].resno1;
661 tcp->pocs[pino].layE = tcp->pocs[pino].layno1;
662 tcp->pocs[pino].prg = tcp->pocs[pino].prg1;
663 if (pino > 0)
664 tcp->pocs[pino].layS = (tcp->pocs[pino].layE > tcp->pocs[pino - 1].layE) ? tcp->pocs[pino - 1].layE : 0;
665 }else {
666 tcp->pocs[pino].compS= 0;
667 tcp->pocs[pino].compE= image->numcomps;
668 tcp->pocs[pino].resS = 0;
669 tcp->pocs[pino].resE = maxres;
670 tcp->pocs[pino].layS = 0;
671 tcp->pocs[pino].layE = tcp->numlayers;
672 tcp->pocs[pino].prg = tcp->prg;
673 }
674 tcp->pocs[pino].prcS = 0;
675 tcp->pocs[pino].prcE = maxprec;;
676 tcp->pocs[pino].txS = pi[pino].tx0;
677 tcp->pocs[pino].txE = pi[pino].tx1;
678 tcp->pocs[pino].tyS = pi[pino].ty0;
679 tcp->pocs[pino].tyE = pi[pino].ty1;
680 tcp->pocs[pino].dx = pi[pino].dx;
681 tcp->pocs[pino].dy = pi[pino].dy;
682 }
683 return pi;
684 }
685
686
687
pi_destroy(opj_pi_iterator_t * pi,opj_cp_t * cp,int tileno)688 void pi_destroy(opj_pi_iterator_t *pi, opj_cp_t *cp, int tileno) {
689 int compno, pino;
690 opj_tcp_t *tcp = &cp->tcps[tileno];
691 if(pi) {
692 for (pino = 0; pino < tcp->numpocs + 1; pino++) {
693 if(pi[pino].comps) {
694 for (compno = 0; compno < pi->numcomps; compno++) {
695 opj_pi_comp_t *comp = &pi[pino].comps[compno];
696 if(comp->resolutions) {
697 opj_free(comp->resolutions);
698 }
699 }
700 opj_free(pi[pino].comps);
701 }
702 }
703 if(pi->include) {
704 opj_free(pi->include);
705 }
706 opj_free(pi);
707 }
708 }
709
pi_next(opj_pi_iterator_t * pi)710 bool pi_next(opj_pi_iterator_t * pi) {
711 switch (pi->poc.prg) {
712 case LRCP:
713 return pi_next_lrcp(pi);
714 case RLCP:
715 return pi_next_rlcp(pi);
716 case RPCL:
717 return pi_next_rpcl(pi);
718 case PCRL:
719 return pi_next_pcrl(pi);
720 case CPRL:
721 return pi_next_cprl(pi);
722 case PROG_UNKNOWN:
723 return false;
724 }
725
726 return false;
727 }
728
pi_create_encode(opj_pi_iterator_t * pi,opj_cp_t * cp,int tileno,int pino,int tpnum,int tppos,J2K_T2_MODE t2_mode,int cur_totnum_tp)729 bool pi_create_encode( opj_pi_iterator_t *pi, opj_cp_t *cp,int tileno, int pino,int tpnum, int tppos, J2K_T2_MODE t2_mode,int cur_totnum_tp){
730 char prog[4];
731 int i;
732 int incr_top=1,resetX=0;
733 opj_tcp_t *tcps =&cp->tcps[tileno];
734 opj_poc_t *tcp= &tcps->pocs[pino];
735
736 pi[pino].first = 1;
737 pi[pino].poc.prg = tcp->prg;
738
739 switch(tcp->prg){
740 case CPRL: strncpy(prog, "CPRL",4);
741 break;
742 case LRCP: strncpy(prog, "LRCP",4);
743 break;
744 case PCRL: strncpy(prog, "PCRL",4);
745 break;
746 case RLCP: strncpy(prog, "RLCP",4);
747 break;
748 case RPCL: strncpy(prog, "RPCL",4);
749 break;
750 case PROG_UNKNOWN:
751 return true;
752 }
753
754 if(!(cp->tp_on && ((!cp->cinema && (t2_mode == FINAL_PASS)) || cp->cinema))){
755 pi[pino].poc.resno0 = tcp->resS;
756 pi[pino].poc.resno1 = tcp->resE;
757 pi[pino].poc.compno0 = tcp->compS;
758 pi[pino].poc.compno1 = tcp->compE;
759 pi[pino].poc.layno0 = tcp->layS;
760 pi[pino].poc.layno1 = tcp->layE;
761 pi[pino].poc.precno0 = tcp->prcS;
762 pi[pino].poc.precno1 = tcp->prcE;
763 pi[pino].poc.tx0 = tcp->txS;
764 pi[pino].poc.ty0 = tcp->tyS;
765 pi[pino].poc.tx1 = tcp->txE;
766 pi[pino].poc.ty1 = tcp->tyE;
767 }else {
768 if( tpnum < cur_totnum_tp){
769 for(i=3;i>=0;i--){
770 switch(prog[i]){
771 case 'C':
772 if (i > tppos){
773 pi[pino].poc.compno0 = tcp->compS;
774 pi[pino].poc.compno1 = tcp->compE;
775 }else{
776 if (tpnum == 0){
777 tcp->comp_t = tcp->compS;
778 pi[pino].poc.compno0 = tcp->comp_t;
779 pi[pino].poc.compno1 = tcp->comp_t+1;
780 tcp->comp_t+=1;
781 }else{
782 if (incr_top == 1){
783 if(tcp->comp_t ==tcp->compE){
784 tcp->comp_t = tcp->compS;
785 pi[pino].poc.compno0 = tcp->comp_t;
786 pi[pino].poc.compno1 = tcp->comp_t+1;
787 tcp->comp_t+=1;
788 incr_top=1;
789 }else{
790 pi[pino].poc.compno0 = tcp->comp_t;
791 pi[pino].poc.compno1 = tcp->comp_t+1;
792 tcp->comp_t+=1;
793 incr_top=0;
794 }
795 }else{
796 pi[pino].poc.compno0 = tcp->comp_t-1;
797 pi[pino].poc.compno1 = tcp->comp_t;
798 }
799 }
800 }
801 break;
802
803 case 'R':
804 if (i > tppos){
805 pi[pino].poc.resno0 = tcp->resS;
806 pi[pino].poc.resno1 = tcp->resE;
807 }else{
808 if (tpnum == 0){
809 tcp->res_t = tcp->resS;
810 pi[pino].poc.resno0 = tcp->res_t;
811 pi[pino].poc.resno1 = tcp->res_t+1;
812 tcp->res_t+=1;
813 }else{
814 if (incr_top == 1){
815 if(tcp->res_t==tcp->resE){
816 tcp->res_t = tcp->resS;
817 pi[pino].poc.resno0 = tcp->res_t;
818 pi[pino].poc.resno1 = tcp->res_t+1;
819 tcp->res_t+=1;
820 incr_top=1;
821 }else{
822 pi[pino].poc.resno0 = tcp->res_t;
823 pi[pino].poc.resno1 = tcp->res_t+1;
824 tcp->res_t+=1;
825 incr_top=0;
826 }
827 }else{
828 pi[pino].poc.resno0 = tcp->res_t - 1;
829 pi[pino].poc.resno1 = tcp->res_t;
830 }
831 }
832 }
833 break;
834
835 case 'L':
836 if (i > tppos){
837 pi[pino].poc.layno0 = tcp->layS;
838 pi[pino].poc.layno1 = tcp->layE;
839 }else{
840 if (tpnum == 0){
841 tcp->lay_t = tcp->layS;
842 pi[pino].poc.layno0 = tcp->lay_t;
843 pi[pino].poc.layno1 = tcp->lay_t+1;
844 tcp->lay_t+=1;
845 }else{
846 if (incr_top == 1){
847 if(tcp->lay_t == tcp->layE){
848 tcp->lay_t = tcp->layS;
849 pi[pino].poc.layno0 = tcp->lay_t;
850 pi[pino].poc.layno1 = tcp->lay_t+1;
851 tcp->lay_t+=1;
852 incr_top=1;
853 }else{
854 pi[pino].poc.layno0 = tcp->lay_t;
855 pi[pino].poc.layno1 = tcp->lay_t+1;
856 tcp->lay_t+=1;
857 incr_top=0;
858 }
859 }else{
860 pi[pino].poc.layno0 = tcp->lay_t - 1;
861 pi[pino].poc.layno1 = tcp->lay_t;
862 }
863 }
864 }
865 break;
866
867 case 'P':
868 switch(tcp->prg){
869 case LRCP:
870 case RLCP:
871 if (i > tppos){
872 pi[pino].poc.precno0 = tcp->prcS;
873 pi[pino].poc.precno1 = tcp->prcE;
874 }else{
875 if (tpnum == 0){
876 tcp->prc_t = tcp->prcS;
877 pi[pino].poc.precno0 = tcp->prc_t;
878 pi[pino].poc.precno1 = tcp->prc_t+1;
879 tcp->prc_t+=1;
880 }else{
881 if (incr_top == 1){
882 if(tcp->prc_t == tcp->prcE){
883 tcp->prc_t = tcp->prcS;
884 pi[pino].poc.precno0 = tcp->prc_t;
885 pi[pino].poc.precno1 = tcp->prc_t+1;
886 tcp->prc_t+=1;
887 incr_top=1;
888 }else{
889 pi[pino].poc.precno0 = tcp->prc_t;
890 pi[pino].poc.precno1 = tcp->prc_t+1;
891 tcp->prc_t+=1;
892 incr_top=0;
893 }
894 }else{
895 pi[pino].poc.precno0 = tcp->prc_t - 1;
896 pi[pino].poc.precno1 = tcp->prc_t;
897 }
898 }
899 }
900 break;
901 default:
902 if (i > tppos){
903 pi[pino].poc.tx0 = tcp->txS;
904 pi[pino].poc.ty0 = tcp->tyS;
905 pi[pino].poc.tx1 = tcp->txE;
906 pi[pino].poc.ty1 = tcp->tyE;
907 }else{
908 if (tpnum == 0){
909 tcp->tx0_t = tcp->txS;
910 tcp->ty0_t = tcp->tyS;
911 pi[pino].poc.tx0 = tcp->tx0_t;
912 pi[pino].poc.tx1 = tcp->tx0_t + tcp->dx - (tcp->tx0_t % tcp->dx);
913 pi[pino].poc.ty0 = tcp->ty0_t;
914 pi[pino].poc.ty1 = tcp->ty0_t + tcp->dy - (tcp->ty0_t % tcp->dy);
915 tcp->tx0_t = pi[pino].poc.tx1;
916 tcp->ty0_t = pi[pino].poc.ty1;
917 }else{
918 if (incr_top == 1){
919 if(tcp->tx0_t >= tcp->txE){
920 if(tcp->ty0_t >= tcp->tyE){
921 tcp->ty0_t = tcp->tyS;
922 pi[pino].poc.ty0 = tcp->ty0_t;
923 pi[pino].poc.ty1 = tcp->ty0_t + tcp->dy - (tcp->ty0_t % tcp->dy);
924 tcp->ty0_t = pi[pino].poc.ty1;
925 incr_top=1;resetX=1;
926 }else{
927 pi[pino].poc.ty0 = tcp->ty0_t;
928 pi[pino].poc.ty1 = tcp->ty0_t + tcp->dy - (tcp->ty0_t % tcp->dy);
929 tcp->ty0_t = pi[pino].poc.ty1;
930 incr_top=0;resetX=1;
931 }
932 if(resetX==1){
933 tcp->tx0_t = tcp->txS;
934 pi[pino].poc.tx0 = tcp->tx0_t;
935 pi[pino].poc.tx1 = tcp->tx0_t + tcp->dx- (tcp->tx0_t % tcp->dx);
936 tcp->tx0_t = pi[pino].poc.tx1;
937 }
938 }else{
939 pi[pino].poc.tx0 = tcp->tx0_t;
940 pi[pino].poc.tx1 = tcp->tx0_t + tcp->dx- (tcp->tx0_t % tcp->dx);
941 tcp->tx0_t = pi[pino].poc.tx1;
942 pi[pino].poc.ty0 = tcp->ty0_t - tcp->dy - (tcp->ty0_t % tcp->dy);
943 pi[pino].poc.ty1 = tcp->ty0_t ;
944 incr_top=0;
945 }
946 }else{
947 pi[pino].poc.tx0 = tcp->tx0_t - tcp->dx - (tcp->tx0_t % tcp->dx);
948 pi[pino].poc.tx1 = tcp->tx0_t ;
949 pi[pino].poc.ty0 = tcp->ty0_t - tcp->dy - (tcp->ty0_t % tcp->dy);
950 pi[pino].poc.ty1 = tcp->ty0_t ;
951 }
952 }
953 }
954 break;
955 }
956 break;
957 }
958 }
959 }
960 }
961 return false;
962 }
963
964