如何用FFmpeg API采集摄像头视频和麦克风音频，并实现录制文件的功能

之前一直用Directshow技术采集摄像头数据，但是觉得涉及的细节比较多，要开发者比较了解Directshow的框架知识，学习起来有一点点难度。最近发现很多人问怎么用FFmpeg采集摄像头图像，事实上FFmpeg很早就支持通过DShow获取采集设备（摄像头、麦克风）的数据了，只是网上提供的例子比较少。如果能用FFmpeg实现采集、编码和录制（或推流），那整个实现方案就简化很多，正因为这个原因，我想尝试做一个FFmpeg采集摄像头视频和麦克风音频的程序。经过一个星期的努力，终于做出来了。我打算把开发的心得和经验分享给大家。我分三部分来讲述：首先第一部分介绍如何用FFmpeg的官方工具（ffmpeg.exe）通过命令行来枚举DShow设备和采集摄像头图像，这部分是基础，能够快速让大家熟悉怎么用FFmpeg测试摄像头采集；第二部分介绍我写的采集程序的功能和用法；第三部分讲解各个模块包括采集、编码、封装和录制是如何实现的。

1.用命令行枚举采集设备和采集数据

打开Cmd命令行控制台，进入FFmpeg的Bin目录，输入如下命令：

ffmpeg -list_devices true -f dshow -i dummy

则在我的机器上显示如下结果：

在上面的命令行窗口中列出了两个设备，一个是视频采集设备，另外是一个音频采集设备。另外，我们发现：音频设备的名称有乱码，因为其中有中文名称，后面在讲到用API采集数据的时候会提到解决这个问题的方法。

接着我们输入另外一个命令行：

ffmpeg -list_options true -f dshow -i video="USB 2861 Device"

这个命令行的作用是获取指定视频采集设备支持的分辨率、帧率和像素格式等属性，返回的是一个列表，结果如下：

这里我们看到采集设备支持的最大分辨率是720x576，输出像素格式是yuyv422，支持的帧率为29.97和25FPS。

下面我们执行另外一条命令，将摄像头的图像和麦克风的音频录制保存成一个文件。命令如下：

ffmpeg -f dshow -i video="USB 2861 Device" -f dshow -i audio="线路 (3- USB Audio Device)" -vcodec libx264 -acodec aac -strict -2 mycamera.mkv

上面的命令行用video=指定视频设备，用audio=指定音频设备，后面的参数是定义编码器的格式和属性，输出为一个名为mycamera.mkv的文件。

命令运行之后，控制台打印FFmpeg的运行日志，按“Q”键则中止命令。

这里有些读者可能会问：采集设备不是支持多个分辨率吗？怎么设置采集时用哪一种分辨率输出？答案是用“-s”参数设置，若在上面的命令行加上“-s 720x576”，则FFmpeg就会以720x576的分辨率进行采集，如果不设置，则以默认的分辨率输出。

注意：如果你运行上面命令ffmpeg报如下错误：Could not run filter

Video=XXX:Input/output error

则说明该版本的ffmpeg不支持该采集设备。这是由于旧版本的FFmpeg一个Bug引起的，不支持需要连接crossbar连接的视频采集设备（详情可参考这个帖子：https://ffmpeg.zeranoe.com/forum/viewtopic.php?t=722）。新版本的FFmpeg（avdevice-58）修复了这个问题。所以如果你遇到这个问题，可以通过升级FFmpeg来解决。

好，关于命令行的内容就介绍完了。

2.采集程序的使用

这个程序叫“AVCapture”，能从视频采集设备（摄像头，采集卡）获取图像，支持图像预览；还可以采集麦克风音频；支持对视频和音频编码，支持录制成文件。这是一个MFC开发的窗口程序，界面比较简洁，如下图：

开始采集前需要选择设备，点击文件菜单的“打开设备”，弹出一个设备选择对话框，如下图所示：

在对话框里选择任意一个视频设备和音频设备，如果想启用某种设备，必须勾选右边的“启用”选项，但如果只需要用其中一种采集设备，则可以把其中一个禁用掉。

按“确定”则开始采集数据了。视频和音频会编码后保存到一个文件中，这个文件的路径是在配置文件中设置的，打开程序目录下的Config.ini文件，则显示如下字段：

[Client]

file_path = D:\camera.mkv

File_path就是录制文件的路径。

采集的图像默认显示到中间的窗口中，如果不想预览，可以在主菜单栏的“编辑”菜单中取消勾选“预览视频”。

3.功能模块实现

该采集程序实现了枚举采集设备，采集控制、显示图像、视频/音频编码和录制的功能，其中输入（Input）、输出（Output）和显示（Paint）这三个模块分别用一个单独的类进行封装：CAVInputStream，CAVOutputStream，CImagePainter。CAVInputStream负责从采集设备获取数据，提供接口获取采集设备的属性，以及提供回调函数把数据传给上层。CAVOutputStream负责对采集的视频和音频流进行编码、封装，保存成一个文件。而CImagePainter则用来显示图像，使用了GDI绘图，把图像显示到主界面的窗口。

3.1 枚举采集设备

采集前我们需要先选择设备，把所有的设备名称列出来，其中一个方法可以用第一节介绍的运行ffmpeg命令行工具来列举，但是这样有两个问题：第一，假如设备名称带中文，则显示的名称有乱码，因此，我们不知道它真实的名称。第二，ffmpeg没有API返回系统中安装的采集设备列表，虽然FFmpeg提供了API把设备名称列举出来，但是是打印到控制台的，不是通过参数来返回，如下面这段代码只能打印输出结果到控制台。但是对于窗口界面程序，没有控制台，怎么获取命令行结果呢？

1. AVFormatContext *pFmtCtx = avformat_alloc_context();
2. AVDictionary* options = NULL;
3. av_dict_set(&options, "list_devices", "true", 0);
4. AVInputFormat *iformat = av_find_input_format("dshow");
5. //printf("Device Info=============\n");
6. avformat_open_input(&pFmtCtx, "video=dummy", iformat, &options);
7. //printf("========================\n");

我用了一种最传统的做法来解决，就是通过Directshow的COM接口来枚举设备，工程里面的EnumDevice接口就实现了枚举设备的功能，函数原型如下：

1. //枚举指定类型的所有采集设备的名称
2. ENUMDEVICE_API HRESULT EnumDevice(CAPTURE_DEVICE_TYPE type, char * deviceList[], int nListLen, int & iNumCapDevices);

当然，如果读者用的采集设备是固定一种，那么可以固定采集设备的名称，这样做可以省点事。

3.2 注册FFmpeg库

1. av_register_all();
2. avdevice_register_all();

这两个API可以在程序的构造函数和窗口初始化里面调用。

3.3 打开输入设备

首先需要指定采集设备的名称。如果是视频设备类型，则名称以“video=”开头；如果是音频设备类型，则名称以“audio=”开头。调用avformat_open_input接口打开设备，将设备名称作为参数传进去，注意这个设备名称需要转成UTF-8编码。然后调用avformat_find_stream_info获取流的信息，得到视频流或音频流的索引号，之后会频繁用到这个索引号来定位视频和音频的Stream信息。接着，调用avcodec_open2打开视频解码器或音频解码器，实际上，我们可以把设备也看成是一般的文件源，而文件一般采用某种封装格式，要播放出来需要进行解复用，分离成裸流，然后对单独的视频流、音频流进行解码。虽然采集出来的图像或音频都是未编码的，但是按照FFmpeg的常规处理流程，我们需要加上“解码”这个步骤。

1. int i;
2. m_pInputFormat = av_find_input_format("dshow");
3. ASSERT(m_pInputFormat != NULL);
4. if(!m_video_device.empty())
5. {
6. int res = 0;
7. string device_name = "video=" + m_video_device;
8. string device_name_utf8 = AnsiToUTF8(device_name.c_str(), device_name.length());  //转成UTF-8，解决设备名称包含中文字符出现乱码的问题
9. //Set own video device's name
10. if ((res = avformat_open_input(&m_pVidFmtCtx, device_name_utf8.c_str(), m_pInputFormat, &device_param)) != 0)
11. {
12. ATLTRACE("Couldn't open input video stream.（无法打开输入流）\n");
13. return false;
14. }
15. //input video initialize
16. if (avformat_find_stream_info(m_pVidFmtCtx, NULL) < 0)
17. {
18. ATLTRACE("Couldn't find video stream information.（无法获取流信息）\n");
19. return false;
20. }
21. m_videoindex = -1;
22. for (i = 0; i < m_pVidFmtCtx->nb_streams; i++)
23. {
24. if (m_pVidFmtCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO)
25. {
26. m_videoindex = i;
27. break;
28. }
29. }
30. if (m_videoindex == -1)
31. {
32. ATLTRACE("Couldn't find a video stream.（没有找到视频流）\n");
33. return false;
34. }
35. if (avcodec_open2(m_pVidFmtCtx->streams[m_videoindex]->codec, avcodec_find_decoder(m_pVidFmtCtx->streams[m_videoindex]->codec->codec_id), NULL) < 0)
36. {
37. ATLTRACE("Could not open video codec.（无法打开解码器）\n");
38. return false;
39. }
40. }
41. //////////////////////////////////////////////////////////
42. if(!m_audio_device.empty())
43. {
44. string device_name = "audio=" + m_audio_device;
45. string device_name_utf8 = AnsiToUTF8(device_name.c_str(), device_name.length());  //转成UTF-8，解决设备名称包含中文字符出现乱码的问题
46. //Set own audio device's name
47. if (avformat_open_input(&m_pAudFmtCtx, device_name_utf8.c_str(), m_pInputFormat, &device_param) != 0){
48. ATLTRACE("Couldn't open input audio stream.（无法打开输入流）\n");
49. return false;
50. }
51. //input audio initialize
52. if (avformat_find_stream_info(m_pAudFmtCtx, NULL) < 0)
53. {
54. ATLTRACE("Couldn't find audio stream information.（无法获取流信息）\n");
55. return false;
56. }
57. m_audioindex = -1;
58. for (i = 0; i < m_pAudFmtCtx->nb_streams; i++)
59. {
60. if (m_pAudFmtCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_AUDIO)
61. {
62. m_audioindex = i;
63. break;
64. }
65. }
66. if (m_audioindex == -1)
67. {
68. ATLTRACE("Couldn't find a audio stream.（没有找到音频流）\n");
69. return false;
70. }
71. if (avcodec_open2(m_pAudFmtCtx->streams[m_audioindex]->codec, avcodec_find_decoder(m_pAudFmtCtx->streams[m_audioindex]->codec->codec_id), NULL) < 0)
72. {
73. ATLTRACE("Could not open audio codec.（无法打开解码器）\n");
74. return false;
75. }
76. }

3.4 初始化输出流

前面我们已经初始化了InputStream，现在需要对OutputStream进行初始化，而要初始化输出流需要知道视频采集的分辨率，帧率，输出像素格式等信息，还有音频采集设备的采样率，声道数，Sample格式，而这些信息可通过CAVInputStream类的接口来获取到。下面是初始化OutputStream的代码：

1. m_InputStream.SetVideoCaptureCB(VideoCaptureCallback);
2. m_InputStream.SetAudioCaptureCB(AudioCaptureCallback);
3. bool bRet;
4. bRet = m_InputStream.OpenInputStream(); //初始化采集设备
5. if(!bRet)
6. {
7. MessageBox(_T("打开采集设备失败"), _T("提示"), MB_OK|MB_ICONWARNING);
8. return 1;
9. }
10. int cx, cy, fps;
11. AVPixelFormat pixel_fmt;
12. if(m_InputStream.GetVideoInputInfo(cx, cy, fps, pixel_fmt)) //获取视频采集源的信息
13. {
14. m_OutputStream.SetVideoCodecProp(AV_CODEC_ID_H264, fps, 500000, 100, cx, cy); //设置视频编码器属性
15. }
16. int sample_rate = 0, channels = 0;
17. AVSampleFormat  sample_fmt;
18. if(m_InputStream.GetAudioInputInfo(sample_fmt, sample_rate, channels)) //获取音频采集源的信息
19. {
20. m_OutputStream.SetAudioCodecProp(AV_CODEC_ID_AAC, sample_rate, channels, 32000); //设置音频编码器属性
21. }
22. //从Config.INI文件中读取录制文件路径
23. P_GetProfileString(_T("Client"), "file_path", m_szFilePath, sizeof(m_szFilePath));
24. bRet  = m_OutputStream.OpenOutputStream(m_szFilePath); //设置输出路径
25. if(!bRet)
26. {
27. MessageBox(_T("初始化输出失败"), _T("提示"), MB_OK|MB_ICONWARNING);
28. return 1;
29. }

在上面的代码片段中，首先设置了视频和音频的数据回调函数。当采集开始时，视频和音频数据就会传递给相应的函数去处理，在该程序中，回调函数主要对图像或音频进行编码，然后封装成FFmpeg支持的容器（例如mkv/avi/mpg/ts/mp4）。另外，需要初始化OutputStream的VideoCodec和AudioCodec的属性，在我的程序中，视频编码器是H264，音频编码器用AAC，通过CAVInputStream对象获得输入流的信息之后再赋值给输出流相应的参数。最后调用m_OutputStream对象的OpenOutputStream成员函数打开编码器和录制的容器，其中我们需要传入一个输出文件路径作为参数，这个为录制的文件路径，路径是在Config.ini文件里配置的。如果OpenOutputStream函数返回true，则表示初始化输出流成功。

3.5 读取采集数据

接着，我们就可以开始采集了。开始采集的函数实现如下：

1. bool  CAVInputStream::StartCapture()
2. {
3. if (m_videoindex == -1 && m_audioindex == -1)
4. {
5. ATLTRACE("错误：你没有打开设备\n");
6. return false;
7. }
8. m_start_time = av_gettime();
9. m_exit_thread = false;
10. if(!m_video_device.empty())
11. {
12. m_hCapVideoThread = CreateThread(
13. NULL,                   // default security attributes
14. 0,                      // use default stack size
15. CaptureVideoThreadFunc,       // thread function name
16. this,          // argument to thread function
17. 0,                      // use default creation flags
18. NULL);   // returns the thread identifier
19. }
20. if(!m_audio_device.empty())
21. {
22. m_hCapAudioThread = CreateThread(
23. NULL,                   // default security attributes
24. 0,                      // use default stack size
25. CaptureAudioThreadFunc,       // thread function name
26. this,          // argument to thread function
27. 0,                      // use default creation flags
28. NULL);   // returns the thread identifier
29. }
30. return true;
31. }

StartCapture函数分别建立了一个读取视频包和读取音频包的线程，两个线程各自独立工作，分别从视频采集设备，音频采集设备获取到数据，然后进行后续的处理。（注意：两个线程同时向一个文件写数据可能会有同步的问题，FFmpeg内部可能没有做多线程安全访问的处理，所以最好在自己线程里加一个锁进行互斥，从而保护临界区的安全）

其中，读取摄像头数据的线程的处理代码如下：

1. DWORD WINAPI CAVInputStream::CaptureVideoThreadFunc(LPVOID lParam)
2. {
3. CAVInputStream * pThis = (CAVInputStream*)lParam;
4. pThis->ReadVideoPackets();
5. return 0;
6. }
7. int  CAVInputStream::ReadVideoPackets()
8. {
9. if(dec_pkt == NULL)
10. {
11. ////prepare before decode and encode
12. dec_pkt = (AVPacket *)av_malloc(sizeof(AVPacket));
13. }
14. int encode_video = 1;
15. int ret;
16. //start decode and encode
17. while (encode_video)
18. {
19. if (m_exit_thread)
20. break;
21. AVFrame * pframe = NULL;
22. if ((ret = av_read_frame(m_pVidFmtCtx, dec_pkt)) >= 0)
23. {
24. pframe = av_frame_alloc();
25. if (!pframe)
26. {
27. ret = AVERROR(ENOMEM);
28. return ret;
29. }
30. int dec_got_frame = 0;
31. ret = avcodec_decode_video2(m_pVidFmtCtx->streams[dec_pkt->stream_index]->codec, pframe, &dec_got_frame, dec_pkt);
32. if (ret < 0)
33. {
34. av_frame_free(&pframe);
35. av_log(NULL, AV_LOG_ERROR, "Decoding failed\n");
36. break;
37. }
38. if (dec_got_frame)
39. {
40. if(m_pVideoCBFunc)
41. {
42. CAutoLock lock(&m_WriteLock);
43. m_pVideoCBFunc(m_pVidFmtCtx->streams[dec_pkt->stream_index], m_pVidFmtCtx->streams[m_videoindex]->codec->pix_fmt, pframe, av_gettime() - m_start_time);
44. }
45. av_frame_free(&pframe);
46. }
47. else
48. {
49. av_frame_free(&pframe);
50. }
51. av_free_packet(dec_pkt);
52. }
53. else
54. {
55. if (ret == AVERROR_EOF)
56. encode_video = 0;
57. else
58. {
59. ATLTRACE("Could not read video frame\n");
60. break;
61. }
62. }
63. }
64. return 0;
65. }

在CAVInputStream::ReadVideoPackets()函数中不停地调用 av_read_frame读取采集到的图像帧，接着调用avcodec_decode_video2进行“解码”，这样获得了原始的图像，图像可能是RGB或YUV格式。解码后的图像通过m_pVideoCBFunc指向的回调函数回调给上层处理，回调函数里可进行后续的一些操作，比如对视频帧编码或直接显示。

3.6 编码、封装成文件

CAVInputStream的工作线程里面读取到的视频帧和音频包通过回调函数传给CAVOuputStream类去处理。下面是两个回调函数的实现：

1. //采集到的视频图像回调
2. LRESULT CALLBACK VideoCaptureCallback(AVStream * input_st, enum PixelFormat pix_fmt, AVFrame *pframe, INT64 lTimeStamp)
3. {
4. if(gpMainFrame->IsPreview())
5. {
6. gpMainFrame->m_Painter.Play(input_st, pframe);
7. }
8. gpMainFrame->m_OutputStream.write_video_frame(input_st, pix_fmt, pframe, lTimeStamp);
9. return 0;
10. }
11. //采集到的音频数据回调
12. LRESULT CALLBACK AudioCaptureCallback(AVStream * input_st, AVFrame *pframe, INT64 lTimeStamp)
13. {
14. gpMainFrame->m_OutputStream.write_audio_frame(input_st, pframe, lTimeStamp);
15. return 0;
16. }

视频回调函数调用了CAVOutputStream的成员函数write_video_frame，这个函数对传入的图像帧进行编码（H264），并且写到指定的封装文件；而音频回调函数则调用了CAVOutputStream的另外一个成员函数write_audio_frame，这个函数负责对音频编码（AAC），然后输出到指定的封装文件。下面是Write_video_frame函数的实现代码：

1. //input_st -- 输入流的信息
2. //input_frame -- 输入视频帧的信息
3. //lTimeStamp -- 时间戳，时间单位为/1000000
4. //
5. int CAVOutputStream::write_video_frame(AVStream * input_st, enum PixelFormat pix_fmt, AVFrame *pframe, INT64 lTimeStamp)
6. {
7. if(video_st == NULL)
8. return -1;
9. //ATLTRACE("Video timestamp: %ld \n", lTimeStamp);
10. if(m_first_vid_time1 == -1)
11. {
12. TRACE("First Video timestamp: %ld \n", lTimeStamp);
13. m_first_vid_time1 = lTimeStamp;
14. }
15. AVRational time_base_q = { 1, AV_TIME_BASE };
16. if(img_convert_ctx == NULL)
17. {
18. //camera data may has a pix fmt of RGB or sth else,convert it to YUV420
19. img_convert_ctx = sws_getContext(m_width, m_height,
20. pix_fmt, pCodecCtx->width, pCodecCtx->height, PIX_FMT_YUV420P, SWS_BICUBIC, NULL, NULL, NULL);
21. }
22. sws_scale(img_convert_ctx, (const uint8_t* const*)pframe->data, pframe->linesize, 0, pCodecCtx->height, pFrameYUV->data, pFrameYUV->linesize);
23. pFrameYUV->width = pframe->width;
24. pFrameYUV->height = pframe->height;
25. pFrameYUV->format = PIX_FMT_YUV420P;
26. enc_pkt.data = NULL;
27. enc_pkt.size = 0;
28. av_init_packet(&enc_pkt);
29. int ret;
30. int enc_got_frame = 0;
31. ret = avcodec_encode_video2(pCodecCtx, &enc_pkt, pFrameYUV, &enc_got_frame);
32. if (enc_got_frame == 1)
33. {
34. //printf("Succeed to encode frame: %5d\tsize:%5d\n", framecnt, enc_pkt.size);
35. if(m_first_vid_time2 == -1)
36. {
37. m_first_vid_time2 = lTimeStamp;
38. }
39. enc_pkt.stream_index = video_st->index;
40. //enc_pkt.pts= av_rescale_q(lTimeStamp, time_base_q, video_st->time_base);
41. enc_pkt.pts = (INT64)video_st->time_base.den * lTimeStamp/AV_TIME_BASE;
42. m_vid_framecnt++;
43. ret = av_interleaved_write_frame(ofmt_ctx, &enc_pkt);
44. if(ret < 0)
45. {
46. char tmpErrString[128] = {0};
47. ATLTRACE("Could not write video frame, error: %s\n", av_make_error_string(tmpErrString, AV_ERROR_MAX_STRING_SIZE, ret));
48. av_packet_unref(&enc_pkt);
49. return ret;
50. }
51. av_free_packet(&enc_pkt);
52. }
53. else if(ret == 0)
54. {
55. ATLTRACE("Buffer video frame, timestamp: %I64d.\n", lTimeStamp); //编码器缓冲帧
56. }
57. return 0;
58. }

Write_video_frame和write_audio_frame是CAVOutputStream的两个很重要的函数，其中对音频包的处理略为复杂一些，主要是因为输入的音频和编码后的音频的frame_size不一样，中间需要一个Fifo作缓冲队列。另外时间戳PTS的计算也是很关键的，弄得不好保存的文件播放视音频就不同步了，具体怎么实现你们看代码吧。工程的代码下载地址：点击打开链接

from:https://blog.csdn.net/zhoubotong2012/article/details/79338093