yading@10: /*
yading@10:  * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
yading@10:  *
yading@10:  * This file is part of FFmpeg.
yading@10:  *
yading@10:  * FFmpeg is free software; you can redistribute it and/or
yading@10:  * modify it under the terms of the GNU Lesser General Public
yading@10:  * License as published by the Free Software Foundation; either
yading@10:  * version 2.1 of the License, or (at your option) any later version.
yading@10:  *
yading@10:  * FFmpeg is distributed in the hope that it will be useful,
yading@10:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@10:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
yading@10:  * Lesser General Public License for more details.
yading@10:  *
yading@10:  * You should have received a copy of the GNU Lesser General Public
yading@10:  * License along with FFmpeg; if not, write to the Free Software
yading@10:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@10:  */
yading@10: 
yading@10: #include <stdint.h>
yading@10: #include <stdio.h>
yading@10: 
yading@10: #include "libavutil/mem.h"
yading@10: 
yading@10: #include "get_bits.h"
yading@10: #include "golomb.h"
yading@10: #include "put_bits.h"
yading@10: 
yading@10: #define COUNT 8191
yading@10: #define SIZE (COUNT * 4)
yading@10: 
yading@10: int main(void)
yading@10: {
yading@10:     int i, ret = 0;
yading@10:     uint8_t *temp;
yading@10:     PutBitContext pb;
yading@10:     GetBitContext gb;
yading@10: 
yading@10:     temp = av_malloc(SIZE);
yading@10:     if (!temp)
yading@10:         return 2;
yading@10: 
yading@10:     init_put_bits(&pb, temp, SIZE);
yading@10:     for (i = 0; i < COUNT; i++)
yading@10:         set_ue_golomb(&pb, i);
yading@10:     flush_put_bits(&pb);
yading@10: 
yading@10:     init_get_bits(&gb, temp, 8 * SIZE);
yading@10:     for (i = 0; i < COUNT; i++) {
yading@10:         int j, s = show_bits(&gb, 25);
yading@10: 
yading@10:         j = get_ue_golomb(&gb);
yading@10:         if (j != i) {
yading@10:             fprintf(stderr, "get_ue_golomb: expected %d, got %d. bits: %7x\n",
yading@10:                     i, j, s);
yading@10:             ret = 1;
yading@10:         }
yading@10:     }
yading@10: 
yading@10: #define EXTEND(i) (i << 3 | i & 7)
yading@10:     init_put_bits(&pb, temp, SIZE);
yading@10:     for (i = 0; i < COUNT; i++)
yading@10:         set_ue_golomb(&pb, EXTEND(i));
yading@10:     flush_put_bits(&pb);
yading@10: 
yading@10:     init_get_bits(&gb, temp, 8 * SIZE);
yading@10:     for (i = 0; i < COUNT; i++) {
yading@10:         int j, s = show_bits_long(&gb, 32);
yading@10: 
yading@10:         j = get_ue_golomb_long(&gb);
yading@10:         if (j != EXTEND(i)) {
yading@10:             fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
yading@10:                     "bits: %8x\n", EXTEND(i), j, s);
yading@10:             ret = 1;
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     init_put_bits(&pb, temp, SIZE);
yading@10:     for (i = 0; i < COUNT; i++)
yading@10:         set_se_golomb(&pb, i - COUNT / 2);
yading@10:     flush_put_bits(&pb);
yading@10: 
yading@10:     init_get_bits(&gb, temp, 8 * SIZE);
yading@10:     for (i = 0; i < COUNT; i++) {
yading@10:         int j, s = show_bits(&gb, 25);
yading@10: 
yading@10:         j = get_se_golomb(&gb);
yading@10:         if (j != i - COUNT / 2) {
yading@10:             fprintf(stderr, "get_se_golomb: expected %d, got %d. bits: %7x\n",
yading@10:                     i - COUNT / 2, j, s);
yading@10:             ret = 1;
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     av_free(temp);
yading@10: 
yading@10:     return ret;
yading@10: }