FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/tests/bitstream_template.c
Date: 2025-04-26 09:35:27
Exec Total Coverage
Lines: 89 96 92.7%
Functions: 1 1 100.0%
Branches: 40 52 76.9%

Line Branch Exec Source
1 /*
2 * cached bitstream reader test
3 * copyright (c) 2022 Anton Khirnov <anton@khirnov.net>
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 #define ASSERT_LEVEL 2
23
24 #include "libavutil/avassert.h"
25 #include "libavutil/lfg.h"
26 #include "libavutil/random_seed.h"
27
28 #include "libavcodec/bitstream.h"
29 #include "libavcodec/defs.h"
30
31 #ifdef BITSTREAM_LE
32 #define BITSTREAM_WRITER_LE
33 #endif
34 #include "libavcodec/put_bits.h"
35
36 #define SIZE 157
37
38 enum Op {
39 OP_READ,
40 OP_READ_NZ,
41 OP_READ_BIT,
42 OP_READ_63,
43 OP_READ_64,
44 OP_READ_SIGNED,
45 OP_READ_SIGNED_NZ,
46 OP_APPLY_SIGN,
47 OP_ALIGN,
48 OP_NB,
49 };
50
51 2 int main(int argc, char **argv)
52 {
53 BitstreamContext bc;
54 PutBitContext pb;
55 AVLFG lfg;
56
57 uint8_t buf[SIZE + AV_INPUT_BUFFER_PADDING_SIZE];
58 uint8_t dst[SIZE + AV_INPUT_BUFFER_PADDING_SIZE];
59
60 uint32_t random_seed;
61 uint64_t val, val1;
62 int32_t sval, sval1;
63 unsigned count;
64 int ret;
65
66 /* generate random input, using a given or random seed */
67
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 if (argc > 1)
68 random_seed = strtoul(argv[1], NULL, 0);
69 else
70 2 random_seed = av_get_random_seed();
71
72 2 fprintf(stderr, "Testing with LFG seed: %"PRIu32"\n", random_seed);
73 2 av_lfg_init(&lfg, random_seed);
74
75
2/2
✓ Branch 0 taken 314 times.
✓ Branch 1 taken 2 times.
316 for (unsigned i = 0; i < SIZE; i++)
76 314 buf[i] = av_lfg_get(&lfg);
77
78 2 ret = bits_init8 (&bc, buf, SIZE);
79
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 av_assert0(ret >= 0);
80 2 init_put_bits(&pb, dst, SIZE);
81
82 /* use a random sequence of bitreading operations to transfer data
83 * from BitstreamContext to PutBitContext */
84
2/2
✓ Branch 1 taken 194 times.
✓ Branch 2 taken 2 times.
196 while (bits_left(&bc) > 0) {
85 194 enum Op op = av_lfg_get(&lfg) % OP_NB;
86
87
9/10
✓ Branch 0 taken 22 times.
✓ Branch 1 taken 21 times.
✓ Branch 2 taken 25 times.
✓ Branch 3 taken 20 times.
✓ Branch 4 taken 15 times.
✓ Branch 5 taken 18 times.
✓ Branch 6 taken 22 times.
✓ Branch 7 taken 24 times.
✓ Branch 8 taken 27 times.
✗ Branch 9 not taken.
194 switch (op) {
88 22 case OP_READ:
89
2/2
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 20 times.
22 count = av_lfg_get(&lfg) % FFMIN(33, bits_left(&bc) + 1);
90 22 val1 = bits_peek(&bc, count);
91 22 val = bits_read(&bc, count);
92
93 22 fprintf(stderr, "%d read %u: %"PRIu64"\n", bits_tell(&bc) - count, count, val);
94
95
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 22 times.
22 av_assert0(val == val1);
96
97 22 put_bits64(&pb, count, val);
98 22 break;
99 21 case OP_READ_NZ:
100
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 21 times.
21 count = av_lfg_get(&lfg) % FFMIN(33, bits_left(&bc) + 1);
101 21 count = FFMAX(count, 1);
102 21 val1 = bits_peek_nz(&bc, count);
103 21 val = bits_read_nz(&bc, count);
104
105 21 fprintf(stderr, "%d read_nz %u: %"PRIu64"\n", bits_tell(&bc) - count, count, val);
106
107
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 21 times.
21 av_assert0(val == val1);
108
109 21 put_bits64(&pb, count, val);
110 21 break;
111 25 case OP_READ_BIT:
112 25 val = bits_read_bit(&bc);
113
114 25 fprintf(stderr, "%d read_bit: %"PRIu64"\n", bits_tell(&bc) - 1, val);
115
116 25 put_bits(&pb, 1, val);
117 25 break;
118 20 case OP_READ_63:
119
2/2
✓ Branch 2 taken 4 times.
✓ Branch 3 taken 16 times.
20 count = av_lfg_get(&lfg) % FFMIN(64, bits_left(&bc) + 1);
120 20 val = bits_read_63(&bc, count);
121
122 20 fprintf(stderr, "%d read_63 %u: %"PRIu64"\n", bits_tell(&bc) - count, count, val);
123
124 20 put_bits64(&pb, count, val);
125 20 break;
126 15 case OP_READ_64:
127
2/2
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 13 times.
15 count = av_lfg_get(&lfg) % FFMIN(65, bits_left(&bc) + 1);
128 15 val = bits_read_64(&bc, count);
129
130 15 fprintf(stderr, "%d read_64 %u: %"PRIu64"\n", bits_tell(&bc) - count, count, val);
131
132 15 put_bits64(&pb, count, val);
133 15 break;
134 18 case OP_READ_SIGNED:
135
2/2
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 16 times.
18 count = av_lfg_get(&lfg) % FFMIN(33, bits_left(&bc) + 1);
136 18 sval1 = bits_peek_signed(&bc, count);
137 18 sval = bits_read_signed(&bc, count);
138
139 18 fprintf(stderr, "%d read_signed %u: %"PRId32"\n", bits_tell(&bc) - count, count, sval);
140
141
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
18 av_assert0(sval == sval1);
142
143
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
18 if (count == 32) put_bits32(&pb, sval);
144 18 else put_sbits(&pb, count, sval);
145 18 break;
146 22 case OP_READ_SIGNED_NZ:
147
2/2
✓ Branch 2 taken 3 times.
✓ Branch 3 taken 19 times.
22 count = av_lfg_get(&lfg) % FFMIN(33, bits_left(&bc) + 1);
148 22 count = FFMAX(count, 1);
149 22 sval1 = bits_peek_signed_nz(&bc, count);
150 22 sval = bits_read_signed_nz(&bc, count);
151
152 22 fprintf(stderr, "%d read_signed_nz %u: %"PRId32"\n", bits_tell(&bc) - count, count, sval);
153
154
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 22 times.
22 av_assert0(sval == sval1);
155
156
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 22 times.
22 if (count == 32) put_bits32(&pb, sval);
157 22 else put_sbits(&pb, count, sval);
158 22 break;
159 24 case OP_ALIGN:
160 24 count = (bits_tell(&bc) + 7) / 8 * 8 - bits_tell(&bc);
161
162 24 fprintf(stderr, "%d align %u\n", bits_tell(&bc), count);
163
164 24 put_bits(&pb, count, bits_peek(&bc, count));
165 24 bits_align(&bc);
166 24 break;
167 27 case OP_APPLY_SIGN:
168
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 27 times.
27 if (bits_left(&bc) < 2)
169 continue;
170
171
2/2
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 25 times.
27 count = av_lfg_get(&lfg) % FFMIN(32, bits_left(&bc));
172 27 count = FFMAX(count, 1);
173
174
2/2
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 25 times.
27 if (!bits_peek(&bc, count))
175 2 continue;
176
177 25 val = bits_read(&bc, count);
178 25 sval = bits_apply_sign(&bc, val);
179
180 25 fprintf(stderr, "%d apply_sign %u %"PRId32"\n",
181 25 bits_tell(&bc) - count - 1, count, sval);
182
183 25 put_bits64(&pb, count, FFABS(sval));
184 25 put_bits(&pb, 1, sval < 0);
185
186 25 break;
187 default:
188 av_assert0(0);
189 }
190 }
191
192 2 flush_put_bits(&pb);
193
194
2/2
✓ Branch 0 taken 314 times.
✓ Branch 1 taken 2 times.
316 for (unsigned i = 0; i < SIZE; i++)
195
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 314 times.
314 if (buf[i] != dst[i]) {
196 fprintf(stderr, "Mismatch at byte %u: %hhu %hhu; seed %"PRIu32"\n",
197 i, buf[i], dst[i], random_seed);
198 return 1;
199 }
200
201 2 return 0;
202 }
203