archive/rpcs3
1
0
Fork 0
mirror of https://github.com/RPCS3/rpcs3.git synced 2025-07-03 21:41:26 +12:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

ps move tracker: draw sphere size range into live image

Browse source
This commit is contained in:
Megamouse 2024-12-21 12:51:27 +01:00
parent ec7f50d925
commit beb3565001
2 changed files with 42 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

40
rpcs3/Input/ps_move_tracker.cpp
View file

@ -313,6 +313,34 @@ static bool is_circular_contour(const std::vector<cv::Point>& contour, f32& area
return true; return true;
} }
template <bool DiagnosticsEnabled>
void ps_move_tracker<DiagnosticsEnabled>::draw_sphere_size_range(f32 result_radius)
{
if constexpr (!DiagnosticsEnabled) return;
if (!m_draw_overlays) return;
// Map memory
cv::Mat rgba(cv::Size(m_width, m_height), CV_8UC4, m_image_rgba_contours.data(), 0);
// Draw result, min and max radius
const f32 min_radius = m_min_radius * m_width;
const f32 max_radius = m_max_radius * m_width;
const f32 min_radius_clamped = std::max(0.0f, std::min(min_radius, max_radius));
const cv::Point2f center = cv::Point2f(m_width - 1 - max_radius, max_radius);
if (result_radius > 0.0f)
{
cv::circle(rgba, center, static_cast<int>(result_radius), cv::Scalar(255, 0, 0, 255), cv::FILLED);
}
if (min_radius_clamped > 0.0f && min_radius_clamped <= max_radius)
{
cv::circle(rgba, center, static_cast<int>(min_radius_clamped), cv::Scalar(0, 0, 0, 255), cv::FILLED);
}
if (max_radius > min_radius_clamped)
{
cv::circle(rgba, center, static_cast<int>(max_radius), cv::Scalar(0, 0, 0, 255), 1);
}
}
template <bool DiagnosticsEnabled> template <bool DiagnosticsEnabled>
void ps_move_tracker<DiagnosticsEnabled>::process_contours(ps_move_info& info, u32 index) void ps_move_tracker<DiagnosticsEnabled>::process_contours(ps_move_info& info, u32 index)
{ {
@ -384,6 +412,11 @@ void ps_move_tracker<DiagnosticsEnabled>::process_contours(ps_move_info& info, u
if (all_contours.empty()) if (all_contours.empty())
{ {
set_valid(info, index, false); set_valid(info, index, false);
if constexpr (DiagnosticsEnabled)
{
draw_sphere_size_range(0.0f);
}
return; return;
} }
@ -431,6 +464,11 @@ void ps_move_tracker<DiagnosticsEnabled>::process_contours(ps_move_info& info, u
if (best_index == umax) if (best_index == umax)
{ {
set_valid(info, index, false); set_valid(info, index, false);
if constexpr (DiagnosticsEnabled)
{
draw_sphere_size_range(0.0f);
}
return; return;
} }
@ -472,6 +510,8 @@ void ps_move_tracker<DiagnosticsEnabled>::process_contours(ps_move_info& info, u
if (!m_draw_contours && !m_draw_overlays) [[likely]] if (!m_draw_contours && !m_draw_overlays) [[likely]]
return; return;
draw_sphere_size_range(info.radius);
// Map memory // Map memory
cv::Mat rgba(cv::Size(width, height), CV_8UC4, m_image_rgba_contours.data(), 0); cv::Mat rgba(cv::Size(width, height), CV_8UC4, m_image_rgba_contours.data(), 0);

2
rpcs3/Input/ps_move_tracker.h
View file

@ -73,6 +73,8 @@ private:
void set_valid(ps_move_info& info, u32 index, bool valid); void set_valid(ps_move_info& info, u32 index, bool valid);
void draw_sphere_size_range(f32 result_radius);
u32 m_width = 0; u32 m_width = 0;
u32 m_height = 0; u32 m_height = 0;
s32 m_format = 0; s32 m_format = 0;