1 /* -*- c++ -*- */
2 /*
3 * Copyright 2004,2009,2010,2012,2018 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
24 #ifdef HAVE_CONFIG_H
25 #include "config.h"
26 #endif
27
28 #include "multiply_impl.h"
29 #include <gnuradio/io_signature.h>
30 #include <volk/volk.h>
31
32 namespace gr {
33 namespace blocks {
34
35 template <class T>
make(size_t vlen)36 typename multiply<T>::sptr multiply<T>::make(size_t vlen)
37 {
38 return gnuradio::get_initial_sptr(new multiply_impl<T>(vlen));
39 }
40
41 template <>
multiply_impl(size_t vlen)42 multiply_impl<float>::multiply_impl(size_t vlen)
43 : sync_block("multiply_ff",
44 io_signature::make(1, -1, sizeof(float) * vlen),
45 io_signature::make(1, 1, sizeof(float) * vlen)),
46 d_vlen(vlen)
47 {
48 const int alignment_multiple = volk_get_alignment() / sizeof(float);
49 set_alignment(std::max(1, alignment_multiple));
50 }
51
52 template <>
work(int noutput_items,gr_vector_const_void_star & input_items,gr_vector_void_star & output_items)53 int multiply_impl<float>::work(int noutput_items,
54 gr_vector_const_void_star& input_items,
55 gr_vector_void_star& output_items)
56 {
57 float* out = (float*)output_items[0];
58 int noi = d_vlen * noutput_items;
59
60 memcpy(out, input_items[0], noi * sizeof(float));
61 for (size_t i = 1; i < input_items.size(); i++)
62 volk_32f_x2_multiply_32f(out, out, (float*)input_items[i], noi);
63
64 return noutput_items;
65 }
66
67 template <>
multiply_impl(size_t vlen)68 multiply_impl<gr_complex>::multiply_impl(size_t vlen)
69 : sync_block("multiply_cc",
70 io_signature::make(1, -1, sizeof(gr_complex) * vlen),
71 io_signature::make(1, 1, sizeof(gr_complex) * vlen)),
72 d_vlen(vlen)
73 {
74 const int alignment_multiple = volk_get_alignment() / sizeof(gr_complex);
75 set_alignment(std::max(1, alignment_multiple));
76 }
77
78 template <>
work(int noutput_items,gr_vector_const_void_star & input_items,gr_vector_void_star & output_items)79 int multiply_impl<gr_complex>::work(int noutput_items,
80 gr_vector_const_void_star& input_items,
81 gr_vector_void_star& output_items)
82 {
83 gr_complex* out = (gr_complex*)output_items[0];
84 int noi = d_vlen * noutput_items;
85
86 memcpy(out, input_items[0], noi * sizeof(gr_complex));
87 for (size_t i = 1; i < input_items.size(); i++)
88 volk_32fc_x2_multiply_32fc(out, out, (gr_complex*)input_items[i], noi);
89
90 return noutput_items;
91 }
92
93 template <class T>
multiply_impl(size_t vlen)94 multiply_impl<T>::multiply_impl(size_t vlen)
95 : sync_block("multiply",
96 io_signature::make(1, -1, sizeof(T) * vlen),
97 io_signature::make(1, 1, sizeof(T) * vlen)),
98 d_vlen(vlen)
99 {
100 }
101
102 template <class T>
work(int noutput_items,gr_vector_const_void_star & input_items,gr_vector_void_star & output_items)103 int multiply_impl<T>::work(int noutput_items,
104 gr_vector_const_void_star& input_items,
105 gr_vector_void_star& output_items)
106 {
107 T* optr = (T*)output_items[0];
108
109 int ninputs = input_items.size();
110
111 for (size_t i = 0; i < noutput_items * d_vlen; i++) {
112 T acc = ((T*)input_items[0])[i];
113 for (int j = 1; j < ninputs; j++)
114 acc *= ((T*)input_items[j])[i];
115
116 *optr++ = (T)acc;
117 }
118
119 return noutput_items;
120 }
121
122 template class multiply<std::int16_t>;
123 template class multiply<std::int32_t>;
124 template class multiply<gr_complex>;
125 template class multiply<float>;
126 } /* namespace blocks */
127 } /* namespace gr */
128