Merge branch 'dfsg-orig'

[debian/gnuradio] / gr-atsc / src / lib / qa_atsci_data_interleaver.cc

diff --git a/gr-atsc/src/lib/qa_atsci_data_interleaver.cc b/gr-atsc/src/lib/qa_atsci_data_interleaver.cc
new file mode 100644 (file)
index 0000000..e76a2f1
--- /dev/null
+++ b/gr-atsc/src/lib/qa_atsci_data_interleaver.cc
@@ -0,0 +1,197 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2002 Free Software Foundation, Inc.
+ * 
+ * This file is part of GNU Radio
+ * 
+ * GNU Radio is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3, or (at your option)
+ * any later version.
+ * 
+ * GNU Radio is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License
+ * along with GNU Radio; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#include <cppunit/TestAssert.h>
+#include <qa_atsci_data_interleaver.h>
+#include <string.h>
+
+
+/*!
+ * write an easy to identify pattern into the packet
+ */
+void
+qa_atsci_data_interleaver::init_test_packet (int counter,
+                                           atsc_mpeg_packet_rs_encoded &out)
+{
+  out.data[0] = 0xf0;
+  out.data[1] = 0xff;
+  out.data[2] = (counter >> 8) & 0xff;
+  out.data[3] = counter & 0xff;
+
+  for (int i = 4; i < 205; i++)
+    out.data[i] = i;
+
+  out.data[205] = 0xa0;
+  out.data[206] = 0xaf;
+}
+
+void
+qa_atsci_data_interleaver::print_packet (FILE *fp,
+                                       int frame_number,
+                                       int field_number,
+                                       int segment_number,
+                                       const atsc_mpeg_packet_rs_encoded &in)
+{
+  fprintf (fp, "%04d:%d:%03d  ", frame_number, field_number, segment_number);
+
+  const unsigned char *p = &in.data[0];
+  for (int i = 0; i < 8; i++)
+    fprintf (fp, "%02X", p[i]);
+
+  fprintf (fp, "  ");
+  p = &in.data[8];
+  for (int i = 0; i < 8; i++)
+    fprintf (fp, "%02X", p[i]);
+
+  
+  fprintf (fp, "  ...  ");
+  p = &in.data[191];
+  for (int i = 0; i < 8; i++)
+    fprintf (fp, "%02X", p[i]);
+
+  fprintf (fp, "  ");
+  p = &in.data[199];
+  for (int i = 0; i < 8; i++)
+    fprintf (fp, "%02X", p[i]);
+
+  fprintf (fp, "\n");
+}
+
+void
+qa_atsci_data_interleaver::chk_assert (const atsc_mpeg_packet_rs_encoded &expected,
+                                     const atsc_mpeg_packet_rs_encoded &actual)
+{
+  if (expected == actual)
+    return;
+
+  FILE *fp = stderr;
+
+  fprintf (fp, "Expected: ");
+  print_packet (fp, 0, 0, 0, expected);
+
+  fprintf (fp, "Actual:   ");
+  print_packet (fp, 0, 0, 0, actual);
+
+  CPPUNIT_ASSERT (expected == actual);
+}
+
+void
+qa_atsci_data_interleaver::t0 ()
+{
+  int  counter = 0;
+  atsc_mpeg_packet_rs_encoded  in, enc, out;
+  atsc_mpeg_packet_rs_encoded  zero;
+
+  memset (&zero, 0, sizeof (zero));
+  
+  for (int frame = 0; frame < 4; frame++){
+    for (int field = 0; field < 2; field++){
+      for (int segment = 0; segment < 312; segment++, counter++){
+
+       // build test packet
+       init_test_packet (counter, in);
+       in.pli.set_regular_seg (field == 1, segment);
+
+       // interleave it
+       outbound.interleave (enc, in);
+
+       // deinterleave it
+       inbound.deinterleave (out, enc);
+
+       if (counter < 52)
+         // CPPUNIT_ASSERT (zero == out);
+         chk_assert (zero, out);
+
+       else if (counter >= 52){
+         atsc_mpeg_packet_rs_encoded expected;
+         init_test_packet (counter - 52, expected);
+         // CPPUNIT_ASSERT (expected == out);
+         chk_assert (expected, out);
+       }
+      }
+    }
+  }
+}
+
+//
+// Note, this assumes that the interleaver and deinterleaver
+// have the state left by t0.
+//
+// This test pushes crap into the interleaver, and then confirms that
+// the deinterleaver recovers.  This would be part of what you'd see
+// on a channel change.
+//
+
+void
+qa_atsci_data_interleaver::t1 ()
+{
+  int  counter = 0;
+  atsc_mpeg_packet_rs_encoded  in, enc, out;
+  atsc_mpeg_packet_rs_encoded  zero;
+
+  memset (&zero, 0, sizeof (zero));
+  
+  static const int NCRAP = 13;
+  
+  // push NCRAP segments of crap into the interleaver,
+  // but don't change the deinterleaver state
+
+  for (int i = 0; i < NCRAP; i++){
+    init_test_packet (i + 2758, in);
+    in.pli.set_regular_seg (false, i + 5);
+
+    outbound.interleave (enc, in);
+  }
+
+  // now run the normal test.
+  // we should get good segments after NCRAP + 52
+  
+  int starting_counter = 3141;
+  counter = starting_counter;
+
+  for (int frame = 0; frame < 4; frame++){
+    for (int field = 0; field < 2; field++){
+      for (int segment = 0; segment < 312; segment++, counter++){
+
+       // build test packet
+       init_test_packet (counter, in);
+       in.pli.set_regular_seg (field == 1, segment);
+
+       // interleave it
+       outbound.interleave (enc, in);
+
+       // deinterleave it
+       inbound.deinterleave (out, enc);
+
+       if (counter < starting_counter + 52 + NCRAP){
+         // don't care...
+       }
+       else if (counter >= starting_counter + 52 + NCRAP){
+         atsc_mpeg_packet_rs_encoded expected;
+         init_test_packet (counter - 52, expected);
+         // CPPUNIT_ASSERT (expected == out);
+         chk_assert (expected, out);
+       }
+      }
+    }
+  }
+}