libdspl-2.0
Библиотека алгоритмов цифровой обработки сигналов
goertzel.c
1 /*
2 * Copyright (c) 2015-2019 Sergey Bakhurin
3 * Digital Signal Processing Library [http://dsplib.org]
4 *
5 * This file is part of libdspl-2.0.
6 *
7 * is free software: you can redistribute it and/or modify
8 * it under the terms of the GNU Lesser General Public License as published by
9 * the Free Software Foundation, either version 3 of the License, or
10 * (at your option) any later version.
11 *
12 * DSPL 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 Lesser General Public License
18 * along with Foobar. If not, see <http://www.gnu.org/licenses/>.
19 */
20 
21 #include <stdlib.h>
22 #include <string.h>
23 #include "dspl.h"
24 
25 
26 #ifdef DOXYGEN_ENGLISH
27 
74 #endif
75 #ifdef DOXYGEN_RUSSIAN
76 
124 #endif
125 int DSPL_API goertzel(double *x, int n, int *ind, int k, complex_t *y)
126 {
127 
128  int m, p;
129  double wR, wI;
130  double alpha;
131  double v[3];
132 
133  if(!x || !y || !ind)
134  return ERROR_PTR;
135 
136  if(n < 1 || k < 1)
137  return ERROR_SIZE;
138 
139  for(p = 0; p < k; p++)
140  {
141  wR = cos(M_2PI * (double)ind[p] / (double)n);
142  wI = sin(M_2PI * (double)ind[p] / (double)n);
143 
144  alpha = 2.0 * wR;
145  v[0] = v[1] = v[2] = 0.0;
146 
147  for(m = 0; m < n; m++)
148  {
149  v[2] = v[1];
150  v[1] = v[0];
151  v[0] = x[m]+alpha*v[1] - v[2];
152  }
153  RE(y[p]) = wR * v[0] - v[1];
154  IM(y[p]) = wI * v[0];
155  }
156  return RES_OK;
157 }
158 
159 
160 
161 
162 
163 #ifdef DOXYGEN_ENGLISH
164 
212 #endif
213 #ifdef DOXYGEN_RUSSIAN
214 
262 #endif
263 int DSPL_API goertzel_cmplx(complex_t *x, int n, int *ind, int k, complex_t *y)
264 {
265 
266  int m, p;
267  complex_t w;
268  double alpha;
269  complex_t v[3];
270 
271  if(!x || !y || !ind)
272  return ERROR_PTR;
273 
274  if(n < 1 || k < 1)
275  return ERROR_SIZE;
276 
277  for(p = 0; p < k; p++)
278  {
279  RE(w) = cos(M_2PI * (double)ind[p] / (double)n);
280  IM(w) = sin(M_2PI * (double)ind[p] / (double)n);
281 
282  alpha = 2.0 * RE(w);
283  memset(v, 0, 3*sizeof(complex_t));
284 
285  for(m = 0; m < n; m++)
286  {
287  RE(v[2]) = RE(v[1]);
288  RE(v[1]) = RE(v[0]);
289  RE(v[0]) = RE(x[m]) + alpha * RE(v[1]) - RE(v[2]);
290 
291  IM(v[2]) = IM(v[1]);
292  IM(v[1]) = IM(v[0]);
293  IM(v[0]) = IM(x[m]) + alpha * IM(v[1]) - IM(v[2]);
294  }
295 
296  RE(y[p]) = CMRE(w, v[0]) - RE(v[1]);
297  IM(y[p]) = CMIM(w, v[0]) - IM(v[1]);
298  }
299 
300  return RES_OK;
301 }
302 
#define RE(x)
Макрос определяющий реальную часть комплексного числа.
Definition: dspl.h:359
#define ERROR_PTR
Ошибка указателя. Данная ошибка означает, что один из обязательных указателей (память под который дол...
Definition: dspl.h:545
#define ERROR_SIZE
Ошибка при передаче размера массива. Данная ошибка возникает когда помимо указателя на массив входных...
Definition: dspl.h:553
int goertzel(double *x, int n, int *ind, int k, complex_t *y)
Алгоритм Гёрцеля для расчета отдельных спектральных отсчетов дискретного преобразования Фурье веществ...
Definition: goertzel.c:125
double complex_t[2]
Описание комплексного типа данных.
Definition: dspl.h:86
#define RES_OK
Функция завершилась корректно. Ошибки отсутствуют.
Definition: dspl.h:497
int goertzel_cmplx(complex_t *x, int n, int *ind, int k, complex_t *y)
Алгоритм Гёрцеля для расчета отдельных спектральных отсчетов дискретного преобразования Фурье комплек...
Definition: goertzel.c:263
#define IM(x)
Макрос определяющий мнимую часть комплексного числа.
Definition: dspl.h:417