FFmpeg
golomb.c
Go to the documentation of this file.
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 #include "libavcodec/put_golomb.h"
31 
32 #define COUNT 8191
33 #define SIZE (COUNT * 4)
34 
35 int main(void)
36 {
37  int i, ret = 0;
38  uint8_t *temp;
39  PutBitContext pb;
40  GetBitContext gb;
41 
42  temp = av_malloc(SIZE);
43  if (!temp)
44  return 2;
45 
46  init_put_bits(&pb, temp, SIZE);
47  for (i = 0; i < COUNT; i++)
48  set_ue_golomb(&pb, i);
49  flush_put_bits(&pb);
50 
51  init_get_bits(&gb, temp, 8 * SIZE);
52  for (i = 0; i < COUNT; i++) {
53  int j, s = show_bits(&gb, 25);
54 
55  j = get_ue_golomb(&gb);
56  if (j != i) {
57  fprintf(stderr, "get_ue_golomb: expected %d, got %d. bits: %7x\n",
58  i, j, s);
59  ret = 1;
60  }
61  }
62 
63 #define EXTEND(i) ((i) << 3 | (i) & 7)
64  init_put_bits(&pb, temp, SIZE);
65  for (i = 0; i < COUNT; i++)
66  set_ue_golomb(&pb, EXTEND(i));
67  flush_put_bits(&pb);
68 
69  init_get_bits(&gb, temp, 8 * SIZE);
70  for (i = 0; i < COUNT; i++) {
71  int j, s = show_bits_long(&gb, 32);
72 
73  j = get_ue_golomb_long(&gb);
74  if (j != EXTEND(i)) {
75  fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
76  "bits: %8x\n", EXTEND(i), j, s);
77  ret = 1;
78  }
79  }
80 
81 #define EXTEND_L(i) ((i) << 4 | (i) & 15)
82  init_put_bits(&pb, temp, SIZE);
83  for (i = 0; i < COUNT; i++)
85  flush_put_bits(&pb);
86 
87  init_get_bits(&gb, temp, 8 * SIZE);
88  for (i = 0; i < COUNT; i++) {
89  int j, s = show_bits_long(&gb, 32);
90 
91  j = get_ue_golomb_long(&gb);
92  if (j != EXTEND_L(i)) {
93  fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
94  "bits: %8x\n", EXTEND_L(i), j, s);
95  ret = 1;
96  }
97  }
98 
99  init_put_bits(&pb, temp, SIZE);
100  for (i = 0; i < COUNT; i++)
101  set_se_golomb(&pb, i - COUNT / 2);
102  flush_put_bits(&pb);
103 
104  init_get_bits(&gb, temp, 8 * SIZE);
105  for (i = 0; i < COUNT; i++) {
106  int j, s = show_bits(&gb, 25);
107 
108  j = get_se_golomb(&gb);
109  if (j != i - COUNT / 2) {
110  fprintf(stderr, "get_se_golomb: expected %d, got %d. bits: %7x\n",
111  i - COUNT / 2, j, s);
112  ret = 1;
113  }
114  }
115 
116  av_free(temp);
117 
118  return ret;
119 }
show_bits_long
static unsigned int show_bits_long(GetBitContext *s, int n)
Show 0-32 bits.
Definition: get_bits.h:602
SIZE
#define SIZE
Definition: golomb.c:33
init_put_bits
static void init_put_bits(PutBitContext *s, uint8_t *buffer, int buffer_size)
Initialize the PutBitContext s.
Definition: put_bits.h:62
get_ue_golomb
static int get_ue_golomb(GetBitContext *gb)
Read an unsigned Exp-Golomb code in the range 0 to 8190.
Definition: golomb.h:53
init_get_bits
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
Definition: get_bits.h:649
av_malloc
#define av_malloc(s)
Definition: tableprint_vlc.h:30
golomb.h
exp golomb vlc stuff
EXTEND
#define EXTEND(i)
GetBitContext
Definition: get_bits.h:61
set_ue_golomb_long
static void set_ue_golomb_long(PutBitContext *pb, uint32_t i)
write unsigned exp golomb code.
Definition: put_golomb.h:57
put_golomb.h
exp golomb vlc writing stuff
s
#define s(width, name)
Definition: cbs_vp9.c:256
get_bits.h
get_se_golomb
static int get_se_golomb(GetBitContext *gb)
read signed exp golomb code.
Definition: golomb.h:239
PutBitContext
Definition: put_bits.h:50
COUNT
#define COUNT
Definition: golomb.c:32
set_se_golomb
static void set_se_golomb(PutBitContext *pb, int i)
write signed exp golomb code.
Definition: put_golomb.h:86
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:269
show_bits
static unsigned int show_bits(GetBitContext *s, int n)
Show 1-25 bits.
Definition: get_bits.h:446
internal.h
set_ue_golomb
static void set_ue_golomb(PutBitContext *pb, int i)
write unsigned exp golomb code.
Definition: put_golomb.h:41
ret
ret
Definition: filter_design.txt:187
main
int main(void)
Definition: golomb.c:35
temp
else temp
Definition: vf_mcdeint.c:248
mem.h
get_ue_golomb_long
static unsigned get_ue_golomb_long(GetBitContext *gb)
Read an unsigned Exp-Golomb code in the range 0 to UINT32_MAX-1.
Definition: golomb.h:104
flush_put_bits
static void flush_put_bits(PutBitContext *s)
Pad the end of the output stream with zeros.
Definition: put_bits.h:143
av_free
#define av_free(p)
Definition: tableprint_vlc.h:33
EXTEND_L
#define EXTEND_L(i)
put_bits.h