tests/check/libs/bitwriter.c

   1 /* GStreamer
   2  *
   3  * unit test for GstBitWriter
   4  *
   5  * Copyright (C) <2014> Intel Corporation
   6  * Copyright (C) <2014> Sreerenj Balachandran <sreerenj.balachandran@intel.com>
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Library General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2 of the License, or (at your option) any later version.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Library General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Library General Public
  19  * License along with this library; if not, write to the
  20  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  21  * Boston, MA 02110-1301, USA.
  22  */
  23
  24 #ifdef HAVE_CONFIG_H
  25 #include "config.h"
  26 #endif
  27
  28 #include <gst/gst.h>
  29 #include <gst/check/gstcheck.h>
  30 #include <gst/base/gstbitwriter.h>
  31 #include <gst/base/gstbitreader.h>
  32 #include "gst/glib-compat-private.h"
  33
  34 GST_START_TEST (test_initialization)
  35 {
  36   GstBitWriter writer;
  37   GstBitReader reader;
  38
  39   static guint8 sdata[] = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
  40
  41   gst_bit_writer_init_with_size (&writer, 4, FALSE);
  42   /* should be aligned to 256 bytes */
  43   fail_unless_equals_int (gst_bit_writer_get_remaining (&writer), 2048);
  44   fail_unless_equals_int (gst_bit_writer_get_size (&writer), 0);
  45   fail_unless_equals_int (gst_bit_writer_set_pos (&writer, 32), 1);
  46   fail_unless_equals_int (gst_bit_writer_get_size (&writer), 32);
  47   fail_unless_equals_int (gst_bit_writer_get_remaining (&writer), 2048 - 32);
  48   gst_bit_writer_reset (&writer);
  49
  50   gst_bit_writer_init_with_data (&writer, sdata, 8, FALSE);
  51   gst_bit_reader_init (&reader, sdata, 8);
  52   fail_unless_equals_int (gst_bit_reader_get_size (&reader), 64);
  53   fail_unless_equals_int (gst_bit_reader_get_remaining (&reader), 64);
  54   gst_bit_writer_reset (&writer);
  55 }
  56
  57 GST_END_TEST;
  58
  59 GST_START_TEST (test_data)
  60 {
  61   GstBitWriter writer;
  62   GstBitReader reader;
  63   guint8 val_uint8 = 0;
  64   guint16 val_uint16 = 0;
  65   guint32 val_uint32 = 0;
  66   guint64 val_uint64 = 0;
  67   static guint8 sdata[] = { 0xff, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7 };
  68   guint8 *data, i;
  69
  70   gst_bit_writer_init_with_size (&writer, 32, TRUE);
  71   fail_unless_equals_int (gst_bit_writer_get_remaining (&writer), 2048);
  72
  73   fail_unless (gst_bit_writer_put_bits_uint8 (&writer, 0x3f, 6));
  74   fail_unless (gst_bit_writer_put_bits_uint8 (&writer, 0x3, 2));
  75   fail_unless (gst_bit_writer_put_bits_uint16 (&writer, 0x15, 5));
  76   fail_unless (gst_bit_writer_put_bits_uint32 (&writer, 0x31, 10));
  77   fail_unless (gst_bit_writer_put_bits_uint64 (&writer, 0x45, 48));
  78   fail_unless_equals_int (gst_bit_writer_get_remaining (&writer), 2048 - 71);
  79   fail_unless (gst_bit_writer_align_bytes (&writer, 0));
  80   data = g_memdup2 (sdata, sizeof (sdata));
  81   fail_unless (gst_bit_writer_put_bytes (&writer, data, sizeof (sdata)));
  82
  83   gst_bit_reader_init (&reader, gst_bit_writer_get_data (&writer), 256);
  84   fail_unless_equals_int (gst_bit_reader_get_size (&reader), 256 * 8);
  85   fail_unless (gst_bit_reader_get_bits_uint8 (&reader, &val_uint8, 6));
  86   fail_unless_equals_int (val_uint8, 0x3f);
  87   fail_unless (gst_bit_reader_get_bits_uint8 (&reader, &val_uint8, 2));
  88   fail_unless_equals_int (val_uint8, 0x3);
  89   fail_unless (gst_bit_reader_get_bits_uint16 (&reader, &val_uint16, 5));
  90   fail_unless_equals_int_hex (val_uint16, 0x15);
  91   fail_unless (gst_bit_reader_get_bits_uint32 (&reader, &val_uint32, 10));
  92   fail_unless_equals_int_hex (val_uint32, 0x31);
  93   fail_unless (gst_bit_reader_get_bits_uint64 (&reader, &val_uint64, 48));
  94   fail_unless_equals_int_hex (val_uint64, 0x45);
  95   fail_unless (gst_bit_reader_set_pos (&reader, 72));
  96
  97   for (i = 0; i < 8; i++) {
  98     fail_unless (gst_bit_reader_get_bits_uint8 (&reader, &val_uint8, 8));
  99     fail_unless (memcmp (&val_uint8, &data[i], 1) == 0);
 100   }
 101
 102   g_free (data);
 103   gst_bit_writer_reset (&writer);
 104 }
 105
 106 GST_END_TEST;
 107
 108 GST_START_TEST (test_reset)
 109 {
 110   GstBitWriter writer, *writer2;
 111   GstBitReader reader;
 112   GstBuffer *buf;
 113   GstMapInfo info;
 114   static guint8 sdata[] = { 0xff, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7 };
 115   guint8 *data;
 116   guint8 val_uint8 = 0;
 117
 118   gst_bit_writer_init_with_data (&writer, sdata, sizeof (sdata), TRUE);
 119   data = gst_bit_writer_reset_and_get_data (&writer);
 120   fail_unless (data != NULL);
 121
 122   gst_bit_writer_init_with_data (&writer, sdata, sizeof (sdata), TRUE);
 123   buf = gst_bit_writer_reset_and_get_buffer (&writer);
 124   fail_unless (buf != NULL);
 125   fail_unless (gst_buffer_map (buf, &info, GST_MAP_READWRITE));
 126   fail_unless (info.data);
 127   fail_unless_equals_int (info.size, 8);
 128   gst_bit_reader_init (&reader, info.data, 8);
 129   fail_unless (gst_bit_reader_set_pos (&reader, 64 - 10));
 130   fail_unless (gst_bit_reader_get_bits_uint8 (&reader, &val_uint8, 8));
 131   fail_unless_equals_int_hex (val_uint8, 0xbd);
 132   fail_unless (gst_bit_reader_get_bits_uint8 (&reader, &val_uint8, 2));
 133   fail_unless_equals_int_hex (val_uint8, 0x3);
 134   gst_buffer_unmap (buf, &info);
 135   gst_buffer_unref (buf);
 136
 137   writer2 = gst_bit_writer_new_with_size (3, TRUE);
 138   data = gst_bit_writer_free_and_get_data (writer2);
 139   /* no data, no buffer allocated */
 140   fail_unless (data == NULL);
 141
 142   writer2 = gst_bit_writer_new_with_size (1, FALSE);
 143   fail_unless (gst_bit_writer_put_bits_uint8 (writer2, 0xff, 8));
 144   data = gst_bit_writer_free_and_get_data (writer2);
 145   fail_unless (data != NULL);
 146   fail_unless_equals_int_hex (*data, 0xff);
 147   g_free (data);
 148 }
 149
 150 GST_END_TEST;
 151
 152
 153 static Suite *
 154 gst_bit_writer_suite (void)
 155 {
 156   Suite *s = suite_create ("GstBitWriter");
 157   TCase *tc_chain = tcase_create ("general");
 158
 159   suite_add_tcase (s, tc_chain);
 160
 161   tcase_add_test (tc_chain, test_initialization);
 162   tcase_add_test (tc_chain, test_data);
 163   tcase_add_test (tc_chain, test_reset);
 164
 165   return s;
 166 }
 167
 168 GST_CHECK_MAIN (gst_bit_writer);