1 /*
2 * Copyright (C) 2004-2010 Michael Niedermayer <michaelni@gmx.at>
3 * Copyright (C) 2008 David Conrad
4 * Copyright (C) 2015 Open Broadcast Systems Ltd.
5 * Author (C) 2015 Rostislav Pehlivanov <atomnuker@gmail.com>
6 *
7 * This file is part of FFmpeg.
8 *
9 * FFmpeg is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
13 *
14 * FFmpeg is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with FFmpeg; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 */
23
24 #if defined(TEMPLATE_8bit)
25
26 # define RENAME(N) N ## _8bit
27 # define TYPE int16_t
28 # undef TEMPLATE_8bit
29
30 #elif defined(TEMPLATE_10bit)
31
32 # define RENAME(N) N ## _10bit
33 # define TYPE int32_t
34 # undef TEMPLATE_10bit
35
36 #elif defined(TEMPLATE_12bit)
37
38 # define RENAME(N) N ## _12bit
39 # define TYPE int32_t
40 # undef TEMPLATE_12bit
41
42 #endif
43
RENAME(vertical_compose53iL0)44 static void RENAME(vertical_compose53iL0)(uint8_t *_b0, uint8_t *_b1, uint8_t *_b2,
45 int width)
46 {
47 int i;
48 TYPE *b0 = (TYPE *)_b0;
49 TYPE *b1 = (TYPE *)_b1;
50 TYPE *b2 = (TYPE *)_b2;
51 for (i = 0; i < width; i++)
52 b1[i] -= (unsigned)((int)(b0[i] + (unsigned)b2[i] + 2) >> 2);
53 }
54
RENAME(interleave)55 static av_always_inline void RENAME(interleave)(TYPE *dst, TYPE *src0, TYPE *src1, int w2,
56 int add, int shift)
57 {
58 int i;
59 for (i = 0; i < w2; i++) {
60 dst[2*i ] = ((int)(src0[i] + (unsigned)add)) >> shift;
61 dst[2*i+1] = ((int)(src1[i] + (unsigned)add)) >> shift;
62 }
63 }
64
RENAME(horizontal_compose_dirac53i)65 static void RENAME(horizontal_compose_dirac53i)(uint8_t *_b, uint8_t *_temp, int w)
66 {
67 int x;
68 const int w2 = w >> 1;
69 TYPE *b = (TYPE *)_b;
70 TYPE *temp = (TYPE *)_temp;
71
72 temp[0] = COMPOSE_53iL0(b[w2], b[0], b[w2]);
73 for (x = 1; x < w2; x++) {
74 temp[x ] = COMPOSE_53iL0 (b[x+w2-1], b[x ], b[x+w2]);
75 temp[x+w2-1] = COMPOSE_DIRAC53iH0(temp[x-1], b[x+w2-1], temp[x]);
76 }
77 temp[w-1] = COMPOSE_DIRAC53iH0(temp[w2-1], b[w-1], temp[w2-1]);
78
79 RENAME(interleave)(b, temp, temp+w2, w2, 1, 1);
80 }
81
RENAME(horizontal_compose_dd97i)82 static void RENAME(horizontal_compose_dd97i)(uint8_t *_b, uint8_t *_tmp, int w)
83 {
84 int x;
85 const int w2 = w >> 1;
86 TYPE *b = (TYPE *)_b;
87 TYPE *tmp = (TYPE *)_tmp;
88
89 tmp[0] = COMPOSE_53iL0(b[w2], b[0], b[w2]);
90 for (x = 1; x < w2; x++)
91 tmp[x] = COMPOSE_53iL0(b[x+w2-1], b[x], b[x+w2]);
92
93 // extend the edges
94 tmp[-1] = tmp[0];
95 tmp[w2+1] = tmp[w2] = tmp[w2-1];
96
97 for (x = 0; x < w2; x++) {
98 b[2*x ] = ((int)(tmp[x] + 1U))>>1;
99 b[2*x+1] = ((int)(COMPOSE_DD97iH0(tmp[x-1], tmp[x], b[x+w2], tmp[x+1], tmp[x+2]) + 1U))>>1;
100 }
101 }
102
RENAME(horizontal_compose_dd137i)103 static void RENAME(horizontal_compose_dd137i)(uint8_t *_b, uint8_t *_tmp, int w)
104 {
105 const int w2 = w >> 1;
106 int x;
107 TYPE *b = (TYPE *)_b;
108 TYPE *tmp = (TYPE *)_tmp;
109
110 tmp[0] = COMPOSE_DD137iL0(b[w2], b[w2], b[0], b[w2 ], b[w2+1]);
111 tmp[1] = COMPOSE_DD137iL0(b[w2], b[w2], b[1], b[w2+1], b[w2+2]);
112 for (x = 2; x < w2-1; x++)
113 tmp[x] = COMPOSE_DD137iL0(b[x+w2-2], b[x+w2-1], b[x], b[x+w2], b[x+w2+1]);
114 tmp[w2-1] = COMPOSE_DD137iL0(b[w-3], b[w-2], b[w2-1], b[w-1], b[w-1]);
115
116 // extend the edges
117 tmp[-1] = tmp[0];
118 tmp[w2+1] = tmp[w2] = tmp[w2-1];
119
120 for (x = 0; x < w2; x++) {
121 b[2*x ] = ((int)(tmp[x] + 1U))>>1;
122 b[2*x+1] = ((int)(COMPOSE_DD97iH0(tmp[x-1], tmp[x], b[x+w2], tmp[x+1], tmp[x+2]) + 1U))>>1;
123 }
124 }
125
RENAME(horizontal_compose_haari)126 static av_always_inline void RENAME(horizontal_compose_haari)(TYPE *b, TYPE *temp,
127 int w, int shift)
128 {
129 const int w2 = w >> 1;
130 int x;
131
132 for (x = 0; x < w2; x++) {
133 temp[x ] = COMPOSE_HAARiL0(b[x ], b[x+w2]);
134 temp[x+w2] = COMPOSE_HAARiH0(b[x+w2], temp[x]);
135 }
136
137 RENAME(interleave)(b, temp, temp+w2, w2, shift, shift);
138 }
139
RENAME(horizontal_compose_haar0i)140 static void RENAME(horizontal_compose_haar0i)(uint8_t *_b, uint8_t *_temp, int w)
141 {
142 TYPE *b = (TYPE *)_b;
143 TYPE *temp = (TYPE *)_temp;
144 RENAME(horizontal_compose_haari)(b, temp, w, 0);
145 }
146
RENAME(horizontal_compose_haar1i)147 static void RENAME(horizontal_compose_haar1i)(uint8_t *_b, uint8_t *_temp, int w)
148 {
149 TYPE *b = (TYPE *)_b;
150 TYPE *temp = (TYPE *)_temp;
151 RENAME(horizontal_compose_haari)(b, temp, w, 1);
152 }
153
RENAME(horizontal_compose_fidelityi)154 static void RENAME(horizontal_compose_fidelityi)(uint8_t *_b, uint8_t *_tmp, int w)
155 {
156 const int w2 = w >> 1;
157 int i, x;
158 TYPE v[8];
159 TYPE *b = (TYPE *)_b;
160 TYPE *tmp = (TYPE *)_tmp;
161
162 for (x = 0; x < w2; x++) {
163 for (i = 0; i < 8; i++)
164 v[i] = b[av_clip(x-3+i, 0, w2-1)];
165 tmp[x] = COMPOSE_FIDELITYiH0(v[0], v[1], v[2], v[3], b[x+w2], v[4], v[5], v[6], v[7]);
166 }
167
168 for (x = 0; x < w2; x++) {
169 for (i = 0; i < 8; i++)
170 v[i] = tmp[av_clip(x-4+i, 0, w2-1)];
171 tmp[x+w2] = COMPOSE_FIDELITYiL0(v[0], v[1], v[2], v[3], b[x], v[4], v[5], v[6], v[7]);
172 }
173
174 RENAME(interleave)(b, tmp+w2, tmp, w2, 0, 0);
175 }
176
RENAME(horizontal_compose_daub97i)177 static void RENAME(horizontal_compose_daub97i)(uint8_t *_b, uint8_t *_temp, int w)
178 {
179 const int w2 = w >> 1;
180 int x, b0, b1, b2;
181 TYPE *b = (TYPE *)_b;
182 TYPE *temp = (TYPE *)_temp;
183
184 temp[0] = COMPOSE_DAUB97iL1(b[w2], b[0], b[w2]);
185 for (x = 1; x < w2; x++) {
186 temp[x ] = COMPOSE_DAUB97iL1(b[x+w2-1], b[x ], b[x+w2]);
187 temp[x+w2-1] = COMPOSE_DAUB97iH1(temp[x-1], b[x+w2-1], temp[x]);
188 }
189 temp[w-1] = COMPOSE_DAUB97iH1(temp[w2-1], b[w-1], temp[w2-1]);
190
191 // second stage combined with interleave and shift
192 b0 = b2 = COMPOSE_DAUB97iL0(temp[w2], temp[0], temp[w2]);
193 b[0] = ~((~b0) >> 1);
194 for (x = 1; x < w2; x++) {
195 b2 = COMPOSE_DAUB97iL0(temp[x+w2-1], temp[x ], temp[x+w2]);
196 b1 = COMPOSE_DAUB97iH0( b0, temp[x+w2-1], b2 );
197 b[2*x-1] = ~((~b1) >> 1);
198 b[2*x ] = ~((~b2) >> 1);
199 b0 = b2;
200 }
201 b[w-1] = ~((~COMPOSE_DAUB97iH0(b2, temp[w-1], b2)) >> 1);
202 }
203
RENAME(vertical_compose_dirac53iH0)204 static void RENAME(vertical_compose_dirac53iH0)(uint8_t *_b0, uint8_t *_b1, uint8_t *_b2,
205 int width)
206 {
207 int i;
208 TYPE *b0 = (TYPE *)_b0;
209 TYPE *b1 = (TYPE *)_b1;
210 TYPE *b2 = (TYPE *)_b2;
211 for(i=0; i<width; i++){
212 b1[i] = COMPOSE_DIRAC53iH0(b0[i], b1[i], b2[i]);
213 }
214 }
215
RENAME(vertical_compose_dd97iH0)216 static void RENAME(vertical_compose_dd97iH0)(uint8_t *_b0, uint8_t *_b1, uint8_t *_b2,
217 uint8_t *_b3, uint8_t *_b4, int width)
218 {
219 int i;
220 TYPE *b0 = (TYPE *)_b0;
221 TYPE *b1 = (TYPE *)_b1;
222 TYPE *b2 = (TYPE *)_b2;
223 TYPE *b3 = (TYPE *)_b3;
224 TYPE *b4 = (TYPE *)_b4;
225 for(i=0; i<width; i++){
226 b2[i] = COMPOSE_DD97iH0(b0[i], b1[i], b2[i], b3[i], b4[i]);
227 }
228 }
229
RENAME(vertical_compose_dd137iL0)230 static void RENAME(vertical_compose_dd137iL0)(uint8_t *_b0, uint8_t *_b1, uint8_t *_b2,
231 uint8_t *_b3, uint8_t *_b4, int width)
232 {
233 int i;
234 TYPE *b0 = (TYPE *)_b0;
235 TYPE *b1 = (TYPE *)_b1;
236 TYPE *b2 = (TYPE *)_b2;
237 TYPE *b3 = (TYPE *)_b3;
238 TYPE *b4 = (TYPE *)_b4;
239 for(i=0; i<width; i++){
240 b2[i] = COMPOSE_DD137iL0(b0[i], b1[i], b2[i], b3[i], b4[i]);
241 }
242 }
243
RENAME(vertical_compose_haar)244 static void RENAME(vertical_compose_haar)(uint8_t *_b0, uint8_t *_b1, int width)
245 {
246 int i;
247 TYPE *b0 = (TYPE *)_b0;
248 TYPE *b1 = (TYPE *)_b1;
249
250 for (i = 0; i < width; i++) {
251 b0[i] = COMPOSE_HAARiL0(b0[i], b1[i]);
252 b1[i] = COMPOSE_HAARiH0(b1[i], b0[i]);
253 }
254 }
255
RENAME(vertical_compose_fidelityiH0)256 static void RENAME(vertical_compose_fidelityiH0)(uint8_t *_dst, uint8_t *_b[8], int width)
257 {
258 int i;
259 TYPE *dst = (TYPE *)_dst;
260 TYPE *b0 = (TYPE *)_b[0];
261 TYPE *b1 = (TYPE *)_b[1];
262 TYPE *b2 = (TYPE *)_b[2];
263 TYPE *b3 = (TYPE *)_b[3];
264 TYPE *b4 = (TYPE *)_b[4];
265 TYPE *b5 = (TYPE *)_b[5];
266 TYPE *b6 = (TYPE *)_b[6];
267 TYPE *b7 = (TYPE *)_b[7];
268 for(i=0; i<width; i++){
269 dst[i] = COMPOSE_FIDELITYiH0(b0[i], b1[i], b2[i], b3[i], dst[i], b4[i], b5[i], b6[i], b7[i]);
270 }
271 }
272
RENAME(vertical_compose_fidelityiL0)273 static void RENAME(vertical_compose_fidelityiL0)(uint8_t *_dst, uint8_t *_b[8], int width)
274 {
275 int i;
276 TYPE *dst = (TYPE *)_dst;
277 TYPE *b0 = (TYPE *)_b[0];
278 TYPE *b1 = (TYPE *)_b[1];
279 TYPE *b2 = (TYPE *)_b[2];
280 TYPE *b3 = (TYPE *)_b[3];
281 TYPE *b4 = (TYPE *)_b[4];
282 TYPE *b5 = (TYPE *)_b[5];
283 TYPE *b6 = (TYPE *)_b[6];
284 TYPE *b7 = (TYPE *)_b[7];
285
286 for(i=0; i<width; i++){
287 dst[i] = COMPOSE_FIDELITYiL0(b0[i], b1[i], b2[i], b3[i], dst[i], b4[i], b5[i], b6[i], b7[i]);
288 }
289 }
290
RENAME(vertical_compose_daub97iH0)291 static void RENAME(vertical_compose_daub97iH0)(uint8_t *_b0, uint8_t *_b1, uint8_t *_b2, int width)
292 {
293 int i;
294 TYPE *b0 = (TYPE *)_b0;
295 TYPE *b1 = (TYPE *)_b1;
296 TYPE *b2 = (TYPE *)_b2;
297
298 for(i=0; i<width; i++){
299 b1[i] = COMPOSE_DAUB97iH0(b0[i], b1[i], b2[i]);
300 }
301 }
302
RENAME(vertical_compose_daub97iH1)303 static void RENAME(vertical_compose_daub97iH1)(uint8_t *_b0, uint8_t *_b1, uint8_t *_b2, int width)
304 {
305 int i;
306 TYPE *b0 = (TYPE *)_b0;
307 TYPE *b1 = (TYPE *)_b1;
308 TYPE *b2 = (TYPE *)_b2;
309
310 for(i=0; i<width; i++){
311 b1[i] = COMPOSE_DAUB97iH1(b0[i], b1[i], b2[i]);
312 }
313 }
314
RENAME(vertical_compose_daub97iL0)315 static void RENAME(vertical_compose_daub97iL0)(uint8_t *_b0, uint8_t *_b1, uint8_t *_b2, int width)
316 {
317 int i;
318 TYPE *b0 = (TYPE *)_b0;
319 TYPE *b1 = (TYPE *)_b1;
320 TYPE *b2 = (TYPE *)_b2;
321
322 for(i=0; i<width; i++){
323 b1[i] = COMPOSE_DAUB97iL0(b0[i], b1[i], b2[i]);
324 }
325 }
326
RENAME(vertical_compose_daub97iL1)327 static void RENAME(vertical_compose_daub97iL1)(uint8_t *_b0, uint8_t *_b1, uint8_t *_b2, int width)
328 {
329 int i;
330 TYPE *b0 = (TYPE *)_b0;
331 TYPE *b1 = (TYPE *)_b1;
332 TYPE *b2 = (TYPE *)_b2;
333
334 for(i=0; i<width; i++){
335 b1[i] = COMPOSE_DAUB97iL1(b0[i], b1[i], b2[i]);
336 }
337 }
338
RENAME(spatial_compose_dd97i_dy)339 static void RENAME(spatial_compose_dd97i_dy)(DWTContext *d, int level, int width, int height, int stride)
340 {
341 vertical_compose_3tap vertical_compose_l0 = (void*)d->vertical_compose_l0;
342 vertical_compose_5tap vertical_compose_h0 = (void*)d->vertical_compose_h0;
343 DWTCompose *cs = d->cs + level;
344
345 int i, y = cs->y;
346 uint8_t *b[8];
347 for (i = 0; i < 6; i++)
348 b[i] = cs->b[i];
349 b[6] = d->buffer + av_clip(y+5, 0, height-2)*stride;
350 b[7] = d->buffer + av_clip(y+6, 1, height-1)*stride;
351
352 if(y+5<(unsigned)height) vertical_compose_l0( b[5], b[6], b[7], width);
353 if(y+1<(unsigned)height) vertical_compose_h0(b[0], b[2], b[3], b[4], b[6], width);
354
355 if(y-1<(unsigned)height) d->horizontal_compose(b[0], d->temp, width);
356 if(y+0<(unsigned)height) d->horizontal_compose(b[1], d->temp, width);
357
358 for (i = 0; i < 6; i++)
359 cs->b[i] = b[i+2];
360 cs->y += 2;
361 }
362
RENAME(spatial_compose_dirac53i_dy)363 static void RENAME(spatial_compose_dirac53i_dy)(DWTContext *d, int level, int width, int height, int stride)
364 {
365 vertical_compose_3tap vertical_compose_l0 = (void*)d->vertical_compose_l0;
366 vertical_compose_3tap vertical_compose_h0 = (void*)d->vertical_compose_h0;
367 DWTCompose *cs = d->cs + level;
368
369 int y= cs->y;
370 uint8_t *b[4] = { cs->b[0], cs->b[1] };
371 b[2] = d->buffer + avpriv_mirror(y+1, height-1)*stride;
372 b[3] = d->buffer + avpriv_mirror(y+2, height-1)*stride;
373
374 if(y+1<(unsigned)height) vertical_compose_l0(b[1], b[2], b[3], width);
375 if(y+0<(unsigned)height) vertical_compose_h0(b[0], b[1], b[2], width);
376
377 if(y-1<(unsigned)height) d->horizontal_compose(b[0], d->temp, width);
378 if(y+0<(unsigned)height) d->horizontal_compose(b[1], d->temp, width);
379
380 cs->b[0] = b[2];
381 cs->b[1] = b[3];
382 cs->y += 2;
383 }
384
RENAME(spatial_compose_dd137i_dy)385 static void RENAME(spatial_compose_dd137i_dy)(DWTContext *d, int level, int width, int height, int stride)
386 {
387 vertical_compose_5tap vertical_compose_l0 = (void*)d->vertical_compose_l0;
388 vertical_compose_5tap vertical_compose_h0 = (void*)d->vertical_compose_h0;
389 DWTCompose *cs = d->cs + level;
390
391 int i, y = cs->y;
392 uint8_t *b[10];
393 for (i = 0; i < 8; i++)
394 b[i] = cs->b[i];
395 b[8] = d->buffer + av_clip(y+7, 0, height-2)*stride;
396 b[9] = d->buffer + av_clip(y+8, 1, height-1)*stride;
397
398 if(y+5<(unsigned)height) vertical_compose_l0(b[3], b[5], b[6], b[7], b[9], width);
399 if(y+1<(unsigned)height) vertical_compose_h0(b[0], b[2], b[3], b[4], b[6], width);
400
401 if(y-1<(unsigned)height) d->horizontal_compose(b[0], d->temp, width);
402 if(y+0<(unsigned)height) d->horizontal_compose(b[1], d->temp, width);
403
404 for (i = 0; i < 8; i++)
405 cs->b[i] = b[i+2];
406 cs->y += 2;
407 }
408
409 // haar makes the assumption that height is even (always true for dirac)
RENAME(spatial_compose_haari_dy)410 static void RENAME(spatial_compose_haari_dy)(DWTContext *d, int level, int width, int height, int stride)
411 {
412 vertical_compose_2tap vertical_compose = (void*)d->vertical_compose;
413 int y = d->cs[level].y;
414 uint8_t *b0 = d->buffer + (y-1)*stride;
415 uint8_t *b1 = d->buffer + (y )*stride;
416
417 vertical_compose(b0, b1, width);
418 d->horizontal_compose(b0, d->temp, width);
419 d->horizontal_compose(b1, d->temp, width);
420
421 d->cs[level].y += 2;
422 }
423
424 // Don't do sliced idwt for fidelity; the 9 tap filter makes it a bit annoying
425 // Fortunately, this filter isn't used in practice.
RENAME(spatial_compose_fidelity)426 static void RENAME(spatial_compose_fidelity)(DWTContext *d, int level, int width, int height, int stride)
427 {
428 vertical_compose_9tap vertical_compose_l0 = (void*)d->vertical_compose_l0;
429 vertical_compose_9tap vertical_compose_h0 = (void*)d->vertical_compose_h0;
430 int i, y;
431 uint8_t *b[8];
432
433 for (y = 1; y < height; y += 2) {
434 for (i = 0; i < 8; i++)
435 b[i] = d->buffer + av_clip((y-7 + 2*i), 0, height-2)*stride;
436 vertical_compose_h0(d->buffer + y*stride, b, width);
437 }
438
439 for (y = 0; y < height; y += 2) {
440 for (i = 0; i < 8; i++)
441 b[i] = d->buffer + av_clip((y-7 + 2*i), 1, height-1)*stride;
442 vertical_compose_l0(d->buffer + y*stride, b, width);
443 }
444
445 for (y = 0; y < height; y++)
446 d->horizontal_compose(d->buffer + y*stride, d->temp, width);
447
448 d->cs[level].y = height+1;
449 }
450
RENAME(spatial_compose_daub97i_dy)451 static void RENAME(spatial_compose_daub97i_dy)(DWTContext *d, int level, int width, int height, int stride)
452 {
453 vertical_compose_3tap vertical_compose_l0 = (void*)d->vertical_compose_l0;
454 vertical_compose_3tap vertical_compose_h0 = (void*)d->vertical_compose_h0;
455 vertical_compose_3tap vertical_compose_l1 = (void*)d->vertical_compose_l1;
456 vertical_compose_3tap vertical_compose_h1 = (void*)d->vertical_compose_h1;
457 DWTCompose *cs = d->cs + level;
458
459 int i, y = cs->y;
460 uint8_t *b[6];
461 for (i = 0; i < 4; i++)
462 b[i] = cs->b[i];
463 b[4] = d->buffer + avpriv_mirror(y+3, height-1)*stride;
464 b[5] = d->buffer + avpriv_mirror(y+4, height-1)*stride;
465
466 if(y+3<(unsigned)height) vertical_compose_l1(b[3], b[4], b[5], width);
467 if(y+2<(unsigned)height) vertical_compose_h1(b[2], b[3], b[4], width);
468 if(y+1<(unsigned)height) vertical_compose_l0(b[1], b[2], b[3], width);
469 if(y+0<(unsigned)height) vertical_compose_h0(b[0], b[1], b[2], width);
470
471 if(y-1<(unsigned)height) d->horizontal_compose(b[0], d->temp, width);
472 if(y+0<(unsigned)height) d->horizontal_compose(b[1], d->temp, width);
473
474 for (i = 0; i < 4; i++)
475 cs->b[i] = b[i+2];
476 cs->y += 2;
477 }
478
RENAME(spatial_compose97i_init)479 static void RENAME(spatial_compose97i_init)(DWTCompose *cs, uint8_t *buffer, int height, int stride)
480 {
481 cs->b[0] = buffer + avpriv_mirror(-3-1, height-1)*stride;
482 cs->b[1] = buffer + avpriv_mirror(-3 , height-1)*stride;
483 cs->b[2] = buffer + avpriv_mirror(-3+1, height-1)*stride;
484 cs->b[3] = buffer + avpriv_mirror(-3+2, height-1)*stride;
485 cs->y = -3;
486 }
487
RENAME(spatial_compose53i_init)488 static void RENAME(spatial_compose53i_init)(DWTCompose *cs, uint8_t *buffer, int height, int stride)
489 {
490 cs->b[0] = buffer + avpriv_mirror(-1-1, height-1)*stride;
491 cs->b[1] = buffer + avpriv_mirror(-1 , height-1)*stride;
492 cs->y = -1;
493 }
494
RENAME(spatial_compose_dd97i_init)495 static void RENAME(spatial_compose_dd97i_init)(DWTCompose *cs, uint8_t *buffer, int height, int stride)
496 {
497 cs->b[0] = buffer + av_clip(-5-1, 0, height-2)*stride;
498 cs->b[1] = buffer + av_clip(-5 , 1, height-1)*stride;
499 cs->b[2] = buffer + av_clip(-5+1, 0, height-2)*stride;
500 cs->b[3] = buffer + av_clip(-5+2, 1, height-1)*stride;
501 cs->b[4] = buffer + av_clip(-5+3, 0, height-2)*stride;
502 cs->b[5] = buffer + av_clip(-5+4, 1, height-1)*stride;
503 cs->y = -5;
504 }
505
RENAME(spatial_compose_dd137i_init)506 static void RENAME(spatial_compose_dd137i_init)(DWTCompose *cs, uint8_t *buffer, int height, int stride)
507 {
508 cs->b[0] = buffer + av_clip(-5-1, 0, height-2)*stride;
509 cs->b[1] = buffer + av_clip(-5 , 1, height-1)*stride;
510 cs->b[2] = buffer + av_clip(-5+1, 0, height-2)*stride;
511 cs->b[3] = buffer + av_clip(-5+2, 1, height-1)*stride;
512 cs->b[4] = buffer + av_clip(-5+3, 0, height-2)*stride;
513 cs->b[5] = buffer + av_clip(-5+4, 1, height-1)*stride;
514 cs->b[6] = buffer + av_clip(-5+5, 0, height-2)*stride;
515 cs->b[7] = buffer + av_clip(-5+6, 1, height-1)*stride;
516 cs->y = -5;
517 }
518
RENAME(ff_spatial_idwt_init)519 static int RENAME(ff_spatial_idwt_init)(DWTContext *d, enum dwt_type type)
520 {
521 int level;
522
523 d->temp = (uint8_t *)(((TYPE *)d->temp) + 8);
524
525 for (level = d->decomposition_count - 1; level >= 0; level--){
526 int hl = d->height >> level;
527 int stride_l = d->stride << level;
528
529 switch(type){
530 case DWT_DIRAC_DD9_7:
531 RENAME(spatial_compose_dd97i_init)(d->cs+level, d->buffer, hl, stride_l);
532 break;
533 case DWT_DIRAC_LEGALL5_3:
534 RENAME(spatial_compose53i_init)(d->cs+level, d->buffer, hl, stride_l);
535 break;
536 case DWT_DIRAC_DD13_7:
537 RENAME(spatial_compose_dd137i_init)(d->cs+level, d->buffer, hl, stride_l);
538 break;
539 case DWT_DIRAC_HAAR0:
540 case DWT_DIRAC_HAAR1:
541 d->cs[level].y = 1;
542 break;
543 case DWT_DIRAC_DAUB9_7:
544 RENAME(spatial_compose97i_init)(d->cs+level, d->buffer, hl, stride_l);
545 break;
546 default:
547 d->cs[level].y = 0;
548 break;
549 }
550 }
551
552 switch (type) {
553 case DWT_DIRAC_DD9_7:
554 d->spatial_compose = RENAME(spatial_compose_dd97i_dy);
555 d->vertical_compose_l0 = (void*)RENAME(vertical_compose53iL0);
556 d->vertical_compose_h0 = (void*)RENAME(vertical_compose_dd97iH0);
557 d->horizontal_compose = RENAME(horizontal_compose_dd97i);
558 d->support = 7;
559 break;
560 case DWT_DIRAC_LEGALL5_3:
561 d->spatial_compose = RENAME(spatial_compose_dirac53i_dy);
562 d->vertical_compose_l0 = (void*)RENAME(vertical_compose53iL0);
563 d->vertical_compose_h0 = (void*)RENAME(vertical_compose_dirac53iH0);
564 d->horizontal_compose = RENAME(horizontal_compose_dirac53i);
565 d->support = 3;
566 break;
567 case DWT_DIRAC_DD13_7:
568 d->spatial_compose = RENAME(spatial_compose_dd137i_dy);
569 d->vertical_compose_l0 = (void*)RENAME(vertical_compose_dd137iL0);
570 d->vertical_compose_h0 = (void*)RENAME(vertical_compose_dd97iH0);
571 d->horizontal_compose = RENAME(horizontal_compose_dd137i);
572 d->support = 7;
573 break;
574 case DWT_DIRAC_HAAR0:
575 case DWT_DIRAC_HAAR1:
576 d->spatial_compose = RENAME(spatial_compose_haari_dy);
577 d->vertical_compose = (void*)RENAME(vertical_compose_haar);
578 if (type == DWT_DIRAC_HAAR0)
579 d->horizontal_compose = RENAME(horizontal_compose_haar0i);
580 else
581 d->horizontal_compose = RENAME(horizontal_compose_haar1i);
582 d->support = 1;
583 break;
584 case DWT_DIRAC_FIDELITY:
585 d->spatial_compose = RENAME(spatial_compose_fidelity);
586 d->vertical_compose_l0 = (void*)RENAME(vertical_compose_fidelityiL0);
587 d->vertical_compose_h0 = (void*)RENAME(vertical_compose_fidelityiH0);
588 d->horizontal_compose = RENAME(horizontal_compose_fidelityi);
589 d->support = 0; // not really used
590 break;
591 case DWT_DIRAC_DAUB9_7:
592 d->spatial_compose = RENAME(spatial_compose_daub97i_dy);
593 d->vertical_compose_l0 = (void*)RENAME(vertical_compose_daub97iL0);
594 d->vertical_compose_h0 = (void*)RENAME(vertical_compose_daub97iH0);
595 d->vertical_compose_l1 = (void*)RENAME(vertical_compose_daub97iL1);
596 d->vertical_compose_h1 = (void*)RENAME(vertical_compose_daub97iH1);
597 d->horizontal_compose = RENAME(horizontal_compose_daub97i);
598 d->support = 5;
599 break;
600 default:
601 return AVERROR_INVALIDDATA;
602 }
603
604 return 0;
605 }
606
607 #undef RENAME
608 #undef TYPE
609