aboutsummaryrefslogtreecommitdiff
path: root/src/mtx_cumprod.c
diff options
context:
space:
mode:
authorFranz Zotter <fzotter@users.sourceforge.net>2010-06-22 06:23:15 +0000
committerFranz Zotter <fzotter@users.sourceforge.net>2010-06-22 06:23:15 +0000
commit3c1d6bd4f73e370464c537f9a3368752bd44bdf0 (patch)
treef02fa82b929d05c8c4ba07ceca6763e3c3f1888a /src/mtx_cumprod.c
parent810388c19a126701120e2fdb3c716fc87de7e282 (diff)
added [mtx_cumprod] and [mtx_polyval] to evaluate polynomials
svn path=/trunk/externals/iem/iemmatrix/; revision=13685
Diffstat (limited to 'src/mtx_cumprod.c')
-rw-r--r--src/mtx_cumprod.c271
1 files changed, 271 insertions, 0 deletions
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_size<size) {
+ post("mtx_cumprod: sparse matrix not yet supported: use \"mtx_check\"");
+ return;
+ }
+ else if ((!x)||(!list_out)||(!y)) {
+ if (!x)
+ x = (t_float *) getbytes (sizeof (t_float) * (size));
+ if (!y)
+ y = (t_float *) getbytes (sizeof (t_float) * (size));
+ if (!list_out)
+ list_out = (t_atom *) getbytes (sizeof (t_atom) * (size+2));
+ }
+ else if (size != mtx_cumprod_obj->size) {
+ 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();
+}