利用FFmpeg 將 rtsp 獲取H264裸流並保存到文件中

時間  2021-04-07
標籤 服務器 網絡 ide 學習 編碼 url spa .net 調試 欄目 系統網絡 简体版
原文   原文鏈接

既然已經能夠經過 RTSP 獲取h264 裸流了。那麼經過 FFmpeg 將其保存到文件中怎麼作呢？服務器

1、首先RTSP獲取 h264 裸流

咱們上面兩篇文章主要講的是經過 rtsp://Your ip:554/stream_chn0.h265 播放H.265視頻流。網絡

PS：我剛試了一下，個人 FFmpeg 程序暫時不支持 h265 ...   以前編譯的時候，只提供了 x264沒有x265ide

若是感興趣參看下面兩篇文章添加。學習

參看：使用VS2015添加對ffmpeg添加h265 支持。ui

參看：ffmpeg 編碼H265和H264對比編碼

再結合以前講的，FFmepg 再學習系列，應該是沒問題的。不過很久沒有弄了，早忘了了。url

 

那如今沒有能夠播放的 H.264 視頻流了啊，怎麼辦？spa

有辦法以前講過一篇文章，參看：LIVE555再學習 -- VLC搭建RTSP服務器（轉） 用VLC搭建一個不就完了。.net

 

固然還能夠直接用 live555，參看：LIVE555再學習 -- live555實現RTSP直播服務器  （推薦）調試

2、FFmpeg 將H.264 裸流保存到文件

這個也好說，以前有講到，參看：FFmpeg再學習 -- SDL 環境搭建和視頻顯示

將其改改就能夠了。

具體代碼以下：

參看：利用ffmpeg將RTSP傳輸的h264原始碼流保存到文件中

 

[cpp]  view plain  copy
 
  1. #include "stdafx.h"  
  2.   
  3. #include <stdio.h>    
  4.   
  5. #define __STDC_CONSTANT_MACROS    
  6.   
  7. #ifdef _WIN32    
  8. //Windows    
  9. extern "C"  
  10. {  
  11. #include "libavcodec/avcodec.h"    
  12. #include "libavformat/avformat.h"    
  13. #include "libswscale/swscale.h"    
  14. #include "SDL2/SDL.h"    
  15. };  
  16. #else    
  17. //Linux...    
  18. #ifdef __cplusplus    
  19. extern "C"  
  20. {  
  21. #endif    
  22. #include <libavcodec/avcodec.h>    
  23. #include <libavformat/avformat.h>    
  24. #include <libswscale/swscale.h>    
  25. #include <SDL2/SDL.h>    
  26. #ifdef __cplusplus    
  27. };  
  28. #endif    
  29. #endif    
  30.   
  31. int main(int argc, char* argv[])  
  32. {  
  33.   
  34.     AVFormatContext *pFormatCtx;  
  35.     int             i, videoindex;  
  36.     AVCodecContext  *pCodecCtx;  
  37.     AVCodec         *pCodec;  
  38.     AVFrame *pFrame, *pFrameYUV;  
  39.     uint8_t *out_buffer;  
  40.     AVPacket *packet;  
  41.     int ret, got_picture;  
  42.   
  43.   
  44.     struct SwsContext *img_convert_ctx;  
  45.     // 改爲你本身的 URL  
  46.     char filepath[] = "rtsp://192.168.2.xx:8554/1";   
  47.     av_register_all();  
  48.     avformat_network_init();  
  49.     pFormatCtx = avformat_alloc_context();  
  50.   
  51.     if (avformat_open_input(&pFormatCtx, filepath, NULL, NULL) != 0)////打開網絡流或文件流    
  52.     {  
  53.         printf("Couldn't open input stream.\n");  
  54.         return -1;  
  55.     }  
  56.     if (avformat_find_stream_info(pFormatCtx, NULL)<0)  
  57.     {  
  58.         printf("Couldn't find stream information.\n");  
  59.         return -1;  
  60.     }  
  61.     videoindex = -1;  
  62.     for (i = 0; i<pFormatCtx->nb_streams; i++)  
  63.         if (pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO)  
  64.         {  
  65.             videoindex = i;  
  66.             break;  
  67.         }  
  68.     if (videoindex == -1)  
  69.     {  
  70.         printf("Didn't find a video stream.\n");  
  71.         return -1;  
  72.     }  
  73.     pCodecCtx = pFormatCtx->streams[videoindex]->codec;  
  74.     pCodec = avcodec_find_decoder(pCodecCtx->codec_id);  
  75.     if (pCodec == NULL)  
  76.     {  
  77.         printf("Codec not found.\n");  
  78.         return -1;  
  79.     }  
  80.     if (avcodec_open2(pCodecCtx, pCodec, NULL)<0)  
  81.     {  
  82.         printf("Could not open codec.\n");  
  83.         return -1;  
  84.     }  
  85.     pFrame = av_frame_alloc();  
  86.     pFrameYUV = av_frame_alloc();  
  87.     out_buffer = (uint8_t *)av_malloc(avpicture_get_size(PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height));  
  88.     avpicture_fill((AVPicture *)pFrameYUV, out_buffer, PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height);  
  89.   
  90.     //Output Info---輸出一些文件（RTSP）信息    
  91.     printf("---------------- File Information ---------------\n");  
  92.     av_dump_format(pFormatCtx, 0, filepath, 0);  
  93.     printf("-------------------------------------------------\n");  
  94.   
  95.     img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt,  
  96.         pCodecCtx->width, pCodecCtx->height, PIX_FMT_YUV420P, SWS_BICUBIC, NULL, NULL, NULL);  
  97.   
  98.   
  99.     packet = (AVPacket *)av_malloc(sizeof(AVPacket));  
  100.   
  101.     FILE *fpSave;  
  102.     if ((fpSave = fopen("geth264.h264", "ab")) == NULL) //h264保存的文件名    
  103.         return 0;  
  104.     for (;;)  
  105.     {  
  106.         //------------------------------    
  107.         if (av_read_frame(pFormatCtx, packet) >= 0)  
  108.         {  
  109.             if (packet->stream_index == videoindex)  
  110.             {  
  111.                 fwrite(packet->data, 1, packet->size, fpSave);//寫數據到文件中    
  112.             }  
  113.             av_free_packet(packet);  
  114.         }  
  115.     }  
  116.   
  117.   
  118.     //--------------    
  119.     av_frame_free(&pFrameYUV);  
  120.     av_frame_free(&pFrame);  
  121.     avcodec_close(pCodecCtx);  
  122.     avformat_close_input(&pFormatCtx);  
  123.   
  124.     return 0;  
  125. }  

 

 

調試結果顯示以下：



生成 geth264.h264 文件，可播放。

3、工程下載

下載：利用FFmpeg 將 rtsp 獲取H264裸流並保存到文件中 工程

思考，這裏就有兩個問題未完成，一個就是怎麼將 H265的裸流保存到文件，再有怎麼保存成其餘格式好比MP4。

保存到MP4文件代碼以下：

 

[cpp]  view plain  copy
 
    1. #include "stdafx.h"  
    2. #ifdef __cplusplus  
    3. extern "C" {  
    4. #endif  
    5. #include <libavcodec/avcodec.h>  
    6. #include <libavdevice/avdevice.h>  
    7. #include <libavformat/avformat.h>  
    8. #include <libavfilter/avfilter.h>  
    9. #include <libavutil/avutil.h>  
    10. #include <libswscale/swscale.h>  
    11. #include <stdlib.h>  
    12. #include <stdio.h>  
    13. #include <string.h>  
    14. #include <math.h>  
    15. #ifdef __cplusplus  
    16. }  
    17. #endif  
    18.   
    19. AVFormatContext *i_fmt_ctx;  
    20. AVStream *i_video_stream;  
    21. AVFormatContext *o_fmt_ctx;  
    22. AVStream *o_video_stream;  
    23.   
    24. int _tmain(int argc, char **argv)  
    25. {  
    26.     avcodec_register_all();  
    27.     av_register_all();  
    28.     avformat_network_init();  
    29.   
    30.     /* should set to NULL so that avformat_open_input() allocate a new one */  
    31.     i_fmt_ctx = NULL;  
    32.     char rtspUrl[] = "rtsp://192.168.2.xx:8554/H264unicast";  
    33.     const char *filename = "1.mp4";  
    34.     if (avformat_open_input(&i_fmt_ctx, rtspUrl, NULL, NULL) != 0)  
    35.     {  
    36.         fprintf(stderr, "could not open input file\n");  
    37.         return -1;  
    38.     }  
    39.   
    40.     if (avformat_find_stream_info(i_fmt_ctx, NULL)<0)  
    41.     {  
    42.         fprintf(stderr, "could not find stream info\n");  
    43.         return -1;  
    44.     }  
    45.   
    46.     //av_dump_format(i_fmt_ctx, 0, argv[1], 0);  
    47.   
    48.     /* find first video stream */  
    49.     for (unsigned i = 0; i<i_fmt_ctx->nb_streams; i++)  
    50.     {  
    51.         if (i_fmt_ctx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO)  
    52.         {  
    53.             i_video_stream = i_fmt_ctx->streams[i];  
    54.             break;  
    55.         }  
    56.     }  
    57.     if (i_video_stream == NULL)  
    58.     {  
    59.         fprintf(stderr, "didn't find any video stream\n");  
    60.         return -1;  
    61.     }  
    62.   
    63.     avformat_alloc_output_context2(&o_fmt_ctx, NULL, NULL, filename);  
    64.   
    65.     /* 
    66.     * since all input files are supposed to be identical (framerate, dimension, color format, ...) 
    67.     * we can safely set output codec values from first input file 
    68.     */  
    69.     o_video_stream = avformat_new_stream(o_fmt_ctx, NULL);  
    70.     {  
    71.         AVCodecContext *c;  
    72.         c = o_video_stream->codec;  
    73.         c->bit_rate = 400000;  
    74.         c->codec_id = i_video_stream->codec->codec_id;  
    75.         c->codec_type = i_video_stream->codec->codec_type;  
    76.         c->time_base.num = i_video_stream->time_base.num;  
    77.         c->time_base.den = i_video_stream->time_base.den;  
    78.         fprintf(stderr, "time_base.num = %d time_base.den = %d\n", c->time_base.num, c->time_base.den);  
    79.         c->width = i_video_stream->codec->width;  
    80.         c->height = i_video_stream->codec->height;  
    81.         c->pix_fmt = i_video_stream->codec->pix_fmt;  
    82.         printf("%d %d %d", c->width, c->height, c->pix_fmt);  
    83.         c->flags = i_video_stream->codec->flags;  
    84.         c->flags |= CODEC_FLAG_GLOBAL_HEADER;  
    85.         c->me_range = i_video_stream->codec->me_range;  
    86.         c->max_qdiff = i_video_stream->codec->max_qdiff;  
    87.         c->qmin = i_video_stream->codec->qmin;  
    88.         c->qmax = i_video_stream->codec->qmax;  
    89.         c->qcompress = i_video_stream->codec->qcompress;  
    90.     }  
    91.   
    92.     avio_open(&o_fmt_ctx->pb, filename, AVIO_FLAG_WRITE);  
    93.     avformat_write_header(o_fmt_ctx, NULL);  
    94.   
    95.     int last_pts = 0;  
    96.     int last_dts = 0;  
    97.     int64_t pts, dts;  
    98.     while (1)  
    99.     {  
    100.         AVPacket i_pkt;  
    101.         av_init_packet(&i_pkt);  
    102.         i_pkt.size = 0;  
    103.         i_pkt.data = NULL;  
    104.         if (av_read_frame(i_fmt_ctx, &i_pkt) <0)  
    105.             break;  
    106.         /* 
    107.         * pts and dts should increase monotonically 
    108.         * pts should be >= dts 
    109.         */  
    110.         i_pkt.flags |= AV_PKT_FLAG_KEY;  
    111.         pts = i_pkt.pts;  
    112.         i_pkt.pts += last_pts;  
    113.         dts = i_pkt.dts;  
    114.         i_pkt.dts += last_dts;  
    115.         i_pkt.stream_index = 0;  
    116.   
    117.         //printf("%lld %lld\n", i_pkt.pts, i_pkt.dts);  
    118.         static int num = 1;  
    119.         printf("frame %d\n", num++);  
    120.         av_interleaved_write_frame(o_fmt_ctx, &i_pkt);  
    121.         //av_free_packet(&i_pkt);  
    122.         //av_init_packet(&i_pkt);  
    123.     }  
    124.     last_dts += dts;  
    125.     last_pts += pts;  
    126.     avformat_close_input(&i_fmt_ctx);  
    127.     av_write_trailer(o_fmt_ctx);  
    128.     avcodec_close(o_fmt_ctx->streams[0]->codec);  
    129.     av_freep(&o_fmt_ctx->streams[0]->codec);  
    130.     av_freep(&o_fmt_ctx->streams[0]);  
    131.     avio_close(o_fmt_ctx->pb);  
    132.     av_free(o_fmt_ctx);  
    133.     return 0;  
    134. }  
相關文章
相關標籤/搜索
每日一句
    每一个你不满意的现在，都有一个你没有努力的曾经。
本站公眾號
   歡迎關注本站公眾號,獲取更多信息
聯繫我們 最近搜索 最新文章 沪ICP备13005482号-10 MyBatis教程 SQL 教程 MySQL教程 Java 教程 Thymeleaf 教程 Hibernate教程 Spring教程 Redis教程