ToB企服应用市场:ToB评测及商务社交产业平台

标题: SDL直接渲染yuv视频帧数据 [打印本页]

---

作者: 西河刘卡车医    时间: 2024-8-21 11:10
标题: SDL直接渲染yuv视频帧数据
用ffmpeg解码出来的视频数据都是yuv视频帧，直接用sdl渲染可以有效的降低系统的cpu资源
直接上代码把 

1. SDLRender::SDLRender()
2. {
3.         SDL_Init(SDL_INIT_VIDEO);
4.         m_spN12Chech = std::make_shared<std::vector<uint8_t>>();
5. }

复制代码

1. bool SDLRender::DrawFrame(AVFrame *frame)
2. {
3.         std::unique_lock<std::mutex> locker(m_mtDrawFrame);
4.         if (!frame)
5.         {
6.                 return false;
7.         }
9.         if (!m_SdlRender)
10.         {
11.                 m_SdlRender = SDL_CreateRenderer(m_SdlWindow, -1, SDL_RendererFlags::SDL_RENDERER_ACCELERATED);
12.         }
14.         if (frame->width != m_tureWidth || frame->height != m_tureHeight || !m_SdlTexture)
15.         {
16.                 m_tureWidth = frame->width;
17.                 m_tureHeight = frame->height;
19.                 if (m_SdlTexture)
20.                 {
21.                         SDL_DestroyTexture(m_SdlTexture);
22.                         m_SdlTexture = nullptr;
23.                 }
24.                
25.                 m_SdlTexture = SDL_CreateTexture(m_SdlRender, GetSDLFormat((AVPixelFormat)frame->format), SDL_TEXTUREACCESS_STREAMING, frame->width, frame->height);
26.         }
28.         int linesize = 0;
29.         switch (frame->format)
30.         {
31.         case AV_PIX_FMT_YUVJ420P:
32.         case AV_PIX_FMT_YUV420P:
33.                 return Draw(frame->data[0], frame->linesize[0], frame->data[1], frame->linesize[1], frame->data[2], frame->linesize[2]);
34.         case AV_PIX_FMT_NV12:
35.                 if (m_spN12Chech)
36.                 {
37.                         m_spN12Chech->clear();
38.                 }
39.                 else
40.                 {
41.                         return false;
42.                 }
43.                 linesize = frame->width;
44.                 if (frame->linesize[0] == frame->width)
45.                 {
46.                         // 拷贝所有Y分量
47.                         m_spN12Chech->insert(m_spN12Chech->end(), frame->data[0], frame->data[0] + frame->linesize[0] * frame->height);
48.                         // 拷贝所有的UV分量
49.                         m_spN12Chech->insert(m_spN12Chech->end(), frame->data[1], frame->data[1] + frame->linesize[1] * frame->height/2);
50.                 }
51.                 return Draw(m_spN12Chech->data(), linesize);
52.         case AV_PIX_FMT_RGBA:
53.         case AV_PIX_FMT_BGRA:
54.         case AV_PIX_FMT_ARGB:
55.                 return Draw(frame->data[0], frame->linesize[0]);
56.         default:
57.                 break;
58.         }
60.         return false;
61. }

复制代码

1. bool SDLRender::Draw(const unsigned char* y, int y_pitch, const unsigned char* u, int u_pitch, const unsigned char* v, int v_pitch)
2. {
3.         if (!y || !u || !v)
4.         {
5.                 return false;
6.         }
8.         std::unique_lock<std::mutex> locker(m_mtSdl);
9.         if (!m_SdlWindow || !m_SdlRender || !m_SdlTexture)
10.         {
11.                 return false;
12.         }
14.         // 复制内存到显存
15.         auto re = SDL_UpdateYUVTexture(m_SdlTexture, NULL, y, y_pitch, u, u_pitch, v, v_pitch);
16.         if (re)
17.         {
18.                 return false;
19.         }
21.         // 清理渲染器
22.         SDL_RenderClear(m_SdlRender);
24.         // 如果用户手动设置了缩放，就按照用户设置的大小显示
25.         // 如果用户没有设置，就传递null, 采用默认的窗口大小
26.         SDL_Rect* prect = nullptr;
27.         if (m_nScaleWidth > 0 || m_nScaleHeight > 0)
28.         {
29.                 SDL_Rect rect;
30.                 rect.x = 0;
31.                 rect.y = 0;
32.                 rect.w = m_nScaleWidth;
33.                 rect.h = m_nScaleHeight;
34.                 prect = &rect;
35.         }
37.         // 拷贝材质到渲染器
38.         re = SDL_RenderCopy(m_SdlRender, m_SdlTexture, NULL, prect);
39.         if (re)
40.         {
41.                 return false;
42.         }
44.         SDL_RenderPresent(m_SdlRender);
45.         return true;
46. }

复制代码

1. bool SDLRender::Draw(const unsigned char* data, int linesize)
2. {
3.         if (!data)
4.         {
5.                 return false;
6.         }
8.         std::unique_lock<std::mutex> locker(m_mtSdl);
9.         if (!m_SdlWindow || !m_SdlRender || !m_SdlTexture)
10.         {
11.                 return false;
12.         }
14.         if (linesize <= 0)
15.         {
16.                 return false;
17.         }
19.         // 复制内存到显存
20.         auto re = SDL_UpdateTexture(m_SdlTexture, NULL, data, linesize);
21.         if (re)
22.         {
23.                 return false;
24.         }
26.         // 清理渲染器
27.         SDL_RenderClear(m_SdlRender);
29.         // 如果用户手动设置了缩放，就按照用户设置的大小显示
30.         // 如果用户没有设置，就传递null, 采用默认的窗口大小
31.         SDL_Rect *prect = nullptr;
32.         if (m_nScaleWidth > 0 || m_nScaleHeight > 0)
33.         {
34.                 SDL_Rect rect;
35.                 rect.x = 0;
36.                 rect.y = 0;
37.                 rect.w = m_nScaleWidth;
38.                 rect.h = m_nScaleHeight;
39.                 prect = &rect;
40.         }
41.         re = SDL_RenderCopy(m_SdlRender, m_SdlTexture, NULL, prect);
42.         if (re)
43.         {
44.                 return false;
45.         }
47.         SDL_RenderPresent(m_SdlRender);
48.         return true;
49. }

复制代码

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

---

欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/)

Powered by Discuz! X3.4