gr-foo/lib/random_periodic_msg_source_impl.cc

/*
 * Copyright (C) 2013 Bastian Bloessl <bloessl@ccs-labs.org>
 * Copyright (C) 2016 Paul Garver <garverp@gatech.edu>
 *
 * This program 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 of the License, or
 * (at your option) any later version.
 *
 * This program 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 program.  If not, see <http://www.gnu.org/licenses/>.
 */
// Just like periodic_msg_source but random data for every message
#include "random_periodic_msg_source_impl.h"

#include <gnuradio/io_signature.h>
#include <gnuradio/block_detail.h>
#include <string.h>

using namespace gr::foo;

#define dout d_debug && std::cout

random_periodic_msg_source_impl::random_periodic_msg_source_impl(int msg_len,
			long interval, int num_msg, bool quit, bool debug, int seed):
		block("random_periodic_msg_source",
				gr::io_signature::make(0, 0, 0),
				gr::io_signature::make(0, 0, 0)),
		d_msg_len(msg_len),
		d_nmsg_total(num_msg),
		d_nmsg_left(num_msg),
		d_interval(interval),
		d_debug(debug),
		d_finished(false),
		d_quit(quit),
		d_seed(static_cast<unsigned int>(seed)),
		d_brange(0,255),
		d_rng(seed),
		d_randbytes(d_rng,d_brange)
{
	message_port_register_out(pmt::mp("out"));

	d_thread = new boost::thread(boost::bind(&random_periodic_msg_source_impl::run, this, this));
}

random_periodic_msg_source_impl::~random_periodic_msg_source_impl() {
	gr::thread::scoped_lock d_mutex;
	d_finished = true;
	d_thread->interrupt();
	d_thread->join();
	delete d_thread;
}

void
random_periodic_msg_source_impl::run(random_periodic_msg_source_impl *instance) {

	// flow graph startup delay
	boost::this_thread::sleep(boost::posix_time::milliseconds(500));

	while(1) {
		long delay;
		{
			gr::thread::scoped_lock d_mutex;
			if(d_finished || !d_nmsg_left) {
				d_finished = true;
				if(d_quit) {
					boost::this_thread::sleep(boost::posix_time::milliseconds(500));
					post(pmt::mp("system"), pmt::cons(pmt::mp("done"), pmt::from_long(1)));
				}
				break;
			}

			dout << "PMS: number of messages left: " << d_nmsg_left << std::endl;
			// Generate a random message
			pmt::pmt_t msg = generate_msg();
			message_port_pub( pmt::mp("out"), msg);

			if(d_nmsg_left > 0) {
				d_nmsg_left--;
			}

			delay = d_interval;
		}
		boost::this_thread::sleep(boost::posix_time::milliseconds(delay));
	}
}

// Generate a random message
pmt::pmt_t
random_periodic_msg_source_impl::generate_msg() {
	std::vector<uint8_t> vec(d_msg_len);

	for(int i = 0; i < d_msg_len; i++) {
		vec[i] = d_randbytes();
	}

	pmt::pmt_t blob = pmt::make_blob(vec.data(), d_msg_len);
	return pmt::cons(pmt::PMT_NIL, blob);
}
void
random_periodic_msg_source_impl::set_nmsg(int nmsg) {
	gr::thread::scoped_lock d_mutex;
	d_nmsg_total = nmsg;
}

int
random_periodic_msg_source_impl::get_nmsg() {
	gr::thread::scoped_lock d_mutex;
	return d_nmsg_total;
}

void
random_periodic_msg_source_impl::set_delay(long delay) {
	gr::thread::scoped_lock d_mutex;
	d_interval = delay;
}

long
random_periodic_msg_source_impl::get_delay() {
	gr::thread::scoped_lock d_mutex;
	return d_interval;
}


void
random_periodic_msg_source_impl::start_tx() {
	gr::thread::scoped_lock d_mutex;

	if(is_running()) return;

	d_nmsg_left = d_nmsg_total;
	d_finished = false;
	d_thread->join();
	delete d_thread;

	d_thread = new boost::thread(boost::bind(&random_periodic_msg_source_impl::run, this, this));
}

void
random_periodic_msg_source_impl::stop_tx() {
	d_thread->interrupt();
}

bool
random_periodic_msg_source_impl::is_running() {
	return !d_finished;
}

random_periodic_msg_source::sptr
random_periodic_msg_source::make(int msg_len, long interval, int num_msg, 
		bool quit, bool debug,int seed) {
	return gnuradio::get_initial_sptr(new random_periodic_msg_source_impl(msg_len, interval, num_msg, quit, debug, seed));
}