diff options
| author | Johnathan Corgan <johnathan@corganlabs.com> | 2014-04-21 11:27:54 -0700 |
|---|---|---|
| committer | Johnathan Corgan <johnathan@corganlabs.com> | 2014-04-22 15:07:42 -0700 |
| commit | 9833f3fb80c788d740dc1328899d50165cfdfa90 (patch) | |
| tree | 31b24e4ecc8604fb3b70ad0cdc72a86483885e65 /gr-zeromq/lib/pull_source_impl.cc | |
| parent | 687b81073d4997a73183cd69b2c467ea728a5ef1 (diff) | |
zeromq: renamed source_pushpull to pull_source
Diffstat (limited to 'gr-zeromq/lib/pull_source_impl.cc')
| -rw-r--r-- | gr-zeromq/lib/pull_source_impl.cc | 96 |
1 files changed, 96 insertions, 0 deletions
diff --git a/gr-zeromq/lib/pull_source_impl.cc b/gr-zeromq/lib/pull_source_impl.cc new file mode 100644 index 0000000000..637b529a72 --- /dev/null +++ b/gr-zeromq/lib/pull_source_impl.cc @@ -0,0 +1,96 @@ +/* -*- 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 "pull_source_impl.h" + +namespace gr { + namespace zeromq { + + pull_source::sptr + pull_source::make(size_t itemsize, char *address, float timeout) + { + return gnuradio::get_initial_sptr + (new pull_source_impl(itemsize, address, timeout)); + } + + pull_source_impl::pull_source_impl(size_t itemsize, char *address, float timeout) + : gr::sync_block("pull_source", + gr::io_signature::make(0, 0, 0), + gr::io_signature::make(1, 1, itemsize)), + d_itemsize(itemsize) + { + d_timeout = timeout >=0 ? (int)(timeout*1e6) : 0; + d_context = new zmq::context_t(1); + d_socket = new zmq::socket_t(*d_context, ZMQ_PULL); + d_socket->connect (address); + } + + /* + * Our virtual destructor. + */ + pull_source_impl::~pull_source_impl() + { + delete d_socket; + delete d_context; + } + + int + pull_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*noutput_items) { + memcpy(out, (void *)msg.data(), d_itemsize*noutput_items); + + return noutput_items; + } + else { + memcpy(out, (void *)msg.data(), msg.size()); + + return msg.size()/d_itemsize; + } + } + else { + return 0; // FIXME: someday when the scheduler does all the poll/selects + } + } + + } /* namespace zeromq */ +} /* namespace gr */ |