使用opencv + ffmpeg 开发视频播放器Demo

临渊慎行2024-04-10 13:24
cpp 复制代码
#include <opencv2/opencv.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/videoio.hpp>

extern "C" {
#include <libavformat/avformat.h>
#include <libavcodec/avcodec.h>
#include <libswscale/swscale.h>
}

int main() {
    // 初始化 FFmpeg
    av_register_all();

    // 打开视频文件
    AVFormatContext* formatContext = avformat_alloc_context();
    if (avformat_open_input(&formatContext, "video.mp4", NULL, NULL) != 0) {
        std::cout << "无法打开视频文件" << std::endl;
        return -1;
    }

    // 获取视频流信息
    if (avformat_find_stream_info(formatContext, NULL) < 0) {
        std::cout << "无法获取视频流信息" << std::endl;
        return -1;
    }

    // 寻找视频流索引
    int videoStreamIndex = -1;
    for (int i = 0; i < formatContext->nb_streams; i++) {
        if (formatContext->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
            videoStreamIndex = i;
            break;
        }
    }

    if (videoStreamIndex == -1) {
        std::cout << "无法找到视频流" << std::endl;
        return -1;
    }

    // 获取视频解码器
    AVCodecParameters* codecParameters = formatContext->streams[videoStreamIndex]->codecpar;
    AVCodec* codec = avcodec_find_decoder(codecParameters->codec_id);
    AVCodecContext* codecContext = avcodec_alloc_context3(codec);
    avcodec_parameters_to_context(codecContext, codecParameters);
    avcodec_open2(codecContext, codec, NULL);

    // 创建窗口
    cv::namedWindow("Video Player", cv::WINDOW_NORMAL);

    // 读取视频帧
    AVPacket packet;
    AVFrame* frame = av_frame_alloc();
    while (av_read_frame(formatContext, &packet) >= 0) {
        if (packet.stream_index == videoStreamIndex) {
            avcodec_send_packet(codecContext, &packet);
            avcodec_receive_frame(codecContext, frame);

            // 转换 AVFrame 到 OpenCV Mat
            cv::Mat image(cv::Size(frame->width, frame->height), CV_8UC3, frame->data[0], frame->linesize[0]);

            // 显示图像
            cv::imshow("Video Player", image);
            cv::waitKey(25);
        }

        av_packet_unref(&packet);
    }

    // 清理
    av_frame_free(&frame);
    avcodec_close(codecContext);
    avformat_close_input(&formatContext);

    return 0;
}