84 lines
2.5 KiB
C
84 lines
2.5 KiB
C
|
// 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;
|
||
|
|
};
|