testsuite/bits3.c

   1
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <time.h>
   5
   6 #include <schroedinger/schro.h>
   7
   8 int fail;
   9
  10 void
  11 dump_bits (SchroBits *bits, int n)
  12 {
  13   int i;
  14
  15   for(i=0;i<n*8;i++){
  16     printf(" %d", (bits->buffer->data[(i>>3)] >> (7 - (i&7))) & 1);
  17   }
  18   printf("\n");
  19 }
  20
  21
  22 int test (void)
  23 {
  24   SchroBuffer *buffer = schro_buffer_new_and_alloc (300);
  25   SchroBits *bits;
  26   SchroBits bits2;
  27   int i;
  28   int chunk1;
  29   int chunk2;
  30   int ref[10];
  31   int x;
  32
  33   chunk1 = rand()&0x1f;
  34
  35   bits = schro_bits_new ();
  36   schro_bits_encode_init (bits, buffer);
  37
  38   for(i=0;i<chunk1;i++){
  39     schro_bits_encode_bit (bits, rand()&1);
  40   }
  41
  42   chunk2 = 0;
  43   for(i=0;i<10;i++){
  44     ref[i] = rand() & 0xf;
  45     schro_bits_encode_sint (bits, ref[i]);
  46     chunk2 += schro_bits_estimate_sint (ref[i]);
  47   }
  48
  49   for(i=0;i<100;i++){
  50     schro_bits_encode_bit (bits, rand()&1);
  51   }
  52
  53   schro_bits_flush (bits);
  54   schro_bits_free (bits);
  55
  56
  57   bits = schro_bits_new ();
  58   schro_bits_decode_init (bits, buffer);
  59
  60   schro_bits_copy (&bits2, bits);
  61
  62   schro_bits_skip_bits (&bits2, chunk1);
  63   schro_bits_set_length (&bits2, chunk2);
  64
  65   printf("chunk1 %d\n", chunk1);
  66   for(i=0;i<10;i++){
  67     x = schro_bits_decode_sint (&bits2);
  68     printf("%d %d\n", ref[i], x);
  69     if (x != ref[i]) fail = TRUE;
  70   }
  71
  72   schro_bits_free (bits);
  73
  74   return 0;
  75 }
  76
  77 int
  78 main (int argc, char *argv[])
  79 {
  80   schro_init();
  81
  82   srand(time(NULL));
  83
  84   test();
  85
  86   return fail;
  87 }
  88
  89
  90