diff options
Diffstat (limited to 'gr-trellis/lib/metrics_impl.cc')
| -rw-r--r-- | gr-trellis/lib/metrics_impl.cc | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/gr-trellis/lib/metrics_impl.cc b/gr-trellis/lib/metrics_impl.cc new file mode 100644 index 0000000000..cf0697c521 --- /dev/null +++ b/gr-trellis/lib/metrics_impl.cc @@ -0,0 +1,135 @@ +/* -*- c++ -*- */ +/* + * Copyright 2004,2010,2012,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 "metrics_impl.h" +#include <gnuradio/io_signature.h> +#include <assert.h> +#include <stdexcept> +#include <iostream> + +namespace gr { + namespace trellis { + + template <class T> + typename metrics<T>::sptr + metrics<T>::make(int O, int D, const std::vector<T> &TABLE, + digital::trellis_metric_type_t TYPE) + { + return gnuradio::get_initial_sptr + (new metrics_impl<T>(O,D,TABLE,TYPE)); + } + + template <class T> + metrics_impl<T>::metrics_impl(int O, int D, const std::vector<T> &TABLE, + digital::trellis_metric_type_t TYPE) + : block("metrics", + io_signature::make(1, -1, sizeof (T)), + io_signature::make(1, -1, sizeof (float))), + d_O(O), d_D(D), d_TYPE(TYPE), d_TABLE(TABLE) + { + this->set_relative_rate (1.0 * d_O / ((double) d_D)); + this->set_output_multiple ((int)d_O); + } + + template <class T> + void metrics_impl<T>::set_O(int O) + { + gr::thread::scoped_lock guard(this->d_setlock); + d_O = O; + this->set_relative_rate (1.0 * d_O / ((double) d_D)); + this->set_output_multiple ((int)d_O); + } + + template <class T> + void metrics_impl<T>::set_D(int D) + { + gr::thread::scoped_lock guard(this->d_setlock); + d_D = D; + this->set_relative_rate (1.0 * d_O / ((double) d_D)); + } + + template <class T> + void + metrics_impl<T>::set_TYPE(digital::trellis_metric_type_t type) + { + gr::thread::scoped_lock guard(this->d_setlock); + d_TYPE = type; + } + + template <class T> + void + metrics_impl<T>::set_TABLE(const std::vector<T> &table) + { + gr::thread::scoped_lock guard(this->d_setlock); + d_TABLE = table; + } + + template <class T> + metrics_impl<T>::~metrics_impl() + { + } + + + template <class T> + void + metrics_impl<T>::forecast(int noutput_items, gr_vector_int &ninput_items_required) + { + int input_required = d_D * noutput_items / d_O; + unsigned ninputs = ninput_items_required.size(); + for(unsigned int i = 0; i < ninputs; i++) + ninput_items_required[i] = input_required; + } + + template <class T> + int + metrics_impl<T>::general_work(int noutput_items, + gr_vector_int &ninput_items, + gr_vector_const_void_star &input_items, + gr_vector_void_star &output_items) + { + gr::thread::scoped_lock guard(this->d_setlock); + int nstreams = input_items.size(); + + for(int m = 0; m < nstreams; m++) { + const T *in = (T*)input_items[m]; + float *out = (float*)output_items[m]; + + for(int i = 0; i < noutput_items / d_O ; i++) { + calc_metric(d_O, d_D, d_TABLE,&(in[i*d_D]), &(out[i*d_O]), d_TYPE); + } + } + + this->consume_each(d_D * noutput_items / d_O); + return noutput_items; + } + +template class metrics<std::int16_t>; +template class metrics<std::int32_t>; +template class metrics<float>; +template class metrics<gr_complex>; + } /* namespace trellis */ +} /* namespace gr */ |