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,
			float 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) {

	try {

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

	while(1) {
		float 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));
					message_port_pub( pmt::mp("out"), pmt::PMT_EOF);
					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));
	} 

	} catch(boost::thread_interrupted) {
		gr::thread::scoped_lock(d_mutex);
		dout << "PMS: thread interrupted" << std::endl;
		d_finished = true;
		if(d_quit) {
			boost::this_thread::sleep(boost::posix_time::milliseconds(500));
			message_port_pub( pmt::mp("out"), pmt::PMT_EOF);
			post(pmt::mp("system"), pmt::cons(pmt::mp("done"), pmt::from_long(1)));
		}
	}
}

// Generate a random message
pmt::pmt_t 
random_periodic_msg_source_impl::generate_msg(){
   // Construct random vector of specified length
   std::vector<unsigned char> vec(d_msg_len);
   // Fill w/random bytes
   for( int i=0; i<d_msg_len; i++ ){
      vec[i] = ((unsigned char) d_randbytes());
   }
   // Construct the PDU
   pmt::pmt_t vecpmt(pmt::make_blob(&vec[0], d_msg_len));
   return pmt::cons(pmt::PMT_NIL, vecpmt);
}
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(float delay) {
	gr::thread::scoped_lock(d_mutex);
	d_interval = delay;
}

float
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, float 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));
}