3 * Copyright 2006,2010 Free Software Foundation, Inc.
5 * This file is part of GNU Radio
7 * GNU Radio is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation; either version 3, or (at your option)
12 * GNU Radio is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with GNU Radio; see the file COPYING. If not, write to
19 * the Free Software Foundation, Inc., 51 Franklin Street,
20 * Boston, MA 02110-1301, USA.
27 #include <gr_stream_mux.h>
28 #include <gr_io_signature.h>
35 gr_make_stream_mux (size_t itemsize, const std::vector<int> &lengths)
37 return gnuradio::get_initial_sptr(new gr_stream_mux (itemsize, lengths));
40 gr_stream_mux::gr_stream_mux (size_t itemsize, const std::vector<int> &lengths)
41 : gr_block ("stream_mux",
42 gr_make_io_signature (1, -1, itemsize),
43 gr_make_io_signature (1, 1, itemsize)),
49 if(d_lengths[d_stream] == 0) {
52 d_residual = d_lengths[d_stream];
55 gr_stream_mux::~gr_stream_mux(void)
60 gr_stream_mux::forecast (int noutput_items, gr_vector_int &ninput_items_required)
62 unsigned ninputs = ninput_items_required.size ();
63 for (unsigned i = 0; i < ninputs; i++)
64 ninput_items_required[i] = (d_lengths[i] == 0 ? 0 : 1);
67 void gr_stream_mux::increment_stream()
70 d_stream = (d_stream+1) % d_lengths.size();
71 } while(d_lengths[d_stream] == 0);
73 d_residual = d_lengths[d_stream];
77 gr_stream_mux::general_work(int noutput_items,
78 gr_vector_int &ninput_items,
79 gr_vector_const_void_star &input_items,
80 gr_vector_void_star &output_items)
82 char *out = (char *) output_items[0];
85 std::vector<int> input_index(d_lengths.size(), 0);
88 printf("mux: nouput_items: %d d_stream: %d\n", noutput_items, d_stream);
89 for(size_t i = 0; i < d_lengths.size(); i++)
90 printf("\tninput_items[%zu]: %d\n", i, ninput_items[i]);
94 int r = std::min(noutput_items - out_index,
96 ninput_items[d_stream] - input_index[d_stream]));
98 printf("mux: r=%d\n", r);
99 printf("\tnoutput_items - out_index: %d\n",
100 noutput_items - out_index);
101 printf("\td_residual: %d\n",
103 printf("\tninput_items[d_stream] - input_index[d_stream]: %d\n",
104 ninput_items[d_stream] - input_index[d_stream]);
111 in = (const char *) input_items[d_stream] + input_index[d_stream]*d_itemsize;
113 memcpy(&out[out_index*d_itemsize], in, r*d_itemsize);
115 input_index[d_stream] += r;
118 consume(d_stream, r);
120 if(d_residual == 0) {