Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * JPEG 2000 encoder and decoder common functions | ||
3 | * Copyright (c) 2007 Kamil Nowosad | ||
4 | * Copyright (c) 2013 Nicolas Bertrand <nicoinattendu@gmail.com> | ||
5 | * | ||
6 | * This file is part of FFmpeg. | ||
7 | * | ||
8 | * FFmpeg is free software; you can redistribute it and/or | ||
9 | * modify it under the terms of the GNU Lesser General Public | ||
10 | * License as published by the Free Software Foundation; either | ||
11 | * version 2.1 of the License, or (at your option) any later version. | ||
12 | * | ||
13 | * FFmpeg is distributed in the hope that it will be useful, | ||
14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
16 | * Lesser General Public License for more details. | ||
17 | * | ||
18 | * You should have received a copy of the GNU Lesser General Public | ||
19 | * License along with FFmpeg; if not, write to the Free Software | ||
20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
21 | */ | ||
22 | |||
23 | /** | ||
24 | * @file | ||
25 | * JPEG 2000 image encoder and decoder common functions | ||
26 | */ | ||
27 | |||
28 | #include "libavutil/attributes.h" | ||
29 | #include "libavutil/avassert.h" | ||
30 | #include "libavutil/common.h" | ||
31 | #include "libavutil/imgutils.h" | ||
32 | #include "libavutil/mem.h" | ||
33 | #include "libavutil/thread.h" | ||
34 | #include "avcodec.h" | ||
35 | #include "internal.h" | ||
36 | #include "jpeg2000.h" | ||
37 | |||
38 | #define SHL(a, n) ((n) >= 0 ? (a) << (n) : (a) >> -(n)) | ||
39 | |||
40 | /* tag tree routines */ | ||
41 | |||
42 | 972706 | static int32_t tag_tree_size(int w, int h) | |
43 | { | ||
44 | 972706 | int64_t res = 0; | |
45 |
4/4✓ Branch 0 taken 652046 times.
✓ Branch 1 taken 1079072 times.
✓ Branch 2 taken 106366 times.
✓ Branch 3 taken 972706 times.
|
1731118 | while (w > 1 || h > 1) { |
46 | 758412 | res += w * (int64_t)h; | |
47 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 758412 times.
|
758412 | av_assert0(res + 1 < INT32_MAX); |
48 | 758412 | w = (w + 1) >> 1; | |
49 | 758412 | h = (h + 1) >> 1; | |
50 | } | ||
51 | 972706 | return (int32_t)(res + 1); | |
52 | } | ||
53 | |||
54 | /* allocate the memory for tag tree */ | ||
55 | 346106 | static Jpeg2000TgtNode *ff_jpeg2000_tag_tree_init(int w, int h) | |
56 | { | ||
57 | 346106 | int pw = w, ph = h; | |
58 | Jpeg2000TgtNode *res, *t, *t2; | ||
59 | int32_t tt_size; | ||
60 | |||
61 | 346106 | tt_size = tag_tree_size(w, h); | |
62 | |||
63 | 346106 | t = res = av_calloc(tt_size, sizeof(*t)); | |
64 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 346106 times.
|
346106 | if (!res) |
65 | ✗ | return NULL; | |
66 | |||
67 |
4/4✓ Branch 0 taken 223046 times.
✓ Branch 1 taken 382272 times.
✓ Branch 2 taken 36166 times.
✓ Branch 3 taken 346106 times.
|
605318 | while (w > 1 || h > 1) { |
68 | int i, j; | ||
69 | 259212 | pw = w; | |
70 | 259212 | ph = h; | |
71 | |||
72 | 259212 | w = (w + 1) >> 1; | |
73 | 259212 | h = (h + 1) >> 1; | |
74 | 259212 | t2 = t + pw * ph; | |
75 | |||
76 |
2/2✓ Branch 0 taken 653834 times.
✓ Branch 1 taken 259212 times.
|
913046 | for (i = 0; i < ph; i++) |
77 |
2/2✓ Branch 0 taken 2245640 times.
✓ Branch 1 taken 653834 times.
|
2899474 | for (j = 0; j < pw; j++) |
78 | 2245640 | t[i * pw + j].parent = &t2[(i >> 1) * w + (j >> 1)]; | |
79 | |||
80 | 259212 | t = t2; | |
81 | } | ||
82 | 346106 | t[0].parent = NULL; | |
83 | 346106 | return res; | |
84 | } | ||
85 | |||
86 | 626600 | void ff_tag_tree_zero(Jpeg2000TgtNode *t, int w, int h, int val) | |
87 | { | ||
88 | 626600 | int i, siz = tag_tree_size(w, h); | |
89 | |||
90 |
2/2✓ Branch 0 taken 4940000 times.
✓ Branch 1 taken 626600 times.
|
5566600 | for (i = 0; i < siz; i++) { |
91 | 4940000 | t[i].val = val; | |
92 | 4940000 | t[i].temp_val = 0; | |
93 | 4940000 | t[i].vis = 0; | |
94 | } | ||
95 | 626600 | } | |
96 | |||
97 | uint8_t ff_jpeg2000_sigctxno_lut[256][4]; | ||
98 | |||
99 | 50176 | static int getsigctxno(int flag, int bandno) | |
100 | { | ||
101 | int h, v, d; | ||
102 | |||
103 | 50176 | h = ((flag & JPEG2000_T1_SIG_E) ? 1 : 0) + | |
104 | 50176 | ((flag & JPEG2000_T1_SIG_W) ? 1 : 0); | |
105 | 50176 | v = ((flag & JPEG2000_T1_SIG_N) ? 1 : 0) + | |
106 | 50176 | ((flag & JPEG2000_T1_SIG_S) ? 1 : 0); | |
107 | 50176 | d = ((flag & JPEG2000_T1_SIG_NE) ? 1 : 0) + | |
108 | 50176 | ((flag & JPEG2000_T1_SIG_NW) ? 1 : 0) + | |
109 | 50176 | ((flag & JPEG2000_T1_SIG_SE) ? 1 : 0) + | |
110 | 50176 | ((flag & JPEG2000_T1_SIG_SW) ? 1 : 0); | |
111 | |||
112 |
2/2✓ Branch 0 taken 37632 times.
✓ Branch 1 taken 12544 times.
|
50176 | if (bandno < 3) { |
113 |
2/2✓ Branch 0 taken 12544 times.
✓ Branch 1 taken 25088 times.
|
37632 | if (bandno == 1) |
114 | 12544 | FFSWAP(int, h, v); | |
115 |
2/2✓ Branch 0 taken 9408 times.
✓ Branch 1 taken 28224 times.
|
37632 | if (h == 2) return 8; |
116 |
2/2✓ Branch 0 taken 18816 times.
✓ Branch 1 taken 9408 times.
|
28224 | if (h == 1) { |
117 |
2/2✓ Branch 0 taken 14112 times.
✓ Branch 1 taken 4704 times.
|
18816 | if (v >= 1) return 7; |
118 |
2/2✓ Branch 0 taken 4410 times.
✓ Branch 1 taken 294 times.
|
4704 | if (d >= 1) return 6; |
119 | 294 | return 5; | |
120 | } | ||
121 |
2/2✓ Branch 0 taken 2352 times.
✓ Branch 1 taken 7056 times.
|
9408 | if (v == 2) return 4; |
122 |
2/2✓ Branch 0 taken 4704 times.
✓ Branch 1 taken 2352 times.
|
7056 | if (v == 1) return 3; |
123 |
2/2✓ Branch 0 taken 1617 times.
✓ Branch 1 taken 735 times.
|
2352 | if (d >= 2) return 2; |
124 |
2/2✓ Branch 0 taken 588 times.
✓ Branch 1 taken 147 times.
|
735 | if (d == 1) return 1; |
125 | } else { | ||
126 |
2/2✓ Branch 0 taken 3920 times.
✓ Branch 1 taken 8624 times.
|
12544 | if (d >= 3) return 8; |
127 |
2/2✓ Branch 0 taken 4704 times.
✓ Branch 1 taken 3920 times.
|
8624 | if (d == 2) { |
128 |
2/2✓ Branch 0 taken 4410 times.
✓ Branch 1 taken 294 times.
|
4704 | if (h+v >= 1) return 7; |
129 | 294 | return 6; | |
130 | } | ||
131 |
2/2✓ Branch 0 taken 3136 times.
✓ Branch 1 taken 784 times.
|
3920 | if (d == 1) { |
132 |
2/2✓ Branch 0 taken 2156 times.
✓ Branch 1 taken 980 times.
|
3136 | if (h+v >= 2) return 5; |
133 |
2/2✓ Branch 0 taken 784 times.
✓ Branch 1 taken 196 times.
|
980 | if (h+v == 1) return 4; |
134 | 196 | return 3; | |
135 | } | ||
136 |
2/2✓ Branch 0 taken 539 times.
✓ Branch 1 taken 245 times.
|
784 | if (h+v >= 2) return 2; |
137 |
2/2✓ Branch 0 taken 196 times.
✓ Branch 1 taken 49 times.
|
245 | if (h+v == 1) return 1; |
138 | } | ||
139 | 196 | return 0; | |
140 | } | ||
141 | |||
142 | uint8_t ff_jpeg2000_sgnctxno_lut[16][16], ff_jpeg2000_xorbit_lut[16][16]; | ||
143 | |||
144 | static const int contribtab[3][3] = { { 0, -1, 1 }, { -1, -1, 0 }, { 1, 0, 1 } }; | ||
145 | static const int ctxlbltab[3][3] = { { 13, 12, 11 }, { 10, 9, 10 }, { 11, 12, 13 } }; | ||
146 | static const int xorbittab[3][3] = { { 1, 1, 1 }, { 1, 0, 0 }, { 0, 0, 0 } }; | ||
147 | |||
148 | 12544 | static int getsgnctxno(int flag, uint8_t *xorbit) | |
149 | { | ||
150 | int vcontrib, hcontrib; | ||
151 | |||
152 |
4/4✓ Branch 0 taken 6272 times.
✓ Branch 1 taken 6272 times.
✓ Branch 2 taken 3136 times.
✓ Branch 3 taken 3136 times.
|
12544 | hcontrib = contribtab[flag & JPEG2000_T1_SIG_E ? flag & JPEG2000_T1_SGN_E ? 1 : 2 : 0] |
153 |
4/4✓ Branch 0 taken 6272 times.
✓ Branch 1 taken 6272 times.
✓ Branch 2 taken 3136 times.
✓ Branch 3 taken 3136 times.
|
12544 | [flag & JPEG2000_T1_SIG_W ? flag & JPEG2000_T1_SGN_W ? 1 : 2 : 0] + 1; |
154 |
4/4✓ Branch 0 taken 6272 times.
✓ Branch 1 taken 6272 times.
✓ Branch 2 taken 3136 times.
✓ Branch 3 taken 3136 times.
|
12544 | vcontrib = contribtab[flag & JPEG2000_T1_SIG_S ? flag & JPEG2000_T1_SGN_S ? 1 : 2 : 0] |
155 |
4/4✓ Branch 0 taken 6272 times.
✓ Branch 1 taken 6272 times.
✓ Branch 2 taken 3136 times.
✓ Branch 3 taken 3136 times.
|
12544 | [flag & JPEG2000_T1_SIG_N ? flag & JPEG2000_T1_SGN_N ? 1 : 2 : 0] + 1; |
156 | 12544 | *xorbit = xorbittab[hcontrib][vcontrib]; | |
157 | |||
158 | 12544 | return ctxlbltab[hcontrib][vcontrib]; | |
159 | } | ||
160 | |||
161 | 49 | static void av_cold jpeg2000_init_tier1_luts(void) | |
162 | { | ||
163 | int i, j; | ||
164 |
2/2✓ Branch 0 taken 12544 times.
✓ Branch 1 taken 49 times.
|
12593 | for (i = 0; i < 256; i++) |
165 |
2/2✓ Branch 0 taken 50176 times.
✓ Branch 1 taken 12544 times.
|
62720 | for (j = 0; j < 4; j++) |
166 | 50176 | ff_jpeg2000_sigctxno_lut[i][j] = getsigctxno(i, j); | |
167 |
2/2✓ Branch 0 taken 784 times.
✓ Branch 1 taken 49 times.
|
833 | for (i = 0; i < 16; i++) |
168 |
2/2✓ Branch 0 taken 12544 times.
✓ Branch 1 taken 784 times.
|
13328 | for (j = 0; j < 16; j++) |
169 | 12544 | ff_jpeg2000_sgnctxno_lut[i][j] = | |
170 | 12544 | getsgnctxno(i + (j << 8), &ff_jpeg2000_xorbit_lut[i][j]); | |
171 | 49 | } | |
172 | |||
173 | 81 | void av_cold ff_jpeg2000_init_tier1_luts(void) | |
174 | { | ||
175 | static AVOnce init_static_once = AV_ONCE_INIT; | ||
176 | 81 | ff_thread_once(&init_static_once, jpeg2000_init_tier1_luts); | |
177 | 81 | } | |
178 | |||
179 | 264927998 | void ff_jpeg2000_set_significance(Jpeg2000T1Context *t1, int x, int y, | |
180 | int negative) | ||
181 | { | ||
182 | 264927998 | x++; | |
183 | 264927998 | y++; | |
184 | 264927998 | t1->flags[(y) * t1->stride + x] |= JPEG2000_T1_SIG; | |
185 |
2/2✓ Branch 0 taken 129758455 times.
✓ Branch 1 taken 135169543 times.
|
264927998 | if (negative) { |
186 | 129758455 | t1->flags[(y) * t1->stride + x + 1] |= JPEG2000_T1_SIG_W | JPEG2000_T1_SGN_W; | |
187 | 129758455 | t1->flags[(y) * t1->stride + x - 1] |= JPEG2000_T1_SIG_E | JPEG2000_T1_SGN_E; | |
188 | 129758455 | t1->flags[(y + 1) * t1->stride + x] |= JPEG2000_T1_SIG_N | JPEG2000_T1_SGN_N; | |
189 | 129758455 | t1->flags[(y - 1) * t1->stride + x] |= JPEG2000_T1_SIG_S | JPEG2000_T1_SGN_S; | |
190 | } else { | ||
191 | 135169543 | t1->flags[(y) * t1->stride + x + 1] |= JPEG2000_T1_SIG_W; | |
192 | 135169543 | t1->flags[(y) * t1->stride + x - 1] |= JPEG2000_T1_SIG_E; | |
193 | 135169543 | t1->flags[(y + 1) * t1->stride + x] |= JPEG2000_T1_SIG_N; | |
194 | 135169543 | t1->flags[(y - 1) * t1->stride + x] |= JPEG2000_T1_SIG_S; | |
195 | } | ||
196 | 264927998 | t1->flags[(y + 1) * t1->stride + x + 1] |= JPEG2000_T1_SIG_NW; | |
197 | 264927998 | t1->flags[(y + 1) * t1->stride + x - 1] |= JPEG2000_T1_SIG_NE; | |
198 | 264927998 | t1->flags[(y - 1) * t1->stride + x + 1] |= JPEG2000_T1_SIG_SW; | |
199 | 264927998 | t1->flags[(y - 1) * t1->stride + x - 1] |= JPEG2000_T1_SIG_SE; | |
200 | 264927998 | } | |
201 | |||
202 | // static const uint8_t lut_gain[2][4] = { { 0, 0, 0, 0 }, { 0, 1, 1, 2 } }; (unused) | ||
203 | |||
204 | 171874 | static void init_band_stepsize(AVCodecContext *avctx, | |
205 | Jpeg2000Band *band, | ||
206 | Jpeg2000CodingStyle *codsty, | ||
207 | Jpeg2000QuantStyle *qntsty, | ||
208 | int bandno, int gbandno, int reslevelno, | ||
209 | int cbps) | ||
210 | { | ||
211 | /* TODO: Implementation of quantization step not finished, | ||
212 | * see ISO/IEC 15444-1:2002 E.1 and A.6.4. */ | ||
213 |
2/3✓ Branch 0 taken 133866 times.
✓ Branch 1 taken 38008 times.
✗ Branch 2 not taken.
|
171874 | switch (qntsty->quantsty) { |
214 | uint8_t gain; | ||
215 | 133866 | case JPEG2000_QSTY_NONE: | |
216 | /* TODO: to verify. No quantization in this case */ | ||
217 | 133866 | band->f_stepsize = 1; | |
218 | 133866 | break; | |
219 | 38008 | case JPEG2000_QSTY_SI: | |
220 | /*TODO: Compute formula to implement. */ | ||
221 | // numbps = cbps + | ||
222 | // lut_gain[codsty->transform == FF_DWT53][bandno + (reslevelno > 0)]; | ||
223 | // band->f_stepsize = SHL(2048 + qntsty->mant[gbandno], | ||
224 | // 2 + numbps - qntsty->expn[gbandno]); | ||
225 | // break; | ||
226 | case JPEG2000_QSTY_SE: | ||
227 | /* Exponent quantization step. | ||
228 | * Formula: | ||
229 | * delta_b = 2 ^ (R_b - expn_b) * (1 + (mant_b / 2 ^ 11)) | ||
230 | * R_b = R_I + log2 (gain_b ) | ||
231 | * see ISO/IEC 15444-1:2002 E.1.1 eqn. E-3 and E-4 */ | ||
232 | 38008 | gain = cbps; | |
233 | 38008 | band->f_stepsize = ff_exp2fi(gain - qntsty->expn[gbandno]); | |
234 | 38008 | band->f_stepsize *= qntsty->mant[gbandno] / 2048.0 + 1.0; | |
235 | 38008 | break; | |
236 | ✗ | default: | |
237 | ✗ | band->f_stepsize = 0; | |
238 | ✗ | av_log(avctx, AV_LOG_ERROR, "Unknown quantization format\n"); | |
239 | ✗ | break; | |
240 | } | ||
241 |
2/2✓ Branch 0 taken 38008 times.
✓ Branch 1 taken 133866 times.
|
171874 | if (codsty->transform != FF_DWT53) { |
242 | 38008 | int lband = 0; | |
243 |
3/3✓ Branch 0 taken 24004 times.
✓ Branch 1 taken 12002 times.
✓ Branch 2 taken 2002 times.
|
38008 | switch (bandno + (reslevelno > 0)) { |
244 | 24004 | case 1: | |
245 | case 2: | ||
246 | 24004 | band->f_stepsize *= F_LFTG_X * 2; | |
247 | 24004 | lband = 1; | |
248 | 24004 | break; | |
249 | 12002 | case 3: | |
250 | 12002 | band->f_stepsize *= F_LFTG_X * F_LFTG_X * 4; | |
251 | 12002 | break; | |
252 | } | ||
253 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 38008 times.
|
38008 | if (codsty->transform == FF_DWT97) { |
254 | ✗ | band->f_stepsize *= pow(F_LFTG_K, 2*(codsty->nreslevels2decode - reslevelno) + lband - 2); | |
255 | } | ||
256 | } | ||
257 | |||
258 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 171874 times.
|
171874 | if (band->f_stepsize > (INT_MAX >> 15)) { |
259 | ✗ | band->f_stepsize = 0; | |
260 | ✗ | av_log(avctx, AV_LOG_ERROR, "stepsize out of range\n"); | |
261 | } | ||
262 | |||
263 | 171874 | band->i_stepsize = band->f_stepsize * (1 << 15); | |
264 | |||
265 | /* FIXME: In OpenJPEG code stepsize = stepsize * 0.5. Why? | ||
266 | * If not set output of entropic decoder is not correct. */ | ||
267 |
2/2✓ Branch 1 taken 168663 times.
✓ Branch 2 taken 3211 times.
|
171874 | if (!av_codec_is_encoder(avctx->codec)) |
268 | 168663 | band->f_stepsize *= 0.5; | |
269 | 171874 | } | |
270 | |||
271 | 173053 | static int init_prec(AVCodecContext *avctx, | |
272 | Jpeg2000Band *band, | ||
273 | Jpeg2000ResLevel *reslevel, | ||
274 | Jpeg2000Component *comp, | ||
275 | Jpeg2000CodingStyle *codsty, | ||
276 | int precno, int bandno, int reslevelno, | ||
277 | int log2_band_prec_width, | ||
278 | int log2_band_prec_height) | ||
279 | { | ||
280 | 173053 | Jpeg2000Prec *prec = band->prec + precno; | |
281 | int nb_codeblocks, cblkno; | ||
282 | |||
283 | 173053 | prec->decoded_layers = 0; | |
284 | |||
285 | /* TODO: Explain formula for JPEG200 DCINEMA. */ | ||
286 | /* TODO: Verify with previous count of codeblocks per band */ | ||
287 | |||
288 | /* Compute P_x0 */ | ||
289 | 173053 | prec->coord[0][0] = ((reslevel->coord[0][0] >> reslevel->log2_prec_width) + precno % reslevel->num_precincts_x) * | |
290 | (1 << log2_band_prec_width); | ||
291 | |||
292 | /* Compute P_y0 */ | ||
293 | 173053 | prec->coord[1][0] = ((reslevel->coord[1][0] >> reslevel->log2_prec_height) + precno / reslevel->num_precincts_x) * | |
294 | (1 << log2_band_prec_height); | ||
295 | |||
296 | /* Compute P_x1 */ | ||
297 | 173053 | prec->coord[0][1] = prec->coord[0][0] + | |
298 | 173053 | (1 << log2_band_prec_width); | |
299 | 173053 | prec->coord[0][0] = FFMAX(prec->coord[0][0], band->coord[0][0]); | |
300 | 173053 | prec->coord[0][1] = FFMIN(prec->coord[0][1], band->coord[0][1]); | |
301 | |||
302 | /* Compute P_y1 */ | ||
303 | 173053 | prec->coord[1][1] = prec->coord[1][0] + | |
304 | 173053 | (1 << log2_band_prec_height); | |
305 | 173053 | prec->coord[1][0] = FFMAX(prec->coord[1][0], band->coord[1][0]); | |
306 | 173053 | prec->coord[1][1] = FFMIN(prec->coord[1][1], band->coord[1][1]); | |
307 | |||
308 | 173053 | prec->nb_codeblocks_width = | |
309 | 173053 | ff_jpeg2000_ceildivpow2(prec->coord[0][1], | |
310 | 173053 | band->log2_cblk_width) | |
311 | 173053 | - (prec->coord[0][0] >> band->log2_cblk_width); | |
312 | 173053 | prec->nb_codeblocks_height = | |
313 | 173053 | ff_jpeg2000_ceildivpow2(prec->coord[1][1], | |
314 | 173053 | band->log2_cblk_height) | |
315 | 173053 | - (prec->coord[1][0] >> band->log2_cblk_height); | |
316 | |||
317 | |||
318 | /* Tag trees initialization */ | ||
319 | 173053 | prec->cblkincl = | |
320 | 173053 | ff_jpeg2000_tag_tree_init(prec->nb_codeblocks_width, | |
321 | prec->nb_codeblocks_height); | ||
322 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 173053 times.
|
173053 | if (!prec->cblkincl) |
323 | ✗ | return AVERROR(ENOMEM); | |
324 | |||
325 | 173053 | prec->zerobits = | |
326 | 173053 | ff_jpeg2000_tag_tree_init(prec->nb_codeblocks_width, | |
327 | prec->nb_codeblocks_height); | ||
328 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 173053 times.
|
173053 | if (!prec->zerobits) |
329 | ✗ | return AVERROR(ENOMEM); | |
330 | |||
331 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 173053 times.
|
173053 | if (prec->nb_codeblocks_width * (uint64_t)prec->nb_codeblocks_height > INT_MAX) { |
332 | ✗ | prec->cblk = NULL; | |
333 | ✗ | return AVERROR(ENOMEM); | |
334 | } | ||
335 | 173053 | nb_codeblocks = prec->nb_codeblocks_width * prec->nb_codeblocks_height; | |
336 | 173053 | prec->cblk = av_calloc(nb_codeblocks, sizeof(*prec->cblk)); | |
337 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 173053 times.
|
173053 | if (!prec->cblk) |
338 | ✗ | return AVERROR(ENOMEM); | |
339 |
2/2✓ Branch 0 taken 947421 times.
✓ Branch 1 taken 173053 times.
|
1120474 | for (cblkno = 0; cblkno < nb_codeblocks; cblkno++) { |
340 | 947421 | Jpeg2000Cblk *cblk = prec->cblk + cblkno; | |
341 | int Cx0, Cy0; | ||
342 | |||
343 | /* Compute coordinates of codeblocks */ | ||
344 | /* Compute Cx0*/ | ||
345 | 947421 | Cx0 = ((prec->coord[0][0]) >> band->log2_cblk_width) << band->log2_cblk_width; | |
346 | 947421 | Cx0 = Cx0 + ((cblkno % prec->nb_codeblocks_width) << band->log2_cblk_width); | |
347 | 947421 | cblk->coord[0][0] = FFMAX(Cx0, prec->coord[0][0]); | |
348 | |||
349 | /* Compute Cy0*/ | ||
350 | 947421 | Cy0 = ((prec->coord[1][0]) >> band->log2_cblk_height) << band->log2_cblk_height; | |
351 | 947421 | Cy0 = Cy0 + ((cblkno / prec->nb_codeblocks_width) << band->log2_cblk_height); | |
352 | 947421 | cblk->coord[1][0] = FFMAX(Cy0, prec->coord[1][0]); | |
353 | |||
354 | /* Compute Cx1 */ | ||
355 | 947421 | cblk->coord[0][1] = FFMIN(Cx0 + (1 << band->log2_cblk_width), | |
356 | prec->coord[0][1]); | ||
357 | |||
358 | /* Compute Cy1 */ | ||
359 | 947421 | cblk->coord[1][1] = FFMIN(Cy0 + (1 << band->log2_cblk_height), | |
360 | prec->coord[1][1]); | ||
361 | /* Update code-blocks coordinates according sub-band position */ | ||
362 |
2/2✓ Branch 0 taken 625194 times.
✓ Branch 1 taken 322227 times.
|
947421 | if ((bandno + !!reslevelno) & 1) { |
363 | 625194 | cblk->coord[0][0] += comp->reslevel[reslevelno-1].coord[0][1] - | |
364 | 625194 | comp->reslevel[reslevelno-1].coord[0][0]; | |
365 | 625194 | cblk->coord[0][1] += comp->reslevel[reslevelno-1].coord[0][1] - | |
366 | 625194 | comp->reslevel[reslevelno-1].coord[0][0]; | |
367 | } | ||
368 |
2/2✓ Branch 0 taken 625369 times.
✓ Branch 1 taken 322052 times.
|
947421 | if ((bandno + !!reslevelno) & 2) { |
369 | 625369 | cblk->coord[1][0] += comp->reslevel[reslevelno-1].coord[1][1] - | |
370 | 625369 | comp->reslevel[reslevelno-1].coord[1][0]; | |
371 | 625369 | cblk->coord[1][1] += comp->reslevel[reslevelno-1].coord[1][1] - | |
372 | 625369 | comp->reslevel[reslevelno-1].coord[1][0]; | |
373 | } | ||
374 | |||
375 | 947421 | cblk->lblock = 3; | |
376 | 947421 | cblk->length = 0; | |
377 | 947421 | cblk->npasses = 0; | |
378 |
2/2✓ Branch 1 taken 18070 times.
✓ Branch 2 taken 929351 times.
|
947421 | if (av_codec_is_encoder(avctx->codec)) { |
379 | 18070 | cblk->layers = av_calloc(codsty->nlayers, sizeof(*cblk->layers)); | |
380 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18070 times.
|
18070 | if (!cblk->layers) |
381 | ✗ | return AVERROR(ENOMEM); | |
382 | } | ||
383 | } | ||
384 | |||
385 | 173053 | return 0; | |
386 | } | ||
387 | |||
388 | 171874 | static int init_band(AVCodecContext *avctx, | |
389 | Jpeg2000ResLevel *reslevel, | ||
390 | Jpeg2000Component *comp, | ||
391 | Jpeg2000CodingStyle *codsty, | ||
392 | Jpeg2000QuantStyle *qntsty, | ||
393 | int bandno, int gbandno, int reslevelno, | ||
394 | int cbps, int dx, int dy) | ||
395 | { | ||
396 | 171874 | Jpeg2000Band *band = reslevel->band + bandno; | |
397 | uint8_t log2_band_prec_width, log2_band_prec_height; | ||
398 | 171874 | int declvl = codsty->nreslevels - reslevelno; // N_L -r see ISO/IEC 15444-1:2002 B.5 | |
399 | int precno; | ||
400 | int nb_precincts; | ||
401 | int i, j, ret; | ||
402 | |||
403 | 171874 | init_band_stepsize(avctx, band, codsty, qntsty, bandno, gbandno, reslevelno, cbps); | |
404 | |||
405 | /* computation of tbx_0, tbx_1, tby_0, tby_1 | ||
406 | * see ISO/IEC 15444-1:2002 B.5 eq. B-15 and tbl B.1 | ||
407 | * codeblock width and height is computed for | ||
408 | * DCI JPEG 2000 codeblock_width = codeblock_width = 32 = 2 ^ 5 */ | ||
409 |
2/2✓ Branch 0 taken 9421 times.
✓ Branch 1 taken 162453 times.
|
171874 | if (reslevelno == 0) { |
410 | /* for reslevelno = 0, only one band, x0_b = y0_b = 0 */ | ||
411 |
2/2✓ Branch 0 taken 18842 times.
✓ Branch 1 taken 9421 times.
|
28263 | for (i = 0; i < 2; i++) |
412 |
2/2✓ Branch 0 taken 37684 times.
✓ Branch 1 taken 18842 times.
|
56526 | for (j = 0; j < 2; j++) |
413 | 37684 | band->coord[i][j] = | |
414 | 37684 | ff_jpeg2000_ceildivpow2(comp->coord_o[i][j], | |
415 | declvl - 1); | ||
416 | 9421 | log2_band_prec_width = reslevel->log2_prec_width; | |
417 | 9421 | log2_band_prec_height = reslevel->log2_prec_height; | |
418 | /* see ISO/IEC 15444-1:2002 eq. B-17 and eq. B-15 */ | ||
419 | 9421 | band->log2_cblk_width = FFMIN(codsty->log2_cblk_width, | |
420 | reslevel->log2_prec_width); | ||
421 | 9421 | band->log2_cblk_height = FFMIN(codsty->log2_cblk_height, | |
422 | reslevel->log2_prec_height); | ||
423 | } else { | ||
424 | /* 3 bands x0_b = 1 y0_b = 0; x0_b = 0 y0_b = 1; x0_b = y0_b = 1 */ | ||
425 | /* x0_b and y0_b are computed with ((bandno + 1 >> i) & 1) */ | ||
426 |
2/2✓ Branch 0 taken 324906 times.
✓ Branch 1 taken 162453 times.
|
487359 | for (i = 0; i < 2; i++) |
427 |
2/2✓ Branch 0 taken 649812 times.
✓ Branch 1 taken 324906 times.
|
974718 | for (j = 0; j < 2; j++) |
428 | /* Formula example for tbx_0 = ceildiv((tcx_0 - 2 ^ (declvl - 1) * x0_b) / declvl) */ | ||
429 | 649812 | band->coord[i][j] = | |
430 | 649812 | ff_jpeg2000_ceildivpow2(comp->coord_o[i][j] - | |
431 | 649812 | (((bandno + 1 >> i) & 1LL) << declvl - 1), | |
432 | declvl); | ||
433 | /* TODO: Manage case of 3 band offsets here or | ||
434 | * in coding/decoding function? */ | ||
435 | |||
436 | /* see ISO/IEC 15444-1:2002 eq. B-17 and eq. B-15 */ | ||
437 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 162453 times.
|
162453 | band->log2_cblk_width = FFMIN(codsty->log2_cblk_width, |
438 | reslevel->log2_prec_width - 1); | ||
439 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 162453 times.
|
162453 | band->log2_cblk_height = FFMIN(codsty->log2_cblk_height, |
440 | reslevel->log2_prec_height - 1); | ||
441 | |||
442 | 162453 | log2_band_prec_width = reslevel->log2_prec_width - 1; | |
443 | 162453 | log2_band_prec_height = reslevel->log2_prec_height - 1; | |
444 | } | ||
445 | |||
446 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 171874 times.
|
171874 | if (reslevel->num_precincts_x * (uint64_t)reslevel->num_precincts_y > INT_MAX) { |
447 | ✗ | band->prec = NULL; | |
448 | ✗ | return AVERROR(ENOMEM); | |
449 | } | ||
450 | 171874 | nb_precincts = reslevel->num_precincts_x * reslevel->num_precincts_y; | |
451 | 171874 | band->prec = av_calloc(nb_precincts, sizeof(*band->prec)); | |
452 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 171874 times.
|
171874 | if (!band->prec) |
453 | ✗ | return AVERROR(ENOMEM); | |
454 | |||
455 |
2/2✓ Branch 0 taken 173053 times.
✓ Branch 1 taken 171874 times.
|
344927 | for (precno = 0; precno < nb_precincts; precno++) { |
456 | 173053 | ret = init_prec(avctx, band, reslevel, comp, codsty, | |
457 | precno, bandno, reslevelno, | ||
458 | log2_band_prec_width, log2_band_prec_height); | ||
459 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 173053 times.
|
173053 | if (ret < 0) |
460 | ✗ | return ret; | |
461 | } | ||
462 | |||
463 | 171874 | return 0; | |
464 | } | ||
465 | |||
466 | 9421 | int ff_jpeg2000_init_component(Jpeg2000Component *comp, | |
467 | Jpeg2000CodingStyle *codsty, | ||
468 | Jpeg2000QuantStyle *qntsty, | ||
469 | int cbps, int dx, int dy, | ||
470 | AVCodecContext *avctx) | ||
471 | { | ||
472 | 9421 | int reslevelno, bandno, gbandno = 0, ret, i, j; | |
473 | uint32_t csize; | ||
474 | |||
475 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9421 times.
|
9421 | if (codsty->nreslevels2decode <= 0) { |
476 | ✗ | av_log(avctx, AV_LOG_ERROR, "nreslevels2decode %d invalid or uninitialized\n", codsty->nreslevels2decode); | |
477 | ✗ | return AVERROR_INVALIDDATA; | |
478 | } | ||
479 | |||
480 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9421 times.
|
9421 | if (ret = ff_jpeg2000_dwt_init(&comp->dwt, comp->coord, |
481 | 9421 | codsty->nreslevels2decode - 1, | |
482 | 9421 | codsty->transform)) | |
483 | ✗ | return ret; | |
484 | |||
485 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9421 times.
|
9421 | if (av_image_check_size(comp->coord[0][1] - comp->coord[0][0], |
486 | 9421 | comp->coord[1][1] - comp->coord[1][0], 0, avctx)) | |
487 | ✗ | return AVERROR_INVALIDDATA; | |
488 | 9421 | csize = (comp->coord[0][1] - comp->coord[0][0]) * | |
489 | 9421 | (comp->coord[1][1] - comp->coord[1][0]); | |
490 |
1/2✓ Branch 0 taken 9421 times.
✗ Branch 1 not taken.
|
9421 | if (comp->coord[0][1] - comp->coord[0][0] > 32768 || |
491 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9421 times.
|
9421 | comp->coord[1][1] - comp->coord[1][0] > 32768) { |
492 | ✗ | av_log(avctx, AV_LOG_ERROR, "component size too large\n"); | |
493 | ✗ | return AVERROR_PATCHWELCOME; | |
494 | } | ||
495 | |||
496 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9421 times.
|
9421 | if (codsty->transform == FF_DWT97) { |
497 | ✗ | csize += AV_INPUT_BUFFER_PADDING_SIZE / sizeof(*comp->f_data); | |
498 | ✗ | comp->i_data = NULL; | |
499 | ✗ | comp->f_data = av_calloc(csize, sizeof(*comp->f_data)); | |
500 | ✗ | if (!comp->f_data) | |
501 | ✗ | return AVERROR(ENOMEM); | |
502 | } else { | ||
503 | 9421 | csize += AV_INPUT_BUFFER_PADDING_SIZE / sizeof(*comp->i_data); | |
504 | 9421 | comp->f_data = NULL; | |
505 | 9421 | comp->i_data = av_calloc(csize, sizeof(*comp->i_data)); | |
506 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9421 times.
|
9421 | if (!comp->i_data) |
507 | ✗ | return AVERROR(ENOMEM); | |
508 | } | ||
509 | 9421 | comp->reslevel = av_calloc(codsty->nreslevels, sizeof(*comp->reslevel)); | |
510 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9421 times.
|
9421 | if (!comp->reslevel) |
511 | ✗ | return AVERROR(ENOMEM); | |
512 | /* LOOP on resolution levels */ | ||
513 |
2/2✓ Branch 0 taken 63572 times.
✓ Branch 1 taken 9421 times.
|
72993 | for (reslevelno = 0; reslevelno < codsty->nreslevels; reslevelno++) { |
514 | 63572 | int declvl = codsty->nreslevels - reslevelno; // N_L -r see ISO/IEC 15444-1:2002 B.5 | |
515 | 63572 | Jpeg2000ResLevel *reslevel = comp->reslevel + reslevelno; | |
516 | |||
517 | /* Compute borders for each resolution level. | ||
518 | * Computation of trx_0, trx_1, try_0 and try_1. | ||
519 | * see ISO/IEC 15444-1:2002 eq. B.5 and B-14 */ | ||
520 |
2/2✓ Branch 0 taken 127144 times.
✓ Branch 1 taken 63572 times.
|
190716 | for (i = 0; i < 2; i++) |
521 |
2/2✓ Branch 0 taken 254288 times.
✓ Branch 1 taken 127144 times.
|
381432 | for (j = 0; j < 2; j++) |
522 | 254288 | reslevel->coord[i][j] = | |
523 | 254288 | ff_jpeg2000_ceildivpow2(comp->coord_o[i][j], declvl - 1); | |
524 | // update precincts size: 2^n value | ||
525 | 63572 | reslevel->log2_prec_width = codsty->log2_prec_widths[reslevelno]; | |
526 | 63572 | reslevel->log2_prec_height = codsty->log2_prec_heights[reslevelno]; | |
527 | |||
528 | /* Number of bands for each resolution level */ | ||
529 |
2/2✓ Branch 0 taken 9421 times.
✓ Branch 1 taken 54151 times.
|
63572 | if (reslevelno == 0) |
530 | 9421 | reslevel->nbands = 1; | |
531 | else | ||
532 | 54151 | reslevel->nbands = 3; | |
533 | |||
534 | /* Number of precincts which span the tile for resolution level reslevelno | ||
535 | * see B.6 in ISO/IEC 15444-1:2002 eq. B-16 | ||
536 | * num_precincts_x = |- trx_1 / 2 ^ log2_prec_width) -| - (trx_0 / 2 ^ log2_prec_width) | ||
537 | * num_precincts_y = |- try_1 / 2 ^ log2_prec_width) -| - (try_0 / 2 ^ log2_prec_width) | ||
538 | * for Dcinema profiles in JPEG 2000 | ||
539 | * num_precincts_x = |- trx_1 / 2 ^ log2_prec_width) -| | ||
540 | * num_precincts_y = |- try_1 / 2 ^ log2_prec_width) -| */ | ||
541 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 63572 times.
|
63572 | if (reslevel->coord[0][1] == reslevel->coord[0][0]) |
542 | ✗ | reslevel->num_precincts_x = 0; | |
543 | else | ||
544 | 63572 | reslevel->num_precincts_x = | |
545 | 63572 | ff_jpeg2000_ceildivpow2(reslevel->coord[0][1], | |
546 | 63572 | reslevel->log2_prec_width) - | |
547 | 63572 | (reslevel->coord[0][0] >> reslevel->log2_prec_width); | |
548 | |||
549 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 63572 times.
|
63572 | if (reslevel->coord[1][1] == reslevel->coord[1][0]) |
550 | ✗ | reslevel->num_precincts_y = 0; | |
551 | else | ||
552 | 63572 | reslevel->num_precincts_y = | |
553 | 63572 | ff_jpeg2000_ceildivpow2(reslevel->coord[1][1], | |
554 | 63572 | reslevel->log2_prec_height) - | |
555 | 63572 | (reslevel->coord[1][0] >> reslevel->log2_prec_height); | |
556 | |||
557 | 63572 | reslevel->band = av_calloc(reslevel->nbands, sizeof(*reslevel->band)); | |
558 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 63572 times.
|
63572 | if (!reslevel->band) |
559 | ✗ | return AVERROR(ENOMEM); | |
560 | |||
561 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 63572 times.
|
63572 | if (reslevel->num_precincts_x * (uint64_t)reslevel->num_precincts_y * reslevel->nbands > avctx->max_pixels / sizeof(*reslevel->band->prec)) |
562 | ✗ | return AVERROR(ENOMEM); | |
563 | |||
564 |
2/2✓ Branch 0 taken 171874 times.
✓ Branch 1 taken 63572 times.
|
235446 | for (bandno = 0; bandno < reslevel->nbands; bandno++, gbandno++) { |
565 | 171874 | ret = init_band(avctx, reslevel, | |
566 | comp, codsty, qntsty, | ||
567 | bandno, gbandno, reslevelno, | ||
568 | cbps, dx, dy); | ||
569 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 171874 times.
|
171874 | if (ret < 0) |
570 | ✗ | return ret; | |
571 | } | ||
572 | } | ||
573 | 9421 | return 0; | |
574 | } | ||
575 | |||
576 | 8450 | void ff_jpeg2000_reinit(Jpeg2000Component *comp, Jpeg2000CodingStyle *codsty) | |
577 | { | ||
578 | int reslevelno, bandno, cblkno, precno; | ||
579 |
2/2✓ Branch 0 taken 59150 times.
✓ Branch 1 taken 8450 times.
|
67600 | for (reslevelno = 0; reslevelno < codsty->nreslevels; reslevelno++) { |
580 | 59150 | Jpeg2000ResLevel *rlevel = comp->reslevel + reslevelno; | |
581 |
2/2✓ Branch 0 taken 160550 times.
✓ Branch 1 taken 59150 times.
|
219700 | for (bandno = 0; bandno < rlevel->nbands; bandno++) { |
582 | 160550 | Jpeg2000Band *band = rlevel->band + bandno; | |
583 |
2/2✓ Branch 0 taken 160550 times.
✓ Branch 1 taken 160550 times.
|
321100 | for(precno = 0; precno < rlevel->num_precincts_x * rlevel->num_precincts_y; precno++) { |
584 | 160550 | Jpeg2000Prec *prec = band->prec + precno; | |
585 | 160550 | ff_tag_tree_zero(prec->zerobits, prec->nb_codeblocks_width, prec->nb_codeblocks_height, 0); | |
586 | 160550 | ff_tag_tree_zero(prec->cblkincl, prec->nb_codeblocks_width, prec->nb_codeblocks_height, 0); | |
587 |
2/2✓ Branch 0 taken 903500 times.
✓ Branch 1 taken 160550 times.
|
1064050 | for (cblkno = 0; cblkno < prec->nb_codeblocks_width * prec->nb_codeblocks_height; cblkno++) { |
588 | 903500 | Jpeg2000Cblk *cblk = prec->cblk + cblkno; | |
589 | 903500 | cblk->length = 0; | |
590 | 903500 | cblk->lblock = 3; | |
591 | } | ||
592 | } | ||
593 | } | ||
594 | } | ||
595 | 8450 | } | |
596 | |||
597 | 10400 | void ff_jpeg2000_cleanup(Jpeg2000Component *comp, Jpeg2000CodingStyle *codsty) | |
598 | { | ||
599 | int reslevelno, bandno, precno; | ||
600 | 10400 | for (reslevelno = 0; | |
601 |
4/4✓ Branch 0 taken 72993 times.
✓ Branch 1 taken 979 times.
✓ Branch 2 taken 63572 times.
✓ Branch 3 taken 9421 times.
|
73972 | comp->reslevel && reslevelno < codsty->nreslevels; |
602 | 63572 | reslevelno++) { | |
603 | Jpeg2000ResLevel *reslevel; | ||
604 | |||
605 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 63572 times.
|
63572 | if (!comp->reslevel) |
606 | ✗ | continue; | |
607 | |||
608 | 63572 | reslevel = comp->reslevel + reslevelno; | |
609 |
2/2✓ Branch 0 taken 171874 times.
✓ Branch 1 taken 63572 times.
|
235446 | for (bandno = 0; bandno < reslevel->nbands; bandno++) { |
610 | Jpeg2000Band *band; | ||
611 | |||
612 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 171874 times.
|
171874 | if (!reslevel->band) |
613 | ✗ | continue; | |
614 | |||
615 | 171874 | band = reslevel->band + bandno; | |
616 |
2/2✓ Branch 0 taken 173053 times.
✓ Branch 1 taken 171874 times.
|
344927 | for (precno = 0; precno < reslevel->num_precincts_x * reslevel->num_precincts_y; precno++) { |
617 |
1/2✓ Branch 0 taken 173053 times.
✗ Branch 1 not taken.
|
173053 | if (band->prec) { |
618 | 173053 | Jpeg2000Prec *prec = band->prec + precno; | |
619 | 173053 | int nb_code_blocks = prec->nb_codeblocks_height * prec->nb_codeblocks_width; | |
620 | |||
621 | 173053 | av_freep(&prec->zerobits); | |
622 | 173053 | av_freep(&prec->cblkincl); | |
623 |
1/2✓ Branch 0 taken 173053 times.
✗ Branch 1 not taken.
|
173053 | if (prec->cblk) { |
624 | int cblkno; | ||
625 |
2/2✓ Branch 0 taken 947421 times.
✓ Branch 1 taken 173053 times.
|
1120474 | for (cblkno = 0; cblkno < nb_code_blocks; cblkno ++) { |
626 | 947421 | Jpeg2000Cblk *cblk = &prec->cblk[cblkno]; | |
627 | 947421 | av_freep(&cblk->data); | |
628 | 947421 | av_freep(&cblk->passes); | |
629 | 947421 | av_freep(&cblk->lengthinc); | |
630 | 947421 | av_freep(&cblk->data_start); | |
631 | 947421 | av_freep(&cblk->layers); | |
632 | } | ||
633 | 173053 | av_freep(&prec->cblk); | |
634 | } | ||
635 | } | ||
636 | } | ||
637 | |||
638 | 171874 | av_freep(&band->prec); | |
639 | } | ||
640 | 63572 | av_freep(&reslevel->band); | |
641 | } | ||
642 | |||
643 | 10400 | ff_dwt_destroy(&comp->dwt); | |
644 | 10400 | av_freep(&comp->reslevel); | |
645 | 10400 | av_freep(&comp->i_data); | |
646 | 10400 | av_freep(&comp->f_data); | |
647 | 10400 | } | |
648 |