OpenPose  1.7.0
The first real-time multi-person system to jointly detect human body, hand, facial, and foot keypoints
wVideoSaver.hpp
Go to the documentation of this file.
1 #ifndef OPENPOSE_FILESTREAM_W_VIDEO_SAVER_HPP
2 #define OPENPOSE_FILESTREAM_W_VIDEO_SAVER_HPP
3 
7 
8 namespace op
9 {
10  template<typename TDatums>
11  class WVideoSaver : public WorkerConsumer<TDatums>
12  {
13  public:
14  explicit WVideoSaver(const std::shared_ptr<VideoSaver>& videoSaver);
15 
16  virtual ~WVideoSaver();
17 
19 
20  void workConsumer(const TDatums& tDatums);
21 
22  private:
23  std::shared_ptr<VideoSaver> spVideoSaver;
24 
25  DELETE_COPY(WVideoSaver);
26  };
27 }
28 
29 
30 
31 
32 
33 // Implementation
35 namespace op
36 {
37  template<typename TDatums>
38  WVideoSaver<TDatums>::WVideoSaver(const std::shared_ptr<VideoSaver>& videoSaver) :
39  spVideoSaver{videoSaver}
40  {
41  }
42 
43  template<typename TDatums>
45  {
46  }
47 
48  template<typename TDatums>
50  {
51  }
52 
53  template<typename TDatums>
54  void WVideoSaver<TDatums>::workConsumer(const TDatums& tDatums)
55  {
56  try
57  {
58  if (checkNoNullNorEmpty(tDatums))
59  {
60  // Debugging log
61  opLogIfDebug("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
62  // Profiling speed
63  const auto profilerKey = Profiler::timerInit(__LINE__, __FUNCTION__, __FILE__);
64  // T* to T
65  auto& tDatumsNoPtr = *tDatums;
66  // Record video(s)
67  std::vector<Matrix> opOutputDatas(tDatumsNoPtr.size());
68  for (auto i = 0u ; i < opOutputDatas.size() ; i++)
69  opOutputDatas[i] = tDatumsNoPtr[i]->cvOutputData;
70  spVideoSaver->write(opOutputDatas);
71  // Profiling speed
72  Profiler::timerEnd(profilerKey);
73  Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__);
74  // Debugging log
75  opLogIfDebug("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
76  }
77  }
78  catch (const std::exception& e)
79  {
80  this->stop();
81  error(e.what(), __LINE__, __FUNCTION__, __FILE__);
82  }
83  }
84 
86 }
87 
88 #endif // OPENPOSE_FILESTREAM_W_VIDEO_SAVER_HPP
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)
static const std::string timerInit(const int line, const std::string &function, const std::string &file)
static void timerEnd(const std::string &key)
WVideoSaver(const std::shared_ptr< VideoSaver > &videoSaver)
Definition: wVideoSaver.hpp:38
void workConsumer(const TDatums &tDatums)
Definition: wVideoSaver.hpp:54
virtual ~WVideoSaver()
Definition: wVideoSaver.hpp:44
void initializationOnThread()
Definition: wVideoSaver.hpp:49
bool checkNoNullNorEmpty(const TPointerContainer &tPointerContainer)
COMPILE_TEMPLATE_DATUM(WPoseTriangulation)
OP_API void error(const std::string &message, const int line=-1, const std::string &function="", const std::string &file="")
void opLogIfDebug(const T &message, const Priority priority=Priority::Max, const int line=-1, const std::string &function="", const std::string &file="")
Definition: errorAndLog.hpp:97