Merged VRT work-in-progress from eb/vrt2 (11518:11598) into trunk.

Passes distcheck.


git-svn-id: http://gnuradio.org/svn/gnuradio/trunk@11600 221aa14e-8319-0410-a670-987f0aec2ac5

1 files changed, 144 insertions, 0 deletions

diff --git a/gr-vrt/src/vrt_source_32fc.cc b/gr-vrt/src/vrt_source_32fc.cc
new file mode 100644
index 000000000..beb5ef26d
--- /dev/null
+++ b/gr-vrt/src/vrt_source_32fc.cc
@@ -0,0 +1,144 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2008,2009 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.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <vrt_source_32fc.h>
+#include <vrt/expanded_header.h>
+#include <vrt/copiers.h>
+#include <gr_io_signature.h>
+#include <missing_pkt_checker.h>
+#include <iostream>
+
+#define VERBOSE 1		// define to 0 or 1
+
+
+class rx_32fc_handler : public vrt::rx_packet_handler
+{
+  int 			 	      d_noutput_items;
+  std::complex<float>		     *d_out;
+  int				     *d_oo;		// output index
+  std::vector< std::complex<float> > &d_remainder;
+  missing_pkt_checker		     &d_checker;
+  
+
+public:
+
+  rx_32fc_handler(int noutput_items, std::complex<float> *out,
+		  int *oo, std::vector< std::complex<float> > &remainder,
+		  missing_pkt_checker &checker)
+    : d_noutput_items(noutput_items), d_out(out),
+      d_oo(oo), d_remainder(remainder), d_checker(checker) {}
+
+  ~rx_32fc_handler();
+
+  bool operator()(const uint32_t *payload,
+		  size_t n32_bit_words,
+		  const vrt::expanded_header *hdr);
+};
+
+rx_32fc_handler::~rx_32fc_handler()
+{
+}
+
+bool
+rx_32fc_handler::operator()(const uint32_t *payload,
+			    size_t n32_bit_words,
+			    const vrt::expanded_header *hdr)
+{
+  int nmissing = d_checker.check(hdr);
+  if (VERBOSE && nmissing != 0){
+    std::cerr << "S" << nmissing;
+  }
+
+  // copy as many as will fit into the output buffer.
+
+  size_t n = std::min(n32_bit_words, (size_t)(d_noutput_items - *d_oo));
+  vrt::copy_net_16sc_to_host_32fc(n, payload, &d_out[*d_oo]);
+  *d_oo += n;
+
+  // if there are any left over, copy them into remainder and tell
+  // our caller we're had enough for now.
+
+  size_t r = n32_bit_words - n;
+  if (r > 0){
+    assert(d_remainder.size() == 0);
+    d_remainder.resize(r);
+    vrt::copy_net_16sc_to_host_32fc(r, &payload[n], &d_remainder[0]);
+    return false;		// Stop calling us.
+  }
+
+  return true;			// Keep calling us, we've got more room
+}
+
+
+// ------------------------------------------------------------------------
+
+vrt_source_32fc::vrt_source_32fc(const char *name)
+
+  : vrt_source_base(name,
+		    gr_make_io_signature(1, 1, sizeof(gr_complex)))
+{
+}
+
+vrt_source_32fc::~vrt_source_32fc()
+{
+  if (VERBOSE){
+    std::cerr << "\nvrt_source_32fc: npackets = " << d_checker.npackets()
+	      << " nwrong_pkt_cnt = " << d_checker.nwrong_pkt_cnt()
+	      << " nmissing_pkt_est = " << d_checker.nmissing_pkt_est()
+	      << std::endl;
+  }
+}
+
+int
+vrt_source_32fc::work(int noutput_items,
+		      gr_vector_const_void_star &input_items,
+		      gr_vector_void_star &output_items)
+{
+  gr_complex *out = (gr_complex *)output_items[0];
+  int	oo = 0;
+
+  // Handle any samples left over from the last call.
+  int t = std::min(noutput_items, (int)d_remainder.size());
+  if (t != 0){
+    for (int i = 0; i < t; i++)
+      out[i] = d_remainder[i];
+    d_remainder.erase(d_remainder.begin(), d_remainder.begin()+t);
+    oo = t;
+  }
+  if (noutput_items - oo == 0)
+    return oo;
+  
+  // While we've got room, and there are packets, handle them
+  rx_32fc_handler h(noutput_items, out, &oo, d_remainder, d_checker);
+  bool ok = vrt_rx()->rx_packets(&h);
+
+  if (!ok){
+    std::cerr << "vrt_source_32fc: vrt::rx_packets() failed" << std::endl;
+    return -1;	// say we're done
+  }
+
+  return oo;
+}