ffmpeg解码播放

王柳  金牌会员 | 2024-6-20 14:55:56 | 显示全部楼层 | 阅读模式
打印 上一主题 下一主题
楼主

主题 542|帖子 542|积分 1626

  1. int VideoDecodeModule::Open(std::string strUrl)
  2. {
  3.         AVFormatContext *pFormatCtx = nullptr;
  4.         AVCodec *pCodec = nullptr;
  5.         AVCodecContext* pCodecCtx = nullptr;
  6.         AVDictionary *opt = nullptr;
  7.         std::string decodeName = "";
  8.         AVBufferRef* pBufferRef = nullptr;
  9.         int ret = 0;
  10.         int videoStream = -1;
  11.         char errorbuf[1024] = { 0 };
  12.         av_dict_set(&opt, "buffer_size", "1024000", 0);        // 缓冲区大小 单位字节 解决高画质模糊的问题
  13.         av_dict_set(&opt, "max_delay", "100", 0);              // 最大延时 单位微妙
  14.         av_dict_set(&opt, "stimeout", "3000000", 0);           // 设置超时断开连接时间 3s 单位微妙
  15.         av_dict_set(&opt, "rtsp_transport", "tcp", 0);         // 以tcp方式打开，如果以udp方式打开将tcp替换为udp
  16.         av_dict_set(&opt, "fflags", "nobuffer", 0);
  17.         av_dict_set(&opt, "rtbufsize", "6", 0);
  18.         av_dict_set(&opt, "start_time_realtime", 0, 0);
  20.         if ((ret = avformat_open_input(&pFormatCtx, strUrl.data(), nullptr, &opt)) != 0)
  21.         {
  22.                 av_strerror(ret, errorbuf, sizeof(errorbuf));
  23.                 return -1;
  24.         }
  25.        
  26.         m_spFormatContext = std::shared_ptr<AVFormatContext>(pFormatCtx, [](AVFormatContext* ctx){
  27.                 avformat_close_input(&ctx);
  28.         });
  30.         /*if ((ret = avformat_find_stream_info(pFormatCtx, nullptr)) < 0)
  31.         {
  32.                 av_strerror(ret, errorbuf, sizeof(errorbuf));
  33.                 return -1;
  34.         }*/
  36.         if ((ret = av_find_best_stream(pFormatCtx, AVMEDIA_TYPE_VIDEO, -1, -1, &pCodec, 0)) < 0)
  37.         {
  38.                 av_strerror(ret, errorbuf, sizeof(errorbuf));
  39.                 return -1;
  40.         }
  41.         videoStream = ret;
  42.         m_video = pFormatCtx->streams[videoStream];
  44.         if (m_decodeType == DT_CPU)
  45.         {
  46.                 if (!(pCodec = avcodec_find_decoder(m_video->codecpar->codec_id)))
  47.                 {
  48.                         int error = AVERROR(ENOMEM);
  49.                         return -1;
  50.                 }
  51.         }
  52.         else if (m_decodeType == DT_INTER_QSV)
  53.         {
  54.                 decodeName = GetCodeName(m_video->codecpar->codec_id, DT_INTER_QSV);
  55.                 if (av_hwdevice_ctx_create(&pBufferRef, AV_HWDEVICE_TYPE_QSV, "auto", nullptr, 0) != 0)
  56.                 {
  57.                         return -1;
  58.                 }
  59.                 if (!(pCodec = avcodec_find_decoder_by_name(decodeName.data())))
  60.                 {
  61.                         int error = AVERROR(ENOMEM);
  62.                         return -1;
  63.                 }
  64.         }
  65.         else if (m_decodeType == DT_NVIDIA_CUDA)
  66.         {
  67.                 decodeName = GetCodeName(m_video->codecpar->codec_id, DT_NVIDIA_CUDA);
  68.                 if (av_hwdevice_ctx_create(&pBufferRef, AV_HWDEVICE_TYPE_CUDA, "auto", nullptr, 0) != 0)
  69.                 {
  70.                         return -1;
  71.                 }
  72.                 if (!(pCodec = avcodec_find_decoder_by_name(decodeName.data())))
  73.                 {
  74.                         int error = AVERROR(ENOMEM);
  75.                         return -1;
  76.                 }
  77.         }
  79.         if (auto frame = av_frame_alloc())
  80.         {
  81.                 m_spSwFrame = std::shared_ptr<AVFrame>(frame, [](AVFrame* p) {av_frame_free(&p); });
  82.         }
  83.        
  84.         if (!(pCodecCtx = avcodec_alloc_context3(pCodec)))
  85.         {
  86.                 return -1;
  87.         }
  89.         m_spCodecContext = std::shared_ptr<AVCodecContext>(pCodecCtx, [](AVCodecContext* ctx)
  90.         {
  91.                 avcodec_free_context(&ctx);
  92.         });
  94.         m_spCodecContext->codec_id = m_video->codecpar->codec_id;
  95.         if (m_video->codecpar->extradata_size)
  96.         {
  97.                 m_spCodecContext->extradata = (uint8_t*)av_mallocz(m_video->codecpar->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
  98.                 if (!m_spCodecContext->extradata)
  99.                 {
  100.                         return -1;
  101.                 }
  102.                 memcpy(m_spCodecContext->extradata, m_video->codecpar->extradata, m_video->codecpar->extradata_size);
  103.                 m_spCodecContext->extradata_size = m_video->codecpar->extradata_size;
  104.         }
  106.         m_spCodecContext->flags2 |= AV_CODEC_FLAG2_FAST;    // 允许不符合规范的加速技巧。
  107.         m_spCodecContext->thread_count = 8;                 // 使用8线程解码
  109.         if ((ret = avcodec_parameters_to_context(m_spCodecContext.get(), m_video->codecpar)) < 0)
  110.         {
  111.                 av_strerror(ret, errorbuf, sizeof(errorbuf));
  112.                 return -1;
  113.         }
  114.        
  115.         if (m_decodeType == DT_INTER_QSV)
  116.         {
  117.                 m_spCodecContext->opaque = pBufferRef;
  118.                 m_spCodecContext->get_format = GetHWQsvFormat;
  119.         }
  120.         else if (m_decodeType == DT_NVIDIA_CUDA)
  121.         {
  122.                 m_spCodecContext->hw_device_ctx = av_buffer_ref(pBufferRef);
  123.                 m_spCodecContext->opaque = this;
  124.                 m_spCodecContext->get_format = GetHWCudaFormat;
  125.         }
  127.         if ((ret = avcodec_open2(m_spCodecContext.get(), pCodec, nullptr)) < 0)
  128.         {
  129.                 av_strerror(ret, errorbuf, sizeof(errorbuf));
  130.                 return -1;
  131.         }
  132.         m_isConnect = true;
  133.         return ret;
  134. }
复制代码
avformat_find_stream_info 比较耗时，播放视频会比较慢，这里将他屏蔽掉，对播放视频没有影响
2:吸收数据
  1. void VideoDecodeModule::PushFrameQueue(std::shared_ptr<AVPacket> spPacket)
  2. {
  3.         if (spPacket && !spPacket->size && spPacket->data)
  4.         {
  5.                 return;
  6.         }
  7.         int ret = avcodec_send_packet(m_spCodecContext.get(), spPacket.get());
  8.         if (ret < 0)
  9.         {
  10.                 char errorbuf[1024] = { 0 };
  11.                 av_strerror(ret, errorbuf, sizeof(errorbuf));
  12.                 return;
  13.         }
  14.         while (true)
  15.         {
  16.                 auto spFrame = std::shared_ptr<AVFrame>(av_frame_alloc(), [](AVFrame* p)
  17.                 {
  18.                         av_frame_free(&p);
  19.                 });
  20.                 if (!spFrame)
  21.                 {
  22.                         return;
  23.                 }
  24.                 ret = avcodec_receive_frame(m_spCodecContext.get(), spFrame.get());
  25.                 if (ret == AVERROR_EOF || ret == AVERROR(EAGAIN))
  26.                 {
  27.                         return;
  28.                 }
  29.                 if (ret < 0)
  30.                 {
  31.                         break;
  32.                 }
  34.                 if (m_decodeType == DT_INTER_QSV || m_decodeType == DT_NVIDIA_CUDA)
  35.                 {
  36.                         HandleHWVideoFrame(spFrame);
  37.                 }
  38.                 else if (m_decodeType == DT_CPU)
  39.                 {
  40.                         HandleVideoFrame(spFrame);
  41.                 }
  42.         }
  43. }
复制代码
3：吸收视频帧
  2. void VideoDecodeModule::HandleVideoFrame(const std::shared_ptr<AVFrame>& spFrame)
  3. {
  4.         if (!spFrame)
  5.                 return;
  6.         //if (GetFrameExtra() % 6 != 0)
  7.         //        return;
  8.         /// 解码的图片宽
  9.         int width = spFrame->width;
  10.         /// 解码的图片高
  11.         int height = spFrame->height;
  12.         // 计算转码后的图片裸数据需要的大小
  13.         int nSrcbuffSize = av_image_get_buffer_size(m_pixelFormat, width, height, 1);
  14.         if (nSrcbuffSize > m_dstBuffer.size())
  15.         {
  16.                 m_dstBuffer.resize(nSrcbuffSize, '\0');
  17.         }
  18.         std::shared_ptr<SwsContext>        spSwsContext = nullptr;
  19.         /// 判断是否需要格式转换
  20.         if (m_pixelFormat != (AVPixelFormat)spFrame->format)
  21.         {
  22.                 AVPixelFormat srcFmt = (AVPixelFormat)spFrame->format;
  23.                 //SwsContext* image_sws_ctx = sws_getContext(width, height, srcFmt, m_scaleWidth, m_scaleHeight, m_pixelFormat, SWS_FAST_BILINEAR, nullptr, nullptr, nullptr);
  24.                
  25.                 SwsContext* image_sws_ctx = sws_getCachedContext(NULL, width, height, srcFmt, m_scaleWidth, m_scaleHeight, m_pixelFormat, SWS_FAST_BILINEAR, nullptr, nullptr, nullptr);
  26.                
  27.                 spSwsContext = std::shared_ptr<SwsContext>(image_sws_ctx, [](SwsContext* p) {sws_freeContext(p); });
  28.                 uint8_t* data[4] = { nullptr };
  29.                 int linesize[4] = { 0 };
  30.                 av_image_fill_arrays(data, linesize, m_dstBuffer.data(), m_pixelFormat, width, height, 1);
  32.                 int ret = sws_scale(spSwsContext.get(), (uint8_t const* const*)spFrame->data, spFrame->linesize, 0, height, data, linesize);
  33.                 if (ret < 0)
  34.                         return;
  35.         }
  37.         if (m_HwndShow)
  38.         {
  39.                 m_spSdlRender->DrawFrame(spFrame.get());
  40.         }
  43.         /*if (m_HwndShow)
  44.         {
  45.                 BYTE** data = spFrame->data;
  46.                 int* linesize = spFrame->linesize;
  48.                 m_SdlTexture = SDL_CreateTexture(m_SdlRender, SDL_PIXELFORMAT_IYUV, SDL_TEXTUREACCESS_STREAMING, width, height);
  49.                 SDL_UpdateYUVTexture(m_SdlTexture, NULL, data[0], linesize[0],
  50.                         data[1], linesize[1],
  51.                         data[2], linesize[2]);
  52.                 SDL_RenderClear(m_SdlRender);
  53.                 SDL_RenderCopy(m_SdlRender, m_SdlTexture, NULL, NULL);
  54.                 SDL_RenderPresent(m_SdlRender);
  55.         }*/
  57.         if (m_pFrameDataCallBack)
  58.         {
  59.                 FrameInfo info;
  60.                 info.format = GetVideoFormatByAVPixelFormat(m_pixelFormat);
  61.                 info.width = width;
  62.                 info.height = height;
  63.                 info.pts = spFrame->pts / 1000;
  65.                 m_pFrameDataCallBack->cbFrameDataCallBack(m_dstBuffer.data(), m_dstBuffer.size(), info, m_pFrameDataCallBack->pUser);
  66.         }
  67. }
复制代码
工程：https://download.csdn.net/download/weixin_38887743/88748651

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
回复

使用道具 举报

0 个回复

倒序浏览
返回列表

快速回复

高级模式
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

王柳

金牌会员
这个人很懒什么都没写!

楼主热帖

标签云

挺好的 服务器
微信订阅号
微信服务号
微信客服
小程序
H5
关于我们 商务合作 网站地图
©Copyright 2022-2024 杭州祥升科技有限公司 版权所有 浙ICP备20004199号
快速回复 返回顶部 返回列表