TestVideo.h

Go to the documentation of this file.

#pragma once
 
#include "pch.h"
 
#include "FFmpeg/AudioFrameSink.h"
#include "FFmpeg/FFmpegRecorder.h"
#include "FFmpeg/VideoFrameSink.h"
 
#include <boost/program_options.hpp>
 
void TestVideo(const boost::program_options::variables_map &vm,
               const std::map<std::string, std::string> &ffmpegOptions) {
    std::vector<std::unique_ptr<FFmpegFrameSink>> sinks;
    sinks.push_back(std::make_unique<VideoFrameSink>());
    auto videoSink = dynamic_cast<VideoFrameSink *>(sinks[0].get());
 
    av_log_set_level(AV_LOG_QUIET);
 
    FFmpegRecorder recorder(vm["inputFormat"].as<std::string>(),
                            vm["deviceName"].as<std::string>(), ffmpegOptions,
                            sinks);
 
    std::vector<uint8_t> data;
    long long lastFrame = videoSink->getData(data);
    cv::Mat mat = cv::Mat(videoSink->getHeight(), videoSink->getWidth(),
                          CV_8UC3, data.data());
 
    std::atomic<bool> running = true;
    std::string title = std::to_string(
        std::chrono::steady_clock::now().time_since_epoch().count());
 
    std::thread t([&]() {
        while (running.load()) {
            lastFrame = videoSink->getNextData(data, lastFrame);
            cv::imshow(title, mat);
            cv::waitKey(1);
        }
    });
 
    std::cout << "Press enter to stop\n";
    std::cin.get();
    running.store(false);
    t.join();
}