Windows软件插件-音视频文件读取器

下载本插件
本插件读取音频和视频文件，输出音频样本和视频样本，音频样本为16位PCM，采样率48000；视频样本为RGB32。大部门音频和视频文件格式都可以读取。本插件类型为DLL。
本插件是通过创建媒体底子“源读取器”对象实现读取音视频文件。使用MFCreateSourceReaderFromURL函数创建源读取器，源读取器接口为IMFSourceReader，如果读取的是音频文件，只创建音频线程，如果读取视频文件，将创建视频线程和音频线程，在线程中使用IMFSourceReader接口的ReadSample方法读取样本，样本为媒体底子样本；使用IMFSourceReader接口的SetCurrentPosition方法更改当前位置。定义了7个导出函数，用于对本读取器的操纵。
使用方法

首先，加载本读取器DLL，获得读取器模块句柄。

1.         HMODULE hSReader = LoadLibrary(L"SReader.dll");

复制代码

定义媒体信息和样本信息结构，定义接收当前位置信息的函数（如果不需要，也可以不定义），定义视频音频样本接收函数：

1. struct INFO
2. {
3.         int VideoWidth=0;//视频宽，单位像素
4.         int VideoHeight=0;//视频高，单位像素
5.         LONGLONG FrameCur=0;//帧持续时间，单位100纳秒
6.         LONGLONG DUR=0;//媒体时长，单位100纳秒
7.         DWORD AudioSamplePerSec=0;//音频样本采样率
8.         DWORD StreamCount=0;//流数量
9. };
11. struct SAMPLE_INFO//样本信息
12. {
13.         BOOL B;//为TRUE，样本为第1个样本
14.         DWORD STAR;//运行开始时间，单位毫秒
15.         LONGLONG star;//样本开始时间，单位100纳秒
16.         LONGLONG end;//样本结束时间，单位100纳秒
17.         BYTE* pB;//样本缓冲区指针
18.         int len;//样本的字节大小
19.         HANDLE hRun;//“运行”事件句柄
20.         HANDLE hSeek;//“定位”事件句柄
21. };
23. void SliderSetPosition(LONGLONG pos)
24. {
25. }
27. int ReceiveVideoSample(SAMPLE_INFO SampleInfo)//视频样本接收函数
28. {
29.     return 0;
30. }
32. int ReceiveAudioSample(SAMPLE_INFO SampleInfo)//音频样本接收函数
33. {
34.     return 0;
35. }

复制代码

定义函数指针：

1.         typedef int(__thiscall *MYPROC_ReceiveSample)(SAMPLE_INFO info);
2.         typedef void(__thiscall *MYPROC_GetPos)(LONGLONG pos);
3.         typedef INFO(__cdecl *MYPROC_SReader_Init)(wchar_t* Path);
4.         typedef int(__cdecl *MYPROC_SReader_Run)(MYPROC_ReceiveSample VideoSample, MYPROC_ReceiveSample AudioSample, MYPROC_GetPos Pos);
5.         typedef int(__cdecl *MYPROC_SReader_Pause)();
6.         typedef int(__cdecl *MYPROC_SReader_Stop)();
7.         typedef int(__cdecl *MYPROC_SReader_Seek)(LONGLONG SeekPos);
8.         typedef int(__cdecl *MYPROC_SReader_GetState)();
9.         typedef int(__cdecl *MYPROC_SReader_Exit)();

复制代码

获取本读取器导出函数的地址：

1.         MYPROC_SReader_Init SReader_Init = NULL;
2.         MYPROC_SReader_GetState SReader_GetState = NULL;
3.         MYPROC_SReader_Run SReader_Run = NULL;
4.         MYPROC_SReader_Pause SReader_Pause = NULL;
5.         MYPROC_SReader_Stop SReader_Stop = NULL;
6.         MYPROC_SReader_Seek SReader_Seek = NULL;
7.         MYPROC_SReader_Exit SReader_Exit = NULL;
8.         MYPROC_GetPos GetPos=NULL;
9.         MYPROC_ReceiveSample ReceiveVideo=NULL;
10.         MYPROC_ReceiveSample ReceiveAudio=NULL;
11.         if (hSReader)
12.         {
13.                 SReader_Init = (MYPROC_SReader_Init)GetProcAddress(hSReader, "Init");
14.                 SReader_GetState = (MYPROC_SReader_GetState)GetProcAddress(hSReader, "GetState");
15.                 SReader_Run = (MYPROC_SReader_Run)GetProcAddress(hSReader, "Run");
16.                 SReader_Pause = (MYPROC_SReader_Pause)GetProcAddress(hSReader, "Pause");
17.                 SReader_Stop = (MYPROC_SReader_Stop)GetProcAddress(hSReader, "Stop");
18.                 SReader_Seek = (MYPROC_SReader_Seek)GetProcAddress(hSReader, "Seek");
19.                 SReader_Exit = (MYPROC_SReader_Exit)GetProcAddress(hSReader, "Exit");
20.                 GetPos = (MYPROC_GetPos)(&SliderSetPosition);       
21.                 ReceiveVideo=(MYPROC_ReceiveSample)(&ReceiveVideoSample);
22.                 ReceiveAudio=(MYPROC_ReceiveSample)(&ReceiveAudioSample);
23.         }

复制代码

初始化本读取器，获取文件媒体信息：

1.         INFO info;
2.         wchar_t* path = L"某视频文件.mp4";
3.         if (SReader_Init != NULL)
4.                 info = SReader_Init(path);

复制代码

为读取器指定视频样本和音频样本接收函数，和接收位置信息函数（不需要位置信息时，GetPos可以为NULL）；运行读取器:

1.         if (SReader_Run)
2.         {
3.                 SReader_Run(ReceiveVideo, ReceiveAudio, GetPos);
4.         }

复制代码

如许视频音频样本接收函数将被反复调用，参数提供样本的数据和信息。别的导出函数用于读取器的停息，制止和退出；获取读取器当前状态，调用SReader_GetState()，返回状态值。
本读取器为后续文章“播放视频”而计划，读者可以在本代码的底子上添加自己想要的功能，也可以进行更改。
本读取器的全部代码

1. #include "windows.h"
2. #include "mmsystem.h"
3. #pragma comment(lib, "winmm")
4. #include "mfapi.h"
5. #include "mfidl.h"
6. #include "mfreadwrite.h"
7. #pragma comment(lib, "mfplat")
8. #pragma comment(lib, "mfreadwrite")
9. #pragma comment(lib, "mfuuid")
11. template <class T> void SafeRelease(T** ppT)
12. {
13.         if (*ppT)
14.         {
15.                 (*ppT)->Release();
16.                 *ppT = NULL;
17.         }
18. }
20. struct INFO//媒体信息
21. {
22.         int VideoWidth;//视频宽，单位像素
23.         int VideoHeight;//视频高，单位像素
24.         LONGLONG FrameCur;//帧持续时间，单位100纳秒
25.         LONGLONG DUR;//媒体时长，单位100纳秒
26.         DWORD AudioSamplePerSec;//音频样本采样率
27.         DWORD StreamCount = 0;//流数量
28. };
30. struct SAMPLE_INFO//样本信息
31. {
32.         BOOL B;//为TRUE，样本为第1个样本
33.         DWORD STAR;//运行开始时间，单位毫秒
34.         LONGLONG star;//样本开始时间，单位100纳秒
35.         LONGLONG end;//样本结束时间，单位100纳秒
36.         BYTE* pB;//样本缓冲区指针
37.         int len;//样本的字节大小
38.         HANDLE hRun;//“运行”事件句柄
39.         HANDLE hSeek;//“定位”事件句柄
40. };
42. typedef int(__cdecl *MYPROC_SendVideo)(SAMPLE_INFO SampleInfo);//定义函数指针
43. typedef int(__cdecl *MYPROC_SendAudio)(SAMPLE_INFO SampleInfo);
44. typedef int(__cdecl *MYPROC_SendPos)(LONGLONG pos);
46. class SReader
47. {
48. public:
49.         SReader()
50.         {
51.                 HRESULT hr = MFStartup(MF_VERSION);
52.                 if (hr != S_OK)
53.                 {
54.                         MessageBox(NULL, L"初始化媒体基础失败", L"SourceReader", MB_OK);
55.                 }
56.                 hRun = CreateEvent(NULL, TRUE, FALSE, NULL);//手动重置，初始无信号
57.                 hExit = CreateEvent(NULL, TRUE, FALSE, NULL);//手动重置，初始无信号
58.                 hSeek = CreateEvent(NULL, TRUE, FALSE, NULL);//手动重置，初始无信号
59.         }
60.         ~SReader()
61.         {
62.                 SafeRelease(&pIMFSourceReader); CloseHandle(hRun); CloseHandle(hExit); CloseHandle(hSeek);
63.                 MFShutdown();//关闭媒体基础
64.         }
65.         HANDLE hRun, hExit, hSeek;
66.         IMFSourceReader* pIMFSourceReader = NULL;//源读取器接口指针
67.         UINT32 Width, Height;//视频宽高
68.         LONGLONG DUR;//媒体持续时间，100纳秒单位
69.         LONGLONG CUR;//当前位置，100纳秒单位
70.         LONGLONG SeekPos;//定位位置
71.         DWORD STAR;//开始时间
72.         BOOL VFirst, AFirst;//为TRUE标明第1个样本
73.         int mState;//状态；0停止，1运行，2暂停
74.         INFO info;//媒体信息对象
75.         MYPROC_SendVideo SendVideo = NULL;//视频样本输出函数指针
76.         MYPROC_SendAudio SendAudio = NULL;//音频样本输出函数指针
77.         MYPROC_SendPos SendPos = NULL;//当前位置输出函数指针
78. };
80. HRESULT GetSample(IMFSourceReader* pIMFSourceReader, UINT SOURCE, LONGLONG& Cur, LONGLONG& Dur)//获取样本
81. {
82.         IMFSample* pMFSample = NULL; DWORD flags;
83.         HRESULT hr = pIMFSourceReader->ReadSample(SOURCE, 0, NULL, &flags, NULL, &pMFSample);//读取视频样本
84.         hr = pMFSample->GetSampleTime(&Cur);//获取显示时间
85.         hr = pMFSample->GetSampleDuration(&Dur);//获取持续时间
86.         SafeRelease(&pMFSample);//释放接口
87.         return hr;
88. }
90. SReader* pSReader = NULL;//SReader类对象指针
91. HANDLE hVthread = NULL, hAthread = NULL;//视频音频线程句柄
93. DWORD WINAPI VideoThread(LPVOID lp)//视频线程
94. {
95.         HRESULT hr; pSReader->VFirst = TRUE; IMFSample* pMFSample = NULL; DWORD flags;
96. Agan:
97.         DWORD mRun = WaitForSingleObject(pSReader->hRun, 0);//检测“运行”信号，不等待
98.         DWORD mSeek = WaitForSingleObject(pSReader->hSeek, 0);//检测“定位”信号，不等待
99.         DWORD mExit = WaitForSingleObject(pSReader->hExit, 0);//检测“退出”信号，不等待
100.         if (mExit == WAIT_OBJECT_0)//有“退出”信号
101.         {
102.                 return 0;//退出线程
103.         }
104.         if (mSeek == WAIT_OBJECT_0)//有“定位”信号
105.         {
106.                 hr = pSReader->pIMFSourceReader->Flush(MF_SOURCE_READER_ALL_STREAMS);//丢弃所有排队的样本并取消所有挂起的样本请求
107.                 PROPVARIANT pv;
108.                 PropVariantInit(&pv);
109.                 pv.vt = 20;
110.                 pv.hVal.QuadPart = pSReader->SeekPos;
111.                 hr = pSReader->pIMFSourceReader->SetCurrentPosition(GUID_NULL, pv);//更改源读取器位置
112.                 PropVariantClear(&pv);
113.                 /*试验发现，源读取器在媒体中间时间段定位后，音频视频样本不同步（时间戳相差约2秒钟），故添加下面代码以使样本时间同步*/
114.                 LONGLONG VCur, VDur;
115.                 hr = GetSample(pSReader->pIMFSourceReader, MF_SOURCE_READER_FIRST_VIDEO_STREAM, VCur, VDur);//获取视频样本时间
116.                 LONGLONG ACur, ADur;
117.                 hr = GetSample(pSReader->pIMFSourceReader, MF_SOURCE_READER_FIRST_AUDIO_STREAM, ACur, ADur);//获取音频样本时间
118.                 if (ACur > VCur + VDur + 100000)//如果音频样本时间过大
119.                 {
120.                         while (ACur > VCur + VDur + 100000)
121.                         {
122.                                 hr = GetSample(pSReader->pIMFSourceReader, MF_SOURCE_READER_FIRST_VIDEO_STREAM, VCur, VDur);//获取下一个视频样本
123.                         }
124.                 }
125.                 else if (VCur > ACur + ADur + 100000)//如果视频样本时间过大
126.                 {
127.                         while (VCur > ACur + ADur + 100000)
128.                         {
129.                                 hr = GetSample(pSReader->pIMFSourceReader, MF_SOURCE_READER_FIRST_AUDIO_STREAM, ACur, ADur);//获取下一个音频样本
130.                         }
131.                 }
132.                 pSReader->STAR = timeGetTime();//记录运行开始时间，单位毫秒
133.                 pSReader->VFirst = TRUE; pSReader->AFirst = TRUE;//设置第1个样本标记为TRUE
134.                 ResetEvent(pSReader->hSeek);//设置“定位”无信号
135.         }
136.         if (mRun == WAIT_OBJECT_0)//有“运行”信号
137.         {
138.                 hr = pSReader->pIMFSourceReader->ReadSample(MF_SOURCE_READER_FIRST_VIDEO_STREAM, 0, NULL, &flags, NULL, &pMFSample);//读取视频样本
139.                 if (hr == S_OK && pMFSample)
140.                 {
141.                         LONGLONG Cur, Dur;
142.                         hr = pMFSample->GetSampleTime(&Cur);//获取显示时间
143.                         hr = pMFSample->GetSampleDuration(&Dur);//获取持续时间
144.                         DWORD Lt;
145.                         hr = pMFSample->GetTotalLength(&Lt);//获取有效长度
146.                         DWORD count;
147.                         hr = pMFSample->GetBufferCount(&count);//获取缓冲区数量
148.                         IMFMediaBuffer* pMFBuffer = NULL;
149.                         if (count == 1)//如果只有1个缓冲区
150.                         {
151.                                 hr = pMFSample->GetBufferByIndex(0, &pMFBuffer);
152.                         }
153.                         else//如果有多个缓冲区
154.                         {
155.                                 hr = pMFSample->ConvertToContiguousBuffer(&pMFBuffer);
156.                         }
157.                         BYTE* pSB = NULL;
158.                         hr = pMFBuffer->Lock(&pSB, NULL, NULL);//锁定缓冲区
159.                         SAMPLE_INFO info;
160.                         info.B = pSReader->VFirst;
161.                         info.STAR = pSReader->STAR;
162.                         info.star = Cur;
163.                         info.end = Cur + Dur;
164.                         info.pB = pSB;
165.                         info.len = Lt;
166.                         info.hRun = pSReader->hRun;
167.                         info.hSeek = pSReader->hSeek;
168.                         pSReader->SendVideo(info);//调用视频输出函数，输出样本
169.                         if (pSReader->VFirst)pSReader->VFirst = FALSE;//只有第1个样本标记为TRUE
170.                         hr = pMFBuffer->Unlock();//解锁缓冲区
171.                         SafeRelease(&pMFBuffer); SafeRelease(&pMFSample);//释放接口
172.                 }
173.         }
174.         goto Agan;
175. }
177. DWORD WINAPI AudioThread(LPVOID lp)
178. {
179.         HRESULT hr; pSReader->AFirst = TRUE; IMFSample* pMFSample = NULL; DWORD flags; pSReader->CUR = 0; int Count = 0;
180. Agan:
181.         DWORD mRun = WaitForSingleObject(pSReader->hRun, 0);//检测“运行”信号，不等待
182.         DWORD mSeek = WaitForSingleObject(pSReader->hSeek, 0);//检测“定位”信号，不等待
183.         DWORD mExit = WaitForSingleObject(pSReader->hExit, 0);//检测“退出”信号，不等待
184.         if (mExit == WAIT_OBJECT_0)//有“退出”信号
185.         {
186.                 return 0;//退出线程
187.         }
188.         if (mSeek == WAIT_OBJECT_0)//有“定位”信号
189.         {
190.                 if (pSReader->info.StreamCount == 2)//如果有视频流和音频流
191.                 {
192.                         goto Agan;//阻塞（更改位置操作在视频线程中，音频线程等待操作完成）
193.                 }
194.                 else//如果只有音频流，在此更改源读取器位置
195.                 {
196.                         hr = pSReader->pIMFSourceReader->Flush(MF_SOURCE_READER_ALL_STREAMS);//丢弃所有排队的样本并取消所有挂起的样本请求
197.                         PROPVARIANT pv;
198.                         PropVariantInit(&pv);
199.                         pv.vt = 20;
200.                         pv.hVal.QuadPart = pSReader->SeekPos;
201.                         hr = pSReader->pIMFSourceReader->SetCurrentPosition(GUID_NULL, pv);//更改源读取器位置
202.                         PropVariantClear(&pv);
203.                         pSReader->STAR = timeGetTime();//记录运行开始时间
204.                         pSReader->AFirst = TRUE;
205.                         ResetEvent(pSReader->hSeek);//设置“定位”无信号
206.                 }
207.         }
208.         if (mRun == WAIT_OBJECT_0)//有“运行”信号
209.         {
210.                 hr = pSReader->pIMFSourceReader->ReadSample(MF_SOURCE_READER_FIRST_AUDIO_STREAM, 0, NULL, &flags, NULL, &pMFSample);//读取音频样本
211.                 if (hr == S_OK && pMFSample)
212.                 {
213.                         LONGLONG Cur, Dur;
214.                         hr = pMFSample->GetSampleTime(&Cur);//获取显示时间
215.                         hr = pMFSample->GetSampleDuration(&Dur);//获取持续时间
216.                         Count++;
217.                         if (Count > 9)//每10个样本，发送一次当前位置
218.                         {
219.                                 pSReader->CUR = Cur;//赋值当前时间
220.                                 if(pSReader->SendPos)pSReader->SendPos(Cur);//发送当前时间位置值
221.                                 Count = 0;
222.                         }
223.                         DWORD Lt;
224.                         hr = pMFSample->GetTotalLength(&Lt);//获取有效长度
225.                         DWORD count;
226.                         hr = pMFSample->GetBufferCount(&count);//获取缓冲区数量
227.                         IMFMediaBuffer* pMFBuffer = NULL;
228.                         if (count == 1)//如果只有1个缓冲区
229.                         {
230.                                 hr = pMFSample->GetBufferByIndex(0, &pMFBuffer);
231.                         }
232.                         else//如果有多个缓冲区
233.                         {
234.                                 hr = pMFSample->ConvertToContiguousBuffer(&pMFBuffer);
235.                         }
236.                         BYTE* pSB = NULL;
237.                         hr = pMFBuffer->Lock(&pSB, NULL, NULL);//锁定缓冲区
238.                         SAMPLE_INFO info;
239.                         info.B = pSReader->AFirst;
240.                         info.STAR = pSReader->STAR;
241.                         info.star = Cur;
242.                         info.end = Cur + Dur;
243.                         info.pB = pSB;
244.                         info.len = Lt;
245.                         info.hRun = pSReader->hRun;
246.                         info.hSeek = pSReader->hSeek;
247.                         pSReader->SendAudio(info);//调用音频样本输出函数，发送音频样本
248.                         if (pSReader->AFirst)pSReader->AFirst = FALSE;//只有第1个样本标记为TRUE
249.                         hr = pMFBuffer->Unlock();//解锁缓冲区
250.                         SafeRelease(&pMFBuffer); SafeRelease(&pMFSample);//释放接口
251.                 }
252.         }
253.         goto Agan;
254. }
257. //下面是导出函数定义
258. #pragma warning(disable:4190)
260. #ifdef __cplusplus    // If used by C++ code,
261. extern "C" {          // we need to export the C interface
262. #endif
264.         __declspec(dllexport) INFO __cdecl Init(wchar_t* Path)
265.         {
266.                 if (pSReader)//如果SReader对象已存在
267.                 {
268.                         SetEvent(pSReader->hExit);//设置“退出”有信号
269.                         WaitForSingleObject(hVthread, INFINITE);//等待视频线程退出
270.                         WaitForSingleObject(hAthread, INFINITE);//等待音频线程退出
271.                         delete pSReader;//删除SReader对象
272.                 }
273.                 pSReader = new SReader();//创建SReader对象
274.                 IMFAttributes* pIMFAttributes = NULL;
275.                 HRESULT hr = MFCreateAttributes(&pIMFAttributes, 0);
276.                 if (SUCCEEDED(hr))
277.                 {
278.                         hr = pIMFAttributes->SetUINT32(MF_READWRITE_ENABLE_HARDWARE_TRANSFORMS, (UINT32)1);//使用基于硬件的媒体基础转换
279.                 }
280.                 if (SUCCEEDED(hr))
281.                 {
282.                         hr = pIMFAttributes->SetUINT32(MF_SOURCE_READER_ENABLE_VIDEO_PROCESSING, (UINT32)1);//允许源读取器进行有限的视频处理
283.                 }
284.                 if (SUCCEEDED(hr))
285.                 {
286.                         hr = MFCreateSourceReaderFromURL(Path, pIMFAttributes, &pSReader->pIMFSourceReader);//创建源读取器
287.                 }
288.                 SafeRelease(&pIMFAttributes);
289.                 PROPVARIANT var;
290.                 PropVariantInit(&var);
291.                 if (SUCCEEDED(hr))
292.                 {
293.                         hr = pSReader->pIMFSourceReader->GetPresentationAttribute(MF_SOURCE_READER_MEDIASOURCE, MF_PD_DURATION, &var);//获取媒体时间长度，100纳秒单位
294.                 }
295.                 if (SUCCEEDED(hr))
296.                 {
297.                         pSReader->info.DUR = pSReader->DUR = (LONGLONG)var.uhVal.QuadPart;
298.                 }
299.                 PropVariantClear(&var);
300.                 IMFMediaType* pAudioMTA = NULL;
301.                 if (SUCCEEDED(hr))
302.                 {
303.                         hr = MFCreateMediaType(&pAudioMTA);//创建空的媒体类型
304.                 }
305.                 if (SUCCEEDED(hr))
306.                 {
307.                         hr = pAudioMTA->SetGUID(MF_MT_MAJOR_TYPE, MFMediaType_Audio);//设置主要类型音频
308.                 }
309.                 if (SUCCEEDED(hr))
310.                 {
311.                         hr = pAudioMTA->SetGUID(MF_MT_SUBTYPE, MFAudioFormat_PCM);//设置子类型PCM
312.                 }
313.                 if (SUCCEEDED(hr))
314.                 {
315.                         hr = pAudioMTA->SetUINT32(MF_MT_AUDIO_BITS_PER_SAMPLE, (UINT32)16);//设置样本16位
316.                 }
317.                 if (SUCCEEDED(hr))
318.                 {
319.                         hr = pAudioMTA->SetUINT32(MF_MT_AUDIO_SAMPLES_PER_SECOND, (UINT32)48000);//设置样本采样率48000
320.                 }
321.                 if (SUCCEEDED(hr))
322.                 {
323.                         pSReader->info.AudioSamplePerSec = 48000;
324.                 }
325.                 if (SUCCEEDED(hr))
326.                 {
327.                         hr = pSReader->pIMFSourceReader->SetCurrentMediaType(MF_SOURCE_READER_FIRST_AUDIO_STREAM, NULL, pAudioMTA);//设置音频输出媒体类型（此时未提供媒体类型全部信息）
328.                 }
329.                 SafeRelease(&pAudioMTA);
330.                 IMFMediaType* pAudioMT = NULL;
331.                 if (SUCCEEDED(hr))
332.                 {
333.                         hr = pSReader->pIMFSourceReader->GetCurrentMediaType(MF_SOURCE_READER_FIRST_AUDIO_STREAM, &pAudioMT);//获取当前音频输出媒体类型（获取的媒体类型将包含全部信息）
334.                 }
335.                 UINT32 SamplePerSec;
336.                 if (SUCCEEDED(hr))
337.                 {
338.                         hr = pAudioMT->GetUINT32(MF_MT_AUDIO_SAMPLES_PER_SECOND, &SamplePerSec);//获取采样率
339.                 }
340.                 SafeRelease(&pAudioMT);
341.                 if (SUCCEEDED(hr))
342.                 {
343.                         pSReader->info.AudioSamplePerSec = SamplePerSec;
344.                 }
345.                 pSReader->info.StreamCount = 0;
346.                 if (SUCCEEDED(hr))
347.                 {
348.                         ResetEvent(pSReader->hExit);//设置“退出”无信号
349.                         pSReader->info.StreamCount++;//流数量
350.                         hAthread = CreateThread(NULL, 0, AudioThread, pSReader, 0, NULL);//创建音频线程
351.                 }
352.                 IMFMediaType* pVideoMTV = NULL;
353.                 if (SUCCEEDED(hr))
354.                 {
355.                         hr = MFCreateMediaType(&pVideoMTV);//创建空的媒体类型
356.                 }
357.                 if (SUCCEEDED(hr))
358.                 {
359.                         hr = pVideoMTV->SetGUID(MF_MT_MAJOR_TYPE, MFMediaType_Video);//设置主要类型为视频
360.                 }
361.                 if (SUCCEEDED(hr))
362.                 {
363.                         hr = pVideoMTV->SetGUID(MF_MT_SUBTYPE, MFVideoFormat_RGB32);//设置子类型RGB32
364.                 }
365.                 if (SUCCEEDED(hr))
366.                 {
367.                         hr = pSReader->pIMFSourceReader->SetCurrentMediaType(MF_SOURCE_READER_FIRST_VIDEO_STREAM, NULL, pVideoMTV);//设置视频输出媒体类型
368.                 }
369.                 SafeRelease(&pVideoMTV);
370.                 if (SUCCEEDED(hr))
371.                 {
372.                         IMFMediaType* pVideoMT = NULL;
373.                         if (SUCCEEDED(hr))
374.                         {
375.                                 hr = pSReader->pIMFSourceReader->GetCurrentMediaType(MF_SOURCE_READER_FIRST_VIDEO_STREAM, &pVideoMT);//获取当前视频媒体类型（此时的媒体类型将包含全部信息）
376.                         }
377.                         if (SUCCEEDED(hr))
378.                         {
379.                                 hr = MFGetAttributeSize(pVideoMT, MF_MT_FRAME_SIZE, &pSReader->Width, &pSReader->Height);//获取视频图像宽高
380.                         }
381.                         SafeRelease(&pVideoMT);
382.                         if (SUCCEEDED(hr))
383.                         {
384.                                 pSReader->info.VideoWidth = pSReader->Width; pSReader->info.VideoHeight = pSReader->Height; pSReader->info.StreamCount++;
385.                         }
386.                         hVthread = CreateThread(NULL, 0, VideoThread, pSReader, 0, NULL);//创建视频线程
387.                 }
388.                 Sleep(1000);//等待线程初始化完成
389.                 return pSReader->info;
390.         }
392.         __declspec(dllexport) int __cdecl Run(MYPROC_SendVideo SendVideo, MYPROC_SendAudio SendAudio, MYPROC_SendPos SendPos)//运行
393.         {
394.                 if (pSReader)
395.                 {
396.                         pSReader->SendVideo = SendVideo; pSReader->SendAudio = SendAudio; pSReader->SendPos = SendPos;
397.                         pSReader->STAR = timeGetTime();//记录运行开始时间
398.                         pSReader->VFirst = TRUE; pSReader->AFirst = TRUE;
399.                         SetEvent(pSReader->hRun);//设置“运行”有信号
400.                         pSReader->mState = 1;//状态标记置1
401.                         return 0;
402.                 }
403.                 return 1;
404.         }
406.         __declspec(dllexport) int __cdecl Pause(void)//暂停
407.         {
408.                 if (pSReader)
409.                 {
410.                         ResetEvent(pSReader->hRun);//设置“运行”无信号
411.                         pSReader->mState = 2;//状态标记置2
412.                         return 0;
413.                 }
414.                 return 1;
415.         }
417.         __declspec(dllexport) int __cdecl Stop(void)//停止
418.         {
419.                 if (pSReader)
420.                 {
421.                         ResetEvent(pSReader->hRun);//设置“运行”无信号
422.                         HRESULT hr = pSReader->pIMFSourceReader->Flush(MF_SOURCE_READER_ALL_STREAMS);//丢弃所有排队的样本并取消所有挂起的样本请求
423.                         Sleep(200);
424.                         pSReader->SeekPos = 0;
425.                         PROPVARIANT pv;
426.                         PropVariantInit(&pv);
427.                         pv.vt = 20;
428.                         pv.hVal.QuadPart = 0;
429.                         hr = pSReader->pIMFSourceReader->SetCurrentPosition(GUID_NULL, pv);//设置源读取器位置0
430.                         PropVariantClear(&pv);
431.                         pSReader->mState = 0;//状态标记置0
432.                         return 0;
433.                 }
434.                 return 1;
435.         }
437.         __declspec(dllexport) int __cdecl Seek(LONGLONG SeekPos)//定位
438.         {
439.                 if (pSReader)
440.                 {
441.                         pSReader->SeekPos = SeekPos;//赋值定位位置
442.                         SetEvent(pSReader->hSeek);//设置“定位”有信号
443.                         return 0;
444.                 }
445.                 return 1;
446.         }
448.         __declspec(dllexport) int __cdecl GetState()//获取状态
449.         {
450.                 if (pSReader)
451.                 {
452.                         return pSReader->mState;
453.                 }
454.                 return -1;
455.         }
457.         __declspec(dllexport) int __cdecl Exit(void)//退出
458.         {
459.                 if (pSReader)
460.                 {
461.                         ResetEvent(pSReader->hRun);//设置“运行”无信号
462.                         SetEvent(pSReader->hExit);//设置“退出”有信号
463.                         WaitForSingleObject(hVthread, INFINITE);//等待视频线程已退出
464.                         WaitForSingleObject(hAthread, INFINITE);//等待音频线程已退出
465.                         delete pSReader; pSReader = NULL;//删除SReader对象
466.                         return 0;
467.                 }
468.                 return 1;
469.         }
471. #ifdef __cplusplus
472. }
473. #endif

复制代码

下载本插件

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