sources/libs/nvmultiobjecttracker/include/modules/VisualTracker/DummyVisualTracker.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 VISUALTRACKERLIB_DUMMYVISUALTRACKER_HPP_
#define VISUALTRACKERLIB_DUMMYVISUALTRACKER_HPP_
 
#include "modules/VisualTracker/IVisualTracker.hpp"
#include "modules/NvMultiObjectTracker/NvTrackerParams.hpp"
#include "modules/NvMultiObjectTracker/NvTrackedObject.hpp"
 
class DummyVisualTrackerConfigParams : public NvTrackerParams
{
public:
    DummyVisualTrackerConfigParams();
 
    DummyVisualTrackerConfigParams(const std::string &configFilePath, const std::string moduleName);
 
    DummyVisualTrackerConfigParams(const YAML::Node &fn);
 
    void read(const YAML::Node &fn) override;
 
    int m_debugVerboseLevel; 
 
    // [Configuration parameters]
    VISUAL_TRACKER_TYPES m_visualTrackerType; 
};
 
class DummyVisualTracker : public IVisualTracker
{
public:
 
    DummyVisualTracker(const std::string &configFilePath);
 
    virtual ~DummyVisualTracker();
 
    void initialize(
        const SourceFrameInfo &frameInfo, const uint maxTargetsToTrack,
        const uint maxDetectorBboxesPerFrame, const std::shared_ptr<NvTrackedObjectManager> &targetManager
    ) override;
 
    void queryParams(NvMOTQuery &paramsQuery);
 
    void localize(
        const std::map<NvMOTStreamId, NvMOTFrame *> inputMeta,
        const std::map<NvMOTStreamId, std::deque<DetectedObjInfo>> &detectedObjects
    ) override;
 
    void update(const std::map<NvMOTStreamId, NvMOTFrame *> inputMeta) override;
 
    bool isInitialized() override { return m_isInit; }
 
    const VISUAL_TRACKER_TYPES getVisualTrackerType() { return m_params.m_visualTrackerType; }
 
    void updateParams(const YAML::Node &fn) { m_params.read(fn); }
 
protected:
 
    bool m_isInit; 
    DummyVisualTrackerConfigParams m_params; 
    std::shared_ptr<NvTrackedObjectManager> m_targetManager; 
};
 
#endif // VISUALTRACKERLIB_DUMMYVISUALTRACKER_HPP_