#include <stdint.h>
#include <stdio.h>
-#include "avcodec.h"
-#include "dsputil.h"
+#include "libavutil/mem.h"
+
#include "get_bits.h"
#include "golomb.h"
#include "put_bits.h"
-#undef printf
-#define COUNT 8000
-#define SIZE (COUNT*40)
+#undef fprintf
+#define COUNT 8191
+#define SIZE (COUNT * 4)
-int main(void){
- int i;
- uint8_t temp[SIZE];
+int main(void)
+{
+ int i, ret = 0;
+ uint8_t *temp;
PutBitContext pb;
GetBitContext gb;
+ temp = av_malloc(SIZE);
+ if (!temp)
+ return 2;
+
init_put_bits(&pb, temp, SIZE);
- printf("testing unsigned exp golomb\n");
- for(i=0; i<COUNT; i++){
+ for (i = 0; i < COUNT; i++)
set_ue_golomb(&pb, i);
- }
flush_put_bits(&pb);
- init_get_bits(&gb, temp, 8*SIZE);
- for(i=0; i<COUNT; i++){
- int j, s = show_bits(&gb, 24);
+ init_get_bits(&gb, temp, 8 * SIZE);
+ for (i = 0; i < COUNT; i++) {
+ int j, s = show_bits(&gb, 25);
- j= get_ue_golomb(&gb);
- if(j != i){
- printf("mismatch! at %d (%d should be %d) bits:%6X\n", i, j, i, s);
+ j = get_ue_golomb(&gb);
+ if (j != i) {
+ fprintf(stderr, "get_ue_golomb: expected %d, got %d. bits: %7x\n",
+ i, j, s);
+ ret = 1;
}
}
+#define EXTEND(i) (i << 3 | i & 7)
init_put_bits(&pb, temp, SIZE);
- printf("testing signed exp golomb\n");
- for(i=0; i<COUNT; i++){
- set_se_golomb(&pb, i - COUNT/2);
+ 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;
+ }
}
+
+ init_put_bits(&pb, temp, SIZE);
+ for (i = 0; i < COUNT; i++)
+ set_se_golomb(&pb, i - COUNT / 2);
flush_put_bits(&pb);
- init_get_bits(&gb, temp, 8*SIZE);
- for(i=0; i<COUNT; i++){
- int j, s = show_bits(&gb, 24);
+ init_get_bits(&gb, temp, 8 * SIZE);
+ for (i = 0; i < COUNT; i++) {
+ int j, s = show_bits(&gb, 25);
- j= get_se_golomb(&gb);
- if(j != i - COUNT/2){
- printf("mismatch! at %d (%d should be %d) bits:%6X\n", i, j, i, s);
+ j = get_se_golomb(&gb);
+ if (j != i - COUNT / 2) {
+ fprintf(stderr, "get_se_golomb: expected %d, got %d. bits: %7x\n",
+ i - COUNT / 2, j, s);
+ ret = 1;
}
}
- return 0;
+ av_free(temp);
+
+ return ret;
}