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