diff options
| author | Ben Reynwar <ben@reynwar.net> | 2013-05-19 02:55:33 -0700 |
|---|---|---|
| committer | Ben Reynwar <ben@reynwar.net> | 2013-05-19 02:55:33 -0700 |
| commit | bb01988e75d50d82cbb44c1a49c86c1d08f05665 (patch) | |
| tree | 0528dac14476d37f2cde7374a8fcb3428f879c69 /gnuradio-runtime/lib/scheduler_tpb.cc | |
| parent | e4f0319eced22c112f7e6a4cc45bc2036d285332 (diff) | |
| parent | 0fa219774dcf9141ae91204f948c029b05673f3f (diff) | |
Merged in next_docs branch.
Diffstat (limited to 'gnuradio-runtime/lib/scheduler_tpb.cc')
| -rw-r--r-- | gnuradio-runtime/lib/scheduler_tpb.cc | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/gnuradio-runtime/lib/scheduler_tpb.cc b/gnuradio-runtime/lib/scheduler_tpb.cc new file mode 100644 index 0000000000..d35c37fb34 --- /dev/null +++ b/gnuradio-runtime/lib/scheduler_tpb.cc @@ -0,0 +1,106 @@ +/* -*- c++ -*- */ +/* + * Copyright 2008,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 this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif + +#include "scheduler_tpb.h" +#include "tpb_thread_body.h" +#include <gnuradio/thread/thread_body_wrapper.h> +#include <sstream> + +namespace gr { + + class tpb_container + { + block_sptr d_block; + int d_max_noutput_items; + + public: + tpb_container(block_sptr block, int max_noutput_items) + : d_block(block), d_max_noutput_items(max_noutput_items) {} + + void operator()() + { + tpb_thread_body body(d_block, d_max_noutput_items); + } + }; + + scheduler_sptr + scheduler_tpb::make(flat_flowgraph_sptr ffg, int max_noutput_items) + { + return scheduler_sptr(new scheduler_tpb(ffg, max_noutput_items)); + } + + scheduler_tpb::scheduler_tpb(flat_flowgraph_sptr ffg, + int max_noutput_items) + : scheduler(ffg, max_noutput_items) + { + // Get a topologically sorted vector of all the blocks in use. + // Being topologically sorted probably isn't going to matter, but + // there's a non-zero chance it might help... + + basic_block_vector_t used_blocks = ffg->calc_used_blocks(); + used_blocks = ffg->topological_sort(used_blocks); + block_vector_t blocks = flat_flowgraph::make_block_vector(used_blocks); + + // Ensure that the done flag is clear on all blocks + + for(size_t i = 0; i < blocks.size(); i++) { + blocks[i]->detail()->set_done(false); + } + + // Fire off a thead for each block + + for(size_t i = 0; i < blocks.size(); i++) { + std::stringstream name; + name << "thread-per-block[" << i << "]: " << blocks[i]; + + // If set, use internal value instead of global value + if(blocks[i]->is_set_max_noutput_items()) + max_noutput_items = blocks[i]->max_noutput_items(); + + d_threads.create_thread( + gr::thread::thread_body_wrapper<tpb_container> + (tpb_container(blocks[i], max_noutput_items), + name.str())); + } + } + + scheduler_tpb::~scheduler_tpb() + { + stop(); + } + + void + scheduler_tpb::stop() + { + d_threads.interrupt_all(); + } + + void + scheduler_tpb::wait() + { + d_threads.join_all(); + } + +} /* namespace gr */ |