92 lines
2.9 KiB
C++
92 lines
2.9 KiB
C++
|
|
#include "HandCalibrator.h"
|
|
#include <iostream>
|
|
|
|
#define MIN_MENU_HAND_SIZE 10000
|
|
#define MIN_GAME_HAND_SIZE 3000 // todo change
|
|
namespace computervision
|
|
{
|
|
namespace handcalibration
|
|
{
|
|
|
|
HandCalibrator::HandCalibrator()
|
|
{
|
|
|
|
}
|
|
|
|
void HandCalibrator::DrawHandCalibrationText(cv::Mat& output_frame)
|
|
{
|
|
cv::rectangle(output_frame, cv::Rect(0, 0, output_frame.cols, 40), cv::Scalar(0, 0, 0), -1);
|
|
cv::putText(output_frame, "Hand calibration", cv::Point(output_frame.cols / 2 - 100, 25), cv::FONT_HERSHEY_PLAIN, 2.0, cv::Scalar(18, 219, 65), 2);
|
|
|
|
cv::rectangle(output_frame, cv::Rect(0, output_frame.rows - 80, 450, output_frame.cols), cv::Scalar(0, 0, 0), -1);
|
|
|
|
cv::putText(output_frame, "hand in frame:", cv::Point(5, output_frame.rows - 50), cv::FONT_HERSHEY_PLAIN, 2.0, cv::Scalar(255, 255, 0), 1);
|
|
cv::rectangle(output_frame, cv::Rect(420, output_frame.rows - 67, 15, 15), hand_present ? cv::Scalar(0, 255, 0) : cv::Scalar(0, 0, 255), -1);
|
|
|
|
DrawBackgroundSkinCalibrated(output_frame);
|
|
|
|
if (hand_present)
|
|
{
|
|
std::string hand_text = fingers_amount > 0 ? "open" : "closed";
|
|
cv::putText(output_frame, hand_text, cv::Point(10, 75), cv::FONT_HERSHEY_PLAIN, 2.0, cv::Scalar(255, 0, 255), 3);
|
|
}
|
|
}
|
|
|
|
void HandCalibrator::DrawBackgroundSkinCalibrated(cv::Mat& output_frame)
|
|
{
|
|
|
|
cv::putText(output_frame, "background calibrated:", cv::Point(5, output_frame.rows - 30), cv::FONT_HERSHEY_PLAIN, 2.0, cv::Scalar(255, 255, 0), 1);
|
|
cv::rectangle(output_frame, cv::Rect(420, output_frame.rows - 47, 15, 15), background_calibrated ? cv::Scalar(0, 255, 0) : cv::Scalar(0, 0, 255), -1);
|
|
|
|
cv::putText(output_frame, "skin color calibrated:", cv::Point(5, output_frame.rows - 10), cv::FONT_HERSHEY_PLAIN, 2.0, cv::Scalar(255, 255, 0), 1);
|
|
cv::rectangle(output_frame, cv::Rect(420, output_frame.rows - 27, 15, 15), skintone_calibrated ? cv::Scalar(0, 255, 0) : cv::Scalar(0, 0, 255), -1);
|
|
}
|
|
|
|
void HandCalibrator::SetSkinCalibration(bool val)
|
|
{
|
|
skintone_calibrated = val;
|
|
}
|
|
|
|
void HandCalibrator::SetBackGroundCalibrated(bool val)
|
|
{
|
|
background_calibrated = val;
|
|
}
|
|
|
|
void HandCalibrator::SetHandPresent(bool val)
|
|
{
|
|
hand_present = val;
|
|
}
|
|
|
|
void HandCalibrator::SetAmountOfFingers(int amount)
|
|
{
|
|
fingers_amount = amount;
|
|
}
|
|
|
|
bool HandCalibrator::CheckIfHandPresent(cv::Mat input_image, HandDetectionType type)
|
|
{
|
|
std::vector<std::vector<cv::Point>> points;
|
|
cv::findContours(input_image, points, cv::RetrievalModes::RETR_LIST, cv::ContourApproximationModes::CHAIN_APPROX_SIMPLE);
|
|
|
|
if (points.size() == 0) return false;
|
|
|
|
for (int p = 0; p < points.size(); p++)
|
|
{
|
|
int area = cv::contourArea(points[p]);
|
|
|
|
if (type == handcalibration::HandDetectionType::MENU)
|
|
if (area > MIN_MENU_HAND_SIZE) return true;
|
|
|
|
if (type == handcalibration::HandDetectionType::GAME)
|
|
if (area > MIN_GAME_HAND_SIZE) return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
}
|