libdspl-2.0
dft.c
1 /*
2 * Copyright (c) 2015-2018 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 
22 #include <stdlib.h>
23 #include <math.h>
24 #include "dspl.h"
25 
26 
27 
28 /******************************************************************************
29 Real vector DFT
30 *******************************************************************************/
31 int DSPL_API dft(double* x, int n, complex_t *y)
32 {
33  int k;
34  int m;
35  double divn;
36  double phi;
37 
38 
39  if(!x || !y)
40  return ERROR_PTR;
41 
42  if(n<1)
43  return ERROR_SIZE;
44 
45  divn = 1.0 / (double)n;
46 
47  for(k = 0; k < n; k++)
48  {
49  RE(y[k]) = IM(y[k]) = 0.0;
50  for(m = 0; m < n; m++)
51  {
52  phi = -M_2PI * divn * (double)k * (double)m;
53  RE(y[k]) += x[m] * cos(phi);
54  IM(y[k]) += x[m] * sin(phi);
55  }
56  }
57  return RES_OK;
58 }
59 
60 
61 
62 /******************************************************************************
63 Complex vector DFT
64 *******************************************************************************/
65 int DSPL_API dft_cmplx(complex_t* x, int n, complex_t *y)
66 {
67  int k;
68  int m;
69  double divn;
70  double phi;
71  complex_t e;
72 
73  if(!x || !y)
74  return ERROR_PTR;
75 
76  if(n<1)
77  return ERROR_SIZE;
78 
79  divn = 1.0 / (double)n;
80 
81  for(k = 0; k < n; k++)
82  {
83  RE(y[k]) = IM(y[k]) = 0.0;
84  for(m = 0; m < n; m++)
85  {
86  phi = -M_2PI * divn * (double)k * (double)m;
87  RE(e) = cos(phi);
88  IM(e) = sin(phi);
89  RE(y[k]) += CMRE(x[m], e);
90  IM(y[k]) += CMIM(x[m], e);
91  }
92  }
93  return RES_OK;
94 }
95 
96 
97 
98 
99 
100 /******************************************************************************
101 Complex vector DFT
102 *******************************************************************************/
103 int DSPL_API idft_cmplx(complex_t* x, int n, complex_t *y)
104 {
105  int k;
106  int m;
107  double divn;
108  double phi;
109  complex_t e;
110 
111  if(!x || !y)
112  return ERROR_PTR;
113 
114  if(n<1)
115  return ERROR_SIZE;
116 
117  divn = 1.0 / (double)n;
118 
119  for(k = 0; k < n; k++)
120  {
121  RE(y[k]) = IM(y[k]) = 0.0;
122  for(m = 0; m < n; m++)
123  {
124  phi = M_2PI * divn * (double)k * (double)m;
125  RE(e) = cos(phi);
126  IM(e) = sin(phi);
127  RE(y[k]) += CMRE(x[m], e);
128  IM(y[k]) += CMIM(x[m], e);
129  }
130  RE(y[k]) /= (double)n;
131  IM(y[k]) /= (double)n;
132  }
133  return RES_OK;
134 }
135 
#define ERROR_SIZE
Ошибка при передаче размера массива. Данная ошибка возникает когда помимо указателя на массив входных...
Definition: dspl.h:124
double complex_t[2]
Описание комплексного типа данных.
Definition: dspl.h:41
#define ERROR_PTR
Ошибка указателя. Данная ошибка означает, что один из обязательных указателей (память под который дол...
Definition: dspl.h:117
#define IM(x)
Макрос определяющий мнимую часть комплексного числа.
Definition: dspl.h:66
#define RE(x)
Макрос определяющий реальную часть комплексного числа.
Definition: dspl.h:65
int dft_cmplx(complex_t *x, int n, complex_t *y)
Дискретное преобразование Фурье комплексного сигнала.
Definition: dft.c:65
int dft(double *x, int n, complex_t *y)
Дискретное преобразование Фурье вещественного сигнала.
Definition: dft.c:31
#define RES_OK
Функция завершилась корректно. Ошибки отсутствуют.
Definition: dspl.h:81