sources/libs/nvmultiobjecttracker/include/modules/NvMultiObjectTracker/localizer.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2021-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 NVMULTIOBJECTTRACKERLIB_NVTLOCALIZER_H
#define NVMULTIOBJECTTRACKERLIB_NVTLOCALIZER_H
 
#include <map>
#include <set>
#include <memory>
 
#include "modules/utility/dataTypes.hpp"
 
#include "nvdstracker.h"
#include "nvds_tracker_meta.h"
 
class IBatchLocalizer
{
public:
    IBatchLocalizer() {};
 
    virtual ~IBatchLocalizer() {};
 
    virtual bool isInitialized() = 0;
 
    virtual void initialize(const NvMotDataTypes::Size srcImgSize, const size_t srcImgPitch) = 0;
 
    virtual void queryParams(NvMOTQuery &paramsQuery) = 0;
 
    virtual void update(
        const std::map<NvMOTStreamId, NvMOTFrame*> &nvFramesInBatch, NvMOTTrackedObjBatch *&pTrackedObjectsBatch
    ) = 0;
 
    virtual void outputPastFrameObjs(
        const std::set<NvMOTStreamId> videoStreamIdList, NvDsTargetMiscDataBatch *&pPastFrameObjBatch
    ) = 0;
 
    virtual void outputTerminatedObjs(
        const std::set<NvMOTStreamId> videoStreamIdList, NvDsTargetMiscDataBatch *&pPastFrameObjBatch
    ) = 0;
 
    virtual void outputShadowObj(
        const std::set<NvMOTStreamId> videoStreamIdList, NvDsTargetMiscDataBatch *&pShadowTracksBatch
    ) = 0;
 
    virtual void deleteRemovedStreamTrackers(const NvMOTStreamId videoStreamID) = 0;
 
    virtual void updateParams(const std::string &configStr) = 0;
 
protected:
};
 
#endif // NVMULTIOBJECTTRACKERLIB_NVTLOCALIZER_H