diff options
Diffstat (limited to 'src/tab_complex_inv.c')
| -rw-r--r-- | src/tab_complex_inv.c | 246 |
1 files changed, 246 insertions, 0 deletions
diff --git a/src/tab_complex_inv.c b/src/tab_complex_inv.c new file mode 100644 index 0000000..a0264b9 --- /dev/null +++ b/src/tab_complex_inv.c @@ -0,0 +1,246 @@ +/* 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_complex_inv ------------------------------ */ + +typedef struct _tab_complex_inv +{ + t_object x_obj; + int x_size_src1_re; + int x_size_dst_re; + int x_size_src1_im; + int x_size_dst_im; + int x_offset_src1_re; + int x_offset_dst_re; + int x_offset_src1_im; + int x_offset_dst_im; + float *x_beg_mem_src1_re; + float *x_beg_mem_dst_re; + float *x_beg_mem_src1_im; + float *x_beg_mem_dst_im; + t_symbol *x_sym_scr1_re; + t_symbol *x_sym_dst_re; + t_symbol *x_sym_scr1_im; + t_symbol *x_sym_dst_im; +} t_tab_complex_inv; + +static t_class *tab_complex_inv_class; + +static void tab_complex_inv_src1_re(t_tab_complex_inv *x, t_symbol *s) +{ + x->x_sym_scr1_re = s; +} + +static void tab_complex_inv_src1_im(t_tab_complex_inv *x, t_symbol *s) +{ + x->x_sym_scr1_im = s; +} + +static void tab_complex_inv_dst_re(t_tab_complex_inv *x, t_symbol *s) +{ + x->x_sym_dst_re = s; +} + +static void tab_complex_inv_dst_im(t_tab_complex_inv *x, t_symbol *s) +{ + x->x_sym_dst_im = s; +} + +static void tab_complex_inv_bang(t_tab_complex_inv *x) +{ + int i, n; + int ok_src1_re, ok_dst_re; + int ok_src1_im, ok_dst_im; + t_float *vec_src1_re, *vec_dst_re; + t_float *vec_src1_im, *vec_dst_im; + + ok_src1_re = iem_tab_check_arrays(gensym("tab_complex_inv"), x->x_sym_scr1_re, &x->x_beg_mem_src1_re, &x->x_size_src1_re, 0); + ok_dst_re = iem_tab_check_arrays(gensym("tab_complex_inv"), x->x_sym_dst_re, &x->x_beg_mem_dst_re, &x->x_size_dst_re, 0); + ok_src1_im = iem_tab_check_arrays(gensym("tab_complex_inv"), x->x_sym_scr1_im, &x->x_beg_mem_src1_im, &x->x_size_src1_im, 0); + ok_dst_im = iem_tab_check_arrays(gensym("tab_complex_inv"), x->x_sym_dst_im, &x->x_beg_mem_dst_im, &x->x_size_dst_im, 0); + + if(ok_src1_re && ok_dst_re && ok_src1_im && ok_dst_im) + { + if(x->x_size_src1_re < x->x_size_dst_re) + n = x->x_size_src1_re; + else + n = x->x_size_dst_re; + if(x->x_size_src1_im < n) + n = x->x_size_src1_im; + if(x->x_size_dst_im < n) + n = x->x_size_dst_im; + + vec_src1_re = x->x_beg_mem_src1_re; + vec_dst_re = x->x_beg_mem_dst_re; + vec_src1_im = x->x_beg_mem_src1_im; + vec_dst_im = x->x_beg_mem_dst_im; + if(n) + { + t_garray *a; + + for(i=0; i<n; i++) + { + t_float re, im, abs; + + re = vec_src1_re[i]; + im = vec_src1_im[i]; + abs = 1.0f / (re*re + im*im); + vec_dst_re[i] = re*abs; + vec_dst_im[i] = -im*abs; + } + 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_complex_inv_list(t_tab_complex_inv *x, t_symbol *s, int argc, t_atom *argv) +{ + int i, n; + int beg_src1_re, beg_dst_re; + int beg_src1_im, beg_dst_im; + int ok_src1_re, ok_dst_re; + int ok_src1_im, ok_dst_im; + t_float *vec_src1_re, *vec_dst_re; + t_float *vec_src1_im, *vec_dst_im; + + if((argc >= 5) && + IS_A_FLOAT(argv,0) && + IS_A_FLOAT(argv,1) && + IS_A_FLOAT(argv,2) && + IS_A_FLOAT(argv,3) && + IS_A_FLOAT(argv,4)) + { + beg_src1_re = (int)atom_getintarg(0, argc, argv); + beg_src1_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); + n = (int)atom_getintarg(4, argc, argv); + if(beg_src1_re < 0) + beg_src1_re = 0; + if(beg_dst_re < 0) + beg_dst_re = 0; + if(beg_src1_im < 0) + beg_src1_im = 0; + if(beg_dst_im < 0) + beg_dst_im = 0; + if(n < 0) + n = 0; + + ok_src1_re = iem_tab_check_arrays(gensym("tab_complex_inv"), x->x_sym_scr1_re, &x->x_beg_mem_src1_re, &x->x_size_src1_re, beg_src1_re+n); + ok_dst_re = iem_tab_check_arrays(gensym("tab_complex_inv"), x->x_sym_dst_re, &x->x_beg_mem_dst_re, &x->x_size_dst_re, beg_dst_re+n); + ok_src1_im = iem_tab_check_arrays(gensym("tab_complex_inv"), x->x_sym_scr1_im, &x->x_beg_mem_src1_im, &x->x_size_src1_im, beg_src1_im+n); + ok_dst_im = iem_tab_check_arrays(gensym("tab_complex_inv"), x->x_sym_dst_im, &x->x_beg_mem_dst_im, &x->x_size_dst_im, beg_dst_im+n); + + if(ok_src1_re && ok_dst_re && ok_src1_im && ok_dst_im) + { + vec_src1_re = x->x_beg_mem_src1_re + beg_src1_re; + vec_dst_re = x->x_beg_mem_dst_re + beg_dst_re; + vec_src1_im = x->x_beg_mem_src1_im + beg_src1_im; + vec_dst_im = x->x_beg_mem_dst_im + beg_dst_im; + + if(n) + { + t_garray *a; + + for(i=0; i<n; i++) + { + t_float re, im, abs; + + re = vec_src1_re[i]; + im = vec_src1_im[i]; + abs = 1.0f / (re*re + im*im); + vec_dst_re[i] = re*abs; + vec_dst_im[i] = -im*abs; + } + 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_complex_inv-ERROR: list need 5 float arguments:"); + post(" source1_real_offset + source1_imag_offset + destination_real_offset + destination_imag_offset + number_of_samples_to_complex_mul"); + } +} + +static void tab_complex_inv_free(t_tab_complex_inv *x) +{ +} + +static void *tab_complex_inv_new(t_symbol *s, int argc, t_atom *argv) +{ + t_tab_complex_inv *x = (t_tab_complex_inv *)pd_new(tab_complex_inv_class); + t_symbol *src1_re, *src2_re, *dst_re, *src1_im, *src2_im, *dst_im; + t_float time; + + if((argc >= 4) && + IS_A_SYMBOL(argv,0) && + IS_A_SYMBOL(argv,1) && + IS_A_SYMBOL(argv,2) && + IS_A_SYMBOL(argv,3)) + { + src1_re = (t_symbol *)atom_getsymbolarg(0, argc, argv); + src1_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); + } + else if((argc >= 2) && + IS_A_SYMBOL(argv,0) && + IS_A_SYMBOL(argv,1)) + { + src1_re = (t_symbol *)atom_getsymbolarg(0, argc, argv); + src1_im = (t_symbol *)atom_getsymbolarg(1, argc, argv); + dst_re = src1_re; + dst_im = src1_im; + } + else + { + post("tab_complex_inv-ERROR: need 4 symbols arguments:"); + post(" source1_real_array_name + source1_imag_array_name + destination_real_array_name + destination_imag_array_name"); + return(0); + } + + x->x_sym_scr1_re = src1_re; + x->x_sym_scr1_im = src1_im; + x->x_sym_dst_re = dst_re; + x->x_sym_dst_im = dst_im; + outlet_new(&x->x_obj, &s_bang); + return(x); +} + +void tab_complex_inv_setup(void) +{ + tab_complex_inv_class = class_new(gensym("tab_complex_inv"), (t_newmethod)tab_complex_inv_new, (t_method)tab_complex_inv_free, + sizeof(t_tab_complex_inv), 0, A_GIMME, 0); + class_addbang(tab_complex_inv_class, (t_method)tab_complex_inv_bang); + class_addlist(tab_complex_inv_class, (t_method)tab_complex_inv_list); + class_addmethod(tab_complex_inv_class, (t_method)tab_complex_inv_src1_re, gensym("src1_re"), A_DEFSYMBOL, 0); + class_addmethod(tab_complex_inv_class, (t_method)tab_complex_inv_dst_re, gensym("dst_re"), A_DEFSYMBOL, 0); + class_addmethod(tab_complex_inv_class, (t_method)tab_complex_inv_src1_im, gensym("src1_im"), A_DEFSYMBOL, 0); + class_addmethod(tab_complex_inv_class, (t_method)tab_complex_inv_dst_im, gensym("dst_im"), A_DEFSYMBOL, 0); + class_sethelpsymbol(tab_complex_inv_class, gensym("iemhelp2/tab_complex_inv-help")); +} |