dsputil: Remove hpel functions (moved to hpeldsp)
[ffmpeg.git] / libavcodec / dsputil_template.c
1 /*
2  * DSP utils
3  * Copyright (c) 2000, 2001 Fabrice Bellard
4  * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
5  *
6  * gmc & q-pel & 32/64 bit based MC by Michael Niedermayer <michaelni@gmx.at>
7  *
8  * This file is part of Libav.
9  *
10  * Libav is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU Lesser General Public
12  * License as published by the Free Software Foundation; either
13  * version 2.1 of the License, or (at your option) any later version.
14  *
15  * Libav 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 GNU
18  * Lesser General Public License for more details.
19  *
20  * You should have received a copy of the GNU Lesser General Public
21  * License along with Libav; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23  */
24
25 /**
26  * @file
27  * DSP utils
28  */
29
30 #include "bit_depth_template.c"
31
32 /* draw the edges of width 'w' of an image of size width, height */
33 //FIXME check that this is ok for mpeg4 interlaced
34 static void FUNCC(draw_edges)(uint8_t *_buf, int _wrap, int width, int height, int w, int h, int sides)
35 {
36     pixel *buf = (pixel*)_buf;
37     int wrap = _wrap / sizeof(pixel);
38     pixel *ptr, *last_line;
39     int i;
40
41     /* left and right */
42     ptr = buf;
43     for(i=0;i<height;i++) {
44 #if BIT_DEPTH > 8
45         int j;
46         for (j = 0; j < w; j++) {
47             ptr[j-w] = ptr[0];
48             ptr[j+width] = ptr[width-1];
49         }
50 #else
51         memset(ptr - w, ptr[0], w);
52         memset(ptr + width, ptr[width-1], w);
53 #endif
54         ptr += wrap;
55     }
56
57     /* top and bottom + corners */
58     buf -= w;
59     last_line = buf + (height - 1) * wrap;
60     if (sides & EDGE_TOP)
61         for(i = 0; i < h; i++)
62             memcpy(buf - (i + 1) * wrap, buf, (width + w + w) * sizeof(pixel)); // top
63     if (sides & EDGE_BOTTOM)
64         for (i = 0; i < h; i++)
65             memcpy(last_line + (i + 1) * wrap, last_line, (width + w + w) * sizeof(pixel)); // bottom
66 }
67
68 #define DCTELEM_FUNCS(dctcoef, suffix)                                  \
69 static void FUNCC(get_pixels ## suffix)(int16_t *restrict _block,       \
70                                         const uint8_t *_pixels,         \
71                                         int line_size)                  \
72 {                                                                       \
73     const pixel *pixels = (const pixel *) _pixels;                      \
74     dctcoef *restrict block = (dctcoef *) _block;                       \
75     int i;                                                              \
76                                                                         \
77     /* read the pixels */                                               \
78     for(i=0;i<8;i++) {                                                  \
79         block[0] = pixels[0];                                           \
80         block[1] = pixels[1];                                           \
81         block[2] = pixels[2];                                           \
82         block[3] = pixels[3];                                           \
83         block[4] = pixels[4];                                           \
84         block[5] = pixels[5];                                           \
85         block[6] = pixels[6];                                           \
86         block[7] = pixels[7];                                           \
87         pixels += line_size / sizeof(pixel);                            \
88         block += 8;                                                     \
89     }                                                                   \
90 }                                                                       \
91                                                                         \
92 static void FUNCC(clear_block ## suffix)(int16_t *block)                \
93 {                                                                       \
94     memset(block, 0, sizeof(dctcoef)*64);                               \
95 }                                                                       \
96                                                                         \
97 /**                                                                     \
98  * memset(blocks, 0, sizeof(int16_t)*6*64)                              \
99  */                                                                     \
100 static void FUNCC(clear_blocks ## suffix)(int16_t *blocks)              \
101 {                                                                       \
102     memset(blocks, 0, sizeof(dctcoef)*6*64);                            \
103 }
104
105 DCTELEM_FUNCS(int16_t, _16)
106 #if BIT_DEPTH > 8
107 DCTELEM_FUNCS(dctcoef, _32)
108 #endif
109
110 #if BIT_DEPTH == 8
111 #include "hpel_template.c"
112 #endif
113
114 #define PIXOP2(OPNAME, OP) \
115 static inline void FUNC(OPNAME ## _no_rnd_pixels8_l2)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, int dst_stride, \
116                                                 int src_stride1, int src_stride2, int h){\
117     int i;\
118     for(i=0; i<h; i++){\
119         pixel4 a,b;\
120         a= AV_RN4P(&src1[i*src_stride1  ]);\
121         b= AV_RN4P(&src2[i*src_stride2  ]);\
122         OP(*((pixel4*)&dst[i*dst_stride  ]), no_rnd_avg_pixel4(a, b));\
123         a= AV_RN4P(&src1[i*src_stride1+4*sizeof(pixel)]);\
124         b= AV_RN4P(&src2[i*src_stride2+4*sizeof(pixel)]);\
125         OP(*((pixel4*)&dst[i*dst_stride+4*sizeof(pixel)]), no_rnd_avg_pixel4(a, b));\
126     }\
127 }\
128 \
129 static inline void FUNC(OPNAME ## _no_rnd_pixels16_l2)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, int dst_stride, \
130                                                 int src_stride1, int src_stride2, int h){\
131     FUNC(OPNAME ## _no_rnd_pixels8_l2)(dst  , src1  , src2  , dst_stride, src_stride1, src_stride2, h);\
132     FUNC(OPNAME ## _no_rnd_pixels8_l2)(dst+8*sizeof(pixel), src1+8*sizeof(pixel), src2+8*sizeof(pixel), dst_stride, src_stride1, src_stride2, h);\
133 }\
134 \
135 static inline void FUNC(OPNAME ## _pixels8_l4)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, const uint8_t *src3, const uint8_t *src4,\
136                  int dst_stride, int src_stride1, int src_stride2,int src_stride3,int src_stride4, int h){\
137     /* FIXME HIGH BIT DEPTH */\
138     int i;\
139     for(i=0; i<h; i++){\
140         uint32_t a, b, c, d, l0, l1, h0, h1;\
141         a= AV_RN32(&src1[i*src_stride1]);\
142         b= AV_RN32(&src2[i*src_stride2]);\
143         c= AV_RN32(&src3[i*src_stride3]);\
144         d= AV_RN32(&src4[i*src_stride4]);\
145         l0=  (a&0x03030303UL)\
146            + (b&0x03030303UL)\
147            + 0x02020202UL;\
148         h0= ((a&0xFCFCFCFCUL)>>2)\
149           + ((b&0xFCFCFCFCUL)>>2);\
150         l1=  (c&0x03030303UL)\
151            + (d&0x03030303UL);\
152         h1= ((c&0xFCFCFCFCUL)>>2)\
153           + ((d&0xFCFCFCFCUL)>>2);\
154         OP(*((uint32_t*)&dst[i*dst_stride]), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
155         a= AV_RN32(&src1[i*src_stride1+4]);\
156         b= AV_RN32(&src2[i*src_stride2+4]);\
157         c= AV_RN32(&src3[i*src_stride3+4]);\
158         d= AV_RN32(&src4[i*src_stride4+4]);\
159         l0=  (a&0x03030303UL)\
160            + (b&0x03030303UL)\
161            + 0x02020202UL;\
162         h0= ((a&0xFCFCFCFCUL)>>2)\
163           + ((b&0xFCFCFCFCUL)>>2);\
164         l1=  (c&0x03030303UL)\
165            + (d&0x03030303UL);\
166         h1= ((c&0xFCFCFCFCUL)>>2)\
167           + ((d&0xFCFCFCFCUL)>>2);\
168         OP(*((uint32_t*)&dst[i*dst_stride+4]), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
169     }\
170 }\
171 \
172 static inline void FUNC(OPNAME ## _no_rnd_pixels8_l4)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, const uint8_t *src3, const uint8_t *src4,\
173                  int dst_stride, int src_stride1, int src_stride2,int src_stride3,int src_stride4, int h){\
174     /* FIXME HIGH BIT DEPTH*/\
175     int i;\
176     for(i=0; i<h; i++){\
177         uint32_t a, b, c, d, l0, l1, h0, h1;\
178         a= AV_RN32(&src1[i*src_stride1]);\
179         b= AV_RN32(&src2[i*src_stride2]);\
180         c= AV_RN32(&src3[i*src_stride3]);\
181         d= AV_RN32(&src4[i*src_stride4]);\
182         l0=  (a&0x03030303UL)\
183            + (b&0x03030303UL)\
184            + 0x01010101UL;\
185         h0= ((a&0xFCFCFCFCUL)>>2)\
186           + ((b&0xFCFCFCFCUL)>>2);\
187         l1=  (c&0x03030303UL)\
188            + (d&0x03030303UL);\
189         h1= ((c&0xFCFCFCFCUL)>>2)\
190           + ((d&0xFCFCFCFCUL)>>2);\
191         OP(*((uint32_t*)&dst[i*dst_stride]), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
192         a= AV_RN32(&src1[i*src_stride1+4]);\
193         b= AV_RN32(&src2[i*src_stride2+4]);\
194         c= AV_RN32(&src3[i*src_stride3+4]);\
195         d= AV_RN32(&src4[i*src_stride4+4]);\
196         l0=  (a&0x03030303UL)\
197            + (b&0x03030303UL)\
198            + 0x01010101UL;\
199         h0= ((a&0xFCFCFCFCUL)>>2)\
200           + ((b&0xFCFCFCFCUL)>>2);\
201         l1=  (c&0x03030303UL)\
202            + (d&0x03030303UL);\
203         h1= ((c&0xFCFCFCFCUL)>>2)\
204           + ((d&0xFCFCFCFCUL)>>2);\
205         OP(*((uint32_t*)&dst[i*dst_stride+4]), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
206     }\
207 }\
208 static inline void FUNC(OPNAME ## _pixels16_l4)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, const uint8_t *src3, const uint8_t *src4,\
209                  int dst_stride, int src_stride1, int src_stride2,int src_stride3,int src_stride4, int h){\
210     FUNC(OPNAME ## _pixels8_l4)(dst  , src1  , src2  , src3  , src4  , dst_stride, src_stride1, src_stride2, src_stride3, src_stride4, h);\
211     FUNC(OPNAME ## _pixels8_l4)(dst+8*sizeof(pixel), src1+8*sizeof(pixel), src2+8*sizeof(pixel), src3+8*sizeof(pixel), src4+8*sizeof(pixel), dst_stride, src_stride1, src_stride2, src_stride3, src_stride4, h);\
212 }\
213 static inline void FUNC(OPNAME ## _no_rnd_pixels16_l4)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, const uint8_t *src3, const uint8_t *src4,\
214                  int dst_stride, int src_stride1, int src_stride2,int src_stride3,int src_stride4, int h){\
215     FUNC(OPNAME ## _no_rnd_pixels8_l4)(dst  , src1  , src2  , src3  , src4  , dst_stride, src_stride1, src_stride2, src_stride3, src_stride4, h);\
216     FUNC(OPNAME ## _no_rnd_pixels8_l4)(dst+8*sizeof(pixel), src1+8*sizeof(pixel), src2+8*sizeof(pixel), src3+8*sizeof(pixel), src4+8*sizeof(pixel), dst_stride, src_stride1, src_stride2, src_stride3, src_stride4, h);\
217 }\
218 \
219 static inline void FUNCC(OPNAME ## _pixels8_xy2)(uint8_t *block, const uint8_t *pixels, ptrdiff_t line_size, int h)\
220 {\
221     /* FIXME HIGH BIT DEPTH */\
222     int j;\
223     for(j=0; j<2; j++){\
224         int i;\
225         const uint32_t a= AV_RN32(pixels  );\
226         const uint32_t b= AV_RN32(pixels+1);\
227         uint32_t l0=  (a&0x03030303UL)\
228                     + (b&0x03030303UL)\
229                     + 0x02020202UL;\
230         uint32_t h0= ((a&0xFCFCFCFCUL)>>2)\
231                    + ((b&0xFCFCFCFCUL)>>2);\
232         uint32_t l1,h1;\
233 \
234         pixels+=line_size;\
235         for(i=0; i<h; i+=2){\
236             uint32_t a= AV_RN32(pixels  );\
237             uint32_t b= AV_RN32(pixels+1);\
238             l1=  (a&0x03030303UL)\
239                + (b&0x03030303UL);\
240             h1= ((a&0xFCFCFCFCUL)>>2)\
241               + ((b&0xFCFCFCFCUL)>>2);\
242             OP(*((uint32_t*)block), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
243             pixels+=line_size;\
244             block +=line_size;\
245             a= AV_RN32(pixels  );\
246             b= AV_RN32(pixels+1);\
247             l0=  (a&0x03030303UL)\
248                + (b&0x03030303UL)\
249                + 0x02020202UL;\
250             h0= ((a&0xFCFCFCFCUL)>>2)\
251               + ((b&0xFCFCFCFCUL)>>2);\
252             OP(*((uint32_t*)block), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
253             pixels+=line_size;\
254             block +=line_size;\
255         }\
256         pixels+=4-line_size*(h+1);\
257         block +=4-line_size*h;\
258     }\
259 }\
260 \
261 CALL_2X_PIXELS(FUNCC(OPNAME ## _pixels16_xy2), FUNCC(OPNAME ## _pixels8_xy2), 8*sizeof(pixel))\
262
263 #define op_avg(a, b) a = rnd_avg_pixel4(a, b)
264 #define op_put(a, b) a = b
265 #if BIT_DEPTH == 8
266 #define put_no_rnd_pixels8_8_c put_pixels8_8_c
267 PIXOP2(avg, op_avg)
268 PIXOP2(put, op_put)
269 #endif
270 #undef op_avg
271 #undef op_put