ClipProcessingJobs.h

Go to the documentation of this file.

 
// Copyright (c) 2008-2019 OpenShot Studios, LLC
//
// SPDX-License-Identifier: LGPL-3.0-or-later
 
#ifdef USE_OPENCV
    #define int64 opencv_broken_int
    #define uint64 opencv_broken_uint
    #include <opencv2/opencv.hpp>
    #include <opencv2/core.hpp>
    #undef uint64
    #undef int64
 
    #include "CVStabilization.h"
    #include "CVTracker.h"
    #include "CVObjectDetection.h"
    #include "CVObjectMask.h"
#endif
 
#include <thread>
#include "ProcessingController.h"
#include "Clip.h"
 
namespace openshot {
 
// Constructor responsible to choose processing type and apply to clip
class ClipProcessingJobs{
    private:
        std::string processInfoJson;
        std::string processingType;
 
        bool processingDone = false;
        bool stopProcessing = false;
        uint processingProgress = 0;
 
        std::thread t;
 
        ProcessingController processingController;
 
        // Apply object tracking to clip
        void trackClip(Clip& clip, ProcessingController& controller);
        // Apply stabilization to clip
        void stabilizeClip(Clip& clip, ProcessingController& controller);
        // Apply object detection to clip
        void detectObjectsClip(Clip& clip, ProcessingController& controller);
        // Apply object segmentation mask to clip
        void maskObjectClip(Clip& clip, ProcessingController& controller);
 
 
    public:
        // Constructor
        ClipProcessingJobs(std::string processingType, std::string processInfoJson);
        static std::string ValidateONNXModel(std::string modelPath);
        static std::shared_ptr<Frame> PreviewObjectMask(std::string processInfoJson, std::shared_ptr<Frame> frame);
        // Process clip accordingly to processingType
        void processClip(Clip& clip, std::string json);
 
        // Thread related variables and methods
        int GetProgress();
        bool IsDone();
        void CancelProcessing();
        bool GetError();
        std::string GetErrorMessage();
};
 
}  // namespace openshot