forked from ITA-Solar/rh
-
Notifications
You must be signed in to change notification settings - Fork 0
/
complex.h
52 lines (34 loc) · 1.54 KB
/
complex.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
/* ------- file: -------------------------- complex.h ---------------
Version: rh2.0
Author: Han Uitenbroek ([email protected])
Last modified: Mon Nov 22 15:35:13 1999 --
-------------------------- ----------RH-- */
#ifndef __COMPLEX_H__
#define __COMPLEX_H__
/* --- Defines structure for complex numbers.
Let z = a + b*i, then the real part is represented as z.r,
the imaginary part as z.i. Initialize as cmplx z = {a, b}. -- */
typedef struct {double r; double i;} complex;
#define FFT_FORWARD -1
#define FFT_REVERSE 1
#define FFT_REAL 0
#define FFT_COMPLEX 1
/* --- Associated function prototypes -- -------------- */
complex **matrix_complex(int Nrow, int Ncol);
/* --- Manipulation of complex numbers -- -------------- */
complex cmplx(double a, double b);
complex cmplx_conj(complex a);
complex cmplx_mult(complex a, complex b);
complex cmplx_sclr(double a, complex b);
complex cmplx_div(complex a, complex b);
complex cmplx_exp(complex a);
complex cmplx_add(complex a, complex b);
complex cmplx_addr(complex a, double b);
complex cmplx_subt(complex a, complex b);
/* --- Fourier transforms -- -------------- */
void four1(int NN, double *data, int sign);
void fourn(int Ndim, int *NN, double *data, int sign);
void fourier_(int *Ndim, int *NN, double *data, int *sign,
int *format, double *work);
#endif /* !__COMPLEX_H__ */
/* ------- end ---------------------------- complex.h --------------- */