ObjectDetectionHandler.h

Go to the documentation of this file.

 
#ifndef OBJECT_DETECTION_HANDLER_H
#define OBJECT_DETECTION_HANDLER_H
 
#include "../vision/objects/ObjectDetector.h"
#include "RecordingHandler.h"
 
 
class ObjectDetectionHandler
{
    private:
        // Declare private class member variables.
 
        std::shared_ptr<ObjectDetector> m_pObjectDetectorMainCam;
        std::shared_ptr<ObjectDetector> m_pObjectDetectorRearCam;
        std::unique_ptr<RecordingHandler> m_pRecordingHandler;
 
    public:
        // Define public enumerators specific to this class.
 
        enum class ObjectDetectors    // Enum for different cameras that detectors are being ran on.
        {
            OBJECTDETECTOR_START,
            eHeadMainCam,
            eRearCam,
            OBJECTDETECTOR_END
        };
 
        // Declare public class methods and variables.
 
        ObjectDetectionHandler();
        ~ObjectDetectionHandler();
        void StartAllDetectors();
        void StartRecording();
        void StopAllDetectors();
        void StopRecording();
 
        // Accessors.
 
        std::shared_ptr<ObjectDetector> GetObjectDetector(ObjectDetectors eDetectorName);
 
        cv::Mat RequestDetectionOverlayFrame(ObjectDetectors eDetector = ObjectDetectors::eHeadMainCam);
};
 
#endif