From 4a2a1b636df7a67722b6b5764efa8cfc63c6a5e1 Mon Sep 17 00:00:00 2001 From: Wei Shuai Date: Fri, 26 Jan 2018 22:02:07 +0800 Subject: tracker/wii: new home --- tracker-wii/wii_point_extractor.h | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 tracker-wii/wii_point_extractor.h (limited to 'tracker-wii/wii_point_extractor.h') diff --git a/tracker-wii/wii_point_extractor.h b/tracker-wii/wii_point_extractor.h new file mode 100644 index 00000000..be0e5f45 --- /dev/null +++ b/tracker-wii/wii_point_extractor.h @@ -0,0 +1,38 @@ +/* +* Copyright (c) 2017-2018 Wei Shuai +* +* 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 + +#include +#include + +namespace pt_module { + +using namespace types; + +class WIIPointExtractor 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& points) override; + WIIPointExtractor(const QString& module_name); +private: + static constexpr int max_blobs = 16; + + pt_settings s; + void _draw_point(cv::Mat& preview_frame, const vec2& p, const cv::Scalar& color, int thinkness = 1); + bool _draw_points(cv::Mat& preview_frame, const struct wii_info &wii, std::vector& points); + void _draw_bg(cv::Mat& preview_frame, const struct wii_info &wii); +}; + +} // ns impl + -- cgit v1.2.3