FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/tests/checkasm/hevc_sao.c
Date: 2024-11-20 23:03:26
Exec Total Coverage
Lines: 57 59 96.6%
Functions: 3 3 100.0%
Branches: 44 60 73.3%
Line Branch Exec Source
1 /*
2 * Copyright (c) 2018 Yingming Fan <yingmingfan@gmail.com>
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
23 #include "libavutil/intreadwrite.h"
24 #include "libavutil/mem_internal.h"
25
26 #include "libavcodec/hevc/dsp.h"
27
28 #include "checkasm.h"
29
30 static const uint32_t pixel_mask[3] = { 0xffffffff, 0x03ff03ff, 0x0fff0fff };
31 static const uint32_t sao_size[5] = {8, 16, 32, 48, 64};
32
33 #define SIZEOF_PIXEL ((bit_depth + 7) / 8)
34 #define PIXEL_STRIDE (2*MAX_PB_SIZE + AV_INPUT_BUFFER_PADDING_SIZE) //same with sao_edge src_stride
35 #define BUF_SIZE (PIXEL_STRIDE * (64+2) * 2) //+2 for top and bottom row, *2 for high bit depth
36 #define OFFSET_THRESH (1 << (bit_depth - 5))
37 #define OFFSET_LENGTH 5
38
39 #define randomize_buffers(buf0, buf1, size) \
40 do { \
41 uint32_t mask = pixel_mask[(bit_depth - 8) >> 1]; \
42 int k; \
43 for (k = 0; k < size; k += 4) { \
44 uint32_t r = rnd() & mask; \
45 AV_WN32A(buf0 + k, r); \
46 AV_WN32A(buf1 + k, r); \
47 } \
48 } while (0)
49
50 #define randomize_buffers2(buf, size) \
51 do { \
52 uint32_t max_offset = OFFSET_THRESH; \
53 int k; \
54 if (bit_depth == 8) { \
55 for (k = 0; k < size; k++) { \
56 uint8_t r = rnd() % max_offset; \
57 buf[k] = r; \
58 } \
59 } else { \
60 for (k = 0; k < size; k++) { \
61 uint16_t r = rnd() % max_offset; \
62 buf[k] = r; \
63 } \
64 } \
65 } while (0)
66
67 39 static void check_sao_band(HEVCDSPContext *h, int bit_depth)
68 {
69 int i;
70 39 LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
71 39 LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
72 39 LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
73 39 LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
74 int16_t offset_val[OFFSET_LENGTH];
75 39 int left_class = rnd()%32;
76
77
2/2
✓ Branch 0 taken 195 times.
✓ Branch 1 taken 39 times.
 234 for (i = 0; i <= 4; i++) {
78 195 int block_size = sao_size[i];
79
2/2
✓ Branch 0 taken 156 times.
✓ Branch 1 taken 39 times.
 195 int prev_size = i > 0 ? sao_size[i - 1] : 0;
80 195 ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
81 195 declare_func(void, uint8_t *dst, const uint8_t *src, ptrdiff_t dst_stride, ptrdiff_t src_stride,
82 const int16_t *sao_offset_val, int sao_left_class, int width, int height);
83
84
2/2
✓ Branch 3 taken 60 times.
✓ Branch 4 taken 135 times.
 195 if (check_func(h->sao_band_filter[i], "hevc_sao_band_%d_%d", block_size, bit_depth)) {
85
86
2/2
✓ Branch 0 taken 192 times.
✓ Branch 1 taken 60 times.
 252 for (int w = prev_size + 4; w <= block_size; w += 4) {
87
2/2
✓ Branch 1 taken 1216512 times.
✓ Branch 2 taken 192 times.
 1216704 randomize_buffers(src0, src1, BUF_SIZE);
88
6/6
✓ Branch 0 taken 64 times.
✓ Branch 1 taken 128 times.
✓ Branch 3 taken 320 times.
✓ Branch 4 taken 64 times.
✓ Branch 6 taken 640 times.
✓ Branch 7 taken 128 times.
 1152 randomize_buffers2(offset_val, OFFSET_LENGTH);
89 192 memset(dst0, 0, BUF_SIZE);
90 192 memset(dst1, 0, BUF_SIZE);
91
92 192 call_ref(dst0, src0, stride, stride, offset_val, left_class, w, block_size);
93 192 call_new(dst1, src1, stride, stride, offset_val, left_class, w, block_size);
94
2/2
✓ Branch 0 taken 7488 times.
✓ Branch 1 taken 192 times.
 7680 for (int j = 0; j < block_size; j++) {
95
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7488 times.
 7488 if (memcmp(dst0 + j*stride, dst1 + j*stride, w*SIZEOF_PIXEL))
96 ✗ fail();
97 }
98 }
99
1/8
✗ Branch 1 not taken.
✓ Branch 2 taken 60 times.
✗ Branch 11 not taken.
✗ Branch 12 not taken.
✗ Branch 13 not taken.
✗ Branch 14 not taken.
✗ Branch 15 not taken.
✗ Branch 16 not taken.
 60 bench_new(dst1, src1, stride, stride, offset_val, left_class, block_size, block_size);
100 }
101 }
102 39 }
103
104 39 static void check_sao_edge(HEVCDSPContext *h, int bit_depth)
105 {
106 int i;
107 39 LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
108 39 LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
109 39 LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
110 39 LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
111 int16_t offset_val[OFFSET_LENGTH];
112 39 int eo = rnd()%4;
113
114
2/2
✓ Branch 0 taken 195 times.
✓ Branch 1 taken 39 times.
 234 for (i = 0; i <= 4; i++) {
115 195 int block_size = sao_size[i];
116
2/2
✓ Branch 0 taken 156 times.
✓ Branch 1 taken 39 times.
 195 int prev_size = i > 0 ? sao_size[i - 1] : 0;
117 195 ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
118 195 int offset = (AV_INPUT_BUFFER_PADDING_SIZE + PIXEL_STRIDE)*SIZEOF_PIXEL;
119 195 declare_func(void, uint8_t *dst, const uint8_t *src, ptrdiff_t stride_dst,
120 const int16_t *sao_offset_val, int eo, int width, int height);
121
122
2/2
✓ Branch 0 taken 624 times.
✓ Branch 1 taken 195 times.
 819 for (int w = prev_size + 4; w <= block_size; w += 4) {
123
2/2
✓ Branch 1 taken 3953664 times.
✓ Branch 2 taken 624 times.
 3954288 randomize_buffers(src0, src1, BUF_SIZE);
124
6/6
✓ Branch 0 taken 208 times.
✓ Branch 1 taken 416 times.
✓ Branch 3 taken 1040 times.
✓ Branch 4 taken 208 times.
✓ Branch 6 taken 2080 times.
✓ Branch 7 taken 416 times.
 3744 randomize_buffers2(offset_val, OFFSET_LENGTH);
125 624 memset(dst0, 0, BUF_SIZE);
126 624 memset(dst1, 0, BUF_SIZE);
127
128
2/2
✓ Branch 3 taken 43 times.
✓ Branch 4 taken 581 times.
 624 if (check_func(h->sao_edge_filter[i], "hevc_sao_edge_%d_%d", block_size, bit_depth)) {
129 43 call_ref(dst0, src0 + offset, stride, offset_val, eo, w, block_size);
130 43 call_new(dst1, src1 + offset, stride, offset_val, eo, w, block_size);
131
2/2
✓ Branch 0 taken 1488 times.
✓ Branch 1 taken 43 times.
 1531 for (int j = 0; j < block_size; j++) {
132
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1488 times.
 1488 if (memcmp(dst0 + j*stride, dst1 + j*stride, w*SIZEOF_PIXEL))
133 ✗ fail();
134 }
135
1/8
✗ Branch 1 not taken.
✓ Branch 2 taken 43 times.
✗ Branch 11 not taken.
✗ Branch 12 not taken.
✗ Branch 13 not taken.
✗ Branch 14 not taken.
✗ Branch 15 not taken.
✗ Branch 16 not taken.
 43 bench_new(dst1, src1 + offset, stride, offset_val, eo, block_size, block_size);
136 }
137 }
138 }
139 39 }
140
141 13 void checkasm_check_hevc_sao(void)
142 {
143 int bit_depth;
144
145
2/2
✓ Branch 0 taken 39 times.
✓ Branch 1 taken 13 times.
 52 for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
146 HEVCDSPContext h;
147
148 39 ff_hevc_dsp_init(&h, bit_depth);
149 39 check_sao_band(&h, bit_depth);
150 }
151 13 report("sao_band");
152
153
2/2
✓ Branch 0 taken 39 times.
✓ Branch 1 taken 13 times.
 52 for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
154 HEVCDSPContext h;
155
156 39 ff_hevc_dsp_init(&h, bit_depth);
157 39 check_sao_edge(&h, bit_depth);
158 }
159 13 report("sao_edge");
160 13 }
161