All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
wJointAngleEstimation.hpp
Go to the documentation of this file.
#ifdef USE_3D_ADAM_MODEL
#ifndef OPENPOSE_3D_W_JOINT_ANGLE_ESTIMATION_HPP
#define OPENPOSE_3D_W_JOINT_ANGLE_ESTIMATION_HPP
#include <openpose/core/common.hpp>
#include <openpose/3d/jointAngleEstimation.hpp>
#include <openpose/thread/worker.hpp>
namespace op
{
template<typename TDatums>
class WJointAngleEstimation : public Worker<TDatums>
{
public:
explicit WJointAngleEstimation(const std::shared_ptr<JointAngleEstimation>& jointAngleEstimation);
virtual ~WJointAngleEstimation();
void initializationOnThread();
void work(TDatums& tDatums);
private:
const std::shared_ptr<JointAngleEstimation> spJointAngleEstimation;
DELETE_COPY(WJointAngleEstimation);
};
}
// Implementation
#include <openpose/utilities/pointerContainer.hpp>
namespace op
{
template<typename TDatums>
WJointAngleEstimation<TDatums>::WJointAngleEstimation(const std::shared_ptr<JointAngleEstimation>& jointAngleEstimation) :
spJointAngleEstimation{jointAngleEstimation}
{
}
template<typename TDatums>
WJointAngleEstimation<TDatums>::~WJointAngleEstimation()
{
}
template<typename TDatums>
void WJointAngleEstimation<TDatums>::initializationOnThread()
{
try
{
spJointAngleEstimation->initializationOnThread();
}
catch (const std::exception& e)
{
error(e.what(), __LINE__, __FUNCTION__, __FILE__);
}
}
template<typename TDatums>
void WJointAngleEstimation<TDatums>::work(TDatums& tDatums)
{
try
{
if (checkNoNullNorEmpty(tDatums))
{
// Debugging log
dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
// Profiling speed
const auto profilerKey = Profiler::timerInit(__LINE__, __FUNCTION__, __FILE__);
// Input
auto& datum = tDatums->at(0);
const auto& poseKeypoints3D = datum.poseKeypoints3D;
const auto& faceKeypoints3D = datum.faceKeypoints3D;
const auto& handKeypoints3D = datum.handKeypoints3D;
// Running Adam model
spJointAngleEstimation->adamFastFit(
datum.adamPose, datum.adamTranslation, datum.vtVec, datum.j0Vec,
datum.adamFaceCoeffsExp, poseKeypoints3D, faceKeypoints3D, handKeypoints3D);
// Profiling speed
Profiler::timerEnd(profilerKey);
Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__);
// Debugging log
dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
}
}
catch (const std::exception& e)
{
this->stop();
tDatums = nullptr;
error(e.what(), __LINE__, __FUNCTION__, __FILE__);
}
}
COMPILE_TEMPLATE_DATUM(WJointAngleEstimation);
}
#endif // OPENPOSE_3D_W_JOINT_ANGLE_ESTIMATION_HPP
#endif
DELETE_COPY
#define DELETE_COPY(className)
Definition: macros.hpp:34
op::Priority::Low
common.hpp
op::Profiler::timerInit
static const std::string timerInit(const int line, const std::string &function, const std::string &file)
op::error
OP_API void error(const std::string &message, const int line=-1, const std::string &function="", const std::string &file="")
pointerContainer.hpp
op::dLog
void dLog(const T &message, const Priority priority=Priority::Max, const int line=-1, const std::string &function="", const std::string &file="")
Definition: errorAndLog.hpp:53
worker.hpp
jointAngleEstimation.hpp
op::checkNoNullNorEmpty
bool checkNoNullNorEmpty(const TPointerContainer &tPointerContainer)
Definition: pointerContainer.hpp:7
op::Profiler::printAveragedTimeMsOnIterationX
static void printAveragedTimeMsOnIterationX(const std::string &key, const int line, const std::string &function, const std::string &file, const unsigned long long x=DEFAULT_X)
op::COMPILE_TEMPLATE_DATUM
COMPILE_TEMPLATE_DATUM(WPoseTriangulation)
op::Profiler::timerEnd
static void timerEnd(const std::string &key)