Split global .gitignore file into per-directory files
[libav.git] / libavcodec / tests / golomb.c
1 /*
2 * This file is part of Libav.
3 *
4 * Libav 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 * Libav 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 Libav; 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 <stdio.h>
21
22 #include "libavutil/mem.h"
23
24 #include "libavcodec/get_bits.h"
25 #include "libavcodec/put_bits.h"
26 #include "libavcodec/golomb.h"
27
28 #define COUNT 8191
29 #define SIZE (COUNT * 4)
30
31 int main(void)
32 {
33 int i, ret = 0;
34 uint8_t *temp;
35 PutBitContext pb;
36 GetBitContext gb;
37
38 temp = av_malloc(SIZE);
39 if (!temp)
40 return 2;
41
42 init_put_bits(&pb, temp, SIZE);
43 for (i = 0; i < COUNT; i++)
44 set_ue_golomb(&pb, i);
45 flush_put_bits(&pb);
46
47 init_get_bits(&gb, temp, 8 * SIZE);
48 for (i = 0; i < COUNT; i++) {
49 int j, s = show_bits(&gb, 25);
50
51 j = get_ue_golomb(&gb);
52 if (j != i) {
53 fprintf(stderr, "get_ue_golomb: expected %d, got %d. bits: %7x\n",
54 i, j, s);
55 ret = 1;
56 }
57 }
58
59 #define EXTEND(i) (i << 3 | i & 7)
60 init_put_bits(&pb, temp, SIZE);
61 for (i = 0; i < COUNT; i++)
62 set_ue_golomb(&pb, EXTEND(i));
63 flush_put_bits(&pb);
64
65 init_get_bits(&gb, temp, 8 * SIZE);
66 for (i = 0; i < COUNT; i++) {
67 int j, s = show_bits_long(&gb, 32);
68
69 j = get_ue_golomb_long(&gb);
70 if (j != EXTEND(i)) {
71 fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
72 "bits: %8x\n", EXTEND(i), j, s);
73 ret = 1;
74 }
75 }
76
77 init_put_bits(&pb, temp, SIZE);
78 for (i = 0; i < COUNT; i++)
79 set_se_golomb(&pb, i - COUNT / 2);
80 flush_put_bits(&pb);
81
82 init_get_bits(&gb, temp, 8 * SIZE);
83 for (i = 0; i < COUNT; i++) {
84 int j, s = show_bits(&gb, 25);
85
86 j = get_se_golomb(&gb);
87 if (j != i - COUNT / 2) {
88 fprintf(stderr, "get_se_golomb: expected %d, got %d. bits: %7x\n",
89 i - COUNT / 2, j, s);
90 ret = 1;
91 }
92 }
93
94 av_free(temp);
95
96 return ret;
97 }