Core Audio声卡输出采集

#define _CRT_SECURE_NO_WARNINGS

#pragma comment(lib,"avrt.lib")
#include<Audioclient.h>
#include <mmdeviceapi.h>
#include<iostream>
#include<avrt.h>

class CSoundCardAudioCapture
{

public:

	CSoundCardAudioCapture()
	{
		m_pAudioCaptureClient = NULL;
		m_pAudioClient = NULL;
		m_pMMDevice = NULL;
		m_hEventStop = NULL;
		m_hTimerWakeUp = NULL;
		m_hTask = NULL;
		m_pwfx = NULL;
	}

	int StartCapture()
	{
		CoInitialize(NULL);
		IMMDeviceEnumerator *pMMDeviceEnumerator = NULL;
		HRESULT hr = CoCreateInstance(__uuidof(MMDeviceEnumerator), NULL, CLSCTX_ALL,
			__uuidof(IMMDeviceEnumerator), (void**)&pMMDeviceEnumerator);
		if (FAILED(hr))
		{
			CoUninitialize();
			return -1;
		}
		hr = pMMDeviceEnumerator->GetDefaultAudioEndpoint(eRender, eConsole, &m_pMMDevice);
		if (FAILED(hr))
		{
			CoUninitialize();
			return -1;
		}
		pMMDeviceEnumerator->Release();

		m_hEventStop = CreateEvent(NULL, TRUE, FALSE, NULL);
		if (m_hEventStop == NULL)
		{
			CoUninitialize();
			return -1;
		}

		hr = m_pMMDevice->Activate(__uuidof(IAudioClient), CLSCTX_ALL, NULL, (void**)&m_pAudioClient);
		if (FAILED(hr))
			goto error;

		REFERENCE_TIME hnsDefaultDevicePeriod(0);
		hr = m_pAudioClient->GetDevicePeriod(&hnsDefaultDevicePeriod, NULL);
		if (FAILED(hr))
			goto error;

		hr = m_pAudioClient->GetMixFormat(&m_pwfx);
		if (FAILED(hr))
			goto error;

		if (!AdjustFormatTo16Bits(m_pwfx))
			goto error;

		m_hTimerWakeUp = CreateWaitableTimer(NULL, FALSE, NULL);
		if (m_hTimerWakeUp == NULL)
			goto error;

		hr = m_pAudioClient->Initialize(AUDCLNT_SHAREMODE_SHARED, AUDCLNT_STREAMFLAGS_LOOPBACK, 0, 0, m_pwfx, 0);
		if (FAILED(hr))
			goto error;

		hr = m_pAudioClient->GetService(__uuidof(IAudioCaptureClient), (void**)&m_pAudioCaptureClient);
		if (FAILED(hr))
			goto error;

		DWORD nTaskIndex = 0;
		m_hTask = AvSetMmThreadCharacteristics(L"Capture", &nTaskIndex);
		if (NULL == m_hTask)
			goto error;

		LARGE_INTEGER liFirstFire;
		liFirstFire.QuadPart = -hnsDefaultDevicePeriod / 2; // negative means relative time
		LONG lTimeBetweenFires = (LONG)hnsDefaultDevicePeriod / 2 / (10 * 1000); // convert to milliseconds

		BOOL bOK = SetWaitableTimer(m_hTimerWakeUp, &liFirstFire, lTimeBetweenFires, NULL, NULL, FALSE);
		if (!bOK)
			goto error;

		hr = m_pAudioClient->Start();
		if (FAILED(hr))
			goto error;

		m_hThread = CreateThread(NULL, 0, PTHREAD_START_ROUTINE_CALLBACK, this, 0, 0);
		if (m_hThread == NULL)
			goto error;

		CoUninitialize();
		return 0;
	error:
		Close();
		CoUninitialize();
		return -1;
	}

	int StopCapture()
	{
		if (m_pAudioClient)
			m_pAudioClient->Stop();
		SetEvent(m_hEventStop);
		WaitForSingleObject(m_hThread, -1);
		Close();
		return 0;
	}

	void DoWork()
	{
		HANDLE waitArray[2] = { m_hEventStop, m_hTimerWakeUp };
		DWORD dwWaitResult;
		UINT32 nNextPacketSize(0);
		BYTE *pData = NULL;
		UINT32 nNumFramesToRead;
		DWORD dwFlags;
		CoInitialize(NULL);
		FILE *file = fopen("D:\\pwm.data", "wb+");
		while (TRUE)
		{
			dwWaitResult = WaitForMultipleObjects(sizeof(waitArray) / sizeof(waitArray[0]), waitArray, FALSE, INFINITE);
			if (WAIT_OBJECT_0 == dwWaitResult) break;

			if (WAIT_OBJECT_0 + 1 != dwWaitResult)
			{
				break;
			}

			HRESULT hr = m_pAudioCaptureClient->GetNextPacketSize(&nNextPacketSize);
			if (FAILED(hr))
			{
				break;
			}

			if (nNextPacketSize == 0) continue;

			hr = m_pAudioCaptureClient->GetBuffer(&pData, &nNumFramesToRead, &dwFlags, NULL, NULL);
			if (FAILED(hr))
			{
				break;
			}

			if (0 != nNumFramesToRead)
			{
				std::cout << "capture data " << nNumFramesToRead * m_pwfx->nBlockAlign << std::endl;
				fwrite(pData, 1, nNumFramesToRead * m_pwfx->nBlockAlign,file);
			}
			m_pAudioCaptureClient->ReleaseBuffer(nNumFramesToRead);
		}
		CoUninitialize();
	}

	static DWORD WINAPI PTHREAD_START_ROUTINE_CALLBACK(LPVOID lpThreadParameter);

protected:

	void Close()
	{
		if (m_hEventStop != NULL)
		{
			CloseHandle(m_hEventStop);
			m_hEventStop = NULL;
		}
		if (m_pAudioClient)
		{
			m_pAudioClient->Release();
			m_pAudioClient = NULL;
		}
		if (m_pwfx != NULL)
		{
			CoTaskMemFree(m_pwfx);
			m_pwfx = NULL;
		}
		if (m_hTimerWakeUp != NULL)
		{
			CancelWaitableTimer(m_hTimerWakeUp);
			CloseHandle(m_hTimerWakeUp);
			m_hTimerWakeUp = NULL;
		}
		if (m_hTask != NULL)
		{
			AvRevertMmThreadCharacteristics(m_hTask);
			m_hTask = NULL;
		}
		if (m_pAudioCaptureClient != NULL)
		{
			m_pAudioCaptureClient->Release();
			m_pAudioCaptureClient = NULL;
		}
	}

	BOOL AdjustFormatTo16Bits(WAVEFORMATEX *pwfx)
	{
		BOOL bRet(FALSE);

		if (pwfx->wFormatTag == WAVE_FORMAT_IEEE_FLOAT)
		{
			pwfx->wFormatTag = WAVE_FORMAT_PCM;
			pwfx->wBitsPerSample = 16;
			pwfx->nBlockAlign = pwfx->nChannels * pwfx->wBitsPerSample / 8;
			pwfx->nAvgBytesPerSec = pwfx->nBlockAlign * pwfx->nSamplesPerSec;
			bRet = TRUE;
		}
		else if (pwfx->wFormatTag == WAVE_FORMAT_EXTENSIBLE)
		{
			PWAVEFORMATEXTENSIBLE pEx = reinterpret_cast<PWAVEFORMATEXTENSIBLE>(pwfx);
			if (IsEqualGUID(KSDATAFORMAT_SUBTYPE_IEEE_FLOAT, pEx->SubFormat))
			{
				pEx->SubFormat = KSDATAFORMAT_SUBTYPE_PCM;
				pEx->Samples.wValidBitsPerSample = 16;
				pwfx->wBitsPerSample = 16;
				pwfx->nBlockAlign = pwfx->nChannels * pwfx->wBitsPerSample / 8;
				pwfx->nAvgBytesPerSec = pwfx->nBlockAlign * pwfx->nSamplesPerSec;

				bRet = TRUE;
			}
		}

		return bRet;
	}


private:

	HANDLE m_hThread;
	HANDLE m_hTask;
	HANDLE m_hTimerWakeUp;
	IAudioCaptureClient * m_pAudioCaptureClient;
	IAudioClient * m_pAudioClient;
	WAVEFORMATEX * m_pwfx;
	HANDLE m_hEventStop;
	IMMDevice* m_pMMDevice;

};

DWORD CSoundCardAudioCapture::PTHREAD_START_ROUTINE_CALLBACK(LPVOID lpThreadParameter)
{
	CSoundCardAudioCapture* pCapture = (CSoundCardAudioCapture*)lpThreadParameter;
	pCapture->DoWork();
	return 0;
}


int main(void)
{
	CSoundCardAudioCapture cap;
	while (true)
	{
		cap.StartCapture();
		getchar();
		cap.StopCapture();
	}
	return 0;
}