miscdatabufmanager.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2019-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */
 
#ifndef _MISCDATABUFMANAGER_H
#define _MISCDATABUFMANAGER_H
 
#include <queue>
#include <mutex>
#include <vector>
#include <condition_variable>
#include <chrono>
 
#include <gst/gst.h>
 
#include "nvdstracker.h"
 
struct NvTrackerMiscDataBuffer
{
  NvDsTargetMiscDataBatch pastFrameObjBatch;
  NvDsTrajectoryBatch trajectoryBatch;
  NvDsReidTensorBatch reidTensorBatch;
  NvDsTargetMiscDataBatch terminatedTrackBatch;
  NvDsTargetMiscDataBatch shadowTracksBatch;
 
};
 
class TrackerMiscDataManager
{
public:
  TrackerMiscDataManager();
  ~TrackerMiscDataManager();
  bool init(uint32_t batchSize, uint32_t gpuId,
    uint32_t maxTargetsPerStream, uint32_t maxShadowTrackingAge,
    uint32_t reidFeatureSize, uint32_t maxBufferPoolSize, bool pastFrame,
    bool outputReidTensor,
    bool outputTerminatedTracks,
    bool outputShadowTracks,
    uint32_t maxTerminatedFrameHistory);
  void returnBuffer(NvTrackerMiscDataBuffer *data);
  NvTrackerMiscDataBuffer *pop();
  void deInit();
 
 
private:
  bool m_IntentionallyEmpty;
  bool m_PastFrame;
  bool m_OutputReidTensor;
  bool m_OutputTerminatedTracks;
  bool m_OutputShadowTracks;
  std::mutex m_Mutex;
  std::condition_variable m_Cond;
  std::queue<NvTrackerMiscDataBuffer *> m_FreeQueue;
  std::vector<NvTrackerMiscDataBuffer *> m_BufferSet;
  bool isQueueFull() { return m_BufferSet.size() == m_FreeQueue.size(); }
  void allocatePastFrame(NvTrackerMiscDataBuffer *pNewBuf,
    uint32_t batchSize, uint32_t maxTargetsPerStream, uint32_t maxShadowTrackingAge);
  void releasePastFrame(NvTrackerMiscDataBuffer *pBuffer);
  void resetPastFrame(NvTrackerMiscDataBuffer *pBuffer);
  void allocateReid(NvTrackerMiscDataBuffer *pNewBuf,
    uint32_t batchSize, uint32_t maxTargetsPerStream,
    uint32_t reidFeatureSize);
  void releaseReid(NvTrackerMiscDataBuffer *pBuffer);
  void resetReid(NvTrackerMiscDataBuffer *pBuffer);
  void allocateTerminatedTracks(NvTrackerMiscDataBuffer *pNewBuf,
    uint32_t batchSize, uint32_t maxTargetsPerStream, uint32_t maxTermTrackFrameHistory);
  void releaseTerminatedTracks(NvTrackerMiscDataBuffer *pBuffer);
  void resetTerminatedTracks(NvTrackerMiscDataBuffer *pBuffer);
  void allocateShadowTracks(NvTrackerMiscDataBuffer *pNewBuf,
                            uint32_t batchSize, uint32_t maxTargetsPerStream );
  void releaseShadowTracks(NvTrackerMiscDataBuffer *pBuffer);
  void resetShadowTracks(NvTrackerMiscDataBuffer *pBuffer);
};
 
struct GstNvTrackerMiscDataObject
{
  GstMiniObject mini_object;
  TrackerMiscDataManager* misc_data_manager;
  NvTrackerMiscDataBuffer* misc_data_buffer;
};
 
#endif