kinect/codes/Azure-Kinect-Samples/body-tracking-samples/jump_analysis_sample/JumpEvaluator.h

84 lines
2.5 KiB
C
Raw Normal View History

2024-03-06 18:05:53 +00:00
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#pragma once
#include <vector>
#include <k4abt.h>
#include "HandRaisedDetector.h"
#include "Window3dWrapper.h"
enum JumpStatus
{
Idle = 0,
CollectJumpData,
EvaluateAndReview
};
struct IndexValueTuple;
struct JumpResultsData;
class JumpEvaluator
{
public:
void UpdateStatus(bool changeStatus);
void UpdateData(k4abt_body_t selectedBody, uint64_t currentTimestampUsec);
private:
void InitiateJump();
JumpResultsData CalculateJumpResults();
void PrintJumpResults(const JumpResultsData& jumpResults);
void ReviewJumpResults(const JumpResultsData& jumpResults);
std::vector<float> GetInverseHeightInfoFromBodies(k4abt_joint_id_t jointId);
int DetermineCalculationWindowWidth(int jumpStartIndex, const std::vector<float>& timeStampInUsec);
float GetMinKneeAngleFromBody(k4abt_body_t body);
IndexValueTuple CalcualateJumpStartingPoint(const std::vector<float>& velocity, const std::vector<IndexValueTuple>& velocityPhases);
IndexValueTuple CalcualateJumpEndingPoint(const std::vector<float>& velocity, const std::vector<IndexValueTuple>& velocityPhases);
// Calculate characteristic points of velocity signal
std::vector<IndexValueTuple> CalculatePhasesFromVelocity(const std::vector<float>& velocity);
float CalculateStartHeight(std::vector<float> signal, size_t startingPoint = 10, size_t endingPoint = 30);
k4a_float3_t CalculateStandingPosition(int jumpStartIndex, int firstSquatIndex);
void CreateRenderWindow(
Window3dWrapper& window,
std::string windowName,
const k4abt_body_t& body,
int windowIndex,
k4a_float3_t standingPosition);
private:
// Constant settings for digial signal processing
const size_t MinimumBodyNumber = 20; // Minimum number of bodies required in the body list to perform the jump analysis
const int AverageFilterWindowSize = 6;
// Internal status
bool m_reviewWindowIsRunning = false;
JumpStatus m_jumpStatus = JumpStatus::Idle;
std::vector<k4abt_body_t> m_listOfBodyPositions;
std::vector<float> m_framesTimestampInUsec;
HandRaisedDetector m_handRaisedDetector;
bool m_previousHandsAreRaised = false;
// Default jump analysis window size
const int m_defaultWindowWidth = 640;
const int m_defaultWindowHeight = 576;
Window3dWrapper m_window3dSquatPose;
Window3dWrapper m_window3dJumpPeakPose;
Window3dWrapper m_window3dReplay;
};