Commit | Line | Data |
---|---|---|
0abc2e73 MN |
1 | /* |
2 | * Chinese AVS video (AVS1-P2, JiZhun profile) decoder. | |
3 | * | |
4 | * DSP functions | |
5 | * | |
6 | * Copyright (c) 2006 Stefan Gehrer <stefan.gehrer@gmx.de> | |
7 | * | |
8 | * This library 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 of the License, or (at your option) any later version. | |
12 | * | |
13 | * This library 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 this library; if not, write to the Free Software | |
20 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA | |
21 | */ | |
22 | ||
23 | #include <stdio.h> | |
24 | #include "dsputil.h" | |
0abc2e73 MN |
25 | |
26 | /***************************************************************************** | |
27 | * | |
28 | * in-loop deblocking filter | |
29 | * | |
30 | ****************************************************************************/ | |
31 | ||
32 | #define P2 p0_p[-3*stride] | |
33 | #define P1 p0_p[-2*stride] | |
34 | #define P0 p0_p[-1*stride] | |
35 | #define Q0 p0_p[ 0*stride] | |
36 | #define Q1 p0_p[ 1*stride] | |
37 | #define Q2 p0_p[ 2*stride] | |
38 | ||
39 | static inline void loop_filter_l2(uint8_t *p0_p,int stride,int alpha, int beta) { | |
40 | int p0 = P0; | |
41 | int q0 = Q0; | |
42 | ||
43 | if(abs(p0-q0)<alpha && abs(P1-p0)<beta && abs(Q1-q0)<beta) { | |
44 | int s = p0 + q0 + 2; | |
45 | alpha = (alpha>>2) + 2; | |
46 | if(abs(P2-p0) < beta && abs(p0-q0) < alpha) { | |
47 | P0 = (P1 + p0 + s) >> 2; | |
48 | P1 = (2*P1 + s) >> 2; | |
49 | } else | |
50 | P0 = (2*P1 + s) >> 2; | |
51 | if(abs(Q2-q0) < beta && abs(q0-p0) < alpha) { | |
52 | Q0 = (Q1 + q0 + s) >> 2; | |
53 | Q1 = (2*Q1 + s) >> 2; | |
54 | } else | |
55 | Q0 = (2*Q1 + s) >> 2; | |
56 | } | |
57 | } | |
58 | ||
59 | static inline void loop_filter_l1(uint8_t *p0_p, int stride, int alpha, int beta, int tc) { | |
60 | int p0 = P0; | |
61 | int q0 = Q0; | |
62 | ||
63 | if(abs(p0-q0)<alpha && abs(P1-p0)<beta && abs(Q1-q0)<beta) { | |
64 | int delta = clip(((q0-p0)*3+P1-Q1+4)>>3,-tc, tc); | |
65 | P0 = clip_uint8(p0+delta); | |
66 | Q0 = clip_uint8(q0-delta); | |
67 | if(abs(P2-p0)<beta) { | |
68 | delta = clip(((P0-P1)*3+P2-Q0+4)>>3, -tc, tc); | |
69 | P1 = clip_uint8(P1+delta); | |
70 | } | |
71 | if(abs(Q2-q0)<beta) { | |
72 | delta = clip(((Q1-Q0)*3+P0-Q2+4)>>3, -tc, tc); | |
73 | Q1 = clip_uint8(Q1-delta); | |
74 | } | |
75 | } | |
76 | } | |
77 | ||
78 | static inline void loop_filter_c2(uint8_t *p0_p,int stride,int alpha, int beta) { | |
79 | int p0 = P0; | |
80 | int q0 = Q0; | |
81 | ||
82 | if(abs(p0-q0)<alpha && abs(P1-p0)<beta && abs(Q1-q0)<beta) { | |
83 | int s = p0 + q0 + 2; | |
84 | alpha = (alpha>>2) + 2; | |
85 | if(abs(P2-p0) < beta && abs(p0-q0) < alpha) { | |
86 | P0 = (P1 + p0 + s) >> 2; | |
87 | } else | |
88 | P0 = (2*P1 + s) >> 2; | |
89 | if(abs(Q2-q0) < beta && abs(q0-p0) < alpha) { | |
90 | Q0 = (Q1 + q0 + s) >> 2; | |
91 | } else | |
92 | Q0 = (2*Q1 + s) >> 2; | |
93 | } | |
94 | } | |
95 | ||
96 | static inline void loop_filter_c1(uint8_t *p0_p,int stride,int alpha, int beta, | |
97 | int tc) { | |
98 | if(abs(P0-Q0)<alpha && abs(P1-P0)<beta && abs(Q1-Q0)<beta) { | |
99 | int delta = clip(((Q0-P0)*3+P1-Q1+4)>>3, -tc, tc); | |
100 | P0 = clip_uint8(P0+delta); | |
101 | Q0 = clip_uint8(Q0-delta); | |
102 | } | |
103 | } | |
104 | ||
105 | #undef P0 | |
106 | #undef P1 | |
107 | #undef P2 | |
108 | #undef Q0 | |
109 | #undef Q1 | |
110 | #undef Q2 | |
111 | ||
b482e2d1 | 112 | static void cavs_filter_lv_c(uint8_t *d, int stride, int alpha, int beta, int tc, |
0abc2e73 MN |
113 | int bs1, int bs2) { |
114 | int i; | |
115 | if(bs1==2) | |
116 | for(i=0;i<16;i++) | |
117 | loop_filter_l2(d + i*stride,1,alpha,beta); | |
118 | else { | |
119 | if(bs1) | |
120 | for(i=0;i<8;i++) | |
121 | loop_filter_l1(d + i*stride,1,alpha,beta,tc); | |
122 | if (bs2) | |
123 | for(i=8;i<16;i++) | |
124 | loop_filter_l1(d + i*stride,1,alpha,beta,tc); | |
125 | } | |
126 | } | |
127 | ||
b482e2d1 | 128 | static void cavs_filter_lh_c(uint8_t *d, int stride, int alpha, int beta, int tc, |
0abc2e73 MN |
129 | int bs1, int bs2) { |
130 | int i; | |
131 | if(bs1==2) | |
132 | for(i=0;i<16;i++) | |
133 | loop_filter_l2(d + i,stride,alpha,beta); | |
134 | else { | |
135 | if(bs1) | |
136 | for(i=0;i<8;i++) | |
137 | loop_filter_l1(d + i,stride,alpha,beta,tc); | |
138 | if (bs2) | |
139 | for(i=8;i<16;i++) | |
140 | loop_filter_l1(d + i,stride,alpha,beta,tc); | |
141 | } | |
142 | } | |
143 | ||
b482e2d1 | 144 | static void cavs_filter_cv_c(uint8_t *d, int stride, int alpha, int beta, int tc, |
0abc2e73 MN |
145 | int bs1, int bs2) { |
146 | int i; | |
147 | if(bs1==2) | |
148 | for(i=0;i<8;i++) | |
149 | loop_filter_c2(d + i*stride,1,alpha,beta); | |
150 | else { | |
151 | if(bs1) | |
152 | for(i=0;i<4;i++) | |
153 | loop_filter_c1(d + i*stride,1,alpha,beta,tc); | |
154 | if (bs2) | |
155 | for(i=4;i<8;i++) | |
156 | loop_filter_c1(d + i*stride,1,alpha,beta,tc); | |
157 | } | |
158 | } | |
159 | ||
b482e2d1 | 160 | static void cavs_filter_ch_c(uint8_t *d, int stride, int alpha, int beta, int tc, |
0abc2e73 MN |
161 | int bs1, int bs2) { |
162 | int i; | |
163 | if(bs1==2) | |
164 | for(i=0;i<8;i++) | |
165 | loop_filter_c2(d + i,stride,alpha,beta); | |
166 | else { | |
167 | if(bs1) | |
168 | for(i=0;i<4;i++) | |
169 | loop_filter_c1(d + i,stride,alpha,beta,tc); | |
170 | if (bs2) | |
171 | for(i=4;i<8;i++) | |
172 | loop_filter_c1(d + i,stride,alpha,beta,tc); | |
173 | } | |
174 | } | |
175 | ||
176 | /***************************************************************************** | |
177 | * | |
178 | * inverse transform | |
179 | * | |
180 | ****************************************************************************/ | |
181 | ||
b482e2d1 | 182 | static void cavs_idct8_add_c(uint8_t *dst, DCTELEM *block, int stride) { |
0abc2e73 MN |
183 | int i; |
184 | DCTELEM (*src)[8] = (DCTELEM(*)[8])block; | |
185 | uint8_t *cm = cropTbl + MAX_NEG_CROP; | |
186 | ||
187 | for( i = 0; i < 8; i++ ) { | |
188 | const int a0 = 3*src[i][1] - (src[i][7]<<1); | |
189 | const int a1 = 3*src[i][3] + (src[i][5]<<1); | |
190 | const int a2 = (src[i][3]<<1) - 3*src[i][5]; | |
191 | const int a3 = (src[i][1]<<1) + 3*src[i][7]; | |
192 | ||
193 | const int b4 = ((a0 + a1 + a3)<<1) + a1; | |
194 | const int b5 = ((a0 - a1 + a2)<<1) + a0; | |
195 | const int b6 = ((a3 - a2 - a1)<<1) + a3; | |
196 | const int b7 = ((a0 - a2 - a3)<<1) - a2; | |
197 | ||
198 | const int a7 = (src[i][2]<<2) - 10*src[i][6]; | |
199 | const int a6 = (src[i][6]<<2) + 10*src[i][2]; | |
200 | const int a5 = (src[i][0] - src[i][4]) << 3; | |
201 | const int a4 = (src[i][0] + src[i][4]) << 3; | |
202 | ||
203 | const int b0 = a4 + a6; | |
204 | const int b1 = a5 + a7; | |
205 | const int b2 = a5 - a7; | |
206 | const int b3 = a4 - a6; | |
207 | ||
208 | src[i][0] = (b0 + b4 + 4) >> 3; | |
209 | src[i][1] = (b1 + b5 + 4) >> 3; | |
210 | src[i][2] = (b2 + b6 + 4) >> 3; | |
211 | src[i][3] = (b3 + b7 + 4) >> 3; | |
212 | src[i][4] = (b3 - b7 + 4) >> 3; | |
213 | src[i][5] = (b2 - b6 + 4) >> 3; | |
214 | src[i][6] = (b1 - b5 + 4) >> 3; | |
215 | src[i][7] = (b0 - b4 + 4) >> 3; | |
216 | } | |
217 | for( i = 0; i < 8; i++ ) { | |
218 | const int a0 = 3*src[1][i] - (src[7][i]<<1); | |
219 | const int a1 = 3*src[3][i] + (src[5][i]<<1); | |
220 | const int a2 = (src[3][i]<<1) - 3*src[5][i]; | |
221 | const int a3 = (src[1][i]<<1) + 3*src[7][i]; | |
222 | ||
223 | const int b4 = ((a0 + a1 + a3)<<1) + a1; | |
224 | const int b5 = ((a0 - a1 + a2)<<1) + a0; | |
225 | const int b6 = ((a3 - a2 - a1)<<1) + a3; | |
226 | const int b7 = ((a0 - a2 - a3)<<1) - a2; | |
227 | ||
228 | const int a7 = (src[2][i]<<2) - 10*src[6][i]; | |
229 | const int a6 = (src[6][i]<<2) + 10*src[2][i]; | |
230 | const int a5 = (src[0][i] - src[4][i]) << 3; | |
231 | const int a4 = (src[0][i] + src[4][i]) << 3; | |
232 | ||
233 | const int b0 = a4 + a6; | |
234 | const int b1 = a5 + a7; | |
235 | const int b2 = a5 - a7; | |
236 | const int b3 = a4 - a6; | |
237 | ||
238 | dst[i + 0*stride] = cm[ dst[i + 0*stride] + ((b0 + b4 + 64) >> 7)]; | |
239 | dst[i + 1*stride] = cm[ dst[i + 1*stride] + ((b1 + b5 + 64) >> 7)]; | |
240 | dst[i + 2*stride] = cm[ dst[i + 2*stride] + ((b2 + b6 + 64) >> 7)]; | |
241 | dst[i + 3*stride] = cm[ dst[i + 3*stride] + ((b3 + b7 + 64) >> 7)]; | |
242 | dst[i + 4*stride] = cm[ dst[i + 4*stride] + ((b3 - b7 + 64) >> 7)]; | |
243 | dst[i + 5*stride] = cm[ dst[i + 5*stride] + ((b2 - b6 + 64) >> 7)]; | |
244 | dst[i + 6*stride] = cm[ dst[i + 6*stride] + ((b1 - b5 + 64) >> 7)]; | |
245 | dst[i + 7*stride] = cm[ dst[i + 7*stride] + ((b0 - b4 + 64) >> 7)]; | |
246 | } | |
247 | } | |
248 | ||
249 | /***************************************************************************** | |
250 | * | |
251 | * motion compensation | |
252 | * | |
253 | ****************************************************************************/ | |
254 | ||
255 | #define CAVS_SUBPIX(OPNAME, OP, NAME, A, B, C, D, E, F) \ | |
256 | static void OPNAME ## cavs_filt8_h_ ## NAME(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\ | |
257 | const int h=8;\ | |
258 | uint8_t *cm = cropTbl + MAX_NEG_CROP;\ | |
259 | int i;\ | |
260 | for(i=0; i<h; i++)\ | |
261 | {\ | |
262 | OP(dst[0], A*src[-2] + B*src[-1] + C*src[0] + D*src[1] + E*src[2] + F*src[3]);\ | |
263 | OP(dst[1], A*src[-1] + B*src[ 0] + C*src[1] + D*src[2] + E*src[3] + F*src[4]);\ | |
264 | OP(dst[2], A*src[ 0] + B*src[ 1] + C*src[2] + D*src[3] + E*src[4] + F*src[5]);\ | |
265 | OP(dst[3], A*src[ 1] + B*src[ 2] + C*src[3] + D*src[4] + E*src[5] + F*src[6]);\ | |
266 | OP(dst[4], A*src[ 2] + B*src[ 3] + C*src[4] + D*src[5] + E*src[6] + F*src[7]);\ | |
267 | OP(dst[5], A*src[ 3] + B*src[ 4] + C*src[5] + D*src[6] + E*src[7] + F*src[8]);\ | |
268 | OP(dst[6], A*src[ 4] + B*src[ 5] + C*src[6] + D*src[7] + E*src[8] + F*src[9]);\ | |
269 | OP(dst[7], A*src[ 5] + B*src[ 6] + C*src[7] + D*src[8] + E*src[9] + F*src[10]);\ | |
270 | dst+=dstStride;\ | |
271 | src+=srcStride;\ | |
272 | }\ | |
273 | }\ | |
274 | \ | |
275 | static void OPNAME ## cavs_filt8_v_ ## NAME(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\ | |
276 | const int w=8;\ | |
277 | uint8_t *cm = cropTbl + MAX_NEG_CROP;\ | |
278 | int i;\ | |
279 | for(i=0; i<w; i++)\ | |
280 | {\ | |
281 | const int srcB= src[-2*srcStride];\ | |
282 | const int srcA= src[-1*srcStride];\ | |
283 | const int src0= src[0 *srcStride];\ | |
284 | const int src1= src[1 *srcStride];\ | |
285 | const int src2= src[2 *srcStride];\ | |
286 | const int src3= src[3 *srcStride];\ | |
287 | const int src4= src[4 *srcStride];\ | |
288 | const int src5= src[5 *srcStride];\ | |
289 | const int src6= src[6 *srcStride];\ | |
290 | const int src7= src[7 *srcStride];\ | |
291 | const int src8= src[8 *srcStride];\ | |
292 | const int src9= src[9 *srcStride];\ | |
293 | const int src10= src[10 *srcStride];\ | |
294 | OP(dst[0*dstStride], A*srcB + B*srcA + C*src0 + D*src1 + E*src2 + F*src3);\ | |
295 | OP(dst[1*dstStride], A*srcA + B*src0 + C*src1 + D*src2 + E*src3 + F*src4);\ | |
296 | OP(dst[2*dstStride], A*src0 + B*src1 + C*src2 + D*src3 + E*src4 + F*src5);\ | |
297 | OP(dst[3*dstStride], A*src1 + B*src2 + C*src3 + D*src4 + E*src5 + F*src6);\ | |
298 | OP(dst[4*dstStride], A*src2 + B*src3 + C*src4 + D*src5 + E*src6 + F*src7);\ | |
299 | OP(dst[5*dstStride], A*src3 + B*src4 + C*src5 + D*src6 + E*src7 + F*src8);\ | |
300 | OP(dst[6*dstStride], A*src4 + B*src5 + C*src6 + D*src7 + E*src8 + F*src9);\ | |
301 | OP(dst[7*dstStride], A*src5 + B*src6 + C*src7 + D*src8 + E*src9 + F*src10);\ | |
302 | dst++;\ | |
303 | src++;\ | |
304 | }\ | |
305 | }\ | |
306 | \ | |
307 | static void OPNAME ## cavs_filt16_v_ ## NAME(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\ | |
308 | OPNAME ## cavs_filt8_v_ ## NAME(dst , src , dstStride, srcStride);\ | |
309 | OPNAME ## cavs_filt8_v_ ## NAME(dst+8, src+8, dstStride, srcStride);\ | |
310 | src += 8*srcStride;\ | |
311 | dst += 8*dstStride;\ | |
312 | OPNAME ## cavs_filt8_v_ ## NAME(dst , src , dstStride, srcStride);\ | |
313 | OPNAME ## cavs_filt8_v_ ## NAME(dst+8, src+8, dstStride, srcStride);\ | |
314 | }\ | |
315 | \ | |
316 | static void OPNAME ## cavs_filt16_h_ ## NAME(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\ | |
317 | OPNAME ## cavs_filt8_h_ ## NAME(dst , src , dstStride, srcStride);\ | |
318 | OPNAME ## cavs_filt8_h_ ## NAME(dst+8, src+8, dstStride, srcStride);\ | |
319 | src += 8*srcStride;\ | |
320 | dst += 8*dstStride;\ | |
321 | OPNAME ## cavs_filt8_h_ ## NAME(dst , src , dstStride, srcStride);\ | |
322 | OPNAME ## cavs_filt8_h_ ## NAME(dst+8, src+8, dstStride, srcStride);\ | |
323 | }\ | |
324 | ||
325 | #define CAVS_SUBPIX_HV(OPNAME, OP, NAME, AH, BH, CH, DH, EH, FH, AV, BV, CV, DV, EV, FV, FULL) \ | |
326 | static void OPNAME ## cavs_filt8_hv_ ## NAME(uint8_t *dst, uint8_t *src1, uint8_t *src2, int dstStride, int srcStride){\ | |
327 | int16_t temp[8*(8+5)];\ | |
328 | int16_t *tmp = temp;\ | |
329 | const int h=8;\ | |
330 | const int w=8;\ | |
331 | uint8_t *cm = cropTbl + MAX_NEG_CROP;\ | |
332 | int i;\ | |
333 | src1 -= 2*srcStride;\ | |
334 | for(i=0; i<h+5; i++)\ | |
335 | {\ | |
336 | tmp[0]= AH*src1[-2] + BH*src1[-1] + CH*src1[0] + DH*src1[1] + EH*src1[2] + FH*src1[3];\ | |
337 | tmp[1]= AH*src1[-1] + BH*src1[ 0] + CH*src1[1] + DH*src1[2] + EH*src1[3] + FH*src1[4];\ | |
338 | tmp[2]= AH*src1[ 0] + BH*src1[ 1] + CH*src1[2] + DH*src1[3] + EH*src1[4] + FH*src1[5];\ | |
339 | tmp[3]= AH*src1[ 1] + BH*src1[ 2] + CH*src1[3] + DH*src1[4] + EH*src1[5] + FH*src1[6];\ | |
340 | tmp[4]= AH*src1[ 2] + BH*src1[ 3] + CH*src1[4] + DH*src1[5] + EH*src1[6] + FH*src1[7];\ | |
341 | tmp[5]= AH*src1[ 3] + BH*src1[ 4] + CH*src1[5] + DH*src1[6] + EH*src1[7] + FH*src1[8];\ | |
342 | tmp[6]= AH*src1[ 4] + BH*src1[ 5] + CH*src1[6] + DH*src1[7] + EH*src1[8] + FH*src1[9];\ | |
343 | tmp[7]= AH*src1[ 5] + BH*src1[ 6] + CH*src1[7] + DH*src1[8] + EH*src1[9] + FH*src1[10];\ | |
344 | tmp+=8;\ | |
345 | src1+=srcStride;\ | |
346 | }\ | |
347 | if(FULL) {\ | |
348 | tmp = temp+8*2; \ | |
349 | for(i=0; i<w; i++) \ | |
350 | { \ | |
351 | const int tmpB= tmp[-2*8]; \ | |
352 | const int tmpA= tmp[-1*8]; \ | |
353 | const int tmp0= tmp[0 *8]; \ | |
354 | const int tmp1= tmp[1 *8]; \ | |
355 | const int tmp2= tmp[2 *8]; \ | |
356 | const int tmp3= tmp[3 *8]; \ | |
357 | const int tmp4= tmp[4 *8]; \ | |
358 | const int tmp5= tmp[5 *8]; \ | |
359 | const int tmp6= tmp[6 *8]; \ | |
360 | const int tmp7= tmp[7 *8]; \ | |
361 | const int tmp8= tmp[8 *8]; \ | |
362 | const int tmp9= tmp[9 *8]; \ | |
363 | const int tmp10=tmp[10*8]; \ | |
364 | OP(dst[0*dstStride], AV*tmpB + BV*tmpA + CV*tmp0 + DV*tmp1 + EV*tmp2 + FV*tmp3 + 64*src2[0*srcStride]); \ | |
365 | OP(dst[1*dstStride], AV*tmpA + BV*tmp0 + CV*tmp1 + DV*tmp2 + EV*tmp3 + FV*tmp4 + 64*src2[1*srcStride]); \ | |
366 | OP(dst[2*dstStride], AV*tmp0 + BV*tmp1 + CV*tmp2 + DV*tmp3 + EV*tmp4 + FV*tmp5 + 64*src2[2*srcStride]); \ | |
367 | OP(dst[3*dstStride], AV*tmp1 + BV*tmp2 + CV*tmp3 + DV*tmp4 + EV*tmp5 + FV*tmp6 + 64*src2[3*srcStride]); \ | |
368 | OP(dst[4*dstStride], AV*tmp2 + BV*tmp3 + CV*tmp4 + DV*tmp5 + EV*tmp6 + FV*tmp7 + 64*src2[4*srcStride]); \ | |
369 | OP(dst[5*dstStride], AV*tmp3 + BV*tmp4 + CV*tmp5 + DV*tmp6 + EV*tmp7 + FV*tmp8 + 64*src2[5*srcStride]); \ | |
370 | OP(dst[6*dstStride], AV*tmp4 + BV*tmp5 + CV*tmp6 + DV*tmp7 + EV*tmp8 + FV*tmp9 + 64*src2[6*srcStride]); \ | |
371 | OP(dst[7*dstStride], AV*tmp5 + BV*tmp6 + CV*tmp7 + DV*tmp8 + EV*tmp9 + FV*tmp10 + 64*src2[7*srcStride]); \ | |
372 | dst++; \ | |
373 | tmp++; \ | |
374 | src2++; \ | |
375 | } \ | |
376 | } else {\ | |
377 | tmp = temp+8*2; \ | |
378 | for(i=0; i<w; i++) \ | |
379 | { \ | |
380 | const int tmpB= tmp[-2*8]; \ | |
381 | const int tmpA= tmp[-1*8]; \ | |
382 | const int tmp0= tmp[0 *8]; \ | |
383 | const int tmp1= tmp[1 *8]; \ | |
384 | const int tmp2= tmp[2 *8]; \ | |
385 | const int tmp3= tmp[3 *8]; \ | |
386 | const int tmp4= tmp[4 *8]; \ | |
387 | const int tmp5= tmp[5 *8]; \ | |
388 | const int tmp6= tmp[6 *8]; \ | |
389 | const int tmp7= tmp[7 *8]; \ | |
390 | const int tmp8= tmp[8 *8]; \ | |
391 | const int tmp9= tmp[9 *8]; \ | |
392 | const int tmp10=tmp[10*8]; \ | |
393 | OP(dst[0*dstStride], AV*tmpB + BV*tmpA + CV*tmp0 + DV*tmp1 + EV*tmp2 + FV*tmp3); \ | |
394 | OP(dst[1*dstStride], AV*tmpA + BV*tmp0 + CV*tmp1 + DV*tmp2 + EV*tmp3 + FV*tmp4); \ | |
395 | OP(dst[2*dstStride], AV*tmp0 + BV*tmp1 + CV*tmp2 + DV*tmp3 + EV*tmp4 + FV*tmp5); \ | |
396 | OP(dst[3*dstStride], AV*tmp1 + BV*tmp2 + CV*tmp3 + DV*tmp4 + EV*tmp5 + FV*tmp6); \ | |
397 | OP(dst[4*dstStride], AV*tmp2 + BV*tmp3 + CV*tmp4 + DV*tmp5 + EV*tmp6 + FV*tmp7); \ | |
398 | OP(dst[5*dstStride], AV*tmp3 + BV*tmp4 + CV*tmp5 + DV*tmp6 + EV*tmp7 + FV*tmp8); \ | |
399 | OP(dst[6*dstStride], AV*tmp4 + BV*tmp5 + CV*tmp6 + DV*tmp7 + EV*tmp8 + FV*tmp9); \ | |
400 | OP(dst[7*dstStride], AV*tmp5 + BV*tmp6 + CV*tmp7 + DV*tmp8 + EV*tmp9 + FV*tmp10); \ | |
401 | dst++; \ | |
402 | tmp++; \ | |
403 | } \ | |
404 | }\ | |
405 | }\ | |
406 | \ | |
407 | static void OPNAME ## cavs_filt16_hv_ ## NAME(uint8_t *dst, uint8_t *src1, uint8_t *src2, int dstStride, int srcStride){ \ | |
408 | OPNAME ## cavs_filt8_hv_ ## NAME(dst , src1, src2 , dstStride, srcStride); \ | |
409 | OPNAME ## cavs_filt8_hv_ ## NAME(dst+8, src1+8, src2+8, dstStride, srcStride); \ | |
410 | src1 += 8*srcStride;\ | |
411 | src2 += 8*srcStride;\ | |
412 | dst += 8*dstStride;\ | |
413 | OPNAME ## cavs_filt8_hv_ ## NAME(dst , src1, src2 , dstStride, srcStride); \ | |
414 | OPNAME ## cavs_filt8_hv_ ## NAME(dst+8, src1+8, src2+8, dstStride, srcStride); \ | |
415 | }\ | |
416 | ||
417 | #define CAVS_MC(OPNAME, SIZE) \ | |
b482e2d1 | 418 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc10_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
419 | OPNAME ## cavs_filt ## SIZE ## _h_qpel_l(dst, src, stride, stride);\ |
420 | }\ | |
421 | \ | |
b482e2d1 | 422 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc20_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
423 | OPNAME ## cavs_filt ## SIZE ## _h_hpel(dst, src, stride, stride);\ |
424 | }\ | |
425 | \ | |
b482e2d1 | 426 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc30_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
427 | OPNAME ## cavs_filt ## SIZE ## _h_qpel_r(dst, src, stride, stride);\ |
428 | }\ | |
429 | \ | |
b482e2d1 | 430 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc01_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
431 | OPNAME ## cavs_filt ## SIZE ## _v_qpel_l(dst, src, stride, stride);\ |
432 | }\ | |
433 | \ | |
b482e2d1 | 434 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc02_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
435 | OPNAME ## cavs_filt ## SIZE ## _v_hpel(dst, src, stride, stride);\ |
436 | }\ | |
437 | \ | |
b482e2d1 | 438 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc03_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
439 | OPNAME ## cavs_filt ## SIZE ## _v_qpel_r(dst, src, stride, stride);\ |
440 | }\ | |
441 | \ | |
b482e2d1 | 442 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc22_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
443 | OPNAME ## cavs_filt ## SIZE ## _hv_jj(dst, src, NULL, stride, stride); \ |
444 | }\ | |
445 | \ | |
b482e2d1 | 446 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc11_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
447 | OPNAME ## cavs_filt ## SIZE ## _hv_egpr(dst, src, src, stride, stride); \ |
448 | }\ | |
449 | \ | |
b482e2d1 | 450 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc13_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
451 | OPNAME ## cavs_filt ## SIZE ## _hv_egpr(dst, src, src+stride, stride, stride); \ |
452 | }\ | |
453 | \ | |
b482e2d1 | 454 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc31_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
455 | OPNAME ## cavs_filt ## SIZE ## _hv_egpr(dst, src, src+1, stride, stride); \ |
456 | }\ | |
457 | \ | |
b482e2d1 | 458 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc33_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
459 | OPNAME ## cavs_filt ## SIZE ## _hv_egpr(dst, src, src+stride+1,stride, stride); \ |
460 | }\ | |
461 | \ | |
b482e2d1 | 462 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc21_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
463 | OPNAME ## cavs_filt ## SIZE ## _hv_ff(dst, src, src+stride+1,stride, stride); \ |
464 | }\ | |
465 | \ | |
b482e2d1 | 466 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc12_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
467 | OPNAME ## cavs_filt ## SIZE ## _hv_ii(dst, src, src+stride+1,stride, stride); \ |
468 | }\ | |
469 | \ | |
b482e2d1 | 470 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc32_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
471 | OPNAME ## cavs_filt ## SIZE ## _hv_kk(dst, src, src+stride+1,stride, stride); \ |
472 | }\ | |
473 | \ | |
b482e2d1 | 474 | static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc23_c(uint8_t *dst, uint8_t *src, int stride){\ |
0abc2e73 MN |
475 | OPNAME ## cavs_filt ## SIZE ## _hv_qq(dst, src, src+stride+1,stride, stride); \ |
476 | }\ | |
477 | ||
478 | #define op_put1(a, b) a = cm[((b)+4)>>3] | |
479 | #define op_put2(a, b) a = cm[((b)+64)>>7] | |
480 | #define op_put3(a, b) a = cm[((b)+32)>>6] | |
481 | #define op_put4(a, b) a = cm[((b)+512)>>10] | |
482 | #define op_avg1(a, b) a = ((a)+cm[((b)+4)>>3] +1)>>1 | |
483 | #define op_avg2(a, b) a = ((a)+cm[((b)+64)>>7] +1)>>1 | |
484 | #define op_avg3(a, b) a = ((a)+cm[((b)+32)>>6] +1)>>1 | |
485 | #define op_avg4(a, b) a = ((a)+cm[((b)+512)>>10]+1)>>1 | |
486 | CAVS_SUBPIX(put_ , op_put1, hpel, 0, -1, 5, 5, -1, 0) | |
487 | CAVS_SUBPIX(put_ , op_put2, qpel_l, -1, -2, 96, 42, -7, 0) | |
488 | CAVS_SUBPIX(put_ , op_put2, qpel_r, 0, -7, 42, 96, -2, -1) | |
489 | CAVS_SUBPIX_HV(put_, op_put3, jj, 0, -1, 5, 5, -1, 0, 0, -1, 5, 5, -1, 0, 0) | |
490 | CAVS_SUBPIX_HV(put_, op_put4, ff, 0, -1, 5, 5, -1, 0, -1, -2, 96, 42, -7, 0, 0) | |
491 | CAVS_SUBPIX_HV(put_, op_put4, ii, -1, -2, 96, 42, -7, 0, 0, -1, 5, 5, -1, 0, 0) | |
492 | CAVS_SUBPIX_HV(put_, op_put4, kk, 0, -7, 42, 96, -2, -1, 0, -1, 5, 5, -1, 0, 0) | |
493 | CAVS_SUBPIX_HV(put_, op_put4, qq, 0, -1, 5, 5, -1, 0, 0, -7, 42, 96, -2,-1, 0) | |
494 | CAVS_SUBPIX_HV(put_, op_put2, egpr, 0, -1, 5, 5, -1, 0, 0, -1, 5, 5, -1, 0, 1) | |
495 | CAVS_SUBPIX(avg_ , op_avg1, hpel, 0, -1, 5, 5, -1, 0) | |
496 | CAVS_SUBPIX(avg_ , op_avg2, qpel_l, -1, -2, 96, 42, -7, 0) | |
497 | CAVS_SUBPIX(avg_ , op_avg2, qpel_r, 0, -7, 42, 96, -2, -1) | |
498 | CAVS_SUBPIX_HV(avg_, op_avg3, jj, 0, -1, 5, 5, -1, 0, 0, -1, 5, 5, -1, 0, 0) | |
499 | CAVS_SUBPIX_HV(avg_, op_avg4, ff, 0, -1, 5, 5, -1, 0, -1, -2, 96, 42, -7, 0, 0) | |
500 | CAVS_SUBPIX_HV(avg_, op_avg4, ii, -1, -2, 96, 42, -7, 0, 0, -1, 5, 5, -1, 0, 0) | |
501 | CAVS_SUBPIX_HV(avg_, op_avg4, kk, 0, -7, 42, 96, -2, -1, 0, -1, 5, 5, -1, 0, 0) | |
502 | CAVS_SUBPIX_HV(avg_, op_avg4, qq, 0, -1, 5, 5, -1, 0, 0, -7, 42, 96, -2,-1, 0) | |
503 | CAVS_SUBPIX_HV(avg_, op_avg2, egpr, 0, -1, 5, 5, -1, 0, 0, -1, 5, 5, -1, 0, 1) | |
504 | CAVS_MC(put_, 8) | |
505 | CAVS_MC(put_, 16) | |
506 | CAVS_MC(avg_, 8) | |
507 | CAVS_MC(avg_, 16) | |
b482e2d1 MN |
508 | |
509 | void ff_put_cavs_qpel8_mc00_c(uint8_t *dst, uint8_t *src, int stride); | |
510 | void ff_avg_cavs_qpel8_mc00_c(uint8_t *dst, uint8_t *src, int stride); | |
511 | void ff_put_cavs_qpel16_mc00_c(uint8_t *dst, uint8_t *src, int stride); | |
512 | void ff_avg_cavs_qpel16_mc00_c(uint8_t *dst, uint8_t *src, int stride); | |
513 | ||
514 | void ff_cavsdsp_init(DSPContext* c, AVCodecContext *avctx) { | |
515 | #define dspfunc(PFX, IDX, NUM) \ | |
516 | c->PFX ## _pixels_tab[IDX][ 0] = ff_ ## PFX ## NUM ## _mc00_c; \ | |
517 | c->PFX ## _pixels_tab[IDX][ 1] = ff_ ## PFX ## NUM ## _mc10_c; \ | |
518 | c->PFX ## _pixels_tab[IDX][ 2] = ff_ ## PFX ## NUM ## _mc20_c; \ | |
519 | c->PFX ## _pixels_tab[IDX][ 3] = ff_ ## PFX ## NUM ## _mc30_c; \ | |
520 | c->PFX ## _pixels_tab[IDX][ 4] = ff_ ## PFX ## NUM ## _mc01_c; \ | |
521 | c->PFX ## _pixels_tab[IDX][ 5] = ff_ ## PFX ## NUM ## _mc11_c; \ | |
522 | c->PFX ## _pixels_tab[IDX][ 6] = ff_ ## PFX ## NUM ## _mc21_c; \ | |
523 | c->PFX ## _pixels_tab[IDX][ 7] = ff_ ## PFX ## NUM ## _mc31_c; \ | |
524 | c->PFX ## _pixels_tab[IDX][ 8] = ff_ ## PFX ## NUM ## _mc02_c; \ | |
525 | c->PFX ## _pixels_tab[IDX][ 9] = ff_ ## PFX ## NUM ## _mc12_c; \ | |
526 | c->PFX ## _pixels_tab[IDX][10] = ff_ ## PFX ## NUM ## _mc22_c; \ | |
527 | c->PFX ## _pixels_tab[IDX][11] = ff_ ## PFX ## NUM ## _mc32_c; \ | |
528 | c->PFX ## _pixels_tab[IDX][12] = ff_ ## PFX ## NUM ## _mc03_c; \ | |
529 | c->PFX ## _pixels_tab[IDX][13] = ff_ ## PFX ## NUM ## _mc13_c; \ | |
530 | c->PFX ## _pixels_tab[IDX][14] = ff_ ## PFX ## NUM ## _mc23_c; \ | |
531 | c->PFX ## _pixels_tab[IDX][15] = ff_ ## PFX ## NUM ## _mc33_c | |
532 | dspfunc(put_cavs_qpel, 0, 16); | |
533 | dspfunc(put_cavs_qpel, 1, 8); | |
534 | dspfunc(avg_cavs_qpel, 0, 16); | |
535 | dspfunc(avg_cavs_qpel, 1, 8); | |
536 | c->cavs_filter_lv = cavs_filter_lv_c; | |
537 | c->cavs_filter_lh = cavs_filter_lh_c; | |
538 | c->cavs_filter_cv = cavs_filter_cv_c; | |
539 | c->cavs_filter_ch = cavs_filter_ch_c; | |
540 | c->cavs_idct8_add = cavs_idct8_add_c; | |
541 | } |