sources/gst-plugins/gst-nvtracker/miscdatabufmanager.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2019-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#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