added the centerofgravity function

MPIBR · Sep 21, 2016 · 770e662 · 770e662
1 parent 9f5231e
commit 770e662
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/centerofgravity.m b/centerofgravity.m
@@ -0,0 +1,21 @@
+function cog = centerofgravity(d,thr)
+%Determine center of mass of a matrix
+% cog = centerofgravity(d,thr)
+
+if nargin == 2
+    d(d<thr) = 0;
+end
+d=double(d);
+
+
+
+for dim = 1:numel(size(d))
+    dd = d;
+    ind = 1:numel(size(d));
+    ind(dim) = [];
+    dd = permute(dd,[dim ind]);
+    dd = reshape(dd,size(d,dim),[]);
+    dsum=nansum(dd,2);
+    distance = 1:size(d,dim);
+    cog(dim) = nansum(dsum(:).*distance(:)) / nansum(dsum);
+end