11 #ifndef _MISCDATABUFMANAGER_H
12 #define _MISCDATABUFMANAGER_H
17 #include <condition_variable>
47 bool init(uint32_t batchSize, uint32_t gpuId,
48 uint32_t maxTargetsPerStream, uint32_t maxShadowTrackingAge,
49 uint32_t reidFeatureSize, uint32_t maxBufferPoolSize,
bool pastFrame,
50 bool outputReidTensor,
51 bool outputTerminatedTracks,
52 bool outputShadowTracks,
53 uint32_t maxTerminatedFrameHistory);
64 bool m_IntentionallyEmpty;
66 bool m_OutputReidTensor;
67 bool m_OutputTerminatedTracks;
68 bool m_OutputShadowTracks;
71 std::condition_variable m_Cond;
73 std::queue<NvTrackerMiscDataBuffer *> m_FreeQueue;
75 std::vector<NvTrackerMiscDataBuffer *> m_BufferSet;
77 bool isQueueFull() {
return m_BufferSet.size() == m_FreeQueue.size(); }
80 uint32_t batchSize, uint32_t maxTargetsPerStream, uint32_t maxShadowTrackingAge);
87 uint32_t batchSize, uint32_t maxTargetsPerStream,
88 uint32_t reidFeatureSize);
95 uint32_t batchSize, uint32_t maxTargetsPerStream, uint32_t maxTermTrackFrameHistory);
102 uint32_t batchSize, uint32_t maxTargetsPerStream );