tests: Move all test programs to a subdirectory
[libav.git] / libavcodec / tests / golomb.c
CommitLineData
631f96f9
DB
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
e67e3a3f
JG
22#include "libavutil/mem.h"
23
a6a750c7
DB
24#include "libavcodec/get_bits.h"
25#include "libavcodec/put_bits.h"
26#include "libavcodec/golomb.h"
631f96f9 27
e67e3a3f
JG
28#define COUNT 8191
29#define SIZE (COUNT * 4)
631f96f9 30
9e3a2736
DB
31int main(void)
32{
e67e3a3f
JG
33 int i, ret = 0;
34 uint8_t *temp;
631f96f9
DB
35 PutBitContext pb;
36 GetBitContext gb;
37
e67e3a3f
JG
38 temp = av_malloc(SIZE);
39 if (!temp)
40 return 2;
41
631f96f9 42 init_put_bits(&pb, temp, SIZE);
9e3a2736 43 for (i = 0; i < COUNT; i++)
631f96f9 44 set_ue_golomb(&pb, i);
631f96f9
DB
45 flush_put_bits(&pb);
46
9e3a2736
DB
47 init_get_bits(&gb, temp, 8 * SIZE);
48 for (i = 0; i < COUNT; i++) {
e67e3a3f 49 int j, s = show_bits(&gb, 25);
631f96f9 50
9e3a2736 51 j = get_ue_golomb(&gb);
e67e3a3f
JG
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 }
631f96f9
DB
75 }
76
77 init_put_bits(&pb, temp, SIZE);
9e3a2736
DB
78 for (i = 0; i < COUNT; i++)
79 set_se_golomb(&pb, i - COUNT / 2);
631f96f9
DB
80 flush_put_bits(&pb);
81
9e3a2736
DB
82 init_get_bits(&gb, temp, 8 * SIZE);
83 for (i = 0; i < COUNT; i++) {
e67e3a3f 84 int j, s = show_bits(&gb, 25);
631f96f9 85
9e3a2736 86 j = get_se_golomb(&gb);
e67e3a3f
JG
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 }
631f96f9
DB
92 }
93
e67e3a3f
JG
94 av_free(temp);
95
96 return ret;
631f96f9 97}