FFmpeg
 All Data Structures Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
h264_qpel.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2004 Romain Dolbeau <romain@dolbeau.org>
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20 
21 #include "config.h"
22 #include "libavutil/attributes.h"
23 #include "libavcodec/h264qpel.h"
24 
25 #if HAVE_ALTIVEC
26 #include "libavutil/cpu.h"
27 #include "libavutil/intreadwrite.h"
30 #include "dsputil_altivec.h"
31 
32 #define PUT_OP_U8_ALTIVEC(d, s, dst) d = s
33 #define AVG_OP_U8_ALTIVEC(d, s, dst) d = vec_avg(dst, s)
34 
35 #define OP_U8_ALTIVEC PUT_OP_U8_ALTIVEC
36 #define PREFIX_h264_qpel16_h_lowpass_altivec put_h264_qpel16_h_lowpass_altivec
37 #define PREFIX_h264_qpel16_h_lowpass_num altivec_put_h264_qpel16_h_lowpass_num
38 #define PREFIX_h264_qpel16_v_lowpass_altivec put_h264_qpel16_v_lowpass_altivec
39 #define PREFIX_h264_qpel16_v_lowpass_num altivec_put_h264_qpel16_v_lowpass_num
40 #define PREFIX_h264_qpel16_hv_lowpass_altivec put_h264_qpel16_hv_lowpass_altivec
41 #define PREFIX_h264_qpel16_hv_lowpass_num altivec_put_h264_qpel16_hv_lowpass_num
42 #include "h264_qpel_template.c"
43 #undef OP_U8_ALTIVEC
44 #undef PREFIX_h264_qpel16_h_lowpass_altivec
45 #undef PREFIX_h264_qpel16_h_lowpass_num
46 #undef PREFIX_h264_qpel16_v_lowpass_altivec
47 #undef PREFIX_h264_qpel16_v_lowpass_num
48 #undef PREFIX_h264_qpel16_hv_lowpass_altivec
49 #undef PREFIX_h264_qpel16_hv_lowpass_num
50 
51 #define OP_U8_ALTIVEC AVG_OP_U8_ALTIVEC
52 #define PREFIX_h264_qpel16_h_lowpass_altivec avg_h264_qpel16_h_lowpass_altivec
53 #define PREFIX_h264_qpel16_h_lowpass_num altivec_avg_h264_qpel16_h_lowpass_num
54 #define PREFIX_h264_qpel16_v_lowpass_altivec avg_h264_qpel16_v_lowpass_altivec
55 #define PREFIX_h264_qpel16_v_lowpass_num altivec_avg_h264_qpel16_v_lowpass_num
56 #define PREFIX_h264_qpel16_hv_lowpass_altivec avg_h264_qpel16_hv_lowpass_altivec
57 #define PREFIX_h264_qpel16_hv_lowpass_num altivec_avg_h264_qpel16_hv_lowpass_num
58 #include "h264_qpel_template.c"
59 #undef OP_U8_ALTIVEC
60 #undef PREFIX_h264_qpel16_h_lowpass_altivec
61 #undef PREFIX_h264_qpel16_h_lowpass_num
62 #undef PREFIX_h264_qpel16_v_lowpass_altivec
63 #undef PREFIX_h264_qpel16_v_lowpass_num
64 #undef PREFIX_h264_qpel16_hv_lowpass_altivec
65 #undef PREFIX_h264_qpel16_hv_lowpass_num
66 
67 #define H264_MC(OPNAME, SIZE, CODETYPE) \
68 static void OPNAME ## h264_qpel ## SIZE ## _mc00_ ## CODETYPE (uint8_t *dst, uint8_t *src, int stride){\
69  ff_ ## OPNAME ## pixels ## SIZE ## _ ## CODETYPE(dst, src, stride, SIZE);\
70 }\
71 \
72 static void OPNAME ## h264_qpel ## SIZE ## _mc10_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){ \
73  DECLARE_ALIGNED(16, uint8_t, half)[SIZE*SIZE];\
74  put_h264_qpel ## SIZE ## _h_lowpass_ ## CODETYPE(half, src, SIZE, stride);\
75  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, src, half, stride, stride, SIZE);\
76 }\
77 \
78 static void OPNAME ## h264_qpel ## SIZE ## _mc20_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
79  OPNAME ## h264_qpel ## SIZE ## _h_lowpass_ ## CODETYPE(dst, src, stride, stride);\
80 }\
81 \
82 static void OPNAME ## h264_qpel ## SIZE ## _mc30_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
83  DECLARE_ALIGNED(16, uint8_t, half)[SIZE*SIZE];\
84  put_h264_qpel ## SIZE ## _h_lowpass_ ## CODETYPE(half, src, SIZE, stride);\
85  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, src+1, half, stride, stride, SIZE);\
86 }\
87 \
88 static void OPNAME ## h264_qpel ## SIZE ## _mc01_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
89  DECLARE_ALIGNED(16, uint8_t, half)[SIZE*SIZE];\
90  put_h264_qpel ## SIZE ## _v_lowpass_ ## CODETYPE(half, src, SIZE, stride);\
91  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, src, half, stride, stride, SIZE);\
92 }\
93 \
94 static void OPNAME ## h264_qpel ## SIZE ## _mc02_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
95  OPNAME ## h264_qpel ## SIZE ## _v_lowpass_ ## CODETYPE(dst, src, stride, stride);\
96 }\
97 \
98 static void OPNAME ## h264_qpel ## SIZE ## _mc03_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
99  DECLARE_ALIGNED(16, uint8_t, half)[SIZE*SIZE];\
100  put_h264_qpel ## SIZE ## _v_lowpass_ ## CODETYPE(half, src, SIZE, stride);\
101  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, src+stride, half, stride, stride, SIZE);\
102 }\
103 \
104 static void OPNAME ## h264_qpel ## SIZE ## _mc11_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
105  DECLARE_ALIGNED(16, uint8_t, halfH)[SIZE*SIZE];\
106  DECLARE_ALIGNED(16, uint8_t, halfV)[SIZE*SIZE];\
107  put_h264_qpel ## SIZE ## _h_lowpass_ ## CODETYPE(halfH, src, SIZE, stride);\
108  put_h264_qpel ## SIZE ## _v_lowpass_ ## CODETYPE(halfV, src, SIZE, stride);\
109  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, halfH, halfV, stride, SIZE, SIZE);\
110 }\
111 \
112 static void OPNAME ## h264_qpel ## SIZE ## _mc31_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
113  DECLARE_ALIGNED(16, uint8_t, halfH)[SIZE*SIZE];\
114  DECLARE_ALIGNED(16, uint8_t, halfV)[SIZE*SIZE];\
115  put_h264_qpel ## SIZE ## _h_lowpass_ ## CODETYPE(halfH, src, SIZE, stride);\
116  put_h264_qpel ## SIZE ## _v_lowpass_ ## CODETYPE(halfV, src+1, SIZE, stride);\
117  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, halfH, halfV, stride, SIZE, SIZE);\
118 }\
119 \
120 static void OPNAME ## h264_qpel ## SIZE ## _mc13_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
121  DECLARE_ALIGNED(16, uint8_t, halfH)[SIZE*SIZE];\
122  DECLARE_ALIGNED(16, uint8_t, halfV)[SIZE*SIZE];\
123  put_h264_qpel ## SIZE ## _h_lowpass_ ## CODETYPE(halfH, src + stride, SIZE, stride);\
124  put_h264_qpel ## SIZE ## _v_lowpass_ ## CODETYPE(halfV, src, SIZE, stride);\
125  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, halfH, halfV, stride, SIZE, SIZE);\
126 }\
127 \
128 static void OPNAME ## h264_qpel ## SIZE ## _mc33_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
129  DECLARE_ALIGNED(16, uint8_t, halfH)[SIZE*SIZE];\
130  DECLARE_ALIGNED(16, uint8_t, halfV)[SIZE*SIZE];\
131  put_h264_qpel ## SIZE ## _h_lowpass_ ## CODETYPE(halfH, src + stride, SIZE, stride);\
132  put_h264_qpel ## SIZE ## _v_lowpass_ ## CODETYPE(halfV, src+1, SIZE, stride);\
133  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, halfH, halfV, stride, SIZE, SIZE);\
134 }\
135 \
136 static void OPNAME ## h264_qpel ## SIZE ## _mc22_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
137  DECLARE_ALIGNED(16, int16_t, tmp)[SIZE*(SIZE+8)];\
138  OPNAME ## h264_qpel ## SIZE ## _hv_lowpass_ ## CODETYPE(dst, tmp, src, stride, SIZE, stride);\
139 }\
140 \
141 static void OPNAME ## h264_qpel ## SIZE ## _mc21_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
142  DECLARE_ALIGNED(16, uint8_t, halfH)[SIZE*SIZE];\
143  DECLARE_ALIGNED(16, uint8_t, halfHV)[SIZE*SIZE];\
144  DECLARE_ALIGNED(16, int16_t, tmp)[SIZE*(SIZE+8)];\
145  put_h264_qpel ## SIZE ## _h_lowpass_ ## CODETYPE(halfH, src, SIZE, stride);\
146  put_h264_qpel ## SIZE ## _hv_lowpass_ ## CODETYPE(halfHV, tmp, src, SIZE, SIZE, stride);\
147  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, halfH, halfHV, stride, SIZE, SIZE);\
148 }\
149 \
150 static void OPNAME ## h264_qpel ## SIZE ## _mc23_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
151  DECLARE_ALIGNED(16, uint8_t, halfH)[SIZE*SIZE];\
152  DECLARE_ALIGNED(16, uint8_t, halfHV)[SIZE*SIZE];\
153  DECLARE_ALIGNED(16, int16_t, tmp)[SIZE*(SIZE+8)];\
154  put_h264_qpel ## SIZE ## _h_lowpass_ ## CODETYPE(halfH, src + stride, SIZE, stride);\
155  put_h264_qpel ## SIZE ## _hv_lowpass_ ## CODETYPE(halfHV, tmp, src, SIZE, SIZE, stride);\
156  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, halfH, halfHV, stride, SIZE, SIZE);\
157 }\
158 \
159 static void OPNAME ## h264_qpel ## SIZE ## _mc12_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
160  DECLARE_ALIGNED(16, uint8_t, halfV)[SIZE*SIZE];\
161  DECLARE_ALIGNED(16, uint8_t, halfHV)[SIZE*SIZE];\
162  DECLARE_ALIGNED(16, int16_t, tmp)[SIZE*(SIZE+8)];\
163  put_h264_qpel ## SIZE ## _v_lowpass_ ## CODETYPE(halfV, src, SIZE, stride);\
164  put_h264_qpel ## SIZE ## _hv_lowpass_ ## CODETYPE(halfHV, tmp, src, SIZE, SIZE, stride);\
165  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, halfV, halfHV, stride, SIZE, SIZE);\
166 }\
167 \
168 static void OPNAME ## h264_qpel ## SIZE ## _mc32_ ## CODETYPE(uint8_t *dst, uint8_t *src, int stride){\
169  DECLARE_ALIGNED(16, uint8_t, halfV)[SIZE*SIZE];\
170  DECLARE_ALIGNED(16, uint8_t, halfHV)[SIZE*SIZE];\
171  DECLARE_ALIGNED(16, int16_t, tmp)[SIZE*(SIZE+8)];\
172  put_h264_qpel ## SIZE ## _v_lowpass_ ## CODETYPE(halfV, src+1, SIZE, stride);\
173  put_h264_qpel ## SIZE ## _hv_lowpass_ ## CODETYPE(halfHV, tmp, src, SIZE, SIZE, stride);\
174  OPNAME ## pixels ## SIZE ## _l2_ ## CODETYPE(dst, halfV, halfHV, stride, SIZE, SIZE);\
175 }\
176 
177 static inline void put_pixels16_l2_altivec( uint8_t * dst, const uint8_t * src1,
178  const uint8_t * src2, int dst_stride,
179  int src_stride1, int h)
180 {
181  int i;
182  vec_u8 a, b, d, tmp1, tmp2, mask, mask_, edges, align;
183 
184  mask_ = vec_lvsl(0, src2);
185 
186  for (i = 0; i < h; i++) {
187 
188  tmp1 = vec_ld(i * src_stride1, src1);
189  mask = vec_lvsl(i * src_stride1, src1);
190  tmp2 = vec_ld(i * src_stride1 + 15, src1);
191 
192  a = vec_perm(tmp1, tmp2, mask);
193 
194  tmp1 = vec_ld(i * 16, src2);
195  tmp2 = vec_ld(i * 16 + 15, src2);
196 
197  b = vec_perm(tmp1, tmp2, mask_);
198 
199  tmp1 = vec_ld(0, dst);
200  mask = vec_lvsl(0, dst);
201  tmp2 = vec_ld(15, dst);
202 
203  d = vec_avg(a, b);
204 
205  edges = vec_perm(tmp2, tmp1, mask);
206 
207  align = vec_lvsr(0, dst);
208 
209  tmp2 = vec_perm(d, edges, align);
210  tmp1 = vec_perm(edges, d, align);
211 
212  vec_st(tmp2, 15, dst);
213  vec_st(tmp1, 0 , dst);
214 
215  dst += dst_stride;
216  }
217 }
218 
219 static inline void avg_pixels16_l2_altivec( uint8_t * dst, const uint8_t * src1,
220  const uint8_t * src2, int dst_stride,
221  int src_stride1, int h)
222 {
223  int i;
224  vec_u8 a, b, d, tmp1, tmp2, mask, mask_, edges, align;
225 
226  mask_ = vec_lvsl(0, src2);
227 
228  for (i = 0; i < h; i++) {
229 
230  tmp1 = vec_ld(i * src_stride1, src1);
231  mask = vec_lvsl(i * src_stride1, src1);
232  tmp2 = vec_ld(i * src_stride1 + 15, src1);
233 
234  a = vec_perm(tmp1, tmp2, mask);
235 
236  tmp1 = vec_ld(i * 16, src2);
237  tmp2 = vec_ld(i * 16 + 15, src2);
238 
239  b = vec_perm(tmp1, tmp2, mask_);
240 
241  tmp1 = vec_ld(0, dst);
242  mask = vec_lvsl(0, dst);
243  tmp2 = vec_ld(15, dst);
244 
245  d = vec_avg(vec_perm(tmp1, tmp2, mask), vec_avg(a, b));
246 
247  edges = vec_perm(tmp2, tmp1, mask);
248 
249  align = vec_lvsr(0, dst);
250 
251  tmp2 = vec_perm(d, edges, align);
252  tmp1 = vec_perm(edges, d, align);
253 
254  vec_st(tmp2, 15, dst);
255  vec_st(tmp1, 0 , dst);
256 
257  dst += dst_stride;
258  }
259 }
260 
261 /* Implemented but could be faster
262 #define put_pixels16_l2_altivec(d,s1,s2,ds,s1s,h) put_pixels16_l2(d,s1,s2,ds,s1s,16,h)
263 #define avg_pixels16_l2_altivec(d,s1,s2,ds,s1s,h) avg_pixels16_l2(d,s1,s2,ds,s1s,16,h)
264  */
265 
266 H264_MC(put_, 16, altivec)
267 H264_MC(avg_, 16, altivec)
268 #endif /* HAVE_ALTIVEC */
269 
271 {
272 #if HAVE_ALTIVEC
273  const int high_bit_depth = bit_depth > 8;
274 
276  if (!high_bit_depth) {
277 #define dspfunc(PFX, IDX, NUM) \
278  c->PFX ## _pixels_tab[IDX][ 0] = PFX ## NUM ## _mc00_altivec; \
279  c->PFX ## _pixels_tab[IDX][ 1] = PFX ## NUM ## _mc10_altivec; \
280  c->PFX ## _pixels_tab[IDX][ 2] = PFX ## NUM ## _mc20_altivec; \
281  c->PFX ## _pixels_tab[IDX][ 3] = PFX ## NUM ## _mc30_altivec; \
282  c->PFX ## _pixels_tab[IDX][ 4] = PFX ## NUM ## _mc01_altivec; \
283  c->PFX ## _pixels_tab[IDX][ 5] = PFX ## NUM ## _mc11_altivec; \
284  c->PFX ## _pixels_tab[IDX][ 6] = PFX ## NUM ## _mc21_altivec; \
285  c->PFX ## _pixels_tab[IDX][ 7] = PFX ## NUM ## _mc31_altivec; \
286  c->PFX ## _pixels_tab[IDX][ 8] = PFX ## NUM ## _mc02_altivec; \
287  c->PFX ## _pixels_tab[IDX][ 9] = PFX ## NUM ## _mc12_altivec; \
288  c->PFX ## _pixels_tab[IDX][10] = PFX ## NUM ## _mc22_altivec; \
289  c->PFX ## _pixels_tab[IDX][11] = PFX ## NUM ## _mc32_altivec; \
290  c->PFX ## _pixels_tab[IDX][12] = PFX ## NUM ## _mc03_altivec; \
291  c->PFX ## _pixels_tab[IDX][13] = PFX ## NUM ## _mc13_altivec; \
292  c->PFX ## _pixels_tab[IDX][14] = PFX ## NUM ## _mc23_altivec; \
293  c->PFX ## _pixels_tab[IDX][15] = PFX ## NUM ## _mc33_altivec
294 
295  dspfunc(put_h264_qpel, 0, 16);
296  dspfunc(avg_h264_qpel, 0, 16);
297 #undef dspfunc
298  }
299  }
300 #endif /* HAVE_ALTIVEC */
301 }