Support for 9 and 10-bit encoding
[x262.git] / common / deblock.c
1 /*****************************************************************************
2  * deblock.c: h264 encoder library
3  *****************************************************************************
4  * Copyright (C) 2003-2008 x264 project
5  *
6  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
7  *          Loren Merritt <lorenm@u.washington.edu>
8  *          Jason Garrett-Glaser <darkshikari@gmail.com>
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
23  *****************************************************************************/
24
25 #include "common.h"
26
27 /* Deblocking filter */
28 static const uint8_t i_alpha_table[52+12*3] =
29 {
30      0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
31      0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
32      0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
33      0,  0,  0,  0,  0,  0,  4,  4,  5,  6,
34      7,  8,  9, 10, 12, 13, 15, 17, 20, 22,
35     25, 28, 32, 36, 40, 45, 50, 56, 63, 71,
36     80, 90,101,113,127,144,162,182,203,226,
37    255,255,
38    255,255,255,255,255,255,255,255,255,255,255,255,
39 };
40 static const uint8_t i_beta_table[52+12*3] =
41 {
42      0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
43      0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
44      0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
45      0,  0,  0,  0,  0,  0,  2,  2,  2,  3,
46      3,  3,  3,  4,  4,  4,  6,  6,  7,  7,
47      8,  8,  9,  9, 10, 10, 11, 11, 12, 12,
48     13, 13, 14, 14, 15, 15, 16, 16, 17, 17,
49     18, 18,
50     18, 18, 18, 18, 18, 18, 18, 18, 18, 18, 18, 18,
51 };
52 static const int8_t i_tc0_table[52+12*3][4] =
53 {
54     {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 },
55     {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 },
56     {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 },
57     {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 },
58     {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 },
59     {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 },
60     {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 0 }, {-1, 0, 0, 1 },
61     {-1, 0, 0, 1 }, {-1, 0, 0, 1 }, {-1, 0, 0, 1 }, {-1, 0, 1, 1 }, {-1, 0, 1, 1 }, {-1, 1, 1, 1 },
62     {-1, 1, 1, 1 }, {-1, 1, 1, 1 }, {-1, 1, 1, 1 }, {-1, 1, 1, 2 }, {-1, 1, 1, 2 }, {-1, 1, 1, 2 },
63     {-1, 1, 1, 2 }, {-1, 1, 2, 3 }, {-1, 1, 2, 3 }, {-1, 2, 2, 3 }, {-1, 2, 2, 4 }, {-1, 2, 3, 4 },
64     {-1, 2, 3, 4 }, {-1, 3, 3, 5 }, {-1, 3, 4, 6 }, {-1, 3, 4, 6 }, {-1, 4, 5, 7 }, {-1, 4, 5, 8 },
65     {-1, 4, 6, 9 }, {-1, 5, 7,10 }, {-1, 6, 8,11 }, {-1, 6, 8,13 }, {-1, 7,10,14 }, {-1, 8,11,16 },
66     {-1, 9,12,18 }, {-1,10,13,20 }, {-1,11,15,23 }, {-1,13,17,25 },
67     {-1,13,17,25 }, {-1,13,17,25 }, {-1,13,17,25 }, {-1,13,17,25 }, {-1,13,17,25 }, {-1,13,17,25 },
68     {-1,13,17,25 }, {-1,13,17,25 }, {-1,13,17,25 }, {-1,13,17,25 }, {-1,13,17,25 }, {-1,13,17,25 },
69 };
70 #define alpha_table(x) i_alpha_table[(x)+24]
71 #define beta_table(x)  i_beta_table[(x)+24]
72 #define tc0_table(x)   i_tc0_table[(x)+24]
73
74 /* From ffmpeg */
75 static inline void deblock_luma_c( pixel *pix, int xstride, int ystride, int alpha, int beta, int8_t *tc0 )
76 {
77     for( int i = 0; i < 4; i++ )
78     {
79         if( tc0[i] < 0 )
80         {
81             pix += 4*ystride;
82             continue;
83         }
84         for( int d = 0; d < 4; d++ )
85         {
86             int p2 = pix[-3*xstride];
87             int p1 = pix[-2*xstride];
88             int p0 = pix[-1*xstride];
89             int q0 = pix[ 0*xstride];
90             int q1 = pix[ 1*xstride];
91             int q2 = pix[ 2*xstride];
92
93             if( abs( p0 - q0 ) < alpha && abs( p1 - p0 ) < beta && abs( q1 - q0 ) < beta )
94             {
95                 int tc = tc0[i];
96                 int delta;
97                 if( abs( p2 - p0 ) < beta )
98                 {
99                     if( tc0[i] )
100                         pix[-2*xstride] = p1 + x264_clip3( (( p2 + ((p0 + q0 + 1) >> 1)) >> 1) - p1, -tc0[i], tc0[i] );
101                     tc++;
102                 }
103                 if( abs( q2 - q0 ) < beta )
104                 {
105                     if( tc0[i] )
106                         pix[ 1*xstride] = q1 + x264_clip3( (( q2 + ((p0 + q0 + 1) >> 1)) >> 1) - q1, -tc0[i], tc0[i] );
107                     tc++;
108                 }
109
110                 delta = x264_clip3( (((q0 - p0 ) << 2) + (p1 - q1) + 4) >> 3, -tc, tc );
111                 pix[-1*xstride] = x264_clip_pixel( p0 + delta );    /* p0' */
112                 pix[ 0*xstride] = x264_clip_pixel( q0 - delta );    /* q0' */
113             }
114             pix += ystride;
115         }
116     }
117 }
118 static void deblock_v_luma_c( pixel *pix, int stride, int alpha, int beta, int8_t *tc0 )
119 {
120     deblock_luma_c( pix, stride, 1, alpha, beta, tc0 );
121 }
122 static void deblock_h_luma_c( pixel *pix, int stride, int alpha, int beta, int8_t *tc0 )
123 {
124     deblock_luma_c( pix, 1, stride, alpha, beta, tc0 );
125 }
126
127 static inline void deblock_chroma_c( pixel *pix, int xstride, int ystride, int alpha, int beta, int8_t *tc0 )
128 {
129     for( int i = 0; i < 4; i++ )
130     {
131         int tc = tc0[i];
132         if( tc <= 0 )
133         {
134             pix += 2*ystride;
135             continue;
136         }
137         for( int d = 0; d < 2; d++ )
138         {
139             int p1 = pix[-2*xstride];
140             int p0 = pix[-1*xstride];
141             int q0 = pix[ 0*xstride];
142             int q1 = pix[ 1*xstride];
143
144             if( abs( p0 - q0 ) < alpha && abs( p1 - p0 ) < beta && abs( q1 - q0 ) < beta )
145             {
146                 int delta = x264_clip3( (((q0 - p0 ) << 2) + (p1 - q1) + 4) >> 3, -tc, tc );
147                 pix[-1*xstride] = x264_clip_pixel( p0 + delta );    /* p0' */
148                 pix[ 0*xstride] = x264_clip_pixel( q0 - delta );    /* q0' */
149             }
150             pix += ystride;
151         }
152     }
153 }
154 static void deblock_v_chroma_c( pixel *pix, int stride, int alpha, int beta, int8_t *tc0 )
155 {
156     deblock_chroma_c( pix, stride, 1, alpha, beta, tc0 );
157 }
158 static void deblock_h_chroma_c( pixel *pix, int stride, int alpha, int beta, int8_t *tc0 )
159 {
160     deblock_chroma_c( pix, 1, stride, alpha, beta, tc0 );
161 }
162
163 static inline void deblock_luma_intra_c( pixel *pix, int xstride, int ystride, int alpha, int beta )
164 {
165     for( int d = 0; d < 16; d++ )
166     {
167         int p2 = pix[-3*xstride];
168         int p1 = pix[-2*xstride];
169         int p0 = pix[-1*xstride];
170         int q0 = pix[ 0*xstride];
171         int q1 = pix[ 1*xstride];
172         int q2 = pix[ 2*xstride];
173
174         if( abs( p0 - q0 ) < alpha && abs( p1 - p0 ) < beta && abs( q1 - q0 ) < beta )
175         {
176             if(abs( p0 - q0 ) < ((alpha >> 2) + 2) )
177             {
178                 if( abs( p2 - p0 ) < beta ) /* p0', p1', p2' */
179                 {
180                     const int p3 = pix[-4*xstride];
181                     pix[-1*xstride] = ( p2 + 2*p1 + 2*p0 + 2*q0 + q1 + 4 ) >> 3;
182                     pix[-2*xstride] = ( p2 + p1 + p0 + q0 + 2 ) >> 2;
183                     pix[-3*xstride] = ( 2*p3 + 3*p2 + p1 + p0 + q0 + 4 ) >> 3;
184                 }
185                 else /* p0' */
186                     pix[-1*xstride] = ( 2*p1 + p0 + q1 + 2 ) >> 2;
187                 if( abs( q2 - q0 ) < beta ) /* q0', q1', q2' */
188                 {
189                     const int q3 = pix[3*xstride];
190                     pix[0*xstride] = ( p1 + 2*p0 + 2*q0 + 2*q1 + q2 + 4 ) >> 3;
191                     pix[1*xstride] = ( p0 + q0 + q1 + q2 + 2 ) >> 2;
192                     pix[2*xstride] = ( 2*q3 + 3*q2 + q1 + q0 + p0 + 4 ) >> 3;
193                 }
194                 else /* q0' */
195                     pix[0*xstride] = ( 2*q1 + q0 + p1 + 2 ) >> 2;
196             }
197             else /* p0', q0' */
198             {
199                 pix[-1*xstride] = ( 2*p1 + p0 + q1 + 2 ) >> 2;
200                 pix[ 0*xstride] = ( 2*q1 + q0 + p1 + 2 ) >> 2;
201             }
202         }
203         pix += ystride;
204     }
205 }
206 static void deblock_v_luma_intra_c( pixel *pix, int stride, int alpha, int beta )
207 {
208     deblock_luma_intra_c( pix, stride, 1, alpha, beta );
209 }
210 static void deblock_h_luma_intra_c( pixel *pix, int stride, int alpha, int beta )
211 {
212     deblock_luma_intra_c( pix, 1, stride, alpha, beta );
213 }
214
215 static inline void deblock_chroma_intra_c( pixel *pix, int xstride, int ystride, int alpha, int beta )
216 {
217     for( int d = 0; d < 8; d++ )
218     {
219         int p1 = pix[-2*xstride];
220         int p0 = pix[-1*xstride];
221         int q0 = pix[ 0*xstride];
222         int q1 = pix[ 1*xstride];
223
224         if( abs( p0 - q0 ) < alpha && abs( p1 - p0 ) < beta && abs( q1 - q0 ) < beta )
225         {
226             pix[-1*xstride] = (2*p1 + p0 + q1 + 2) >> 2;   /* p0' */
227             pix[ 0*xstride] = (2*q1 + q0 + p1 + 2) >> 2;   /* q0' */
228         }
229         pix += ystride;
230     }
231 }
232 static void deblock_v_chroma_intra_c( pixel *pix, int stride, int alpha, int beta )
233 {
234     deblock_chroma_intra_c( pix, stride, 1, alpha, beta );
235 }
236 static void deblock_h_chroma_intra_c( pixel *pix, int stride, int alpha, int beta )
237 {
238     deblock_chroma_intra_c( pix, 1, stride, alpha, beta );
239 }
240
241 static void deblock_strength_c( uint8_t nnz[X264_SCAN8_SIZE], int8_t ref[2][X264_SCAN8_LUMA_SIZE],
242                                 int16_t mv[2][X264_SCAN8_LUMA_SIZE][2], uint8_t bs[2][4][4], int mvy_limit,
243                                 int bframe )
244 {
245     for( int dir = 0; dir < 2; dir++ )
246     {
247         int s1 = dir ? 1 : 8;
248         int s2 = dir ? 8 : 1;
249         for( int edge = 0; edge < 4; edge++ )
250             for( int i = 0, loc = X264_SCAN8_0+edge*s2; i < 4; i++, loc += s1 )
251             {
252                 int locn = loc - s2;
253                 if( nnz[loc] || nnz[locn] )
254                     bs[dir][edge][i] = 2;
255                 else if( ref[0][loc] != ref[0][locn] ||
256                          abs( mv[0][loc][0] - mv[0][locn][0] ) >= 4 ||
257                          abs( mv[0][loc][1] - mv[0][locn][1] ) >= mvy_limit ||
258                         (bframe && (ref[1][loc] != ref[1][locn] ||
259                          abs( mv[1][loc][0] - mv[1][locn][0] ) >= 4 ||
260                          abs( mv[1][loc][1] - mv[1][locn][1] ) >= mvy_limit )))
261                 {
262                     bs[dir][edge][i] = 1;
263                 }
264                 else
265                     bs[dir][edge][i] = 0;
266             }
267     }
268 }
269
270 static inline void deblock_edge( x264_t *h, pixel *pix1, pixel *pix2, int i_stride, uint8_t bS[4], int i_qp, int b_chroma, x264_deblock_inter_t pf_inter )
271 {
272     int index_a = i_qp-QP_BD_OFFSET + h->sh.i_alpha_c0_offset;
273     int index_b = i_qp-QP_BD_OFFSET + h->sh.i_beta_offset;
274     int alpha = alpha_table(index_a) << (BIT_DEPTH-8);
275     int beta  = beta_table(index_b) << (BIT_DEPTH-8);
276     int8_t tc[4];
277
278     if( !M32(bS) || !alpha || !beta )
279         return;
280
281     tc[0] = (tc0_table(index_a)[bS[0]] << (BIT_DEPTH-8)) + b_chroma;
282     tc[1] = (tc0_table(index_a)[bS[1]] << (BIT_DEPTH-8)) + b_chroma;
283     tc[2] = (tc0_table(index_a)[bS[2]] << (BIT_DEPTH-8)) + b_chroma;
284     tc[3] = (tc0_table(index_a)[bS[3]] << (BIT_DEPTH-8)) + b_chroma;
285
286     pf_inter( pix1, i_stride, alpha, beta, tc );
287     if( b_chroma )
288         pf_inter( pix2, i_stride, alpha, beta, tc );
289 }
290
291 static inline void deblock_edge_intra( x264_t *h, pixel *pix1, pixel *pix2, int i_stride, uint8_t bS[4], int i_qp, int b_chroma, x264_deblock_intra_t pf_intra )
292 {
293     int index_a = i_qp-QP_BD_OFFSET + h->sh.i_alpha_c0_offset;
294     int index_b = i_qp-QP_BD_OFFSET + h->sh.i_beta_offset;
295     int alpha = alpha_table(index_a) << (BIT_DEPTH-8);
296     int beta  = beta_table(index_b) << (BIT_DEPTH-8);
297
298     if( !alpha || !beta )
299         return;
300
301     pf_intra( pix1, i_stride, alpha, beta );
302     if( b_chroma )
303         pf_intra( pix2, i_stride, alpha, beta );
304 }
305
306 void x264_frame_deblock_row( x264_t *h, int mb_y )
307 {
308     int b_interlaced = h->sh.b_mbaff;
309     int qp_thresh = 15 - X264_MIN( h->sh.i_alpha_c0_offset, h->sh.i_beta_offset ) - X264_MAX( 0, h->param.analyse.i_chroma_qp_offset );
310     int stridey   = h->fdec->i_stride[0];
311     int stride2y  = stridey << b_interlaced;
312     int strideuv  = h->fdec->i_stride[1];
313     int stride2uv = strideuv << b_interlaced;
314
315     for( int mb_x = 0; mb_x < h->mb.i_mb_width; mb_x += (~b_interlaced | mb_y)&1, mb_y ^= b_interlaced )
316     {
317         x264_prefetch_fenc( h, h->fdec, mb_x, mb_y );
318         x264_macroblock_cache_load_neighbours_deblock( h, mb_x, mb_y );
319
320         int mb_xy = h->mb.i_mb_xy;
321         int transform_8x8 = h->mb.mb_transform_size[h->mb.i_mb_xy];
322         int intra_cur = IS_INTRA( h->mb.type[mb_xy] );
323         uint8_t (*bs)[4][4] = h->deblock_strength[mb_y&b_interlaced][mb_x];
324
325         pixel *pixy = h->fdec->plane[0] + 16*mb_y*stridey  + 16*mb_x;
326         pixel *pixu = h->fdec->plane[1] +  8*mb_y*strideuv +  8*mb_x;
327         pixel *pixv = h->fdec->plane[2] +  8*mb_y*strideuv +  8*mb_x;
328         if( mb_y & b_interlaced )
329         {
330             pixy -= 15*stridey;
331             pixu -=  7*strideuv;
332             pixv -=  7*strideuv;
333         }
334
335         int qp = h->mb.qp[mb_xy];
336         int qpc = h->chroma_qp_table[qp];
337         int first_edge_only = h->mb.type[mb_xy] == P_SKIP || qp <= qp_thresh;
338
339         #define FILTER( intra, dir, edge, qp, chroma_qp )\
340         do\
341         {\
342             deblock_edge##intra( h, pixy + 4*edge*(dir?stride2y:1), NULL,\
343                                  stride2y, bs[dir][edge], qp, 0,\
344                                  h->loopf.deblock_luma##intra[dir] );\
345             if( !(edge & 1) )\
346                 deblock_edge##intra( h, pixu + 2*edge*(dir?stride2uv:1), pixv + 2*edge*(dir?stride2uv:1),\
347                                      stride2uv, bs[dir][edge], chroma_qp, 1,\
348                                      h->loopf.deblock_chroma##intra[dir] );\
349         } while(0)
350
351         if( h->mb.i_neighbour & MB_LEFT )
352         {
353             int qpl = h->mb.qp[h->mb.i_mb_left_xy];
354             int qp_left = (qp + qpl + 1) >> 1;
355             int qpc_left = (h->chroma_qp_table[qp] + h->chroma_qp_table[qpl] + 1) >> 1;
356             int intra_left = IS_INTRA( h->mb.type[h->mb.i_mb_left_xy] );
357             if( intra_cur || intra_left )
358                 FILTER( _intra, 0, 0, qp_left, qpc_left );
359             else
360                 FILTER(       , 0, 0, qp_left, qpc_left );
361         }
362
363         if( !first_edge_only )
364         {
365             if( !transform_8x8 ) FILTER( , 0, 1, qp, qpc );
366                                  FILTER( , 0, 2, qp, qpc );
367             if( !transform_8x8 ) FILTER( , 0, 3, qp, qpc );
368         }
369
370         if( h->mb.i_neighbour & MB_TOP )
371         {
372             int qpt = h->mb.qp[h->mb.i_mb_top_xy];
373             int qp_top = (qp + qpt + 1) >> 1;
374             int qpc_top = (h->chroma_qp_table[qp] + h->chroma_qp_table[qpt] + 1) >> 1;
375             int intra_top = IS_INTRA( h->mb.type[h->mb.i_mb_top_xy] );
376             if( ~b_interlaced & (intra_cur | intra_top) )
377                 FILTER( _intra, 1, 0, qp_top, qpc_top );
378             else
379             {
380                 if( intra_top )
381                     M32( bs[1][0] ) = 0x03030303;
382                 FILTER(       , 1, 0, qp_top, qpc_top );
383             }
384         }
385
386         if( !first_edge_only )
387         {
388             if( !transform_8x8 ) FILTER( , 1, 1, qp, qpc );
389                                  FILTER( , 1, 2, qp, qpc );
390             if( !transform_8x8 ) FILTER( , 1, 3, qp, qpc );
391         }
392     }
393 }
394
395 #if HAVE_MMX
396 void x264_deblock_v_chroma_mmxext( uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0 );
397 void x264_deblock_h_chroma_mmxext( uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0 );
398 void x264_deblock_v_chroma_intra_mmxext( uint8_t *pix, int stride, int alpha, int beta );
399 void x264_deblock_h_chroma_intra_mmxext( uint8_t *pix, int stride, int alpha, int beta );
400
401 void x264_deblock_v_luma_sse2( uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0 );
402 void x264_deblock_h_luma_sse2( uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0 );
403 void x264_deblock_v_luma_intra_sse2( uint8_t *pix, int stride, int alpha, int beta );
404 void x264_deblock_h_luma_intra_sse2( uint8_t *pix, int stride, int alpha, int beta );
405 void x264_deblock_strength_mmxext( uint8_t nnz[X264_SCAN8_SIZE], int8_t ref[2][X264_SCAN8_LUMA_SIZE],
406                                    int16_t mv[2][X264_SCAN8_LUMA_SIZE][2], uint8_t bs[2][4][4],
407                                    int mvy_limit, int bframe );
408 void x264_deblock_strength_sse2  ( uint8_t nnz[X264_SCAN8_SIZE], int8_t ref[2][X264_SCAN8_LUMA_SIZE],
409                                    int16_t mv[2][X264_SCAN8_LUMA_SIZE][2], uint8_t bs[2][4][4],
410                                    int mvy_limit, int bframe );
411 void x264_deblock_strength_ssse3 ( uint8_t nnz[X264_SCAN8_SIZE], int8_t ref[2][X264_SCAN8_LUMA_SIZE],
412                                    int16_t mv[2][X264_SCAN8_LUMA_SIZE][2], uint8_t bs[2][4][4],
413                                    int mvy_limit, int bframe );
414 #if ARCH_X86
415 void x264_deblock_h_luma_mmxext( uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0 );
416 void x264_deblock_v8_luma_mmxext( uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0 );
417 void x264_deblock_h_luma_intra_mmxext( uint8_t *pix, int stride, int alpha, int beta );
418 void x264_deblock_v8_luma_intra_mmxext( uint8_t *pix, int stride, int alpha, int beta );
419
420 static void x264_deblock_v_luma_mmxext( uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0 )
421 {
422     x264_deblock_v8_luma_mmxext( pix,   stride, alpha, beta, tc0   );
423     x264_deblock_v8_luma_mmxext( pix+8, stride, alpha, beta, tc0+2 );
424 }
425 static void x264_deblock_v_luma_intra_mmxext( uint8_t *pix, int stride, int alpha, int beta )
426 {
427     x264_deblock_v8_luma_intra_mmxext( pix,   stride, alpha, beta );
428     x264_deblock_v8_luma_intra_mmxext( pix+8, stride, alpha, beta );
429 }
430 #endif
431 #endif
432
433 #if ARCH_PPC
434 void x264_deblock_v_luma_altivec( uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0 );
435 void x264_deblock_h_luma_altivec( uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0 );
436 #endif // ARCH_PPC
437
438 #if HAVE_ARMV6
439 void x264_deblock_v_luma_neon( uint8_t *, int, int, int, int8_t * );
440 void x264_deblock_h_luma_neon( uint8_t *, int, int, int, int8_t * );
441 void x264_deblock_v_chroma_neon( uint8_t *, int, int, int, int8_t * );
442 void x264_deblock_h_chroma_neon( uint8_t *, int, int, int, int8_t * );
443 #endif
444
445 void x264_deblock_init( int cpu, x264_deblock_function_t *pf )
446 {
447     pf->deblock_luma[1] = deblock_v_luma_c;
448     pf->deblock_luma[0] = deblock_h_luma_c;
449     pf->deblock_chroma[1] = deblock_v_chroma_c;
450     pf->deblock_chroma[0] = deblock_h_chroma_c;
451     pf->deblock_luma_intra[1] = deblock_v_luma_intra_c;
452     pf->deblock_luma_intra[0] = deblock_h_luma_intra_c;
453     pf->deblock_chroma_intra[1] = deblock_v_chroma_intra_c;
454     pf->deblock_chroma_intra[0] = deblock_h_chroma_intra_c;
455     pf->deblock_strength = deblock_strength_c;
456
457 #if HAVE_MMX
458     if( cpu&X264_CPU_MMXEXT )
459     {
460 #if !X264_HIGH_BIT_DEPTH
461         pf->deblock_chroma[1] = x264_deblock_v_chroma_mmxext;
462         pf->deblock_chroma[0] = x264_deblock_h_chroma_mmxext;
463         pf->deblock_chroma_intra[1] = x264_deblock_v_chroma_intra_mmxext;
464         pf->deblock_chroma_intra[0] = x264_deblock_h_chroma_intra_mmxext;
465 #if ARCH_X86
466         pf->deblock_luma[1] = x264_deblock_v_luma_mmxext;
467         pf->deblock_luma[0] = x264_deblock_h_luma_mmxext;
468         pf->deblock_luma_intra[1] = x264_deblock_v_luma_intra_mmxext;
469         pf->deblock_luma_intra[0] = x264_deblock_h_luma_intra_mmxext;
470 #endif
471 #endif // !X264_HIGH_BIT_DEPTH
472         pf->deblock_strength = x264_deblock_strength_mmxext;
473         if( cpu&X264_CPU_SSE2 )
474         {
475             pf->deblock_strength = x264_deblock_strength_sse2;
476 #if !X264_HIGH_BIT_DEPTH
477             if( !(cpu&X264_CPU_STACK_MOD4) )
478             {
479                 pf->deblock_luma[1] = x264_deblock_v_luma_sse2;
480                 pf->deblock_luma[0] = x264_deblock_h_luma_sse2;
481                 pf->deblock_luma_intra[1] = x264_deblock_v_luma_intra_sse2;
482                 pf->deblock_luma_intra[0] = x264_deblock_h_luma_intra_sse2;
483             }
484 #endif // !X264_HIGH_BIT_DEPTH
485         }
486         if( cpu&X264_CPU_SSSE3 )
487             pf->deblock_strength = x264_deblock_strength_ssse3;
488     }
489 #endif
490
491 #if !X264_HIGH_BIT_DEPTH
492 #if HAVE_ALTIVEC
493     if( cpu&X264_CPU_ALTIVEC )
494     {
495         pf->deblock_luma[1] = x264_deblock_v_luma_altivec;
496         pf->deblock_luma[0] = x264_deblock_h_luma_altivec;
497    }
498 #endif // HAVE_ALTIVEC
499
500 #if HAVE_ARMV6
501    if( cpu&X264_CPU_NEON )
502    {
503         pf->deblock_luma[1] = x264_deblock_v_luma_neon;
504         pf->deblock_luma[0] = x264_deblock_h_luma_neon;
505         pf->deblock_chroma[1] = x264_deblock_v_chroma_neon;
506         pf->deblock_chroma[0] = x264_deblock_h_chroma_neon;
507    }
508 #endif
509 #endif // !X264_HIGH_BIT_DEPTH
510 }