libs/audiographer/tests/general/interleaver_deinterleaver_test.cc

   1 #include "tests/utils.h"
   2
   3 #include "audiographer/general/interleaver.h"
   4 #include "audiographer/general/deinterleaver.h"
   5
   6 using namespace AudioGrapher;
   7
   8 class InterleaverDeInterleaverTest : public CppUnit::TestFixture
   9 {
  10   CPPUNIT_TEST_SUITE (InterleaverDeInterleaverTest);
  11   CPPUNIT_TEST (testInterleavedInput);
  12   CPPUNIT_TEST (testDeInterleavedInput);
  13   CPPUNIT_TEST_SUITE_END ();
  14
  15   public:
  16         void setUp()
  17         {
  18                 channels = 3;
  19                 frames_per_channel = 128;
  20                 total_frames = channels * frames_per_channel;
  21
  22                 random_data_a = TestUtils::init_random_data (total_frames, 1.0);
  23                 random_data_b = TestUtils::init_random_data (frames_per_channel, 1.0);
  24                 random_data_c = TestUtils::init_random_data (frames_per_channel, 1.0);
  25
  26                 deinterleaver.reset (new DeInterleaver<float>());
  27                 interleaver.reset (new Interleaver<float>());
  28
  29                 sink_a.reset (new VectorSink<float>());
  30                 sink_b.reset (new VectorSink<float>());
  31                 sink_c.reset (new VectorSink<float>());
  32         }
  33
  34         void tearDown()
  35         {
  36                 delete [] random_data_a;
  37                 delete [] random_data_b;
  38                 delete [] random_data_c;
  39         }
  40
  41         void testInterleavedInput()
  42         {
  43                 deinterleaver->init (channels, frames_per_channel);
  44                 interleaver->init (channels, frames_per_channel);
  45
  46                 deinterleaver->output (0)->add_output (interleaver->input (0));
  47                 deinterleaver->output (1)->add_output (interleaver->input (1));
  48                 deinterleaver->output (2)->add_output (interleaver->input (2));
  49
  50                 interleaver->add_output (sink_a);
  51
  52                 // Process and assert
  53                 ProcessContext<float> c (random_data_a, total_frames, channels);
  54                 deinterleaver->process (c);
  55                 CPPUNIT_ASSERT (TestUtils::array_equals (random_data_a, sink_a->get_array(), total_frames));
  56
  57                 // And a second round...
  58                 framecnt_t less_frames = (frames_per_channel / 10) * channels;
  59                 deinterleaver->process (c.beginning (less_frames));
  60                 CPPUNIT_ASSERT (TestUtils::array_equals (random_data_a, sink_a->get_array(), less_frames));
  61         }
  62
  63         void testDeInterleavedInput()
  64         {
  65                 deinterleaver->init (channels, frames_per_channel);
  66                 interleaver->init (channels, frames_per_channel);
  67
  68                 interleaver->add_output (deinterleaver);
  69
  70                 deinterleaver->output (0)->add_output (sink_a);
  71                 deinterleaver->output (1)->add_output (sink_b);
  72                 deinterleaver->output (2)->add_output (sink_c);
  73
  74                 ProcessContext<float> c_a (random_data_a, frames_per_channel, 1);
  75                 ProcessContext<float> c_b (random_data_b, frames_per_channel, 1);
  76                 ProcessContext<float> c_c (random_data_c, frames_per_channel, 1);
  77
  78                 // Process and assert
  79                 interleaver->input (0)->process (c_a);
  80                 interleaver->input (1)->process (c_b);
  81                 interleaver->input (2)->process (c_c);
  82
  83                 CPPUNIT_ASSERT (TestUtils::array_equals (random_data_a, sink_a->get_array(), frames_per_channel));
  84                 CPPUNIT_ASSERT (TestUtils::array_equals (random_data_b, sink_b->get_array(), frames_per_channel));
  85                 CPPUNIT_ASSERT (TestUtils::array_equals (random_data_c, sink_c->get_array(), frames_per_channel));
  86
  87                 // And a second round...
  88                 framecnt_t less_frames = frames_per_channel / 5;
  89                 interleaver->input (0)->process (c_a.beginning (less_frames));
  90                 interleaver->input (1)->process (c_b.beginning (less_frames));
  91                 interleaver->input (2)->process (c_c.beginning (less_frames));
  92
  93                 CPPUNIT_ASSERT (TestUtils::array_equals (random_data_a, sink_a->get_array(), less_frames));
  94                 CPPUNIT_ASSERT (TestUtils::array_equals (random_data_b, sink_b->get_array(), less_frames));
  95                 CPPUNIT_ASSERT (TestUtils::array_equals (random_data_c, sink_c->get_array(), less_frames));
  96
  97         }
  98
  99   private:
 100         boost::shared_ptr<Interleaver<float> > interleaver;
 101         boost::shared_ptr<DeInterleaver<float> > deinterleaver;
 102
 103         boost::shared_ptr<VectorSink<float> > sink_a;
 104         boost::shared_ptr<VectorSink<float> > sink_b;
 105         boost::shared_ptr<VectorSink<float> > sink_c;
 106
 107         float * random_data_a;
 108         float * random_data_b;
 109         float * random_data_c;
 110
 111         framecnt_t frames_per_channel;
 112         framecnt_t total_frames;
 113         unsigned int channels;
 114 };
 115
 116 CPPUNIT_TEST_SUITE_REGISTRATION (InterleaverDeInterleaverTest);
 117