在linux下,用ffmpeg解码视频,例子。(借助opencv可显示)

本文介绍如何使用FFmpeg库解码视频并利用OpenCV显示。通过C++实现,程序首先注册FFmpeg支持的文件格式及解码器,接着打开视频文件并获取流信息。找到视频流后,选择合适的解码器进行初始化,并分配原始数据帧和RGB帧内存。最后,文章展示了如何将解码后的视频帧转换为RGB格式并在OpenCV窗口中显示。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

#include <stdio.h>
#include <iostream>
#include <opencv2/core/core.hpp>    
#include <opencv2/highgui/highgui.hpp>   
extern "C"
{
	#include <libavformat/avformat.h>    
	#include <libavcodec/avcodec.h>    
	#include <libswscale/swscale.h>     
	#include <libavutil/opt.h>  

	#include <libavutil/channel_layout.h>  
	#include <libavutil/common.h>  
	#include <libavutil/imgutils.h>  
	#include <libavutil/mathematics.h>  
	#include <libavutil/samplefmt.h> 
}

using namespace std;
using namespace cv;
int main(int argc, char const *argv[])
{
    int result = 0;  
    av_register_all(); // 注册所有FFmpeg库所支持的文件格式和codec  
  
    AVFormatContext* pFormatCtx = NULL;  
    char* filename = "123.wmv"; //输入文件名  
  
    // step1: 打开媒体文件,最后2个参数是用来指定文件格式,buffer大小和格式参数,设置成NULL的话,libavformat库会自动去探测它们  
    result = avformat_open_input(&pFormatCtx, filename, NULL, NULL);  
    if (result != 0)  
    {  
        cout << "open file fail" << endl;  
        return -1;  
    }  
    cout << "open file succ" << endl;  
  
    // step2: 查找信息流的信息  
    result = avformat_find_stream_info(pFormatCtx, NULL);  
    if (result != 0)  
    {  
        cout << "find stream fail" << endl;  
        return -1;  
    }  
    cout << "find stream succ" << endl;  
  
    // step3: 打印信息  
    //av_dump_format(pFormatCtx, 0, filename, 0);  
  
    // step4:找到video流数据  
    int i = 0;  
    int videoStream = -1;  
    AVCodecContext* pCodecCtx = NULL;  
    for (i = 0; i < pFormatCtx->nb_streams; i++)  
    {  
        if (pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO)  
        {  
            videoStream = i;  
            break;  
        }  
    }  
  
    if (videoStream == -1)  
    {  
        cout << "find stream video fail" << endl;  
        return -1;  
    }  
    cout << "find stream video succ." << endl;  
  
    // 得到video编码格式  
    pCodecCtx = pFormatCtx->streams[videoStream]->codec;  
      
    // step5: 得到解码器  
    AVCodec* pCodec = NULL;  
    pCodec = avcodec_find_decoder(pCodecCtx->codec_id);  
    if (pCodec == NULL)  
    {  
        cout << "find decoder fail" << endl;  
        return -1;  
    }  
    cout << "find decoder succ" << endl;  
  
    result = avcodec_open2(pCodecCtx, pCodec, NULL);  
    if (result != 0)  
    {  
        cout << "open codec fail" << endl;  
        return -1;  
    }  
    cout << "open codec succ" << endl;  
  
    // step6: 申请原始数据帧 和 RGB帧内存  
    AVFrame* pFrame = NULL;  
    AVFrame* pFrameRGB = NULL;  
    pFrame = av_frame_alloc();  
    pFrameRGB = av_frame_alloc();  
    if (pFrame == NULL || pFrameRGB == NULL)  
    {  
        return -1;  
    }  
  
    int numBytes = avpicture_get_size(AV_PIX_FMT_RGB24, pCodecCtx->width, pCodecCtx->height);  
    uint8_t* buffer = (uint8_t*)av_malloc(numBytes * sizeof(uint8_t));  
    avpicture_fill((AVPicture*)pFrameRGB, buffer, AV_PIX_FMT_RGB24, pCodecCtx->width, pCodecCtx->height);  
  
    int frameFinishsed = 0;  
    AVPacket packet;  
    i = 0;   
  
    // step7: 创建格式转化文本  
    SwsContext * pSwxCtx = sws_getContext(  
        pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt,  
        pCodecCtx->width, pCodecCtx->height, AV_PIX_FMT_RGB24,  
        SWS_BILINEAR, 0, 0, 0);  
  
    cv::Mat image(pCodecCtx->height, pCodecCtx->width, CV_8UC3);  
    int b = 0;  
    int g = 1;  
    int r = 2;  
  
    while (true)  
    {  
        // 得到数据包  
        result = av_read_frame(pFormatCtx, &packet);  
        if (result != 0)  
        {  
            break;  
        }  
        if (packet.stream_index == videoStream)  
        {  
            // 解码  
            avcodec_decode_video2(pCodecCtx, pFrame, &frameFinishsed, &packet);   
            if (frameFinishsed)  
            {  
                // 转换  
                sws_scale(pSwxCtx, pFrame->data, pFrame->linesize, 0, pCodecCtx->height,  
                    pFrameRGB->data, pFrameRGB->linesize);  
  
                for (int m = 0; m < pCodecCtx->height; m++)  
                {  
                    for (int n = 0; n < pCodecCtx->width; n++)  
                    {  
                        image.at<Vec3b>(m, n)[r] = pFrameRGB->data[0][3 * (m*pCodecCtx->width + n) + 0];  
                        image.at<Vec3b>(m, n)[g] = pFrameRGB->data[0][3 * (m*pCodecCtx->width + n) + 1];  
                        image.at<Vec3b>(m, n)[b] = pFrameRGB->data[0][3 * (m*pCodecCtx->width + n) + 2];  
                    }  
                }  
                cv::imshow("pic", image);  
                cv::waitKey(1);  
            }  
        }  
  
        av_free_packet(&packet);  
    }  
      
    avformat_close_input(&pFormatCtx);  
  
    //system("pause");  
    return 0;  
}  

g++ test.cpp `pkg-config --cflags --libs opencv`  -lavformat -lavcodec -lswscale -lavutil  -o test.run


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Teleger

你的支持是我前进的方向

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值