sources/libs/nvmultiobjecttracker/include/modules/Segmenter/DummySegmenter.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2024-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 SEGMENTERLIB_DUMMYSEGMENTER_HPP_
#define SEGMENTERLIB_DUMMYSEGMENTER_HPP_
 
#include "modules/Segmenter/ISegmenter.hpp"
#include "modules/NvMultiObjectTracker/NvTrackerParams.hpp"
#include "modules/NvMultiObjectTracker/NvTrackedObject.hpp"
 
class DummySegmenterConfigParams : public NvTrackerParams
{
public:
    DummySegmenterConfigParams();
 
    DummySegmenterConfigParams(const std::string &configFilePath, const std::string moduleName);
 
    DummySegmenterConfigParams(const YAML::Node &fn);
 
    void read(const YAML::Node &fn) override;
 
    int m_debugVerboseLevel; 
    SEGMENTER_TYPES m_segmenterType; 
};
 
class DummySegmenter : public ISegmenter
{
public:
    DummySegmenter(const std::string &configFilePath);
 
    virtual ~DummySegmenter();
 
    void initialize(
        const SourceFrameInfo &frameInfo, const uint maxTargetsPerStream,
        const std::shared_ptr<NvTrackedObjectManager> &targetManager
    ) override;
 
    void queryParams(NvMOTQuery &paramsQuery) override;
 
    void localize(const std::map<NvMOTStreamId, NvMOTFrame *>& inputMeta) override;
 
    void update(const std::map<NvMOTStreamId, NvMOTFrame *>& inputMeta) override;
 
    void outputMaskObj(const std::shared_ptr<NvTrackedObject> target, NvMOTMask& mask) override;
 
    void removeStream(const NvMOTStreamId videoStreamId) override;
 
    bool isInitialized() override { return m_isInit; }
 
    const SEGMENTER_TYPES getSegmenterType() override { return m_params.m_segmenterType; }
 
protected:
    bool m_isInit; 
    DummySegmenterConfigParams m_params; 
};
 
#endif // SEGMENTERLIB_DUMMYSEGMENTER_HPP_