aboutsummaryrefslogtreecommitdiff
path: root/src/mtx_find.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/mtx_find.c')
-rw-r--r--src/mtx_find.c286
1 files changed, 286 insertions, 0 deletions
diff --git a/src/mtx_find.c b/src/mtx_find.c
new file mode 100644
index 0000000..6fdbee7
--- /dev/null
+++ b/src/mtx_find.c
@@ -0,0 +1,286 @@
+/*
+ * 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_find_class;
+
+typedef struct _MTXfind_ MTXfind;
+struct _MTXfind_
+{
+ t_object x_obj;
+ int size;
+ int outsize;
+ int find_dimension;
+ int find_direction;
+
+ t_outlet *list_outlet;
+
+ t_atom *list_out;
+ t_atom *list_in;
+};
+
+static void deleteMTXFind (MTXfind *mtx_find_obj)
+{
+ if (mtx_find_obj->list_out)
+ freebytes (mtx_find_obj->list_out, sizeof(t_atom)*(mtx_find_obj->size+2));
+}
+
+static void mTXSetFindDirection (MTXfind *mtx_find_obj, t_float c_dir)
+{
+ int direction = (int) c_dir;
+ if ((direction != -1) && (direction != 1))
+ direction = 1;
+ mtx_find_obj->find_direction = direction;
+}
+
+static void mTXSetFindDimension (MTXfind *mtx_find_obj, t_float c_dim)
+{
+ int dimension = (int) c_dim;
+ dimension = (dimension > 0)?dimension:0;
+ dimension = (dimension < 3)?dimension:3;
+ mtx_find_obj->find_dimension = dimension;
+}
+
+static void *newMTXFind (t_symbol *s, int argc, t_atom *argv)
+{
+ MTXfind *mtx_find_obj = (MTXfind *) pd_new (mtx_find_class);
+ int c_dim = 0;
+ int c_dir = 1;
+
+ switch ((argc>2)?2:argc) {
+ case 2:
+ c_dir = atom_getint(argv+1);
+ case 1:
+ c_dim = atom_getint(argv);
+ }
+ mTXSetFindDimension (mtx_find_obj, (t_float) c_dim);
+ mTXSetFindDirection (mtx_find_obj, (t_float) c_dir);
+
+ mtx_find_obj->list_outlet = outlet_new (&mtx_find_obj->x_obj, gensym("matrix"));
+
+ error("[mtx_find]: this object is likely to change! not really for use yet");
+ return ((void *) mtx_find_obj);
+}
+
+static void mTXFindBang (MTXfind *mtx_find_obj)
+{
+ if (mtx_find_obj->list_out)
+ outlet_anything(mtx_find_obj->list_outlet, gensym("matrix"),
+ mtx_find_obj->outsize+2, mtx_find_obj->list_out);
+}
+/*
+static void copyList (int size, t_atom *x, t_atom *y)
+{
+ while(size--)
+ *y++=*x++;
+ }
+ */
+static int findPreviousNonZero (const int n, t_atom *x, int offset)
+{
+ x+=offset;
+ for (; offset > n; offset--, x--)
+ if (atom_getfloat(x))
+ return offset;
+ return -1;
+}
+static int findPreviousNonZeroStep (const int step, t_atom *x, int offset)
+{
+ x += offset;
+ for (; offset > 0; offset-=step, x-=step)
+ if (atom_getfloat(x))
+ return offset;
+ return -1;
+}
+static int findNextNonZero (const int n, t_atom *x, int offset)
+{
+ x+=offset;
+ for (; offset < n; offset++, x++)
+ if (atom_getfloat(x))
+ return offset;
+ return -1;
+}
+static int findNextNonZeroStep (const int n, const int step, t_atom *x, int offset)
+{
+ x += offset;
+ for (; offset < n; offset+=step, x+=step)
+ if (atom_getfloat(x))
+ return offset;
+ return -1;
+}
+
+static void findFirstNonZeroRow (const int rows, const int columns, t_atom *x, t_atom *y)
+{
+ int offset;
+ int pos;
+ const int size = rows*columns;
+ for (offset = 0; offset < size; y++, offset+=columns) {
+ pos = findNextNonZero(offset+columns,x,offset)+1;
+ SETFLOAT(y,(t_float)pos);
+ }
+}
+static void findLastNonZeroRow (const int rows, const int columns, t_atom *x, t_atom *y)
+{
+ int offset;
+ int pos;
+ const int size = rows*columns;
+ for (offset = columns-1; offset < size; y++, offset+=columns) {
+ pos = findPreviousNonZero(offset-columns,x,offset)+1;
+ SETFLOAT(y,(t_float)pos);
+ }
+}
+static void findFirstNonZeroColumn (const int rows, const int columns, t_atom *x, t_atom *y)
+{
+ int offset;
+ int pos;
+ const int size = rows*columns;
+ for (offset = 0; offset < columns; y++, offset++) {
+ pos = findNextNonZeroStep(size,columns,x,offset)+1;
+ SETFLOAT(y,(t_float)pos);
+ }
+}
+static void findLastNonZeroColumn (const int rows, const int columns, t_atom *x, t_atom *y)
+{
+ int offset;
+ int pos;
+ const int size = rows*columns;
+ for (offset = size-columns; offset < size; y++, offset++) {
+ pos = findPreviousNonZeroStep(columns,x,offset)+1;
+ SETFLOAT(y,(t_float)pos);
+ }
+}
+
+static int findAllNonZeros (int n, t_atom *x, t_atom *y)
+{
+ int outsize = 0;
+ int pos = 0;
+ while ((pos = findNextNonZero(n,x,pos)) != -1) {
+ pos++;
+ SETFLOAT(y,(t_float)pos);
+ y++;
+ outsize++;
+ }
+ return outsize;
+}
+
+static void zeroFloatList (int n, t_atom *x)
+{
+ for (;n--;x++)
+ SETFLOAT(x,0);
+}
+static void findReplaceNonZerosWithIndex (int n, t_atom *x, t_atom *y)
+{
+ int pos = 0;
+ zeroFloatList(n,y);
+ while ((pos = findNextNonZero(n,x,pos)) != -1) {
+ SETFLOAT(y+pos,(t_float)pos+1);
+ pos++;
+ }
+}
+
+static void mTXFindMatrix (MTXfind *mtx_find_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_in = argv;
+ t_atom *list_out = mtx_find_obj->list_out;
+ int count;
+ int rows_out;
+ int columns_out;
+
+ // size check
+ if (!size) {
+ post("mtx_find: invalid dimensions");
+ return;
+ }
+ else if (list_size<size) {
+ post("mtx_find: sparse matrix not yet supported: use \"mtx_check\"");
+ return;
+ }
+
+ if (size != mtx_find_obj->size) {
+ if (!list_out)
+ list_out = (t_atom *) getbytes (sizeof (t_atom) * (size + 2));
+ else
+ list_out = (t_atom *) resizebytes (list_out,
+ sizeof (t_atom) * (mtx_find_obj->size+2),
+ sizeof (t_atom) * (size + 2));
+ }
+
+ mtx_find_obj->size = size;
+ mtx_find_obj->list_out = list_out;
+
+ // main part
+ list_out += 2;
+ //copyList (size, argv, list_out);
+ rows_out = 1;
+ switch (mtx_find_obj->find_dimension) {
+ case 0:
+ columns_out = findAllNonZeros (size, list_in, list_out);
+ rows_out = 1;
+ break;
+ case 3:
+ findReplaceNonZerosWithIndex (size, list_in, list_out);
+ rows_out = rows;
+ columns_out = columns;
+ break;
+ case 2:
+ if (mtx_find_obj->find_direction == -1)
+ findLastNonZeroColumn (rows, columns, list_in, list_out);
+ else
+ findFirstNonZeroColumn (rows, columns, list_in, list_out);
+ columns_out = columns;
+ rows_out = 1;
+ break;
+ case 1:
+ if (mtx_find_obj->find_direction == -1)
+ findLastNonZeroRow (rows, columns, list_in, list_out);
+ else
+ findFirstNonZeroRow (rows, columns, list_in, list_out);
+ rows_out = rows;
+ columns_out = 1;
+ break;
+ }
+ mtx_find_obj->outsize = columns_out * rows_out;
+ list_out = mtx_find_obj->list_out;
+
+ SETSYMBOL(list_out, gensym("matrix"));
+ SETFLOAT(list_out, rows_out);
+ SETFLOAT(&list_out[1], columns_out);
+ outlet_anything(mtx_find_obj->list_outlet, gensym("matrix"),
+ mtx_find_obj->outsize+2, list_out);
+}
+
+void mtx_find_setup (void)
+{
+ mtx_find_class = class_new
+ (gensym("mtx_find"),
+ (t_newmethod) newMTXFind,
+ (t_method) deleteMTXFind,
+ sizeof (MTXfind),
+ CLASS_DEFAULT, A_GIMME, 0);
+ class_addbang (mtx_find_class, (t_method) mTXFindBang);
+ class_addmethod (mtx_find_class, (t_method) mTXFindMatrix, gensym("matrix"), A_GIMME,0);
+ class_addmethod (mtx_find_class, (t_method) mTXSetFindDimension, gensym("dimension"), A_DEFFLOAT,0);
+ class_addmethod (mtx_find_class, (t_method) mTXSetFindDirection, gensym("direction"), A_DEFFLOAT,0);
+ class_sethelpsymbol (mtx_find_class, gensym("iemmatrix/mtx_find"));
+}
+
+void iemtx_find_setup(void){
+ mtx_find_setup();
+}
generated by cgit v1.2.1 (git 2.18.0) at 2024-04-19 00:57:24 +0000