git.sesse.net Git - ffmpeg/blob - libavcodec/tests/golomb.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/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
  31 #define COUNT 8191
  32 #define SIZE (COUNT * 4)
  33
  34 int main(void)
  35 {
  36     int i, ret = 0;
  37     uint8_t *temp;
  38     PutBitContext pb;
  39     GetBitContext gb;
  40
  41     temp = av_malloc(SIZE);
  42     if (!temp)
  43         return 2;
  44
  45     init_put_bits(&pb, temp, SIZE);
  46     for (i = 0; i < COUNT; i++)
  47         set_ue_golomb(&pb, i);
  48     flush_put_bits(&pb);
  49
  50     init_get_bits(&gb, temp, 8 * SIZE);
  51     for (i = 0; i < COUNT; i++) {
  52         int j, s = show_bits(&gb, 25);
  53
  54         j = get_ue_golomb(&gb);
  55         if (j != i) {
  56             fprintf(stderr, "get_ue_golomb: expected %d, got %d. bits: %7x\n",
  57                     i, j, s);
  58             ret = 1;
  59         }
  60     }
  61
  62 #define EXTEND(i) ((i) << 3 | (i) & 7)
  63     init_put_bits(&pb, temp, SIZE);
  64     for (i = 0; i < COUNT; i++)
  65         set_ue_golomb(&pb, EXTEND(i));
  66     flush_put_bits(&pb);
  67
  68     init_get_bits(&gb, temp, 8 * SIZE);
  69     for (i = 0; i < COUNT; i++) {
  70         int j, s = show_bits_long(&gb, 32);
  71
  72         j = get_ue_golomb_long(&gb);
  73         if (j != EXTEND(i)) {
  74             fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
  75                     "bits: %8x\n", EXTEND(i), j, s);
  76             ret = 1;
  77         }
  78     }
  79
  80 #define EXTEND_L(i) ((i) << 4 | (i) & 15)
  81     init_put_bits(&pb, temp, SIZE);
  82     for (i = 0; i < COUNT; i++)
  83         set_ue_golomb_long(&pb, EXTEND_L(i));
  84     flush_put_bits(&pb);
  85
  86     init_get_bits(&gb, temp, 8 * SIZE);
  87     for (i = 0; i < COUNT; i++) {
  88         int j, s = show_bits_long(&gb, 32);
  89
  90         j = get_ue_golomb_long(&gb);
  91         if (j != EXTEND_L(i)) {
  92             fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
  93                     "bits: %8x\n", EXTEND_L(i), j, s);
  94             ret = 1;
  95         }
  96     }
  97
  98     init_put_bits(&pb, temp, SIZE);
  99     for (i = 0; i < COUNT; i++)
 100         set_se_golomb(&pb, i - COUNT / 2);
 101     flush_put_bits(&pb);
 102
 103     init_get_bits(&gb, temp, 8 * SIZE);
 104     for (i = 0; i < COUNT; i++) {
 105         int j, s = show_bits(&gb, 25);
 106
 107         j = get_se_golomb(&gb);
 108         if (j != i - COUNT / 2) {
 109             fprintf(stderr, "get_se_golomb: expected %d, got %d. bits: %7x\n",
 110                     i - COUNT / 2, j, s);
 111             ret = 1;
 112         }
 113     }
 114
 115     av_free(temp);
 116
 117     return ret;
 118 }