openpose-usr / usr /local /include /openpose /hand /wHandDetectorUpdate.hpp
camenduru's picture
thanks to openpose ❤
7fc5a59
#ifndef OPENPOSE_HAND_W_HAND_DETECTOR_UPDATE_HPP
#define OPENPOSE_HAND_W_HAND_DETECTOR_UPDATE_HPP
#include <openpose/core/common.hpp>
#include <openpose/hand/handRenderer.hpp>
#include <openpose/thread/worker.hpp>
namespace op
{
template<typename TDatums>
class WHandDetectorUpdate : public Worker<TDatums>
{
public:
explicit WHandDetectorUpdate(const std::shared_ptr<HandDetector>& handDetector);
virtual ~WHandDetectorUpdate();
void initializationOnThread();
void work(TDatums& tDatums);
private:
std::shared_ptr<HandDetector> spHandDetector;
DELETE_COPY(WHandDetectorUpdate);
};
}
// Implementation
#include <openpose/utilities/pointerContainer.hpp>
namespace op
{
template<typename TDatums>
WHandDetectorUpdate<TDatums>::WHandDetectorUpdate(const std::shared_ptr<HandDetector>& handDetector) :
spHandDetector{handDetector}
{
}
template<typename TDatums>
WHandDetectorUpdate<TDatums>::~WHandDetectorUpdate()
{
}
template<typename TDatums>
void WHandDetectorUpdate<TDatums>::initializationOnThread()
{
}
template<typename TDatums>
void WHandDetectorUpdate<TDatums>::work(TDatums& tDatums)
{
try
{
if (checkNoNullNorEmpty(tDatums))
{
// Debugging log
opLogIfDebug("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
// Profiling speed
const auto profilerKey = Profiler::timerInit(__LINE__, __FUNCTION__, __FILE__);
// Detect people hand
for (auto& tDatumPtr : *tDatums)
spHandDetector->updateTracker(tDatumPtr->handKeypoints, tDatumPtr->id);
// Profiling speed
Profiler::timerEnd(profilerKey);
Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__);
// Debugging log
opLogIfDebug("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
}
}
catch (const std::exception& e)
{
this->stop();
tDatums = nullptr;
error(e.what(), __LINE__, __FUNCTION__, __FILE__);
}
}
COMPILE_TEMPLATE_DATUM(WHandDetectorUpdate);
}
#endif // OPENPOSE_HAND_W_HAND_DETECTOR_UPDATE_HPP