From ade55607696baa64b4bfa07529aec389b996e6e5 Mon Sep 17 00:00:00 2001
From: Scott Klum <scott.klum@gmail.com>
Date: Thu, 19 Feb 2015 16:19:32 -0500
Subject: [PATCH] Added heatmap

---
 openbr/plugins/imgproc/heatmap.cpp | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+), 0 deletions(-)
 create mode 100644 openbr/plugins/imgproc/heatmap.cpp

diff --git a/openbr/plugins/imgproc/heatmap.cpp b/openbr/plugins/imgproc/heatmap.cpp
new file mode 100644
index 0000000..b0ceb4a
--- /dev/null
+++ b/openbr/plugins/imgproc/heatmap.cpp
@@ -0,0 +1,39 @@
+#include "openbr/plugins/openbr_internal.h"
+#include "openbr/core/opencvutils.h"
+
+#include <opencv2/ml/ml.hpp>
+#include <opencv2/contrib/contrib.hpp>
+
+using namespace cv;
+
+namespace br
+{
+
+class HeatmapTransform : public UntrainableTransform
+{
+    Q_OBJECT
+
+    Q_PROPERTY(float alpha READ get_alpha WRITE set_alpha RESET reset_alpha STORED false)
+    BR_PROPERTY(float, alpha, .5)
+
+    void project(const Template &src, Template &dst) const
+    {
+        dst = src;
+
+        const float beta = 1-alpha;
+
+        Mat buffer;
+        applyColorMap(src,buffer,COLORMAP_JET);
+
+        Mat colored;
+        cvtColor(src, colored, CV_GRAY2BGR);
+
+        addWeighted(colored,alpha,buffer,beta,0.0,dst);
+    }
+};
+
+BR_REGISTER(Transform, HeatmapTransform)
+
+} // namespace br
+
+#include "imgproc/heatmap.moc"
--
libgit2 0.21.4