Lucas Rooyakkers
154 lines
3.8 KiB
C++
154 lines
3.8 KiB
C++
/*
|
|
* Copyright (C) 2013 Bastian Bloessl <bloessl@ccs-labs.org>
|
|
*
|
|
* 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/>.
|
|
*/
|
|
#include "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
|
|
|
|
periodic_msg_source_impl::periodic_msg_source_impl(pmt::pmt_t msg,
|
|
long interval, int num_msg, bool quit, bool debug) :
|
|
block("periodic_msg_source",
|
|
gr::io_signature::make(0, 0, 0),
|
|
gr::io_signature::make(0, 0, 0)),
|
|
d_msg(msg),
|
|
d_nmsg_total(num_msg),
|
|
d_nmsg_left(num_msg),
|
|
d_interval(interval),
|
|
d_debug(debug),
|
|
d_finished(false),
|
|
d_quit(quit) {
|
|
|
|
message_port_register_out(pmt::mp("out"));
|
|
|
|
d_thread = new boost::thread(boost::bind(&periodic_msg_source_impl::run, this, this));
|
|
}
|
|
|
|
periodic_msg_source_impl::~periodic_msg_source_impl() {
|
|
gr::thread::scoped_lock(d_mutex);
|
|
|
|
d_finished = true;
|
|
d_thread->interrupt();
|
|
d_thread->join();
|
|
delete d_thread;
|
|
}
|
|
|
|
void
|
|
periodic_msg_source_impl::run(periodic_msg_source_impl *instance) {
|
|
|
|
try {
|
|
|
|
// 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));
|
|
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;
|
|
|
|
message_port_pub( pmt::mp("out"), d_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)));
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
periodic_msg_source_impl::set_nmsg(int nmsg) {
|
|
gr::thread::scoped_lock(d_mutex);
|
|
d_nmsg_total = nmsg;
|
|
}
|
|
|
|
int
|
|
periodic_msg_source_impl::get_nmsg() {
|
|
gr::thread::scoped_lock(d_mutex);
|
|
return d_nmsg_total;
|
|
}
|
|
|
|
void
|
|
periodic_msg_source_impl::set_delay(long delay) {
|
|
gr::thread::scoped_lock(d_mutex);
|
|
d_interval = delay;
|
|
}
|
|
|
|
long
|
|
periodic_msg_source_impl::get_delay() {
|
|
gr::thread::scoped_lock(d_mutex);
|
|
return d_interval;
|
|
}
|
|
|
|
|
|
void
|
|
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(&periodic_msg_source_impl::run, this, this));
|
|
}
|
|
|
|
void
|
|
periodic_msg_source_impl::stop_tx() {
|
|
d_thread->interrupt();
|
|
}
|
|
|
|
bool
|
|
periodic_msg_source_impl::is_running() {
|
|
return !d_finished;
|
|
}
|
|
|
|
periodic_msg_source::sptr
|
|
periodic_msg_source::make(pmt::pmt_t msg, long interval, int num_msg, bool quit, bool debug) {
|
|
return gnuradio::get_initial_sptr(new periodic_msg_source_impl(msg, interval, num_msg, quit, debug));
|
|
}
|