16 files changed, 934 insertions, 0 deletions
diff --git a/tracker-points/module/CMakeLists.txt b/tracker-points/module/CMakeLists.txt
new file mode 100644
index 00000000..1eec9616
--- /dev/null
+++ b/tracker-points/module/CMakeLists.txt
@@ -0,0 +1,6 @@
+find_package(OpenCV QUIET)
+if(OpenCV_FOUND)
+    otr_module(tracker-points)
+    target_link_libraries(${self} opentrack-tracker-points-base)
+    target_include_directories(${self} PUBLIC "${CMAKE_SOURCE_DIR}/tracker-pt")
+endif()
diff --git a/tracker-points/module/Resources/Logo_IR.png b/tracker-points/module/Resources/Logo_IR.png
new file mode 100644
index 00000000..95032a25
--- /dev/null
+++ b/tracker-points/module/Resources/Logo_IR.png
diff --git a/tracker-points/module/camera.cpp b/tracker-points/module/camera.cpp
new file mode 100644
index 00000000..a70698de
--- /dev/null
+++ b/tracker-points/module/camera.cpp
@@ -0,0 +1,152 @@
+/* Copyright (c) 2012 Patrick Ruoff
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ */
+
+#include "camera.h"
+#include "frame.hpp"
+
+#include "compat/math-imports.hpp"
+
+#include <opencv2/core.hpp>
+
+namespace pt_module {
+
+Camera::Camera(const QString& module_name) : s { module_name }
+{
+}
+
+QString Camera::get_desired_name() const
+{
+    return cam_desired.name;
+}
+
+QString Camera::get_active_name() const
+{
+    return cam_info.name;
+}
+
+void Camera::show_camera_settings()
+{
+    if (cap)
+        (void)cap->show_dialog();
+}
+
+Camera::result Camera::get_info() const
+{
+    if (cam_info.res_x == 0 || cam_info.res_y == 0)
+        return { false, pt_camera_info() };
+    else
+        return { true, cam_info };
+}
+
+Camera::result Camera::get_frame(pt_frame& frame_)
+{
+    cv::Mat& frame = frame_.as<Frame>()->mat;
+
+    const bool new_frame = get_frame_(frame);
+
+    if (new_frame)
+    {
+        const f dt = (f)t.elapsed_seconds();
+        t.start();
+
+        // measure fps of valid frames
+        constexpr f RC = f{1}/10; // seconds
+        const f alpha = dt/(dt + RC);
+
+        if (dt_mean < dt_eps)
+            dt_mean = dt;
+        else
+            dt_mean = (1-alpha) * dt_mean + alpha * dt;
+
+        cam_info.fps = dt_mean > dt_eps ? 1 / dt_mean : 0;
+        cam_info.res_x = frame.cols;
+        cam_info.res_y = frame.rows;
+        cam_info.fov = fov;
+
+        return { true, cam_info };
+    }
+    else
+        return { false, {} };
+}
+
+bool Camera::start(const QString& name, int fps, int res_x, int res_y)
+{
+    if (fps >= 0 && res_x >= 0 && res_y >= 0)
+    {
+        if (cam_desired.name != name ||
+            (int)cam_desired.fps != fps ||
+            cam_desired.res_x != res_x ||
+            cam_desired.res_y != res_y ||
+            !cap || !cap->is_open())
+        {
+            stop();
+
+            cam_desired.name = name;
+            cam_desired.fps = fps;
+            cam_desired.res_x = res_x;
+            cam_desired.res_y = res_y;
+            cam_desired.fov = fov;
+
+            cap = video::make_camera(name);
+
+            if (!cap)
+                goto fail;
+
+            camera::info info {};
+            info.fps = fps;
+            info.width = res_x;
+            info.height = res_y;
+
+            if (!cap->start(info))
+                goto fail;
+
+            cam_info = pt_camera_info();
+            cam_info.name = name;
+            dt_mean = 0;
+
+            cv::Mat tmp;
+
+            if (!get_frame_(tmp))
+                goto fail;
+
+            t.start();
+        }
+    }
+
+    return true;
+
+fail:
+    stop();
+    return false;
+}
+
+void Camera::stop()
+{
+    cap = nullptr;
+    cam_info = {};
+    cam_desired = {};
+}
+
+bool Camera::get_frame_(cv::Mat& img)
+{
+    if (cap && cap->is_open())
+    {
+        auto [ frame, ret ] = cap->get_frame();
+        if (ret)
+        {
+            int stride = frame.stride;
+            if (stride == 0)
+                stride = cv::Mat::AUTO_STEP;
+            img = cv::Mat(frame.height, frame.width, CV_8UC(frame.channels), (void*)frame.data, stride);
+            return true;
+        }
+    }
+
+    return false;
+}
+
+} // ns pt_module
diff --git a/tracker-points/module/camera.h b/tracker-points/module/camera.h
new file mode 100644
index 00000000..02e2fe4d
--- /dev/null
+++ b/tracker-points/module/camera.h
@@ -0,0 +1,55 @@
+/* Copyright (c) 2012 Patrick Ruoff
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ */
+
+#pragma once
+
+#include "pt-api.hpp"
+#include "compat/timer.hpp"
+#include "video/camera.hpp"
+
+#include <memory>
+
+#include <opencv2/core.hpp>
+
+#include <QString>
+
+namespace pt_module {
+
+struct Camera final : pt_camera
+{
+    Camera(const QString& module_name);
+
+    bool start(const QString& name, int fps, int res_x, int res_y) override;
+    void stop() override;
+
+    result get_frame(pt_frame& Frame) override;
+    result get_info() const override;
+
+    pt_camera_info get_desired() const override { return cam_desired; }
+    QString get_desired_name() const override;
+    QString get_active_name() const override;
+
+    void set_fov(f value) override { fov = value; }
+    void show_camera_settings() override;
+
+private:
+    using camera = typename video::impl::camera;
+
+    [[nodiscard]] bool get_frame_(cv::Mat& frame);
+
+    f dt_mean = 0, fov = 30;
+    Timer t;
+    pt_camera_info cam_info;
+    pt_camera_info cam_desired;
+
+    std::unique_ptr<camera> cap;
+    pt_settings s;
+
+    static constexpr f dt_eps = f{1}/256;
+};
+
+} // ns pt_module
diff --git a/tracker-points/module/export.hpp b/tracker-points/module/export.hpp
new file mode 100644
index 00000000..a733c9fe
--- /dev/null
+++ b/tracker-points/module/export.hpp
@@ -0,0 +1,11 @@
+// generates export.hpp for each module from compat/linkage.hpp
+
+#pragma once
+
+#include "compat/linkage-macros.hpp"
+
+#ifdef BUILD_TRACKER_PT
+#   define OTR_PT_EXPORT OTR_GENERIC_EXPORT
+#else
+#   define OTR_PT_EXPORT OTR_GENERIC_IMPORT
+#endif
diff --git a/tracker-points/module/frame.cpp b/tracker-points/module/frame.cpp
new file mode 100644
index 00000000..a045b783
--- /dev/null
+++ b/tracker-points/module/frame.cpp
@@ -0,0 +1,80 @@
+#include "frame.hpp"
+
+#include "compat/math.hpp"
+
+#include <opencv2/imgproc.hpp>
+
+namespace pt_module {
+
+Preview& Preview::operator=(const pt_frame& frame_)
+{
+    const cv::Mat& frame = frame_.as_const<const Frame>()->mat;
+
+    if (frame.channels() != 3)
+    {
+        eval_once(qDebug() << "tracker/pt: camera frame depth: 3 !=" << frame.channels());
+        return *this;
+    }
+
+    const bool need_resize = frame.cols != frame_out.cols || frame.rows != frame_out.rows;
+    if (need_resize)
+        cv::resize(frame, frame_copy, cv::Size(frame_out.cols, frame_out.rows), 0, 0, cv::INTER_NEAREST);
+    else
+        frame.copyTo(frame_copy);
+
+    return *this;
+}
+
+Preview::Preview(int w, int h)
+{
+    ensure_size(frame_out, w, h, CV_8UC4);
+    ensure_size(frame_copy, w, h, CV_8UC3);
+
+    frame_copy.setTo(cv::Scalar(0, 0, 0));
+}
+
+QImage Preview::get_bitmap()
+{
+    int stride = frame_out.step.p[0];
+
+    if (stride < 64 || stride < frame_out.cols * 4)
+    {
+        eval_once(qDebug() << "bad stride" << stride
+                           << "for bitmap size" << frame_copy.cols << frame_copy.rows);
+        return QImage();
+    }
+
+    cv::cvtColor(frame_copy, frame_out, cv::COLOR_BGR2BGRA);
+
+    return QImage((const unsigned char*) frame_out.data,
+                  frame_out.cols, frame_out.rows,
+                  stride,
+                  QImage::Format_ARGB32);
+}
+
+void Preview::draw_head_center(f x, f y)
+{
+    auto [px_, py_] = to_pixel_pos(x, y, frame_copy.cols, frame_copy.rows);
+
+    int px = iround(px_), py = iround(py_);
+
+    constexpr int len = 9;
+
+    static const cv::Scalar color(0, 255, 255);
+    cv::line(frame_copy,
+             cv::Point(px - len, py),
+             cv::Point(px + len, py),
+             color, 1);
+    cv::line(frame_copy,
+             cv::Point(px, py - len),
+             cv::Point(px, py + len),
+             color, 1);
+}
+
+void Preview::ensure_size(cv::Mat& frame, int w, int h, int type)
+{
+    if (frame.cols != w || frame.rows != h)
+        frame = cv::Mat(h, w, type);
+}
+
+} // ns pt_module
diff --git a/tracker-points/module/frame.hpp b/tracker-points/module/frame.hpp
new file mode 100644
index 00000000..89334599
--- /dev/null
+++ b/tracker-points/module/frame.hpp
@@ -0,0 +1,44 @@
+#pragma once
+
+#include "pt-api.hpp"
+
+#include <opencv2/core.hpp>
+#include <QImage>
+
+#ifdef __clang__
+#   pragma clang diagnostic push
+#   pragma clang diagnostic ignored "-Wweak-vtables"
+#endif
+
+namespace pt_module {
+
+struct Frame final : pt_frame
+{
+    cv::Mat mat;
+
+    operator const cv::Mat&() const& { return mat; }
+    operator cv::Mat&() & { return mat; }
+};
+
+struct Preview final : pt_preview
+{
+    Preview(int w, int h);
+
+    Preview& operator=(const pt_frame& frame) override;
+    QImage get_bitmap() override;
+    void draw_head_center(f x, f y) override;
+
+    operator cv::Mat&() { return frame_copy; }
+    operator cv::Mat const&() const { return frame_copy; }
+
+private:
+    static void ensure_size(cv::Mat& frame, int w, int h, int type);
+
+    cv::Mat frame_copy, frame_out;
+};
+
+} // ns pt_module
+
+#ifdef __clang__
+#   pragma clang diagnostic pop
+#endif
diff --git a/tracker-points/module/lang/nl_NL.ts b/tracker-points/module/lang/nl_NL.ts
new file mode 100644
index 00000000..aaaead85
--- /dev/null
+++ b/tracker-points/module/lang/nl_NL.ts
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!DOCTYPE TS>
+<TS version="2.1" language="nl_NL">
+<context>
+    <name>pt_module::metadata_pt</name>
+    <message>
+        <source>Points Tracker 0.1</source>
+        <translation type="unfinished"></translation>
+    </message>
+</context>
+</TS>
diff --git a/tracker-points/module/lang/ru_RU.ts b/tracker-points/module/lang/ru_RU.ts
new file mode 100644
index 00000000..b71de82c
--- /dev/null
+++ b/tracker-points/module/lang/ru_RU.ts
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!DOCTYPE TS>
+<TS version="2.1" language="ru_RU">
+<context>
+    <name>pt_module::metadata_pt</name>
+    <message>
+        <source>Points Tracker 0.1</source>
+        <translation type="unfinished"></translation>
+    </message>
+</context>
+</TS>
diff --git a/tracker-points/module/lang/stub.ts b/tracker-points/module/lang/stub.ts
new file mode 100644
index 00000000..9a27c78c
--- /dev/null
+++ b/tracker-points/module/lang/stub.ts
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!DOCTYPE TS>
+<TS version="2.1">
+<context>
+    <name>pt_module::metadata_pt</name>
+    <message>
+        <source>Points Tracker 0.1</source>
+        <translation type="unfinished"></translation>
+    </message>
+</context>
+</TS>
diff --git a/tracker-points/module/lang/zh_CN.ts b/tracker-points/module/lang/zh_CN.ts
new file mode 100644
index 00000000..9a27c78c
--- /dev/null
+++ b/tracker-points/module/lang/zh_CN.ts
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!DOCTYPE TS>
+<TS version="2.1">
+<context>
+    <name>pt_module::metadata_pt</name>
+    <message>
+        <source>Points Tracker 0.1</source>
+        <translation type="unfinished"></translation>
+    </message>
+</context>
+</TS>
diff --git a/tracker-points/module/module.cpp b/tracker-points/module/module.cpp
new file mode 100644
index 00000000..06cd003b
--- /dev/null
+++ b/tracker-points/module/module.cpp
@@ -0,0 +1,72 @@
+#include "ftnoir_tracker_pt.h"
+
+#include "module.hpp"
+#include "camera.h"
+#include "frame.hpp"
+#include "point_extractor.h"
+#include "ftnoir_tracker_pt_dialog.h"
+
+#include "pt-api.hpp"
+
+#include <memory>
+
+static const QString module_name = "tracker-pt";
+
+#ifdef __clang__
+#   pragma clang diagnostic ignored "-Wweak-vtables"
+#endif
+
+namespace pt_module {
+
+struct pt_module_traits final : pt_runtime_traits
+{
+    pointer<pt_camera> make_camera() const override
+    {
+        return pointer<pt_camera>(new Camera(module_name));
+    }
+
+    pointer<pt_point_extractor> make_point_extractor() const override
+    {
+        return pointer<pt_point_extractor>(new PointExtractor(module_name));
+    }
+
+    QString get_module_name() const override
+    {
+        return module_name;
+    }
+
+    pointer<pt_frame> make_frame() const override
+    {
+        return pointer<pt_frame>(new Frame);
+    }
+
+    pointer<pt_preview> make_preview(int w, int h) const override
+    {
+        return pointer<pt_preview>(new Preview(w, h));
+    }
+};
+
+struct tracker_pt : Tracker_PT
+{
+    tracker_pt() : Tracker_PT(pointer<pt_runtime_traits>(new pt_module_traits))
+    {
+    }
+};
+
+struct dialog_pt : TrackerDialog_PT
+{
+    dialog_pt();
+};
+
+dialog_pt::dialog_pt() : TrackerDialog_PT(module_name) {}
+
+QString metadata_pt::name() { return tr("Points Tracker 0.1"); }
+QIcon metadata_pt::icon() { return QIcon(":/Resources/Logo_IR.png"); }
+
+}
+
+// ns pt_module
+
+using namespace pt_module;
+
+OPENTRACK_DECLARE_TRACKER(tracker_pt, dialog_pt, metadata_pt)
diff --git a/tracker-points/module/module.hpp b/tracker-points/module/module.hpp
new file mode 100644
index 00000000..0b3f12cf
--- /dev/null
+++ b/tracker-points/module/module.hpp
@@ -0,0 +1,20 @@
+#pragma once
+
+#include "api/plugin-api.hpp"
+#include <QIcon>
+#include <QString>
+
+#include "compat/linkage-macros.hpp"
+
+namespace pt_module
+{
+
+class OTR_GENERIC_EXPORT metadata_pt : public Metadata
+{
+    Q_OBJECT
+
+    QString name() override;
+    QIcon icon() override;
+};
+
+} // ns pt_module
diff --git a/tracker-points/module/point_extractor.cpp b/tracker-points/module/point_extractor.cpp
new file mode 100644
index 00000000..1a75a3e3
--- /dev/null
+++ b/tracker-points/module/point_extractor.cpp
@@ -0,0 +1,387 @@
+/* Copyright (c) 2012 Patrick Ruoff
+ * Copyright (c) 2015-2017 Stanislaw Halik <sthalik@misaki.pl>
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ */
+
+#include "point_extractor.h"
+#include "point_tracker.h"
+#include "frame.hpp"
+
+#include "cv/numeric.hpp"
+#include "compat/math.hpp"
+
+#undef PREVIEW
+//#define PREVIEW
+
+#if defined PREVIEW
+#   include <opencv2/highgui.hpp>
+#endif
+
+#include <cmath>
+#include <algorithm>
+#include <cinttypes>
+#include <memory>
+
+#include <QDebug>
+
+using namespace numeric_types;
+
+// meanshift code written by Michael Welter
+
+/*
+http://en.wikipedia.org/wiki/Mean-shift
+In this application the idea, is to eliminate any bias of the point estimate 
+which is introduced by the rather arbitrary thresholded area. One must recognize
+that the thresholded area can only move in one pixel increments since it is
+binary. Thus, its center of mass might make "jumps" as pixels are added/removed
+from the thresholded area.
+With mean-shift, a moving "window" or kernel is multiplied with the gray-scale
+image, and the COM is calculated of the result. This is iterated where the
+kernel center is set the previously computed COM. Thus, peaks in the image intensity
+distribution "pull" the kernel towards themselves. Eventually it stops moving, i.e.
+then the computed COM coincides with the kernel center. We hope that the 
+corresponding location is a good candidate for the extracted point.
+The idea similar to the window scaling suggested in  Berglund et al. "Fast, bias-free 
+algorithm for tracking single particles with variable size and shape." (2008).
+*/
+static vec2 MeanShiftIteration(const cv::Mat1b &frame_gray, const vec2 &current_center, f filter_width)
+{
+    const f s = 1 / filter_width;
+
+    f m = 0;
+    vec2 com { 0, 0 };
+    for (int i = 0; i < frame_gray.rows; i++)
+    {
+        uint8_t const* const __restrict frame_ptr = frame_gray.ptr(i);
+        for (int j = 0; j < frame_gray.cols; j++)
+        {
+            f val = frame_ptr[j];
+            val = val * val; // taking the square weighs brighter parts of the image stronger.
+            f dx = (j - current_center[0])*s;
+            f dy = (i - current_center[1])*s;
+            f max = std::fmax(f(0), 1 - dx*dx - dy*dy);
+            val *= max;
+            m += val;
+            com[0] += j * val;
+            com[1] += i * val;
+        }
+    }
+    if (m > f(.1))
+    {
+        com *= 1 / m;
+        return com;
+    }
+    else
+        return current_center;
+}
+
+namespace pt_module {
+
+PointExtractor::PointExtractor(const QString& module_name) : s(module_name)
+{
+    blobs.reserve(max_blobs);
+}
+
+void PointExtractor::ensure_channel_buffers(const cv::Mat& orig_frame)
+{
+    if (ch[0].rows != orig_frame.rows || ch[0].cols != orig_frame.cols)
+        for (cv::Mat1b& x : ch)
+            x = cv::Mat1b(orig_frame.rows, orig_frame.cols);
+}
+
+void PointExtractor::ensure_buffers(const cv::Mat& frame)
+{
+    const int W = frame.cols, H = frame.rows;
+
+    if (frame_gray.rows != W || frame_gray.cols != H)
+    {
+        frame_gray = cv::Mat1b(H, W);
+        frame_bin = cv::Mat1b(H, W);
+        frame_gray_unmasked = cv::Mat1b(H, W);
+    }
+}
+
+void PointExtractor::extract_single_channel(const cv::Mat& orig_frame, int idx, cv::Mat1b& dest)
+{
+    ensure_channel_buffers(orig_frame);
+
+    const int from_to[] = {
+        idx, 0,
+    };
+
+    cv::mixChannels(&orig_frame, 1, &dest, 1, from_to, 1);
+}
+
+void PointExtractor::color_to_grayscale(const cv::Mat& frame, cv::Mat1b& output)
+{
+    switch (s.blob_color)
+    {
+    case pt_color_green_only:
+    {
+        extract_single_channel(frame, 1, output);
+        break;
+    }
+    case pt_color_blue_only:
+    {
+        extract_single_channel(frame, 0, output);
+        break;
+    }
+    case pt_color_red_only:
+    {
+        extract_single_channel(frame, 2, output);
+        break;
+    }
+    case pt_color_average:
+    {
+        const int W = frame.cols, H = frame.rows, sz = W*H;
+        cv::reduce(frame.reshape(1, sz),
+                   output.reshape(1, sz),
+                   1, cv::REDUCE_AVG);
+        break;
+    }
+    default:
+        eval_once(qDebug() << "wrong pt_color_type enum value" << int(s.blob_color));
+    [[fallthrough]];
+    case pt_color_natural:
+        cv::cvtColor(frame, output, cv::COLOR_BGR2GRAY);
+        break;
+    }
+}
+
+void PointExtractor::threshold_image(const cv::Mat& frame_gray, cv::Mat1b& output)
+{
+    const int threshold_slider_value = s.threshold_slider.to<int>();
+
+    if (!s.auto_threshold)
+    {
+        cv::threshold(frame_gray, output, threshold_slider_value, 255, cv::THRESH_BINARY);
+    }
+    else
+    {
+        const int hist_size = 256;
+        const float ranges_[] = { 0, 256 };
+        float const* ranges = (const float*) ranges_;
+
+        cv::calcHist(&frame_gray,
+                     1,
+                     nullptr,
+                     cv::noArray(),
+                     hist,
+                     1,
+                     &hist_size,
+                     &ranges);
+
+        const f radius = threshold_radius_value(frame_gray.cols, frame_gray.rows, threshold_slider_value);
+
+        float const* const __restrict ptr = hist.ptr<float>(0);
+        const unsigned area = uround(3 * pi * radius*radius);
+        const unsigned sz = unsigned(hist.cols * hist.rows);
+        constexpr unsigned min_thres = 64;
+        unsigned thres = min_thres;
+        for (unsigned i = sz-1, cnt = 0; i > 32; i--)
+        {
+            cnt += (unsigned)ptr[i];
+            if (cnt >= area)
+                break;
+            thres = i;
+        }
+
+        cv::threshold(frame_gray, output, thres, 255, cv::THRESH_BINARY);
+    }
+}
+
+static void draw_blobs(cv::Mat& preview_frame, const blob* blobs, unsigned nblobs, const cv::Size& size)
+{
+    for (unsigned k = 0; k < nblobs; k++)
+    {
+        const blob& b = blobs[k];
+
+        if (b.radius < 0)
+            continue;
+
+        const f dpi = preview_frame.cols / f(320);
+        const f offx = 10 * dpi, offy = f(7.5) * dpi;
+
+        const f cx = preview_frame.cols / f(size.width),
+                cy = preview_frame.rows / f(size.height),
+                c  = std::fmax(f(1), cx+cy)/2;
+
+        constexpr unsigned fract_bits = 8;
+        constexpr int c_fract(1 << fract_bits);
+
+        cv::Point p(iround(b.pos[0] * cx * c_fract), iround(b.pos[1] * cy * c_fract));
+
+        auto circle_color = k >= PointModel::N_POINTS
+                            ? cv::Scalar(192, 192, 192)
+                            : cv::Scalar(255, 255, 0);
+
+        const int overlay_size = iround(dpi);
+
+        cv::circle(preview_frame, p, iround((b.radius + f(3.3) * c) * c_fract),
+                   circle_color, overlay_size,
+                   cv::LINE_AA, fract_bits);
+
+        char buf[16];
+        buf[sizeof(buf)-1] = '\0';
+        std::snprintf(buf, sizeof(buf) - 1, "%.2fpx", (double)b.radius);
+
+        auto text_color = k >= PointModel::N_POINTS
+                          ? cv::Scalar(160, 160, 160)
+                          : cv::Scalar(0, 0, 255);
+
+        cv::Point pos(iround(b.pos[0]*cx+offx), iround(b.pos[1]*cy+offy));
+        cv::putText(preview_frame, buf, pos,
+                    cv::FONT_HERSHEY_PLAIN, overlay_size, text_color,
+                    1);
+    }
+}
+
+void PointExtractor::extract_points(const pt_frame& frame_, pt_preview& preview_frame_, std::vector<vec2>& points)
+{
+    const cv::Mat& frame = frame_.as_const<Frame>()->mat;
+
+    ensure_buffers(frame);
+    color_to_grayscale(frame, frame_gray_unmasked);
+
+#if defined PREVIEW
+    cv::imshow("capture", frame_gray);
+    cv::waitKey(1);
+#endif
+
+    threshold_image(frame_gray_unmasked, frame_bin);
+    frame_gray_unmasked.copyTo(frame_gray, frame_bin);
+
+    const f region_size_min = (f)s.min_point_size;
+    const f region_size_max = (f)s.max_point_size;
+
+    unsigned idx = 0;
+
+    blobs.clear();
+
+    for (int y=0; y < frame_bin.rows; y++)
+    {
+        const unsigned char* __restrict ptr_bin = frame_bin.ptr(y);
+        for (int x=0; x < frame_bin.cols; x++)
+        {
+            if (ptr_bin[x] != 255)
+                continue;
+            idx = blobs.size() + 1;
+
+            cv::Rect rect;
+            cv::floodFill(frame_bin,
+                          cv::Point(x,y),
+                          cv::Scalar(idx),
+                          &rect,
+                          cv::Scalar(0),
+                          cv::Scalar(0),
+                          4 | cv::FLOODFILL_FIXED_RANGE);
+
+            unsigned cnt = 0;
+            unsigned norm = 0;
+
+            const int ymax = rect.y+rect.height,
+                      xmax = rect.x+rect.width;
+
+            for (int i=rect.y; i < ymax; i++)
+            {
+                unsigned char const* const __restrict ptr_blobs = frame_bin.ptr(i);
+                unsigned char const* const __restrict ptr_gray = frame_gray.ptr(i);
+                for (int j=rect.x; j < xmax; j++)
+                {
+                    if (ptr_blobs[j] != idx)
+                        continue;
+
+                    //ptr_blobs[j] = 0;
+                    norm += ptr_gray[j];
+                    cnt++;
+                }
+            }
+
+            const f radius = std::sqrt(cnt / pi);
+            if (radius > region_size_max || radius < region_size_min)
+                continue;
+
+            blobs.emplace_back(radius,
+                               vec2(rect.width/f(2), rect.height/f(2)),
+                               std::pow(f(norm), f(1.1))/cnt,
+                               rect);
+
+            if (idx >= max_blobs)
+                goto end;
+
+            // XXX we could go to the next scanline unless the points are really small.
+            // i'd expect each point being present on at least one unique scanline
+            // but it turns out some people are using 2px points -sh 20180110
+            //break;
+        }
+    }
+end:
+
+    const int W = frame_gray.cols;
+    const int H = frame_gray.rows;
+
+    const unsigned sz = blobs.size();
+
+    std::sort(blobs.begin(), blobs.end(), [](const blob& b1, const blob& b2) { return b2.brightness < b1.brightness; });
+
+    for (idx = 0; idx < sz; ++idx)
+    {
+        blob& b = blobs[idx];
+        cv::Rect rect = b.rect;
+
+        rect.x -= rect.width / 2;
+        rect.y -= rect.height / 2;
+        rect.width *= 2;
+        rect.height *= 2;
+        rect &= cv::Rect(0, 0, W, H);  // crop at frame boundaries
+
+        cv::Mat frame_roi = frame_gray(rect);
+
+        // smaller values mean more changes. 1 makes too many changes while 1.5 makes about .1
+        static constexpr f radius_c = f(1.75);
+
+        const f kernel_radius = b.radius * radius_c;
+        vec2 pos(rect.width/f(2), rect.height/f(2)); // position relative to ROI.
+
+        for (int iter = 0; iter < 10; ++iter)
+        {
+            vec2 com_new = MeanShiftIteration(frame_roi, pos, kernel_radius);
+            vec2 delta = com_new - pos;
+            pos = com_new;
+            if (delta.dot(delta) < f(1e-3))
+                break;
+        }
+
+        b.pos[0] = pos[0] + rect.x;
+        b.pos[1] = pos[1] + rect.y;
+    }
+
+    draw_blobs(preview_frame_.as<Frame>()->mat,
+               blobs.data(), blobs.size(),
+               frame_gray.size());
+
+
+    // End of mean shift code. At this point, blob positions are updated with hopefully less noisy less biased values.
+    points.reserve(max_blobs);
+    points.clear();
+
+    for (const auto& b : blobs)
+    {
+        // note: H/W is equal to fx/fy
+
+        vec2 p;
+        std::tie(p[0], p[1]) = to_screen_pos(b.pos[0], b.pos[1], W, H);
+        points.push_back(p);
+    }
+}
+
+blob::blob(f radius, const vec2& pos, f brightness, const cv::Rect& rect) :
+    radius(radius), brightness(brightness), pos(pos), rect(rect)
+{
+    //qDebug() << "radius" << radius << "pos" << pos[0] << pos[1];
+}
+
+} // ns pt_module
diff --git a/tracker-points/module/point_extractor.h b/tracker-points/module/point_extractor.h
new file mode 100644
index 00000000..a6103667
--- /dev/null
+++ b/tracker-points/module/point_extractor.h
@@ -0,0 +1,58 @@
+/* Copyright (c) 2012 Patrick Ruoff
+ * Copyright (c) 2015-2016 Stanislaw Halik <sthalik@misaki.pl>
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ */
+
+#pragma once
+
+#include "pt-api.hpp"
+
+#include <vector>
+
+#include <opencv2/core.hpp>
+#include <opencv2/imgproc.hpp>
+
+namespace pt_module {
+
+using namespace numeric_types;
+
+struct blob final
+{
+    f radius, brightness;
+    vec2 pos;
+    cv::Rect rect;
+
+    blob(f radius, const vec2& pos, f brightness, const cv::Rect& rect);
+};
+
+class PointExtractor final : public pt_point_extractor
+{
+public:
+    // extracts points from frame and draws some processing info into frame, if draw_output is set
+    // dt: time since last call in seconds
+    void extract_points(const pt_frame& frame, pt_preview& preview_frame, std::vector<vec2>& points) override;
+    PointExtractor(const QString& module_name);
+private:
+    static constexpr int max_blobs = 16;
+
+    pt_settings s;
+
+    cv::Mat1b frame_gray_unmasked, frame_bin, frame_gray;
+    cv::Mat1f hist;
+    std::vector<blob> blobs;
+    cv::Mat1b ch[3];
+
+    void ensure_channel_buffers(const cv::Mat& orig_frame);
+    void ensure_buffers(const cv::Mat& frame);
+
+    void extract_single_channel(const cv::Mat& orig_frame, int idx, cv::Mat1b& dest);
+
+    void color_to_grayscale(const cv::Mat& frame, cv::Mat1b& output);
+    void threshold_image(const cv::Mat& frame_gray, cv::Mat1b& output);
+};
+
+} // ns impl
+
diff --git a/tracker-points/module/tracker_pt.qrc b/tracker-points/module/tracker_pt.qrc
new file mode 100644
index 00000000..dfeb7369
--- /dev/null
+++ b/tracker-points/module/tracker_pt.qrc
@@ -0,0 +1,5 @@
+<RCC>
+    <qresource prefix="/">
+        <file>Resources/Logo_IR.png</file>
+    </qresource>
+</RCC>