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
wDatumProducer.hpp
Go to the documentation of this file.
1 #ifndef OPENPOSE_PRODUCER_W_DATUM_PRODUCER_HPP
2 #define OPENPOSE_PRODUCER_W_DATUM_PRODUCER_HPP
3 
4 #include <limits> // std::numeric_limits
5 #include <queue> // std::queue
9 
10 namespace op
11 {
12  template<typename TDatums, typename TDatumsNoPtr>
13  class WDatumProducer : public WorkerProducer<TDatums>
14  {
15  public:
16  explicit WDatumProducer(const std::shared_ptr<DatumProducer<TDatumsNoPtr>>& datumProducer);
17 
19 
20  TDatums workProducer();
21 
22  private:
23  std::shared_ptr<DatumProducer<TDatumsNoPtr>> spDatumProducer;
24  std::queue<TDatums> mQueuedElements;
25 
26  DELETE_COPY(WDatumProducer);
27  };
28 }
29 
30 
31 
32 
33 
34 // Implementation
35 #include <openpose/core/datum.hpp>
36 namespace op
37 {
38  template<typename TDatums, typename TDatumsNoPtr>
40  spDatumProducer{datumProducer}
41  {
42  }
43 
44  template<typename TDatums, typename TDatumsNoPtr>
46  {
47  }
48 
49  template<typename TDatums, typename TDatumsNoPtr>
51  {
52  try
53  {
54  // Debugging log
55  dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
56  // Profiling speed
57  const auto profilerKey = Profiler::timerInit(__LINE__, __FUNCTION__, __FILE__);
58  // Create and fill TDatums
59  std::shared_ptr<TDatumsNoPtr> tDatums;
60  // Producer
61  if (mQueuedElements.empty())
62  {
63  bool isRunning;
64  std::tie(isRunning, tDatums) = spDatumProducer->checkIfRunningAndGetDatum();
65  // Stop Worker if producer finished
66  if (!isRunning)
67  this->stop();
68  // Profiling speed
69  Profiler::timerEnd(profilerKey);
70  Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__);
71  // Debugging log
72  dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
73  }
74  // Equivalent to WQueueSplitter
75  // Queued elements - Multiple views --> Split views into different TDatums
76  if (tDatums != nullptr && tDatums->size() > 1)
77  {
78  // Add tDatums to mQueuedElements
79  for (auto i = 0u ; i < tDatums->size() ; i++)
80  {
81  auto& tDatum = (*tDatums)[i];
82  tDatum.subId = i;
83  tDatum.subIdMax = tDatums->size()-1;
84  mQueuedElements.emplace(
85  std::make_shared<TDatumsNoPtr>(TDatumsNoPtr{tDatum}));
86  }
87  }
88  // Queued elements - Multiple views --> Return oldest view
89  if (!mQueuedElements.empty())
90  {
91  tDatums = mQueuedElements.front();
92  mQueuedElements.pop();
93  }
94  // Return TDatums
95  return tDatums;
96  }
97  catch (const std::exception& e)
98  {
99  this->stop();
100  error(e.what(), __LINE__, __FUNCTION__, __FILE__);
101  return TDatums{};
102  }
103  }
104 
105  extern template class WDatumProducer<DATUM_BASE, DATUM_BASE_NO_PTR>;
106 }
107 
108 #endif // OPENPOSE_PRODUCER_W_DATUM_PRODUCER_HPP
void initializationOnThread()
Definition: wDatumProducer.hpp:45
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="")
WDatumProducer(const std::shared_ptr< DatumProducer< TDatumsNoPtr >> &datumProducer)
Definition: wDatumProducer.hpp:39
Definition: workerProducer.hpp:10
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: datumProducer.hpp:14
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 void timerEnd(const std::string &key)
Definition: wDatumProducer.hpp:13
TDatums workProducer()
Definition: wDatumProducer.hpp:50