aboutsummaryrefslogtreecommitdiff
path: root/src/mtx_cumsum.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/mtx_cumsum.c')
-rw-r--r--src/mtx_cumsum.c238
1 files changed, 238 insertions, 0 deletions
diff --git a/src/mtx_cumsum.c b/src/mtx_cumsum.c
new file mode 100644
index 0000000..a5c12ba
--- /dev/null
+++ b/src/mtx_cumsum.c
@@ -0,0 +1,238 @@
+/*
+ * 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_cumsum_class;
+
+typedef struct _MTXCumsum_ MTXCumsum;
+struct _MTXCumsum_
+{
+ t_object x_obj;
+ int rows;
+ int columns;
+ int size;
+ int cumsum_dimension;
+ int cumsum_direction;
+
+ t_outlet *list_outlet;
+
+ t_atom *list_out;
+ t_atom *list_in;
+ t_float *x;
+ t_float *y;
+};
+
+static void deleteMTXCumsum (MTXCumsum *mtx_cumsum_obj)
+{
+ if (mtx_cumsum_obj->list_out)
+ freebytes (mtx_cumsum_obj->list_out, sizeof(t_atom)*(mtx_cumsum_obj->size+2));
+ if (mtx_cumsum_obj->x)
+ freebytes (mtx_cumsum_obj->x, sizeof(t_float)*(mtx_cumsum_obj->size));
+ if (mtx_cumsum_obj->y)
+ freebytes (mtx_cumsum_obj->y, sizeof(t_float)*(mtx_cumsum_obj->size));
+}
+
+static void mTXSetCumsumDirection (MTXCumsum *mtx_cumsum_obj, t_float c_dir)
+{
+ int direction = (int) c_dir;
+ mtx_cumsum_obj->cumsum_direction = (direction==-1)?direction:1;
+}
+static void mTXSetCumsumDimension (MTXCumsum *mtx_cumsum_obj, t_float c_dim)
+{
+ int dimension = (int) c_dim;
+ mtx_cumsum_obj->cumsum_dimension = (dimension==2)?dimension:1;
+}
+
+static void *newMTXCumsum (t_symbol *s, int argc, t_atom *argv)
+{
+ MTXCumsum *mtx_cumsum_obj = (MTXCumsum *) pd_new (mtx_cumsum_class);
+ int c_dir = 1;
+ int c_dim = 1;
+
+ mtx_cumsum_obj->cumsum_dimension = c_dim;
+ switch ((argc>2)?2:argc) {
+ case 2:
+ c_dir = atom_getint(argv+1);
+ case 1:
+ c_dim = atom_getint(argv);
+ }
+ mTXSetCumsumDirection (mtx_cumsum_obj, (t_float) c_dir);
+ mTXSetCumsumDimension (mtx_cumsum_obj, (t_float) c_dim);
+
+ mtx_cumsum_obj->list_outlet = outlet_new (&mtx_cumsum_obj->x_obj, gensym("matrix"));
+ return ((void *) mtx_cumsum_obj);
+}
+
+static void mTXCumsumBang (MTXCumsum *mtx_cumsum_obj)
+{
+ if (mtx_cumsum_obj->list_out)
+ outlet_anything(mtx_cumsum_obj->list_outlet, gensym("matrix"),
+ mtx_cumsum_obj->size+2, mtx_cumsum_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 cumSum (int n, t_float *x, t_float *y)
+{
+ t_float accu = 0.0f;
+ for (;n--; x++, y++) {
+ accu += *x;
+ *y = accu;
+ }
+}
+static void cumSumReverse (int n, t_float *x, t_float *y)
+{
+ t_float accu = 0.0f;
+ for (;n--; x--, y--) {
+ accu += *x;
+ *y = accu;
+ }
+}
+
+static void mTXCumsumMatrix (MTXCumsum *mtx_cumsum_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_cumsum_obj->list_out;
+ t_float *x = mtx_cumsum_obj->x;
+ t_float *y = mtx_cumsum_obj->y;
+ int count;
+
+ // size check
+ if (!size) {
+ post("mtx_cumsum: invalid dimensions");
+ return;
+ }
+ else if (list_size<size) {
+ post("mtx_cumsum: 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_cumsum_obj->size) {
+ x = (t_float *) resizebytes (x,
+ sizeof (t_float) * (mtx_cumsum_obj->size),
+ sizeof (t_float) * (size));
+ y = (t_float *) resizebytes (y,
+ sizeof (t_float) * (mtx_cumsum_obj->size),
+ sizeof (t_float) * (size));
+ list_out = (t_atom *) resizebytes (list_out,
+ sizeof (t_atom) * (mtx_cumsum_obj->size+2),
+ sizeof (t_atom) * (size + 2));
+ }
+ mtx_cumsum_obj->size = size;
+ mtx_cumsum_obj->rows = rows;
+ mtx_cumsum_obj->columns = columns;
+ mtx_cumsum_obj->list_out = list_out;
+ mtx_cumsum_obj->x = x;
+ mtx_cumsum_obj->y = y;
+
+ // main part
+ // reading matrix from inlet
+ if (mtx_cumsum_obj->cumsum_dimension == 2) {
+ readFloatFromListModulo (size, columns, list_ptr, x);
+ columns = mtx_cumsum_obj->rows;
+ rows = mtx_cumsum_obj->columns;
+ }
+ else
+ readFloatFromList (size, list_ptr, x);
+
+ // calculating cumsum
+ if (mtx_cumsum_obj->cumsum_direction == -1) {
+ x += columns-1;
+ y += columns-1;
+ for (count = rows; count--; x += columns, y += columns)
+ cumSumReverse (columns,x,y);
+ }
+ else
+ for (count = rows; count--; x += columns, y += columns)
+ cumSum (columns,x,y);
+ x = mtx_cumsum_obj->x;
+ y = mtx_cumsum_obj->y;
+
+ // writing matrix to outlet
+ if (mtx_cumsum_obj->cumsum_dimension == 2) {
+ columns = mtx_cumsum_obj->columns;
+ rows = mtx_cumsum_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_cumsum_obj->list_outlet, gensym("matrix"),
+ mtx_cumsum_obj->size+2, list_out);
+}
+
+void mtx_cumsum_setup (void)
+{
+ mtx_cumsum_class = class_new
+ (gensym("mtx_cumsum"),
+ (t_newmethod) newMTXCumsum,
+ (t_method) deleteMTXCumsum,
+ sizeof (MTXCumsum),
+ CLASS_DEFAULT, A_GIMME, 0);
+ class_addbang (mtx_cumsum_class, (t_method) mTXCumsumBang);
+ class_addmethod (mtx_cumsum_class, (t_method) mTXCumsumMatrix, gensym("matrix"), A_GIMME,0);
+ class_addmethod (mtx_cumsum_class, (t_method) mTXSetCumsumDimension, gensym("dimension"), A_DEFFLOAT,0);
+ class_addmethod (mtx_cumsum_class, (t_method) mTXSetCumsumDirection, gensym("direction"), A_DEFFLOAT,0);
+ class_sethelpsymbol (mtx_cumsum_class, gensym("iemmatrix/mtx_cumsum"));
+}
+
+void iemtx_cumsum_setup(void){
+ mtx_cumsum_setup();
+}