diff options
author | Johnathan Corgan <johnathan@corganlabs.com> | 2013-02-28 14:31:39 -0800 |
---|---|---|
committer | Johnathan Corgan <johnathan@corganlabs.com> | 2013-02-28 14:31:39 -0800 |
commit | 1d093decc5bf37b8d173ce7769a14af6765f45a6 (patch) | |
tree | d0d45936b81a8a429746cff573aa7c9c5f9d179f /gr-blocks/lib | |
parent | 4a99e708ad0dd54bae4a4275ae2c06e44decbebf (diff) |
blocks: added gr::blocks::random_pdu
Diffstat (limited to 'gr-blocks/lib')
-rw-r--r-- | gr-blocks/lib/CMakeLists.txt | 1 | ||||
-rw-r--r-- | gr-blocks/lib/random_pdu_impl.cc | 80 | ||||
-rw-r--r-- | gr-blocks/lib/random_pdu_impl.h | 54 | ||||
-rw-r--r-- | gr-blocks/lib/socket_pdu_impl.cc | 2 | ||||
-rw-r--r-- | gr-blocks/lib/tcp_connection.cc | 4 |
5 files changed, 137 insertions, 4 deletions
diff --git a/gr-blocks/lib/CMakeLists.txt b/gr-blocks/lib/CMakeLists.txt index 4ebaf3dcb8..008f00f361 100644 --- a/gr-blocks/lib/CMakeLists.txt +++ b/gr-blocks/lib/CMakeLists.txt @@ -169,6 +169,7 @@ list(APPEND gr_blocks_sources pdu.cc pdu_to_tagged_stream_impl.cc peak_detector2_fb_impl.cc + random_pdu_impl.cc regenerate_bb_impl.cc repeat_impl.cc rms_cf_impl.cc diff --git a/gr-blocks/lib/random_pdu_impl.cc b/gr-blocks/lib/random_pdu_impl.cc new file mode 100644 index 0000000000..0b194a0537 --- /dev/null +++ b/gr-blocks/lib/random_pdu_impl.cc @@ -0,0 +1,80 @@ +/* -*- c++ -*- */ +/* + * Copyright 2013 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 "random_pdu_impl.h" +#include <gr_io_signature.h> +#include <blocks/pdu.h> + +namespace gr { + namespace blocks { + + random_pdu::sptr + random_pdu::make(int min_items, int max_items) + { + return gnuradio::get_initial_sptr(new random_pdu_impl(min_items, max_items)); + } + + random_pdu_impl::random_pdu_impl(int min_items, int max_items) + : gr_block("random_pdu", + gr_make_io_signature (0, 0, 0), + gr_make_io_signature (0, 0, 0)), + d_urange(min_items, max_items), + d_brange(0, 255), + d_rvar(d_rng, d_urange), + d_bvar(d_rng, d_brange) + { + message_port_register_out(PDU_PORT_ID); + message_port_register_in(pmt::mp("generate")); + set_msg_handler(pmt::mp("generate"), boost::bind(&random_pdu_impl::generate_pdu, this, _1)); + } + + bool + random_pdu_impl::start() + { + output_random(); + return true; + } + + void + random_pdu_impl::output_random() + { + // pick a random vector length + int len = d_rvar(); + + // fill it with random bytes + std::vector<unsigned char> vec; + for (int i=0; i<len; i++) + vec.push_back((unsigned char) d_bvar()); + + // send the vector + pmt::pmt_t vecpmt(pmt::make_blob(&vec[0], len)); + pmt::pmt_t pdu(pmt::cons(pmt::PMT_NIL, vecpmt)); + + message_port_pub(PDU_PORT_ID, pdu); + } + + } /* namespace blocks */ +}/* namespace gr */ diff --git a/gr-blocks/lib/random_pdu_impl.h b/gr-blocks/lib/random_pdu_impl.h new file mode 100644 index 0000000000..3197f740a0 --- /dev/null +++ b/gr-blocks/lib/random_pdu_impl.h @@ -0,0 +1,54 @@ +/* -*- c++ -*- */ +/* + * Copyright 2013 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. + */ + +#ifndef INCLUDED_BLOCKS_SOCKET_PDU_IMPL_H +#define INCLUDED_BLOCKS_RANDOM_PDU_IMPL_H + +#include <blocks/random_pdu.h> +#include <boost/random.hpp> +#include <boost/generator_iterator.hpp> + +namespace gr { + namespace blocks { + + class random_pdu_impl : public random_pdu + { + private: + boost::mt19937 d_rng; + boost::uniform_int<> d_urange; + boost::uniform_int<> d_brange; + boost::variate_generator< boost::mt19937, boost::uniform_int<> > d_rvar; // pdu length + boost::variate_generator< boost::mt19937, boost::uniform_int<> > d_bvar; // pdu contents + + public: + random_pdu_impl(int min_items, int max_items); + + bool start(); + void output_random(); + void generate_pdu(pmt::pmt_t msg) { output_random(); } + void generate_pdu() { output_random(); } + }; + + } /* namespace blocks */ +} /* namespace gr */ + +#endif /* INCLUDED_BLOCKS_RANDOM_PDU_IMPL_H */ diff --git a/gr-blocks/lib/socket_pdu_impl.cc b/gr-blocks/lib/socket_pdu_impl.cc index 20338f9968..4ff5ce6eb9 100644 --- a/gr-blocks/lib/socket_pdu_impl.cc +++ b/gr-blocks/lib/socket_pdu_impl.cc @@ -27,7 +27,7 @@ #include "socket_pdu_impl.h" #include "tcp_connection.h" #include <gr_io_signature.h> -#include <gr_pdu.h> +#include <blocks/pdu.h> namespace gr { namespace blocks { diff --git a/gr-blocks/lib/tcp_connection.cc b/gr-blocks/lib/tcp_connection.cc index f7af371250..33bb6f10f1 100644 --- a/gr-blocks/lib/tcp_connection.cc +++ b/gr-blocks/lib/tcp_connection.cc @@ -26,9 +26,7 @@ #include "tcp_connection.h" #include <gr_basic_block.h> -#include <gr_pdu.h> -//#include <boost/asio.hpp> -//#include <boost/bind.hpp> +#include <blocks/pdu.h> namespace gr { namespace blocks { |