libavcodec/golomb-test.c

   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/mem.h"
  25
  26 #include "get_bits.h"
  27 #include "golomb.h"
  28 #include "put_bits.h"
  29
  30 #define COUNT 8191
  31 #define SIZE (COUNT * 4)
  32
  33 int main(void)
  34 {
  35     int i, ret = 0;
  36     uint8_t *temp;
  37     PutBitContext pb;
  38     GetBitContext gb;
  39
  40     temp = av_malloc(SIZE);
  41     if (!temp)
  42         return 2;
  43
  44     init_put_bits(&pb, temp, SIZE);
  45     for (i = 0; i < COUNT; i++)
  46         set_ue_golomb(&pb, i);
  47     flush_put_bits(&pb);
  48
  49     init_get_bits(&gb, temp, 8 * SIZE);
  50     for (i = 0; i < COUNT; i++) {
  51         int j, s = show_bits(&gb, 25);
  52
  53         j = get_ue_golomb(&gb);
  54         if (j != i) {
  55             fprintf(stderr, "get_ue_golomb: expected %d, got %d. bits: %7x\n",
  56                     i, j, s);
  57             ret = 1;
  58         }
  59     }
  60
  61 #define EXTEND(i) ((i) << 3 | (i) & 7)
  62     init_put_bits(&pb, temp, SIZE);
  63     for (i = 0; i < COUNT; i++)
  64         set_ue_golomb(&pb, EXTEND(i));
  65     flush_put_bits(&pb);
  66
  67     init_get_bits(&gb, temp, 8 * SIZE);
  68     for (i = 0; i < COUNT; i++) {
  69         int j, s = show_bits_long(&gb, 32);
  70
  71         j = get_ue_golomb_long(&gb);
  72         if (j != EXTEND(i)) {
  73             fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
  74                     "bits: %8x\n", EXTEND(i), j, s);
  75             ret = 1;
  76         }
  77     }
  78
  79     init_put_bits(&pb, temp, SIZE);
  80     for (i = 0; i < COUNT; i++)
  81         set_se_golomb(&pb, i - COUNT / 2);
  82     flush_put_bits(&pb);
  83
  84     init_get_bits(&gb, temp, 8 * SIZE);
  85     for (i = 0; i < COUNT; i++) {
  86         int j, s = show_bits(&gb, 25);
  87
  88         j = get_se_golomb(&gb);
  89         if (j != i - COUNT / 2) {
  90             fprintf(stderr, "get_se_golomb: expected %d, got %d. bits: %7x\n",
  91                     i - COUNT / 2, j, s);
  92             ret = 1;
  93         }
  94     }
  95
  96     av_free(temp);
  97
  98     return ret;
  99 }