OpenPose  1.0.0rc2
OpenPose: A Real-Time Multi-Person Key-Point Detection And Multi-Threading C++ Library
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
wCocoJsonSaver.hpp
Go to the documentation of this file.
1 #ifndef OPENPOSE_FILESTREAM_W_COCO_JSON_SAVER_HPP
2 #define OPENPOSE_FILESTREAM_W_COCO_JSON_SAVER_HPP
3 
7 
8 namespace op
9 {
10  template<typename TDatums>
11  class WCocoJsonSaver : public WorkerConsumer<TDatums>
12  {
13  public:
14  explicit WCocoJsonSaver(const std::shared_ptr<CocoJsonSaver>& cocoJsonSaver);
15 
17 
18  void workConsumer(const TDatums& tDatums);
19 
20  private:
21  std::shared_ptr<CocoJsonSaver> spCocoJsonSaver;
22 
23  DELETE_COPY(WCocoJsonSaver);
24  };
25 }
26 
27 
28 
29 
30 
31 // Implementation
33 namespace op
34 {
35  template<typename TDatums>
36  WCocoJsonSaver<TDatums>::WCocoJsonSaver(const std::shared_ptr<CocoJsonSaver>& cocoJsonSaver) :
37  spCocoJsonSaver{cocoJsonSaver}
38  {
39  }
40 
41  template<typename TDatums>
43  {
44  }
45 
46  template<typename TDatums>
47  void WCocoJsonSaver<TDatums>::workConsumer(const TDatums& tDatums)
48  {
49  try
50  {
51  if (checkNoNullNorEmpty(tDatums))
52  {
53  // Check tDatums->size() == 1
54  if (tDatums->size() > 1)
55  error("Function only ready for tDatums->size() == 1", __LINE__, __FUNCTION__, __FILE__);
56  // Debugging log
57  dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
58  // Profiling speed
59  const auto profilerKey = Profiler::timerInit(__LINE__, __FUNCTION__, __FILE__);
60  // T* to T
61  const auto& tDatum = tDatums->at(0);
62  // Record json in COCO format
63  spCocoJsonSaver->record(tDatum.poseKeypoints, tDatum.poseScores, tDatum.name);
64  // Profiling speed
65  Profiler::timerEnd(profilerKey);
66  Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__);
67  // Debugging log
68  dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
69  }
70  }
71  catch (const std::exception& e)
72  {
73  this->stop();
74  error(e.what(), __LINE__, __FUNCTION__, __FILE__);
75  }
76  }
77 
79 }
80 
81 #endif // OPENPOSE_FILESTREAM_W_COCO_JSON_SAVER_HPP
Definition: workerConsumer.hpp:10
WCocoJsonSaver(const std::shared_ptr< CocoJsonSaver > &cocoJsonSaver)
Definition: wCocoJsonSaver.hpp:36
static const std::string timerInit(const int line, const std::string &function, const std::string &file)
OP_API void error(const std::string &message, const int line=-1, const std::string &function="", const std::string &file="")
void initializationOnThread()
Definition: wCocoJsonSaver.hpp:42
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
Definition: wCocoJsonSaver.hpp:11
bool checkNoNullNorEmpty(const TPointerContainer &tPointerContainer)
Definition: pointerContainer.hpp:7
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)
COMPILE_TEMPLATE_DATUM(WPoseTriangulation)
static void timerEnd(const std::string &key)
void workConsumer(const TDatums &tDatums)
Definition: wCocoJsonSaver.hpp:47