FFmpeg
h264pred.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2015 Henrik Gramner
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (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
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License along
17  * with FFmpeg; if not, write to the Free Software Foundation, Inc.,
18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19  */
20 
21 #include <string.h>
22 #include "checkasm.h"
23 #include "libavcodec/avcodec.h"
24 #include "libavcodec/h264pred.h"
25 #include "libavutil/common.h"
26 #include "libavutil/internal.h"
27 #include "libavutil/intreadwrite.h"
28 #include "libavutil/mem_internal.h"
29 
30 static const int codec_ids[4] = { AV_CODEC_ID_H264, AV_CODEC_ID_VP8, AV_CODEC_ID_RV40, AV_CODEC_ID_SVQ3 };
31 
32 static const char * const pred4x4_modes[4][15] = {
33  { /* H.264 */
34  [VERT_PRED ] = "vertical",
35  [HOR_PRED ] = "horizontal",
36  [DC_PRED ] = "dc",
37  [DIAG_DOWN_LEFT_PRED ] = "down_left",
38  [DIAG_DOWN_RIGHT_PRED] = "down_right",
39  [VERT_RIGHT_PRED ] = "vertical_right",
40  [HOR_DOWN_PRED ] = "horizontal_right",
41  [VERT_LEFT_PRED ] = "vertical_left",
42  [HOR_UP_PRED ] = "horizontal_up",
43  [LEFT_DC_PRED ] = "left_dc",
44  [TOP_DC_PRED ] = "top_dc",
45  [DC_128_PRED ] = "dc_128",
46  },
47  { /* VP8 */
48  [VERT_PRED ] = "vertical_vp8",
49  [HOR_PRED ] = "horizontal_vp8",
50  [VERT_LEFT_PRED] = "vertical_left_vp8",
51  [TM_VP8_PRED ] = "tm_vp8",
52  [DC_127_PRED ] = "dc_127_vp8",
53  [DC_129_PRED ] = "dc_129_vp8",
54  },
55  { /* RV40 */
56  [DIAG_DOWN_LEFT_PRED ] = "down_left_rv40",
57  [VERT_LEFT_PRED ] = "vertical_left_rv40",
58  [HOR_UP_PRED ] = "horizontal_up_rv40",
59  [DIAG_DOWN_LEFT_PRED_RV40_NODOWN] = "down_left_nodown_rv40",
60  [HOR_UP_PRED_RV40_NODOWN ] = "horizontal_up_nodown_rv40",
61  [VERT_LEFT_PRED_RV40_NODOWN ] = "vertical_left_nodown_rv40",
62  },
63  { /* SVQ3 */
64  [DIAG_DOWN_LEFT_PRED] = "down_left_svq3",
65  },
66 };
67 
68 static const char * const pred8x8_modes[4][11] = {
69  { /* H.264 */
70  [DC_PRED8x8 ] = "dc",
71  [HOR_PRED8x8 ] = "horizontal",
72  [VERT_PRED8x8 ] = "vertical",
73  [PLANE_PRED8x8 ] = "plane",
74  [LEFT_DC_PRED8x8 ] = "left_dc",
75  [TOP_DC_PRED8x8 ] = "top_dc",
76  [DC_128_PRED8x8 ] = "dc_128",
77  [ALZHEIMER_DC_L0T_PRED8x8] = "mad_cow_dc_l0t",
78  [ALZHEIMER_DC_0LT_PRED8x8] = "mad_cow_dc_0lt",
79  [ALZHEIMER_DC_L00_PRED8x8] = "mad_cow_dc_l00",
80  [ALZHEIMER_DC_0L0_PRED8x8] = "mad_cow_dc_0l0",
81  },
82  { /* VP8 */
83  [PLANE_PRED8x8 ] = "tm_vp8",
84  [DC_127_PRED8x8] = "dc_127_vp8",
85  [DC_129_PRED8x8] = "dc_129_vp8",
86  },
87  { /* RV40 */
88  [DC_PRED8x8 ] = "dc_rv40",
89  [LEFT_DC_PRED8x8] = "left_dc_rv40",
90  [TOP_DC_PRED8x8 ] = "top_dc_rv40",
91  },
92  /* nothing for SVQ3 */
93 };
94 
95 static const char * const pred16x16_modes[4][9] = {
96  { /* H.264 */
97  [DC_PRED8x8 ] = "dc",
98  [HOR_PRED8x8 ] = "horizontal",
99  [VERT_PRED8x8 ] = "vertical",
100  [PLANE_PRED8x8 ] = "plane",
101  [LEFT_DC_PRED8x8] = "left_dc",
102  [TOP_DC_PRED8x8 ] = "top_dc",
103  [DC_128_PRED8x8 ] = "dc_128",
104  },
105  { /* VP8 */
106  [PLANE_PRED8x8 ] = "tm_vp8",
107  [DC_127_PRED8x8] = "dc_127_vp8",
108  [DC_129_PRED8x8] = "dc_129_vp8",
109  },
110  { /* RV40 */
111  [PLANE_PRED8x8] = "plane_rv40",
112  },
113  { /* SVQ3 */
114  [PLANE_PRED8x8] = "plane_svq3",
115  },
116 };
117 
118 static const uint32_t pixel_mask[3] = { 0xffffffff, 0x01ff01ff, 0x03ff03ff };
119 
120 #define SIZEOF_PIXEL ((bit_depth + 7) / 8)
121 #define BUF_SIZE (3 * 16 * 17)
122 
123 #define check_pred_func(func, name, mode_name) \
124  (mode_name && ((codec_ids[codec] == AV_CODEC_ID_H264) ? \
125  check_func(func, "pred%s_%s_%d", name, mode_name, bit_depth) : \
126  check_func(func, "pred%s_%s", name, mode_name)))
127 
128 #define randomize_buffers() \
129  do { \
130  uint32_t mask = pixel_mask[bit_depth - 8]; \
131  int i; \
132  for (i = 0; i < BUF_SIZE; i += 4) { \
133  uint32_t r = rnd() & mask; \
134  AV_WN32A(buf0 + i, r); \
135  AV_WN32A(buf1 + i, r); \
136  } \
137  } while (0)
138 
139 #define src0 (buf0 + 4 * 16) /* Offset to allow room for top and left */
140 #define src1 (buf1 + 4 * 16)
141 
142 static void check_pred4x4(H264PredContext *h, uint8_t *buf0, uint8_t *buf1,
143  int codec, int chroma_format, int bit_depth)
144 {
145  if (chroma_format == 1) {
146  uint8_t *topright = buf0 + 2*16;
147  int pred_mode;
148  declare_func_emms(AV_CPU_FLAG_MMX | AV_CPU_FLAG_MMXEXT, void, uint8_t *src, const uint8_t *topright, ptrdiff_t stride);
149 
150  for (pred_mode = 0; pred_mode < 15; pred_mode++) {
151  if (check_pred_func(h->pred4x4[pred_mode], "4x4", pred4x4_modes[codec][pred_mode])) {
152  randomize_buffers();
153  call_ref(src0, topright, 12*SIZEOF_PIXEL);
154  call_new(src1, topright, 12*SIZEOF_PIXEL);
155  if (memcmp(buf0, buf1, BUF_SIZE))
156  fail();
157  bench_new(src1, topright, 12*SIZEOF_PIXEL);
158  }
159  }
160  }
161 }
162 
163 static void check_pred8x8(H264PredContext *h, uint8_t *buf0, uint8_t *buf1,
164  int codec, int chroma_format, int bit_depth)
165 {
166  int pred_mode;
167  declare_func_emms(AV_CPU_FLAG_MMX | AV_CPU_FLAG_MMXEXT, void, uint8_t *src, ptrdiff_t stride);
168 
169  for (pred_mode = 0; pred_mode < 11; pred_mode++) {
170  if (check_pred_func(h->pred8x8[pred_mode], (chroma_format == 2) ? "8x16" : "8x8",
171  pred8x8_modes[codec][pred_mode])) {
172  randomize_buffers();
173  call_ref(src0, 24*SIZEOF_PIXEL);
174  call_new(src1, 24*SIZEOF_PIXEL);
175  if (memcmp(buf0, buf1, BUF_SIZE))
176  fail();
177  bench_new(src1, 24*SIZEOF_PIXEL);
178  }
179  }
180 }
181 
182 static void check_pred16x16(H264PredContext *h, uint8_t *buf0, uint8_t *buf1,
183  int codec, int chroma_format, int bit_depth)
184 {
185  if (chroma_format == 1) {
186  int pred_mode;
187  declare_func_emms(AV_CPU_FLAG_MMX | AV_CPU_FLAG_MMXEXT, void, uint8_t *src, ptrdiff_t stride);
188 
189  for (pred_mode = 0; pred_mode < 9; pred_mode++) {
190  if (check_pred_func(h->pred16x16[pred_mode], "16x16", pred16x16_modes[codec][pred_mode])) {
191  randomize_buffers();
192  call_ref(src0, 48);
193  call_new(src1, 48);
194  if (memcmp(buf0, buf1, BUF_SIZE))
195  fail();
196  bench_new(src1, 48);
197  }
198  }
199  }
200 }
201 
202 static void check_pred8x8l(H264PredContext *h, uint8_t *buf0, uint8_t *buf1,
203  int codec, int chroma_format, int bit_depth)
204 {
205  if (chroma_format == 1 && codec_ids[codec] == AV_CODEC_ID_H264) {
206  int pred_mode;
207  declare_func_emms(AV_CPU_FLAG_MMXEXT, void, uint8_t *src, int topleft, int topright, ptrdiff_t stride);
208 
209  for (pred_mode = 0; pred_mode < 12; pred_mode++) {
210  if (check_pred_func(h->pred8x8l[pred_mode], "8x8l", pred4x4_modes[codec][pred_mode])) {
211  int neighbors;
212  for (neighbors = 0; neighbors <= 0xc000; neighbors += 0x4000) {
213  int has_topleft = neighbors & 0x8000;
214  int has_topright = neighbors & 0x4000;
215 
216  if ((pred_mode == DIAG_DOWN_RIGHT_PRED || pred_mode == VERT_RIGHT_PRED) && !has_topleft)
217  continue; /* Those aren't allowed according to the spec */
218 
219  randomize_buffers();
220  call_ref(src0, has_topleft, has_topright, 24*SIZEOF_PIXEL);
221  call_new(src1, has_topleft, has_topright, 24*SIZEOF_PIXEL);
222  if (memcmp(buf0, buf1, BUF_SIZE))
223  fail();
224  bench_new(src1, has_topleft, has_topright, 24*SIZEOF_PIXEL);
225  }
226  }
227  }
228  }
229 }
230 
231 /* TODO: Add tests for H.264 lossless H/V prediction */
232 
233 void checkasm_check_h264pred(void)
234 {
235  static const struct {
236  void (*func)(H264PredContext*, uint8_t*, uint8_t*, int, int, int);
237  const char *name;
238  } tests[] = {
239  { check_pred4x4, "pred4x4" },
240  { check_pred8x8, "pred8x8" },
241  { check_pred16x16, "pred16x16" },
242  { check_pred8x8l, "pred8x8l" },
243  };
244 
245  LOCAL_ALIGNED_16(uint8_t, buf0, [BUF_SIZE]);
246  LOCAL_ALIGNED_16(uint8_t, buf1, [BUF_SIZE]);
247  H264PredContext h;
248  int test, codec, chroma_format, bit_depth;
249 
250  for (test = 0; test < FF_ARRAY_ELEMS(tests); test++) {
251  for (codec = 0; codec < 4; codec++) {
252  int codec_id = codec_ids[codec];
253  for (bit_depth = 8; bit_depth <= (codec_id == AV_CODEC_ID_H264 ? 10 : 8); bit_depth++)
254  for (chroma_format = 1; chroma_format <= (codec_id == AV_CODEC_ID_H264 ? 2 : 1); chroma_format++) {
255  ff_h264_pred_init(&h, codec_id, bit_depth, chroma_format);
256  tests[test].func(&h, buf0, buf1, codec, chroma_format, bit_depth);
257  }
258  }
259  report("%s", tests[test].name);
260  }
261 }
func
int(* func)(AVBPrint *dst, const char *in, const char *arg)
Definition: jacosubdec.c:68
declare_func_emms
#define declare_func_emms(cpu_flags, ret,...)
Definition: checkasm.h:128
HOR_PRED8x8
#define HOR_PRED8x8
Definition: h264pred.h:69
bit_depth
static void bit_depth(AudioStatsContext *s, uint64_t mask, uint64_t imask, AVRational *depth)
Definition: af_astats.c:226
name
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf default minimum maximum flags name is the option name
Definition: writing_filters.txt:88
DC_PRED8x8
#define DC_PRED8x8
Definition: h264pred.h:68
mem_internal.h
DC_128_PRED
@ DC_128_PRED
Definition: vp9.h:58
pred16x16_modes
static const char *const pred16x16_modes[4][9]
Definition: h264pred.c:95
TM_VP8_PRED
@ TM_VP8_PRED
Definition: vp9.h:55
DC_PRED
@ DC_PRED
Definition: vp9.h:48
test
Definition: idctdsp.c:34
VERT_LEFT_PRED
@ VERT_LEFT_PRED
Definition: vp9.h:53
SIZEOF_PIXEL
#define SIZEOF_PIXEL
Definition: h264pred.c:120
call_ref
#define call_ref(...)
Definition: checkasm.h:137
DC_127_PRED
@ DC_127_PRED
Definition: vp9.h:59
fail
#define fail()
Definition: checkasm.h:131
VERT_PRED
@ VERT_PRED
Definition: vp9.h:46
checkasm.h
DIAG_DOWN_RIGHT_PRED
@ DIAG_DOWN_RIGHT_PRED
Definition: vp9.h:50
pred4x4_modes
static const char *const pred4x4_modes[4][15]
Definition: h264pred.c:32
FF_ARRAY_ELEMS
#define FF_ARRAY_ELEMS(a)
Definition: sinewin_tablegen.c:29
HOR_PRED
@ HOR_PRED
Definition: vp9.h:47
pixel_mask
static const uint32_t pixel_mask[3]
Definition: h264pred.c:118
intreadwrite.h
pred8x8_modes
static const char *const pred8x8_modes[4][11]
Definition: h264pred.c:68
DC_129_PRED8x8
#define DC_129_PRED8x8
Definition: h264pred.h:86
VERT_LEFT_PRED_RV40_NODOWN
#define VERT_LEFT_PRED_RV40_NODOWN
Definition: h264pred.h:56
LOCAL_ALIGNED_16
#define LOCAL_ALIGNED_16(t, v,...)
Definition: mem_internal.h:130
LEFT_DC_PRED
@ LEFT_DC_PRED
Definition: vp9.h:56
check_pred8x8
static void check_pred8x8(H264PredContext *h, uint8_t *buf0, uint8_t *buf1, int codec, int chroma_format, int bit_depth)
Definition: h264pred.c:163
codec_id
enum AVCodecID codec_id
Definition: vaapi_decode.c:371
AV_CODEC_ID_SVQ3
@ AV_CODEC_ID_SVQ3
Definition: codec_id.h:73
check_pred4x4
static void check_pred4x4(H264PredContext *h, uint8_t *buf0, uint8_t *buf1, int codec, int chroma_format, int bit_depth)
Definition: h264pred.c:142
AV_CODEC_ID_H264
@ AV_CODEC_ID_H264
Definition: codec_id.h:77
TOP_DC_PRED8x8
#define TOP_DC_PRED8x8
Definition: h264pred.h:75
codec_ids
static const int codec_ids[4]
Definition: h264pred.c:30
BUF_SIZE
#define BUF_SIZE
Definition: h264pred.c:121
call_new
#define call_new(...)
Definition: checkasm.h:209
check_pred_func
#define check_pred_func(func, name, mode_name)
Definition: h264pred.c:123
checkasm_check_h264pred
void checkasm_check_h264pred(void)
Definition: h264pred.c:233
VERT_PRED8x8
#define VERT_PRED8x8
Definition: h264pred.h:70
DC_127_PRED8x8
#define DC_127_PRED8x8
Definition: h264pred.h:85
VERT_RIGHT_PRED
@ VERT_RIGHT_PRED
Definition: vp9.h:51
DC_128_PRED8x8
#define DC_128_PRED8x8
Definition: h264pred.h:76
PLANE_PRED8x8
#define PLANE_PRED8x8
Definition: h264pred.h:71
src0
#define src0
Definition: h264pred.c:139
DIAG_DOWN_LEFT_PRED_RV40_NODOWN
#define DIAG_DOWN_LEFT_PRED_RV40_NODOWN
Definition: h264pred.h:54
src1
#define src1
Definition: h264pred.c:140
tests
const TestCase tests[]
Definition: fifo_muxer.c:242
check_pred16x16
static void check_pred16x16(H264PredContext *h, uint8_t *buf0, uint8_t *buf1, int codec, int chroma_format, int bit_depth)
Definition: h264pred.c:182
report
#define report
Definition: checkasm.h:134
AV_CODEC_ID_RV40
@ AV_CODEC_ID_RV40
Definition: codec_id.h:119
check_pred8x8l
static void check_pred8x8l(H264PredContext *h, uint8_t *buf0, uint8_t *buf1, int codec, int chroma_format, int bit_depth)
Definition: h264pred.c:202
bench_new
#define bench_new(...)
Definition: checkasm.h:272
DC_129_PRED
@ DC_129_PRED
Definition: vp9.h:60
internal.h
HOR_UP_PRED_RV40_NODOWN
#define HOR_UP_PRED_RV40_NODOWN
Definition: h264pred.h:55
common.h
ALZHEIMER_DC_L00_PRED8x8
#define ALZHEIMER_DC_L00_PRED8x8
Definition: h264pred.h:81
LEFT_DC_PRED8x8
#define LEFT_DC_PRED8x8
Definition: h264pred.h:74
avcodec.h
stride
#define stride
Definition: h264pred_template.c:537
randomize_buffers
#define randomize_buffers()
Definition: h264pred.c:128
h264pred.h
AV_CPU_FLAG_MMX
#define AV_CPU_FLAG_MMX
standard MMX
Definition: cpu.h:29
test
static int test(void)
Definition: dnn-layer-dense.c:28
ff_h264_pred_init
av_cold void ff_h264_pred_init(H264PredContext *h, int codec_id, const int bit_depth, int chroma_format_idc)
Set the intra prediction function pointers.
Definition: h264pred.c:437
HOR_UP_PRED
@ HOR_UP_PRED
Definition: vp9.h:54
HOR_DOWN_PRED
@ HOR_DOWN_PRED
Definition: vp9.h:52
AV_CPU_FLAG_MMXEXT
#define AV_CPU_FLAG_MMXEXT
SSE integer functions or AMD MMX ext.
Definition: cpu.h:30
ALZHEIMER_DC_0L0_PRED8x8
#define ALZHEIMER_DC_0L0_PRED8x8
Definition: h264pred.h:82
H264PredContext
Context for storing H.264 prediction functions.
Definition: h264pred.h:94
ALZHEIMER_DC_L0T_PRED8x8
#define ALZHEIMER_DC_L0T_PRED8x8
Definition: h264pred.h:79
TOP_DC_PRED
@ TOP_DC_PRED
Definition: vp9.h:57
src
INIT_CLIP pixel * src
Definition: h264pred_template.c:418
DIAG_DOWN_LEFT_PRED
@ DIAG_DOWN_LEFT_PRED
Definition: vp9.h:49
AV_CODEC_ID_VP8
@ AV_CODEC_ID_VP8
Definition: codec_id.h:190
ALZHEIMER_DC_0LT_PRED8x8
#define ALZHEIMER_DC_0LT_PRED8x8
Definition: h264pred.h:80
h
h
Definition: vp9dsp_template.c:2038
int
int
Definition: ffmpeg_filter.c:153
Generated on Wed Aug 24 2022 21:41:27 for FFmpeg by   doxygen 1.8.17