diff options
Diffstat (limited to 'gr-trellis/lib/viterbi_X_impl.cc.t')
-rw-r--r-- | gr-trellis/lib/viterbi_X_impl.cc.t | 99 |
1 files changed, 99 insertions, 0 deletions
diff --git a/gr-trellis/lib/viterbi_X_impl.cc.t b/gr-trellis/lib/viterbi_X_impl.cc.t new file mode 100644 index 0000000000..5097cc2053 --- /dev/null +++ b/gr-trellis/lib/viterbi_X_impl.cc.t @@ -0,0 +1,99 @@ +/* -*- c++ -*- */ +/* + * Copyright 2004,2010,2012 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. + */ + +// @WARNING@ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif + +#include "@NAME@.h" +#include <gr_io_signature.h> +#include <iostream> + +namespace gr { + namespace trellis { + + static const float INF = 1.0e9; + + @BASE_NAME@::sptr + @BASE_NAME@::make(const fsm &FSM, int K, + int S0, int SK) + { + return gnuradio::get_initial_sptr + (new @IMPL_NAME@(FSM, K, S0, SK)); + } + + @IMPL_NAME@::@IMPL_NAME@(const fsm &FSM, int K, + int S0, int SK) + : gr_block("@BASE_NAME@", + gr_make_io_signature(1, -1, sizeof(float)), + gr_make_io_signature(1, -1, sizeof(@TYPE@))), + d_FSM(FSM), d_K(K), d_S0(S0), d_SK(SK)//, + //d_trace(FSM.S()*K) + { + set_relative_rate(1.0 / ((double)d_FSM.O())); + set_output_multiple(d_K); + } + + @IMPL_NAME@::~@IMPL_NAME@() + { + } + + void + @IMPL_NAME@::forecast(int noutput_items, + gr_vector_int &ninput_items_required) + { + int input_required = d_FSM.O() * noutput_items; + unsigned ninputs = ninput_items_required.size(); + for(unsigned int i = 0; i < ninputs; i++) { + ninput_items_required[i] = input_required; + } + } + + int + @IMPL_NAME@::general_work(int noutput_items, + gr_vector_int &ninput_items, + gr_vector_const_void_star &input_items, + gr_vector_void_star &output_items) + { + int nstreams = input_items.size(); + int nblocks = noutput_items / d_K; + + for(int m = 0; m < nstreams; m++) { + const float *in = (const float*)input_items[m]; + @TYPE@ *out = (@TYPE@*)output_items[m]; + + for(int n = 0; n < nblocks; n++) { + viterbi_algorithm(d_FSM.I(), d_FSM.S(), d_FSM.O(), + d_FSM.NS(), d_FSM.OS(), d_FSM.PS(), + d_FSM.PI(), d_K,d_S0, d_SK, + &(in[n*d_K*d_FSM.O()]), &(out[n*d_K])); + } + } + + consume_each(d_FSM.O() * noutput_items); + return noutput_items; + } + + } /* namespace trellis */ +} /* namespace gr */ |