diff options
| author | Tim O'Shea <tim.oshea753@gmail.com> | 2014-12-29 16:01:43 +0100 |
|---|---|---|
| committer | Tim O'Shea <tim.oshea753@gmail.com> | 2014-12-29 16:01:43 +0100 |
| commit | cd7e604d527f6324138f778dd15253d78df49350 (patch) | |
| tree | 9f9deb7da50c7c921cddd0d6d450679f3b1947f7 /gr-zeromq/lib/sub_msg_source_impl.cc | |
| parent | 232a9bc8f9e05671e54fc204ed209dc8e7948f7f (diff) | |
zmq: Adding zmq pub/sub blocks for message passing
Diffstat (limited to 'gr-zeromq/lib/sub_msg_source_impl.cc')
| -rw-r--r-- | gr-zeromq/lib/sub_msg_source_impl.cc | 115 |
1 files changed, 115 insertions, 0 deletions
diff --git a/gr-zeromq/lib/sub_msg_source_impl.cc b/gr-zeromq/lib/sub_msg_source_impl.cc new file mode 100644 index 0000000000..1c63f2d97c --- /dev/null +++ b/gr-zeromq/lib/sub_msg_source_impl.cc @@ -0,0 +1,115 @@ +/* -*- 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_msg_source_impl.h" +#include "tag_headers.h" + +namespace gr { + namespace zeromq { + + sub_msg_source::sptr + sub_msg_source::make(char *address, int timeout) + { + return gnuradio::get_initial_sptr + (new sub_msg_source_impl(address, timeout)); + } + + sub_msg_source_impl::sub_msg_source_impl(char *address, int timeout) + : gr::sync_block("sub_msg_source", + gr::io_signature::make(0, 0, 0), + gr::io_signature::make(0, 0, 0)), + 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); + + message_port_register_out(pmt::mp("out")); + } + + sub_msg_source_impl::~sub_msg_source_impl() + { + d_socket->close(); + delete d_socket; + delete d_context; + } + + bool sub_msg_source_impl::start(){ + d_finished = false; + d_thread = new boost::thread( boost::bind( &sub_msg_source_impl::readloop , this ) ); + } + + bool sub_msg_source_impl::stop(){ + d_finished = true; + d_thread->join(); + } + + void sub_msg_source_impl::readloop(){ + while(!d_finished){ + //std::cout << "readloop\n"; + + 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); + + //std::cout << "got msg...\n"; + + std::string buf(static_cast<char*>(msg.data()), msg.size()); + std::stringbuf sb(buf); + pmt::pmt_t m = pmt::deserialize(sb); + //std::cout << m << "\n"; + message_port_pub(pmt::mp("out"), m); + + } else { + usleep(100); + } + } + } + + int + sub_msg_source_impl::work(int noutput_items, + gr_vector_const_void_star &input_items, + gr_vector_void_star &output_items) + { + return noutput_items; + } + + } /* namespace zeromq */ +} /* namespace gr */ |