• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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) 2007, Callum Lerwick <seg@haxxed.com>
15  * Copyright (c) 2012, Carl Hetherington
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 
41 #define OPJ_SKIP_POISON
42 #include "opj_includes.h"
43 
44 #ifdef __SSE__
45 #include <xmmintrin.h>
46 #endif
47 #ifdef __SSE2__
48 #include <emmintrin.h>
49 #endif
50 
51 #if defined(__GNUC__)
52 #pragma GCC poison malloc calloc realloc free
53 #endif
54 
55 #include "t1_luts.h"
56 
57 /** @defgroup T1 T1 - Implementation of the tier-1 coding */
58 /*@{*/
59 
60 #define T1_FLAGS(x, y) (t1->flags[x + 1 + ((y / 4) + 1) * (t1->w+2)])
61 
62 #define opj_t1_setcurctx(curctx, ctxno)  curctx = &(mqc)->ctxs[(OPJ_UINT32)(ctxno)]
63 
64 /** @name Local static functions */
65 /*@{*/
66 
67 static INLINE OPJ_BYTE opj_t1_getctxno_zc(opj_mqc_t *mqc, OPJ_UINT32 f);
68 static INLINE OPJ_UINT32 opj_t1_getctxno_mag(OPJ_UINT32 f);
69 static OPJ_INT16 opj_t1_getnmsedec_sig(OPJ_UINT32 x, OPJ_UINT32 bitpos);
70 static OPJ_INT16 opj_t1_getnmsedec_ref(OPJ_UINT32 x, OPJ_UINT32 bitpos);
71 static INLINE void opj_t1_update_flags(opj_flag_t *flagsp, OPJ_UINT32 ci,
72                                        OPJ_UINT32 s, OPJ_UINT32 stride,
73                                        OPJ_UINT32 vsc);
74 
75 
76 /**
77 Decode significant pass
78 */
79 
80 static INLINE void opj_t1_dec_sigpass_step_raw(
81     opj_t1_t *t1,
82     opj_flag_t *flagsp,
83     OPJ_INT32 *datap,
84     OPJ_INT32 oneplushalf,
85     OPJ_UINT32 vsc,
86     OPJ_UINT32 row);
87 static INLINE void opj_t1_dec_sigpass_step_mqc(
88     opj_t1_t *t1,
89     opj_flag_t *flagsp,
90     OPJ_INT32 *datap,
91     OPJ_INT32 oneplushalf,
92     OPJ_UINT32 row,
93     OPJ_UINT32 flags_stride,
94     OPJ_UINT32 vsc);
95 
96 /**
97 Encode significant pass
98 */
99 static void opj_t1_enc_sigpass(opj_t1_t *t1,
100                                OPJ_INT32 bpno,
101                                OPJ_INT32 *nmsedec,
102                                OPJ_BYTE type,
103                                OPJ_UINT32 cblksty);
104 
105 /**
106 Decode significant pass
107 */
108 static void opj_t1_dec_sigpass_raw(
109     opj_t1_t *t1,
110     OPJ_INT32 bpno,
111     OPJ_INT32 cblksty);
112 
113 /**
114 Encode refinement pass
115 */
116 static void opj_t1_enc_refpass(opj_t1_t *t1,
117                                OPJ_INT32 bpno,
118                                OPJ_INT32 *nmsedec,
119                                OPJ_BYTE type);
120 
121 /**
122 Decode refinement pass
123 */
124 static void opj_t1_dec_refpass_raw(
125     opj_t1_t *t1,
126     OPJ_INT32 bpno);
127 
128 
129 /**
130 Decode refinement pass
131 */
132 
133 static INLINE void  opj_t1_dec_refpass_step_raw(
134     opj_t1_t *t1,
135     opj_flag_t *flagsp,
136     OPJ_INT32 *datap,
137     OPJ_INT32 poshalf,
138     OPJ_UINT32 row);
139 static INLINE void opj_t1_dec_refpass_step_mqc(
140     opj_t1_t *t1,
141     opj_flag_t *flagsp,
142     OPJ_INT32 *datap,
143     OPJ_INT32 poshalf,
144     OPJ_UINT32 row);
145 
146 
147 /**
148 Decode clean-up pass
149 */
150 
151 static void opj_t1_dec_clnpass_step(
152     opj_t1_t *t1,
153     opj_flag_t *flagsp,
154     OPJ_INT32 *datap,
155     OPJ_INT32 oneplushalf,
156     OPJ_UINT32 row,
157     OPJ_UINT32 vsc);
158 
159 /**
160 Encode clean-up pass
161 */
162 static void opj_t1_enc_clnpass(
163     opj_t1_t *t1,
164     OPJ_INT32 bpno,
165     OPJ_INT32 *nmsedec,
166     OPJ_UINT32 cblksty);
167 
168 static OPJ_FLOAT64 opj_t1_getwmsedec(
169     OPJ_INT32 nmsedec,
170     OPJ_UINT32 compno,
171     OPJ_UINT32 level,
172     OPJ_UINT32 orient,
173     OPJ_INT32 bpno,
174     OPJ_UINT32 qmfbid,
175     OPJ_FLOAT64 stepsize,
176     OPJ_UINT32 numcomps,
177     const OPJ_FLOAT64 * mct_norms,
178     OPJ_UINT32 mct_numcomps);
179 
180 static void opj_t1_encode_cblk(opj_t1_t *t1,
181                                opj_tcd_cblk_enc_t* cblk,
182                                OPJ_UINT32 orient,
183                                OPJ_UINT32 compno,
184                                OPJ_UINT32 level,
185                                OPJ_UINT32 qmfbid,
186                                OPJ_FLOAT64 stepsize,
187                                OPJ_UINT32 cblksty,
188                                OPJ_UINT32 numcomps,
189                                opj_tcd_tile_t * tile,
190                                const OPJ_FLOAT64 * mct_norms,
191                                OPJ_UINT32 mct_numcomps);
192 
193 /**
194 Decode 1 code-block
195 @param t1 T1 handle
196 @param cblk Code-block coding parameters
197 @param orient
198 @param roishift Region of interest shifting value
199 @param cblksty Code-block style
200 @param p_manager the event manager
201 @param p_manager_mutex mutex for the event manager
202 @param check_pterm whether PTERM correct termination should be checked
203 */
204 static OPJ_BOOL opj_t1_decode_cblk(opj_t1_t *t1,
205                                    opj_tcd_cblk_dec_t* cblk,
206                                    OPJ_UINT32 orient,
207                                    OPJ_UINT32 roishift,
208                                    OPJ_UINT32 cblksty,
209                                    opj_event_mgr_t *p_manager,
210                                    opj_mutex_t* p_manager_mutex,
211                                    OPJ_BOOL check_pterm);
212 
213 static OPJ_BOOL opj_t1_allocate_buffers(opj_t1_t *t1,
214                                         OPJ_UINT32 w,
215                                         OPJ_UINT32 h);
216 
217 /*@}*/
218 
219 /*@}*/
220 
221 /* ----------------------------------------------------------------------- */
222 
opj_t1_getctxno_zc(opj_mqc_t * mqc,OPJ_UINT32 f)223 static INLINE OPJ_BYTE opj_t1_getctxno_zc(opj_mqc_t *mqc, OPJ_UINT32 f)
224 {
225     return mqc->lut_ctxno_zc_orient[(f & T1_SIGMA_NEIGHBOURS)];
226 }
227 
opj_t1_getctxtno_sc_or_spb_index(OPJ_UINT32 fX,OPJ_UINT32 pfX,OPJ_UINT32 nfX,OPJ_UINT32 ci)228 static INLINE OPJ_UINT32 opj_t1_getctxtno_sc_or_spb_index(OPJ_UINT32 fX,
229         OPJ_UINT32 pfX,
230         OPJ_UINT32 nfX,
231         OPJ_UINT32 ci)
232 {
233     /*
234       0 pfX T1_CHI_THIS           T1_LUT_SGN_W
235       1 tfX T1_SIGMA_1            T1_LUT_SIG_N
236       2 nfX T1_CHI_THIS           T1_LUT_SGN_E
237       3 tfX T1_SIGMA_3            T1_LUT_SIG_W
238       4  fX T1_CHI_(THIS - 1)     T1_LUT_SGN_N
239       5 tfX T1_SIGMA_5            T1_LUT_SIG_E
240       6  fX T1_CHI_(THIS + 1)     T1_LUT_SGN_S
241       7 tfX T1_SIGMA_7            T1_LUT_SIG_S
242     */
243 
244     OPJ_UINT32 lu = (fX >> (ci * 3U)) & (T1_SIGMA_1 | T1_SIGMA_3 | T1_SIGMA_5 |
245                                          T1_SIGMA_7);
246 
247     lu |= (pfX >> (T1_CHI_THIS_I      + (ci * 3U))) & (1U << 0);
248     lu |= (nfX >> (T1_CHI_THIS_I - 2U + (ci * 3U))) & (1U << 2);
249     if (ci == 0U) {
250         lu |= (fX >> (T1_CHI_0_I - 4U)) & (1U << 4);
251     } else {
252         lu |= (fX >> (T1_CHI_1_I - 4U + ((ci - 1U) * 3U))) & (1U << 4);
253     }
254     lu |= (fX >> (T1_CHI_2_I - 6U + (ci * 3U))) & (1U << 6);
255     return lu;
256 }
257 
opj_t1_getctxno_sc(OPJ_UINT32 lu)258 static INLINE OPJ_BYTE opj_t1_getctxno_sc(OPJ_UINT32 lu)
259 {
260     return lut_ctxno_sc[lu];
261 }
262 
opj_t1_getctxno_mag(OPJ_UINT32 f)263 static INLINE OPJ_UINT32 opj_t1_getctxno_mag(OPJ_UINT32 f)
264 {
265     OPJ_UINT32 tmp = (f & T1_SIGMA_NEIGHBOURS) ? T1_CTXNO_MAG + 1 : T1_CTXNO_MAG;
266     OPJ_UINT32 tmp2 = (f & T1_MU_0) ? T1_CTXNO_MAG + 2 : tmp;
267     return tmp2;
268 }
269 
opj_t1_getspb(OPJ_UINT32 lu)270 static INLINE OPJ_BYTE opj_t1_getspb(OPJ_UINT32 lu)
271 {
272     return lut_spb[lu];
273 }
274 
opj_t1_getnmsedec_sig(OPJ_UINT32 x,OPJ_UINT32 bitpos)275 static OPJ_INT16 opj_t1_getnmsedec_sig(OPJ_UINT32 x, OPJ_UINT32 bitpos)
276 {
277     if (bitpos > 0) {
278         return lut_nmsedec_sig[(x >> (bitpos)) & ((1 << T1_NMSEDEC_BITS) - 1)];
279     }
280 
281     return lut_nmsedec_sig0[x & ((1 << T1_NMSEDEC_BITS) - 1)];
282 }
283 
opj_t1_getnmsedec_ref(OPJ_UINT32 x,OPJ_UINT32 bitpos)284 static OPJ_INT16 opj_t1_getnmsedec_ref(OPJ_UINT32 x, OPJ_UINT32 bitpos)
285 {
286     if (bitpos > 0) {
287         return lut_nmsedec_ref[(x >> (bitpos)) & ((1 << T1_NMSEDEC_BITS) - 1)];
288     }
289 
290     return lut_nmsedec_ref0[x & ((1 << T1_NMSEDEC_BITS) - 1)];
291 }
292 
293 #define opj_t1_update_flags_macro(flags, flagsp, ci, s, stride, vsc) \
294 { \
295     /* east */ \
296     flagsp[-1] |= T1_SIGMA_5 << (3U * ci); \
297  \
298     /* mark target as significant */ \
299     flags |= ((s << T1_CHI_1_I) | T1_SIGMA_4) << (3U * ci); \
300  \
301     /* west */ \
302     flagsp[1] |= T1_SIGMA_3 << (3U * ci); \
303  \
304     /* north-west, north, north-east */ \
305     if (ci == 0U && !(vsc)) { \
306         opj_flag_t* north = flagsp - (stride); \
307         *north |= (s << T1_CHI_5_I) | T1_SIGMA_16; \
308         north[-1] |= T1_SIGMA_17; \
309         north[1] |= T1_SIGMA_15; \
310     } \
311  \
312     /* south-west, south, south-east */ \
313     if (ci == 3U) { \
314         opj_flag_t* south = flagsp + (stride); \
315         *south |= (s << T1_CHI_0_I) | T1_SIGMA_1; \
316         south[-1] |= T1_SIGMA_2; \
317         south[1] |= T1_SIGMA_0; \
318     } \
319 }
320 
321 
opj_t1_update_flags(opj_flag_t * flagsp,OPJ_UINT32 ci,OPJ_UINT32 s,OPJ_UINT32 stride,OPJ_UINT32 vsc)322 static INLINE void opj_t1_update_flags(opj_flag_t *flagsp, OPJ_UINT32 ci,
323                                        OPJ_UINT32 s, OPJ_UINT32 stride,
324                                        OPJ_UINT32 vsc)
325 {
326     opj_t1_update_flags_macro(*flagsp, flagsp, ci, s, stride, vsc);
327 }
328 
329 /**
330 Encode significant pass
331 */
opj_t1_enc_sigpass_step(opj_t1_t * t1,opj_flag_t * flagsp,OPJ_INT32 * datap,OPJ_INT32 bpno,OPJ_INT32 one,OPJ_INT32 * nmsedec,OPJ_BYTE type,OPJ_UINT32 ci,OPJ_UINT32 vsc)332 static INLINE void opj_t1_enc_sigpass_step(opj_t1_t *t1,
333         opj_flag_t *flagsp,
334         OPJ_INT32 *datap,
335         OPJ_INT32 bpno,
336         OPJ_INT32 one,
337         OPJ_INT32 *nmsedec,
338         OPJ_BYTE type,
339         OPJ_UINT32 ci,
340         OPJ_UINT32 vsc)
341 {
342     OPJ_UINT32 v;
343 
344     opj_mqc_t *mqc = &(t1->mqc);   /* MQC component */
345 
346     OPJ_UINT32 const flags = *flagsp;
347 
348     if ((flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U))) == 0U &&
349             (flags & (T1_SIGMA_NEIGHBOURS << (ci * 3U))) != 0U) {
350         OPJ_UINT32 ctxt1 = opj_t1_getctxno_zc(mqc, flags >> (ci * 3U));
351         v = (opj_int_abs(*datap) & one) ? 1 : 0;
352 #ifdef DEBUG_ENC_SIG
353         fprintf(stderr, "   ctxt1=%d\n", ctxt1);
354 #endif
355         opj_mqc_setcurctx(mqc, ctxt1);
356         if (type == T1_TYPE_RAW) {  /* BYPASS/LAZY MODE */
357             opj_mqc_bypass_enc(mqc, v);
358         } else {
359             opj_mqc_encode(mqc, v);
360         }
361         if (v) {
362             OPJ_UINT32 lu = opj_t1_getctxtno_sc_or_spb_index(
363                                 *flagsp,
364                                 flagsp[-1], flagsp[1],
365                                 ci);
366             OPJ_UINT32 ctxt2 = opj_t1_getctxno_sc(lu);
367             v = *datap < 0 ? 1U : 0U;
368             *nmsedec += opj_t1_getnmsedec_sig((OPJ_UINT32)opj_int_abs(*datap),
369                                               (OPJ_UINT32)bpno);
370 #ifdef DEBUG_ENC_SIG
371             fprintf(stderr, "   ctxt2=%d\n", ctxt2);
372 #endif
373             opj_mqc_setcurctx(mqc, ctxt2);
374             if (type == T1_TYPE_RAW) {  /* BYPASS/LAZY MODE */
375                 opj_mqc_bypass_enc(mqc, v);
376             } else {
377                 OPJ_UINT32 spb = opj_t1_getspb(lu);
378 #ifdef DEBUG_ENC_SIG
379                 fprintf(stderr, "   spb=%d\n", spb);
380 #endif
381                 opj_mqc_encode(mqc, v ^ spb);
382             }
383             opj_t1_update_flags(flagsp, ci, v, t1->w + 2, vsc);
384         }
385         *flagsp |= T1_PI_THIS << (ci * 3U);
386     }
387 }
388 
opj_t1_dec_sigpass_step_raw(opj_t1_t * t1,opj_flag_t * flagsp,OPJ_INT32 * datap,OPJ_INT32 oneplushalf,OPJ_UINT32 vsc,OPJ_UINT32 ci)389 static INLINE void opj_t1_dec_sigpass_step_raw(
390     opj_t1_t *t1,
391     opj_flag_t *flagsp,
392     OPJ_INT32 *datap,
393     OPJ_INT32 oneplushalf,
394     OPJ_UINT32 vsc,
395     OPJ_UINT32 ci)
396 {
397     OPJ_UINT32 v;
398     opj_mqc_t *mqc = &(t1->mqc);       /* RAW component */
399 
400     OPJ_UINT32 const flags = *flagsp;
401 
402     if ((flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U))) == 0U &&
403             (flags & (T1_SIGMA_NEIGHBOURS << (ci * 3U))) != 0U) {
404         if (opj_mqc_raw_decode(mqc)) {
405             v = opj_mqc_raw_decode(mqc);
406             *datap = v ? -oneplushalf : oneplushalf;
407             opj_t1_update_flags(flagsp, ci, v, t1->w + 2, vsc);
408         }
409         *flagsp |= T1_PI_THIS << (ci * 3U);
410     }
411 }
412 
413 #define opj_t1_dec_sigpass_step_mqc_macro(flags, flagsp, flags_stride, data, \
414                                           data_stride, ci, mqc, curctx, \
415                                           v, a, c, ct, oneplushalf, vsc) \
416 { \
417     if ((flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U))) == 0U && \
418         (flags & (T1_SIGMA_NEIGHBOURS << (ci * 3U))) != 0U) { \
419         OPJ_UINT32 ctxt1 = opj_t1_getctxno_zc(mqc, flags >> (ci * 3U)); \
420         opj_t1_setcurctx(curctx, ctxt1); \
421         opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \
422         if (v) { \
423             OPJ_UINT32 lu = opj_t1_getctxtno_sc_or_spb_index( \
424                                 flags, \
425                                 flagsp[-1], flagsp[1], \
426                                 ci); \
427             OPJ_UINT32 ctxt2 = opj_t1_getctxno_sc(lu); \
428             OPJ_UINT32 spb = opj_t1_getspb(lu); \
429             opj_t1_setcurctx(curctx, ctxt2); \
430             opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \
431             v = v ^ spb; \
432             data[ci*data_stride] = v ? -oneplushalf : oneplushalf; \
433             opj_t1_update_flags_macro(flags, flagsp, ci, v, flags_stride, vsc); \
434         } \
435         flags |= T1_PI_THIS << (ci * 3U); \
436     } \
437 }
438 
opj_t1_dec_sigpass_step_mqc(opj_t1_t * t1,opj_flag_t * flagsp,OPJ_INT32 * datap,OPJ_INT32 oneplushalf,OPJ_UINT32 ci,OPJ_UINT32 flags_stride,OPJ_UINT32 vsc)439 static INLINE void opj_t1_dec_sigpass_step_mqc(
440     opj_t1_t *t1,
441     opj_flag_t *flagsp,
442     OPJ_INT32 *datap,
443     OPJ_INT32 oneplushalf,
444     OPJ_UINT32 ci,
445     OPJ_UINT32 flags_stride,
446     OPJ_UINT32 vsc)
447 {
448     OPJ_UINT32 v;
449 
450     opj_mqc_t *mqc = &(t1->mqc);       /* MQC component */
451     opj_t1_dec_sigpass_step_mqc_macro(*flagsp, flagsp, flags_stride, datap,
452                                       0, ci, mqc, mqc->curctx,
453                                       v, mqc->a, mqc->c, mqc->ct, oneplushalf, vsc);
454 }
455 
opj_t1_enc_sigpass(opj_t1_t * t1,OPJ_INT32 bpno,OPJ_INT32 * nmsedec,OPJ_BYTE type,OPJ_UINT32 cblksty)456 static void opj_t1_enc_sigpass(opj_t1_t *t1,
457                                OPJ_INT32 bpno,
458                                OPJ_INT32 *nmsedec,
459                                OPJ_BYTE type,
460                                OPJ_UINT32 cblksty
461                               )
462 {
463     OPJ_UINT32 i, k;
464     OPJ_INT32 const one = 1 << (bpno + T1_NMSEDEC_FRACBITS);
465     opj_flag_t* f = &T1_FLAGS(0, 0);
466     OPJ_UINT32 const extra = 2;
467 
468     *nmsedec = 0;
469 #ifdef DEBUG_ENC_SIG
470     fprintf(stderr, "enc_sigpass: bpno=%d\n", bpno);
471 #endif
472     for (k = 0; k < (t1->h & ~3U); k += 4) {
473 #ifdef DEBUG_ENC_SIG
474         fprintf(stderr, " k=%d\n", k);
475 #endif
476         for (i = 0; i < t1->w; ++i) {
477 #ifdef DEBUG_ENC_SIG
478             fprintf(stderr, " i=%d\n", i);
479 #endif
480             if (*f == 0U) {
481                 /* Nothing to do for any of the 4 data points */
482                 f++;
483                 continue;
484             }
485             opj_t1_enc_sigpass_step(
486                 t1,
487                 f,
488                 &t1->data[((k + 0) * t1->data_stride) + i],
489                 bpno,
490                 one,
491                 nmsedec,
492                 type,
493                 0, cblksty & J2K_CCP_CBLKSTY_VSC);
494             opj_t1_enc_sigpass_step(
495                 t1,
496                 f,
497                 &t1->data[((k + 1) * t1->data_stride) + i],
498                 bpno,
499                 one,
500                 nmsedec,
501                 type,
502                 1, 0);
503             opj_t1_enc_sigpass_step(
504                 t1,
505                 f,
506                 &t1->data[((k + 2) * t1->data_stride) + i],
507                 bpno,
508                 one,
509                 nmsedec,
510                 type,
511                 2, 0);
512             opj_t1_enc_sigpass_step(
513                 t1,
514                 f,
515                 &t1->data[((k + 3) * t1->data_stride) + i],
516                 bpno,
517                 one,
518                 nmsedec,
519                 type,
520                 3, 0);
521             ++f;
522         }
523         f += extra;
524     }
525 
526     if (k < t1->h) {
527         OPJ_UINT32 j;
528 #ifdef DEBUG_ENC_SIG
529         fprintf(stderr, " k=%d\n", k);
530 #endif
531         for (i = 0; i < t1->w; ++i) {
532 #ifdef DEBUG_ENC_SIG
533             fprintf(stderr, " i=%d\n", i);
534 #endif
535             if (*f == 0U) {
536                 /* Nothing to do for any of the 4 data points */
537                 f++;
538                 continue;
539             }
540             for (j = k; j < t1->h; ++j) {
541                 opj_t1_enc_sigpass_step(
542                     t1,
543                     f,
544                     &t1->data[(j * t1->data_stride) + i],
545                     bpno,
546                     one,
547                     nmsedec,
548                     type,
549                     j - k,
550                     (j == k && (cblksty & J2K_CCP_CBLKSTY_VSC) != 0));
551             }
552             ++f;
553         }
554     }
555 }
556 
opj_t1_dec_sigpass_raw(opj_t1_t * t1,OPJ_INT32 bpno,OPJ_INT32 cblksty)557 static void opj_t1_dec_sigpass_raw(
558     opj_t1_t *t1,
559     OPJ_INT32 bpno,
560     OPJ_INT32 cblksty)
561 {
562     OPJ_INT32 one, half, oneplushalf;
563     OPJ_UINT32 i, j, k;
564     OPJ_INT32 *data = t1->data;
565     opj_flag_t *flagsp = &T1_FLAGS(0, 0);
566     const OPJ_UINT32 l_w = t1->w;
567     one = 1 << bpno;
568     half = one >> 1;
569     oneplushalf = one | half;
570 
571     for (k = 0; k < (t1->h & ~3U); k += 4, flagsp += 2, data += 3 * l_w) {
572         for (i = 0; i < l_w; ++i, ++flagsp, ++data) {
573             opj_flag_t flags = *flagsp;
574             if (flags != 0) {
575                 opj_t1_dec_sigpass_step_raw(
576                     t1,
577                     flagsp,
578                     data,
579                     oneplushalf,
580                     cblksty & J2K_CCP_CBLKSTY_VSC, /* vsc */
581                     0U);
582                 opj_t1_dec_sigpass_step_raw(
583                     t1,
584                     flagsp,
585                     data + l_w,
586                     oneplushalf,
587                     OPJ_FALSE, /* vsc */
588                     1U);
589                 opj_t1_dec_sigpass_step_raw(
590                     t1,
591                     flagsp,
592                     data + 2 * l_w,
593                     oneplushalf,
594                     OPJ_FALSE, /* vsc */
595                     2U);
596                 opj_t1_dec_sigpass_step_raw(
597                     t1,
598                     flagsp,
599                     data + 3 * l_w,
600                     oneplushalf,
601                     OPJ_FALSE, /* vsc */
602                     3U);
603             }
604         }
605     }
606     if (k < t1->h) {
607         for (i = 0; i < l_w; ++i, ++flagsp, ++data) {
608             for (j = 0; j < t1->h - k; ++j) {
609                 opj_t1_dec_sigpass_step_raw(
610                     t1,
611                     flagsp,
612                     data + j * l_w,
613                     oneplushalf,
614                     cblksty & J2K_CCP_CBLKSTY_VSC, /* vsc */
615                     j);
616             }
617         }
618     }
619 }
620 
621 #define opj_t1_dec_sigpass_mqc_internal(t1, bpno, vsc, w, h, flags_stride) \
622 { \
623         OPJ_INT32 one, half, oneplushalf; \
624         OPJ_UINT32 i, j, k; \
625         register OPJ_INT32 *data = t1->data; \
626         register opj_flag_t *flagsp = &t1->flags[(flags_stride) + 1]; \
627         const OPJ_UINT32 l_w = w; \
628         opj_mqc_t* mqc = &(t1->mqc); \
629         DOWNLOAD_MQC_VARIABLES(mqc, curctx, c, a, ct); \
630         register OPJ_UINT32 v; \
631         one = 1 << bpno; \
632         half = one >> 1; \
633         oneplushalf = one | half; \
634         for (k = 0; k < (h & ~3u); k += 4, data += 3*l_w, flagsp += 2) { \
635                 for (i = 0; i < l_w; ++i, ++data, ++flagsp) { \
636                         opj_flag_t flags = *flagsp; \
637                         if( flags != 0 ) { \
638                             opj_t1_dec_sigpass_step_mqc_macro( \
639                                 flags, flagsp, flags_stride, data, \
640                                 l_w, 0, mqc, curctx, v, a, c, ct, oneplushalf, vsc); \
641                             opj_t1_dec_sigpass_step_mqc_macro( \
642                                 flags, flagsp, flags_stride, data, \
643                                 l_w, 1, mqc, curctx, v, a, c, ct, oneplushalf, OPJ_FALSE); \
644                             opj_t1_dec_sigpass_step_mqc_macro( \
645                                 flags, flagsp, flags_stride, data, \
646                                 l_w, 2, mqc, curctx, v, a, c, ct, oneplushalf, OPJ_FALSE); \
647                             opj_t1_dec_sigpass_step_mqc_macro( \
648                                 flags, flagsp, flags_stride, data, \
649                                 l_w, 3, mqc, curctx, v, a, c, ct, oneplushalf, OPJ_FALSE); \
650                             *flagsp = flags; \
651                         } \
652                 } \
653         } \
654         UPLOAD_MQC_VARIABLES(mqc, curctx, c, a, ct); \
655         if( k < h ) { \
656             for (i = 0; i < l_w; ++i, ++data, ++flagsp) { \
657                 for (j = 0; j < h - k; ++j) { \
658                         opj_t1_dec_sigpass_step_mqc(t1, flagsp, \
659                             data + j * l_w, oneplushalf, j, flags_stride, vsc); \
660                 } \
661             } \
662         } \
663 }
664 
opj_t1_dec_sigpass_mqc_64x64_novsc(opj_t1_t * t1,OPJ_INT32 bpno)665 static void opj_t1_dec_sigpass_mqc_64x64_novsc(
666     opj_t1_t *t1,
667     OPJ_INT32 bpno)
668 {
669     opj_t1_dec_sigpass_mqc_internal(t1, bpno, OPJ_FALSE, 64, 64, 66);
670 }
671 
opj_t1_dec_sigpass_mqc_64x64_vsc(opj_t1_t * t1,OPJ_INT32 bpno)672 static void opj_t1_dec_sigpass_mqc_64x64_vsc(
673     opj_t1_t *t1,
674     OPJ_INT32 bpno)
675 {
676     opj_t1_dec_sigpass_mqc_internal(t1, bpno, OPJ_TRUE, 64, 64, 66);
677 }
678 
opj_t1_dec_sigpass_mqc_generic_novsc(opj_t1_t * t1,OPJ_INT32 bpno)679 static void opj_t1_dec_sigpass_mqc_generic_novsc(
680     opj_t1_t *t1,
681     OPJ_INT32 bpno)
682 {
683     opj_t1_dec_sigpass_mqc_internal(t1, bpno, OPJ_FALSE, t1->w, t1->h,
684                                     t1->w + 2U);
685 }
686 
opj_t1_dec_sigpass_mqc_generic_vsc(opj_t1_t * t1,OPJ_INT32 bpno)687 static void opj_t1_dec_sigpass_mqc_generic_vsc(
688     opj_t1_t *t1,
689     OPJ_INT32 bpno)
690 {
691     opj_t1_dec_sigpass_mqc_internal(t1, bpno, OPJ_TRUE, t1->w, t1->h,
692                                     t1->w + 2U);
693 }
694 
opj_t1_dec_sigpass_mqc(opj_t1_t * t1,OPJ_INT32 bpno,OPJ_INT32 cblksty)695 static void opj_t1_dec_sigpass_mqc(
696     opj_t1_t *t1,
697     OPJ_INT32 bpno,
698     OPJ_INT32 cblksty)
699 {
700     if (t1->w == 64 && t1->h == 64) {
701         if (cblksty & J2K_CCP_CBLKSTY_VSC) {
702             opj_t1_dec_sigpass_mqc_64x64_vsc(t1, bpno);
703         } else {
704             opj_t1_dec_sigpass_mqc_64x64_novsc(t1, bpno);
705         }
706     } else {
707         if (cblksty & J2K_CCP_CBLKSTY_VSC) {
708             opj_t1_dec_sigpass_mqc_generic_vsc(t1, bpno);
709         } else {
710             opj_t1_dec_sigpass_mqc_generic_novsc(t1, bpno);
711         }
712     }
713 }
714 
715 /**
716 Encode refinement pass step
717 */
opj_t1_enc_refpass_step(opj_t1_t * t1,opj_flag_t * flagsp,OPJ_INT32 * datap,OPJ_INT32 bpno,OPJ_INT32 one,OPJ_INT32 * nmsedec,OPJ_BYTE type,OPJ_UINT32 ci)718 static INLINE void opj_t1_enc_refpass_step(opj_t1_t *t1,
719         opj_flag_t *flagsp,
720         OPJ_INT32 *datap,
721         OPJ_INT32 bpno,
722         OPJ_INT32 one,
723         OPJ_INT32 *nmsedec,
724         OPJ_BYTE type,
725         OPJ_UINT32 ci)
726 {
727     OPJ_UINT32 v;
728 
729     opj_mqc_t *mqc = &(t1->mqc);   /* MQC component */
730 
731     OPJ_UINT32 const shift_flags =
732         (*flagsp >> (ci * 3U));
733 
734     if ((shift_flags & (T1_SIGMA_THIS | T1_PI_THIS)) == T1_SIGMA_THIS) {
735         OPJ_UINT32 ctxt = opj_t1_getctxno_mag(shift_flags);
736         *nmsedec += opj_t1_getnmsedec_ref((OPJ_UINT32)opj_int_abs(*datap),
737                                           (OPJ_UINT32)bpno);
738         v = (opj_int_abs(*datap) & one) ? 1 : 0;
739 #ifdef DEBUG_ENC_REF
740         fprintf(stderr, "  ctxt=%d\n", ctxt);
741 #endif
742         opj_mqc_setcurctx(mqc, ctxt);
743         if (type == T1_TYPE_RAW) {  /* BYPASS/LAZY MODE */
744             opj_mqc_bypass_enc(mqc, v);
745         } else {
746             opj_mqc_encode(mqc, v);
747         }
748         *flagsp |= T1_MU_THIS << (ci * 3U);
749     }
750 }
751 
752 
opj_t1_dec_refpass_step_raw(opj_t1_t * t1,opj_flag_t * flagsp,OPJ_INT32 * datap,OPJ_INT32 poshalf,OPJ_UINT32 ci)753 static INLINE void opj_t1_dec_refpass_step_raw(
754     opj_t1_t *t1,
755     opj_flag_t *flagsp,
756     OPJ_INT32 *datap,
757     OPJ_INT32 poshalf,
758     OPJ_UINT32 ci)
759 {
760     OPJ_UINT32 v;
761 
762     opj_mqc_t *mqc = &(t1->mqc);       /* RAW component */
763 
764     if ((*flagsp & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U))) ==
765             (T1_SIGMA_THIS << (ci * 3U))) {
766         v = opj_mqc_raw_decode(mqc);
767         *datap += (v ^ (*datap < 0)) ? poshalf : -poshalf;
768         *flagsp |= T1_MU_THIS << (ci * 3U);
769     }
770 }
771 
772 #define opj_t1_dec_refpass_step_mqc_macro(flags, data, data_stride, ci, \
773                                           mqc, curctx, v, a, c, ct, poshalf) \
774 { \
775     if ((flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U))) == \
776             (T1_SIGMA_THIS << (ci * 3U))) { \
777         OPJ_UINT32 ctxt = opj_t1_getctxno_mag(flags >> (ci * 3U)); \
778         opj_t1_setcurctx(curctx, ctxt); \
779         opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \
780         data[ci*data_stride] += (v ^ (data[ci*data_stride] < 0)) ? poshalf : -poshalf; \
781         flags |= T1_MU_THIS << (ci * 3U); \
782     } \
783 }
784 
opj_t1_dec_refpass_step_mqc(opj_t1_t * t1,opj_flag_t * flagsp,OPJ_INT32 * datap,OPJ_INT32 poshalf,OPJ_UINT32 ci)785 static INLINE void opj_t1_dec_refpass_step_mqc(
786     opj_t1_t *t1,
787     opj_flag_t *flagsp,
788     OPJ_INT32 *datap,
789     OPJ_INT32 poshalf,
790     OPJ_UINT32 ci)
791 {
792     OPJ_UINT32 v;
793 
794     opj_mqc_t *mqc = &(t1->mqc);       /* MQC component */
795     opj_t1_dec_refpass_step_mqc_macro(*flagsp, datap, 0, ci,
796                                       mqc, mqc->curctx, v, mqc->a, mqc->c,
797                                       mqc->ct, poshalf);
798 }
799 
opj_t1_enc_refpass(opj_t1_t * t1,OPJ_INT32 bpno,OPJ_INT32 * nmsedec,OPJ_BYTE type)800 static void opj_t1_enc_refpass(
801     opj_t1_t *t1,
802     OPJ_INT32 bpno,
803     OPJ_INT32 *nmsedec,
804     OPJ_BYTE type)
805 {
806     OPJ_UINT32 i, k;
807     const OPJ_INT32 one = 1 << (bpno + T1_NMSEDEC_FRACBITS);
808     opj_flag_t* f = &T1_FLAGS(0, 0);
809     const OPJ_UINT32 extra = 2U;
810 
811     *nmsedec = 0;
812 #ifdef DEBUG_ENC_REF
813     fprintf(stderr, "enc_refpass: bpno=%d\n", bpno);
814 #endif
815     for (k = 0; k < (t1->h & ~3U); k += 4) {
816 #ifdef DEBUG_ENC_REF
817         fprintf(stderr, " k=%d\n", k);
818 #endif
819         for (i = 0; i < t1->w; ++i) {
820 #ifdef DEBUG_ENC_REF
821             fprintf(stderr, " i=%d\n", i);
822 #endif
823             if ((*f & (T1_SIGMA_4 | T1_SIGMA_7 | T1_SIGMA_10 | T1_SIGMA_13)) == 0) {
824                 /* none significant */
825                 f++;
826                 continue;
827             }
828             if ((*f & (T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3)) ==
829                     (T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3)) {
830                 /* all processed by sigpass */
831                 f++;
832                 continue;
833             }
834 
835             opj_t1_enc_refpass_step(
836                 t1,
837                 f,
838                 &t1->data[((k + 0) * t1->data_stride) + i],
839                 bpno,
840                 one,
841                 nmsedec,
842                 type,
843                 0);
844             opj_t1_enc_refpass_step(
845                 t1,
846                 f,
847                 &t1->data[((k + 1) * t1->data_stride) + i],
848                 bpno,
849                 one,
850                 nmsedec,
851                 type,
852                 1);
853             opj_t1_enc_refpass_step(
854                 t1,
855                 f,
856                 &t1->data[((k + 2) * t1->data_stride) + i],
857                 bpno,
858                 one,
859                 nmsedec,
860                 type,
861                 2);
862             opj_t1_enc_refpass_step(
863                 t1,
864                 f,
865                 &t1->data[((k + 3) * t1->data_stride) + i],
866                 bpno,
867                 one,
868                 nmsedec,
869                 type,
870                 3);
871             ++f;
872         }
873         f += extra;
874     }
875 
876     if (k < t1->h) {
877         OPJ_UINT32 j;
878 #ifdef DEBUG_ENC_REF
879         fprintf(stderr, " k=%d\n", k);
880 #endif
881         for (i = 0; i < t1->w; ++i) {
882 #ifdef DEBUG_ENC_REF
883             fprintf(stderr, " i=%d\n", i);
884 #endif
885             if ((*f & (T1_SIGMA_4 | T1_SIGMA_7 | T1_SIGMA_10 | T1_SIGMA_13)) == 0) {
886                 /* none significant */
887                 f++;
888                 continue;
889             }
890             for (j = k; j < t1->h; ++j) {
891                 opj_t1_enc_refpass_step(
892                     t1,
893                     f,
894                     &t1->data[(j * t1->data_stride) + i],
895                     bpno,
896                     one,
897                     nmsedec,
898                     type,
899                     j - k);
900             }
901             ++f;
902         }
903     }
904 }
905 
906 
opj_t1_dec_refpass_raw(opj_t1_t * t1,OPJ_INT32 bpno)907 static void opj_t1_dec_refpass_raw(
908     opj_t1_t *t1,
909     OPJ_INT32 bpno)
910 {
911     OPJ_INT32 one, poshalf;
912     OPJ_UINT32 i, j, k;
913     OPJ_INT32 *data = t1->data;
914     opj_flag_t *flagsp = &T1_FLAGS(0, 0);
915     const OPJ_UINT32 l_w = t1->w;
916     one = 1 << bpno;
917     poshalf = one >> 1;
918     for (k = 0; k < (t1->h & ~3U); k += 4, flagsp += 2, data += 3 * l_w) {
919         for (i = 0; i < l_w; ++i, ++flagsp, ++data) {
920             opj_flag_t flags = *flagsp;
921             if (flags != 0) {
922                 opj_t1_dec_refpass_step_raw(
923                     t1,
924                     flagsp,
925                     data,
926                     poshalf,
927                     0U);
928                 opj_t1_dec_refpass_step_raw(
929                     t1,
930                     flagsp,
931                     data + l_w,
932                     poshalf,
933                     1U);
934                 opj_t1_dec_refpass_step_raw(
935                     t1,
936                     flagsp,
937                     data + 2 * l_w,
938                     poshalf,
939                     2U);
940                 opj_t1_dec_refpass_step_raw(
941                     t1,
942                     flagsp,
943                     data + 3 * l_w,
944                     poshalf,
945                     3U);
946             }
947         }
948     }
949     if (k < t1->h) {
950         for (i = 0; i < l_w; ++i, ++flagsp, ++data) {
951             for (j = 0; j < t1->h - k; ++j) {
952                 opj_t1_dec_refpass_step_raw(
953                     t1,
954                     flagsp,
955                     data + j * l_w,
956                     poshalf,
957                     j);
958             }
959         }
960     }
961 }
962 
963 #define opj_t1_dec_refpass_mqc_internal(t1, bpno, w, h, flags_stride) \
964 { \
965         OPJ_INT32 one, poshalf; \
966         OPJ_UINT32 i, j, k; \
967         register OPJ_INT32 *data = t1->data; \
968         register opj_flag_t *flagsp = &t1->flags[flags_stride + 1]; \
969         const OPJ_UINT32 l_w = w; \
970         opj_mqc_t* mqc = &(t1->mqc); \
971         DOWNLOAD_MQC_VARIABLES(mqc, curctx, c, a, ct); \
972         register OPJ_UINT32 v; \
973         one = 1 << bpno; \
974         poshalf = one >> 1; \
975         for (k = 0; k < (h & ~3u); k += 4, data += 3*l_w, flagsp += 2) { \
976                 for (i = 0; i < l_w; ++i, ++data, ++flagsp) { \
977                         opj_flag_t flags = *flagsp; \
978                         if( flags != 0 ) { \
979                             opj_t1_dec_refpass_step_mqc_macro( \
980                                 flags, data, l_w, 0, \
981                                 mqc, curctx, v, a, c, ct, poshalf); \
982                             opj_t1_dec_refpass_step_mqc_macro( \
983                                 flags, data, l_w, 1, \
984                                 mqc, curctx, v, a, c, ct, poshalf); \
985                             opj_t1_dec_refpass_step_mqc_macro( \
986                                 flags, data, l_w, 2, \
987                                 mqc, curctx, v, a, c, ct, poshalf); \
988                             opj_t1_dec_refpass_step_mqc_macro( \
989                                 flags, data, l_w, 3, \
990                                 mqc, curctx, v, a, c, ct, poshalf); \
991                             *flagsp = flags; \
992                         } \
993                 } \
994         } \
995         UPLOAD_MQC_VARIABLES(mqc, curctx, c, a, ct); \
996         if( k < h ) { \
997             for (i = 0; i < l_w; ++i, ++data, ++flagsp) { \
998                 for (j = 0; j < h - k; ++j) { \
999                         opj_t1_dec_refpass_step_mqc(t1, flagsp, data + j * l_w, poshalf, j); \
1000                 } \
1001             } \
1002         } \
1003 }
1004 
opj_t1_dec_refpass_mqc_64x64(opj_t1_t * t1,OPJ_INT32 bpno)1005 static void opj_t1_dec_refpass_mqc_64x64(
1006     opj_t1_t *t1,
1007     OPJ_INT32 bpno)
1008 {
1009     opj_t1_dec_refpass_mqc_internal(t1, bpno, 64, 64, 66);
1010 }
1011 
opj_t1_dec_refpass_mqc_generic(opj_t1_t * t1,OPJ_INT32 bpno)1012 static void opj_t1_dec_refpass_mqc_generic(
1013     opj_t1_t *t1,
1014     OPJ_INT32 bpno)
1015 {
1016     opj_t1_dec_refpass_mqc_internal(t1, bpno, t1->w, t1->h, t1->w + 2U);
1017 }
1018 
opj_t1_dec_refpass_mqc(opj_t1_t * t1,OPJ_INT32 bpno)1019 static void opj_t1_dec_refpass_mqc(
1020     opj_t1_t *t1,
1021     OPJ_INT32 bpno)
1022 {
1023     if (t1->w == 64 && t1->h == 64) {
1024         opj_t1_dec_refpass_mqc_64x64(t1, bpno);
1025     } else {
1026         opj_t1_dec_refpass_mqc_generic(t1, bpno);
1027     }
1028 }
1029 
1030 /**
1031 Encode clean-up pass step
1032 */
opj_t1_enc_clnpass_step(opj_t1_t * t1,opj_flag_t * flagsp,OPJ_INT32 * datap,OPJ_INT32 bpno,OPJ_INT32 one,OPJ_INT32 * nmsedec,OPJ_UINT32 agg,OPJ_UINT32 runlen,OPJ_UINT32 lim,OPJ_UINT32 cblksty)1033 static void opj_t1_enc_clnpass_step(
1034     opj_t1_t *t1,
1035     opj_flag_t *flagsp,
1036     OPJ_INT32 *datap,
1037     OPJ_INT32 bpno,
1038     OPJ_INT32 one,
1039     OPJ_INT32 *nmsedec,
1040     OPJ_UINT32 agg,
1041     OPJ_UINT32 runlen,
1042     OPJ_UINT32 lim,
1043     OPJ_UINT32 cblksty)
1044 {
1045     OPJ_UINT32 v;
1046     OPJ_UINT32 ci;
1047     opj_mqc_t *mqc = &(t1->mqc);   /* MQC component */
1048 
1049     const OPJ_UINT32 check = (T1_SIGMA_4 | T1_SIGMA_7 | T1_SIGMA_10 | T1_SIGMA_13 |
1050                               T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3);
1051 
1052     if ((*flagsp & check) == check) {
1053         if (runlen == 0) {
1054             *flagsp &= ~(T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3);
1055         } else if (runlen == 1) {
1056             *flagsp &= ~(T1_PI_1 | T1_PI_2 | T1_PI_3);
1057         } else if (runlen == 2) {
1058             *flagsp &= ~(T1_PI_2 | T1_PI_3);
1059         } else if (runlen == 3) {
1060             *flagsp &= ~(T1_PI_3);
1061         }
1062         return;
1063     }
1064 
1065     for (ci = runlen; ci < lim; ++ci) {
1066         OPJ_UINT32 vsc;
1067         opj_flag_t flags;
1068         OPJ_UINT32 ctxt1;
1069 
1070         flags = *flagsp;
1071 
1072         if ((agg != 0) && (ci == runlen)) {
1073             goto LABEL_PARTIAL;
1074         }
1075 
1076         if (!(flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U)))) {
1077             ctxt1 = opj_t1_getctxno_zc(mqc, flags >> (ci * 3U));
1078 #ifdef DEBUG_ENC_CLN
1079             printf("   ctxt1=%d\n", ctxt1);
1080 #endif
1081             opj_mqc_setcurctx(mqc, ctxt1);
1082             v = (opj_int_abs(*datap) & one) ? 1 : 0;
1083             opj_mqc_encode(mqc, v);
1084             if (v) {
1085                 OPJ_UINT32 ctxt2, spb;
1086                 OPJ_UINT32 lu;
1087 LABEL_PARTIAL:
1088                 lu = opj_t1_getctxtno_sc_or_spb_index(
1089                          *flagsp,
1090                          flagsp[-1], flagsp[1],
1091                          ci);
1092                 *nmsedec += opj_t1_getnmsedec_sig((OPJ_UINT32)opj_int_abs(*datap),
1093                                                   (OPJ_UINT32)bpno);
1094                 ctxt2 = opj_t1_getctxno_sc(lu);
1095 #ifdef DEBUG_ENC_CLN
1096                 printf("   ctxt2=%d\n", ctxt2);
1097 #endif
1098                 opj_mqc_setcurctx(mqc, ctxt2);
1099 
1100                 v = *datap < 0 ? 1U : 0U;
1101                 spb = opj_t1_getspb(lu);
1102 #ifdef DEBUG_ENC_CLN
1103                 printf("   spb=%d\n", spb);
1104 #endif
1105                 opj_mqc_encode(mqc, v ^ spb);
1106                 vsc = ((cblksty & J2K_CCP_CBLKSTY_VSC) && (ci == 0)) ? 1 : 0;
1107                 opj_t1_update_flags(flagsp, ci, v, t1->w + 2U, vsc);
1108             }
1109         }
1110         *flagsp &= ~(T1_PI_THIS << (3U * ci));
1111         datap += t1->data_stride;
1112     }
1113 }
1114 
1115 #define opj_t1_dec_clnpass_step_macro(check_flags, partial, \
1116                                       flags, flagsp, flags_stride, data, \
1117                                       data_stride, ci, mqc, curctx, \
1118                                       v, a, c, ct, oneplushalf, vsc) \
1119 { \
1120     if ( !check_flags || !(flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U)))) {\
1121         do { \
1122             if( !partial ) { \
1123                 OPJ_UINT32 ctxt1 = opj_t1_getctxno_zc(mqc, flags >> (ci * 3U)); \
1124                 opj_t1_setcurctx(curctx, ctxt1); \
1125                 opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \
1126                 if( !v ) \
1127                     break; \
1128             } \
1129             { \
1130                 OPJ_UINT32 lu = opj_t1_getctxtno_sc_or_spb_index( \
1131                                     flags, flagsp[-1], flagsp[1], \
1132                                     ci); \
1133                 opj_t1_setcurctx(curctx, opj_t1_getctxno_sc(lu)); \
1134                 opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \
1135                 v = v ^ opj_t1_getspb(lu); \
1136                 data[ci*data_stride] = v ? -oneplushalf : oneplushalf; \
1137                 opj_t1_update_flags_macro(flags, flagsp, ci, v, flags_stride, vsc); \
1138             } \
1139         } while(0); \
1140     } \
1141 }
1142 
opj_t1_dec_clnpass_step(opj_t1_t * t1,opj_flag_t * flagsp,OPJ_INT32 * datap,OPJ_INT32 oneplushalf,OPJ_UINT32 ci,OPJ_UINT32 vsc)1143 static void opj_t1_dec_clnpass_step(
1144     opj_t1_t *t1,
1145     opj_flag_t *flagsp,
1146     OPJ_INT32 *datap,
1147     OPJ_INT32 oneplushalf,
1148     OPJ_UINT32 ci,
1149     OPJ_UINT32 vsc)
1150 {
1151     OPJ_UINT32 v;
1152 
1153     opj_mqc_t *mqc = &(t1->mqc);   /* MQC component */
1154     opj_t1_dec_clnpass_step_macro(OPJ_TRUE, OPJ_FALSE,
1155                                   *flagsp, flagsp, t1->w + 2U, datap,
1156                                   0, ci, mqc, mqc->curctx,
1157                                   v, mqc->a, mqc->c, mqc->ct, oneplushalf, vsc);
1158 }
1159 
opj_t1_enc_clnpass(opj_t1_t * t1,OPJ_INT32 bpno,OPJ_INT32 * nmsedec,OPJ_UINT32 cblksty)1160 static void opj_t1_enc_clnpass(
1161     opj_t1_t *t1,
1162     OPJ_INT32 bpno,
1163     OPJ_INT32 *nmsedec,
1164     OPJ_UINT32 cblksty)
1165 {
1166     OPJ_UINT32 i, k;
1167     const OPJ_INT32 one = 1 << (bpno + T1_NMSEDEC_FRACBITS);
1168     OPJ_UINT32 agg, runlen;
1169 
1170     opj_mqc_t *mqc = &(t1->mqc);   /* MQC component */
1171 
1172     *nmsedec = 0;
1173 #ifdef DEBUG_ENC_CLN
1174     printf("enc_clnpass: bpno=%d\n", bpno);
1175 #endif
1176     for (k = 0; k < (t1->h & ~3U); k += 4) {
1177 #ifdef DEBUG_ENC_CLN
1178         printf(" k=%d\n", k);
1179 #endif
1180         for (i = 0; i < t1->w; ++i) {
1181 #ifdef DEBUG_ENC_CLN
1182             printf("  i=%d\n", i);
1183 #endif
1184             agg = !(T1_FLAGS(i, k));
1185 #ifdef DEBUG_ENC_CLN
1186             printf("   agg=%d\n", agg);
1187 #endif
1188             if (agg) {
1189                 for (runlen = 0; runlen < 4; ++runlen) {
1190                     if (opj_int_abs(t1->data[((k + runlen)*t1->data_stride) + i]) & one) {
1191                         break;
1192                     }
1193                 }
1194                 opj_mqc_setcurctx(mqc, T1_CTXNO_AGG);
1195                 opj_mqc_encode(mqc, runlen != 4);
1196                 if (runlen == 4) {
1197                     continue;
1198                 }
1199                 opj_mqc_setcurctx(mqc, T1_CTXNO_UNI);
1200                 opj_mqc_encode(mqc, runlen >> 1);
1201                 opj_mqc_encode(mqc, runlen & 1);
1202             } else {
1203                 runlen = 0;
1204             }
1205             opj_t1_enc_clnpass_step(
1206                 t1,
1207                 &T1_FLAGS(i, k),
1208                 &t1->data[((k + runlen) * t1->data_stride) + i],
1209                 bpno,
1210                 one,
1211                 nmsedec,
1212                 agg,
1213                 runlen,
1214                 4U,
1215                 cblksty);
1216         }
1217     }
1218     if (k < t1->h) {
1219         agg = 0;
1220         runlen = 0;
1221 #ifdef DEBUG_ENC_CLN
1222         printf(" k=%d\n", k);
1223 #endif
1224         for (i = 0; i < t1->w; ++i) {
1225 #ifdef DEBUG_ENC_CLN
1226             printf("  i=%d\n", i);
1227             printf("   agg=%d\n", agg);
1228 #endif
1229             opj_t1_enc_clnpass_step(
1230                 t1,
1231                 &T1_FLAGS(i, k),
1232                 &t1->data[((k + runlen) * t1->data_stride) + i],
1233                 bpno,
1234                 one,
1235                 nmsedec,
1236                 agg,
1237                 runlen,
1238                 t1->h - k,
1239                 cblksty);
1240         }
1241     }
1242 }
1243 
1244 #define opj_t1_dec_clnpass_internal(t1, bpno, vsc, w, h, flags_stride) \
1245 { \
1246     OPJ_INT32 one, half, oneplushalf; \
1247     OPJ_UINT32 runlen; \
1248     OPJ_UINT32 i, j, k; \
1249     const OPJ_UINT32 l_w = w; \
1250     opj_mqc_t* mqc = &(t1->mqc); \
1251     register OPJ_INT32 *data = t1->data; \
1252     register opj_flag_t *flagsp = &t1->flags[flags_stride + 1]; \
1253     DOWNLOAD_MQC_VARIABLES(mqc, curctx, c, a, ct); \
1254     register OPJ_UINT32 v; \
1255     one = 1 << bpno; \
1256     half = one >> 1; \
1257     oneplushalf = one | half; \
1258     for (k = 0; k < (h & ~3u); k += 4, data += 3*l_w, flagsp += 2) { \
1259         for (i = 0; i < l_w; ++i, ++data, ++flagsp) { \
1260             opj_flag_t flags = *flagsp; \
1261             if (flags == 0) { \
1262                 OPJ_UINT32 partial = OPJ_TRUE; \
1263                 opj_t1_setcurctx(curctx, T1_CTXNO_AGG); \
1264                 opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \
1265                 if (!v) { \
1266                     continue; \
1267                 } \
1268                 opj_t1_setcurctx(curctx, T1_CTXNO_UNI); \
1269                 opj_mqc_decode_macro(runlen, mqc, curctx, a, c, ct); \
1270                 opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \
1271                 runlen = (runlen << 1) | v; \
1272                 switch(runlen) { \
1273                     case 0: \
1274                         opj_t1_dec_clnpass_step_macro(OPJ_FALSE, OPJ_TRUE,\
1275                                             flags, flagsp, flags_stride, data, \
1276                                             l_w, 0, mqc, curctx, \
1277                                             v, a, c, ct, oneplushalf, vsc); \
1278                         partial = OPJ_FALSE; \
1279                         /* FALLTHRU */ \
1280                     case 1: \
1281                         opj_t1_dec_clnpass_step_macro(OPJ_FALSE, partial,\
1282                                             flags, flagsp, flags_stride, data, \
1283                                             l_w, 1, mqc, curctx, \
1284                                             v, a, c, ct, oneplushalf, OPJ_FALSE); \
1285                         partial = OPJ_FALSE; \
1286                         /* FALLTHRU */ \
1287                     case 2: \
1288                         opj_t1_dec_clnpass_step_macro(OPJ_FALSE, partial,\
1289                                             flags, flagsp, flags_stride, data, \
1290                                             l_w, 2, mqc, curctx, \
1291                                             v, a, c, ct, oneplushalf, OPJ_FALSE); \
1292                         partial = OPJ_FALSE; \
1293                         /* FALLTHRU */ \
1294                     case 3: \
1295                         opj_t1_dec_clnpass_step_macro(OPJ_FALSE, partial,\
1296                                             flags, flagsp, flags_stride, data, \
1297                                             l_w, 3, mqc, curctx, \
1298                                             v, a, c, ct, oneplushalf, OPJ_FALSE); \
1299                         break; \
1300                 } \
1301             } else { \
1302                 opj_t1_dec_clnpass_step_macro(OPJ_TRUE, OPJ_FALSE, \
1303                                     flags, flagsp, flags_stride, data, \
1304                                     l_w, 0, mqc, curctx, \
1305                                     v, a, c, ct, oneplushalf, vsc); \
1306                 opj_t1_dec_clnpass_step_macro(OPJ_TRUE, OPJ_FALSE, \
1307                                     flags, flagsp, flags_stride, data, \
1308                                     l_w, 1, mqc, curctx, \
1309                                     v, a, c, ct, oneplushalf, OPJ_FALSE); \
1310                 opj_t1_dec_clnpass_step_macro(OPJ_TRUE, OPJ_FALSE, \
1311                                     flags, flagsp, flags_stride, data, \
1312                                     l_w, 2, mqc, curctx, \
1313                                     v, a, c, ct, oneplushalf, OPJ_FALSE); \
1314                 opj_t1_dec_clnpass_step_macro(OPJ_TRUE, OPJ_FALSE, \
1315                                     flags, flagsp, flags_stride, data, \
1316                                     l_w, 3, mqc, curctx, \
1317                                     v, a, c, ct, oneplushalf, OPJ_FALSE); \
1318             } \
1319             *flagsp = flags & ~(T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3); \
1320         } \
1321     } \
1322     UPLOAD_MQC_VARIABLES(mqc, curctx, c, a, ct); \
1323     if( k < h ) { \
1324         for (i = 0; i < l_w; ++i, ++flagsp, ++data) { \
1325             for (j = 0; j < h - k; ++j) { \
1326                 opj_t1_dec_clnpass_step(t1, flagsp, data + j * l_w, oneplushalf, j, vsc); \
1327             } \
1328             *flagsp &= ~(T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3); \
1329         } \
1330     } \
1331 }
1332 
opj_t1_dec_clnpass_check_segsym(opj_t1_t * t1,OPJ_INT32 cblksty)1333 static void opj_t1_dec_clnpass_check_segsym(opj_t1_t *t1, OPJ_INT32 cblksty)
1334 {
1335     if (cblksty & J2K_CCP_CBLKSTY_SEGSYM) {
1336         opj_mqc_t* mqc = &(t1->mqc);
1337         OPJ_UINT32 v, v2;
1338         opj_mqc_setcurctx(mqc, T1_CTXNO_UNI);
1339         opj_mqc_decode(v, mqc);
1340         opj_mqc_decode(v2, mqc);
1341         v = (v << 1) | v2;
1342         opj_mqc_decode(v2, mqc);
1343         v = (v << 1) | v2;
1344         opj_mqc_decode(v2, mqc);
1345         v = (v << 1) | v2;
1346         /*
1347         if (v!=0xa) {
1348             opj_event_msg(t1->cinfo, EVT_WARNING, "Bad segmentation symbol %x\n", v);
1349         }
1350         */
1351     }
1352 }
1353 
opj_t1_dec_clnpass_64x64_novsc(opj_t1_t * t1,OPJ_INT32 bpno)1354 static void opj_t1_dec_clnpass_64x64_novsc(
1355     opj_t1_t *t1,
1356     OPJ_INT32 bpno)
1357 {
1358     opj_t1_dec_clnpass_internal(t1, bpno, OPJ_FALSE, 64, 64, 66);
1359 }
1360 
opj_t1_dec_clnpass_64x64_vsc(opj_t1_t * t1,OPJ_INT32 bpno)1361 static void opj_t1_dec_clnpass_64x64_vsc(
1362     opj_t1_t *t1,
1363     OPJ_INT32 bpno)
1364 {
1365     opj_t1_dec_clnpass_internal(t1, bpno, OPJ_TRUE, 64, 64, 66);
1366 }
1367 
opj_t1_dec_clnpass_generic_novsc(opj_t1_t * t1,OPJ_INT32 bpno)1368 static void opj_t1_dec_clnpass_generic_novsc(
1369     opj_t1_t *t1,
1370     OPJ_INT32 bpno)
1371 {
1372     opj_t1_dec_clnpass_internal(t1, bpno, OPJ_FALSE, t1->w, t1->h,
1373                                 t1->w + 2U);
1374 }
1375 
opj_t1_dec_clnpass_generic_vsc(opj_t1_t * t1,OPJ_INT32 bpno)1376 static void opj_t1_dec_clnpass_generic_vsc(
1377     opj_t1_t *t1,
1378     OPJ_INT32 bpno)
1379 {
1380     opj_t1_dec_clnpass_internal(t1, bpno, OPJ_TRUE, t1->w, t1->h,
1381                                 t1->w + 2U);
1382 }
1383 
opj_t1_dec_clnpass(opj_t1_t * t1,OPJ_INT32 bpno,OPJ_INT32 cblksty)1384 static void opj_t1_dec_clnpass(
1385     opj_t1_t *t1,
1386     OPJ_INT32 bpno,
1387     OPJ_INT32 cblksty)
1388 {
1389     if (t1->w == 64 && t1->h == 64) {
1390         if (cblksty & J2K_CCP_CBLKSTY_VSC) {
1391             opj_t1_dec_clnpass_64x64_vsc(t1, bpno);
1392         } else {
1393             opj_t1_dec_clnpass_64x64_novsc(t1, bpno);
1394         }
1395     } else {
1396         if (cblksty & J2K_CCP_CBLKSTY_VSC) {
1397             opj_t1_dec_clnpass_generic_vsc(t1, bpno);
1398         } else {
1399             opj_t1_dec_clnpass_generic_novsc(t1, bpno);
1400         }
1401     }
1402     opj_t1_dec_clnpass_check_segsym(t1, cblksty);
1403 }
1404 
1405 
1406 /** mod fixed_quality */
opj_t1_getwmsedec(OPJ_INT32 nmsedec,OPJ_UINT32 compno,OPJ_UINT32 level,OPJ_UINT32 orient,OPJ_INT32 bpno,OPJ_UINT32 qmfbid,OPJ_FLOAT64 stepsize,OPJ_UINT32 numcomps,const OPJ_FLOAT64 * mct_norms,OPJ_UINT32 mct_numcomps)1407 static OPJ_FLOAT64 opj_t1_getwmsedec(
1408     OPJ_INT32 nmsedec,
1409     OPJ_UINT32 compno,
1410     OPJ_UINT32 level,
1411     OPJ_UINT32 orient,
1412     OPJ_INT32 bpno,
1413     OPJ_UINT32 qmfbid,
1414     OPJ_FLOAT64 stepsize,
1415     OPJ_UINT32 numcomps,
1416     const OPJ_FLOAT64 * mct_norms,
1417     OPJ_UINT32 mct_numcomps)
1418 {
1419     OPJ_FLOAT64 w1 = 1, w2, wmsedec;
1420     OPJ_ARG_NOT_USED(numcomps);
1421 
1422     if (mct_norms && (compno < mct_numcomps)) {
1423         w1 = mct_norms[compno];
1424     }
1425 
1426     if (qmfbid == 1) {
1427         w2 = opj_dwt_getnorm(level, orient);
1428     } else {    /* if (qmfbid == 0) */
1429         w2 = opj_dwt_getnorm_real(level, orient);
1430     }
1431 
1432     wmsedec = w1 * w2 * stepsize * (1 << bpno);
1433     wmsedec *= wmsedec * nmsedec / 8192.0;
1434 
1435     return wmsedec;
1436 }
1437 
opj_t1_allocate_buffers(opj_t1_t * t1,OPJ_UINT32 w,OPJ_UINT32 h)1438 static OPJ_BOOL opj_t1_allocate_buffers(
1439     opj_t1_t *t1,
1440     OPJ_UINT32 w,
1441     OPJ_UINT32 h)
1442 {
1443     OPJ_UINT32 flagssize;
1444     OPJ_UINT32 flags_stride;
1445 
1446     /* No risk of overflow. Prior checks ensure those assert are met */
1447     /* They are per the specification */
1448     assert(w <= 1024);
1449     assert(h <= 1024);
1450     assert(w * h <= 4096);
1451 
1452     /* encoder uses tile buffer, so no need to allocate */
1453     if (!t1->encoder) {
1454         OPJ_UINT32 datasize = w * h;
1455 
1456         if (datasize > t1->datasize) {
1457             opj_aligned_free(t1->data);
1458             t1->data = (OPJ_INT32*) opj_aligned_malloc(datasize * sizeof(OPJ_INT32));
1459             if (!t1->data) {
1460                 /* FIXME event manager error callback */
1461                 return OPJ_FALSE;
1462             }
1463             t1->datasize = datasize;
1464         }
1465         /* memset first arg is declared to never be null by gcc */
1466         if (t1->data != NULL) {
1467             memset(t1->data, 0, datasize * sizeof(OPJ_INT32));
1468         }
1469     }
1470 
1471     flags_stride = w + 2U; /* can't be 0U */
1472 
1473     flagssize = (h + 3U) / 4U + 2U;
1474 
1475     flagssize *= flags_stride;
1476     {
1477         opj_flag_t* p;
1478         OPJ_UINT32 x;
1479         OPJ_UINT32 flags_height = (h + 3U) / 4U;
1480 
1481         if (flagssize > t1->flagssize) {
1482 
1483             opj_aligned_free(t1->flags);
1484             t1->flags = (opj_flag_t*) opj_aligned_malloc(flagssize * sizeof(
1485                             opj_flag_t));
1486             if (!t1->flags) {
1487                 /* FIXME event manager error callback */
1488                 return OPJ_FALSE;
1489             }
1490         }
1491         t1->flagssize = flagssize;
1492 
1493         memset(t1->flags, 0, flagssize * sizeof(opj_flag_t));
1494 
1495         p = &t1->flags[0];
1496         for (x = 0; x < flags_stride; ++x) {
1497             /* magic value to hopefully stop any passes being interested in this entry */
1498             *p++ = (T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3);
1499         }
1500 
1501         p = &t1->flags[((flags_height + 1) * flags_stride)];
1502         for (x = 0; x < flags_stride; ++x) {
1503             /* magic value to hopefully stop any passes being interested in this entry */
1504             *p++ = (T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3);
1505         }
1506 
1507         if (h % 4) {
1508             OPJ_UINT32 v = 0;
1509             p = &t1->flags[((flags_height) * flags_stride)];
1510             if (h % 4 == 1) {
1511                 v |= T1_PI_1 | T1_PI_2 | T1_PI_3;
1512             } else if (h % 4 == 2) {
1513                 v |= T1_PI_2 | T1_PI_3;
1514             } else if (h % 4 == 3) {
1515                 v |= T1_PI_3;
1516             }
1517             for (x = 0; x < flags_stride; ++x) {
1518                 *p++ = v;
1519             }
1520         }
1521     }
1522 
1523     t1->w = w;
1524     t1->h = h;
1525 
1526     return OPJ_TRUE;
1527 }
1528 
1529 /* ----------------------------------------------------------------------- */
1530 
1531 /* ----------------------------------------------------------------------- */
1532 /**
1533  * Creates a new Tier 1 handle
1534  * and initializes the look-up tables of the Tier-1 coder/decoder
1535  * @return a new T1 handle if successful, returns NULL otherwise
1536 */
opj_t1_create(OPJ_BOOL isEncoder)1537 opj_t1_t* opj_t1_create(OPJ_BOOL isEncoder)
1538 {
1539     opj_t1_t *l_t1 = 00;
1540 
1541     l_t1 = (opj_t1_t*) opj_calloc(1, sizeof(opj_t1_t));
1542     if (!l_t1) {
1543         return 00;
1544     }
1545 
1546     l_t1->encoder = isEncoder;
1547 
1548     return l_t1;
1549 }
1550 
1551 
1552 /**
1553  * Destroys a previously created T1 handle
1554  *
1555  * @param p_t1 Tier 1 handle to destroy
1556 */
opj_t1_destroy(opj_t1_t * p_t1)1557 void opj_t1_destroy(opj_t1_t *p_t1)
1558 {
1559     if (! p_t1) {
1560         return;
1561     }
1562 
1563     /* encoder uses tile buffer, so no need to free */
1564     if (!p_t1->encoder && p_t1->data) {
1565         opj_aligned_free(p_t1->data);
1566         p_t1->data = 00;
1567     }
1568 
1569     if (p_t1->flags) {
1570         opj_aligned_free(p_t1->flags);
1571         p_t1->flags = 00;
1572     }
1573 
1574     opj_free(p_t1->cblkdatabuffer);
1575 
1576     opj_free(p_t1);
1577 }
1578 
1579 typedef struct {
1580     OPJ_BOOL whole_tile_decoding;
1581     OPJ_UINT32 resno;
1582     opj_tcd_cblk_dec_t* cblk;
1583     opj_tcd_band_t* band;
1584     opj_tcd_tilecomp_t* tilec;
1585     opj_tccp_t* tccp;
1586     OPJ_BOOL mustuse_cblkdatabuffer;
1587     volatile OPJ_BOOL* pret;
1588     opj_event_mgr_t *p_manager;
1589     opj_mutex_t* p_manager_mutex;
1590     OPJ_BOOL check_pterm;
1591 } opj_t1_cblk_decode_processing_job_t;
1592 
opj_t1_destroy_wrapper(void * t1)1593 static void opj_t1_destroy_wrapper(void* t1)
1594 {
1595     opj_t1_destroy((opj_t1_t*) t1);
1596 }
1597 
opj_t1_clbl_decode_processor(void * user_data,opj_tls_t * tls)1598 static void opj_t1_clbl_decode_processor(void* user_data, opj_tls_t* tls)
1599 {
1600     opj_tcd_cblk_dec_t* cblk;
1601     opj_tcd_band_t* band;
1602     opj_tcd_tilecomp_t* tilec;
1603     opj_tccp_t* tccp;
1604     OPJ_INT32* OPJ_RESTRICT datap;
1605     OPJ_UINT32 cblk_w, cblk_h;
1606     OPJ_INT32 x, y;
1607     OPJ_UINT32 i, j;
1608     opj_t1_cblk_decode_processing_job_t* job;
1609     opj_t1_t* t1;
1610     OPJ_UINT32 resno;
1611     OPJ_UINT32 tile_w;
1612 
1613     job = (opj_t1_cblk_decode_processing_job_t*) user_data;
1614 
1615     cblk = job->cblk;
1616 
1617     if (!job->whole_tile_decoding) {
1618         cblk_w = (OPJ_UINT32)(cblk->x1 - cblk->x0);
1619         cblk_h = (OPJ_UINT32)(cblk->y1 - cblk->y0);
1620 
1621         cblk->decoded_data = (OPJ_INT32*)opj_aligned_malloc(sizeof(OPJ_INT32) *
1622                              cblk_w * cblk_h);
1623         if (cblk->decoded_data == NULL) {
1624             if (job->p_manager_mutex) {
1625                 opj_mutex_lock(job->p_manager_mutex);
1626             }
1627             opj_event_msg(job->p_manager, EVT_ERROR,
1628                           "Cannot allocate cblk->decoded_data\n");
1629             if (job->p_manager_mutex) {
1630                 opj_mutex_unlock(job->p_manager_mutex);
1631             }
1632             *(job->pret) = OPJ_FALSE;
1633             opj_free(job);
1634             return;
1635         }
1636         /* Zero-init required */
1637         memset(cblk->decoded_data, 0, sizeof(OPJ_INT32) * cblk_w * cblk_h);
1638     } else if (cblk->decoded_data) {
1639         /* Not sure if that code path can happen, but better be */
1640         /* safe than sorry */
1641         opj_aligned_free(cblk->decoded_data);
1642         cblk->decoded_data = NULL;
1643     }
1644 
1645     resno = job->resno;
1646     band = job->band;
1647     tilec = job->tilec;
1648     tccp = job->tccp;
1649     tile_w = (OPJ_UINT32)(tilec->resolutions[tilec->minimum_num_resolutions - 1].x1
1650                           -
1651                           tilec->resolutions[tilec->minimum_num_resolutions - 1].x0);
1652 
1653     if (!*(job->pret)) {
1654         opj_free(job);
1655         return;
1656     }
1657 
1658     t1 = (opj_t1_t*) opj_tls_get(tls, OPJ_TLS_KEY_T1);
1659     if (t1 == NULL) {
1660         t1 = opj_t1_create(OPJ_FALSE);
1661         opj_tls_set(tls, OPJ_TLS_KEY_T1, t1, opj_t1_destroy_wrapper);
1662     }
1663     t1->mustuse_cblkdatabuffer = job->mustuse_cblkdatabuffer;
1664 
1665     if (OPJ_FALSE == opj_t1_decode_cblk(
1666                 t1,
1667                 cblk,
1668                 band->bandno,
1669                 (OPJ_UINT32)tccp->roishift,
1670                 tccp->cblksty,
1671                 job->p_manager,
1672                 job->p_manager_mutex,
1673                 job->check_pterm)) {
1674         *(job->pret) = OPJ_FALSE;
1675         opj_free(job);
1676         return;
1677     }
1678 
1679     x = cblk->x0 - band->x0;
1680     y = cblk->y0 - band->y0;
1681     if (band->bandno & 1) {
1682         opj_tcd_resolution_t* pres = &tilec->resolutions[resno - 1];
1683         x += pres->x1 - pres->x0;
1684     }
1685     if (band->bandno & 2) {
1686         opj_tcd_resolution_t* pres = &tilec->resolutions[resno - 1];
1687         y += pres->y1 - pres->y0;
1688     }
1689 
1690     datap = cblk->decoded_data ? cblk->decoded_data : t1->data;
1691     cblk_w = t1->w;
1692     cblk_h = t1->h;
1693 
1694     if (tccp->roishift) {
1695         if (tccp->roishift >= 31) {
1696             for (j = 0; j < cblk_h; ++j) {
1697                 for (i = 0; i < cblk_w; ++i) {
1698                     datap[(j * cblk_w) + i] = 0;
1699                 }
1700             }
1701         } else {
1702             OPJ_INT32 thresh = 1 << tccp->roishift;
1703             for (j = 0; j < cblk_h; ++j) {
1704                 for (i = 0; i < cblk_w; ++i) {
1705                     OPJ_INT32 val = datap[(j * cblk_w) + i];
1706                     OPJ_INT32 mag = abs(val);
1707                     if (mag >= thresh) {
1708                         mag >>= tccp->roishift;
1709                         datap[(j * cblk_w) + i] = val < 0 ? -mag : mag;
1710                     }
1711                 }
1712             }
1713         }
1714     }
1715 
1716     /* Both can be non NULL if for example decoding a full tile and then */
1717     /* partially a tile. In which case partial decoding should be the */
1718     /* priority */
1719     assert((cblk->decoded_data != NULL) || (tilec->data != NULL));
1720 
1721     if (cblk->decoded_data) {
1722         OPJ_UINT32 cblk_size = cblk_w * cblk_h;
1723         if (tccp->qmfbid == 1) {
1724             for (i = 0; i < cblk_size; ++i) {
1725                 datap[i] /= 2;
1726             }
1727         } else {        /* if (tccp->qmfbid == 0) */
1728             i = 0;
1729 #ifdef __SSE2__
1730             {
1731                 const __m128 xmm_stepsize = _mm_set1_ps(band->stepsize);
1732                 for (; i < (cblk_size & ~15U); i += 16) {
1733                     __m128 xmm0_data = _mm_cvtepi32_ps(_mm_load_si128((__m128i * const)(
1734                                                            datap + 0)));
1735                     __m128 xmm1_data = _mm_cvtepi32_ps(_mm_load_si128((__m128i * const)(
1736                                                            datap + 4)));
1737                     __m128 xmm2_data = _mm_cvtepi32_ps(_mm_load_si128((__m128i * const)(
1738                                                            datap + 8)));
1739                     __m128 xmm3_data = _mm_cvtepi32_ps(_mm_load_si128((__m128i * const)(
1740                                                            datap + 12)));
1741                     _mm_store_ps((float*)(datap +  0), _mm_mul_ps(xmm0_data, xmm_stepsize));
1742                     _mm_store_ps((float*)(datap +  4), _mm_mul_ps(xmm1_data, xmm_stepsize));
1743                     _mm_store_ps((float*)(datap +  8), _mm_mul_ps(xmm2_data, xmm_stepsize));
1744                     _mm_store_ps((float*)(datap + 12), _mm_mul_ps(xmm3_data, xmm_stepsize));
1745                     datap += 16;
1746                 }
1747             }
1748 #endif
1749             for (; i < cblk_size; ++i) {
1750                 OPJ_FLOAT32 tmp = ((OPJ_FLOAT32)(*datap)) * band->stepsize;
1751                 memcpy(datap, &tmp, sizeof(tmp));
1752                 datap++;
1753             }
1754         }
1755     } else if (tccp->qmfbid == 1) {
1756         OPJ_INT32* OPJ_RESTRICT tiledp = &tilec->data[(OPJ_SIZE_T)y * tile_w +
1757                                                        (OPJ_SIZE_T)x];
1758         for (j = 0; j < cblk_h; ++j) {
1759             i = 0;
1760             for (; i < (cblk_w & ~(OPJ_UINT32)3U); i += 4U) {
1761                 OPJ_INT32 tmp0 = datap[(j * cblk_w) + i + 0U];
1762                 OPJ_INT32 tmp1 = datap[(j * cblk_w) + i + 1U];
1763                 OPJ_INT32 tmp2 = datap[(j * cblk_w) + i + 2U];
1764                 OPJ_INT32 tmp3 = datap[(j * cblk_w) + i + 3U];
1765                 ((OPJ_INT32*)tiledp)[(j * (OPJ_SIZE_T)tile_w) + i + 0U] = tmp0 / 2;
1766                 ((OPJ_INT32*)tiledp)[(j * (OPJ_SIZE_T)tile_w) + i + 1U] = tmp1 / 2;
1767                 ((OPJ_INT32*)tiledp)[(j * (OPJ_SIZE_T)tile_w) + i + 2U] = tmp2 / 2;
1768                 ((OPJ_INT32*)tiledp)[(j * (OPJ_SIZE_T)tile_w) + i + 3U] = tmp3 / 2;
1769             }
1770             for (; i < cblk_w; ++i) {
1771                 OPJ_INT32 tmp = datap[(j * cblk_w) + i];
1772                 ((OPJ_INT32*)tiledp)[(j * (OPJ_SIZE_T)tile_w) + i] = tmp / 2;
1773             }
1774         }
1775     } else {        /* if (tccp->qmfbid == 0) */
1776         OPJ_FLOAT32* OPJ_RESTRICT tiledp = (OPJ_FLOAT32*) &tilec->data[(OPJ_SIZE_T)y *
1777                                                          tile_w + (OPJ_SIZE_T)x];
1778         for (j = 0; j < cblk_h; ++j) {
1779             OPJ_FLOAT32* OPJ_RESTRICT tiledp2 = tiledp;
1780             for (i = 0; i < cblk_w; ++i) {
1781                 OPJ_FLOAT32 tmp = (OPJ_FLOAT32) * datap * band->stepsize;
1782                 *tiledp2 = tmp;
1783                 datap++;
1784                 tiledp2++;
1785             }
1786             tiledp += tile_w;
1787         }
1788     }
1789 
1790     opj_free(job);
1791 }
1792 
1793 
opj_t1_decode_cblks(opj_tcd_t * tcd,volatile OPJ_BOOL * pret,opj_tcd_tilecomp_t * tilec,opj_tccp_t * tccp,opj_event_mgr_t * p_manager,opj_mutex_t * p_manager_mutex,OPJ_BOOL check_pterm)1794 void opj_t1_decode_cblks(opj_tcd_t* tcd,
1795                          volatile OPJ_BOOL* pret,
1796                          opj_tcd_tilecomp_t* tilec,
1797                          opj_tccp_t* tccp,
1798                          opj_event_mgr_t *p_manager,
1799                          opj_mutex_t* p_manager_mutex,
1800                          OPJ_BOOL check_pterm
1801                         )
1802 {
1803     opj_thread_pool_t* tp = tcd->thread_pool;
1804     OPJ_UINT32 resno, bandno, precno, cblkno;
1805 
1806 #ifdef DEBUG_VERBOSE
1807     OPJ_UINT32 codeblocks_decoded = 0;
1808     printf("Enter opj_t1_decode_cblks()\n");
1809 #endif
1810 
1811     for (resno = 0; resno < tilec->minimum_num_resolutions; ++resno) {
1812         opj_tcd_resolution_t* res = &tilec->resolutions[resno];
1813 
1814         for (bandno = 0; bandno < res->numbands; ++bandno) {
1815             opj_tcd_band_t* OPJ_RESTRICT band = &res->bands[bandno];
1816 
1817             for (precno = 0; precno < res->pw * res->ph; ++precno) {
1818                 opj_tcd_precinct_t* precinct = &band->precincts[precno];
1819 
1820                 if (!opj_tcd_is_subband_area_of_interest(tcd,
1821                         tilec->compno,
1822                         resno,
1823                         band->bandno,
1824                         (OPJ_UINT32)precinct->x0,
1825                         (OPJ_UINT32)precinct->y0,
1826                         (OPJ_UINT32)precinct->x1,
1827                         (OPJ_UINT32)precinct->y1)) {
1828                     for (cblkno = 0; cblkno < precinct->cw * precinct->ch; ++cblkno) {
1829                         opj_tcd_cblk_dec_t* cblk = &precinct->cblks.dec[cblkno];
1830                         if (cblk->decoded_data) {
1831 #ifdef DEBUG_VERBOSE
1832                             printf("Discarding codeblock %d,%d at resno=%d, bandno=%d\n",
1833                                    cblk->x0, cblk->y0, resno, bandno);
1834 #endif
1835                             opj_aligned_free(cblk->decoded_data);
1836                             cblk->decoded_data = NULL;
1837                         }
1838                     }
1839                     continue;
1840                 }
1841 
1842                 for (cblkno = 0; cblkno < precinct->cw * precinct->ch; ++cblkno) {
1843                     opj_tcd_cblk_dec_t* cblk = &precinct->cblks.dec[cblkno];
1844                     opj_t1_cblk_decode_processing_job_t* job;
1845 
1846                     if (!opj_tcd_is_subband_area_of_interest(tcd,
1847                             tilec->compno,
1848                             resno,
1849                             band->bandno,
1850                             (OPJ_UINT32)cblk->x0,
1851                             (OPJ_UINT32)cblk->y0,
1852                             (OPJ_UINT32)cblk->x1,
1853                             (OPJ_UINT32)cblk->y1)) {
1854                         if (cblk->decoded_data) {
1855 #ifdef DEBUG_VERBOSE
1856                             printf("Discarding codeblock %d,%d at resno=%d, bandno=%d\n",
1857                                    cblk->x0, cblk->y0, resno, bandno);
1858 #endif
1859                             opj_aligned_free(cblk->decoded_data);
1860                             cblk->decoded_data = NULL;
1861                         }
1862                         continue;
1863                     }
1864 
1865                     if (!tcd->whole_tile_decoding) {
1866                         OPJ_UINT32 cblk_w = (OPJ_UINT32)(cblk->x1 - cblk->x0);
1867                         OPJ_UINT32 cblk_h = (OPJ_UINT32)(cblk->y1 - cblk->y0);
1868                         if (cblk->decoded_data != NULL) {
1869 #ifdef DEBUG_VERBOSE
1870                             printf("Reusing codeblock %d,%d at resno=%d, bandno=%d\n",
1871                                    cblk->x0, cblk->y0, resno, bandno);
1872 #endif
1873                             continue;
1874                         }
1875                         if (cblk_w == 0 || cblk_h == 0) {
1876                             continue;
1877                         }
1878 #ifdef DEBUG_VERBOSE
1879                         printf("Decoding codeblock %d,%d at resno=%d, bandno=%d\n",
1880                                cblk->x0, cblk->y0, resno, bandno);
1881 #endif
1882                     }
1883 
1884                     job = (opj_t1_cblk_decode_processing_job_t*) opj_calloc(1,
1885                             sizeof(opj_t1_cblk_decode_processing_job_t));
1886                     if (!job) {
1887                         *pret = OPJ_FALSE;
1888                         return;
1889                     }
1890                     job->whole_tile_decoding = tcd->whole_tile_decoding;
1891                     job->resno = resno;
1892                     job->cblk = cblk;
1893                     job->band = band;
1894                     job->tilec = tilec;
1895                     job->tccp = tccp;
1896                     job->pret = pret;
1897                     job->p_manager_mutex = p_manager_mutex;
1898                     job->p_manager = p_manager;
1899                     job->check_pterm = check_pterm;
1900                     job->mustuse_cblkdatabuffer = opj_thread_pool_get_thread_count(tp) > 1;
1901                     opj_thread_pool_submit_job(tp, opj_t1_clbl_decode_processor, job);
1902 #ifdef DEBUG_VERBOSE
1903                     codeblocks_decoded ++;
1904 #endif
1905                     if (!(*pret)) {
1906                         return;
1907                     }
1908                 } /* cblkno */
1909             } /* precno */
1910         } /* bandno */
1911     } /* resno */
1912 
1913 #ifdef DEBUG_VERBOSE
1914     printf("Leave opj_t1_decode_cblks(). Number decoded: %d\n", codeblocks_decoded);
1915 #endif
1916     return;
1917 }
1918 
1919 
opj_t1_decode_cblk(opj_t1_t * t1,opj_tcd_cblk_dec_t * cblk,OPJ_UINT32 orient,OPJ_UINT32 roishift,OPJ_UINT32 cblksty,opj_event_mgr_t * p_manager,opj_mutex_t * p_manager_mutex,OPJ_BOOL check_pterm)1920 static OPJ_BOOL opj_t1_decode_cblk(opj_t1_t *t1,
1921                                    opj_tcd_cblk_dec_t* cblk,
1922                                    OPJ_UINT32 orient,
1923                                    OPJ_UINT32 roishift,
1924                                    OPJ_UINT32 cblksty,
1925                                    opj_event_mgr_t *p_manager,
1926                                    opj_mutex_t* p_manager_mutex,
1927                                    OPJ_BOOL check_pterm)
1928 {
1929     opj_mqc_t *mqc = &(t1->mqc);   /* MQC component */
1930 
1931     OPJ_INT32 bpno_plus_one;
1932     OPJ_UINT32 passtype;
1933     OPJ_UINT32 segno, passno;
1934     OPJ_BYTE* cblkdata = NULL;
1935     OPJ_UINT32 cblkdataindex = 0;
1936     OPJ_BYTE type = T1_TYPE_MQ; /* BYPASS mode */
1937     OPJ_INT32* original_t1_data = NULL;
1938 
1939     mqc->lut_ctxno_zc_orient = lut_ctxno_zc + (orient << 9);
1940 
1941     if (!opj_t1_allocate_buffers(
1942                 t1,
1943                 (OPJ_UINT32)(cblk->x1 - cblk->x0),
1944                 (OPJ_UINT32)(cblk->y1 - cblk->y0))) {
1945         return OPJ_FALSE;
1946     }
1947 
1948     bpno_plus_one = (OPJ_INT32)(roishift + cblk->numbps);
1949     if (bpno_plus_one >= 31) {
1950         if (p_manager_mutex) {
1951             opj_mutex_lock(p_manager_mutex);
1952         }
1953         opj_event_msg(p_manager, EVT_WARNING,
1954                       "opj_t1_decode_cblk(): unsupported bpno_plus_one = %d >= 31\n",
1955                       bpno_plus_one);
1956         if (p_manager_mutex) {
1957             opj_mutex_unlock(p_manager_mutex);
1958         }
1959         return OPJ_FALSE;
1960     }
1961     passtype = 2;
1962 
1963     opj_mqc_resetstates(mqc);
1964     opj_mqc_setstate(mqc, T1_CTXNO_UNI, 0, 46);
1965     opj_mqc_setstate(mqc, T1_CTXNO_AGG, 0, 3);
1966     opj_mqc_setstate(mqc, T1_CTXNO_ZC, 0, 4);
1967 
1968     /* Even if we have a single chunk, in multi-threaded decoding */
1969     /* the insertion of our synthetic marker might potentially override */
1970     /* valid codestream of other codeblocks decoded in parallel. */
1971     if (cblk->numchunks > 1 || t1->mustuse_cblkdatabuffer) {
1972         OPJ_UINT32 i;
1973         OPJ_UINT32 cblk_len;
1974 
1975         /* Compute whole codeblock length from chunk lengths */
1976         cblk_len = 0;
1977         for (i = 0; i < cblk->numchunks; i++) {
1978             cblk_len += cblk->chunks[i].len;
1979         }
1980 
1981         /* Allocate temporary memory if needed */
1982         if (cblk_len + OPJ_COMMON_CBLK_DATA_EXTRA > t1->cblkdatabuffersize) {
1983             cblkdata = (OPJ_BYTE*)opj_realloc(t1->cblkdatabuffer,
1984                                               cblk_len + OPJ_COMMON_CBLK_DATA_EXTRA);
1985             if (cblkdata == NULL) {
1986                 return OPJ_FALSE;
1987             }
1988             t1->cblkdatabuffer = cblkdata;
1989             memset(t1->cblkdatabuffer + cblk_len, 0, OPJ_COMMON_CBLK_DATA_EXTRA);
1990             t1->cblkdatabuffersize = cblk_len + OPJ_COMMON_CBLK_DATA_EXTRA;
1991         }
1992 
1993         /* Concatenate all chunks */
1994         cblkdata = t1->cblkdatabuffer;
1995         cblk_len = 0;
1996         for (i = 0; i < cblk->numchunks; i++) {
1997             memcpy(cblkdata + cblk_len, cblk->chunks[i].data, cblk->chunks[i].len);
1998             cblk_len += cblk->chunks[i].len;
1999         }
2000     } else if (cblk->numchunks == 1) {
2001         cblkdata = cblk->chunks[0].data;
2002     } else {
2003         /* Not sure if that can happen in practice, but avoid Coverity to */
2004         /* think we will dereference a null cblkdta pointer */
2005         return OPJ_TRUE;
2006     }
2007 
2008     /* For subtile decoding, directly decode in the decoded_data buffer of */
2009     /* the code-block. Hack t1->data to point to it, and restore it later */
2010     if (cblk->decoded_data) {
2011         original_t1_data = t1->data;
2012         t1->data = cblk->decoded_data;
2013     }
2014 
2015     for (segno = 0; segno < cblk->real_num_segs; ++segno) {
2016         opj_tcd_seg_t *seg = &cblk->segs[segno];
2017 
2018         /* BYPASS mode */
2019         type = ((bpno_plus_one <= ((OPJ_INT32)(cblk->numbps)) - 4) && (passtype < 2) &&
2020                 (cblksty & J2K_CCP_CBLKSTY_LAZY)) ? T1_TYPE_RAW : T1_TYPE_MQ;
2021 
2022         if (type == T1_TYPE_RAW) {
2023             opj_mqc_raw_init_dec(mqc, cblkdata + cblkdataindex, seg->len,
2024                                  OPJ_COMMON_CBLK_DATA_EXTRA);
2025         } else {
2026             opj_mqc_init_dec(mqc, cblkdata + cblkdataindex, seg->len,
2027                              OPJ_COMMON_CBLK_DATA_EXTRA);
2028         }
2029         cblkdataindex += seg->len;
2030 
2031         for (passno = 0; (passno < seg->real_num_passes) &&
2032                 (bpno_plus_one >= 1); ++passno) {
2033             switch (passtype) {
2034             case 0:
2035                 if (type == T1_TYPE_RAW) {
2036                     opj_t1_dec_sigpass_raw(t1, bpno_plus_one, (OPJ_INT32)cblksty);
2037                 } else {
2038                     opj_t1_dec_sigpass_mqc(t1, bpno_plus_one, (OPJ_INT32)cblksty);
2039                 }
2040                 break;
2041             case 1:
2042                 if (type == T1_TYPE_RAW) {
2043                     opj_t1_dec_refpass_raw(t1, bpno_plus_one);
2044                 } else {
2045                     opj_t1_dec_refpass_mqc(t1, bpno_plus_one);
2046                 }
2047                 break;
2048             case 2:
2049                 opj_t1_dec_clnpass(t1, bpno_plus_one, (OPJ_INT32)cblksty);
2050                 break;
2051             }
2052 
2053             if ((cblksty & J2K_CCP_CBLKSTY_RESET) && type == T1_TYPE_MQ) {
2054                 opj_mqc_resetstates(mqc);
2055                 opj_mqc_setstate(mqc, T1_CTXNO_UNI, 0, 46);
2056                 opj_mqc_setstate(mqc, T1_CTXNO_AGG, 0, 3);
2057                 opj_mqc_setstate(mqc, T1_CTXNO_ZC, 0, 4);
2058             }
2059             if (++passtype == 3) {
2060                 passtype = 0;
2061                 bpno_plus_one--;
2062             }
2063         }
2064 
2065         opq_mqc_finish_dec(mqc);
2066     }
2067 
2068     if (check_pterm) {
2069         if (mqc->bp + 2 < mqc->end) {
2070             if (p_manager_mutex) {
2071                 opj_mutex_lock(p_manager_mutex);
2072             }
2073             opj_event_msg(p_manager, EVT_WARNING,
2074                           "PTERM check failure: %d remaining bytes in code block (%d used / %d)\n",
2075                           (int)(mqc->end - mqc->bp) - 2,
2076                           (int)(mqc->bp - mqc->start),
2077                           (int)(mqc->end - mqc->start));
2078             if (p_manager_mutex) {
2079                 opj_mutex_unlock(p_manager_mutex);
2080             }
2081         } else if (mqc->end_of_byte_stream_counter > 2) {
2082             if (p_manager_mutex) {
2083                 opj_mutex_lock(p_manager_mutex);
2084             }
2085             opj_event_msg(p_manager, EVT_WARNING,
2086                           "PTERM check failure: %d synthetized 0xFF markers read\n",
2087                           mqc->end_of_byte_stream_counter);
2088             if (p_manager_mutex) {
2089                 opj_mutex_unlock(p_manager_mutex);
2090             }
2091         }
2092     }
2093 
2094     /* Restore original t1->data is needed */
2095     if (cblk->decoded_data) {
2096         t1->data = original_t1_data;
2097     }
2098 
2099     return OPJ_TRUE;
2100 }
2101 
2102 
2103 
2104 
opj_t1_encode_cblks(opj_t1_t * t1,opj_tcd_tile_t * tile,opj_tcp_t * tcp,const OPJ_FLOAT64 * mct_norms,OPJ_UINT32 mct_numcomps)2105 OPJ_BOOL opj_t1_encode_cblks(opj_t1_t *t1,
2106                              opj_tcd_tile_t *tile,
2107                              opj_tcp_t *tcp,
2108                              const OPJ_FLOAT64 * mct_norms,
2109                              OPJ_UINT32 mct_numcomps
2110                             )
2111 {
2112     OPJ_UINT32 compno, resno, bandno, precno, cblkno;
2113 
2114     tile->distotile = 0;        /* fixed_quality */
2115 
2116     for (compno = 0; compno < tile->numcomps; ++compno) {
2117         opj_tcd_tilecomp_t* tilec = &tile->comps[compno];
2118         opj_tccp_t* tccp = &tcp->tccps[compno];
2119         OPJ_UINT32 tile_w = (OPJ_UINT32)(tilec->x1 - tilec->x0);
2120 
2121         for (resno = 0; resno < tilec->numresolutions; ++resno) {
2122             opj_tcd_resolution_t *res = &tilec->resolutions[resno];
2123 
2124             for (bandno = 0; bandno < res->numbands; ++bandno) {
2125                 opj_tcd_band_t* OPJ_RESTRICT band = &res->bands[bandno];
2126                 OPJ_INT32 bandconst;
2127 
2128                 /* Skip empty bands */
2129                 if (opj_tcd_is_band_empty(band)) {
2130                     continue;
2131                 }
2132 
2133                 bandconst = 8192 * 8192 / ((OPJ_INT32) floor(band->stepsize * 8192));
2134                 for (precno = 0; precno < res->pw * res->ph; ++precno) {
2135                     opj_tcd_precinct_t *prc = &band->precincts[precno];
2136 
2137                     for (cblkno = 0; cblkno < prc->cw * prc->ch; ++cblkno) {
2138                         opj_tcd_cblk_enc_t* cblk = &prc->cblks.enc[cblkno];
2139                         OPJ_INT32* OPJ_RESTRICT tiledp;
2140                         OPJ_UINT32 cblk_w;
2141                         OPJ_UINT32 cblk_h;
2142                         OPJ_UINT32 i, j, tileLineAdvance;
2143                         OPJ_SIZE_T tileIndex = 0;
2144 
2145                         OPJ_INT32 x = cblk->x0 - band->x0;
2146                         OPJ_INT32 y = cblk->y0 - band->y0;
2147                         if (band->bandno & 1) {
2148                             opj_tcd_resolution_t *pres = &tilec->resolutions[resno - 1];
2149                             x += pres->x1 - pres->x0;
2150                         }
2151                         if (band->bandno & 2) {
2152                             opj_tcd_resolution_t *pres = &tilec->resolutions[resno - 1];
2153                             y += pres->y1 - pres->y0;
2154                         }
2155 
2156                         if (!opj_t1_allocate_buffers(
2157                                     t1,
2158                                     (OPJ_UINT32)(cblk->x1 - cblk->x0),
2159                                     (OPJ_UINT32)(cblk->y1 - cblk->y0))) {
2160                             return OPJ_FALSE;
2161                         }
2162 
2163                         cblk_w = t1->w;
2164                         cblk_h = t1->h;
2165                         tileLineAdvance = tile_w - cblk_w;
2166 
2167                         tiledp = &tilec->data[(OPJ_SIZE_T)y * tile_w + (OPJ_SIZE_T)x];
2168                         t1->data = tiledp;
2169                         t1->data_stride = tile_w;
2170                         if (tccp->qmfbid == 1) {
2171                             /* Do multiplication on unsigned type, even if the
2172                              * underlying type is signed, to avoid potential
2173                              * int overflow on large value (the output will be
2174                              * incorrect in such situation, but whatever...)
2175                              * This assumes complement-to-2 signed integer
2176                              * representation
2177                              * Fixes https://github.com/uclouvain/openjpeg/issues/1053
2178                              */
2179                             OPJ_UINT32* OPJ_RESTRICT tiledp_u = (OPJ_UINT32*) tiledp;
2180                             for (j = 0; j < cblk_h; ++j) {
2181                                 for (i = 0; i < cblk_w; ++i) {
2182                                     tiledp_u[tileIndex] <<= T1_NMSEDEC_FRACBITS;
2183                                     tileIndex++;
2184                                 }
2185                                 tileIndex += tileLineAdvance;
2186                             }
2187                         } else {        /* if (tccp->qmfbid == 0) */
2188                             for (j = 0; j < cblk_h; ++j) {
2189                                 for (i = 0; i < cblk_w; ++i) {
2190                                     OPJ_INT32 tmp = tiledp[tileIndex];
2191                                     tiledp[tileIndex] =
2192                                         opj_int_fix_mul_t1(
2193                                             tmp,
2194                                             bandconst);
2195                                     tileIndex++;
2196                                 }
2197                                 tileIndex += tileLineAdvance;
2198                             }
2199                         }
2200 
2201                         opj_t1_encode_cblk(
2202                             t1,
2203                             cblk,
2204                             band->bandno,
2205                             compno,
2206                             tilec->numresolutions - 1 - resno,
2207                             tccp->qmfbid,
2208                             band->stepsize,
2209                             tccp->cblksty,
2210                             tile->numcomps,
2211                             tile,
2212                             mct_norms,
2213                             mct_numcomps);
2214 
2215                     } /* cblkno */
2216                 } /* precno */
2217             } /* bandno */
2218         } /* resno  */
2219     } /* compno  */
2220     return OPJ_TRUE;
2221 }
2222 
2223 /* Returns whether the pass (bpno, passtype) is terminated */
opj_t1_enc_is_term_pass(opj_tcd_cblk_enc_t * cblk,OPJ_UINT32 cblksty,OPJ_INT32 bpno,OPJ_UINT32 passtype)2224 static int opj_t1_enc_is_term_pass(opj_tcd_cblk_enc_t* cblk,
2225                                    OPJ_UINT32 cblksty,
2226                                    OPJ_INT32 bpno,
2227                                    OPJ_UINT32 passtype)
2228 {
2229     /* Is it the last cleanup pass ? */
2230     if (passtype == 2 && bpno == 0) {
2231         return OPJ_TRUE;
2232     }
2233 
2234     if (cblksty & J2K_CCP_CBLKSTY_TERMALL) {
2235         return OPJ_TRUE;
2236     }
2237 
2238     if ((cblksty & J2K_CCP_CBLKSTY_LAZY)) {
2239         /* For bypass arithmetic bypass, terminate the 4th cleanup pass */
2240         if ((bpno == ((OPJ_INT32)cblk->numbps - 4)) && (passtype == 2)) {
2241             return OPJ_TRUE;
2242         }
2243         /* and beyond terminate all the magnitude refinement passes (in raw) */
2244         /* and cleanup passes (in MQC) */
2245         if ((bpno < ((OPJ_INT32)(cblk->numbps) - 4)) && (passtype > 0)) {
2246             return OPJ_TRUE;
2247         }
2248     }
2249 
2250     return OPJ_FALSE;
2251 }
2252 
2253 
2254 /** mod fixed_quality */
opj_t1_encode_cblk(opj_t1_t * t1,opj_tcd_cblk_enc_t * cblk,OPJ_UINT32 orient,OPJ_UINT32 compno,OPJ_UINT32 level,OPJ_UINT32 qmfbid,OPJ_FLOAT64 stepsize,OPJ_UINT32 cblksty,OPJ_UINT32 numcomps,opj_tcd_tile_t * tile,const OPJ_FLOAT64 * mct_norms,OPJ_UINT32 mct_numcomps)2255 static void opj_t1_encode_cblk(opj_t1_t *t1,
2256                                opj_tcd_cblk_enc_t* cblk,
2257                                OPJ_UINT32 orient,
2258                                OPJ_UINT32 compno,
2259                                OPJ_UINT32 level,
2260                                OPJ_UINT32 qmfbid,
2261                                OPJ_FLOAT64 stepsize,
2262                                OPJ_UINT32 cblksty,
2263                                OPJ_UINT32 numcomps,
2264                                opj_tcd_tile_t * tile,
2265                                const OPJ_FLOAT64 * mct_norms,
2266                                OPJ_UINT32 mct_numcomps)
2267 {
2268     OPJ_FLOAT64 cumwmsedec = 0.0;
2269 
2270     opj_mqc_t *mqc = &(t1->mqc);   /* MQC component */
2271 
2272     OPJ_UINT32 passno;
2273     OPJ_INT32 bpno;
2274     OPJ_UINT32 passtype;
2275     OPJ_INT32 nmsedec = 0;
2276     OPJ_INT32 max;
2277     OPJ_UINT32 i, j;
2278     OPJ_BYTE type = T1_TYPE_MQ;
2279     OPJ_FLOAT64 tempwmsedec;
2280 
2281 #ifdef EXTRA_DEBUG
2282     printf("encode_cblk(x=%d,y=%d,x1=%d,y1=%d,orient=%d,compno=%d,level=%d\n",
2283            cblk->x0, cblk->y0, cblk->x1, cblk->y1, orient, compno, level);
2284 #endif
2285 
2286     mqc->lut_ctxno_zc_orient = lut_ctxno_zc + (orient << 9);
2287 
2288     max = 0;
2289     for (i = 0; i < t1->w; ++i) {
2290         for (j = 0; j < t1->h; ++j) {
2291             OPJ_INT32 tmp = abs(t1->data[i + j * t1->data_stride]);
2292             max = opj_int_max(max, tmp);
2293         }
2294     }
2295 
2296     cblk->numbps = max ? (OPJ_UINT32)((opj_int_floorlog2(max) + 1) -
2297                                       T1_NMSEDEC_FRACBITS) : 0;
2298     if (cblk->numbps == 0) {
2299         cblk->totalpasses = 0;
2300         return;
2301     }
2302 
2303     bpno = (OPJ_INT32)(cblk->numbps - 1);
2304     passtype = 2;
2305 
2306     opj_mqc_resetstates(mqc);
2307     opj_mqc_setstate(mqc, T1_CTXNO_UNI, 0, 46);
2308     opj_mqc_setstate(mqc, T1_CTXNO_AGG, 0, 3);
2309     opj_mqc_setstate(mqc, T1_CTXNO_ZC, 0, 4);
2310     opj_mqc_init_enc(mqc, cblk->data);
2311 
2312     for (passno = 0; bpno >= 0; ++passno) {
2313         opj_tcd_pass_t *pass = &cblk->passes[passno];
2314         type = ((bpno < ((OPJ_INT32)(cblk->numbps) - 4)) && (passtype < 2) &&
2315                 (cblksty & J2K_CCP_CBLKSTY_LAZY)) ? T1_TYPE_RAW : T1_TYPE_MQ;
2316 
2317         /* If the previous pass was terminating, we need to reset the encoder */
2318         if (passno > 0 && cblk->passes[passno - 1].term) {
2319             if (type == T1_TYPE_RAW) {
2320                 opj_mqc_bypass_init_enc(mqc);
2321             } else {
2322                 opj_mqc_restart_init_enc(mqc);
2323             }
2324         }
2325 
2326         switch (passtype) {
2327         case 0:
2328             opj_t1_enc_sigpass(t1, bpno, &nmsedec, type, cblksty);
2329             break;
2330         case 1:
2331             opj_t1_enc_refpass(t1, bpno, &nmsedec, type);
2332             break;
2333         case 2:
2334             opj_t1_enc_clnpass(t1, bpno, &nmsedec, cblksty);
2335             /* code switch SEGMARK (i.e. SEGSYM) */
2336             if (cblksty & J2K_CCP_CBLKSTY_SEGSYM) {
2337                 opj_mqc_segmark_enc(mqc);
2338             }
2339             break;
2340         }
2341 
2342         /* fixed_quality */
2343         tempwmsedec = opj_t1_getwmsedec(nmsedec, compno, level, orient, bpno, qmfbid,
2344                                         stepsize, numcomps, mct_norms, mct_numcomps) ;
2345         cumwmsedec += tempwmsedec;
2346         tile->distotile += tempwmsedec;
2347         pass->distortiondec = cumwmsedec;
2348 
2349         if (opj_t1_enc_is_term_pass(cblk, cblksty, bpno, passtype)) {
2350             /* If it is a terminated pass, terminate it */
2351             if (type == T1_TYPE_RAW) {
2352                 opj_mqc_bypass_flush_enc(mqc, cblksty & J2K_CCP_CBLKSTY_PTERM);
2353             } else {
2354                 if (cblksty & J2K_CCP_CBLKSTY_PTERM) {
2355                     opj_mqc_erterm_enc(mqc);
2356                 } else {
2357                     opj_mqc_flush(mqc);
2358                 }
2359             }
2360             pass->term = 1;
2361             pass->rate = opj_mqc_numbytes(mqc);
2362         } else {
2363             /* Non terminated pass */
2364             OPJ_UINT32 rate_extra_bytes;
2365             if (type == T1_TYPE_RAW) {
2366                 rate_extra_bytes = opj_mqc_bypass_get_extra_bytes(
2367                                        mqc, (cblksty & J2K_CCP_CBLKSTY_PTERM));
2368             } else {
2369                 rate_extra_bytes = 3;
2370             }
2371             pass->term = 0;
2372             pass->rate = opj_mqc_numbytes(mqc) + rate_extra_bytes;
2373         }
2374 
2375         if (++passtype == 3) {
2376             passtype = 0;
2377             bpno--;
2378         }
2379 
2380         /* Code-switch "RESET" */
2381         if (cblksty & J2K_CCP_CBLKSTY_RESET) {
2382             opj_mqc_reset_enc(mqc);
2383         }
2384     }
2385 
2386     cblk->totalpasses = passno;
2387 
2388     if (cblk->totalpasses) {
2389         /* Make sure that pass rates are increasing */
2390         OPJ_UINT32 last_pass_rate = opj_mqc_numbytes(mqc);
2391         for (passno = cblk->totalpasses; passno > 0;) {
2392             opj_tcd_pass_t *pass = &cblk->passes[--passno];
2393             if (pass->rate > last_pass_rate) {
2394                 pass->rate = last_pass_rate;
2395             } else {
2396                 last_pass_rate = pass->rate;
2397             }
2398         }
2399     }
2400 
2401     for (passno = 0; passno < cblk->totalpasses; passno++) {
2402         opj_tcd_pass_t *pass = &cblk->passes[passno];
2403 
2404         /* Prevent generation of FF as last data byte of a pass*/
2405         /* For terminating passes, the flushing procedure ensured this already */
2406         assert(pass->rate > 0);
2407         if (cblk->data[pass->rate - 1] == 0xFF) {
2408             pass->rate--;
2409         }
2410         pass->len = pass->rate - (passno == 0 ? 0 : cblk->passes[passno - 1].rate);
2411     }
2412 
2413 #ifdef EXTRA_DEBUG
2414     printf(" len=%d\n", (cblk->totalpasses) ? opj_mqc_numbytes(mqc) : 0);
2415 
2416     /* Check that there not 0xff >=0x90 sequences */
2417     if (cblk->totalpasses) {
2418         OPJ_UINT32 i;
2419         OPJ_UINT32 len = opj_mqc_numbytes(mqc);
2420         for (i = 1; i < len; ++i) {
2421             if (cblk->data[i - 1] == 0xff && cblk->data[i] >= 0x90) {
2422                 printf("0xff %02x at offset %d\n", cblk->data[i], i - 1);
2423                 abort();
2424             }
2425         }
2426     }
2427 #endif
2428 }
2429