added [mtx_cumprod] and [mtx_polyval] to evaluate polynomials

svn path=/trunk/externals/iem/iemmatrix/; revision=13685

3 files changed, 273 insertions, 0 deletions

diff --git a/src/iemmatrix_sources.c b/src/iemmatrix_sources.c
index 5e05345..2e6b4f4 100644
--- a/src/iemmatrix_sources.c
+++ b/src/iemmatrix_sources.c
@@ -24,6 +24,7 @@ void iemmatrix_sources_setup(void)
 	iemtx_concat_setup(); /* mtx_concat.c */
 	iemtx_conv_setup(); /* mtx_conv.c */
 	iemtx_cos_setup(); /* mtx_cos.c */
+	iemtx_cumprod_setup(); /* mtx_cumprod.c */
 	iemtx_cumsum_setup(); /* mtx_cumsum.c */
 	iemtx_dbtopow_setup(); /* mtx_dbtopow.c */
 	iemtx_dbtorms_setup(); /* mtx_dbtorms.c */
diff --git a/src/iemmatrix_sources.h b/src/iemmatrix_sources.h
index 3e56eba..df6dc7b 100644
--- a/src/iemmatrix_sources.h
+++ b/src/iemmatrix_sources.h
@@ -22,6 +22,7 @@ void iemtx_colon_setup(void); /* mtx_colon.c */
 void iemtx_concat_setup(void); /* mtx_concat.c */
 void iemtx_conv_setup(void); /* mtx_conv.c */
 void iemtx_cos_setup(void); /* mtx_cos.c */
+void iemtx_cumprod_setup(void); /* mtx_cumprod.c */
 void iemtx_cumsum_setup(void); /* mtx_cumsum.c */
 void iemtx_dbtopow_setup(void); /* mtx_dbtopow.c */
 void iemtx_dbtorms_setup(void); /* mtx_dbtorms.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_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();
+}