Libav
golomb-test.c
Go to the documentation of this file.
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 "get_bits.h"
25 #include "golomb.h"
26 #include "put_bits.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 }