From 3c1d6bd4f73e370464c537f9a3368752bd44bdf0 Mon Sep 17 00:00:00 2001 From: Franz Zotter Date: Tue, 22 Jun 2010 06:23:15 +0000 Subject: added [mtx_cumprod] and [mtx_polyval] to evaluate polynomials svn path=/trunk/externals/iem/iemmatrix/; revision=13685 --- src/mtx_cumprod.c | 271 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 271 insertions(+) create mode 100644 src/mtx_cumprod.c (limited to 'src/mtx_cumprod.c') diff --git a/src/mtx_cumprod.c b/src/mtx_cumprod.c new file mode 100644 index 0000000..03f0c9f --- /dev/null +++ b/src/mtx_cumprod.c @@ -0,0 +1,271 @@ +/* + * 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_cumprod_class; +static t_symbol *row_sym; +static t_symbol *col_sym; +static t_symbol *col_sym2; + +typedef struct _MTXCumprod_ MTXCumprod; +struct _MTXCumprod_ +{ + t_object x_obj; + int rows; + int columns; + int size; + int cumprod_direction; + t_symbol *cumprod_mode; + + t_outlet *list_outlet; + + t_atom *list_out; + t_atom *list_in; + t_float *x; + t_float *y; +}; + +static void deleteMTXCumprod (MTXCumprod *mtx_cumprod_obj) +{ + if (mtx_cumprod_obj->list_out) + freebytes (mtx_cumprod_obj->list_out, sizeof(t_atom)*(mtx_cumprod_obj->size+2)); + if (mtx_cumprod_obj->x) + freebytes (mtx_cumprod_obj->x, sizeof(t_float)*(mtx_cumprod_obj->size)); + if (mtx_cumprod_obj->y) + freebytes (mtx_cumprod_obj->y, sizeof(t_float)*(mtx_cumprod_obj->size)); +} + +static void mTXSetCumprodDirection (MTXCumprod *mtx_cumprod_obj, t_float c_dir) +{ + int direction = (int) c_dir; + mtx_cumprod_obj->cumprod_direction = (direction==-1)?direction:1; +} + +static void mTXSetCumprodMode (MTXCumprod *mtx_cumprod_obj, t_symbol *m_sym) +{ + mtx_cumprod_obj->cumprod_mode = m_sym; +} + +static void *newMTXCumprod (t_symbol *s, int argc, t_atom *argv) +{ + MTXCumprod *mtx_cumprod_obj = (MTXCumprod *) pd_new (mtx_cumprod_class); + mTXSetCumprodMode (mtx_cumprod_obj, gensym(":")); + mTXSetCumprodDirection (mtx_cumprod_obj, 1.0f); + if (argc>=1) { + if (argv[0].a_type == A_SYMBOL) { + mTXSetCumprodMode (mtx_cumprod_obj, atom_getsymbol (argv)); + if (argc>=2) { + if (argv[1].a_type != A_SYMBOL) + mTXSetCumprodDirection (mtx_cumprod_obj, atom_getfloat (argv+1)); + else + post("mtx_cumprod: 2nd arg ignored. supposed to be float"); + } + } + else { + mTXSetCumprodDirection (mtx_cumprod_obj, atom_getfloat (argv)); + if (argc>=2) { + if (argv[1].a_type == A_SYMBOL) + mTXSetCumprodMode (mtx_cumprod_obj, atom_getsymbol (argv+1)); + else + post("mtx_cumprod: 2nd arg ignored. supposed to be symbolic, e.g. \"row\", \"col\", \":\""); + } + } + } + + mtx_cumprod_obj->list_outlet = outlet_new (&mtx_cumprod_obj->x_obj, gensym("matrix")); + return ((void *) mtx_cumprod_obj); +} + +static void mTXCumprodBang (MTXCumprod *mtx_cumprod_obj) +{ + if (mtx_cumprod_obj->list_out) + outlet_anything(mtx_cumprod_obj->list_outlet, gensym("matrix"), + mtx_cumprod_obj->size+2, mtx_cumprod_obj->list_out); +} + +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 readFloatFromListModulo (int n, int m, t_atom *l, t_float *f) +{ + t_atom *ptr = l; + int count1, count2; + n /= m; + count1 = m; + while (count1--) + for (count2 = n, ptr = l++; count2--; ptr += m, f++) + *f = atom_getfloat (ptr); +} +static void writeFloatIntoListModulo (int n, int m, t_atom *l, t_float *f) +{ + t_atom *ptr = l; + int count1, count2; + n /= m; + count1 = m; + while (count1--) + for (count2 = n, ptr = l++; count2--; ptr += m, f++) + SETFLOAT(ptr,*f); +} + +static void cumProd (int n, t_float *x, t_float *y) +{ + t_float accu = 1.0f; + for (;n--; x++, y++) { + accu *= *x; + *y = accu; + } +} +static void cumProdReverse (int n, t_float *x, t_float *y) +{ + t_float accu = 1.0f; + for (;n--; x--, y--) { + accu *= *x; + *y = accu; + } +} + +static void mTXCumprodMatrix (MTXCumprod *mtx_cumprod_obj, t_symbol *s, + int argc, t_atom *argv) +{ + int rows = atom_getint (argv++); + int columns = atom_getint (argv++); + int size = rows * columns; + int list_size = argc - 2; + t_atom *list_ptr = argv; + t_atom *list_out = mtx_cumprod_obj->list_out; + t_float *x = mtx_cumprod_obj->x; + t_float *y = mtx_cumprod_obj->y; + int count; + + /* size check */ + if (!size) { + post("mtx_cumprod: invalid dimensions"); + return; + } + else if (list_sizesize) { + x = (t_float *) resizebytes (x, + sizeof (t_float) * (mtx_cumprod_obj->size), + sizeof (t_float) * (size)); + y = (t_float *) resizebytes (y, + sizeof (t_float) * (mtx_cumprod_obj->size), + sizeof (t_float) * (size)); + list_out = (t_atom *) resizebytes (list_out, + sizeof (t_atom) * (mtx_cumprod_obj->size+2), + sizeof (t_atom) * (size + 2)); + } + mtx_cumprod_obj->size = size; + mtx_cumprod_obj->rows = rows; + mtx_cumprod_obj->columns = columns; + mtx_cumprod_obj->list_out = list_out; + mtx_cumprod_obj->x = x; + mtx_cumprod_obj->y = y; + + /* main part */ + /* reading matrix from inlet */ + if ((mtx_cumprod_obj->cumprod_mode == col_sym) || + (mtx_cumprod_obj->cumprod_mode == col_sym2)) { + readFloatFromListModulo (size, columns, list_ptr, x); + columns = mtx_cumprod_obj->rows; + rows = mtx_cumprod_obj->columns; + } + else + readFloatFromList (size, list_ptr, x); + + /* calculating cumprod */ + if (mtx_cumprod_obj->cumprod_direction == -1) { + if ((mtx_cumprod_obj->cumprod_mode == row_sym) || + (mtx_cumprod_obj->cumprod_mode == col_sym) || + (mtx_cumprod_obj->cumprod_mode == col_sym2)) { + x += columns-1; + y += columns-1; + + for (count = rows; count--; x += columns, y += columns) + cumProdReverse (columns,x,y); + } + else { + x += size-1; + y += size-1; + cumProdReverse (size, x, y); + } + } + else if ((mtx_cumprod_obj->cumprod_mode == row_sym) || + (mtx_cumprod_obj->cumprod_mode == col_sym) || + (mtx_cumprod_obj->cumprod_mode == col_sym2)) + for (count = rows; count--; x += columns, y += columns) + cumProd (columns,x,y); + else + cumProd (size, x, y); + + x = mtx_cumprod_obj->x; + y = mtx_cumprod_obj->y; + + /* writing matrix to outlet */ + if ((mtx_cumprod_obj->cumprod_mode == col_sym) || + (mtx_cumprod_obj->cumprod_mode == col_sym2)) { + columns = mtx_cumprod_obj->columns; + rows = mtx_cumprod_obj->rows; + writeFloatIntoListModulo (size, columns, list_out+2, y); + } + else + writeFloatIntoList (size, list_out+2, y); + + SETSYMBOL(list_out, gensym("matrix")); + SETFLOAT(list_out, rows); + SETFLOAT(&list_out[1], columns); + outlet_anything(mtx_cumprod_obj->list_outlet, gensym("matrix"), + mtx_cumprod_obj->size+2, list_out); +} + +void mtx_cumprod_setup (void) +{ + mtx_cumprod_class = class_new + (gensym("mtx_cumprod"), + (t_newmethod) newMTXCumprod, + (t_method) deleteMTXCumprod, + sizeof (MTXCumprod), + CLASS_DEFAULT, A_GIMME, 0); + class_addbang (mtx_cumprod_class, (t_method) mTXCumprodBang); + class_addmethod (mtx_cumprod_class, (t_method) mTXCumprodMatrix, gensym("matrix"), A_GIMME,0); + class_addmethod (mtx_cumprod_class, (t_method) mTXSetCumprodMode, gensym("mode"), A_DEFSYMBOL,0); + class_addmethod (mtx_cumprod_class, (t_method) mTXSetCumprodDirection, gensym("direction"), A_DEFFLOAT,0); + + row_sym = gensym("row"); + col_sym = gensym("col"); + col_sym2 = gensym("column"); +} + +void iemtx_cumprod_setup(void){ + mtx_cumprod_setup(); +} -- cgit v1.2.1