1 /* -*- c++ -*- */
2 /*
3 * Copyright 2015 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 3, 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 #ifdef HAVE_CONFIG_H
24 #include "config.h"
25 #endif
26
27 #include "complex_to_magphase_impl.h"
28 #include <gnuradio/io_signature.h>
29 #include <gnuradio/math.h>
30 #include <volk/volk.h>
31
32 namespace gr {
33 namespace blocks {
34
make(size_t vlen)35 complex_to_magphase::sptr complex_to_magphase::make(size_t vlen)
36 {
37 return gnuradio::get_initial_sptr(new complex_to_magphase_impl(vlen));
38 }
39
complex_to_magphase_impl(size_t vlen)40 complex_to_magphase_impl::complex_to_magphase_impl(size_t vlen)
41 : sync_block("complex_to_magphase",
42 io_signature::make(1, 1, sizeof(gr_complex) * vlen),
43 io_signature::make(2, 2, sizeof(float) * vlen)),
44 d_vlen(vlen)
45 {
46 const int alignment_multiple = volk_get_alignment() / sizeof(float);
47 set_alignment(std::max(1, alignment_multiple));
48 }
49
work(int noutput_items,gr_vector_const_void_star & input_items,gr_vector_void_star & output_items)50 int complex_to_magphase_impl::work(int noutput_items,
51 gr_vector_const_void_star& input_items,
52 gr_vector_void_star& output_items)
53 {
54 const gr_complex* in = (const gr_complex*)input_items[0];
55 float* out0 = (float*)output_items[0];
56 float* out1 = (float*)output_items[1];
57 int noi = noutput_items * d_vlen;
58
59 volk_32fc_magnitude_32f_u(out0, in, noi);
60
61 // The fast_atan2f is faster than Volk
62 for (int i = 0; i < noi; i++) {
63 // out[i] = std::arg (in[i]);
64 out1[i] = gr::fast_atan2f(in[i]);
65 }
66
67 return noutput_items;
68 }
69
70 } /* namespace blocks */
71 } /* namespace gr */
72