aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/mtx_distance2.c107
1 files changed, 107 insertions, 0 deletions
diff --git a/src/mtx_distance2.c b/src/mtx_distance2.c
new file mode 100644
index 0000000..8d91c96
--- /dev/null
+++ b/src/mtx_distance2.c
@@ -0,0 +1,107 @@
+/*
+ * iemmatrix
+ *
+ * objects for manipulating simple matrices
+ * mostly refering to matlab/octave matrix functions
+ *
+ * (c) IOhannes m zmölnig, forum::für::umläute
+ *
+ * IEM, Graz
+ *
+ * this code is published under the LGPL
+ *
+ */
+#include "iemmatrix.h"
+
+/*
+ mtx_distance2: gets the euclidian distances (squared) between 2 sets of n-dimensional vectors
+*/
+
+/* -­------------------------------------------------------------- */
+/* matrix math */
+
+/* mtx_distance2 */
+static t_class *mtx_distance2_class;
+
+static void mtx_distance2_matrix(t_mtx_binmtx *x, t_symbol *s, int argc, t_atom *argv)
+{
+ int row=atom_getfloat(argv);
+ int col=atom_getfloat(argv+1);
+ int row2=0, col2=0;
+ t_atom *m;
+ t_atom *m1 = argv+2;
+ t_atom *m2 = x->m2.atombuffer+2;
+ int i, j;
+
+ if (argc<2){
+ post("mtx_distance2: crippled matrix");
+ return;
+ }
+ if ((col<1)||(row<1)) {
+ post("mtx_distance2: invalid dimensions");
+ return;
+ }
+ if (col*row>argc-2){
+ post("sparse matrix not yet supported : use \"mtx_check\"");
+ return;
+ }
+
+ row2=x->m2.row;
+ col2=x->m2.col;
+
+ if (!(col2*row2)) {
+ /* 2nd matrix is NULL; take the 1st matrix instead (distance between it's own vectors) */
+ m2=argv+2;
+ row2=row;
+ col2=col;
+ } else if (col!=col2){
+ post("mtx_distance2: matrix dimensions do not match");
+ return;
+ }
+
+ adjustsize(&x->m, row, row2);
+ m = x->m.atombuffer+2;
+
+ for(i=0; i<row; i++)
+ for(j=0; j<row2; j++){
+ t_float f=0.f;
+ int c1=col*i;
+ int c2=col*j;
+
+ int n;
+ for(n=0; n<col; n++){
+ t_float x=atom_getfloat(&m1[c1+n])-atom_getfloat(&m2[c2+n]);
+ f+=x*x;
+ }
+ SETFLOAT(m,f);
+ m++;
+ }
+
+ outlet_anything(x->x_obj.ob_outlet, gensym("matrix"), row*row2+2, x->m.atombuffer);
+}
+static void *mtx_distance2_new(void)
+{
+ t_mtx_binmtx *x = (t_mtx_binmtx *)pd_new(mtx_distance2_class);
+ inlet_new(&x->x_obj, &x->x_obj.ob_pd, gensym("matrix"), gensym(""));
+ outlet_new(&x->x_obj, 0);
+ x->m.col = x->m.row = x->m2.col = x->m2.row = 0;
+ x->m.atombuffer = x->m2.atombuffer = 0;
+ return(x);
+}
+
+void mtx_distance2_setup(void)
+{
+ mtx_distance2_class = class_new(gensym("mtx_distance2"),
+ (t_newmethod)mtx_distance2_new, (t_method)mtx_binmtx_free,
+ sizeof(t_mtx_binmtx), 0, A_NULL, 0);
+ class_addmethod(mtx_distance2_class, (t_method)mtx_distance2_matrix, gensym("matrix"), A_GIMME, 0);
+ class_addmethod(mtx_distance2_class, (t_method)mtx_bin_matrix2, gensym(""), A_GIMME, 0);
+ class_addbang (mtx_distance2_class, mtx_binmtx_bang);
+
+ class_sethelpsymbol(mtx_distance2_class, gensym("iemmatrix/mtx_distasnce2"));
+}
+
+void iemtx_distance2_setup(void)
+{
+ mtx_distance2_setup();
+}