diff options
Diffstat (limited to 'src/tab_fft.c')
| -rw-r--r-- | src/tab_fft.c | 400 |
1 files changed, 400 insertions, 0 deletions
diff --git a/src/tab_fft.c b/src/tab_fft.c new file mode 100644 index 0000000..0948821 --- /dev/null +++ b/src/tab_fft.c @@ -0,0 +1,400 @@ +/* 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 - 2005 */ + +#ifdef NT +#pragma warning( disable : 4244 ) +#pragma warning( disable : 4305 ) +#endif + + +#include "m_pd.h" +#include "iemlib.h" +#include "iem_tab.h" +#include <math.h> +#include <stdio.h> +#include <string.h> + + +/* -------------------------- tab_fft ------------------------------ */ + +typedef struct _tab_fft +{ + t_object x_obj; + int x_size_src_re; + int x_size_src_im; + int x_size_dst_re; + int x_size_dst_im; + int x_offset_src_re; + int x_offset_src_im; + int x_offset_dst_re; + int x_offset_dst_im; + int x_fftsize; + float *x_beg_mem_src_re; + float *x_beg_mem_src_im; + float *x_beg_mem_dst_re; + float *x_beg_mem_dst_im; + TAB_COMPLEX *x_sin_cos; + t_symbol *x_sym_src_re; + t_symbol *x_sym_src_im; + t_symbol *x_sym_dst_re; + t_symbol *x_sym_dst_im; +} t_tab_fft; + +static t_class *tab_fft_class; + +static void tab_fft_init(t_tab_fft *x) +{ + int i, fftsize = x->x_fftsize; + float f, g; + TAB_COMPLEX *sincos = x->x_sin_cos; + + g = 2.0f * 3.1415926538f / (float)fftsize; + for(i=0; i<fftsize; i++) + { + f = g * (float)i; + (*sincos).real = cos(f); + (*sincos).imag = -sin(f);/*FFT*/ + sincos++; + } +} + +static void tab_fft_fftsize(t_tab_fft *x, t_floatarg f) +{ + int i=1, fftsize = (int)f; + + if(fftsize < 8) + fftsize = 8; + + while(i <= fftsize) + i *= 2; + i /= 2; + + if(i != x->x_fftsize) + { + x->x_sin_cos = (TAB_COMPLEX *)resizebytes(x->x_sin_cos, x->x_fftsize*sizeof(TAB_COMPLEX), i*sizeof(TAB_COMPLEX)); + x->x_fftsize = i; + } + tab_fft_init(x); +} + +static void tab_fft_src_re(t_tab_fft *x, t_symbol *s) +{ + x->x_sym_src_re = s; +} + +static void tab_fft_src_im(t_tab_fft *x, t_symbol *s) +{ + x->x_sym_src_im = s; +} + +static void tab_fft_dst_re(t_tab_fft *x, t_symbol *s) +{ + x->x_sym_dst_re = s; +} + +static void tab_fft_dst_im(t_tab_fft *x, t_symbol *s) +{ + x->x_sym_dst_im = s; +} + +static void tab_fft_bang(t_tab_fft *x) +{ + int i, j, k; + int ok_src_re, ok_src_im, ok_dst_re, ok_dst_im; + int w_index, w_inc, i_inc, v_index; + int fftsize = x->x_fftsize; + int fs1 = fftsize - 1; + 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; + 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); + ok_src_im = iem_tab_check_arrays(gensym("tab_fft"), x->x_sym_src_im, &x->x_beg_mem_src_im, &x->x_size_src_im, fftsize); + ok_dst_re = iem_tab_check_arrays(gensym("tab_fft"), x->x_sym_dst_re, &x->x_beg_mem_dst_re, &x->x_size_dst_re, fftsize); + ok_dst_im = iem_tab_check_arrays(gensym("tab_fft"), x->x_sym_dst_im, &x->x_beg_mem_dst_im, &x->x_size_dst_im, fftsize); + if(ok_src_re && ok_src_im && ok_dst_re && ok_dst_im) + { + t_garray *a; + + vec_src_re=x->x_beg_mem_src_re; + vec_src_im=x->x_beg_mem_src_im; + vec_dst_re=x->x_beg_mem_dst_re; + vec_dst_im=x->x_beg_mem_dst_im; + + for(k=0; k<fftsize; k++) + { + vec_dst_re[k] = vec_src_re[k]; + vec_dst_im[k] = vec_src_im[k]; + } + + i_inc = fs2; + w_inc = 1; + for(i=1; i<fftsize; i<<=1) + { + v_index = 0; + for(j=0; j<i; j++) + { + 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]; + 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; + w_index += w_inc; + v_index++; + } + v_index += i_inc; + } + w_inc <<= 1; + i_inc >>= 1; + } + + j = 0; + for(i=1;i<fs1;i++) + { + k = fs2; + while(k <= j) + { + j = j - k; + k >>= 1; + } + 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; + } + } + +// g = 2.0f / (float)fftsize; +/* + ein fehler tritt auf beim 0.sample, hier sollte nur mal 1.0 multipliziert werden + wenn gelten soll : Energie im zeitfenster == Energie im Frequenz-dichte-fenster + + g = 1.0f; + for(i = 0; i < fs2; i++) + { + vec_dst_re[i] *= g; + vec_dst_im[i] *= g; + } + */ + + + outlet_bang(x->x_obj.ob_outlet); + a = (t_garray *)pd_findbyclass(x->x_sym_dst_re, garray_class); + garray_redraw(a); + a = (t_garray *)pd_findbyclass(x->x_sym_dst_im, garray_class); + garray_redraw(a); + } +} + +static void tab_fft_list(t_tab_fft *x, t_symbol *s, int argc, t_atom *argv) +{ + int beg_src_re, beg_src_im, beg_dst_re, beg_dst_im; + int i, j, k; + int ok_src_re, ok_src_im, ok_dst_re, ok_dst_im; + int w_index, w_inc, i_inc, v_index; + int fftsize = x->x_fftsize; + int fs1 = fftsize - 1; + 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; + t_float old1_re, old1_im, old2_re, old2_im; + + if((argc >= 4) && + IS_A_FLOAT(argv,0) && + IS_A_FLOAT(argv,1) && + IS_A_FLOAT(argv,2) && + IS_A_FLOAT(argv,3)) + { + beg_src_re = (int)atom_getintarg(0, argc, argv); + beg_src_im = (int)atom_getintarg(1, argc, argv); + beg_dst_re = (int)atom_getintarg(2, argc, argv); + beg_dst_im = (int)atom_getintarg(3, argc, argv); + if(beg_src_re < 0) + beg_src_re = 0; + if(beg_src_im < 0) + beg_src_im = 0; + if(beg_dst_re < 0) + beg_dst_re = 0; + if(beg_dst_im < 0) + beg_dst_im = 0; + + 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, beg_src_re+fftsize); + ok_src_im = iem_tab_check_arrays(gensym("tab_fft"), x->x_sym_src_im, &x->x_beg_mem_src_im, &x->x_size_src_im, beg_src_im+fftsize); + ok_dst_re = iem_tab_check_arrays(gensym("tab_fft"), x->x_sym_dst_re, &x->x_beg_mem_dst_re, &x->x_size_dst_re, beg_dst_re+fftsize); + ok_dst_im = iem_tab_check_arrays(gensym("tab_fft"), x->x_sym_dst_im, &x->x_beg_mem_dst_im, &x->x_size_dst_im, beg_dst_im+fftsize); + + if(ok_src_re && ok_src_im && ok_dst_re && ok_dst_im) + { + t_garray *a; + + vec_src_re=x->x_beg_mem_src_re + beg_src_re; + vec_src_im=x->x_beg_mem_src_im + beg_src_im; + vec_dst_re=x->x_beg_mem_dst_re + beg_dst_re; + vec_dst_im=x->x_beg_mem_dst_im + beg_dst_im; + + for(k=0; k<fftsize; k++) + { + vec_dst_re[k] = vec_src_re[k]; + vec_dst_im[k] = vec_src_im[k]; + } + + i_inc = fs2; + w_inc = 1; + for(i=1; i<fftsize; i<<=1) + { + v_index = 0; + for(j=0; j<i; j++) + { + 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]; + 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; + w_index += w_inc; + v_index++; + } + v_index += i_inc; + } + w_inc <<= 1; + i_inc >>= 1; + } + + j = 0; + for(i=1;i<fs1;i++) + { + k = fs2; + while(k <= j) + { + j = j - k; + k >>= 1; + } + 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; + } + } + +// g = 2.0f / (float)fftsize; +/* + ein fehler tritt auf beim 0.sample, hier sollte nur mal 1.0 multipliziert werden + wenn gelten soll : Energie im zeitfenster == Energie im Frequenz-dichte-fenster + + g = 1.0f; + for(i = 0; i < fs2; i++) + { + vec_dst_re[i] *= g; + vec_dst_im[i] *= g; + } + */ + + outlet_bang(x->x_obj.ob_outlet); + a = (t_garray *)pd_findbyclass(x->x_sym_dst_re, garray_class); + garray_redraw(a); + a = (t_garray *)pd_findbyclass(x->x_sym_dst_im, garray_class); + garray_redraw(a); + } + } + else + { + post("tab_fft-ERROR: list need 4 float arguments:"); + post(" source_real_offset + source_imag_offset + destination_real_offset + destination_imag_offset"); + } +} + +static void tab_fft_free(t_tab_fft *x) +{ + freebytes(x->x_sin_cos, x->x_fftsize * sizeof(TAB_COMPLEX)); +} + +static void *tab_fft_new(t_symbol *s, int argc, t_atom *argv) +{ + t_tab_fft *x = (t_tab_fft *)pd_new(tab_fft_class); + t_symbol *src_re, *src_im, *dst_re, *dst_im; + int fftsize, i=1; + + if((argc >= 5) && + IS_A_SYMBOL(argv,0) && + IS_A_SYMBOL(argv,1) && + IS_A_SYMBOL(argv,2) && + IS_A_SYMBOL(argv,3) && + IS_A_FLOAT(argv,4)) + { + src_re = (t_symbol *)atom_getsymbolarg(0, argc, argv); + src_im = (t_symbol *)atom_getsymbolarg(1, argc, argv); + dst_re = (t_symbol *)atom_getsymbolarg(2, argc, argv); + dst_im = (t_symbol *)atom_getsymbolarg(3, argc, argv); + fftsize = (int)atom_getintarg(4, argc, argv); + } + else + { + post("tab_fft-ERROR: need 4 symbols + 1 float arguments:"); + post(" source_real_array_name + source_imag_array_name + destination_real_array_name + destination_imag_array_name + FFT-size"); + return(0); + } + + if(fftsize < 8) + fftsize = 8; + + while(i <= fftsize) + i *= 2; + i /= 2; + fftsize = i; + + x->x_fftsize = fftsize; + x->x_sym_src_re = src_re; + x->x_sym_src_im = src_im; + x->x_sym_dst_re = dst_re; + x->x_sym_dst_im = dst_im; + x->x_sin_cos = (TAB_COMPLEX *)getbytes(x->x_fftsize * sizeof(TAB_COMPLEX)); + tab_fft_init(x); + outlet_new(&x->x_obj, &s_bang); + return(x); +} + +void tab_fft_setup(void) +{ + tab_fft_class = class_new(gensym("tab_fft"), (t_newmethod)tab_fft_new, (t_method)tab_fft_free, + sizeof(t_tab_fft), 0, A_GIMME, 0); + class_addbang(tab_fft_class, (t_method)tab_fft_bang); + class_addlist(tab_fft_class, (t_method)tab_fft_list); + class_addmethod(tab_fft_class, (t_method)tab_fft_fftsize, gensym("fftsize"), A_DEFFLOAT, 0); + class_addmethod(tab_fft_class, (t_method)tab_fft_src_re, gensym("src_re"), A_DEFSYMBOL, 0); + class_addmethod(tab_fft_class, (t_method)tab_fft_src_im, gensym("src_im"), A_DEFSYMBOL, 0); + class_addmethod(tab_fft_class, (t_method)tab_fft_src_re, gensym("src1"), A_DEFSYMBOL, 0); + class_addmethod(tab_fft_class, (t_method)tab_fft_src_im, gensym("src2"), A_DEFSYMBOL, 0); + class_addmethod(tab_fft_class, (t_method)tab_fft_dst_re, gensym("dst_re"), A_DEFSYMBOL, 0); + class_addmethod(tab_fft_class, (t_method)tab_fft_dst_im, gensym("dst_im"), A_DEFSYMBOL, 0); + class_addmethod(tab_fft_class, (t_method)tab_fft_dst_re, gensym("dst1"), A_DEFSYMBOL, 0); + class_addmethod(tab_fft_class, (t_method)tab_fft_dst_im, gensym("dst2"), A_DEFSYMBOL, 0); + class_sethelpsymbol(tab_fft_class, gensym("iemhelp2/tab_fft-help")); +} |