From c5f168d6af6b3ee55baf78d5db75b994f8dbfc02 Mon Sep 17 00:00:00 2001
From: Josh Klontz <josh.klontz@gmail.com>
Date: Fri, 15 Mar 2013 20:13:29 -0400
Subject: [PATCH] added more plugins

---
 sdk/plugins/algorithms.cpp |  2 +-
 sdk/plugins/sentence.cpp   | 85 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 86 insertions(+), 1 deletion(-)
 create mode 100644 sdk/plugins/sentence.cpp

diff --git a/sdk/plugins/algorithms.cpp b/sdk/plugins/algorithms.cpp
index cd513b2..b7a9a95 100644
--- a/sdk/plugins/algorithms.cpp
+++ b/sdk/plugins/algorithms.cpp
@@ -50,7 +50,7 @@ class AlgorithmsInitializer : public Initializer
         Globals->abbreviations.insert("SmallSIFT", "Open+LimitSize(512)+KeyPointDetector(SIFT)+KeyPointDescriptor(SIFT):KeyPointMatcher(BruteForce)");
         Globals->abbreviations.insert("SmallSURF", "Open+LimitSize(512)+KeyPointDetector(SURF)+KeyPointDescriptor(SURF):KeyPointMatcher(BruteForce)");
         Globals->abbreviations.insert("ColorHist", "Open+LimitSize(512)!EnsureChannels(3)+SplitChannels+Hist(256,0,8)+Cat+Normalize(L1):L2");
-        Globals->abbreviations.insert("IHH", "Open+(RG+MAdd(0.5))/(Cvt(Gray)+Gradient+Bin(0,360,8,true))+Merge+Integral+IntegralSampler+CvtFloat+WordWise(RowWisePCA(8)+RowWiseMeanCenter+Binarize,RowWisePCA):L2");
+        Globals->abbreviations.insert("IHH", "Open+(RG+MAdd(0.5))/(Cvt(Gray)+Gradient+Bin(0,360,8,true))+Merge+Integral+IntegralSampler+CvtFloat+WordWise(RowWisePCA(8)+RowWiseMeanCenter+Binarize,RowWisePCA)+Sentence:SentenceSimilarity");
 
         // Hash
         Globals->abbreviations.insert("FileName", "Name+Identity:Identical");
diff --git a/sdk/plugins/sentence.cpp b/sdk/plugins/sentence.cpp
new file mode 100644
index 0000000..628cbaf
--- /dev/null
+++ b/sdk/plugins/sentence.cpp
@@ -0,0 +1,85 @@
+#include <openbr_plugin.h>
+
+using namespace cv;
+
+namespace br
+{
+
+/*!
+ * \ingroup transforms
+ * \brief Ordered words
+ * \author Josh Klontz \cite jklontz
+ */
+class SentenceTransform : public UntrainableMetaTransform
+{
+    Q_OBJECT
+
+    void project(const Template &src, Template &dst) const
+    {
+        QByteArray sentence;
+        QDataStream stream(&sentence, QIODevice::WriteOnly);
+        for (int i=0; i<src.size(); i++) {
+            const Mat &m = src[i];
+            if (!m.data) continue;
+            stream << i << m.rows << m.cols;
+            stream.writeRawData((const char*)m.data, 4*m.rows*m.cols);
+        }
+        dst.file = src.file;
+        dst.m() = Mat(1, sentence.size(), CV_8UC1, sentence.data()).clone();
+    }
+};
+
+BR_REGISTER(Transform, SentenceTransform)
+
+/*!
+ * \ingroup distances
+ * \brief Distance between sentences
+ * \author Josh Klontz \cite jklontz
+ */
+class SentenceSimilarityDistance : public Distance
+{
+    Q_OBJECT
+
+    float compare(const Template &a, const Template &b) const
+    {
+        const int aSize = a.m().cols;
+        const int bSize = b.m().cols;
+        uchar *aData = a.m().data;
+        uchar *bData = b.m().data;
+        const uchar *aEnd = aData + aSize;
+        const uchar *bEnd = bData + bSize;
+
+        int32_t aWord, bWord, aRows, bRows, aColumns, bColumns;
+        aWord = aRows = aColumns = -2;
+        bWord = bRows = bColumns = -1;
+        float distance = 0;
+        int comparisons = 0;
+        while ((aData != aEnd) && (bData != bEnd)) {
+            if (aWord < bWord) {
+                aWord = *reinterpret_cast<int32_t*>(aData);
+                aRows = *reinterpret_cast<int32_t*>(aData+4);
+                aColumns = *reinterpret_cast<int32_t*>(aData+8);
+                aData += 12;
+            } else if (bWord < aWord) {
+                bWord = *reinterpret_cast<int32_t*>(bData);
+                bRows = *reinterpret_cast<int32_t*>(bData+4);
+                bColumns = *reinterpret_cast<int32_t*>(bData+8);
+                bData += 12;
+            } else {
+                for (int i=0; i<aRows; i++)
+                    for (int j=0; j<bRows; j++)
+                        for (int k=0; k<aColumns; k++)
+                            distance += *reinterpret_cast<float*>(aData+4*k) * *reinterpret_cast<float*>(bData+4*k);
+                comparisons += aRows * bRows;
+            }
+        }
+
+        return comparisons / distance;
+    }
+};
+
+BR_REGISTER(Distance, SentenceSimilarityDistance)
+
+} // namespace br
+
+#include "sentence.moc"
--
libgit2 0.21.4