diff maths/CosineDistance.cpp @ 256:43943a4382ef

* Add cosine distance and the self-similarity matrix used for SB rhythmic similarity * Pull out SB timbral similarity KL divergence into its own file
author Chris Cannam <c.cannam@qmul.ac.uk>
date Fri, 18 Jan 2008 14:40:20 +0000
parents
children 769da847732b
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/maths/CosineDistance.cpp	Fri Jan 18 14:40:20 2008 +0000
@@ -0,0 +1,47 @@
+/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
+
+/*
+    QM DSP Library
+
+    Centre for Digital Music, Queen Mary, University of London.
+    This file copyright 2008 Kurt Jacobson.
+    All rights reserved.
+*/
+
+#include "CosineDistance.h"
+
+#include <iostream>
+#include <limits>
+
+using std::cerr;
+
+double CosineDistance::distance(const vector<double> &v1,
+                                const vector<double> &v2)
+{
+    dist = 1.0; dDenTot = 0; dDen1 = 0; dDen2 = 0; dSum1 =0;
+
+    //check if v1, v2 same size
+    if (v1.size() != v2.size())
+    {
+        cerr << "CosineDistance::distance: ERROR: vectors not the same size\n";
+        return 1.0;
+    }
+    else
+    {
+        for(int i=0; i<v1.size(); i++)
+        {
+            dSum1 += v1[i]*v2[i];
+            dDen1 += v1[i]*v1[i];
+            dDen2 += v2[i]*v2[i];
+        }
+        dDenTot = sqrt(fabs(dDen1*dDen2));
+        if(dDenTot == 0)
+        {
+            cerr << "CosineDistance::distance: WARNING: dividing by zero in cosine dist\n";
+            return 1.0;
+        }
+	
+        dist = 1-((dSum1)/dDenTot);
+        return dist;
+    }
+}