vp9: allocate 'b', 'block/uvblock' and 'eob/uveob' dynamically.
[ffmpeg.git] / libavcodec / h264chroma_template.c
1 /*
2  * Copyright (c) 2000, 2001 Fabrice Bellard
3  * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
4  *
5  * This file is part of Libav.
6  *
7  * Libav is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * Libav is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include <assert.h>
23
24 #include "bit_depth_template.c"
25
26 #define H264_CHROMA_MC(OPNAME, OP)\
27 static void FUNCC(OPNAME ## h264_chroma_mc2)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\
28     pixel *dst = (pixel*)_dst;\
29     pixel *src = (pixel*)_src;\
30     const int A=(8-x)*(8-y);\
31     const int B=(  x)*(8-y);\
32     const int C=(8-x)*(  y);\
33     const int D=(  x)*(  y);\
34     int i;\
35     stride /= sizeof(pixel);\
36     \
37     assert(x<8 && y<8 && x>=0 && y>=0);\
38 \
39     if(D){\
40         for(i=0; i<h; i++){\
41             OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
42             OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
43             dst+= stride;\
44             src+= stride;\
45         }\
46     } else if (B + C) {\
47         const int E= B+C;\
48         const int step= C ? stride : 1;\
49         for(i=0; i<h; i++){\
50             OP(dst[0], (A*src[0] + E*src[step+0]));\
51             OP(dst[1], (A*src[1] + E*src[step+1]));\
52             dst+= stride;\
53             src+= stride;\
54         }\
55     } else {\
56         for ( i = 0; i < h; i++){\
57             OP(dst[0], A * src[0]);\
58             OP(dst[1], A * src[1]);\
59             dst += stride;\
60             src += stride;\
61         }\
62     }\
63 }\
64 \
65 static void FUNCC(OPNAME ## h264_chroma_mc4)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\
66     pixel *dst = (pixel*)_dst;\
67     pixel *src = (pixel*)_src;\
68     const int A=(8-x)*(8-y);\
69     const int B=(  x)*(8-y);\
70     const int C=(8-x)*(  y);\
71     const int D=(  x)*(  y);\
72     int i;\
73     stride /= sizeof(pixel);\
74     \
75     assert(x<8 && y<8 && x>=0 && y>=0);\
76 \
77     if(D){\
78         for(i=0; i<h; i++){\
79             OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
80             OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
81             OP(dst[2], (A*src[2] + B*src[3] + C*src[stride+2] + D*src[stride+3]));\
82             OP(dst[3], (A*src[3] + B*src[4] + C*src[stride+3] + D*src[stride+4]));\
83             dst+= stride;\
84             src+= stride;\
85         }\
86     } else if (B + C) {\
87         const int E= B+C;\
88         const int step= C ? stride : 1;\
89         for(i=0; i<h; i++){\
90             OP(dst[0], (A*src[0] + E*src[step+0]));\
91             OP(dst[1], (A*src[1] + E*src[step+1]));\
92             OP(dst[2], (A*src[2] + E*src[step+2]));\
93             OP(dst[3], (A*src[3] + E*src[step+3]));\
94             dst+= stride;\
95             src+= stride;\
96         }\
97     } else {\
98         for ( i = 0; i < h; i++){\
99             OP(dst[0], A * src[0]);\
100             OP(dst[1], A * src[1]);\
101             OP(dst[2], A * src[2]);\
102             OP(dst[3], A * src[3]);\
103             dst += stride;\
104             src += stride;\
105         }\
106     }\
107 }\
108 \
109 static void FUNCC(OPNAME ## h264_chroma_mc8)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\
110     pixel *dst = (pixel*)_dst;\
111     pixel *src = (pixel*)_src;\
112     const int A=(8-x)*(8-y);\
113     const int B=(  x)*(8-y);\
114     const int C=(8-x)*(  y);\
115     const int D=(  x)*(  y);\
116     int i;\
117     stride /= sizeof(pixel);\
118     \
119     assert(x<8 && y<8 && x>=0 && y>=0);\
120 \
121     if(D){\
122         for(i=0; i<h; i++){\
123             OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
124             OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
125             OP(dst[2], (A*src[2] + B*src[3] + C*src[stride+2] + D*src[stride+3]));\
126             OP(dst[3], (A*src[3] + B*src[4] + C*src[stride+3] + D*src[stride+4]));\
127             OP(dst[4], (A*src[4] + B*src[5] + C*src[stride+4] + D*src[stride+5]));\
128             OP(dst[5], (A*src[5] + B*src[6] + C*src[stride+5] + D*src[stride+6]));\
129             OP(dst[6], (A*src[6] + B*src[7] + C*src[stride+6] + D*src[stride+7]));\
130             OP(dst[7], (A*src[7] + B*src[8] + C*src[stride+7] + D*src[stride+8]));\
131             dst+= stride;\
132             src+= stride;\
133         }\
134     } else if (B + C) {\
135         const int E= B+C;\
136         const int step= C ? stride : 1;\
137         for(i=0; i<h; i++){\
138             OP(dst[0], (A*src[0] + E*src[step+0]));\
139             OP(dst[1], (A*src[1] + E*src[step+1]));\
140             OP(dst[2], (A*src[2] + E*src[step+2]));\
141             OP(dst[3], (A*src[3] + E*src[step+3]));\
142             OP(dst[4], (A*src[4] + E*src[step+4]));\
143             OP(dst[5], (A*src[5] + E*src[step+5]));\
144             OP(dst[6], (A*src[6] + E*src[step+6]));\
145             OP(dst[7], (A*src[7] + E*src[step+7]));\
146             dst+= stride;\
147             src+= stride;\
148         }\
149     } else {\
150         for ( i = 0; i < h; i++){\
151             OP(dst[0], A * src[0]);\
152             OP(dst[1], A * src[1]);\
153             OP(dst[2], A * src[2]);\
154             OP(dst[3], A * src[3]);\
155             OP(dst[4], A * src[4]);\
156             OP(dst[5], A * src[5]);\
157             OP(dst[6], A * src[6]);\
158             OP(dst[7], A * src[7]);\
159             dst += stride;\
160             src += stride;\
161         }\
162     }\
163 }
164
165 #define op_avg(a, b) a = (((a)+(((b) + 32)>>6)+1)>>1)
166 #define op_put(a, b) a = (((b) + 32)>>6)
167
168 H264_CHROMA_MC(put_       , op_put)
169 H264_CHROMA_MC(avg_       , op_avg)
170 #undef op_avg
171 #undef op_put