From 8c5d0621d0d305bb62997b4ad12af874ec253432 Mon Sep 17 00:00:00 2001 From: musil Date: Fri, 10 Mar 2006 03:05:55 +0000 Subject: initial check in svn path=/trunk/externals/iem/iem_tab/; revision=4683 --- src/tab_rifft.c | 406 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 406 insertions(+) create mode 100644 src/tab_rifft.c (limited to 'src/tab_rifft.c') diff --git a/src/tab_rifft.c b/src/tab_rifft.c new file mode 100644 index 0000000..1b92a05 --- /dev/null +++ b/src/tab_rifft.c @@ -0,0 +1,406 @@ +/* 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 +#include +#include + + +/* -------------------------- tab_rifft ------------------------------ */ + +typedef struct _tab_rifft +{ + t_object x_obj; + int x_size_src_re; + int x_size_src_im; + int x_size_dst; + int x_offset_src_re; + int x_offset_src_im; + int x_offset_dst; + 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; +} t_tab_rifft; + +static t_class *tab_rifft_class; + +static void tab_rifft_init(t_tab_rifft *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; ix_fftsize) + { + x->x_sin_cos = (TAB_COMPLEX *)resizebytes(x->x_sin_cos, x->x_fftsize*sizeof(TAB_COMPLEX), i*sizeof(TAB_COMPLEX)); + x->x_beg_mem_dst_im = (float *)resizebytes(x->x_beg_mem_dst_im, x->x_fftsize*sizeof(float), i*sizeof(float)); + x->x_fftsize = i; + } + tab_rifft_init(x); +} + +static void tab_rifft_dst(t_tab_rifft *x, t_symbol *s) +{ + x->x_sym_dst = s; +} + +static void tab_rifft_src_re(t_tab_rifft *x, t_symbol *s) +{ + x->x_sym_src_re = s; +} + +static void tab_rifft_src_im(t_tab_rifft *x, t_symbol *s) +{ + x->x_sym_src_im = s; +} + +static void tab_rifft_bang(t_tab_rifft *x) +{ + int i, j, k; + int ok_src_re, ok_src_im, ok_dst; + 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, 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); + ok_src_im = iem_tab_check_arrays(gensym("tab_rifft"), x->x_sym_src_im, &x->x_beg_mem_src_im, &x->x_size_src_im, fftsize); + ok_dst = iem_tab_check_arrays(gensym("tab_rifft"), x->x_sym_dst, &x->x_beg_mem_dst_re, &x->x_size_dst, fftsize); + + if(ok_src_re && ok_src_im && ok_dst) + { + 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(j=0; j>= 1; + } + + j = 0; + for(i=1;i>= 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; + } + */ + + g = 1.0f / (float)fftsize; + for(i = 0; i < fftsize; i++) + { + vec_dst_re[i] *= g; + } + + outlet_bang(x->x_obj.ob_outlet); + a = (t_garray *)pd_findbyclass(x->x_sym_dst, garray_class); + garray_redraw(a); + } +} + +static void tab_rifft_list(t_tab_rifft *x, t_symbol *s, int argc, t_atom *argv) +{ + int beg_src_re, beg_src_im, beg_dst; + int i, j, k; + int ok_src_re, ok_src_im, ok_dst; + 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, g; + + if((argc >= 3) && + IS_A_FLOAT(argv,0) && + IS_A_FLOAT(argv,1) && + IS_A_FLOAT(argv,2)) + { + beg_src_re = (int)atom_getintarg(0, argc, argv); + beg_src_im = (int)atom_getintarg(1, argc, argv); + beg_dst = (int)atom_getintarg(2, argc, argv); + if(beg_src_re < 0) + beg_src_re = 0; + if(beg_src_im < 0) + beg_src_im = 0; + if(beg_dst < 0) + beg_dst = 0; + + 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, beg_src_re+fftsize); + ok_src_im = iem_tab_check_arrays(gensym("tab_rifft"), x->x_sym_src_im, &x->x_beg_mem_src_im, &x->x_size_src_im, beg_src_im+fftsize); + ok_dst = iem_tab_check_arrays(gensym("tab_rifft"), x->x_sym_dst, &x->x_beg_mem_dst_re, &x->x_size_dst, beg_dst+fftsize); + + if(ok_src_re && ok_src_im && ok_dst) + { + 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; + vec_dst_im=x->x_beg_mem_dst_im; + + for(j=0; j>= 1; + } + + j = 0; + for(i=1;i>= 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_src_re[i] *= g; + vec_src_im[i] *= g; + } + */ + + g = 1.0f / (float)fftsize; + for(i = 0; i < fftsize; i++) + { + vec_dst_re[i] *= g; + } + + outlet_bang(x->x_obj.ob_outlet); + a = (t_garray *)pd_findbyclass(x->x_sym_dst, garray_class); + garray_redraw(a); + } + } + else + { + post("tab_rifft-ERROR: list need 3 float arguments:"); + post(" source_real_offset + source_imag_offset + destination_offset"); + } +} + +static void tab_rifft_free(t_tab_rifft *x) +{ + freebytes(x->x_sin_cos, x->x_fftsize * sizeof(TAB_COMPLEX)); + freebytes(x->x_beg_mem_dst_im, x->x_fftsize * sizeof(float)); +} + +static void *tab_rifft_new(t_symbol *s, int argc, t_atom *argv) +{ + t_tab_rifft *x = (t_tab_rifft *)pd_new(tab_rifft_class); + t_symbol *src_re, *src_im, *dst; + int fftsize, i=1; + + if((argc >= 4) && + IS_A_SYMBOL(argv,0) && + IS_A_SYMBOL(argv,1) && + IS_A_SYMBOL(argv,2) && + IS_A_FLOAT(argv,3)) + { + src_re = (t_symbol *)atom_getsymbolarg(0, argc, argv); + src_im = (t_symbol *)atom_getsymbolarg(1, argc, argv); + dst = (t_symbol *)atom_getsymbolarg(2, argc, argv); + fftsize = (int)atom_getintarg(3, argc, argv); + } + else + { + post("tab_rifft-ERROR: need 3 symbols + 1 float arguments:"); + post(" source_real_array_name + source_imag_array_name + destination_array_name + IFFT-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 = dst; + x->x_sin_cos = (TAB_COMPLEX *)getbytes(x->x_fftsize * sizeof(TAB_COMPLEX)); + x->x_beg_mem_dst_im = (float *)getbytes(x->x_fftsize * sizeof(float)); + tab_rifft_init(x); + outlet_new(&x->x_obj, &s_bang); + return(x); +} + +void tab_rifft_setup(void) +{ + tab_rifft_class = class_new(gensym("tab_rifft"), (t_newmethod)tab_rifft_new, (t_method)tab_rifft_free, + sizeof(t_tab_rifft), 0, A_GIMME, 0); + class_addbang(tab_rifft_class, (t_method)tab_rifft_bang); + class_addlist(tab_rifft_class, (t_method)tab_rifft_list); + class_addmethod(tab_rifft_class, (t_method)tab_rifft_ifftsize, gensym("ifftsize"), A_DEFFLOAT, 0); + class_addmethod(tab_rifft_class, (t_method)tab_rifft_src_re, gensym("src_re"), A_DEFSYMBOL, 0); + class_addmethod(tab_rifft_class, (t_method)tab_rifft_src_im, gensym("src_im"), A_DEFSYMBOL, 0); + class_addmethod(tab_rifft_class, (t_method)tab_rifft_src_re, gensym("src1"), A_DEFSYMBOL, 0); + class_addmethod(tab_rifft_class, (t_method)tab_rifft_src_im, gensym("src2"), A_DEFSYMBOL, 0); + class_addmethod(tab_rifft_class, (t_method)tab_rifft_dst, gensym("dst"), A_DEFSYMBOL, 0); + class_addmethod(tab_rifft_class, (t_method)tab_rifft_dst, gensym("dst1"), A_DEFSYMBOL, 0); + class_sethelpsymbol(tab_rifft_class, gensym("iemhelp2/tab_rifft-help")); +} -- cgit v1.2.1