VideoManager.hpp

#pragma once
 
#include <chrono>
#include <cstdint>
#include <string>
#include <vector>
 
#include <opencv2/core/mat.hpp>
#include <opencv2/videoio.hpp>
 
#include "detectors/LaneDetector.hpp"
#include "detectors/ObjectDetector.hpp"
#include "helpers/Globals.hpp"
#include "helpers/Information.hpp"
#include "helpers/Performance.hpp"

namespace LaneAndObjectDetection
{
    class VideoManager
    {
    public:
        explicit VideoManager();

        explicit VideoManager(const int32_t& p_inputVideoCamera,
                              const std::string& p_yoloFolderPath,
                              const Globals::ObjectDetectorTypes& p_objectDetectorTypes,
                              const Globals::ObjectDetectorBackEnds& p_objectDetectorBackEnds = Globals::ObjectDetectorBackEnds::CPU,
                              const Globals::ObjectDetectorBlobSizes& p_objectDetectorBlobSizes = Globals::ObjectDetectorBlobSizes::ONE);

        explicit VideoManager(const std::string& p_inputVideoFilePath,
                              const std::string& p_yoloFolderPath,
                              const Globals::ObjectDetectorTypes& p_objectDetectorTypes,
                              const Globals::ObjectDetectorBackEnds& p_objectDetectorBackEnds = Globals::ObjectDetectorBackEnds::CPU,
                              const Globals::ObjectDetectorBlobSizes& p_objectDetectorBlobSizes = Globals::ObjectDetectorBlobSizes::ONE);

        explicit VideoManager(const std::vector<std::string>& p_commandLineArguments);

        ~VideoManager();

        void SetProperties(const int32_t& p_inputVideoCamera,
                           const std::string& p_yoloFolderPath,
                           const Globals::ObjectDetectorTypes& p_objectDetectorTypes,
                           const Globals::ObjectDetectorBackEnds& p_objectDetectorBackEnds,
                           const Globals::ObjectDetectorBlobSizes& p_objectDetectorBlobSizes);

        void SetProperties(const std::string& p_inputVideoFilePath,
                           const std::string& p_yoloFolderPath,
                           const Globals::ObjectDetectorTypes& p_objectDetectorTypes,
                           const Globals::ObjectDetectorBackEnds& p_objectDetectorBackEnds,
                           const Globals::ObjectDetectorBlobSizes& p_objectDetectorBlobSizes);

        void RunLaneAndObjectDetector();

        Performance GetPerformance();

        VideoManager(const VideoManager& p_videoManager) = delete;

        VideoManager(const VideoManager&& p_videoManager) = delete;

        VideoManager& operator=(const VideoManager& p_videoManager) = delete;

        VideoManager& operator=(const VideoManager&& p_videoManager) = delete;
 
    private:
        void ToggleDebugMode();

        void ToggleSaveOutput();

        void Quit();

        cv::VideoCapture m_inputVideo;

        cv::VideoWriter m_outputVideo;

        cv::Mat m_currentFrame;

        LaneDetector m_laneDetector;

        ObjectDetector m_objectDetector;

        Performance m_performance;

        VideoManagerInformation m_videoManagerInformation;

        std::chrono::time_point<std::chrono::high_resolution_clock> m_saveOutputStartTime;
    };
}