sources/gst-plugins/gst-nvtracker/invtracker_proc.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 _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;
 
  std::vector<gboolean> operateOnClassIds;
};
 
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;
  virtual bool updateDynamicConfig(const std::string& configStr, uint32_t sourceId)=0;
};
 
#endif