Dehazer C++ lib. Fast video dehaze/defog algorithm

namespace cr
{
namespace video
{
class Dehazer : public VFilter
{
public:

    /// Class constructor.
    Dehazer();

    /// Class destructor.
    ~Dehazer();

    /// Get string of current library version.
    static std::string getVersion();

    /// Initialize video filter.
    bool initVFilter(VFilterParams& params) override;

    /// Set parameter.
    bool setParam(VFilterParam id, float value) override;

    /// Get parameter value.
    float getParam(VFilterParam id) override;

    /// Get all library parameters.
    void getParams(VFilterParams& params) override;

    /// Execute action command.
    bool executeCommand(VFilterCommand id) override;

    /// Process frame.
    bool processFrame(Frame& frame) override;

    /// Set mask for the filter.
    bool setMask(cr::video::Frame mask) override;

    /// Decode and execute command.
    bool decodeAndExecuteCommand(uint8_t* data, int size) override;
};
}
}

#include <iostream>
#include <opencv2/opencv.hpp>
#include "Dehazer.h"

int main(void)
{
    // Open video source.
    cv::VideoCapture source;
    if (!source.open("test.mp4"))
        return -1;

    // Init dehazer and set initial params.
    cr::video::Dehazer dehazer;
    dehazer.setParam(cr::video::VFilterParam::TYPE, 1);
    dehazer.setParam(cr::video::VFilterParam::LEVEL, 30);

    // Get frame size.
    int width = (int)source.get(cv::CAP_PROP_FRAME_WIDTH);
    int height = (int)source.get(cv::CAP_PROP_FRAME_HEIGHT);

    // Init frames.
    cv::Mat openCvBgrImg(height, width, CV_8UC3);
    cr::video::Frame frame(width, height, cr::video::Fourcc::YUV24);

    // Main loop.
    while (true)
    {
        // Capture next video frame.
        source >> openCvBgrImg;
        if (openCvBgrImg.empty())
            return 0;

        // Convert BGR to YUV.
        cv::Mat yuvImg(frame.height, frame.width, CV_8UC3, frame.data);
        cvtColor(openCvBgrImg, yuvImg, cv::COLOR_BGR2YUV);

        // Dehaze.
        dehazer.processFrame(frame);

        // Convert YUV to BGR.
        cvtColor(yuvImg, openCvBgrImg, cv::COLOR_YUV2BGR);

        // Show results.
        imshow("Result", openCvBgrImg);
        if (cv::waitKey(1) == 27)
            return 0;
    }
    return 1;
}