summaryrefslogtreecommitdiff
path: root/gr-zeromq/lib/sub_source_impl.cc
diff options
context:
space:
mode:
authorCamilo Solano <solano@ti.rwth-aachen.de>2014-05-13 12:53:23 +0200
committerCamilo Solano <solano@ti.rwth-aachen.de>2014-05-13 12:53:23 +0200
commit9970854f8938fcc015608dc163442467427cb9a6 (patch)
tree31390248ae3252291b15fcb1a3a2819eaa5ce2c6 /gr-zeromq/lib/sub_source_impl.cc
parent746737b4a9fc74bf4e86debdf35e67b8238da8b9 (diff)
Add sub_source block and fix pub_sink
Diffstat (limited to 'gr-zeromq/lib/sub_source_impl.cc')
-rw-r--r--gr-zeromq/lib/sub_source_impl.cc102
1 files changed, 102 insertions, 0 deletions
diff --git a/gr-zeromq/lib/sub_source_impl.cc b/gr-zeromq/lib/sub_source_impl.cc
new file mode 100644
index 0000000000..38ddc78e59
--- /dev/null
+++ b/gr-zeromq/lib/sub_source_impl.cc
@@ -0,0 +1,102 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2013,2014 Free Software Foundation, Inc.
+ *
+ * This file is part of GNU Radio.
+ *
+ * This 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.
+ *
+ * This software 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 this software; 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 <gnuradio/io_signature.h>
+#include "sub_source_impl.h"
+
+namespace gr {
+ namespace zeromq {
+
+ sub_source::sptr
+ sub_source::make(size_t itemsize, size_t vlen, char *address, int timeout)
+ {
+ return gnuradio::get_initial_sptr
+ (new sub_source_impl(itemsize, vlen, address, timeout));
+ }
+
+ sub_source_impl::sub_source_impl(size_t itemsize, size_t vlen, char *address, int timeout)
+ : gr::sync_block("sub_source",
+ gr::io_signature::make(0, 0, 0),
+ gr::io_signature::make(1, 1, itemsize * vlen)),
+ d_itemsize(itemsize), d_vlen(vlen), d_timeout(timeout)
+ {
+ int major, minor, patch;
+ zmq::version (&major, &minor, &patch);
+ if (major < 3) {
+ d_timeout = timeout*1000;
+ }
+ d_context = new zmq::context_t(1);
+ d_socket = new zmq::socket_t(*d_context, ZMQ_SUB);
+ //int time = 0;
+ d_socket->setsockopt(ZMQ_SUBSCRIBE, "", 0);
+ d_socket->connect (address);
+ }
+
+ /*
+ * Our virtual destructor.
+ */
+ sub_source_impl::~sub_source_impl()
+ {
+ d_socket->close();
+ delete d_socket;
+ delete d_context;
+ }
+
+ int
+ sub_source_impl::work(int noutput_items,
+ gr_vector_const_void_star &input_items,
+ gr_vector_void_star &output_items)
+ {
+ char *out = (char*)output_items[0];
+
+ zmq::pollitem_t items[] = { { *d_socket, 0, ZMQ_POLLIN, 0 } };
+ zmq::poll (&items[0], 1, d_timeout);
+
+ // If we got a reply, process
+ if (items[0].revents & ZMQ_POLLIN) {
+
+ // Receive data
+ zmq::message_t msg;
+ d_socket->recv(&msg);
+ // Copy to ouput buffer and return
+ if (msg.size() >= d_itemsize*d_vlen*noutput_items) {
+ memcpy(out, (void *)msg.data(), d_itemsize*d_vlen*noutput_items);
+
+ return noutput_items;
+ }
+ else {
+ memcpy(out, (void *)msg.data(), msg.size());
+
+ return msg.size()/(d_itemsize*d_vlen);
+ }
+ }
+ else {
+ return 0; // FIXME: someday when the scheduler does all the poll/selects
+ }
+ }
+
+ } /* namespace zeromq */
+} /* namespace gr */