diff options
Diffstat (limited to 'src/tab_rifft.c')
-rw-r--r-- | src/tab_rifft.c | 106 |
1 files changed, 56 insertions, 50 deletions
diff --git a/src/tab_rifft.c b/src/tab_rifft.c index 9b717a1..0730ef1 100644 --- a/src/tab_rifft.c +++ b/src/tab_rifft.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" @@ -10,6 +10,7 @@ iem_tab written by Thomas Musil, Copyright (c) IEM KUG Graz Austria 2000 - 2006 /* -------------------------- tab_rifft ------------------------------ */ +/* complex spectral domain inverse FFT to real time domain */ typedef struct _tab_rifft { @@ -21,10 +22,10 @@ typedef struct _tab_rifft int x_offset_src_im; int x_offset_dst; 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; @@ -94,7 +95,7 @@ static void tab_rifft_bang(t_tab_rifft *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, g; ok_src_re = iem_tab_check_arrays(gensym("tab_rifft"), x->x_sym_src_re, &x->x_beg_mem_src_re, &x->x_size_src_re, fftsize); @@ -110,17 +111,20 @@ static void tab_rifft_bang(t_tab_rifft *x) vec_dst_re=x->x_beg_mem_dst_re; vec_dst_im=x->x_beg_mem_dst_im; - for(j=0; j<fs2; j++) + iemarray_setfloat(vec_dst_re, 0, iemarray_getfloat(vec_src_re, 0)); + iemarray_setfloat(vec_dst_im, 0, 0.0f); + for(j=1; j<fs2; j++) { - vec_dst_re[j] = vec_src_re[j]; - vec_dst_im[j] = vec_src_im[j]; + iemarray_setfloat(vec_dst_re, j, iemarray_getfloat(vec_src_re, j)); + iemarray_setfloat(vec_dst_im, j, iemarray_getfloat(vec_src_im, j)); } - vec_dst_re[fs2] = vec_src_re[fs2]; - vec_dst_im[fs2] = 0.0f; + iemarray_setfloat(vec_dst_re, fs2, iemarray_getfloat(vec_src_re, fs2)); + iemarray_setfloat(vec_dst_im, fs2, 0.0f); + for(k=1, j=fftsize-1; k<fs2; k++, j--) { - vec_dst_re[j] = vec_src_re[k]; - vec_dst_im[j] = -vec_src_im[k]; + iemarray_setfloat(vec_dst_re, j, iemarray_getfloat(vec_dst_re, k)); + iemarray_setfloat(vec_dst_im, j, -iemarray_getfloat(vec_dst_im, k)); } i_inc = fs2; @@ -133,15 +137,16 @@ static void tab_rifft_bang(t_tab_rifft *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++; } @@ -163,12 +168,12 @@ static void tab_rifft_bang(t_tab_rifft *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); } } @@ -188,7 +193,7 @@ static void tab_rifft_bang(t_tab_rifft *x) g = 1.0f / (t_float)fftsize; for(i = 0; i < fftsize; i++) { - vec_dst_re[i] *= g; + iemarray_setfloat(vec_dst_re, i, iemarray_getfloat(vec_dst_re, i)*g); } outlet_bang(x->x_obj.ob_outlet); @@ -208,7 +213,7 @@ static void tab_rifft_list(t_tab_rifft *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, g; if((argc >= 3) && @@ -241,15 +246,15 @@ static void tab_rifft_list(t_tab_rifft *x, t_symbol *s, int argc, t_atom *argv) for(j=0; j<fs2; j++) { - vec_dst_re[j] = vec_src_re[j]; - vec_dst_im[j] = vec_src_im[j]; + iemarray_setfloat(vec_dst_re, j, iemarray_getfloat(vec_src_re, j)); + iemarray_setfloat(vec_dst_im, j, iemarray_getfloat(vec_src_im, j)); } - vec_dst_re[fs2] = vec_src_re[fs2]; - vec_dst_im[fs2] = 0.0f; + iemarray_setfloat(vec_dst_re, fs2, iemarray_getfloat(vec_src_re, fs2)); + iemarray_setfloat(vec_dst_im, fs2, 0.0f); for(k=1, j=fftsize-1; k<fs2; k++, j--) { - vec_dst_re[j] = vec_src_re[k]; - vec_dst_im[j] = -vec_src_im[k]; + iemarray_setfloat(vec_dst_re, j, iemarray_getfloat(vec_src_re, k)); + iemarray_setfloat(vec_dst_im, j, -iemarray_getfloat(vec_src_im, k)); } i_inc = fs2; @@ -262,15 +267,16 @@ static void tab_rifft_list(t_tab_rifft *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++; } @@ -292,12 +298,12 @@ static void tab_rifft_list(t_tab_rifft *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); } } @@ -317,7 +323,7 @@ static void tab_rifft_list(t_tab_rifft *x, t_symbol *s, int argc, t_atom *argv) g = 1.0f / (t_float)fftsize; for(i = 0; i < fftsize; i++) { - vec_dst_re[i] *= g; + iemarray_setfloat(vec_dst_re, i, iemarray_getfloat(vec_dst_re, i)*g); } outlet_bang(x->x_obj.ob_outlet); |