封装的声音采集播放类

#include "stdafx.h" #include "WaveInput.h" #define INP_BUFFER_SIZE 16384 CWaveInput::CWaveInput() : //m_hWaveIn(NULL), m_pBuffer1(NULL), m_pBuffer2(NULL), m_bRecording(false), m_bInit(false), m_pEvent(NULL) { } CWaveInput::~CWaveInput() { UnInit(); } bool CWaveInput::Init(IWaveInputEvent* pEvent) { if (m_bInit) return true; m_pEvent = pEvent; m_pBuffer1 = new BYTE[INP_BUFFER_SIZE]; m_pBuffer2 = new BYTE[INP_BUFFER_SIZE]; m_waveHdr1.lpData = (LPSTR)m_pBuffer1 ; m_waveHdr1.dwBufferLength = INP_BUFFER_SIZE ; m_waveHdr1.dwBytesRecorded = 0 ; m_waveHdr1.dwUser = 0 ; m_waveHdr1.dwFlags = 0 ; m_waveHdr1.dwLoops = 1 ; m_waveHdr1.lpNext = NULL ; m_waveHdr1.reserved = 0 ; m_waveHdr2.lpData = (LPSTR) m_pBuffer2 ; m_waveHdr2.dwBufferLength = INP_BUFFER_SIZE ; m_waveHdr2.dwBytesRecorded = 0 ; m_waveHdr2.dwUser = 0 ; m_waveHdr2.dwFlags = 0 ; m_waveHdr2.dwLoops = 1 ; m_waveHdr2.lpNext = NULL ; m_waveHdr2.reserved = 0 ; m_bInit = true; return true; } void CWaveInput::UnInit() { if (!m_bInit) return ; StopRecord(); if (m_pBuffer1 != NULL) { delete m_pBuffer1; m_pBuffer1 = NULL; } if (m_pBuffer2 != NULL) { delete m_pBuffer2; m_pBuffer2 = NULL; } } bool CWaveInput::StartRecord() { if (m_bRecording) return true; WAVEFORMATEX waveform; // Open waveform audio for input waveform.wFormatTag = WAVE_FORMAT_PCM ; waveform.nChannels = 1 ; waveform.nSamplesPerSec = 11025 ; waveform.nAvgBytesPerSec = 11025 ; waveform.nBlockAlign = 1 ; waveform.wBitsPerSample = 8 ; waveform.cbSize = 0 ; if (waveInOpen (&m_hWaveIn, WAVE_MAPPER, &waveform, (DWORD) WaveInProc, (DWORD)this, CALLBACK_FUNCTION)) { return false; } waveInPrepareHeader (m_hWaveIn, &(m_waveHdr1), sizeof(WAVEHDR)) ; waveInAddBuffer (m_hWaveIn, &(m_waveHdr1), sizeof (WAVEHDR)) ; waveInPrepareHeader (m_hWaveIn, &(m_waveHdr2), sizeof (WAVEHDR)) ; waveInAddBuffer (m_hWaveIn, &(m_waveHdr2), sizeof (WAVEHDR)) ; MMRESULT result = waveInStart (m_hWaveIn) ; if (MMSYSERR_NOERROR == result) { } m_bRecording = true; return true; } void CWaveInput::StopRecord() { m_bRecording = false; MMRESULT mmReturn = ::waveInStop(m_hWaveIn); if(!mmReturn) { mmReturn = ::waveInReset(m_hWaveIn); } Sleep(500); if(!mmReturn) { mmReturn = waveInClose(m_hWaveIn); } } void CALLBACK CWaveInput::WaveInProc(HWAVEIN hwi, UINT uMsg, DWORD dwInstance, DWORD dwParm1, DWORD dwParm2) { CWaveInput* pThis = (CWaveInput*)dwInstance; switch(uMsg) { case WIM_CLOSE: { if (pThis->m_pEvent) pThis->m_pEvent->OnWaveInput_WIM_CLOSE(); } break; case WIM_DATA: { if (!pThis->m_bRecording) return; LPWAVEHDR lpHdr = (LPWAVEHDR) dwParm1; if(lpHdr->dwBytesRecorded==0 || lpHdr==NULL) return ; // 发送一个通知告诉设备驱动程序该数据块已不能用于录音 ::waveInUnprepareHeader(hwi, lpHdr, sizeof(WAVEHDR)); // 处理数据 if (pThis->m_pEvent) { pThis->m_pEvent->OnWaveInput_WIM_DATA( (PBYTE)lpHdr->lpData, (DWORD)lpHdr->dwBytesRecorded ); } // 这个数据块使用完后再重新添加到设备中 if (pThis->m_bRecording) { ::waveInPrepareHeader(hwi, lpHdr, sizeof(WAVEHDR)); ::waveInAddBuffer(hwi, lpHdr, sizeof(WAVEHDR)); } else { int i = 0; // do nothing } } break; case WIM_OPEN: { if (pThis->m_pEvent) pThis->m_pEvent->OnWaveInput_WIM_OPEN(); } break; } }