invtracker_proc.h

Go to the documentation of this file.

 
#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* trackerConfigFile;
  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 = {};
 
  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