summaryrefslogtreecommitdiffhomepage
path: root/tracker-wii/wii_frame.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'tracker-wii/wii_frame.hpp')
-rw-r--r--tracker-wii/wii_frame.hpp64
1 files changed, 64 insertions, 0 deletions
diff --git a/tracker-wii/wii_frame.hpp b/tracker-wii/wii_frame.hpp
new file mode 100644
index 00000000..2bffa862
--- /dev/null
+++ b/tracker-wii/wii_frame.hpp
@@ -0,0 +1,64 @@
+/*
+* Copyright (c) 2017-2018 Wei Shuai <cpuwolf@gmail.com>
+*
+* 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 <opencv2/core.hpp>
+#include <QImage>
+
+
+
+namespace pt_module {
+
+enum wii_camera_status : unsigned { wii_cam_wait_for_connect, wii_cam_data_no_change, wii_cam_data_change };
+
+struct wii_info_points {
+ unsigned ux;
+ unsigned uy;
+ int isize;
+ bool bvis;
+};
+
+struct wii_info {
+ struct wii_info_points Points[4];
+ bool bBatteryDrained;
+ unsigned char BatteryPercent;
+ float Pitch;
+ float Roll;
+ wii_camera_status status;
+};
+
+struct WIIFrame final : pt_frame
+{
+ cv::Mat mat;
+ struct wii_info wii;
+
+ operator const cv::Mat&() const& { return mat; }
+ operator cv::Mat&() & { return mat; }
+};
+
+struct WIIPreview final : pt_preview
+{
+ WIIPreview(int w, int h);
+
+ WIIPreview& operator=(const pt_frame& frame) override;
+ QImage get_bitmap() override;
+ void draw_head_center(double x, double 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);
+
+ bool fresh = true;
+ cv::Mat frame_copy, frame_color, frame_resize, frame_out;
+};
+
+} // ns pt_module