diff options
Diffstat (limited to 'src/mtx_rowrfft.c')
| -rw-r--r-- | src/mtx_rowrfft.c | 205 |
1 files changed, 205 insertions, 0 deletions
diff --git a/src/mtx_rowrfft.c b/src/mtx_rowrfft.c new file mode 100644 index 0000000..e7f0551 --- /dev/null +++ b/src/mtx_rowrfft.c @@ -0,0 +1,205 @@ +/* + * iemmatrix + * + * objects for manipulating simple matrices + * mostly refering to matlab/octave matrix functions + * + * Copyright (c) 2005, Franz Zotter + * IEM, Graz, Austria + * + * For information on usage and redistribution, and for a DISCLAIMER OF ALL + * WARRANTIES, see the file, "LICENSE.txt," in this distribution. + * + */ + +#include "iemmatrix.h" + +static t_class *mtx_rowrfft_class; + +typedef struct _MTXRowrfft_ MTXRowrfft; +struct _MTXRowrfft_ +{ + t_object x_obj; + int rows; + int columns; + int columns_re; + int size; + int size2; + t_float renorm_fac; + + t_float *f_re; + t_float *f_im; + + t_outlet *list_re_out; + t_outlet *list_im_out; + + t_atom *list_re; + t_atom *list_im; +}; + +static void deleteMTXRowrfft (MTXRowrfft *mtx_rowrfft_obj) +{ + if (mtx_rowrfft_obj->f_re) + freebytes (mtx_rowrfft_obj->f_re, sizeof(t_float)*mtx_rowrfft_obj->size); + if (mtx_rowrfft_obj->f_im) + freebytes (mtx_rowrfft_obj->f_im, sizeof(t_float)*mtx_rowrfft_obj->size); + if (mtx_rowrfft_obj->list_re) + freebytes (mtx_rowrfft_obj->list_re, sizeof(t_atom)*(mtx_rowrfft_obj->size2+2)); + if (mtx_rowrfft_obj->list_im) + freebytes (mtx_rowrfft_obj->list_im, sizeof(t_atom)*(mtx_rowrfft_obj->size2+2)); +} + +static void *newMTXRowrfft (t_symbol *s, int argc, t_atom *argv) +{ + MTXRowrfft *mtx_rowrfft_obj = (MTXRowrfft *) pd_new (mtx_rowrfft_class); + mtx_rowrfft_obj->list_re_out = outlet_new (&mtx_rowrfft_obj->x_obj, gensym("matrix")); + mtx_rowrfft_obj->list_im_out = outlet_new (&mtx_rowrfft_obj->x_obj, gensym("matrix")); + return ((void *) mtx_rowrfft_obj); +} + +static void mTXrowrfftBang (MTXRowrfft *mtx_rowrfft_obj) +{ + if (mtx_rowrfft_obj->list_im) { + outlet_anything(mtx_rowrfft_obj->list_im_out, gensym("matrix"), mtx_rowrfft_obj->size2+2, mtx_rowrfft_obj->list_im); + outlet_anything(mtx_rowrfft_obj->list_re_out, gensym("matrix"), mtx_rowrfft_obj->size2+2, mtx_rowrfft_obj->list_re); + } +} + +static void fftRestoreImag (int n, t_float *re, t_float *im) +{ + t_float *im2; + n >>= 1; + *im=0; + re += n; + im += n; + im2 = im; + *im=0; + while (--n) { + *--im = -*++re; + *++im2 = 0; + *re = 0; + } +} + +static void zeroFloatArray (int n, t_float *f) +{ + while (n--) + *f++ = 0.0f; +} + +static void writeFloatIntoList (int n, t_atom *l, t_float *f) +{ + for (;n--;f++, l++) + SETFLOAT (l, *f); +} +static void readFloatFromList (int n, t_atom *l, t_float *f) +{ + while (n--) + *f++ = atom_getfloat (l++); +} + +static void mTXrowrfftMatrix (MTXRowrfft *mtx_rowrfft_obj, t_symbol *s, + int argc, t_atom *argv) +{ + //mTXrowrfftList (mtx_rowrfft_obj, s, argc-2, argv+2); + int rows = atom_getint (argv++); + int columns = atom_getint (argv++); + int columns_re = (columns>>1)+1; + int size = rows * columns; + int in_size = argc-2; + int size2 = columns_re * rows; + int fft_count; + t_atom *ptr_re = mtx_rowrfft_obj->list_re; + t_atom *ptr_im = mtx_rowrfft_obj->list_im; + t_float *f_re = mtx_rowrfft_obj->f_re; + t_float *f_im = mtx_rowrfft_obj->f_im; + + // fftsize check + if (!size) + post("mtx_rowrfft: invalid dimensions"); + else if (in_size<size) + post("mtx_rowrfft: sparse matrix not yet supported: use \"mtx_check\""); + else if (columns == (1 << ilog2(columns))) { + // memory things + if (f_re) { + if (size != mtx_rowrfft_obj->size) { + f_re = (t_float *) resizebytes (f_re, + sizeof (t_float) * mtx_rowrfft_obj->size, + sizeof (t_float) * size); + f_im = (t_float *) resizebytes (f_im, + sizeof (t_float) * mtx_rowrfft_obj->size, + sizeof (t_float) * size); + ptr_re = (t_atom *) resizebytes (ptr_re, + sizeof (t_atom) * (mtx_rowrfft_obj->size2+2), + sizeof (t_atom) * (size2 + 2)); + ptr_im = (t_atom *) resizebytes (ptr_im, + sizeof (t_atom) * (mtx_rowrfft_obj->size2+2), + sizeof (t_atom) * (size2 + 2)); + } + } + else { + f_re = (t_float *) getbytes (sizeof (t_float) * size); + f_im = (t_float *) getbytes (sizeof (t_float) * size); + ptr_re = (t_atom *) getbytes (sizeof (t_atom) * (size2+2)); + ptr_im = (t_atom *) getbytes (sizeof (t_atom) * (size2+2)); + } + mtx_rowrfft_obj->size = size; + mtx_rowrfft_obj->size2 = size2; + mtx_rowrfft_obj->rows = rows; + mtx_rowrfft_obj->columns = columns; + mtx_rowrfft_obj->columns_re = columns_re; + mtx_rowrfft_obj->list_im = ptr_im; + mtx_rowrfft_obj->list_re = ptr_re; + mtx_rowrfft_obj->f_re = f_re; + mtx_rowrfft_obj->f_im = f_im; + + // main part + readFloatFromList (size, argv, f_re); + + fft_count = rows; + ptr_re += 2; + ptr_im += 2; + while (fft_count--){ + mayer_realfft (columns, f_re); + fftRestoreImag (columns, f_re, f_im); + writeFloatIntoList (columns_re, ptr_re, f_re); + writeFloatIntoList (columns_re, ptr_im, f_im); + f_im += columns; + f_re += columns; + ptr_re += columns_re; + ptr_im += columns_re; + } + ptr_re = mtx_rowrfft_obj->list_re; + ptr_im = mtx_rowrfft_obj->list_im; + + SETSYMBOL(ptr_re, gensym("matrix")); + SETSYMBOL(ptr_im, gensym("matrix")); + SETFLOAT(ptr_re, rows); + SETFLOAT(ptr_im, rows); + SETFLOAT(&ptr_re[1], columns_re); + SETFLOAT(&ptr_im[1], columns_re); + outlet_anything(mtx_rowrfft_obj->list_im_out, gensym("matrix"), + mtx_rowrfft_obj->size2+2, ptr_im); + outlet_anything(mtx_rowrfft_obj->list_re_out, gensym("matrix"), + mtx_rowrfft_obj->size2+2, ptr_re); + } + else + post("mtx_rowfft: rowvector size no power of 2!"); +} + +void mtx_rowrfft_setup (void) +{ + mtx_rowrfft_class = class_new + (gensym("mtx_rowrfft"), + (t_newmethod) newMTXRowrfft, + (t_method) deleteMTXRowrfft, + sizeof (MTXRowrfft), + CLASS_DEFAULT, A_GIMME, 0); + class_addbang (mtx_rowrfft_class, (t_method) mTXrowrfftBang); + class_addmethod (mtx_rowrfft_class, (t_method) mTXrowrfftMatrix, gensym("matrix"), A_GIMME,0); + class_sethelpsymbol (mtx_rowrfft_class, gensym("iemmatrix/mtx_rowrfft")); +} + +void iemtx_rowrfft_setup(void){ + mtx_rowrfft_setup(); +} |