h264: Split h264-test off into a separate file - golomb-test.c.
[ffmpeg.git] / libavcodec / golomb-test.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 "avcodec.h"
23 #include "dsputil.h"
24 #include "get_bits.h"
25 #include "golomb.h"
26 #include "put_bits.h"
27
28 #undef printf
29 #define COUNT 8000
30 #define SIZE (COUNT*40)
31
32 int main(void){
33     int i;
34     uint8_t temp[SIZE];
35     PutBitContext pb;
36     GetBitContext gb;
37
38     init_put_bits(&pb, temp, SIZE);
39     printf("testing unsigned exp golomb\n");
40     for(i=0; i<COUNT; i++){
41         set_ue_golomb(&pb, i);
42     }
43     flush_put_bits(&pb);
44
45     init_get_bits(&gb, temp, 8*SIZE);
46     for(i=0; i<COUNT; i++){
47         int j, s = show_bits(&gb, 24);
48
49         j= get_ue_golomb(&gb);
50         if(j != i){
51             printf("mismatch! at %d (%d should be %d) bits:%6X\n", i, j, i, s);
52         }
53     }
54
55     init_put_bits(&pb, temp, SIZE);
56     printf("testing signed exp golomb\n");
57     for(i=0; i<COUNT; i++){
58         set_se_golomb(&pb, i - COUNT/2);
59     }
60     flush_put_bits(&pb);
61
62     init_get_bits(&gb, temp, 8*SIZE);
63     for(i=0; i<COUNT; i++){
64         int j, s = show_bits(&gb, 24);
65
66         j= get_se_golomb(&gb);
67         if(j != i - COUNT/2){
68             printf("mismatch! at %d (%d should be %d) bits:%6X\n", i, j, i, s);
69         }
70     }
71
72     return 0;
73 }