利用DirectSound实现声卡录音【老毕改版】【Part1】

这几天一直在研究DSOUND的录音及播放，为了能实现网络音视频打基础，呵呵，在网上查找了大量的相关资料，有一定的收货，其中大量的工作用来修改从网上找出的源码，因为太多错误的了，不太适合初学者，为此我特别封装成类，供初学者学习参考，欢迎批评指正！

/*类接口文件 ISound.h*/

1. #pragma once
2. #pragma comment(lib, "dxguid.lib")
3. #pragma comment(lib, "dsound.lib")
5. #include <mmsystem.h>
6. #include <dsound.h>
7. #include <string>
8. #include "WaveFile.h"
10. using namespace std;
12. #define NOTIFICATIONS_PORTION 4
14. class ISound
15. {
16. public: 
17.     ISound()
18.         :m_dwBufferOffset(0),
19.         m_dwNotifySize(1000)
20.         
21.     {
22.         
23.     }
24.     virtual ~ISound();
25. public:
26.     void SetFile(const char*);
28. protected:
29.     virtual BOOL InitializeDevice() = 0;
30.     virtual BOOL UninitializeDevice() = 0;
31.     virtual BOOL CreateBuffer() = 0;
32.     virtual void WriteData() = 0;
34. protected:
35.     string g_strfile;
36.     WAVEFORMATEX g_wformat;
38.     // 偏移量
39.     DWORD m_dwBufferOffset;
40.     // buffer size
41.     DWORD m_dwBufferSize;
42.     //
43.     DWORD m_dwNotifySize;
45.     // event handle
46.     HANDLE m_hEventNotify;
47. };

/*类接口文件 ISound.h*/

1. #include "StdAfx.h"
2. #include "ISound.h"
4. ISound::~ISound()
5. {
6.     
7. }
9. void ISound::SetFile(const char* lpfile)
10. {   
11.     g_strfile = lpfile;
12. }

 

1. /*录音 CSoundRecord.h*/
3. #pragma once
4. #include "isound.h"
6. class CSoundRecord :
7.     public ISound
8. {
9. public:
10.     CSoundRecord(void);
11.     ~CSoundRecord(void);
13. private:
14.     LPDIRECTSOUNDCAPTURE8 m_lpCapture;
15.     LPDIRECTSOUNDCAPTUREBUFFER m_lpBuffer;
16.     LPDIRECTSOUNDCAPTUREBUFFER8 m_lpBuffer8;
17.     LPDIRECTSOUNDNOTIFY8 m_lpNotify;
18.     
19.     DSBPOSITIONNOTIFY m_NotifyPosition[NOTIFICATIONS_PORTION + 1];  
21.     CWinThread* pThread;
22.     CWaveFile *m_pWaveFile;
24. public:
25.     BOOL Record();
26.     BOOL Stop();
28. protected:
29.     BOOL InitializeDevice();
30.     BOOL UninitializeDevice();
32. protected:
33.     BOOL CreateBuffer();
34.     BOOL CreateNotification();
35.     void WriteData();
36.     void CorrectLocksize(LONG*);
38.     static UINT ThreadProc(LPVOID);
40. };

 

1. /*录音 CSoundRecord.cpp*/
3. #include "StdAfx.h"
4. #include "SoundRecord.h"
5. //#include <boost/lexical_cast.hpp>
6. //using namespace boost;
8. CSoundRecord::CSoundRecord(void)
9. {
10.     m_dwBufferSize = m_dwNotifySize*NOTIFICATIONS_PORTION;
12.     BOOL bResult = InitializeDevice();
14.     if(!bResult)
15.     {
16.         return;
17.     }
19.     bResult = CreateBuffer();
21.     if(!bResult)
22.     {
23.         return;
24.     }
26.     bResult = CreateNotification();
28.     if(!bResult)
29.     {
30.         return;
31.     }
33.     pThread = AfxBeginThread((AFX_THREADPROC)CSoundRecord::ThreadProc, this);
35.     m_pWaveFile = new CWaveFile;
36. }
38. CSoundRecord::~CSoundRecord(void)
39. {
40.     UninitializeDevice();
41. }
43. BOOL CSoundRecord::Record()
44. {
45.     if(g_strfile.empty())
46.     {
47.         return FALSE;
48.     }
50.     LPTSTR lpfile = (LPTSTR)g_strfile.c_str();
51.     HRESULT hr = m_pWaveFile->Open(lpfile, &g_wformat, WAVEFILE_WRITE);
53.     if(FAILED(hr))
54.     {
55.         m_pWaveFile->Close();
56.         return FALSE;
57.     }
59.     m_lpBuffer8->Start(DSCBSTART_LOOPING);
61.     return TRUE;
62. }
64. BOOL CSoundRecord::Stop()
65. {
66.     m_lpBuffer8->Stop();
68.     m_pWaveFile->Close();
70.     return TRUE;
71. }
73. BOOL CSoundRecord::InitializeDevice()
74. {
75.     HRESULT hr = DirectSoundCaptureCreate8(&DSDEVID_DefaultCapture, &m_lpCapture, NULL);
77.     if(FAILED(hr))
78.     {
79.         return FALSE;
80.     }
82.     return TRUE;
83. }
85. BOOL CSoundRecord::UninitializeDevice()
86. {
87.     if(m_pWaveFile != NULL) 
88.     {
89.         delete m_pWaveFile;
90.     }
92.     if(pThread)
93.     {
94.         pThread->Delete();
95.     }
97.     m_lpCapture->Release();
99.     return TRUE;
100. }
102. BOOL CSoundRecord::CreateBuffer()
103. {
104.     ZeroMemory(&g_wformat, sizeof(g_wformat));
105.     g_wformat.wFormatTag = WAVE_FORMAT_PCM;
106.     g_wformat.nSamplesPerSec = 11025;//22050;
107.     g_wformat.wBitsPerSample = 8;
108.     g_wformat.nChannels = 1;
109.     g_wformat.nBlockAlign = g_wformat.nChannels * (g_wformat.wBitsPerSample / 8);
110.     g_wformat.nAvgBytesPerSec = g_wformat.nBlockAlign * g_wformat.nSamplesPerSec;
111.     g_wformat.cbSize = 0;
113.     DSCBUFFERDESC dscBufDesc;
114.     ZeroMemory(&dscBufDesc, sizeof(dscBufDesc));
115.     dscBufDesc.dwSize = sizeof(DSCBUFFERDESC);
116.     dscBufDesc.dwFlags = 0;// | DSBCAPS_CTRLVOLUME;
117.     dscBufDesc.dwBufferBytes = m_dwBufferSize;//m_wfx.nAvgBytesPerSec;
118.     dscBufDesc.dwReserved = 0;
119.     dscBufDesc.lpwfxFormat = &g_wformat;
120.     dscBufDesc.dwFXCount = 0;
121.     dscBufDesc.lpDSCFXDesc = NULL;  
122.     
123.     HRESULT hr = m_lpCapture->CreateCaptureBuffer(&dscBufDesc, &m_lpBuffer, NULL);
125.     if(FAILED(hr))
126.     {
127.         return FALSE;
128.     }
130.     hr = m_lpBuffer->QueryInterface(IID_IDirectSoundCaptureBuffer8, (LPVOID*)&m_lpBuffer8);
132.     if(FAILED(hr))
133.     {
134.         return FALSE;
135.     }
137.     m_lpBuffer->Release();
138.     
139.     return TRUE;
140. }
142. BOOL CSoundRecord::CreateNotification()
143. {
144.     m_hEventNotify = CreateEvent(NULL, FALSE, FALSE, NULL);
146.     HRESULT hr = m_lpBuffer8->QueryInterface(IID_IDirectSoundNotify, (VOID**)&m_lpNotify);
148.     if(FAILED(hr))
149.     {
150.         return FALSE;
151.     }
153.     for (int i=0; i<NOTIFICATIONS_PORTION; i++)
154.     {
155.         m_NotifyPosition[i].dwOffset = (i + 1)*m_dwNotifySize - 1;
156.         m_NotifyPosition[i].hEventNotify = m_hEventNotify;
157.     }
159.     hr = m_lpNotify->SetNotificationPositions(NOTIFICATIONS_PORTION, m_NotifyPosition);
161.     if(FAILED(hr))
162.     {
163.         return FALSE;
164.     }
166.     return TRUE;
167. }
169. void CSoundRecord::WriteData()
170. {
171.     DWORD dwReadPos;
172.     DWORD dwCapturePos;
173.     LONG lgLockSize;
175.     HRESULT hr = m_lpBuffer8->GetCurrentPosition(&dwCapturePos, &dwReadPos);
177.     if(FAILED(hr))
178.     {
179.         return;
180.     }
182.     lgLockSize = dwReadPos - m_dwBufferOffset;
184.     CorrectLocksize(&lgLockSize);
187.     VOID *pbCaptureData = NULL;
188.     DWORD dwCaptureLength = 0;
189.     VOID *pbCaptureData2 = NULL;
190.     DWORD dwCaptureLength2 = 0;
191.     UINT dwDataWrote;   
193.     // lock 
194.     hr = m_lpBuffer8->Lock(m_dwBufferOffset, lgLockSize, 
195.         &pbCaptureData, &dwCaptureLength, &pbCaptureData2, &dwCaptureLength2, 
196.         0L);
198.     hr = m_pWaveFile->Write(dwCaptureLength, (BYTE*)pbCaptureData, &dwDataWrote);
200.     m_dwBufferOffset+=dwCaptureLength;
201.     m_dwBufferOffset%=m_dwBufferSize;
203.     if(pbCaptureData2 != NULL)
204.     {
205.         hr = m_pWaveFile->Write(dwCaptureLength2, (BYTE*)pbCaptureData2, &dwDataWrote);
206.         if(FAILED(hr))
207.             return;
209.         m_dwBufferOffset+=dwCaptureLength2;
210.         m_dwBufferOffset%=m_dwBufferSize;
211.     }
213.     // unlock
214.     m_lpBuffer8->Unlock(pbCaptureData, dwCaptureLength, 
215.         pbCaptureData2, dwCaptureLength2);  
216. }
218. UINT CSoundRecord::ThreadProc(LPVOID pParam)
219. {
220.     CSoundRecord* pDSoundEx = (CSoundRecord*)pParam;
222.     DWORD dwResult;
224.     while (TRUE)
225.     {
226.         dwResult = MsgWaitForMultipleObjects(1, &pDSoundEx->m_hEventNotify, FALSE, INFINITE, NULL);
228.         switch(dwResult)
229.         {
230.         case WAIT_OBJECT_0 + 0:
231.             pDSoundEx->WriteData();
232.         }
233.     }
235.     return 0;
236. }
238. void CSoundRecord::CorrectLocksize(LONG* lpsize)
239. {
240.     if(*lpsize < 0)
241.     {
242.         *lpsize += m_dwBufferSize;
243.     }
245.     //锁住内存的大小
246.     //这里取模是为了使得我们读取的数据大小为g_dwNotifySize整数倍，这样buffer里剩下的也是notify的倍数
247.     *lpsize -= (*lpsize%m_dwNotifySize);
249.     if (*lpsize <= 0)
250.     {
251.         return;
252.     }
253. }