File size: 3,456 Bytes
7fc5a59 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 | #ifdef USE_3D_ADAM_MODEL
#ifndef OPENPOSE_GUI_W_GUI_ADAM_HPP
#define OPENPOSE_GUI_W_GUI_ADAM_HPP
#include <openpose/core/common.hpp>
#include <openpose/gui/guiAdam.hpp>
#include <openpose/thread/workerConsumer.hpp>
namespace op
{
template<typename TDatums>
class WGuiAdam : public WorkerConsumer<TDatums>
{
public:
explicit WGuiAdam(const std::shared_ptr<GuiAdam>& guiAdam);
virtual ~WGuiAdam();
void initializationOnThread();
void workConsumer(const TDatums& tDatums);
private:
std::shared_ptr<GuiAdam> spGuiAdam;
DELETE_COPY(WGuiAdam);
};
}
// Implementation
#include <openpose/utilities/pointerContainer.hpp>
namespace op
{
template<typename TDatums>
WGuiAdam<TDatums>::WGuiAdam(const std::shared_ptr<GuiAdam>& guiAdam) :
spGuiAdam{guiAdam}
{
}
template<typename TDatums>
WGuiAdam<TDatums>::~WGuiAdam()
{
}
template<typename TDatums>
void WGuiAdam<TDatums>::initializationOnThread()
{
try
{
spGuiAdam->initializationOnThread();
}
catch (const std::exception& e)
{
error(e.what(), __LINE__, __FUNCTION__, __FILE__);
}
}
template<typename TDatums>
void WGuiAdam<TDatums>::workConsumer(const TDatums& tDatums)
{
try
{
// tDatums might be empty but we still wanna update the GUI
if (tDatums != nullptr)
{
// Debugging log
opLogIfDebug("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
// Profiling speed
const auto profilerKey = Profiler::timerInit(__LINE__, __FUNCTION__, __FILE__);
// Update cvMat & keypoints
if (!tDatums->empty())
{
// Update cvMat
std::vector<Mat> cvOutputDatas;
for (auto& tDatum : *tDatums)
cvOutputDatas.emplace_back(tDatumPtr->cvOutputData);
spGuiAdam->setImage(cvOutputDatas);
// Update keypoints
const auto& tDatumPtr = (*tDatums)[0];
if (!tDatumPtr->poseKeypoints3D.empty())
spGuiAdam->generateMesh(
tDatumPtr->poseKeypoints3D, tDatumPtr->faceKeypoints3D, tDatumPtr->handKeypoints3D,
tDatumPtr->adamPose.data(), tDatumPtr->adamTranslation.data(), tDatumPtr->vtVec.data(),
tDatumPtr->vtVec.rows(), tDatumPtr->j0Vec.data(), tDatumPtr->j0Vec.rows(),
tDatumPtr->adamFaceCoeffsExp.data());
}
// Refresh/update GUI
spGuiAdam->update();
// Profiling speed
if (!tDatums->empty())
{
Profiler::timerEnd(profilerKey);
Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__);
}
// Debugging log
opLogIfDebug("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
}
}
catch (const std::exception& e)
{
this->stop();
error(e.what(), __LINE__, __FUNCTION__, __FILE__);
}
}
COMPILE_TEMPLATE_DATUM(WGuiAdam);
}
#endif // OPENPOSE_GUI_W_GUI_ADAM_HPP
#endif
|