1 /*
2 * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (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 GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include <stdint.h>
22 #include <stdio.h>
23
24 #include "libavutil/internal.h"
25 #include "libavutil/mem.h"
26
27 #include "libavcodec/get_bits.h"
28 #include "libavcodec/golomb.h"
29 #include "libavcodec/put_bits.h"
30
31 #define COUNT 8191
32 #define SIZE (COUNT * 4)
33
main(void)34 int main(void)
35 {
36 int i, ret = 0;
37 uint8_t *temp;
38 PutBitContext pb;
39 GetBitContext gb;
40
41 temp = av_malloc(SIZE);
42 if (!temp)
43 return 2;
44
45 init_put_bits(&pb, temp, SIZE);
46 for (i = 0; i < COUNT; i++)
47 set_ue_golomb(&pb, i);
48 flush_put_bits(&pb);
49
50 init_get_bits(&gb, temp, 8 * SIZE);
51 for (i = 0; i < COUNT; i++) {
52 int j, s = show_bits(&gb, 25);
53
54 j = get_ue_golomb(&gb);
55 if (j != i) {
56 fprintf(stderr, "get_ue_golomb: expected %d, got %d. bits: %7x\n",
57 i, j, s);
58 ret = 1;
59 }
60 }
61
62 #define EXTEND(i) ((i) << 3 | (i) & 7)
63 init_put_bits(&pb, temp, SIZE);
64 for (i = 0; i < COUNT; i++)
65 set_ue_golomb(&pb, EXTEND(i));
66 flush_put_bits(&pb);
67
68 init_get_bits(&gb, temp, 8 * SIZE);
69 for (i = 0; i < COUNT; i++) {
70 int j, s = show_bits_long(&gb, 32);
71
72 j = get_ue_golomb_long(&gb);
73 if (j != EXTEND(i)) {
74 fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
75 "bits: %8x\n", EXTEND(i), j, s);
76 ret = 1;
77 }
78 }
79
80 #define EXTEND_L(i) ((i) << 4 | (i) & 15)
81 init_put_bits(&pb, temp, SIZE);
82 for (i = 0; i < COUNT; i++)
83 set_ue_golomb_long(&pb, EXTEND_L(i));
84 flush_put_bits(&pb);
85
86 init_get_bits(&gb, temp, 8 * SIZE);
87 for (i = 0; i < COUNT; i++) {
88 int j, s = show_bits_long(&gb, 32);
89
90 j = get_ue_golomb_long(&gb);
91 if (j != EXTEND_L(i)) {
92 fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
93 "bits: %8x\n", EXTEND_L(i), j, s);
94 ret = 1;
95 }
96 }
97
98 init_put_bits(&pb, temp, SIZE);
99 for (i = 0; i < COUNT; i++)
100 set_se_golomb(&pb, i - COUNT / 2);
101 flush_put_bits(&pb);
102
103 init_get_bits(&gb, temp, 8 * SIZE);
104 for (i = 0; i < COUNT; i++) {
105 int j, s = show_bits(&gb, 25);
106
107 j = get_se_golomb(&gb);
108 if (j != i - COUNT / 2) {
109 fprintf(stderr, "get_se_golomb: expected %d, got %d. bits: %7x\n",
110 i - COUNT / 2, j, s);
111 ret = 1;
112 }
113 }
114
115 av_free(temp);
116
117 return ret;
118 }
119