invtracker_proc.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 _INVTRACKERPROC_H
#define _INVTRACKERPROC_H
 
#include "nvbufsurface.h"
#include "nvdstracker.h"
 
#include <vector>
 
struct InputParams
{
  NvBufSurface *pSurfaceBatch;
  NvDsBatchMeta *pBatchMeta;
  void* pPreservedData;
  bool eventMarker;
};
 
enum CompletionStatus
{
  CompletionStatus_OK,
  CompletionStatus_Error,
  CompletionStatus_Exit
};
 
enum TrackingIdResetMode
{
  TrackingIdResetMode_Default = 0,
  TrackingIdResetMode_NewIdAfterStreamReset = 1 << 0,
  TrackingIdResetMode_FromZeroAfterEOS = 1 << 1,
  TrackingIdResetMode_MaxValue = 3
};
 
struct TrackerConfig
{
  uint32_t batchSize;
  uint32_t trackerWidth;
  uint32_t trackerHeight;
  char* trackerLibFile;
  char* trackerConfigFileList;
  std::vector<std::string> trackerConfigFilePerSubBatch;
 
  bool displayTrackingId;
  TrackingIdResetMode trackingIdResetMode;
 
  NvMOTCompute computeTarget;
  int32_t gpuId;
  int32_t compute_hw;
 
  uint32_t trackingSurfType;
  bool trackingSurfTypeFromConfig = false;
 
  bool inputTensorMeta = false;
  uint32_t tensorMetaGieId = 0;
  std::vector<std::vector<int>> subBatchesConfig = {};
  std::vector<uint32_t> subBatchSizes = {};
  bool dynamicSubBatching = false;
  int subBatchErrRecoveryTrialCnt;
 
  NvBufSurfaceColorFormat colorFormat;
  NvBufSurfaceMemType memType;
  uint32_t numTransforms;
  uint32_t maxTargetsPerStream;
  uint32_t maxShadowTrackingAge;
  bool pastFrame;
  bool outputTerminatedTracks;
  uint32_t maxTrajectoryBufferLength;
 
  bool outputShadowTracks;
 
  uint32_t maxConvBufPoolSize;
  uint32_t maxMiscDataPoolSize;
  uint32_t reidFeatureSize;
  uint32_t maxConvexHullSize;
  bool outputReidTensor;
  bool outputVisibility;
  bool outputFootLocation;
  bool outputConvexHull;
 
  char* gstName;
};
 
class INvTrackerProc
{
public:
  virtual ~INvTrackerProc() {};
 
  virtual bool init(const TrackerConfig& config) = 0;
  virtual void deInit() = 0;
 
  virtual bool addSource(uint32_t sourceId) = 0;
  virtual bool removeSource(uint32_t sourceId, bool removeObjectIdMapping=true) = 0;
  virtual bool resetSource(uint32_t sourceId) = 0;
  virtual bool submitInput(const InputParams& inputParams) = 0;
  virtual CompletionStatus waitForCompletion(InputParams& inputParams) = 0;
  virtual bool flushReqs() = 0;
};
 
#endif