diff options
| author | Andrej Rode <mail@andrejro.de> | 2018-02-22 13:28:01 +0100 |
|---|---|---|
| committer | Andrej Rode <mail@andrejro.de> | 2018-08-25 20:21:23 +0200 |
| commit | 1043222e0de551f00683a1cc9060f194adc4fc85 (patch) | |
| tree | 663d4157d907f391cbe8934d4c131b981f350ae3 /gr-blocks/lib/vector_source_impl.cc | |
| parent | aa844b050d5b658ab8d6961a3678251b65db7ce7 (diff) | |
blocks: replace gengen with C++ templates
Diffstat (limited to 'gr-blocks/lib/vector_source_impl.cc')
| -rw-r--r-- | gr-blocks/lib/vector_source_impl.cc | 153 |
1 files changed, 153 insertions, 0 deletions
diff --git a/gr-blocks/lib/vector_source_impl.cc b/gr-blocks/lib/vector_source_impl.cc new file mode 100644 index 0000000000..fd1ad16981 --- /dev/null +++ b/gr-blocks/lib/vector_source_impl.cc @@ -0,0 +1,153 @@ +/* -*- c++ -*- */ +/* + * Copyright 2004,2008,2010,2013,2018 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 "vector_source_impl.h" +#include <algorithm> +#include <gnuradio/io_signature.h> +#include <stdexcept> + +namespace gr { + namespace blocks { + + template <class T> + typename vector_source<T>::sptr + vector_source<T>::make(const std::vector<T> &data, + bool repeat, int vlen, + const std::vector<tag_t> &tags) + { + return gnuradio::get_initial_sptr + (new vector_source_impl<T> (data, repeat, vlen, tags)); + } + + template <class T> + vector_source_impl<T> ::vector_source_impl(const std::vector<T> &data, + bool repeat, int vlen, + const std::vector<tag_t> &tags) + : sync_block("vector_source", + io_signature::make(0, 0, 0), + io_signature::make(1, 1, sizeof(T) * vlen)), + d_data(data), + d_repeat(repeat), + d_offset(0), + d_vlen(vlen), + d_tags(tags) + { + if(tags.size() == 0) { + d_settags = 0; + } + else { + d_settags = 1; + this->set_output_multiple(data.size() / vlen); + } + if((data.size() % vlen) != 0) + throw std::invalid_argument("data length must be a multiple of vlen"); + } + + template <class T> + vector_source_impl<T> ::~vector_source_impl() + {} + + template <class T> + void + vector_source_impl<T> ::set_data (const std::vector<T> &data, + const std::vector<tag_t> &tags) + { + d_data = data; + d_tags = tags; + rewind(); + if(tags.size() == 0) { + d_settags = false; + } + else { + d_settags = true; + } + } + + template <class T> + int + vector_source_impl<T> ::work(int noutput_items, + gr_vector_const_void_star &input_items, + gr_vector_void_star &output_items) + { + T *optr = (T *) output_items[0]; + + if(d_repeat) { + unsigned int size = d_data.size (); + unsigned int offset = d_offset; + if(size == 0) + return -1; + + if(d_settags) { + int n_outputitems_per_vector = d_data.size() / d_vlen; + for(int i = 0; i < noutput_items; i += n_outputitems_per_vector) { + // FIXME do proper vector copy + memcpy((void *) optr, (const void*)&d_data[0], size*sizeof (T)); + optr += size; + for(unsigned t = 0; t < d_tags.size(); t++) { + this->add_item_tag(0, this->nitems_written(0)+i+d_tags[t].offset, + d_tags[t].key, d_tags[t].value, d_tags[t].srcid); + } + } + } + else { + for(int i = 0; i < noutput_items*d_vlen; i++) { + optr[i] = d_data[offset++]; + if(offset >= size) { + offset = 0; + } + } + } + + d_offset = offset; + return noutput_items; + } + else { + if(d_offset >= d_data.size ()) + return -1; // Done! + + unsigned n = std::min((unsigned)d_data.size() - d_offset, + (unsigned)noutput_items*d_vlen); + for(unsigned i = 0; i < n; i++) { + optr[i] = d_data[d_offset + i]; + } + for(unsigned t = 0; t < d_tags.size(); t++) { + if((d_tags[t].offset >= d_offset) && (d_tags[t].offset < d_offset+n)) + this->add_item_tag(0, d_tags[t].offset, d_tags[t].key, d_tags[t].value, + d_tags[t].srcid); + } + d_offset += n; + return n/d_vlen; + } + } + +template class vector_source<std::uint8_t>; +template class vector_source<std::int16_t>; +template class vector_source<std::int32_t>; +template class vector_source<float>; +template class vector_source<gr_complex>; + } /* namespace blocks */ +} /* namespace gr */ |