Imported Upstream version 3.0
[debian/gnuradio] / gnuradio-core / src / lib / general / gr_vector_source_f.cc
1 /* -*- c++ -*- */
2 /*
3  * Copyright 2004 Free Software Foundation, Inc.
4  * 
5  * This file is part of GNU Radio
6  * 
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 2, or (at your option)
10  * any later version.
11  * 
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.
16  * 
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.
21  */
22
23 // WARNING: this file is machine generated.  Edits will be over written
24
25 #ifdef HAVE_CONFIG_H
26 #include <config.h>
27 #endif
28 #include <gr_vector_source_f.h>
29 #include <algorithm>
30 #include <gr_io_signature.h>
31
32
33 gr_vector_source_f::gr_vector_source_f (const std::vector<float> &data, bool repeat)
34   : gr_sync_block ("vector_source_f",
35                gr_make_io_signature (0, 0, 0),
36                gr_make_io_signature (1, 1, sizeof (float))),
37     d_data (data),
38     d_repeat (repeat),
39     d_offset (0)
40 {
41 }
42
43 int
44 gr_vector_source_f::work (int noutput_items,
45                     gr_vector_const_void_star &input_items,
46                     gr_vector_void_star &output_items)
47 {
48   float *optr = (float *) output_items[0];
49
50   if (d_repeat){
51     unsigned int size = d_data.size ();
52     unsigned int offset = d_offset;
53     
54     if (size == 0)
55       return -1;
56     
57     for (int i = 0; i < noutput_items; i++){
58       optr[i] = d_data[offset++];
59       if (offset >= size)
60         offset = 0;
61     }
62     d_offset = offset;
63     return noutput_items;
64   }
65
66   else {
67     if (d_offset >= d_data.size ())
68       return -1;                        // Done!
69
70     unsigned n = std::min ((unsigned) d_data.size () - d_offset,
71                            (unsigned) noutput_items);
72     for (unsigned i = 0; i < n; i++)
73       optr[i] = d_data[d_offset + i];
74
75     d_offset += n;
76     return n;
77   }
78 }
79
80 gr_vector_source_f_sptr
81 gr_make_vector_source_f (const std::vector<float> &data, bool repeat)
82 {
83   return gr_vector_source_f_sptr (new gr_vector_source_f (data, repeat));
84 }
85