FFmpeg
 All Data Structures Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
h264chroma_template.c
Go to the documentation of this file.
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 FFmpeg.
6  *
7  * FFmpeg 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  * FFmpeg 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 FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include "libavutil/avassert.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)-1;\
36  \
37  av_assert2(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{\
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  }\
56 }\
57 \
58 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){\
59  pixel *dst = (pixel*)_dst;\
60  pixel *src = (pixel*)_src;\
61  const int A=(8-x)*(8-y);\
62  const int B=( x)*(8-y);\
63  const int C=(8-x)*( y);\
64  const int D=( x)*( y);\
65  int i;\
66  stride >>= sizeof(pixel)-1;\
67  \
68  av_assert2(x<8 && y<8 && x>=0 && y>=0);\
69 \
70  if(D){\
71  for(i=0; i<h; i++){\
72  OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
73  OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
74  OP(dst[2], (A*src[2] + B*src[3] + C*src[stride+2] + D*src[stride+3]));\
75  OP(dst[3], (A*src[3] + B*src[4] + C*src[stride+3] + D*src[stride+4]));\
76  dst+= stride;\
77  src+= stride;\
78  }\
79  }else{\
80  const int E= B+C;\
81  const int step= C ? stride : 1;\
82  for(i=0; i<h; i++){\
83  OP(dst[0], (A*src[0] + E*src[step+0]));\
84  OP(dst[1], (A*src[1] + E*src[step+1]));\
85  OP(dst[2], (A*src[2] + E*src[step+2]));\
86  OP(dst[3], (A*src[3] + E*src[step+3]));\
87  dst+= stride;\
88  src+= stride;\
89  }\
90  }\
91 }\
92 \
93 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){\
94  pixel *dst = (pixel*)_dst;\
95  pixel *src = (pixel*)_src;\
96  const int A=(8-x)*(8-y);\
97  const int B=( x)*(8-y);\
98  const int C=(8-x)*( y);\
99  const int D=( x)*( y);\
100  int i;\
101  stride >>= sizeof(pixel)-1;\
102  \
103  av_assert2(x<8 && y<8 && x>=0 && y>=0);\
104 \
105  if(D){\
106  for(i=0; i<h; i++){\
107  OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
108  OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
109  OP(dst[2], (A*src[2] + B*src[3] + C*src[stride+2] + D*src[stride+3]));\
110  OP(dst[3], (A*src[3] + B*src[4] + C*src[stride+3] + D*src[stride+4]));\
111  OP(dst[4], (A*src[4] + B*src[5] + C*src[stride+4] + D*src[stride+5]));\
112  OP(dst[5], (A*src[5] + B*src[6] + C*src[stride+5] + D*src[stride+6]));\
113  OP(dst[6], (A*src[6] + B*src[7] + C*src[stride+6] + D*src[stride+7]));\
114  OP(dst[7], (A*src[7] + B*src[8] + C*src[stride+7] + D*src[stride+8]));\
115  dst+= stride;\
116  src+= stride;\
117  }\
118  }else{\
119  const int E= B+C;\
120  const int step= C ? stride : 1;\
121  for(i=0; i<h; i++){\
122  OP(dst[0], (A*src[0] + E*src[step+0]));\
123  OP(dst[1], (A*src[1] + E*src[step+1]));\
124  OP(dst[2], (A*src[2] + E*src[step+2]));\
125  OP(dst[3], (A*src[3] + E*src[step+3]));\
126  OP(dst[4], (A*src[4] + E*src[step+4]));\
127  OP(dst[5], (A*src[5] + E*src[step+5]));\
128  OP(dst[6], (A*src[6] + E*src[step+6]));\
129  OP(dst[7], (A*src[7] + E*src[step+7]));\
130  dst+= stride;\
131  src+= stride;\
132  }\
133  }\
134 }
135 
136 #define op_avg(a, b) a = (((a)+(((b) + 32)>>6)+1)>>1)
137 #define op_put(a, b) a = (((b) + 32)>>6)
138 
139 H264_CHROMA_MC(put_ , op_put)
140 H264_CHROMA_MC(avg_ , op_avg)
141 #undef op_avg
142 #undef op_put