diff options
Diffstat (limited to 'src/tab_fft.c')
-rw-r--r-- | src/tab_fft.c | 81 |
1 files changed, 42 insertions, 39 deletions
diff --git a/src/tab_fft.c b/src/tab_fft.c index 372e68c..a8107d4 100644 --- a/src/tab_fft.c +++ b/src/tab_fft.c @@ -1,7 +1,7 @@ /* For information on usage and redistribution, and for a DISCLAIMER OF ALL * WARRANTIES, see the file, "LICENSE.txt," in this distribution. -iem_tab written by Thomas Musil, Copyright (c) IEM KUG Graz Austria 2000 - 2006 */ +iem_tab written by Thomas Musil, Copyright (c) IEM KUG Graz Austria 2000 - 2009 */ #include "m_pd.h" #include "iemlib.h" @@ -9,6 +9,7 @@ iem_tab written by Thomas Musil, Copyright (c) IEM KUG Graz Austria 2000 - 2006 #include <math.h> /* -------------------------- tab_fft ------------------------------ */ +/* complex FFT */ typedef struct _tab_fft { @@ -22,10 +23,10 @@ typedef struct _tab_fft int x_offset_dst_re; int x_offset_dst_im; int x_fftsize; - t_float *x_beg_mem_src_re; - t_float *x_beg_mem_src_im; - t_float *x_beg_mem_dst_re; - t_float *x_beg_mem_dst_im; + iemarray_t *x_beg_mem_src_re; + iemarray_t *x_beg_mem_src_im; + iemarray_t *x_beg_mem_dst_re; + iemarray_t *x_beg_mem_dst_im; TAB_COMPLEX *x_sin_cos; t_symbol *x_sym_src_re; t_symbol *x_sym_src_im; @@ -100,7 +101,7 @@ static void tab_fft_bang(t_tab_fft *x) int fs2 = fftsize / 2; TAB_COMPLEX w; TAB_COMPLEX *sincos = x->x_sin_cos; - t_float *vec_src_re, *vec_src_im, *vec_dst_re, *vec_dst_im; + iemarray_t *vec_src_re, *vec_src_im, *vec_dst_re, *vec_dst_im; t_float old1_re, old1_im, old2_re, old2_im; ok_src_re = iem_tab_check_arrays(gensym("tab_fft"), x->x_sym_src_re, &x->x_beg_mem_src_re, &x->x_size_src_re, fftsize); @@ -118,8 +119,8 @@ static void tab_fft_bang(t_tab_fft *x) for(k=0; k<fftsize; k++) { - vec_dst_re[k] = vec_src_re[k]; - vec_dst_im[k] = vec_src_im[k]; + iemarray_setfloat(vec_dst_re, k, iemarray_getfloat(vec_src_re, k)); + iemarray_setfloat(vec_dst_im, k, iemarray_getfloat(vec_src_im, k)); } i_inc = fs2; @@ -132,15 +133,16 @@ static void tab_fft_bang(t_tab_fft *x) w_index = 0; for(k=0; k<i_inc; k++) { - old1_re = vec_dst_re[v_index]; - old1_im = vec_dst_im[v_index]; - old2_re = vec_dst_re[v_index+i_inc]; - old2_im = vec_dst_im[v_index+i_inc]; + old1_re = iemarray_getfloat(vec_dst_re, v_index); + old1_im = iemarray_getfloat(vec_dst_im, v_index); + old2_re = iemarray_getfloat(vec_dst_re, v_index+i_inc); + old2_im = iemarray_getfloat(vec_dst_im, v_index+i_inc); + w = sincos[w_index]; - vec_dst_re[v_index+i_inc] = (old1_re - old2_re)*w.real - (old1_im - old2_im)*w.imag; - vec_dst_im[v_index+i_inc] = (old1_im - old2_im)*w.real + (old1_re - old2_re)*w.imag; - vec_dst_re[v_index] = old1_re + old2_re; - vec_dst_im[v_index] = old1_im + old2_im; + iemarray_setfloat(vec_dst_re, v_index+i_inc, (old1_re - old2_re)*w.real - (old1_im - old2_im)*w.imag); + iemarray_setfloat(vec_dst_im, v_index+i_inc, (old1_im - old2_im)*w.real + (old1_re - old2_re)*w.imag); + iemarray_setfloat(vec_dst_re, v_index, old1_re + old2_re); + iemarray_setfloat(vec_dst_im, v_index, old1_im + old2_im); w_index += w_inc; v_index++; } @@ -162,12 +164,12 @@ static void tab_fft_bang(t_tab_fft *x) j = j + k; if(i < j) { - old1_re = vec_dst_re[j]; - old1_im = vec_dst_im[j]; - vec_dst_re[j] = vec_dst_re[i]; - vec_dst_im[j] = vec_dst_im[i]; - vec_dst_re[i] = old1_re; - vec_dst_im[i] = old1_im; + old1_re = iemarray_getfloat(vec_dst_re, j); + old1_im = iemarray_getfloat(vec_dst_im, j); + iemarray_setfloat(vec_dst_re, j, iemarray_getfloat(vec_dst_re, i)); + iemarray_setfloat(vec_dst_im, j, iemarray_getfloat(vec_dst_im, i)); + iemarray_setfloat(vec_dst_re, i, old1_re); + iemarray_setfloat(vec_dst_im, i, old1_im); } } @@ -204,7 +206,7 @@ static void tab_fft_list(t_tab_fft *x, t_symbol *s, int argc, t_atom *argv) int fs2 = fftsize / 2; TAB_COMPLEX w; TAB_COMPLEX *sincos = x->x_sin_cos; - t_float *vec_src_re, *vec_src_im, *vec_dst_re, *vec_dst_im; + iemarray_t *vec_src_re, *vec_src_im, *vec_dst_re, *vec_dst_im; t_float old1_re, old1_im, old2_re, old2_im; if((argc >= 4) && @@ -242,8 +244,8 @@ static void tab_fft_list(t_tab_fft *x, t_symbol *s, int argc, t_atom *argv) for(k=0; k<fftsize; k++) { - vec_dst_re[k] = vec_src_re[k]; - vec_dst_im[k] = vec_src_im[k]; + iemarray_setfloat(vec_dst_re, k, iemarray_getfloat(vec_src_re, k)); + iemarray_setfloat(vec_dst_im, k, iemarray_getfloat(vec_src_im, k)); } i_inc = fs2; @@ -256,15 +258,16 @@ static void tab_fft_list(t_tab_fft *x, t_symbol *s, int argc, t_atom *argv) w_index = 0; for(k=0; k<i_inc; k++) { - old1_re = vec_dst_re[v_index]; - old1_im = vec_dst_im[v_index]; - old2_re = vec_dst_re[v_index+i_inc]; - old2_im = vec_dst_im[v_index+i_inc]; + old1_re = iemarray_getfloat(vec_dst_re, v_index); + old1_im = iemarray_getfloat(vec_dst_im, v_index); + old2_re = iemarray_getfloat(vec_dst_re, v_index+i_inc); + old2_im = iemarray_getfloat(vec_dst_im, v_index+i_inc); + w = sincos[w_index]; - vec_dst_re[v_index+i_inc] = (old1_re - old2_re)*w.real - (old1_im - old2_im)*w.imag; - vec_dst_im[v_index+i_inc] = (old1_im - old2_im)*w.real + (old1_re - old2_re)*w.imag; - vec_dst_re[v_index] = old1_re + old2_re; - vec_dst_im[v_index] = old1_im + old2_im; + iemarray_setfloat(vec_dst_re, v_index+i_inc, (old1_re - old2_re)*w.real - (old1_im - old2_im)*w.imag); + iemarray_setfloat(vec_dst_im, v_index+i_inc, (old1_im - old2_im)*w.real + (old1_re - old2_re)*w.imag); + iemarray_setfloat(vec_dst_re, v_index, old1_re + old2_re); + iemarray_setfloat(vec_dst_im, v_index, old1_im + old2_im); w_index += w_inc; v_index++; } @@ -286,12 +289,12 @@ static void tab_fft_list(t_tab_fft *x, t_symbol *s, int argc, t_atom *argv) j = j + k; if(i < j) { - old1_re = vec_dst_re[j]; - old1_im = vec_dst_im[j]; - vec_dst_re[j] = vec_dst_re[i]; - vec_dst_im[j] = vec_dst_im[i]; - vec_dst_re[i] = old1_re; - vec_dst_im[i] = old1_im; + old1_re = iemarray_getfloat(vec_dst_re, j); + old1_im = iemarray_getfloat(vec_dst_im, j); + iemarray_setfloat(vec_dst_re, j, iemarray_getfloat(vec_dst_re, i)); + iemarray_setfloat(vec_dst_im, j, iemarray_getfloat(vec_dst_im, i)); + iemarray_setfloat(vec_dst_re, i, old1_re); + iemarray_setfloat(vec_dst_im, i, old1_im); } } |