FFmpeg 如何探测网络流格式/如何从内存中获取数据

文章转自：http://blog.****.net/rootusers/article/details/42551935

一般ffmpeg都是直接从文件中读取或者从网络流中读取，比如rtp://xx.xx.xx.xx:xxxx。

事实上也支持从内存中获取。

函数avio_alloc_context()实现该功能。

1. AVIOContext *avio_alloc_context(
2. unsigned char *buffer,
3. int buffer_size,
4. int write_flag,
5. void *opaque,
6. int (*read_packet)(void *opaque, uint8_t *buf, int buf_size),//重写该函数，指定从内存中读取的方法，将buf_size字节大小的数据保存到buf
7. int (*write_packet)(void *opaque, uint8_t *buf, int buf_size),//对应的这是写内存的函数
8. int64_t (*seek)(void *opaque, int64_t offset, int whence));

对于探测网络媒体流个格式，也可以用此种方法，先接收数据，然后探测。

下面贴出代码：

1. /*
2. *author tongli
3. */
4. extern "C"{
5. #include "libavformat/avformat.h"
6. #include "libavcodec/avcodec.h"
7. #include "libavutil/avutil.h"
8. }
9. #define BUF_SIZE 4096*500
10. FILE* file;
11. //实现read_packet函数，从文件中读取模拟的是从内存中获取，同样可以实现为接收网络流
12. int read_packet(void *opaque, uint8_t *buf, int buf_size)
13. {
14. int n = 0;
15. if (!feof(file)){
16. n = fread(buf, 1, buf_size, file);
17. return n;
18. }else
19. return -1;
20. }
21. int main(int argc, char** argv)
22. {
23. file = fopen("2.mp4", "rb");
24. if (file == NULL)
25. return -1;
26. av_register_all();
27. AVIOContext* pb = NULL;
28. AVInputFormat* piFmt = NULL;
29. AVInputFormat* pFmt = NULL;
30. uint8_t* buf = (uint8_t*)av_mallocz(sizeof(uint8_t)* BUF_SIZE);
31. pb = avio_alloc_context(buf, BUF_SIZE, 0, NULL, read_packet, NULL, NULL);
32. if (av_probe_input_buffer(pb, &piFmt, "", NULL, 0, 0) < 0)//探测从内存中获取到的媒体流的格式
33. {
34. fprintf(stderr, "probe format failed\n");
35. return -1;
36. }
37. else{
38. fprintf(stdout, "format:%s[%s]\n", piFmt->name, piFmt->long_name);
39. }
40. return 0;
41. }

下面实现一个简单的例子，从内存中读取，然后播放。

1. <pre name="code" class="cpp">/*
2. *author tongli
3. */
4. #include <stdio.h>
5. #include <direct.h>
6. #include <io.h>
7. extern "C"{
8. #include "libavformat/avformat.h"
9. #include "libavcodec/avcodec.h"
10. #include "libavutil/avutil.h"
11. #include "libswscale/swscale.h"
12. #include "libavformat/avio.h"
13. #include "sdl/SDL.h"
14. }
15. #define BUF_SIZE 4096 * 500
16. FILE* file;
17. int read_data(void *opaque, uint8_t *buf, int buf_size)
18. {
19. int n = 0;
20. if (!feof(file)){
21. n = fread(buf, 1, buf_size, file);
22. return n;
23. }
24. else
25. return -1;
26. }
27. int main(int argc, char* argv[])
28. {
29. av_register_all();
30. //file = fopen("h2.dat", "rb");
31. file = fopen("3.mp4", "rb+");
32. if (file == NULL)
33. return -1;
34. AVFormatContext *pFormatCtx;
35. int             i, videoindex;
36. AVCodecContext  *pCodecCtx;
37. AVCodec         *pCodec;
38. AVIOContext* pb = NULL;
39. AVInputFormat* piFmt = NULL;
40. uint8_t* buf = (uint8_t*)av_mallocz(sizeof(uint8_t)* BUF_SIZE);
41. pb = avio_alloc_context(buf, BUF_SIZE, 0, NULL, read_data, NULL, NULL);
42. if (av_probe_input_buffer(pb, &piFmt, "", NULL, 0, 0) < 0)
43. {
44. fprintf(stderr, "probe format failed\n");
45. return -1;
46. }
47. else{
48. fprintf(stdout, "format:%s[%s]\n", piFmt->name, piFmt->long_name);
49. }
50. pFormatCtx = avformat_alloc_context();
51. pFormatCtx->pb = pb;
52. if (avformat_open_input(&pFormatCtx, "", piFmt, NULL) != 0){//iformat，priv_data赋值，pb, nbstreams,streams为null
53. printf("Couldn't open input stream.（无法打开输入流）\n");
54. return -1;
55. }
56. if (avformat_find_stream_info(pFormatCtx, NULL)<0)//nbstreams,streams赋值， pb还是为null
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++) //一般情况下，一个媒体只有两个流，视频和音频流即streams[0],stream[1]
63. if (pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO)
64. {//找到视频流
65. videoindex = i;//在nb_streams视频流的索引
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. AVFrame *pFrame, *pFrameYUV;
86. pFrame = av_frame_alloc();
87. pFrameYUV = av_frame_alloc();
88. uint8_t *out_buffer;
89. out_buffer = new uint8_t[avpicture_get_size(PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height)];
90. avpicture_fill((AVPicture *)pFrameYUV, out_buffer, PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height);
91. //------------SDL----------------
92. if (SDL_Init(SDL_INIT_VIDEO | SDL_INIT_AUDIO | SDL_INIT_TIMER)) {
93. printf("Could not initialize SDL - %s\n", SDL_GetError());
94. return -1;
95. }
96. SDL_Surface *screen;
97. screen = SDL_SetVideoMode(pCodecCtx->width, pCodecCtx->height, 0, 0);
98. if (!screen) {
99. printf("SDL: could not set video mode - exiting\n");
100. return -1;
101. }
102. SDL_Overlay *bmp;
103. bmp = SDL_CreateYUVOverlay(pCodecCtx->width, pCodecCtx->height, SDL_YV12_OVERLAY, screen);
104. SDL_Rect rect;
105. //---------------
106. int ret, got_picture;
107. int y_size = pCodecCtx->width * pCodecCtx->height;
108. AVPacket *packet = (AVPacket *)av_malloc(sizeof(AVPacket));
109. av_new_packet(packet, y_size);
110. struct SwsContext *img_convert_ctx;
111. img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height,
112. AV_PIX_FMT_YUVJ420P/*pCodecCtx->pix_fmt*/, pCodecCtx->width, pCodecCtx->height,
113. PIX_FMT_YUV420P, SWS_BICUBIC, NULL, NULL, NULL);
114. //------------------------------
115. while (av_read_frame(pFormatCtx, packet) >= 0)
116. {
117. if (packet->stream_index == videoindex)
118. {
119. ret = avcodec_decode_video2(pCodecCtx, pFrame, &got_picture, packet);
120. if (ret < 0)
121. {
122. printf("Decode Error.（解码错误）\n");
123. return -1;
124. }
125. if (got_picture)
126. {
127. sws_scale(img_convert_ctx, (const uint8_t* const*)pFrame->data, pFrame->linesize, 0, pCodecCtx->height, pFrameYUV->data, pFrameYUV->linesize);
128. SDL_LockYUVOverlay(bmp);
129. bmp->pixels[0] = pFrameYUV->data[0];
130. bmp->pixels[2] = pFrameYUV->data[1];
131. bmp->pixels[1] = pFrameYUV->data[2];
132. bmp->pitches[0] = pFrameYUV->linesize[0];
133. bmp->pitches[2] = pFrameYUV->linesize[1];
134. bmp->pitches[1] = pFrameYUV->linesize[2];
135. SDL_UnlockYUVOverlay(bmp);
136. rect.x = 0;
137. rect.y = 0;
138. rect.w = pCodecCtx->width;
139. rect.h = pCodecCtx->height;
140. SDL_DisplayYUVOverlay(bmp, &rect);
141. //延时40ms
142. SDL_Delay(40);
143. }
144. }
145. av_free_packet(packet);
146. }
147. sws_freeContext(img_convert_ctx);
148. delete[] out_buffer;
149. av_free(pFrameYUV);
150. avcodec_close(pCodecCtx);
151. avformat_close_input(&pFormatCtx);
152. return 0;
153. }