mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2588 lines
86 KiB
2588 lines
86 KiB
/* |
|
* The copyright in this software is being made available under the 2-clauses |
|
* BSD License, included below. This software may be subject to other third |
|
* party and contributor rights, including patent rights, and no such rights |
|
* are granted under this license. |
|
* |
|
* Copyright (c) 2002-2014, Universite catholique de Louvain (UCL), Belgium |
|
* Copyright (c) 2002-2014, Professor Benoit Macq |
|
* Copyright (c) 2001-2003, David Janssens |
|
* Copyright (c) 2002-2003, Yannick Verschueren |
|
* Copyright (c) 2003-2007, Francois-Olivier Devaux |
|
* Copyright (c) 2003-2014, Antonin Descampe |
|
* Copyright (c) 2005, Herve Drolon, FreeImage Team |
|
* Copyright (c) 2007, Callum Lerwick <seg@haxxed.com> |
|
* Copyright (c) 2012, Carl Hetherington |
|
* Copyright (c) 2017, IntoPIX SA <support@intopix.com> |
|
* All rights reserved. |
|
* |
|
* Redistribution and use in source and binary forms, with or without |
|
* modification, are permitted provided that the following conditions |
|
* are met: |
|
* 1. Redistributions of source code must retain the above copyright |
|
* notice, this list of conditions and the following disclaimer. |
|
* 2. Redistributions in binary form must reproduce the above copyright |
|
* notice, this list of conditions and the following disclaimer in the |
|
* documentation and/or other materials provided with the distribution. |
|
* |
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS' |
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE |
|
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
|
* POSSIBILITY OF SUCH DAMAGE. |
|
*/ |
|
|
|
#define OPJ_SKIP_POISON |
|
#include "opj_includes.h" |
|
|
|
#ifdef __SSE__ |
|
#include <xmmintrin.h> |
|
#endif |
|
#ifdef __SSE2__ |
|
#include <emmintrin.h> |
|
#endif |
|
|
|
#if defined(__GNUC__) |
|
#pragma GCC poison malloc calloc realloc free |
|
#endif |
|
|
|
#include "t1_luts.h" |
|
|
|
/** @defgroup T1 T1 - Implementation of the tier-1 coding */ |
|
/*@{*/ |
|
|
|
#define T1_FLAGS(x, y) (t1->flags[x + 1 + ((y / 4) + 1) * (t1->w+2)]) |
|
|
|
#define opj_t1_setcurctx(curctx, ctxno) curctx = &(mqc)->ctxs[(OPJ_UINT32)(ctxno)] |
|
|
|
/* Macros to deal with signed integer with just MSB bit set for |
|
* negative values (smr = signed magnitude representation) */ |
|
#define opj_smr_abs(x) (((OPJ_UINT32)(x)) & 0x7FFFFFFFU) |
|
#define opj_smr_sign(x) (((OPJ_UINT32)(x)) >> 31) |
|
#define opj_to_smr(x) ((x) >= 0 ? (OPJ_UINT32)(x) : ((OPJ_UINT32)(-x) | 0x80000000U)) |
|
|
|
|
|
/** @name Local static functions */ |
|
/*@{*/ |
|
|
|
static INLINE OPJ_BYTE opj_t1_getctxno_zc(opj_mqc_t *mqc, OPJ_UINT32 f); |
|
static INLINE OPJ_UINT32 opj_t1_getctxno_mag(OPJ_UINT32 f); |
|
static OPJ_INT16 opj_t1_getnmsedec_sig(OPJ_UINT32 x, OPJ_UINT32 bitpos); |
|
static OPJ_INT16 opj_t1_getnmsedec_ref(OPJ_UINT32 x, OPJ_UINT32 bitpos); |
|
static INLINE void opj_t1_update_flags(opj_flag_t *flagsp, OPJ_UINT32 ci, |
|
OPJ_UINT32 s, OPJ_UINT32 stride, |
|
OPJ_UINT32 vsc); |
|
|
|
|
|
/** |
|
Decode significant pass |
|
*/ |
|
|
|
static INLINE void 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 row); |
|
static INLINE void opj_t1_dec_sigpass_step_mqc( |
|
opj_t1_t *t1, |
|
opj_flag_t *flagsp, |
|
OPJ_INT32 *datap, |
|
OPJ_INT32 oneplushalf, |
|
OPJ_UINT32 row, |
|
OPJ_UINT32 flags_stride, |
|
OPJ_UINT32 vsc); |
|
|
|
/** |
|
Encode significant pass |
|
*/ |
|
static void opj_t1_enc_sigpass(opj_t1_t *t1, |
|
OPJ_INT32 bpno, |
|
OPJ_INT32 *nmsedec, |
|
OPJ_BYTE type, |
|
OPJ_UINT32 cblksty); |
|
|
|
/** |
|
Decode significant pass |
|
*/ |
|
static void opj_t1_dec_sigpass_raw( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno, |
|
OPJ_INT32 cblksty); |
|
|
|
/** |
|
Encode refinement pass |
|
*/ |
|
static void opj_t1_enc_refpass(opj_t1_t *t1, |
|
OPJ_INT32 bpno, |
|
OPJ_INT32 *nmsedec, |
|
OPJ_BYTE type); |
|
|
|
/** |
|
Decode refinement pass |
|
*/ |
|
static void opj_t1_dec_refpass_raw( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno); |
|
|
|
|
|
/** |
|
Decode refinement pass |
|
*/ |
|
|
|
static INLINE void opj_t1_dec_refpass_step_raw( |
|
opj_t1_t *t1, |
|
opj_flag_t *flagsp, |
|
OPJ_INT32 *datap, |
|
OPJ_INT32 poshalf, |
|
OPJ_UINT32 row); |
|
static INLINE void opj_t1_dec_refpass_step_mqc( |
|
opj_t1_t *t1, |
|
opj_flag_t *flagsp, |
|
OPJ_INT32 *datap, |
|
OPJ_INT32 poshalf, |
|
OPJ_UINT32 row); |
|
|
|
|
|
/** |
|
Decode clean-up pass |
|
*/ |
|
|
|
static void opj_t1_dec_clnpass_step( |
|
opj_t1_t *t1, |
|
opj_flag_t *flagsp, |
|
OPJ_INT32 *datap, |
|
OPJ_INT32 oneplushalf, |
|
OPJ_UINT32 row, |
|
OPJ_UINT32 vsc); |
|
|
|
/** |
|
Encode clean-up pass |
|
*/ |
|
static void opj_t1_enc_clnpass( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno, |
|
OPJ_INT32 *nmsedec, |
|
OPJ_UINT32 cblksty); |
|
|
|
static OPJ_FLOAT64 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); |
|
|
|
/** Return "cumwmsedec" that should be used to increase tile->distotile */ |
|
static double 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, |
|
const OPJ_FLOAT64 * mct_norms, |
|
OPJ_UINT32 mct_numcomps); |
|
|
|
/** |
|
Decode 1 code-block |
|
@param t1 T1 handle |
|
@param cblk Code-block coding parameters |
|
@param orient |
|
@param roishift Region of interest shifting value |
|
@param cblksty Code-block style |
|
@param p_manager the event manager |
|
@param p_manager_mutex mutex for the event manager |
|
@param check_pterm whether PTERM correct termination should be checked |
|
*/ |
|
static OPJ_BOOL 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); |
|
|
|
/** |
|
Decode 1 HT code-block |
|
@param t1 T1 handle |
|
@param cblk Code-block coding parameters |
|
@param orient |
|
@param roishift Region of interest shifting value |
|
@param cblksty Code-block style |
|
@param p_manager the event manager |
|
@param p_manager_mutex mutex for the event manager |
|
@param check_pterm whether PTERM correct termination should be checked |
|
*/ |
|
OPJ_BOOL opj_t1_ht_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); |
|
|
|
|
|
static OPJ_BOOL opj_t1_allocate_buffers(opj_t1_t *t1, |
|
OPJ_UINT32 w, |
|
OPJ_UINT32 h); |
|
|
|
/*@}*/ |
|
|
|
/*@}*/ |
|
|
|
/* ----------------------------------------------------------------------- */ |
|
|
|
static INLINE OPJ_BYTE opj_t1_getctxno_zc(opj_mqc_t *mqc, OPJ_UINT32 f) |
|
{ |
|
return mqc->lut_ctxno_zc_orient[(f & T1_SIGMA_NEIGHBOURS)]; |
|
} |
|
|
|
static INLINE OPJ_UINT32 opj_t1_getctxtno_sc_or_spb_index(OPJ_UINT32 fX, |
|
OPJ_UINT32 pfX, |
|
OPJ_UINT32 nfX, |
|
OPJ_UINT32 ci) |
|
{ |
|
/* |
|
0 pfX T1_CHI_THIS T1_LUT_SGN_W |
|
1 tfX T1_SIGMA_1 T1_LUT_SIG_N |
|
2 nfX T1_CHI_THIS T1_LUT_SGN_E |
|
3 tfX T1_SIGMA_3 T1_LUT_SIG_W |
|
4 fX T1_CHI_(THIS - 1) T1_LUT_SGN_N |
|
5 tfX T1_SIGMA_5 T1_LUT_SIG_E |
|
6 fX T1_CHI_(THIS + 1) T1_LUT_SGN_S |
|
7 tfX T1_SIGMA_7 T1_LUT_SIG_S |
|
*/ |
|
|
|
OPJ_UINT32 lu = (fX >> (ci * 3U)) & (T1_SIGMA_1 | T1_SIGMA_3 | T1_SIGMA_5 | |
|
T1_SIGMA_7); |
|
|
|
lu |= (pfX >> (T1_CHI_THIS_I + (ci * 3U))) & (1U << 0); |
|
lu |= (nfX >> (T1_CHI_THIS_I - 2U + (ci * 3U))) & (1U << 2); |
|
if (ci == 0U) { |
|
lu |= (fX >> (T1_CHI_0_I - 4U)) & (1U << 4); |
|
} else { |
|
lu |= (fX >> (T1_CHI_1_I - 4U + ((ci - 1U) * 3U))) & (1U << 4); |
|
} |
|
lu |= (fX >> (T1_CHI_2_I - 6U + (ci * 3U))) & (1U << 6); |
|
return lu; |
|
} |
|
|
|
static INLINE OPJ_BYTE opj_t1_getctxno_sc(OPJ_UINT32 lu) |
|
{ |
|
return lut_ctxno_sc[lu]; |
|
} |
|
|
|
static INLINE OPJ_UINT32 opj_t1_getctxno_mag(OPJ_UINT32 f) |
|
{ |
|
OPJ_UINT32 tmp = (f & T1_SIGMA_NEIGHBOURS) ? T1_CTXNO_MAG + 1 : T1_CTXNO_MAG; |
|
OPJ_UINT32 tmp2 = (f & T1_MU_0) ? T1_CTXNO_MAG + 2 : tmp; |
|
return tmp2; |
|
} |
|
|
|
static INLINE OPJ_BYTE opj_t1_getspb(OPJ_UINT32 lu) |
|
{ |
|
return lut_spb[lu]; |
|
} |
|
|
|
static OPJ_INT16 opj_t1_getnmsedec_sig(OPJ_UINT32 x, OPJ_UINT32 bitpos) |
|
{ |
|
if (bitpos > 0) { |
|
return lut_nmsedec_sig[(x >> (bitpos)) & ((1 << T1_NMSEDEC_BITS) - 1)]; |
|
} |
|
|
|
return lut_nmsedec_sig0[x & ((1 << T1_NMSEDEC_BITS) - 1)]; |
|
} |
|
|
|
static OPJ_INT16 opj_t1_getnmsedec_ref(OPJ_UINT32 x, OPJ_UINT32 bitpos) |
|
{ |
|
if (bitpos > 0) { |
|
return lut_nmsedec_ref[(x >> (bitpos)) & ((1 << T1_NMSEDEC_BITS) - 1)]; |
|
} |
|
|
|
return lut_nmsedec_ref0[x & ((1 << T1_NMSEDEC_BITS) - 1)]; |
|
} |
|
|
|
#define opj_t1_update_flags_macro(flags, flagsp, ci, s, stride, vsc) \ |
|
{ \ |
|
/* east */ \ |
|
flagsp[-1] |= T1_SIGMA_5 << (3U * ci); \ |
|
\ |
|
/* mark target as significant */ \ |
|
flags |= ((s << T1_CHI_1_I) | T1_SIGMA_4) << (3U * ci); \ |
|
\ |
|
/* west */ \ |
|
flagsp[1] |= T1_SIGMA_3 << (3U * ci); \ |
|
\ |
|
/* north-west, north, north-east */ \ |
|
if (ci == 0U && !(vsc)) { \ |
|
opj_flag_t* north = flagsp - (stride); \ |
|
*north |= (s << T1_CHI_5_I) | T1_SIGMA_16; \ |
|
north[-1] |= T1_SIGMA_17; \ |
|
north[1] |= T1_SIGMA_15; \ |
|
} \ |
|
\ |
|
/* south-west, south, south-east */ \ |
|
if (ci == 3U) { \ |
|
opj_flag_t* south = flagsp + (stride); \ |
|
*south |= (s << T1_CHI_0_I) | T1_SIGMA_1; \ |
|
south[-1] |= T1_SIGMA_2; \ |
|
south[1] |= T1_SIGMA_0; \ |
|
} \ |
|
} |
|
|
|
|
|
static INLINE void opj_t1_update_flags(opj_flag_t *flagsp, OPJ_UINT32 ci, |
|
OPJ_UINT32 s, OPJ_UINT32 stride, |
|
OPJ_UINT32 vsc) |
|
{ |
|
opj_t1_update_flags_macro(*flagsp, flagsp, ci, s, stride, vsc); |
|
} |
|
|
|
/** |
|
Encode significant pass |
|
*/ |
|
#define opj_t1_enc_sigpass_step_macro(mqc, curctx, a, c, ct, flagspIn, datapIn, bpno, one, nmsedec, type, ciIn, vscIn) \ |
|
{ \ |
|
OPJ_UINT32 v; \ |
|
const OPJ_UINT32 ci = (ciIn); \ |
|
const OPJ_UINT32 vsc = (vscIn); \ |
|
const OPJ_INT32* l_datap = (datapIn); \ |
|
opj_flag_t* flagsp = (flagspIn); \ |
|
OPJ_UINT32 const flags = *flagsp; \ |
|
if ((flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U))) == 0U && \ |
|
(flags & (T1_SIGMA_NEIGHBOURS << (ci * 3U))) != 0U) { \ |
|
OPJ_UINT32 ctxt1 = opj_t1_getctxno_zc(mqc, flags >> (ci * 3U)); \ |
|
v = (opj_smr_abs(*l_datap) & (OPJ_UINT32)one) ? 1 : 0; \ |
|
/* #ifdef DEBUG_ENC_SIG */ \ |
|
/* fprintf(stderr, " ctxt1=%d\n", ctxt1); */ \ |
|
/* #endif */ \ |
|
opj_t1_setcurctx(curctx, ctxt1); \ |
|
if (type == T1_TYPE_RAW) { /* BYPASS/LAZY MODE */ \ |
|
opj_mqc_bypass_enc_macro(mqc, c, ct, v); \ |
|
} else { \ |
|
opj_mqc_encode_macro(mqc, curctx, a, c, ct, v); \ |
|
} \ |
|
if (v) { \ |
|
OPJ_UINT32 lu = opj_t1_getctxtno_sc_or_spb_index( \ |
|
*flagsp, \ |
|
flagsp[-1], flagsp[1], \ |
|
ci); \ |
|
OPJ_UINT32 ctxt2 = opj_t1_getctxno_sc(lu); \ |
|
v = opj_smr_sign(*l_datap); \ |
|
*nmsedec += opj_t1_getnmsedec_sig(opj_smr_abs(*l_datap), \ |
|
(OPJ_UINT32)bpno); \ |
|
/* #ifdef DEBUG_ENC_SIG */ \ |
|
/* fprintf(stderr, " ctxt2=%d\n", ctxt2); */ \ |
|
/* #endif */ \ |
|
opj_t1_setcurctx(curctx, ctxt2); \ |
|
if (type == T1_TYPE_RAW) { /* BYPASS/LAZY MODE */ \ |
|
opj_mqc_bypass_enc_macro(mqc, c, ct, v); \ |
|
} else { \ |
|
OPJ_UINT32 spb = opj_t1_getspb(lu); \ |
|
/* #ifdef DEBUG_ENC_SIG */ \ |
|
/* fprintf(stderr, " spb=%d\n", spb); */ \ |
|
/* #endif */ \ |
|
opj_mqc_encode_macro(mqc, curctx, a, c, ct, v ^ spb); \ |
|
} \ |
|
opj_t1_update_flags(flagsp, ci, v, t1->w + 2, vsc); \ |
|
} \ |
|
*flagsp |= T1_PI_THIS << (ci * 3U); \ |
|
} \ |
|
} |
|
|
|
static INLINE void 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) |
|
{ |
|
OPJ_UINT32 v; |
|
opj_mqc_t *mqc = &(t1->mqc); /* RAW component */ |
|
|
|
OPJ_UINT32 const flags = *flagsp; |
|
|
|
if ((flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U))) == 0U && |
|
(flags & (T1_SIGMA_NEIGHBOURS << (ci * 3U))) != 0U) { |
|
if (opj_mqc_raw_decode(mqc)) { |
|
v = opj_mqc_raw_decode(mqc); |
|
*datap = v ? -oneplushalf : oneplushalf; |
|
opj_t1_update_flags(flagsp, ci, v, t1->w + 2, vsc); |
|
} |
|
*flagsp |= T1_PI_THIS << (ci * 3U); |
|
} |
|
} |
|
|
|
#define opj_t1_dec_sigpass_step_mqc_macro(flags, flagsp, flags_stride, data, \ |
|
data_stride, ci, mqc, curctx, \ |
|
v, a, c, ct, oneplushalf, vsc) \ |
|
{ \ |
|
if ((flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U))) == 0U && \ |
|
(flags & (T1_SIGMA_NEIGHBOURS << (ci * 3U))) != 0U) { \ |
|
OPJ_UINT32 ctxt1 = opj_t1_getctxno_zc(mqc, flags >> (ci * 3U)); \ |
|
opj_t1_setcurctx(curctx, ctxt1); \ |
|
opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \ |
|
if (v) { \ |
|
OPJ_UINT32 lu = opj_t1_getctxtno_sc_or_spb_index( \ |
|
flags, \ |
|
flagsp[-1], flagsp[1], \ |
|
ci); \ |
|
OPJ_UINT32 ctxt2 = opj_t1_getctxno_sc(lu); \ |
|
OPJ_UINT32 spb = opj_t1_getspb(lu); \ |
|
opj_t1_setcurctx(curctx, ctxt2); \ |
|
opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \ |
|
v = v ^ spb; \ |
|
data[ci*data_stride] = v ? -oneplushalf : oneplushalf; \ |
|
opj_t1_update_flags_macro(flags, flagsp, ci, v, flags_stride, vsc); \ |
|
} \ |
|
flags |= T1_PI_THIS << (ci * 3U); \ |
|
} \ |
|
} |
|
|
|
static INLINE void 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) |
|
{ |
|
OPJ_UINT32 v; |
|
|
|
opj_mqc_t *mqc = &(t1->mqc); /* MQC component */ |
|
opj_t1_dec_sigpass_step_mqc_macro(*flagsp, flagsp, flags_stride, datap, |
|
0, ci, mqc, mqc->curctx, |
|
v, mqc->a, mqc->c, mqc->ct, oneplushalf, vsc); |
|
} |
|
|
|
static void opj_t1_enc_sigpass(opj_t1_t *t1, |
|
OPJ_INT32 bpno, |
|
OPJ_INT32 *nmsedec, |
|
OPJ_BYTE type, |
|
OPJ_UINT32 cblksty |
|
) |
|
{ |
|
OPJ_UINT32 i, k; |
|
OPJ_INT32 const one = 1 << (bpno + T1_NMSEDEC_FRACBITS); |
|
opj_flag_t* f = &T1_FLAGS(0, 0); |
|
OPJ_UINT32 const extra = 2; |
|
opj_mqc_t* mqc = &(t1->mqc); |
|
DOWNLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); |
|
const OPJ_INT32* datap = t1->data; |
|
|
|
*nmsedec = 0; |
|
#ifdef DEBUG_ENC_SIG |
|
fprintf(stderr, "enc_sigpass: bpno=%d\n", bpno); |
|
#endif |
|
for (k = 0; k < (t1->h & ~3U); k += 4, f += extra) { |
|
const OPJ_UINT32 w = t1->w; |
|
#ifdef DEBUG_ENC_SIG |
|
fprintf(stderr, " k=%d\n", k); |
|
#endif |
|
for (i = 0; i < w; ++i, ++f, datap += 4) { |
|
#ifdef DEBUG_ENC_SIG |
|
fprintf(stderr, " i=%d\n", i); |
|
#endif |
|
if (*f == 0U) { |
|
/* Nothing to do for any of the 4 data points */ |
|
continue; |
|
} |
|
opj_t1_enc_sigpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
f, |
|
&datap[0], |
|
bpno, |
|
one, |
|
nmsedec, |
|
type, |
|
0, cblksty & J2K_CCP_CBLKSTY_VSC); |
|
opj_t1_enc_sigpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
f, |
|
&datap[1], |
|
bpno, |
|
one, |
|
nmsedec, |
|
type, |
|
1, 0); |
|
opj_t1_enc_sigpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
f, |
|
&datap[2], |
|
bpno, |
|
one, |
|
nmsedec, |
|
type, |
|
2, 0); |
|
opj_t1_enc_sigpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
f, |
|
&datap[3], |
|
bpno, |
|
one, |
|
nmsedec, |
|
type, |
|
3, 0); |
|
} |
|
} |
|
|
|
if (k < t1->h) { |
|
OPJ_UINT32 j; |
|
#ifdef DEBUG_ENC_SIG |
|
fprintf(stderr, " k=%d\n", k); |
|
#endif |
|
for (i = 0; i < t1->w; ++i, ++f) { |
|
#ifdef DEBUG_ENC_SIG |
|
fprintf(stderr, " i=%d\n", i); |
|
#endif |
|
if (*f == 0U) { |
|
/* Nothing to do for any of the 4 data points */ |
|
datap += (t1->h - k); |
|
continue; |
|
} |
|
for (j = k; j < t1->h; ++j, ++datap) { |
|
opj_t1_enc_sigpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
f, |
|
&datap[0], |
|
bpno, |
|
one, |
|
nmsedec, |
|
type, |
|
j - k, |
|
(j == k && (cblksty & J2K_CCP_CBLKSTY_VSC) != 0)); |
|
} |
|
} |
|
} |
|
|
|
UPLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); |
|
} |
|
|
|
static void opj_t1_dec_sigpass_raw( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno, |
|
OPJ_INT32 cblksty) |
|
{ |
|
OPJ_INT32 one, half, oneplushalf; |
|
OPJ_UINT32 i, j, k; |
|
OPJ_INT32 *data = t1->data; |
|
opj_flag_t *flagsp = &T1_FLAGS(0, 0); |
|
const OPJ_UINT32 l_w = t1->w; |
|
one = 1 << bpno; |
|
half = one >> 1; |
|
oneplushalf = one | half; |
|
|
|
for (k = 0; k < (t1->h & ~3U); k += 4, flagsp += 2, data += 3 * l_w) { |
|
for (i = 0; i < l_w; ++i, ++flagsp, ++data) { |
|
opj_flag_t flags = *flagsp; |
|
if (flags != 0) { |
|
opj_t1_dec_sigpass_step_raw( |
|
t1, |
|
flagsp, |
|
data, |
|
oneplushalf, |
|
cblksty & J2K_CCP_CBLKSTY_VSC, /* vsc */ |
|
0U); |
|
opj_t1_dec_sigpass_step_raw( |
|
t1, |
|
flagsp, |
|
data + l_w, |
|
oneplushalf, |
|
OPJ_FALSE, /* vsc */ |
|
1U); |
|
opj_t1_dec_sigpass_step_raw( |
|
t1, |
|
flagsp, |
|
data + 2 * l_w, |
|
oneplushalf, |
|
OPJ_FALSE, /* vsc */ |
|
2U); |
|
opj_t1_dec_sigpass_step_raw( |
|
t1, |
|
flagsp, |
|
data + 3 * l_w, |
|
oneplushalf, |
|
OPJ_FALSE, /* vsc */ |
|
3U); |
|
} |
|
} |
|
} |
|
if (k < t1->h) { |
|
for (i = 0; i < l_w; ++i, ++flagsp, ++data) { |
|
for (j = 0; j < t1->h - k; ++j) { |
|
opj_t1_dec_sigpass_step_raw( |
|
t1, |
|
flagsp, |
|
data + j * l_w, |
|
oneplushalf, |
|
cblksty & J2K_CCP_CBLKSTY_VSC, /* vsc */ |
|
j); |
|
} |
|
} |
|
} |
|
} |
|
|
|
#define opj_t1_dec_sigpass_mqc_internal(t1, bpno, vsc, w, h, flags_stride) \ |
|
{ \ |
|
OPJ_INT32 one, half, oneplushalf; \ |
|
OPJ_UINT32 i, j, k; \ |
|
register OPJ_INT32 *data = t1->data; \ |
|
register opj_flag_t *flagsp = &t1->flags[(flags_stride) + 1]; \ |
|
const OPJ_UINT32 l_w = w; \ |
|
opj_mqc_t* mqc = &(t1->mqc); \ |
|
DOWNLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); \ |
|
register OPJ_UINT32 v; \ |
|
one = 1 << bpno; \ |
|
half = one >> 1; \ |
|
oneplushalf = one | half; \ |
|
for (k = 0; k < (h & ~3u); k += 4, data += 3*l_w, flagsp += 2) { \ |
|
for (i = 0; i < l_w; ++i, ++data, ++flagsp) { \ |
|
opj_flag_t flags = *flagsp; \ |
|
if( flags != 0 ) { \ |
|
opj_t1_dec_sigpass_step_mqc_macro( \ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 0, mqc, curctx, v, a, c, ct, oneplushalf, vsc); \ |
|
opj_t1_dec_sigpass_step_mqc_macro( \ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 1, mqc, curctx, v, a, c, ct, oneplushalf, OPJ_FALSE); \ |
|
opj_t1_dec_sigpass_step_mqc_macro( \ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 2, mqc, curctx, v, a, c, ct, oneplushalf, OPJ_FALSE); \ |
|
opj_t1_dec_sigpass_step_mqc_macro( \ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 3, mqc, curctx, v, a, c, ct, oneplushalf, OPJ_FALSE); \ |
|
*flagsp = flags; \ |
|
} \ |
|
} \ |
|
} \ |
|
UPLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); \ |
|
if( k < h ) { \ |
|
for (i = 0; i < l_w; ++i, ++data, ++flagsp) { \ |
|
for (j = 0; j < h - k; ++j) { \ |
|
opj_t1_dec_sigpass_step_mqc(t1, flagsp, \ |
|
data + j * l_w, oneplushalf, j, flags_stride, vsc); \ |
|
} \ |
|
} \ |
|
} \ |
|
} |
|
|
|
static void opj_t1_dec_sigpass_mqc_64x64_novsc( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
opj_t1_dec_sigpass_mqc_internal(t1, bpno, OPJ_FALSE, 64, 64, 66); |
|
} |
|
|
|
static void opj_t1_dec_sigpass_mqc_64x64_vsc( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
opj_t1_dec_sigpass_mqc_internal(t1, bpno, OPJ_TRUE, 64, 64, 66); |
|
} |
|
|
|
static void opj_t1_dec_sigpass_mqc_generic_novsc( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
opj_t1_dec_sigpass_mqc_internal(t1, bpno, OPJ_FALSE, t1->w, t1->h, |
|
t1->w + 2U); |
|
} |
|
|
|
static void opj_t1_dec_sigpass_mqc_generic_vsc( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
opj_t1_dec_sigpass_mqc_internal(t1, bpno, OPJ_TRUE, t1->w, t1->h, |
|
t1->w + 2U); |
|
} |
|
|
|
static void opj_t1_dec_sigpass_mqc( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno, |
|
OPJ_INT32 cblksty) |
|
{ |
|
if (t1->w == 64 && t1->h == 64) { |
|
if (cblksty & J2K_CCP_CBLKSTY_VSC) { |
|
opj_t1_dec_sigpass_mqc_64x64_vsc(t1, bpno); |
|
} else { |
|
opj_t1_dec_sigpass_mqc_64x64_novsc(t1, bpno); |
|
} |
|
} else { |
|
if (cblksty & J2K_CCP_CBLKSTY_VSC) { |
|
opj_t1_dec_sigpass_mqc_generic_vsc(t1, bpno); |
|
} else { |
|
opj_t1_dec_sigpass_mqc_generic_novsc(t1, bpno); |
|
} |
|
} |
|
} |
|
|
|
/** |
|
Encode refinement pass step |
|
*/ |
|
#define opj_t1_enc_refpass_step_macro(mqc, curctx, a, c, ct, flags, flagsUpdated, datap, bpno, one, nmsedec, type, ci) \ |
|
{\ |
|
OPJ_UINT32 v; \ |
|
if ((flags & ((T1_SIGMA_THIS | T1_PI_THIS) << ((ci) * 3U))) == (T1_SIGMA_THIS << ((ci) * 3U))) { \ |
|
const OPJ_UINT32 shift_flags = (flags >> ((ci) * 3U)); \ |
|
OPJ_UINT32 ctxt = opj_t1_getctxno_mag(shift_flags); \ |
|
OPJ_UINT32 abs_data = opj_smr_abs(*datap); \ |
|
*nmsedec += opj_t1_getnmsedec_ref(abs_data, \ |
|
(OPJ_UINT32)bpno); \ |
|
v = ((OPJ_INT32)abs_data & one) ? 1 : 0; \ |
|
/* #ifdef DEBUG_ENC_REF */ \ |
|
/* fprintf(stderr, " ctxt=%d\n", ctxt); */ \ |
|
/* #endif */ \ |
|
opj_t1_setcurctx(curctx, ctxt); \ |
|
if (type == T1_TYPE_RAW) { /* BYPASS/LAZY MODE */ \ |
|
opj_mqc_bypass_enc_macro(mqc, c, ct, v); \ |
|
} else { \ |
|
opj_mqc_encode_macro(mqc, curctx, a, c, ct, v); \ |
|
} \ |
|
flagsUpdated |= T1_MU_THIS << ((ci) * 3U); \ |
|
} \ |
|
} |
|
|
|
|
|
static INLINE void opj_t1_dec_refpass_step_raw( |
|
opj_t1_t *t1, |
|
opj_flag_t *flagsp, |
|
OPJ_INT32 *datap, |
|
OPJ_INT32 poshalf, |
|
OPJ_UINT32 ci) |
|
{ |
|
OPJ_UINT32 v; |
|
|
|
opj_mqc_t *mqc = &(t1->mqc); /* RAW component */ |
|
|
|
if ((*flagsp & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U))) == |
|
(T1_SIGMA_THIS << (ci * 3U))) { |
|
v = opj_mqc_raw_decode(mqc); |
|
*datap += (v ^ (*datap < 0)) ? poshalf : -poshalf; |
|
*flagsp |= T1_MU_THIS << (ci * 3U); |
|
} |
|
} |
|
|
|
#define opj_t1_dec_refpass_step_mqc_macro(flags, data, data_stride, ci, \ |
|
mqc, curctx, v, a, c, ct, poshalf) \ |
|
{ \ |
|
if ((flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U))) == \ |
|
(T1_SIGMA_THIS << (ci * 3U))) { \ |
|
OPJ_UINT32 ctxt = opj_t1_getctxno_mag(flags >> (ci * 3U)); \ |
|
opj_t1_setcurctx(curctx, ctxt); \ |
|
opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \ |
|
data[ci*data_stride] += (v ^ (data[ci*data_stride] < 0)) ? poshalf : -poshalf; \ |
|
flags |= T1_MU_THIS << (ci * 3U); \ |
|
} \ |
|
} |
|
|
|
static INLINE void opj_t1_dec_refpass_step_mqc( |
|
opj_t1_t *t1, |
|
opj_flag_t *flagsp, |
|
OPJ_INT32 *datap, |
|
OPJ_INT32 poshalf, |
|
OPJ_UINT32 ci) |
|
{ |
|
OPJ_UINT32 v; |
|
|
|
opj_mqc_t *mqc = &(t1->mqc); /* MQC component */ |
|
opj_t1_dec_refpass_step_mqc_macro(*flagsp, datap, 0, ci, |
|
mqc, mqc->curctx, v, mqc->a, mqc->c, |
|
mqc->ct, poshalf); |
|
} |
|
|
|
static void opj_t1_enc_refpass( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno, |
|
OPJ_INT32 *nmsedec, |
|
OPJ_BYTE type) |
|
{ |
|
OPJ_UINT32 i, k; |
|
const OPJ_INT32 one = 1 << (bpno + T1_NMSEDEC_FRACBITS); |
|
opj_flag_t* f = &T1_FLAGS(0, 0); |
|
const OPJ_UINT32 extra = 2U; |
|
opj_mqc_t* mqc = &(t1->mqc); |
|
DOWNLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); |
|
const OPJ_INT32* datap = t1->data; |
|
|
|
*nmsedec = 0; |
|
#ifdef DEBUG_ENC_REF |
|
fprintf(stderr, "enc_refpass: bpno=%d\n", bpno); |
|
#endif |
|
for (k = 0; k < (t1->h & ~3U); k += 4, f += extra) { |
|
#ifdef DEBUG_ENC_REF |
|
fprintf(stderr, " k=%d\n", k); |
|
#endif |
|
for (i = 0; i < t1->w; ++i, f++, datap += 4) { |
|
const OPJ_UINT32 flags = *f; |
|
OPJ_UINT32 flagsUpdated = flags; |
|
#ifdef DEBUG_ENC_REF |
|
fprintf(stderr, " i=%d\n", i); |
|
#endif |
|
if ((flags & (T1_SIGMA_4 | T1_SIGMA_7 | T1_SIGMA_10 | T1_SIGMA_13)) == 0) { |
|
/* none significant */ |
|
continue; |
|
} |
|
if ((flags & (T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3)) == |
|
(T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3)) { |
|
/* all processed by sigpass */ |
|
continue; |
|
} |
|
|
|
opj_t1_enc_refpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
flags, flagsUpdated, |
|
&datap[0], |
|
bpno, |
|
one, |
|
nmsedec, |
|
type, |
|
0); |
|
opj_t1_enc_refpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
flags, flagsUpdated, |
|
&datap[1], |
|
bpno, |
|
one, |
|
nmsedec, |
|
type, |
|
1); |
|
opj_t1_enc_refpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
flags, flagsUpdated, |
|
&datap[2], |
|
bpno, |
|
one, |
|
nmsedec, |
|
type, |
|
2); |
|
opj_t1_enc_refpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
flags, flagsUpdated, |
|
&datap[3], |
|
bpno, |
|
one, |
|
nmsedec, |
|
type, |
|
3); |
|
*f = flagsUpdated; |
|
} |
|
} |
|
|
|
if (k < t1->h) { |
|
OPJ_UINT32 j; |
|
const OPJ_UINT32 remaining_lines = t1->h - k; |
|
#ifdef DEBUG_ENC_REF |
|
fprintf(stderr, " k=%d\n", k); |
|
#endif |
|
for (i = 0; i < t1->w; ++i, ++f) { |
|
#ifdef DEBUG_ENC_REF |
|
fprintf(stderr, " i=%d\n", i); |
|
#endif |
|
if ((*f & (T1_SIGMA_4 | T1_SIGMA_7 | T1_SIGMA_10 | T1_SIGMA_13)) == 0) { |
|
/* none significant */ |
|
datap += remaining_lines; |
|
continue; |
|
} |
|
for (j = 0; j < remaining_lines; ++j, datap ++) { |
|
opj_t1_enc_refpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
*f, *f, |
|
&datap[0], |
|
bpno, |
|
one, |
|
nmsedec, |
|
type, |
|
j); |
|
} |
|
} |
|
} |
|
|
|
UPLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); |
|
} |
|
|
|
|
|
static void opj_t1_dec_refpass_raw( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
OPJ_INT32 one, poshalf; |
|
OPJ_UINT32 i, j, k; |
|
OPJ_INT32 *data = t1->data; |
|
opj_flag_t *flagsp = &T1_FLAGS(0, 0); |
|
const OPJ_UINT32 l_w = t1->w; |
|
one = 1 << bpno; |
|
poshalf = one >> 1; |
|
for (k = 0; k < (t1->h & ~3U); k += 4, flagsp += 2, data += 3 * l_w) { |
|
for (i = 0; i < l_w; ++i, ++flagsp, ++data) { |
|
opj_flag_t flags = *flagsp; |
|
if (flags != 0) { |
|
opj_t1_dec_refpass_step_raw( |
|
t1, |
|
flagsp, |
|
data, |
|
poshalf, |
|
0U); |
|
opj_t1_dec_refpass_step_raw( |
|
t1, |
|
flagsp, |
|
data + l_w, |
|
poshalf, |
|
1U); |
|
opj_t1_dec_refpass_step_raw( |
|
t1, |
|
flagsp, |
|
data + 2 * l_w, |
|
poshalf, |
|
2U); |
|
opj_t1_dec_refpass_step_raw( |
|
t1, |
|
flagsp, |
|
data + 3 * l_w, |
|
poshalf, |
|
3U); |
|
} |
|
} |
|
} |
|
if (k < t1->h) { |
|
for (i = 0; i < l_w; ++i, ++flagsp, ++data) { |
|
for (j = 0; j < t1->h - k; ++j) { |
|
opj_t1_dec_refpass_step_raw( |
|
t1, |
|
flagsp, |
|
data + j * l_w, |
|
poshalf, |
|
j); |
|
} |
|
} |
|
} |
|
} |
|
|
|
#define opj_t1_dec_refpass_mqc_internal(t1, bpno, w, h, flags_stride) \ |
|
{ \ |
|
OPJ_INT32 one, poshalf; \ |
|
OPJ_UINT32 i, j, k; \ |
|
register OPJ_INT32 *data = t1->data; \ |
|
register opj_flag_t *flagsp = &t1->flags[flags_stride + 1]; \ |
|
const OPJ_UINT32 l_w = w; \ |
|
opj_mqc_t* mqc = &(t1->mqc); \ |
|
DOWNLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); \ |
|
register OPJ_UINT32 v; \ |
|
one = 1 << bpno; \ |
|
poshalf = one >> 1; \ |
|
for (k = 0; k < (h & ~3u); k += 4, data += 3*l_w, flagsp += 2) { \ |
|
for (i = 0; i < l_w; ++i, ++data, ++flagsp) { \ |
|
opj_flag_t flags = *flagsp; \ |
|
if( flags != 0 ) { \ |
|
opj_t1_dec_refpass_step_mqc_macro( \ |
|
flags, data, l_w, 0, \ |
|
mqc, curctx, v, a, c, ct, poshalf); \ |
|
opj_t1_dec_refpass_step_mqc_macro( \ |
|
flags, data, l_w, 1, \ |
|
mqc, curctx, v, a, c, ct, poshalf); \ |
|
opj_t1_dec_refpass_step_mqc_macro( \ |
|
flags, data, l_w, 2, \ |
|
mqc, curctx, v, a, c, ct, poshalf); \ |
|
opj_t1_dec_refpass_step_mqc_macro( \ |
|
flags, data, l_w, 3, \ |
|
mqc, curctx, v, a, c, ct, poshalf); \ |
|
*flagsp = flags; \ |
|
} \ |
|
} \ |
|
} \ |
|
UPLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); \ |
|
if( k < h ) { \ |
|
for (i = 0; i < l_w; ++i, ++data, ++flagsp) { \ |
|
for (j = 0; j < h - k; ++j) { \ |
|
opj_t1_dec_refpass_step_mqc(t1, flagsp, data + j * l_w, poshalf, j); \ |
|
} \ |
|
} \ |
|
} \ |
|
} |
|
|
|
static void opj_t1_dec_refpass_mqc_64x64( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
opj_t1_dec_refpass_mqc_internal(t1, bpno, 64, 64, 66); |
|
} |
|
|
|
static void opj_t1_dec_refpass_mqc_generic( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
opj_t1_dec_refpass_mqc_internal(t1, bpno, t1->w, t1->h, t1->w + 2U); |
|
} |
|
|
|
static void opj_t1_dec_refpass_mqc( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
if (t1->w == 64 && t1->h == 64) { |
|
opj_t1_dec_refpass_mqc_64x64(t1, bpno); |
|
} else { |
|
opj_t1_dec_refpass_mqc_generic(t1, bpno); |
|
} |
|
} |
|
|
|
/** |
|
Encode clean-up pass step |
|
*/ |
|
#define opj_t1_enc_clnpass_step_macro(mqc, curctx, a, c, ct, flagspIn, datapIn, bpno, one, nmsedec, agg, runlen, lim, cblksty) \ |
|
{ \ |
|
OPJ_UINT32 v; \ |
|
OPJ_UINT32 ci; \ |
|
opj_flag_t* const flagsp = (flagspIn); \ |
|
const OPJ_INT32* l_datap = (datapIn); \ |
|
const OPJ_UINT32 check = (T1_SIGMA_4 | T1_SIGMA_7 | T1_SIGMA_10 | T1_SIGMA_13 | \ |
|
T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3); \ |
|
\ |
|
if ((*flagsp & check) == check) { \ |
|
if (runlen == 0) { \ |
|
*flagsp &= ~(T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3); \ |
|
} else if (runlen == 1) { \ |
|
*flagsp &= ~(T1_PI_1 | T1_PI_2 | T1_PI_3); \ |
|
} else if (runlen == 2) { \ |
|
*flagsp &= ~(T1_PI_2 | T1_PI_3); \ |
|
} else if (runlen == 3) { \ |
|
*flagsp &= ~(T1_PI_3); \ |
|
} \ |
|
} \ |
|
else \ |
|
for (ci = runlen; ci < lim; ++ci) { \ |
|
OPJ_BOOL goto_PARTIAL = OPJ_FALSE; \ |
|
if ((agg != 0) && (ci == runlen)) { \ |
|
goto_PARTIAL = OPJ_TRUE; \ |
|
} \ |
|
else if (!(*flagsp & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U)))) { \ |
|
OPJ_UINT32 ctxt1 = opj_t1_getctxno_zc(mqc, *flagsp >> (ci * 3U)); \ |
|
/* #ifdef DEBUG_ENC_CLN */ \ |
|
/* printf(" ctxt1=%d\n", ctxt1); */ \ |
|
/* #endif */ \ |
|
opj_t1_setcurctx(curctx, ctxt1); \ |
|
v = (opj_smr_abs(*l_datap) & (OPJ_UINT32)one) ? 1 : 0; \ |
|
opj_mqc_encode_macro(mqc, curctx, a, c, ct, v); \ |
|
if (v) { \ |
|
goto_PARTIAL = OPJ_TRUE; \ |
|
} \ |
|
} \ |
|
if( goto_PARTIAL ) { \ |
|
OPJ_UINT32 vsc; \ |
|
OPJ_UINT32 ctxt2, spb; \ |
|
OPJ_UINT32 lu = opj_t1_getctxtno_sc_or_spb_index( \ |
|
*flagsp, \ |
|
flagsp[-1], flagsp[1], \ |
|
ci); \ |
|
*nmsedec += opj_t1_getnmsedec_sig(opj_smr_abs(*l_datap), \ |
|
(OPJ_UINT32)bpno); \ |
|
ctxt2 = opj_t1_getctxno_sc(lu); \ |
|
/* #ifdef DEBUG_ENC_CLN */ \ |
|
/* printf(" ctxt2=%d\n", ctxt2); */ \ |
|
/* #endif */ \ |
|
opj_t1_setcurctx(curctx, ctxt2); \ |
|
\ |
|
v = opj_smr_sign(*l_datap); \ |
|
spb = opj_t1_getspb(lu); \ |
|
/* #ifdef DEBUG_ENC_CLN */ \ |
|
/* printf(" spb=%d\n", spb); */\ |
|
/* #endif */ \ |
|
opj_mqc_encode_macro(mqc, curctx, a, c, ct, v ^ spb); \ |
|
vsc = ((cblksty & J2K_CCP_CBLKSTY_VSC) && (ci == 0)) ? 1 : 0; \ |
|
opj_t1_update_flags(flagsp, ci, v, t1->w + 2U, vsc); \ |
|
} \ |
|
*flagsp &= ~(T1_PI_THIS << (3U * ci)); \ |
|
l_datap ++; \ |
|
} \ |
|
} |
|
|
|
#define opj_t1_dec_clnpass_step_macro(check_flags, partial, \ |
|
flags, flagsp, flags_stride, data, \ |
|
data_stride, ci, mqc, curctx, \ |
|
v, a, c, ct, oneplushalf, vsc) \ |
|
{ \ |
|
if ( !check_flags || !(flags & ((T1_SIGMA_THIS | T1_PI_THIS) << (ci * 3U)))) {\ |
|
do { \ |
|
if( !partial ) { \ |
|
OPJ_UINT32 ctxt1 = opj_t1_getctxno_zc(mqc, flags >> (ci * 3U)); \ |
|
opj_t1_setcurctx(curctx, ctxt1); \ |
|
opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \ |
|
if( !v ) \ |
|
break; \ |
|
} \ |
|
{ \ |
|
OPJ_UINT32 lu = opj_t1_getctxtno_sc_or_spb_index( \ |
|
flags, flagsp[-1], flagsp[1], \ |
|
ci); \ |
|
opj_t1_setcurctx(curctx, opj_t1_getctxno_sc(lu)); \ |
|
opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \ |
|
v = v ^ opj_t1_getspb(lu); \ |
|
data[ci*data_stride] = v ? -oneplushalf : oneplushalf; \ |
|
opj_t1_update_flags_macro(flags, flagsp, ci, v, flags_stride, vsc); \ |
|
} \ |
|
} while(0); \ |
|
} \ |
|
} |
|
|
|
static void 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) |
|
{ |
|
OPJ_UINT32 v; |
|
|
|
opj_mqc_t *mqc = &(t1->mqc); /* MQC component */ |
|
opj_t1_dec_clnpass_step_macro(OPJ_TRUE, OPJ_FALSE, |
|
*flagsp, flagsp, t1->w + 2U, datap, |
|
0, ci, mqc, mqc->curctx, |
|
v, mqc->a, mqc->c, mqc->ct, oneplushalf, vsc); |
|
} |
|
|
|
static void opj_t1_enc_clnpass( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno, |
|
OPJ_INT32 *nmsedec, |
|
OPJ_UINT32 cblksty) |
|
{ |
|
OPJ_UINT32 i, k; |
|
const OPJ_INT32 one = 1 << (bpno + T1_NMSEDEC_FRACBITS); |
|
opj_mqc_t* mqc = &(t1->mqc); |
|
DOWNLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); |
|
const OPJ_INT32* datap = t1->data; |
|
opj_flag_t *f = &T1_FLAGS(0, 0); |
|
const OPJ_UINT32 extra = 2U; |
|
|
|
*nmsedec = 0; |
|
#ifdef DEBUG_ENC_CLN |
|
printf("enc_clnpass: bpno=%d\n", bpno); |
|
#endif |
|
for (k = 0; k < (t1->h & ~3U); k += 4, f += extra) { |
|
#ifdef DEBUG_ENC_CLN |
|
printf(" k=%d\n", k); |
|
#endif |
|
for (i = 0; i < t1->w; ++i, f++) { |
|
OPJ_UINT32 agg, runlen; |
|
#ifdef DEBUG_ENC_CLN |
|
printf(" i=%d\n", i); |
|
#endif |
|
agg = !*f; |
|
#ifdef DEBUG_ENC_CLN |
|
printf(" agg=%d\n", agg); |
|
#endif |
|
if (agg) { |
|
for (runlen = 0; runlen < 4; ++runlen, ++datap) { |
|
if (opj_smr_abs(*datap) & (OPJ_UINT32)one) { |
|
break; |
|
} |
|
} |
|
opj_t1_setcurctx(curctx, T1_CTXNO_AGG); |
|
opj_mqc_encode_macro(mqc, curctx, a, c, ct, runlen != 4); |
|
if (runlen == 4) { |
|
continue; |
|
} |
|
opj_t1_setcurctx(curctx, T1_CTXNO_UNI); |
|
opj_mqc_encode_macro(mqc, curctx, a, c, ct, runlen >> 1); |
|
opj_mqc_encode_macro(mqc, curctx, a, c, ct, runlen & 1); |
|
} else { |
|
runlen = 0; |
|
} |
|
opj_t1_enc_clnpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
f, |
|
datap, |
|
bpno, |
|
one, |
|
nmsedec, |
|
agg, |
|
runlen, |
|
4U, |
|
cblksty); |
|
datap += 4 - runlen; |
|
} |
|
} |
|
if (k < t1->h) { |
|
const OPJ_UINT32 agg = 0; |
|
const OPJ_UINT32 runlen = 0; |
|
#ifdef DEBUG_ENC_CLN |
|
printf(" k=%d\n", k); |
|
#endif |
|
for (i = 0; i < t1->w; ++i, f++) { |
|
#ifdef DEBUG_ENC_CLN |
|
printf(" i=%d\n", i); |
|
printf(" agg=%d\n", agg); |
|
#endif |
|
opj_t1_enc_clnpass_step_macro( |
|
mqc, curctx, a, c, ct, |
|
f, |
|
datap, |
|
bpno, |
|
one, |
|
nmsedec, |
|
agg, |
|
runlen, |
|
t1->h - k, |
|
cblksty); |
|
datap += t1->h - k; |
|
} |
|
} |
|
|
|
UPLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); |
|
} |
|
|
|
#define opj_t1_dec_clnpass_internal(t1, bpno, vsc, w, h, flags_stride) \ |
|
{ \ |
|
OPJ_INT32 one, half, oneplushalf; \ |
|
OPJ_UINT32 runlen; \ |
|
OPJ_UINT32 i, j, k; \ |
|
const OPJ_UINT32 l_w = w; \ |
|
opj_mqc_t* mqc = &(t1->mqc); \ |
|
register OPJ_INT32 *data = t1->data; \ |
|
register opj_flag_t *flagsp = &t1->flags[flags_stride + 1]; \ |
|
DOWNLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); \ |
|
register OPJ_UINT32 v; \ |
|
one = 1 << bpno; \ |
|
half = one >> 1; \ |
|
oneplushalf = one | half; \ |
|
for (k = 0; k < (h & ~3u); k += 4, data += 3*l_w, flagsp += 2) { \ |
|
for (i = 0; i < l_w; ++i, ++data, ++flagsp) { \ |
|
opj_flag_t flags = *flagsp; \ |
|
if (flags == 0) { \ |
|
OPJ_UINT32 partial = OPJ_TRUE; \ |
|
opj_t1_setcurctx(curctx, T1_CTXNO_AGG); \ |
|
opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \ |
|
if (!v) { \ |
|
continue; \ |
|
} \ |
|
opj_t1_setcurctx(curctx, T1_CTXNO_UNI); \ |
|
opj_mqc_decode_macro(runlen, mqc, curctx, a, c, ct); \ |
|
opj_mqc_decode_macro(v, mqc, curctx, a, c, ct); \ |
|
runlen = (runlen << 1) | v; \ |
|
switch(runlen) { \ |
|
case 0: \ |
|
opj_t1_dec_clnpass_step_macro(OPJ_FALSE, OPJ_TRUE,\ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 0, mqc, curctx, \ |
|
v, a, c, ct, oneplushalf, vsc); \ |
|
partial = OPJ_FALSE; \ |
|
/* FALLTHRU */ \ |
|
case 1: \ |
|
opj_t1_dec_clnpass_step_macro(OPJ_FALSE, partial,\ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 1, mqc, curctx, \ |
|
v, a, c, ct, oneplushalf, OPJ_FALSE); \ |
|
partial = OPJ_FALSE; \ |
|
/* FALLTHRU */ \ |
|
case 2: \ |
|
opj_t1_dec_clnpass_step_macro(OPJ_FALSE, partial,\ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 2, mqc, curctx, \ |
|
v, a, c, ct, oneplushalf, OPJ_FALSE); \ |
|
partial = OPJ_FALSE; \ |
|
/* FALLTHRU */ \ |
|
case 3: \ |
|
opj_t1_dec_clnpass_step_macro(OPJ_FALSE, partial,\ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 3, mqc, curctx, \ |
|
v, a, c, ct, oneplushalf, OPJ_FALSE); \ |
|
break; \ |
|
} \ |
|
} else { \ |
|
opj_t1_dec_clnpass_step_macro(OPJ_TRUE, OPJ_FALSE, \ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 0, mqc, curctx, \ |
|
v, a, c, ct, oneplushalf, vsc); \ |
|
opj_t1_dec_clnpass_step_macro(OPJ_TRUE, OPJ_FALSE, \ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 1, mqc, curctx, \ |
|
v, a, c, ct, oneplushalf, OPJ_FALSE); \ |
|
opj_t1_dec_clnpass_step_macro(OPJ_TRUE, OPJ_FALSE, \ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 2, mqc, curctx, \ |
|
v, a, c, ct, oneplushalf, OPJ_FALSE); \ |
|
opj_t1_dec_clnpass_step_macro(OPJ_TRUE, OPJ_FALSE, \ |
|
flags, flagsp, flags_stride, data, \ |
|
l_w, 3, mqc, curctx, \ |
|
v, a, c, ct, oneplushalf, OPJ_FALSE); \ |
|
} \ |
|
*flagsp = flags & ~(T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3); \ |
|
} \ |
|
} \ |
|
UPLOAD_MQC_VARIABLES(mqc, curctx, a, c, ct); \ |
|
if( k < h ) { \ |
|
for (i = 0; i < l_w; ++i, ++flagsp, ++data) { \ |
|
for (j = 0; j < h - k; ++j) { \ |
|
opj_t1_dec_clnpass_step(t1, flagsp, data + j * l_w, oneplushalf, j, vsc); \ |
|
} \ |
|
*flagsp &= ~(T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3); \ |
|
} \ |
|
} \ |
|
} |
|
|
|
static void opj_t1_dec_clnpass_check_segsym(opj_t1_t *t1, OPJ_INT32 cblksty) |
|
{ |
|
if (cblksty & J2K_CCP_CBLKSTY_SEGSYM) { |
|
opj_mqc_t* mqc = &(t1->mqc); |
|
OPJ_UINT32 v, v2; |
|
opj_mqc_setcurctx(mqc, T1_CTXNO_UNI); |
|
opj_mqc_decode(v, mqc); |
|
opj_mqc_decode(v2, mqc); |
|
v = (v << 1) | v2; |
|
opj_mqc_decode(v2, mqc); |
|
v = (v << 1) | v2; |
|
opj_mqc_decode(v2, mqc); |
|
v = (v << 1) | v2; |
|
/* |
|
if (v!=0xa) { |
|
opj_event_msg(t1->cinfo, EVT_WARNING, "Bad segmentation symbol %x\n", v); |
|
} |
|
*/ |
|
} |
|
} |
|
|
|
static void opj_t1_dec_clnpass_64x64_novsc( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
opj_t1_dec_clnpass_internal(t1, bpno, OPJ_FALSE, 64, 64, 66); |
|
} |
|
|
|
static void opj_t1_dec_clnpass_64x64_vsc( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
opj_t1_dec_clnpass_internal(t1, bpno, OPJ_TRUE, 64, 64, 66); |
|
} |
|
|
|
static void opj_t1_dec_clnpass_generic_novsc( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
opj_t1_dec_clnpass_internal(t1, bpno, OPJ_FALSE, t1->w, t1->h, |
|
t1->w + 2U); |
|
} |
|
|
|
static void opj_t1_dec_clnpass_generic_vsc( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno) |
|
{ |
|
opj_t1_dec_clnpass_internal(t1, bpno, OPJ_TRUE, t1->w, t1->h, |
|
t1->w + 2U); |
|
} |
|
|
|
static void opj_t1_dec_clnpass( |
|
opj_t1_t *t1, |
|
OPJ_INT32 bpno, |
|
OPJ_INT32 cblksty) |
|
{ |
|
if (t1->w == 64 && t1->h == 64) { |
|
if (cblksty & J2K_CCP_CBLKSTY_VSC) { |
|
opj_t1_dec_clnpass_64x64_vsc(t1, bpno); |
|
} else { |
|
opj_t1_dec_clnpass_64x64_novsc(t1, bpno); |
|
} |
|
} else { |
|
if (cblksty & J2K_CCP_CBLKSTY_VSC) { |
|
opj_t1_dec_clnpass_generic_vsc(t1, bpno); |
|
} else { |
|
opj_t1_dec_clnpass_generic_novsc(t1, bpno); |
|
} |
|
} |
|
opj_t1_dec_clnpass_check_segsym(t1, cblksty); |
|
} |
|
|
|
|
|
/** mod fixed_quality */ |
|
static OPJ_FLOAT64 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) |
|
{ |
|
OPJ_FLOAT64 w1 = 1, w2, wmsedec; |
|
OPJ_ARG_NOT_USED(numcomps); |
|
|
|
if (mct_norms && (compno < mct_numcomps)) { |
|
w1 = mct_norms[compno]; |
|
} |
|
|
|
if (qmfbid == 1) { |
|
w2 = opj_dwt_getnorm(level, orient); |
|
} else { /* if (qmfbid == 0) */ |
|
const OPJ_INT32 log2_gain = (orient == 0) ? 0 : |
|
(orient == 3) ? 2 : 1; |
|
w2 = opj_dwt_getnorm_real(level, orient); |
|
/* Not sure this is right. But preserves past behaviour */ |
|
stepsize /= (1 << log2_gain); |
|
} |
|
|
|
wmsedec = w1 * w2 * stepsize * (1 << bpno); |
|
wmsedec *= wmsedec * nmsedec / 8192.0; |
|
|
|
return wmsedec; |
|
} |
|
|
|
static OPJ_BOOL opj_t1_allocate_buffers( |
|
opj_t1_t *t1, |
|
OPJ_UINT32 w, |
|
OPJ_UINT32 h) |
|
{ |
|
OPJ_UINT32 flagssize; |
|
OPJ_UINT32 flags_stride; |
|
|
|
/* No risk of overflow. Prior checks ensure those assert are met */ |
|
/* They are per the specification */ |
|
assert(w <= 1024); |
|
assert(h <= 1024); |
|
assert(w * h <= 4096); |
|
|
|
/* encoder uses tile buffer, so no need to allocate */ |
|
{ |
|
OPJ_UINT32 datasize = w * h; |
|
|
|
if (datasize > t1->datasize) { |
|
opj_aligned_free(t1->data); |
|
t1->data = (OPJ_INT32*) opj_aligned_malloc(datasize * sizeof(OPJ_INT32)); |
|
if (!t1->data) { |
|
/* FIXME event manager error callback */ |
|
return OPJ_FALSE; |
|
} |
|
t1->datasize = datasize; |
|
} |
|
/* memset first arg is declared to never be null by gcc */ |
|
if (t1->data != NULL) { |
|
memset(t1->data, 0, datasize * sizeof(OPJ_INT32)); |
|
} |
|
} |
|
|
|
flags_stride = w + 2U; /* can't be 0U */ |
|
|
|
flagssize = (h + 3U) / 4U + 2U; |
|
|
|
flagssize *= flags_stride; |
|
{ |
|
opj_flag_t* p; |
|
OPJ_UINT32 x; |
|
OPJ_UINT32 flags_height = (h + 3U) / 4U; |
|
|
|
if (flagssize > t1->flagssize) { |
|
|
|
opj_aligned_free(t1->flags); |
|
t1->flags = (opj_flag_t*) opj_aligned_malloc(flagssize * sizeof( |
|
opj_flag_t)); |
|
if (!t1->flags) { |
|
/* FIXME event manager error callback */ |
|
return OPJ_FALSE; |
|
} |
|
} |
|
t1->flagssize = flagssize; |
|
|
|
memset(t1->flags, 0, flagssize * sizeof(opj_flag_t)); |
|
|
|
p = &t1->flags[0]; |
|
for (x = 0; x < flags_stride; ++x) { |
|
/* magic value to hopefully stop any passes being interested in this entry */ |
|
*p++ = (T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3); |
|
} |
|
|
|
p = &t1->flags[((flags_height + 1) * flags_stride)]; |
|
for (x = 0; x < flags_stride; ++x) { |
|
/* magic value to hopefully stop any passes being interested in this entry */ |
|
*p++ = (T1_PI_0 | T1_PI_1 | T1_PI_2 | T1_PI_3); |
|
} |
|
|
|
if (h % 4) { |
|
OPJ_UINT32 v = 0; |
|
p = &t1->flags[((flags_height) * flags_stride)]; |
|
if (h % 4 == 1) { |
|
v |= T1_PI_1 | T1_PI_2 | T1_PI_3; |
|
} else if (h % 4 == 2) { |
|
v |= T1_PI_2 | T1_PI_3; |
|
} else if (h % 4 == 3) { |
|
v |= T1_PI_3; |
|
} |
|
for (x = 0; x < flags_stride; ++x) { |
|
*p++ = v; |
|
} |
|
} |
|
} |
|
|
|
t1->w = w; |
|
t1->h = h; |
|
|
|
return OPJ_TRUE; |
|
} |
|
|
|
/* ----------------------------------------------------------------------- */ |
|
|
|
/* ----------------------------------------------------------------------- */ |
|
/** |
|
* Creates a new Tier 1 handle |
|
* and initializes the look-up tables of the Tier-1 coder/decoder |
|
* @return a new T1 handle if successful, returns NULL otherwise |
|
*/ |
|
opj_t1_t* opj_t1_create(OPJ_BOOL isEncoder) |
|
{ |
|
opj_t1_t *l_t1 = 00; |
|
|
|
l_t1 = (opj_t1_t*) opj_calloc(1, sizeof(opj_t1_t)); |
|
if (!l_t1) { |
|
return 00; |
|
} |
|
|
|
l_t1->encoder = isEncoder; |
|
|
|
return l_t1; |
|
} |
|
|
|
|
|
/** |
|
* Destroys a previously created T1 handle |
|
* |
|
* @param p_t1 Tier 1 handle to destroy |
|
*/ |
|
void opj_t1_destroy(opj_t1_t *p_t1) |
|
{ |
|
if (! p_t1) { |
|
return; |
|
} |
|
|
|
if (p_t1->data) { |
|
opj_aligned_free(p_t1->data); |
|
p_t1->data = 00; |
|
} |
|
|
|
if (p_t1->flags) { |
|
opj_aligned_free(p_t1->flags); |
|
p_t1->flags = 00; |
|
} |
|
|
|
opj_free(p_t1->cblkdatabuffer); |
|
|
|
opj_free(p_t1); |
|
} |
|
|
|
typedef struct { |
|
OPJ_BOOL whole_tile_decoding; |
|
OPJ_UINT32 resno; |
|
opj_tcd_cblk_dec_t* cblk; |
|
opj_tcd_band_t* band; |
|
opj_tcd_tilecomp_t* tilec; |
|
opj_tccp_t* tccp; |
|
OPJ_BOOL mustuse_cblkdatabuffer; |
|
volatile OPJ_BOOL* pret; |
|
opj_event_mgr_t *p_manager; |
|
opj_mutex_t* p_manager_mutex; |
|
OPJ_BOOL check_pterm; |
|
} opj_t1_cblk_decode_processing_job_t; |
|
|
|
static void opj_t1_destroy_wrapper(void* t1) |
|
{ |
|
opj_t1_destroy((opj_t1_t*) t1); |
|
} |
|
|
|
static void opj_t1_clbl_decode_processor(void* user_data, opj_tls_t* tls) |
|
{ |
|
opj_tcd_cblk_dec_t* cblk; |
|
opj_tcd_band_t* band; |
|
opj_tcd_tilecomp_t* tilec; |
|
opj_tccp_t* tccp; |
|
OPJ_INT32* OPJ_RESTRICT datap; |
|
OPJ_UINT32 cblk_w, cblk_h; |
|
OPJ_INT32 x, y; |
|
OPJ_UINT32 i, j; |
|
opj_t1_cblk_decode_processing_job_t* job; |
|
opj_t1_t* t1; |
|
OPJ_UINT32 resno; |
|
OPJ_UINT32 tile_w; |
|
|
|
job = (opj_t1_cblk_decode_processing_job_t*) user_data; |
|
|
|
cblk = job->cblk; |
|
|
|
if (!job->whole_tile_decoding) { |
|
cblk_w = (OPJ_UINT32)(cblk->x1 - cblk->x0); |
|
cblk_h = (OPJ_UINT32)(cblk->y1 - cblk->y0); |
|
|
|
cblk->decoded_data = (OPJ_INT32*)opj_aligned_malloc(sizeof(OPJ_INT32) * |
|
cblk_w * cblk_h); |
|
if (cblk->decoded_data == NULL) { |
|
if (job->p_manager_mutex) { |
|
opj_mutex_lock(job->p_manager_mutex); |
|
} |
|
opj_event_msg(job->p_manager, EVT_ERROR, |
|
"Cannot allocate cblk->decoded_data\n"); |
|
if (job->p_manager_mutex) { |
|
opj_mutex_unlock(job->p_manager_mutex); |
|
} |
|
*(job->pret) = OPJ_FALSE; |
|
opj_free(job); |
|
return; |
|
} |
|
/* Zero-init required */ |
|
memset(cblk->decoded_data, 0, sizeof(OPJ_INT32) * cblk_w * cblk_h); |
|
} else if (cblk->decoded_data) { |
|
/* Not sure if that code path can happen, but better be */ |
|
/* safe than sorry */ |
|
opj_aligned_free(cblk->decoded_data); |
|
cblk->decoded_data = NULL; |
|
} |
|
|
|
resno = job->resno; |
|
band = job->band; |
|
tilec = job->tilec; |
|
tccp = job->tccp; |
|
tile_w = (OPJ_UINT32)(tilec->resolutions[tilec->minimum_num_resolutions - 1].x1 |
|
- |
|
tilec->resolutions[tilec->minimum_num_resolutions - 1].x0); |
|
|
|
if (!*(job->pret)) { |
|
opj_free(job); |
|
return; |
|
} |
|
|
|
t1 = (opj_t1_t*) opj_tls_get(tls, OPJ_TLS_KEY_T1); |
|
if (t1 == NULL) { |
|
t1 = opj_t1_create(OPJ_FALSE); |
|
if (t1 == NULL) { |
|
opj_event_msg(job->p_manager, EVT_ERROR, |
|
"Cannot allocate Tier 1 handle\n"); |
|
*(job->pret) = OPJ_FALSE; |
|
opj_free(job); |
|
return; |
|
} |
|
if (!opj_tls_set(tls, OPJ_TLS_KEY_T1, t1, opj_t1_destroy_wrapper)) { |
|
opj_event_msg(job->p_manager, EVT_ERROR, |
|
"Unable to set t1 handle as TLS\n"); |
|
opj_t1_destroy(t1); |
|
*(job->pret) = OPJ_FALSE; |
|
opj_free(job); |
|
return; |
|
} |
|
} |
|
t1->mustuse_cblkdatabuffer = job->mustuse_cblkdatabuffer; |
|
|
|
if ((tccp->cblksty & J2K_CCP_CBLKSTY_HT) != 0) { |
|
if (OPJ_FALSE == opj_t1_ht_decode_cblk( |
|
t1, |
|
cblk, |
|
band->bandno, |
|
(OPJ_UINT32)tccp->roishift, |
|
tccp->cblksty, |
|
job->p_manager, |
|
job->p_manager_mutex, |
|
job->check_pterm)) { |
|
*(job->pret) = OPJ_FALSE; |
|
opj_free(job); |
|
return; |
|
} |
|
} else { |
|
if (OPJ_FALSE == opj_t1_decode_cblk( |
|
t1, |
|
cblk, |
|
band->bandno, |
|
(OPJ_UINT32)tccp->roishift, |
|
tccp->cblksty, |
|
job->p_manager, |
|
job->p_manager_mutex, |
|
job->check_pterm)) { |
|
*(job->pret) = OPJ_FALSE; |
|
opj_free(job); |
|
return; |
|
} |
|
} |
|
|
|
x = cblk->x0 - band->x0; |
|
y = cblk->y0 - band->y0; |
|
if (band->bandno & 1) { |
|
opj_tcd_resolution_t* pres = &tilec->resolutions[resno - 1]; |
|
x += pres->x1 - pres->x0; |
|
} |
|
if (band->bandno & 2) { |
|
opj_tcd_resolution_t* pres = &tilec->resolutions[resno - 1]; |
|
y += pres->y1 - pres->y0; |
|
} |
|
|
|
datap = cblk->decoded_data ? cblk->decoded_data : t1->data; |
|
cblk_w = t1->w; |
|
cblk_h = t1->h; |
|
|
|
if (tccp->roishift) { |
|
if (tccp->roishift >= 31) { |
|
for (j = 0; j < cblk_h; ++j) { |
|
for (i = 0; i < cblk_w; ++i) { |
|
datap[(j * cblk_w) + i] = 0; |
|
} |
|
} |
|
} else { |
|
OPJ_INT32 thresh = 1 << tccp->roishift; |
|
for (j = 0; j < cblk_h; ++j) { |
|
for (i = 0; i < cblk_w; ++i) { |
|
OPJ_INT32 val = datap[(j * cblk_w) + i]; |
|
OPJ_INT32 mag = abs(val); |
|
if (mag >= thresh) { |
|
mag >>= tccp->roishift; |
|
datap[(j * cblk_w) + i] = val < 0 ? -mag : mag; |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
/* Both can be non NULL if for example decoding a full tile and then */ |
|
/* partially a tile. In which case partial decoding should be the */ |
|
/* priority */ |
|
assert((cblk->decoded_data != NULL) || (tilec->data != NULL)); |
|
|
|
if (cblk->decoded_data) { |
|
OPJ_UINT32 cblk_size = cblk_w * cblk_h; |
|
if (tccp->qmfbid == 1) { |
|
for (i = 0; i < cblk_size; ++i) { |
|
datap[i] /= 2; |
|
} |
|
} else { /* if (tccp->qmfbid == 0) */ |
|
const float stepsize = 0.5f * band->stepsize; |
|
i = 0; |
|
#ifdef __SSE2__ |
|
{ |
|
const __m128 xmm_stepsize = _mm_set1_ps(stepsize); |
|
for (; i < (cblk_size & ~15U); i += 16) { |
|
__m128 xmm0_data = _mm_cvtepi32_ps(_mm_load_si128((__m128i * const)( |
|
datap + 0))); |
|
__m128 xmm1_data = _mm_cvtepi32_ps(_mm_load_si128((__m128i * const)( |
|
datap + 4))); |
|
__m128 xmm2_data = _mm_cvtepi32_ps(_mm_load_si128((__m128i * const)( |
|
datap + 8))); |
|
__m128 xmm3_data = _mm_cvtepi32_ps(_mm_load_si128((__m128i * const)( |
|
datap + 12))); |
|
_mm_store_ps((float*)(datap + 0), _mm_mul_ps(xmm0_data, xmm_stepsize)); |
|
_mm_store_ps((float*)(datap + 4), _mm_mul_ps(xmm1_data, xmm_stepsize)); |
|
_mm_store_ps((float*)(datap + 8), _mm_mul_ps(xmm2_data, xmm_stepsize)); |
|
_mm_store_ps((float*)(datap + 12), _mm_mul_ps(xmm3_data, xmm_stepsize)); |
|
datap += 16; |
|
} |
|
} |
|
#endif |
|
for (; i < cblk_size; ++i) { |
|
OPJ_FLOAT32 tmp = ((OPJ_FLOAT32)(*datap)) * stepsize; |
|
memcpy(datap, &tmp, sizeof(tmp)); |
|
datap++; |
|
} |
|
} |
|
} else if (tccp->qmfbid == 1) { |
|
OPJ_INT32* OPJ_RESTRICT tiledp = &tilec->data[(OPJ_SIZE_T)y * tile_w + |
|
(OPJ_SIZE_T)x]; |
|
for (j = 0; j < cblk_h; ++j) { |
|
i = 0; |
|
for (; i < (cblk_w & ~(OPJ_UINT32)3U); i += 4U) { |
|
OPJ_INT32 tmp0 = datap[(j * cblk_w) + i + 0U]; |
|
OPJ_INT32 tmp1 = datap[(j * cblk_w) + i + 1U]; |
|
OPJ_INT32 tmp2 = datap[(j * cblk_w) + i + 2U]; |
|
OPJ_INT32 tmp3 = datap[(j * cblk_w) + i + 3U]; |
|
((OPJ_INT32*)tiledp)[(j * (OPJ_SIZE_T)tile_w) + i + 0U] = tmp0 / 2; |
|
((OPJ_INT32*)tiledp)[(j * (OPJ_SIZE_T)tile_w) + i + 1U] = tmp1 / 2; |
|
((OPJ_INT32*)tiledp)[(j * (OPJ_SIZE_T)tile_w) + i + 2U] = tmp2 / 2; |
|
((OPJ_INT32*)tiledp)[(j * (OPJ_SIZE_T)tile_w) + i + 3U] = tmp3 / 2; |
|
} |
|
for (; i < cblk_w; ++i) { |
|
OPJ_INT32 tmp = datap[(j * cblk_w) + i]; |
|
((OPJ_INT32*)tiledp)[(j * (OPJ_SIZE_T)tile_w) + i] = tmp / 2; |
|
} |
|
} |
|
} else { /* if (tccp->qmfbid == 0) */ |
|
const float stepsize = 0.5f * band->stepsize; |
|
OPJ_FLOAT32* OPJ_RESTRICT tiledp = (OPJ_FLOAT32*) &tilec->data[(OPJ_SIZE_T)y * |
|
tile_w + (OPJ_SIZE_T)x]; |
|
for (j = 0; j < cblk_h; ++j) { |
|
OPJ_FLOAT32* OPJ_RESTRICT tiledp2 = tiledp; |
|
for (i = 0; i < cblk_w; ++i) { |
|
OPJ_FLOAT32 tmp = (OPJ_FLOAT32) * datap * stepsize; |
|
*tiledp2 = tmp; |
|
datap++; |
|
tiledp2++; |
|
} |
|
tiledp += tile_w; |
|
} |
|
} |
|
|
|
opj_free(job); |
|
} |
|
|
|
|
|
void 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 |
|
) |
|
{ |
|
opj_thread_pool_t* tp = tcd->thread_pool; |
|
OPJ_UINT32 resno, bandno, precno, cblkno; |
|
|
|
#ifdef DEBUG_VERBOSE |
|
OPJ_UINT32 codeblocks_decoded = 0; |
|
printf("Enter opj_t1_decode_cblks()\n"); |
|
#endif |
|
|
|
for (resno = 0; resno < tilec->minimum_num_resolutions; ++resno) { |
|
opj_tcd_resolution_t* res = &tilec->resolutions[resno]; |
|
|
|
for (bandno = 0; bandno < res->numbands; ++bandno) { |
|
opj_tcd_band_t* OPJ_RESTRICT band = &res->bands[bandno]; |
|
|
|
for (precno = 0; precno < res->pw * res->ph; ++precno) { |
|
opj_tcd_precinct_t* precinct = &band->precincts[precno]; |
|
|
|
if (!opj_tcd_is_subband_area_of_interest(tcd, |
|
tilec->compno, |
|
resno, |
|
band->bandno, |
|
(OPJ_UINT32)precinct->x0, |
|
(OPJ_UINT32)precinct->y0, |
|
(OPJ_UINT32)precinct->x1, |
|
(OPJ_UINT32)precinct->y1)) { |
|
for (cblkno = 0; cblkno < precinct->cw * precinct->ch; ++cblkno) { |
|
opj_tcd_cblk_dec_t* cblk = &precinct->cblks.dec[cblkno]; |
|
if (cblk->decoded_data) { |
|
#ifdef DEBUG_VERBOSE |
|
printf("Discarding codeblock %d,%d at resno=%d, bandno=%d\n", |
|
cblk->x0, cblk->y0, resno, bandno); |
|
#endif |
|
opj_aligned_free(cblk->decoded_data); |
|
cblk->decoded_data = NULL; |
|
} |
|
} |
|
continue; |
|
} |
|
|
|
for (cblkno = 0; cblkno < precinct->cw * precinct->ch; ++cblkno) { |
|
opj_tcd_cblk_dec_t* cblk = &precinct->cblks.dec[cblkno]; |
|
opj_t1_cblk_decode_processing_job_t* job; |
|
|
|
if (!opj_tcd_is_subband_area_of_interest(tcd, |
|
tilec->compno, |
|
resno, |
|
band->bandno, |
|
(OPJ_UINT32)cblk->x0, |
|
(OPJ_UINT32)cblk->y0, |
|
(OPJ_UINT32)cblk->x1, |
|
(OPJ_UINT32)cblk->y1)) { |
|
if (cblk->decoded_data) { |
|
#ifdef DEBUG_VERBOSE |
|
printf("Discarding codeblock %d,%d at resno=%d, bandno=%d\n", |
|
cblk->x0, cblk->y0, resno, bandno); |
|
#endif |
|
opj_aligned_free(cblk->decoded_data); |
|
cblk->decoded_data = NULL; |
|
} |
|
continue; |
|
} |
|
|
|
if (!tcd->whole_tile_decoding) { |
|
OPJ_UINT32 cblk_w = (OPJ_UINT32)(cblk->x1 - cblk->x0); |
|
OPJ_UINT32 cblk_h = (OPJ_UINT32)(cblk->y1 - cblk->y0); |
|
if (cblk->decoded_data != NULL) { |
|
#ifdef DEBUG_VERBOSE |
|
printf("Reusing codeblock %d,%d at resno=%d, bandno=%d\n", |
|
cblk->x0, cblk->y0, resno, bandno); |
|
#endif |
|
continue; |
|
} |
|
if (cblk_w == 0 || cblk_h == 0) { |
|
continue; |
|
} |
|
#ifdef DEBUG_VERBOSE |
|
printf("Decoding codeblock %d,%d at resno=%d, bandno=%d\n", |
|
cblk->x0, cblk->y0, resno, bandno); |
|
#endif |
|
} |
|
|
|
job = (opj_t1_cblk_decode_processing_job_t*) opj_calloc(1, |
|
sizeof(opj_t1_cblk_decode_processing_job_t)); |
|
if (!job) { |
|
*pret = OPJ_FALSE; |
|
return; |
|
} |
|
job->whole_tile_decoding = tcd->whole_tile_decoding; |
|
job->resno = resno; |
|
job->cblk = cblk; |
|
job->band = band; |
|
job->tilec = tilec; |
|
job->tccp = tccp; |
|
job->pret = pret; |
|
job->p_manager_mutex = p_manager_mutex; |
|
job->p_manager = p_manager; |
|
job->check_pterm = check_pterm; |
|
job->mustuse_cblkdatabuffer = opj_thread_pool_get_thread_count(tp) > 1; |
|
opj_thread_pool_submit_job(tp, opj_t1_clbl_decode_processor, job); |
|
#ifdef DEBUG_VERBOSE |
|
codeblocks_decoded ++; |
|
#endif |
|
if (!(*pret)) { |
|
return; |
|
} |
|
} /* cblkno */ |
|
} /* precno */ |
|
} /* bandno */ |
|
} /* resno */ |
|
|
|
#ifdef DEBUG_VERBOSE |
|
printf("Leave opj_t1_decode_cblks(). Number decoded: %d\n", codeblocks_decoded); |
|
#endif |
|
return; |
|
} |
|
|
|
|
|
static OPJ_BOOL 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) |
|
{ |
|
opj_mqc_t *mqc = &(t1->mqc); /* MQC component */ |
|
|
|
OPJ_INT32 bpno_plus_one; |
|
OPJ_UINT32 passtype; |
|
OPJ_UINT32 segno, passno; |
|
OPJ_BYTE* cblkdata = NULL; |
|
OPJ_UINT32 cblkdataindex = 0; |
|
OPJ_BYTE type = T1_TYPE_MQ; /* BYPASS mode */ |
|
OPJ_INT32* original_t1_data = NULL; |
|
|
|
mqc->lut_ctxno_zc_orient = lut_ctxno_zc + (orient << 9); |
|
|
|
if (!opj_t1_allocate_buffers( |
|
t1, |
|
(OPJ_UINT32)(cblk->x1 - cblk->x0), |
|
(OPJ_UINT32)(cblk->y1 - cblk->y0))) { |
|
return OPJ_FALSE; |
|
} |
|
|
|
bpno_plus_one = (OPJ_INT32)(roishift + cblk->numbps); |
|
if (bpno_plus_one >= 31) { |
|
if (p_manager_mutex) { |
|
opj_mutex_lock(p_manager_mutex); |
|
} |
|
opj_event_msg(p_manager, EVT_WARNING, |
|
"opj_t1_decode_cblk(): unsupported bpno_plus_one = %d >= 31\n", |
|
bpno_plus_one); |
|
if (p_manager_mutex) { |
|
opj_mutex_unlock(p_manager_mutex); |
|
} |
|
return OPJ_FALSE; |
|
} |
|
passtype = 2; |
|
|
|
opj_mqc_resetstates(mqc); |
|
opj_mqc_setstate(mqc, T1_CTXNO_UNI, 0, 46); |
|
opj_mqc_setstate(mqc, T1_CTXNO_AGG, 0, 3); |
|
opj_mqc_setstate(mqc, T1_CTXNO_ZC, 0, 4); |
|
|
|
/* Even if we have a single chunk, in multi-threaded decoding */ |
|
/* the insertion of our synthetic marker might potentially override */ |
|
/* valid codestream of other codeblocks decoded in parallel. */ |
|
if (cblk->numchunks > 1 || t1->mustuse_cblkdatabuffer) { |
|
OPJ_UINT32 i; |
|
OPJ_UINT32 cblk_len; |
|
|
|
/* Compute whole codeblock length from chunk lengths */ |
|
cblk_len = 0; |
|
for (i = 0; i < cblk->numchunks; i++) { |
|
cblk_len += cblk->chunks[i].len; |
|
} |
|
|
|
/* Allocate temporary memory if needed */ |
|
if (cblk_len + OPJ_COMMON_CBLK_DATA_EXTRA > t1->cblkdatabuffersize) { |
|
cblkdata = (OPJ_BYTE*)opj_realloc(t1->cblkdatabuffer, |
|
cblk_len + OPJ_COMMON_CBLK_DATA_EXTRA); |
|
if (cblkdata == NULL) { |
|
return OPJ_FALSE; |
|
} |
|
t1->cblkdatabuffer = cblkdata; |
|
memset(t1->cblkdatabuffer + cblk_len, 0, OPJ_COMMON_CBLK_DATA_EXTRA); |
|
t1->cblkdatabuffersize = cblk_len + OPJ_COMMON_CBLK_DATA_EXTRA; |
|
} |
|
|
|
/* Concatenate all chunks */ |
|
cblkdata = t1->cblkdatabuffer; |
|
cblk_len = 0; |
|
for (i = 0; i < cblk->numchunks; i++) { |
|
memcpy(cblkdata + cblk_len, cblk->chunks[i].data, cblk->chunks[i].len); |
|
cblk_len += cblk->chunks[i].len; |
|
} |
|
} else if (cblk->numchunks == 1) { |
|
cblkdata = cblk->chunks[0].data; |
|
} else { |
|
/* Not sure if that can happen in practice, but avoid Coverity to */ |
|
/* think we will dereference a null cblkdta pointer */ |
|
return OPJ_TRUE; |
|
} |
|
|
|
/* For subtile decoding, directly decode in the decoded_data buffer of */ |
|
/* the code-block. Hack t1->data to point to it, and restore it later */ |
|
if (cblk->decoded_data) { |
|
original_t1_data = t1->data; |
|
t1->data = cblk->decoded_data; |
|
} |
|
|
|
for (segno = 0; segno < cblk->real_num_segs; ++segno) { |
|
opj_tcd_seg_t *seg = &cblk->segs[segno]; |
|
|
|
/* BYPASS mode */ |
|
type = ((bpno_plus_one <= ((OPJ_INT32)(cblk->numbps)) - 4) && (passtype < 2) && |
|
(cblksty & J2K_CCP_CBLKSTY_LAZY)) ? T1_TYPE_RAW : T1_TYPE_MQ; |
|
|
|
if (type == T1_TYPE_RAW) { |
|
opj_mqc_raw_init_dec(mqc, cblkdata + cblkdataindex, seg->len, |
|
OPJ_COMMON_CBLK_DATA_EXTRA); |
|
} else { |
|
opj_mqc_init_dec(mqc, cblkdata + cblkdataindex, seg->len, |
|
OPJ_COMMON_CBLK_DATA_EXTRA); |
|
} |
|
cblkdataindex += seg->len; |
|
|
|
for (passno = 0; (passno < seg->real_num_passes) && |
|
(bpno_plus_one >= 1); ++passno) { |
|
switch (passtype) { |
|
case 0: |
|
if (type == T1_TYPE_RAW) { |
|
opj_t1_dec_sigpass_raw(t1, bpno_plus_one, (OPJ_INT32)cblksty); |
|
} else { |
|
opj_t1_dec_sigpass_mqc(t1, bpno_plus_one, (OPJ_INT32)cblksty); |
|
} |
|
break; |
|
case 1: |
|
if (type == T1_TYPE_RAW) { |
|
opj_t1_dec_refpass_raw(t1, bpno_plus_one); |
|
} else { |
|
opj_t1_dec_refpass_mqc(t1, bpno_plus_one); |
|
} |
|
break; |
|
case 2: |
|
opj_t1_dec_clnpass(t1, bpno_plus_one, (OPJ_INT32)cblksty); |
|
break; |
|
} |
|
|
|
if ((cblksty & J2K_CCP_CBLKSTY_RESET) && type == T1_TYPE_MQ) { |
|
opj_mqc_resetstates(mqc); |
|
opj_mqc_setstate(mqc, T1_CTXNO_UNI, 0, 46); |
|
opj_mqc_setstate(mqc, T1_CTXNO_AGG, 0, 3); |
|
opj_mqc_setstate(mqc, T1_CTXNO_ZC, 0, 4); |
|
} |
|
if (++passtype == 3) { |
|
passtype = 0; |
|
bpno_plus_one--; |
|
} |
|
} |
|
|
|
opq_mqc_finish_dec(mqc); |
|
} |
|
|
|
if (check_pterm) { |
|
if (mqc->bp + 2 < mqc->end) { |
|
if (p_manager_mutex) { |
|
opj_mutex_lock(p_manager_mutex); |
|
} |
|
opj_event_msg(p_manager, EVT_WARNING, |
|
"PTERM check failure: %d remaining bytes in code block (%d used / %d)\n", |
|
(int)(mqc->end - mqc->bp) - 2, |
|
(int)(mqc->bp - mqc->start), |
|
(int)(mqc->end - mqc->start)); |
|
if (p_manager_mutex) { |
|
opj_mutex_unlock(p_manager_mutex); |
|
} |
|
} else if (mqc->end_of_byte_stream_counter > 2) { |
|
if (p_manager_mutex) { |
|
opj_mutex_lock(p_manager_mutex); |
|
} |
|
opj_event_msg(p_manager, EVT_WARNING, |
|
"PTERM check failure: %d synthetized 0xFF markers read\n", |
|
mqc->end_of_byte_stream_counter); |
|
if (p_manager_mutex) { |
|
opj_mutex_unlock(p_manager_mutex); |
|
} |
|
} |
|
} |
|
|
|
/* Restore original t1->data is needed */ |
|
if (cblk->decoded_data) { |
|
t1->data = original_t1_data; |
|
} |
|
|
|
return OPJ_TRUE; |
|
} |
|
|
|
|
|
typedef struct { |
|
OPJ_UINT32 compno; |
|
OPJ_UINT32 resno; |
|
opj_tcd_cblk_enc_t* cblk; |
|
opj_tcd_tile_t *tile; |
|
opj_tcd_band_t* band; |
|
opj_tcd_tilecomp_t* tilec; |
|
opj_tccp_t* tccp; |
|
const OPJ_FLOAT64 * mct_norms; |
|
OPJ_UINT32 mct_numcomps; |
|
volatile OPJ_BOOL* pret; |
|
opj_mutex_t* mutex; |
|
} opj_t1_cblk_encode_processing_job_t; |
|
|
|
/** Procedure to deal with a asynchronous code-block encoding job. |
|
* |
|
* @param user_data Pointer to a opj_t1_cblk_encode_processing_job_t* structure |
|
* @param tls TLS handle. |
|
*/ |
|
static void opj_t1_cblk_encode_processor(void* user_data, opj_tls_t* tls) |
|
{ |
|
opj_t1_cblk_encode_processing_job_t* job = |
|
(opj_t1_cblk_encode_processing_job_t*)user_data; |
|
opj_tcd_cblk_enc_t* cblk = job->cblk; |
|
const opj_tcd_band_t* band = job->band; |
|
const opj_tcd_tilecomp_t* tilec = job->tilec; |
|
const opj_tccp_t* tccp = job->tccp; |
|
const OPJ_UINT32 resno = job->resno; |
|
opj_t1_t* t1; |
|
const OPJ_UINT32 tile_w = (OPJ_UINT32)(tilec->x1 - tilec->x0); |
|
|
|
OPJ_INT32* OPJ_RESTRICT tiledp; |
|
OPJ_UINT32 cblk_w; |
|
OPJ_UINT32 cblk_h; |
|
OPJ_UINT32 i, j; |
|
|
|
OPJ_INT32 x = cblk->x0 - band->x0; |
|
OPJ_INT32 y = cblk->y0 - band->y0; |
|
|
|
if (!*(job->pret)) { |
|
opj_free(job); |
|
return; |
|
} |
|
|
|
t1 = (opj_t1_t*) opj_tls_get(tls, OPJ_TLS_KEY_T1); |
|
if (t1 == NULL) { |
|
t1 = opj_t1_create(OPJ_TRUE); /* OPJ_TRUE == T1 for encoding */ |
|
opj_tls_set(tls, OPJ_TLS_KEY_T1, t1, opj_t1_destroy_wrapper); |
|
} |
|
|
|
if (band->bandno & 1) { |
|
opj_tcd_resolution_t *pres = &tilec->resolutions[resno - 1]; |
|
x += pres->x1 - pres->x0; |
|
} |
|
if (band->bandno & 2) { |
|
opj_tcd_resolution_t *pres = &tilec->resolutions[resno - 1]; |
|
y += pres->y1 - pres->y0; |
|
} |
|
|
|
if (!opj_t1_allocate_buffers( |
|
t1, |
|
(OPJ_UINT32)(cblk->x1 - cblk->x0), |
|
(OPJ_UINT32)(cblk->y1 - cblk->y0))) { |
|
*(job->pret) = OPJ_FALSE; |
|
opj_free(job); |
|
return; |
|
} |
|
|
|
cblk_w = t1->w; |
|
cblk_h = t1->h; |
|
|
|
tiledp = &tilec->data[(OPJ_SIZE_T)y * tile_w + (OPJ_SIZE_T)x]; |
|
|
|
if (tccp->qmfbid == 1) { |
|
/* Do multiplication on unsigned type, even if the |
|
* underlying type is signed, to avoid potential |
|
* int overflow on large value (the output will be |
|
* incorrect in such situation, but whatever...) |
|
* This assumes complement-to-2 signed integer |
|
* representation |
|
* Fixes https://github.com/uclouvain/openjpeg/issues/1053 |
|
*/ |
|
OPJ_UINT32* OPJ_RESTRICT tiledp_u = (OPJ_UINT32*) tiledp; |
|
OPJ_UINT32* OPJ_RESTRICT t1data = (OPJ_UINT32*) t1->data; |
|
/* Change from "natural" order to "zigzag" order of T1 passes */ |
|
for (j = 0; j < (cblk_h & ~3U); j += 4) { |
|
for (i = 0; i < cblk_w; ++i) { |
|
t1data[0] = tiledp_u[(j + 0) * tile_w + i] << T1_NMSEDEC_FRACBITS; |
|
t1data[1] = tiledp_u[(j + 1) * tile_w + i] << T1_NMSEDEC_FRACBITS; |
|
t1data[2] = tiledp_u[(j + 2) * tile_w + i] << T1_NMSEDEC_FRACBITS; |
|
t1data[3] = tiledp_u[(j + 3) * tile_w + i] << T1_NMSEDEC_FRACBITS; |
|
t1data += 4; |
|
} |
|
} |
|
if (j < cblk_h) { |
|
for (i = 0; i < cblk_w; ++i) { |
|
OPJ_UINT32 k; |
|
for (k = j; k < cblk_h; k++) { |
|
t1data[0] = tiledp_u[k * tile_w + i] << T1_NMSEDEC_FRACBITS; |
|
t1data ++; |
|
} |
|
} |
|
} |
|
} else { /* if (tccp->qmfbid == 0) */ |
|
OPJ_FLOAT32* OPJ_RESTRICT tiledp_f = (OPJ_FLOAT32*) tiledp; |
|
OPJ_INT32* OPJ_RESTRICT t1data = t1->data; |
|
/* Change from "natural" order to "zigzag" order of T1 passes */ |
|
for (j = 0; j < (cblk_h & ~3U); j += 4) { |
|
for (i = 0; i < cblk_w; ++i) { |
|
t1data[0] = (OPJ_INT32)opj_lrintf((tiledp_f[(j + 0) * tile_w + i] / |
|
band->stepsize) * (1 << T1_NMSEDEC_FRACBITS)); |
|
t1data[1] = (OPJ_INT32)opj_lrintf((tiledp_f[(j + 1) * tile_w + i] / |
|
band->stepsize) * (1 << T1_NMSEDEC_FRACBITS)); |
|
t1data[2] = (OPJ_INT32)opj_lrintf((tiledp_f[(j + 2) * tile_w + i] / |
|
band->stepsize) * (1 << T1_NMSEDEC_FRACBITS)); |
|
t1data[3] = (OPJ_INT32)opj_lrintf((tiledp_f[(j + 3) * tile_w + i] / |
|
band->stepsize) * (1 << T1_NMSEDEC_FRACBITS)); |
|
t1data += 4; |
|
} |
|
} |
|
if (j < cblk_h) { |
|
for (i = 0; i < cblk_w; ++i) { |
|
OPJ_UINT32 k; |
|
for (k = j; k < cblk_h; k++) { |
|
t1data[0] = (OPJ_INT32)opj_lrintf((tiledp_f[k * tile_w + i] / band->stepsize) |
|
* (1 << T1_NMSEDEC_FRACBITS)); |
|
t1data ++; |
|
} |
|
} |
|
} |
|
} |
|
|
|
{ |
|
OPJ_FLOAT64 cumwmsedec = |
|
opj_t1_encode_cblk( |
|
t1, |
|
cblk, |
|
band->bandno, |
|
job->compno, |
|
tilec->numresolutions - 1 - resno, |
|
tccp->qmfbid, |
|
band->stepsize, |
|
tccp->cblksty, |
|
job->tile->numcomps, |
|
job->mct_norms, |
|
job->mct_numcomps); |
|
if (job->mutex) { |
|
opj_mutex_lock(job->mutex); |
|
} |
|
job->tile->distotile += cumwmsedec; |
|
if (job->mutex) { |
|
opj_mutex_unlock(job->mutex); |
|
} |
|
} |
|
|
|
opj_free(job); |
|
} |
|
|
|
|
|
OPJ_BOOL opj_t1_encode_cblks(opj_tcd_t* tcd, |
|
opj_tcd_tile_t *tile, |
|
opj_tcp_t *tcp, |
|
const OPJ_FLOAT64 * mct_norms, |
|
OPJ_UINT32 mct_numcomps |
|
) |
|
{ |
|
volatile OPJ_BOOL ret = OPJ_TRUE; |
|
opj_thread_pool_t* tp = tcd->thread_pool; |
|
OPJ_UINT32 compno, resno, bandno, precno, cblkno; |
|
opj_mutex_t* mutex = opj_mutex_create(); |
|
|
|
tile->distotile = 0; /* fixed_quality */ |
|
|
|
for (compno = 0; compno < tile->numcomps; ++compno) { |
|
opj_tcd_tilecomp_t* tilec = &tile->comps[compno]; |
|
opj_tccp_t* tccp = &tcp->tccps[compno]; |
|
|
|
for (resno = 0; resno < tilec->numresolutions; ++resno) { |
|
opj_tcd_resolution_t *res = &tilec->resolutions[resno]; |
|
|
|
for (bandno = 0; bandno < res->numbands; ++bandno) { |
|
opj_tcd_band_t* OPJ_RESTRICT band = &res->bands[bandno]; |
|
|
|
/* Skip empty bands */ |
|
if (opj_tcd_is_band_empty(band)) { |
|
continue; |
|
} |
|
for (precno = 0; precno < res->pw * res->ph; ++precno) { |
|
opj_tcd_precinct_t *prc = &band->precincts[precno]; |
|
|
|
for (cblkno = 0; cblkno < prc->cw * prc->ch; ++cblkno) { |
|
opj_tcd_cblk_enc_t* cblk = &prc->cblks.enc[cblkno]; |
|
|
|
opj_t1_cblk_encode_processing_job_t* job = |
|
(opj_t1_cblk_encode_processing_job_t*) opj_calloc(1, |
|
sizeof(opj_t1_cblk_encode_processing_job_t)); |
|
if (!job) { |
|
ret = OPJ_FALSE; |
|
goto end; |
|
} |
|
job->compno = compno; |
|
job->tile = tile; |
|
job->resno = resno; |
|
job->cblk = cblk; |
|
job->band = band; |
|
job->tilec = tilec; |
|
job->tccp = tccp; |
|
job->mct_norms = mct_norms; |
|
job->mct_numcomps = mct_numcomps; |
|
job->pret = &ret; |
|
job->mutex = mutex; |
|
opj_thread_pool_submit_job(tp, opj_t1_cblk_encode_processor, job); |
|
|
|
} /* cblkno */ |
|
} /* precno */ |
|
} /* bandno */ |
|
} /* resno */ |
|
} /* compno */ |
|
|
|
end: |
|
opj_thread_pool_wait_completion(tcd->thread_pool, 0); |
|
if (mutex) { |
|
opj_mutex_destroy(mutex); |
|
} |
|
|
|
return ret; |
|
} |
|
|
|
/* Returns whether the pass (bpno, passtype) is terminated */ |
|
static int opj_t1_enc_is_term_pass(opj_tcd_cblk_enc_t* cblk, |
|
OPJ_UINT32 cblksty, |
|
OPJ_INT32 bpno, |
|
OPJ_UINT32 passtype) |
|
{ |
|
/* Is it the last cleanup pass ? */ |
|
if (passtype == 2 && bpno == 0) { |
|
return OPJ_TRUE; |
|
} |
|
|
|
if (cblksty & J2K_CCP_CBLKSTY_TERMALL) { |
|
return OPJ_TRUE; |
|
} |
|
|
|
if ((cblksty & J2K_CCP_CBLKSTY_LAZY)) { |
|
/* For bypass arithmetic bypass, terminate the 4th cleanup pass */ |
|
if ((bpno == ((OPJ_INT32)cblk->numbps - 4)) && (passtype == 2)) { |
|
return OPJ_TRUE; |
|
} |
|
/* and beyond terminate all the magnitude refinement passes (in raw) */ |
|
/* and cleanup passes (in MQC) */ |
|
if ((bpno < ((OPJ_INT32)(cblk->numbps) - 4)) && (passtype > 0)) { |
|
return OPJ_TRUE; |
|
} |
|
} |
|
|
|
return OPJ_FALSE; |
|
} |
|
|
|
|
|
/** mod fixed_quality */ |
|
static OPJ_FLOAT64 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, |
|
const OPJ_FLOAT64 * mct_norms, |
|
OPJ_UINT32 mct_numcomps) |
|
{ |
|
OPJ_FLOAT64 cumwmsedec = 0.0; |
|
|
|
opj_mqc_t *mqc = &(t1->mqc); /* MQC component */ |
|
|
|
OPJ_UINT32 passno; |
|
OPJ_INT32 bpno; |
|
OPJ_UINT32 passtype; |
|
OPJ_INT32 nmsedec = 0; |
|
OPJ_INT32 max; |
|
OPJ_UINT32 i, j; |
|
OPJ_BYTE type = T1_TYPE_MQ; |
|
OPJ_FLOAT64 tempwmsedec; |
|
OPJ_INT32* datap; |
|
|
|
#ifdef EXTRA_DEBUG |
|
printf("encode_cblk(x=%d,y=%d,x1=%d,y1=%d,orient=%d,compno=%d,level=%d\n", |
|
cblk->x0, cblk->y0, cblk->x1, cblk->y1, orient, compno, level); |
|
#endif |
|
|
|
mqc->lut_ctxno_zc_orient = lut_ctxno_zc + (orient << 9); |
|
|
|
max = 0; |
|
datap = t1->data; |
|
for (j = 0; j < t1->h; ++j) { |
|
const OPJ_UINT32 w = t1->w; |
|
for (i = 0; i < w; ++i, ++datap) { |
|
OPJ_INT32 tmp = *datap; |
|
if (tmp < 0) { |
|
OPJ_UINT32 tmp_unsigned; |
|
max = opj_int_max(max, -tmp); |
|
tmp_unsigned = opj_to_smr(tmp); |
|
memcpy(datap, &tmp_unsigned, sizeof(OPJ_INT32)); |
|
} else { |
|
max = opj_int_max(max, tmp); |
|
} |
|
} |
|
} |
|
|
|
cblk->numbps = max ? (OPJ_UINT32)((opj_int_floorlog2(max) + 1) - |
|
T1_NMSEDEC_FRACBITS) : 0; |
|
if (cblk->numbps == 0) { |
|
cblk->totalpasses = 0; |
|
return cumwmsedec; |
|
} |
|
|
|
bpno = (OPJ_INT32)(cblk->numbps - 1); |
|
passtype = 2; |
|
|
|
opj_mqc_resetstates(mqc); |
|
opj_mqc_setstate(mqc, T1_CTXNO_UNI, 0, 46); |
|
opj_mqc_setstate(mqc, T1_CTXNO_AGG, 0, 3); |
|
opj_mqc_setstate(mqc, T1_CTXNO_ZC, 0, 4); |
|
opj_mqc_init_enc(mqc, cblk->data); |
|
|
|
for (passno = 0; bpno >= 0; ++passno) { |
|
opj_tcd_pass_t *pass = &cblk->passes[passno]; |
|
type = ((bpno < ((OPJ_INT32)(cblk->numbps) - 4)) && (passtype < 2) && |
|
(cblksty & J2K_CCP_CBLKSTY_LAZY)) ? T1_TYPE_RAW : T1_TYPE_MQ; |
|
|
|
/* If the previous pass was terminating, we need to reset the encoder */ |
|
if (passno > 0 && cblk->passes[passno - 1].term) { |
|
if (type == T1_TYPE_RAW) { |
|
opj_mqc_bypass_init_enc(mqc); |
|
} else { |
|
opj_mqc_restart_init_enc(mqc); |
|
} |
|
} |
|
|
|
switch (passtype) { |
|
case 0: |
|
opj_t1_enc_sigpass(t1, bpno, &nmsedec, type, cblksty); |
|
break; |
|
case 1: |
|
opj_t1_enc_refpass(t1, bpno, &nmsedec, type); |
|
break; |
|
case 2: |
|
opj_t1_enc_clnpass(t1, bpno, &nmsedec, cblksty); |
|
/* code switch SEGMARK (i.e. SEGSYM) */ |
|
if (cblksty & J2K_CCP_CBLKSTY_SEGSYM) { |
|
opj_mqc_segmark_enc(mqc); |
|
} |
|
break; |
|
} |
|
|
|
/* fixed_quality */ |
|
tempwmsedec = opj_t1_getwmsedec(nmsedec, compno, level, orient, bpno, qmfbid, |
|
stepsize, numcomps, mct_norms, mct_numcomps) ; |
|
cumwmsedec += tempwmsedec; |
|
pass->distortiondec = cumwmsedec; |
|
|
|
if (opj_t1_enc_is_term_pass(cblk, cblksty, bpno, passtype)) { |
|
/* If it is a terminated pass, terminate it */ |
|
if (type == T1_TYPE_RAW) { |
|
opj_mqc_bypass_flush_enc(mqc, cblksty & J2K_CCP_CBLKSTY_PTERM); |
|
} else { |
|
if (cblksty & J2K_CCP_CBLKSTY_PTERM) { |
|
opj_mqc_erterm_enc(mqc); |
|
} else { |
|
opj_mqc_flush(mqc); |
|
} |
|
} |
|
pass->term = 1; |
|
pass->rate = opj_mqc_numbytes(mqc); |
|
} else { |
|
/* Non terminated pass */ |
|
OPJ_UINT32 rate_extra_bytes; |
|
if (type == T1_TYPE_RAW) { |
|
rate_extra_bytes = opj_mqc_bypass_get_extra_bytes( |
|
mqc, (cblksty & J2K_CCP_CBLKSTY_PTERM)); |
|
} else { |
|
rate_extra_bytes = 3; |
|
} |
|
pass->term = 0; |
|
pass->rate = opj_mqc_numbytes(mqc) + rate_extra_bytes; |
|
} |
|
|
|
if (++passtype == 3) { |
|
passtype = 0; |
|
bpno--; |
|
} |
|
|
|
/* Code-switch "RESET" */ |
|
if (cblksty & J2K_CCP_CBLKSTY_RESET) { |
|
opj_mqc_reset_enc(mqc); |
|
} |
|
} |
|
|
|
cblk->totalpasses = passno; |
|
|
|
if (cblk->totalpasses) { |
|
/* Make sure that pass rates are increasing */ |
|
OPJ_UINT32 last_pass_rate = opj_mqc_numbytes(mqc); |
|
for (passno = cblk->totalpasses; passno > 0;) { |
|
opj_tcd_pass_t *pass = &cblk->passes[--passno]; |
|
if (pass->rate > last_pass_rate) { |
|
pass->rate = last_pass_rate; |
|
} else { |
|
last_pass_rate = pass->rate; |
|
} |
|
} |
|
} |
|
|
|
for (passno = 0; passno < cblk->totalpasses; passno++) { |
|
opj_tcd_pass_t *pass = &cblk->passes[passno]; |
|
|
|
/* Prevent generation of FF as last data byte of a pass*/ |
|
/* For terminating passes, the flushing procedure ensured this already */ |
|
assert(pass->rate > 0); |
|
if (cblk->data[pass->rate - 1] == 0xFF) { |
|
pass->rate--; |
|
} |
|
pass->len = pass->rate - (passno == 0 ? 0 : cblk->passes[passno - 1].rate); |
|
} |
|
|
|
#ifdef EXTRA_DEBUG |
|
printf(" len=%d\n", (cblk->totalpasses) ? opj_mqc_numbytes(mqc) : 0); |
|
|
|
/* Check that there not 0xff >=0x90 sequences */ |
|
if (cblk->totalpasses) { |
|
OPJ_UINT32 i; |
|
OPJ_UINT32 len = opj_mqc_numbytes(mqc); |
|
for (i = 1; i < len; ++i) { |
|
if (cblk->data[i - 1] == 0xff && cblk->data[i] >= 0x90) { |
|
printf("0xff %02x at offset %d\n", cblk->data[i], i - 1); |
|
abort(); |
|
} |
|
} |
|
} |
|
#endif |
|
|
|
return cumwmsedec; |
|
}
|
|
|