ffmpeg实现接收RTSP流并抓拍生存图片的功能

rtsp_capture.h

1. #ifndef RTSP_CAPTURE_H
2. #define RTSP_CAPTURE_H
4. #pragma once
5. #include <string>
6. #include <thread>
7. #include <atomic>
9. extern "C" {
10. #include <libavcodec/avcodec.h>
11. #include <libavformat/avformat.h>
12. #include <libavutil/imgutils.h>
13. #include <libswscale/swscale.h>
14. }
16. class RtspCapture {
17. public:
18.   RtspCapture(const std::string& url);
19.   ~RtspCapture();
21.   void take(uint32_t width, uint32_t height);
23. private:
24.   bool initFFmpeg();
25.   void cleanup();
26.   bool saveFrame(const std::string& filename, AVFrame* frame, uint32_t width, uint32_t height);
28. private:
29.   std::string m_url;
30.   std::string m_outputDir;
32.   AVFormatContext* m_formatContext;
33.   AVCodecContext* m_codecContext;
34.   SwsContext* m_swsContext;
35.   int m_videoStreamIndex;
36.   std::string m_rtspUrl;
37. };
39. #endif // RTSPCAPTURE_H

复制代码

rtsp_capture.cpp

1. #include "rtsp_capture.h"
2. #include <chrono>
3. #include <iostream>
5. RtspCapture::RtspCapture(const std::string& url)
6.     : m_url(url)
7.       , m_formatContext(nullptr)
8.       , m_codecContext(nullptr)
9.       , m_swsContext(nullptr)
10.       , m_videoStreamIndex(-1)
11. {
12.   initFFmpeg();
13. }
15. RtspCapture::~RtspCapture() {
16.   cleanup();
17. }
19. bool RtspCapture::initFFmpeg() {
21. #if LIBAVFORMAT_VERSION_MAJOR < 58
22.   av_register_all(); // 对于旧版本，显式调用以确保所有编解码器和格式被注册
23. #endif
25.   avformat_network_init();
27.   // 设置RTSP超时选项
28.   AVDictionary* options = nullptr;
29.   // av_dict_set(&options, "rtsp_transport", "tcp", 0);
30.   av_dict_set(&options, "stimeout", "30000000", 0);
31.   av_dict_set(&options, "max_delay", "500000", 0);  // 设置最大延迟
32.   av_dict_set(&options, "buffer_size", "1024000", 0);  // 设置缓冲区大小
34.   // av_log_set_level(AV_LOG_DEBUG);
36.   // 打开RTSP流
37.   m_formatContext = avformat_alloc_context();
38.   if (avformat_open_input(&m_formatContext, m_url.c_str(), nullptr, &options) != 0) {
39.     std::cerr << "无法打开RTSP流:" << m_url << std::endl;
40.     return false;
41.   }
43.   std::cout << "打开RTSP流成功:" << m_url << std::endl;
45.   // 获取流信息
46.   if (avformat_find_stream_info(m_formatContext, nullptr) < 0) {
47.     std::cerr << "无法获取流信息" << std::endl;
48.     return false;
49.   }
51.   // 查找视频流
52.   m_videoStreamIndex = -1;
53.   for (unsigned int i = 0; i < m_formatContext->nb_streams; i++) {
54.     if (m_formatContext->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
55.       m_videoStreamIndex = i;
56.       break;
57.     }
58.   }
60.   if (m_videoStreamIndex == -1) {
61.     std::cerr << "找不到视频流" << std::endl;
62.     return false;
63.   }
65.   // 获取解码器
66.   const AVCodec* codec = avcodec_find_decoder(m_formatContext->streams[m_videoStreamIndex]->codecpar->codec_id);
67.   if (!codec) {
68.     std::cerr << "找不到解码器" << std::endl;
69.     return false;
70.   }
72.   // 创建解码器上下文
73.   m_codecContext = avcodec_alloc_context3(codec);
74.   avcodec_parameters_to_context(m_codecContext, m_formatContext->streams[m_videoStreamIndex]->codecpar);
76.   // 打开解码器
77.   if (avcodec_open2(m_codecContext, codec, nullptr) < 0) {
78.     std::cerr << "无法打开解码器" << std::endl;
79.     return false;
80.   }
82.   std::cout << "FFmpeg初始化成功!" << std::endl;
84.   return true;
85. }
87. static uint64_t GetCurTick()
88. {
89.   auto now = std::chrono::system_clock::now();
90.   auto duration = now.time_since_epoch();
91.   return std::chrono::duration_cast<std::chrono::milliseconds>(duration).count();
92. }
94. void RtspCapture::take(uint32_t width, uint32_t height) {
95.   AVPacket* packet = av_packet_alloc();
96.   AVFrame* frame = av_frame_alloc();
97.   if (av_read_frame(m_formatContext, packet) >= 0) {
98.     if (packet->stream_index == m_videoStreamIndex) {
99.       // 解码视频帧
100.       if (avcodec_send_packet(m_codecContext, packet) >= 0) {
101.         if (avcodec_receive_frame(m_codecContext, frame) >= 0) {
102.           // 生成文件名
103.           std::string filename = "img.jpg";
104.           // std::string filename = "snapshot_" +
105.           //                        std::to_string(GetCurTick()) + ".jpg";
107.           // 保存帧为JPEG
108.           saveFrame(filename, frame, width, height);
109.         }
110.       }
111.     }
112.     av_packet_unref(packet);
113.   }
115.   av_frame_free(&frame);
116.   av_packet_free(&packet);
118. }
120. bool RtspCapture::saveFrame(const std::string& filename,  AVFrame* frame, uint32_t width, uint32_t height) {
122.   // 获取JPEG编码器
123.   const AVCodec* jpegCodec = avcodec_find_encoder(AV_CODEC_ID_MJPEG);
124.   if (!jpegCodec) {
125.     std::cerr << "无法找到JPEG编码器" << std::endl;
126.     return false;
127.   }
129.   // 创建编码器上下文
130.   AVCodecContext* jpegContext = avcodec_alloc_context3(jpegCodec);
131.   if (!jpegContext) {
132.     std::cerr << "无法创建JPEG编码器上下文" << std::endl;
133.     return false;
134.   }
136.   // 设置编码器参数
137.   jpegContext->width = width;
138.   jpegContext->height = height;
139.   jpegContext->time_base = (AVRational){1, 1};
140.   jpegContext->pix_fmt = AV_PIX_FMT_YUVJ420P;  // JPEG使用的颜色空间
141.   jpegContext->codec_type = AVMEDIA_TYPE_VIDEO;
142.   jpegContext->codec_id = AV_CODEC_ID_MJPEG;
144.   // 打开编码器
145.   if (avcodec_open2(jpegContext, jpegCodec, nullptr) < 0) {
146.     std::cerr << "无法打开JPEG编码器" << std::endl;
147.     avcodec_free_context(&jpegContext);
148.     return false;
149.   }
151.   // 分配转换后的帧
152.   AVFrame* yuvFrame = av_frame_alloc();
153.   yuvFrame->format = AV_PIX_FMT_YUVJ420P;
154.   yuvFrame->width = m_codecContext->width;
155.   yuvFrame->height = m_codecContext->height;
156.   av_frame_get_buffer(yuvFrame, 0);
158.   // 创建缩放上下文
159.   SwsContext* swsContext = sws_getContext(
160.       m_codecContext->width, m_codecContext->height, m_codecContext->pix_fmt,
161.       m_codecContext->width, m_codecContext->height, AV_PIX_FMT_YUVJ420P,
162.       SWS_BILINEAR, nullptr, nullptr, nullptr
163.       );
165.   if (!swsContext) {
166.     std::cerr << "无法创建转换上下文" << std::endl;
167.     av_frame_free(&yuvFrame);
168.     avcodec_free_context(&jpegContext);
169.     return false;
170.   }
172.   // 转换颜色空间
173.   sws_scale(swsContext, frame->data, frame->linesize,
174.             0, m_codecContext->height, yuvFrame->data, yuvFrame->linesize);
176.   // 编码为JPEG
177.   AVPacket* jpegPacket = av_packet_alloc();
178.   int ret = avcodec_send_frame(jpegContext, yuvFrame);
179.   if (ret < 0) {
180.     std::cerr << "发送帧到编码器失败" << std::endl;
181.     av_packet_free(&jpegPacket);
182.     av_frame_free(&yuvFrame);
183.     avcodec_free_context(&jpegContext);
184.     sws_freeContext(swsContext);
185.     return false;
186.   }
188.   ret = avcodec_receive_packet(jpegContext, jpegPacket);
189.   if (ret < 0) {
190.     std::cerr << "从编码器接收数据失败" << std::endl;
191.     av_packet_free(&jpegPacket);
192.     av_frame_free(&yuvFrame);
193.     avcodec_free_context(&jpegContext);
194.     sws_freeContext(swsContext);
195.     return false;
196.   }
198.   // 写入文件
199.   FILE* outFile = fopen(filename.c_str(), "wb");
200.   if (outFile) {
201.     fwrite(jpegPacket->data, 1, jpegPacket->size, outFile);
202.     fclose(outFile);
203.     // std::cout << "已保存图片: " << filename << std::endl;
204.   } else {
205.     std::cerr << "无法创建输出文件" << std::endl;
206.   }
208.   return true;
209. }
211. void RtspCapture::cleanup() {
212.   if (m_codecContext) {
213.     avcodec_free_context(&m_codecContext);
214.   }
215.   if (m_formatContext) {
216.     avformat_close_input(&m_formatContext);
217.     avformat_free_context(m_formatContext);
218.   }
219. }

复制代码

main.cpp

1. #include "rtsp_capture.h"
3. int main()
4. {
5.     std::unique_ptr<RtspCapture> rtsp_capture_ptr_;
6.     std::string url = "rtsp://admin:passwd@127.0.0.1:554/live1";
7.     rtsp_capture_ptr_ = std::make_unique<RtspCapture>(url);
8.     rtsp_capture_ptr_->take(640, 480);
10.     getchar();
11.     return 0;
12. }

复制代码

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