golomb-test.c 2.58 KB
Newer Older
1
/*
2
 * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
3
 *
4 5 6
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
7 8 9 10
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
11
 * FFmpeg is distributed in the hope that it will be useful,
12 13 14 15 16
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * 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 19 20 21 22 23
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include <stdint.h>
#include <stdio.h>

24 25
#include "libavutil/mem.h"

26 27 28 29
#include "get_bits.h"
#include "golomb.h"
#include "put_bits.h"

30 31
#define COUNT 8191
#define SIZE (COUNT * 4)
32

33 34
int main(void)
{
35 36
    int i, ret = 0;
    uint8_t *temp;
37 38 39
    PutBitContext pb;
    GetBitContext gb;

40 41 42 43
    temp = av_malloc(SIZE);
    if (!temp)
        return 2;

44
    init_put_bits(&pb, temp, SIZE);
45
    for (i = 0; i < COUNT; i++)
46 47 48
        set_ue_golomb(&pb, i);
    flush_put_bits(&pb);

49 50
    init_get_bits(&gb, temp, 8 * SIZE);
    for (i = 0; i < COUNT; i++) {
51
        int j, s = show_bits(&gb, 25);
52

53
        j = get_ue_golomb(&gb);
54 55 56 57 58 59 60
        if (j != i) {
            fprintf(stderr, "get_ue_golomb: expected %d, got %d. bits: %7x\n",
                    i, j, s);
            ret = 1;
        }
    }

61
#define EXTEND(i) ((i) << 3 | (i) & 7)
62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
    init_put_bits(&pb, temp, SIZE);
    for (i = 0; i < COUNT; i++)
        set_ue_golomb(&pb, EXTEND(i));
    flush_put_bits(&pb);

    init_get_bits(&gb, temp, 8 * SIZE);
    for (i = 0; i < COUNT; i++) {
        int j, s = show_bits_long(&gb, 32);

        j = get_ue_golomb_long(&gb);
        if (j != EXTEND(i)) {
            fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
                    "bits: %8x\n", EXTEND(i), j, s);
            ret = 1;
        }
77 78 79
    }

    init_put_bits(&pb, temp, SIZE);
80 81
    for (i = 0; i < COUNT; i++)
        set_se_golomb(&pb, i - COUNT / 2);
82 83
    flush_put_bits(&pb);

84 85
    init_get_bits(&gb, temp, 8 * SIZE);
    for (i = 0; i < COUNT; i++) {
86
        int j, s = show_bits(&gb, 25);
87

88
        j = get_se_golomb(&gb);
89 90 91 92 93
        if (j != i - COUNT / 2) {
            fprintf(stderr, "get_se_golomb: expected %d, got %d. bits: %7x\n",
                    i - COUNT / 2, j, s);
            ret = 1;
        }
94 95
    }

96 97 98
    av_free(temp);

    return ret;
99
}