1 /*
2 * This file is part of FFmpeg.
3 *
4 * FFmpeg is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * FFmpeg is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with FFmpeg; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19 #include <stdint.h>
20 #include <string.h>
21
22 #include "libavutil/lfg.h"
23 #include "libavutil/log.h"
24
25 #include "libavcodec/rangecoder.h"
26
27 #define SIZE 1240
28
main(void)29 int main(void)
30 {
31 RangeCoder c;
32 uint8_t b[9 * SIZE] = {0};
33 uint8_t r[9 * SIZE];
34 int i, p, actual_length, version;
35 uint8_t state[10];
36 AVLFG prng;
37
38 av_lfg_init(&prng, 1);
39 for (version = 0; version < 2; version++) {
40 for (p = 0; p< 1024; p++) {
41 ff_init_range_encoder(&c, b, SIZE);
42 ff_build_rac_states(&c, (1LL << 32) / 20, 128 + 64 + 32 + 16);
43
44 memset(state, 128, sizeof(state));
45
46 for (i = 0; i < SIZE; i++)
47 r[i] = av_lfg_get(&prng) % 7;
48
49 for (i = 0; i < SIZE; i++)
50 put_rac(&c, state, r[i] & 1);
51
52 actual_length = ff_rac_terminate(&c, version);
53
54 ff_init_range_decoder(&c, b, version ? SIZE : actual_length);
55
56 memset(state, 128, sizeof(state));
57
58 for (i = 0; i < SIZE; i++)
59 if ((r[i] & 1) != get_rac(&c, state)) {
60 av_log(NULL, AV_LOG_ERROR, "rac failure at %d pass %d version %d\n", i, p, version);
61 return 1;
62 }
63
64 if (ff_rac_check_termination(&c, version) < 0) {
65 av_log(NULL, AV_LOG_ERROR, "rac failure at termination pass %d version %d\n", p, version);
66 return 1;
67 }
68 if (c.bytestream - c.bytestream_start - actual_length != version) {
69 av_log(NULL, AV_LOG_ERROR, "rac failure at pass %d version %d\n", p, version);
70 return 1;
71 }
72 }
73 }
74
75 return 0;
76 }
77