forked from LeenkxTeam/LNXSDK
		
	
		
			
				
	
	
		
			309 lines
		
	
	
		
			9.4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			309 lines
		
	
	
		
			9.4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| #include <kinc/audio2/audio.h>
 | |
| 
 | |
| #include <kinc/backend/SystemMicrosoft.h>
 | |
| 
 | |
| #include <kinc/error.h>
 | |
| #include <kinc/log.h>
 | |
| 
 | |
| // Windows 7
 | |
| #define WINVER 0x0601
 | |
| #ifdef _WIN32_WINNT
 | |
| #undef _WIN32_WINNT
 | |
| #endif
 | |
| #define _WIN32_WINNT 0x0601
 | |
| 
 | |
| #define NOATOM
 | |
| #define NOCLIPBOARD
 | |
| #define NOCOLOR
 | |
| #define NOCOMM
 | |
| #define NOCTLMGR
 | |
| #define NODEFERWINDOWPOS
 | |
| #define NODRAWTEXT
 | |
| // #define NOGDI
 | |
| #define NOGDICAPMASKS
 | |
| #define NOHELP
 | |
| #define NOICONS
 | |
| #define NOKANJI
 | |
| #define NOKEYSTATES
 | |
| #define NOMB
 | |
| #define NOMCX
 | |
| #define NOMEMMGR
 | |
| #define NOMENUS
 | |
| #define NOMETAFILE
 | |
| #define NOMINMAX
 | |
| // #define NOMSG
 | |
| #define NONLS
 | |
| #define NOOPENFILE
 | |
| #define NOPROFILER
 | |
| #define NORASTEROPS
 | |
| #define NOSCROLL
 | |
| #define NOSERVICE
 | |
| #define NOSHOWWINDOW
 | |
| #define NOSOUND
 | |
| #define NOSYSCOMMANDS
 | |
| #define NOSYSMETRICS
 | |
| #define NOTEXTMETRIC
 | |
| // #define NOUSER
 | |
| #define NOVIRTUALKEYCODES
 | |
| #define NOWH
 | |
| #define NOWINMESSAGES
 | |
| #define NOWINOFFSETS
 | |
| #define NOWINSTYLES
 | |
| #define WIN32_LEAN_AND_MEAN
 | |
| 
 | |
| #include <initguid.h>
 | |
| 
 | |
| #include <AudioClient.h>
 | |
| #include <mmdeviceapi.h>
 | |
| 
 | |
| #ifndef __MINGW32__
 | |
| // MIDL_INTERFACE("1CB9AD4C-DBFA-4c32-B178-C2F568A703B2")
 | |
| DEFINE_GUID(IID_IAudioClient, 0x1CB9AD4C, 0xDBFA, 0x4c32, 0xB1, 0x78, 0xC2, 0xF5, 0x68, 0xA7, 0x03, 0xB2);
 | |
| // MIDL_INTERFACE("F294ACFC-3146-4483-A7BF-ADDCA7C260E2")
 | |
| DEFINE_GUID(IID_IAudioRenderClient, 0xF294ACFC, 0x3146, 0x4483, 0xA7, 0xBF, 0xAD, 0xDC, 0xA7, 0xC2, 0x60, 0xE2);
 | |
| // MIDL_INTERFACE("A95664D2-9614-4F35-A746-DE8DB63617E6")
 | |
| DEFINE_GUID(IID_IMMDeviceEnumerator, 0xA95664D2, 0x9614, 0x4F35, 0xA7, 0x46, 0xDE, 0x8D, 0xB6, 0x36, 0x17, 0xE6);
 | |
| // DECLSPEC_UUID("BCDE0395-E52F-467C-8E3D-C4579291692E")
 | |
| DEFINE_GUID(CLSID_MMDeviceEnumerator, 0xBCDE0395, 0xE52F, 0x467C, 0x8E, 0x3D, 0xC4, 0x57, 0x92, 0x91, 0x69, 0x2E);
 | |
| #endif
 | |
| 
 | |
| // based on the implementation in soloud and Microsoft sample code
 | |
| static kinc_a2_buffer_t a2_buffer;
 | |
| 
 | |
| static IMMDeviceEnumerator *deviceEnumerator;
 | |
| static IMMDevice *device;
 | |
| static IAudioClient *audioClient = NULL;
 | |
| static IAudioRenderClient *renderClient = NULL;
 | |
| static HANDLE bufferEndEvent = 0;
 | |
| static HANDLE audioProcessingDoneEvent;
 | |
| static UINT32 bufferFrames;
 | |
| static WAVEFORMATEX requestedFormat;
 | |
| static WAVEFORMATEX *closestFormat;
 | |
| static WAVEFORMATEX *format;
 | |
| static uint32_t samples_per_second = 44100;
 | |
| 
 | |
| uint32_t kinc_a2_samples_per_second(void) {
 | |
| 	return samples_per_second;
 | |
| }
 | |
| 
 | |
| static bool initDefaultDevice() {
 | |
| 	if (renderClient != NULL) {
 | |
| 		renderClient->lpVtbl->Release(renderClient);
 | |
| 		renderClient = NULL;
 | |
| 	}
 | |
| 
 | |
| 	if (audioClient != NULL) {
 | |
| 		audioClient->lpVtbl->Release(audioClient);
 | |
| 		audioClient = NULL;
 | |
| 	}
 | |
| 
 | |
| 	if (bufferEndEvent != 0) {
 | |
| 		CloseHandle(bufferEndEvent);
 | |
| 		bufferEndEvent = 0;
 | |
| 	}
 | |
| 
 | |
| 	kinc_log(KINC_LOG_LEVEL_INFO, "Initializing a new default audio device.");
 | |
| 
 | |
| 	HRESULT hr = deviceEnumerator->lpVtbl->GetDefaultAudioEndpoint(deviceEnumerator, eRender, eConsole, &device);
 | |
| 	if (hr == S_OK) {
 | |
| 		hr = device->lpVtbl->Activate(device, &IID_IAudioClient, CLSCTX_ALL, 0, (void **)&audioClient);
 | |
| 	}
 | |
| 
 | |
| 	if (hr == S_OK) {
 | |
| 		const int sampleRate = 48000;
 | |
| 
 | |
| 		format = &requestedFormat;
 | |
| 		memset(&requestedFormat, 0, sizeof(WAVEFORMATEX));
 | |
| 		requestedFormat.nChannels = 2;
 | |
| 		requestedFormat.nSamplesPerSec = sampleRate;
 | |
| 		requestedFormat.wFormatTag = WAVE_FORMAT_PCM;
 | |
| 		requestedFormat.wBitsPerSample = sizeof(short) * 8;
 | |
| 		requestedFormat.nBlockAlign = (requestedFormat.nChannels * requestedFormat.wBitsPerSample) / 8;
 | |
| 		requestedFormat.nAvgBytesPerSec = requestedFormat.nSamplesPerSec * requestedFormat.nBlockAlign;
 | |
| 		requestedFormat.cbSize = 0;
 | |
| 
 | |
| 		HRESULT supported = audioClient->lpVtbl->IsFormatSupported(audioClient, AUDCLNT_SHAREMODE_SHARED, format, &closestFormat);
 | |
| 		if (supported == S_FALSE) {
 | |
| 			kinc_log(KINC_LOG_LEVEL_WARNING, "Falling back to the system's preferred WASAPI mix format.", supported);
 | |
| 			if (closestFormat != NULL) {
 | |
| 				format = closestFormat;
 | |
| 			}
 | |
| 			else {
 | |
| 				audioClient->lpVtbl->GetMixFormat(audioClient, &format);
 | |
| 			}
 | |
| 		}
 | |
| 		HRESULT result =
 | |
| 		    audioClient->lpVtbl->Initialize(audioClient, AUDCLNT_SHAREMODE_SHARED, AUDCLNT_STREAMFLAGS_EVENTCALLBACK, 40 * 1000 * 10, 0, format, 0);
 | |
| 		if (result != S_OK) {
 | |
| 			kinc_log(KINC_LOG_LEVEL_WARNING, "Could not initialize WASAPI audio, going silent (error code 0x%x).", result);
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		uint32_t old_samples_per_second = samples_per_second;
 | |
| 		samples_per_second = format->nSamplesPerSec;
 | |
| 		if (samples_per_second != old_samples_per_second) {
 | |
| 			kinc_a2_internal_sample_rate_callback();
 | |
| 		}
 | |
| 		a2_buffer.channel_count = 2;
 | |
| 
 | |
| 		bufferFrames = 0;
 | |
| 		kinc_microsoft_affirm(audioClient->lpVtbl->GetBufferSize(audioClient, &bufferFrames));
 | |
| 		kinc_microsoft_affirm(audioClient->lpVtbl->GetService(audioClient, &IID_IAudioRenderClient, (void **)&renderClient));
 | |
| 
 | |
| 		bufferEndEvent = CreateEvent(0, FALSE, FALSE, 0);
 | |
| 		kinc_affirm(bufferEndEvent != 0);
 | |
| 
 | |
| 		kinc_microsoft_affirm(audioClient->lpVtbl->SetEventHandle(audioClient, bufferEndEvent));
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 	else {
 | |
| 		kinc_log(KINC_LOG_LEVEL_WARNING, "Could not initialize WASAPI audio.");
 | |
| 		return false;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| static void copyS16Sample(int16_t *left, int16_t *right) {
 | |
| 	float left_value = *(float *)&a2_buffer.channels[0][a2_buffer.read_location];
 | |
| 	float right_value = *(float *)&a2_buffer.channels[1][a2_buffer.read_location];
 | |
| 	a2_buffer.read_location += 1;
 | |
| 	if (a2_buffer.read_location >= a2_buffer.data_size) {
 | |
| 		a2_buffer.read_location = 0;
 | |
| 	}
 | |
| 	*left = (int16_t)(left_value * 32767);
 | |
| 	*right = (int16_t)(right_value * 32767);
 | |
| }
 | |
| 
 | |
| static void copyFloatSample(float *left, float *right) {
 | |
| 	float left_value = *(float *)&a2_buffer.channels[0][a2_buffer.read_location];
 | |
| 	float right_value = *(float *)&a2_buffer.channels[1][a2_buffer.read_location];
 | |
| 	a2_buffer.read_location += 1;
 | |
| 	if (a2_buffer.read_location >= a2_buffer.data_size) {
 | |
| 		a2_buffer.read_location = 0;
 | |
| 	}
 | |
| 	*left = left_value;
 | |
| 	*right = right_value;
 | |
| }
 | |
| 
 | |
| static void submitEmptyBuffer(unsigned frames) {
 | |
| 	BYTE *buffer = NULL;
 | |
| 	HRESULT result = renderClient->lpVtbl->GetBuffer(renderClient, frames, &buffer);
 | |
| 	if (FAILED(result)) {
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	memset(buffer, 0, frames * format->nBlockAlign);
 | |
| 
 | |
| 	result = renderClient->lpVtbl->ReleaseBuffer(renderClient, frames, 0);
 | |
| }
 | |
| 
 | |
| static void submitBuffer(unsigned frames) {
 | |
| 	BYTE *buffer = NULL;
 | |
| 	HRESULT result = renderClient->lpVtbl->GetBuffer(renderClient, frames, &buffer);
 | |
| 	if (FAILED(result)) {
 | |
| 		if (result == AUDCLNT_E_DEVICE_INVALIDATED) {
 | |
| 			initDefaultDevice();
 | |
| 			submitEmptyBuffer(bufferFrames);
 | |
| 			audioClient->lpVtbl->Start(audioClient);
 | |
| 		}
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	if (kinc_a2_internal_callback(&a2_buffer, frames)) {
 | |
| 		if (format->wFormatTag == WAVE_FORMAT_PCM) {
 | |
| 			for (UINT32 i = 0; i < frames; ++i) {
 | |
| 				copyS16Sample((int16_t *)&buffer[i * format->nBlockAlign], (int16_t *)&buffer[i * format->nBlockAlign + 2]);
 | |
| 			}
 | |
| 		}
 | |
| 		else {
 | |
| 			for (UINT32 i = 0; i < frames; ++i) {
 | |
| 				copyFloatSample((float *)&buffer[i * format->nBlockAlign], (float *)&buffer[i * format->nBlockAlign + 4]);
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	else {
 | |
| 		memset(buffer, 0, frames * format->nBlockAlign);
 | |
| 	}
 | |
| 
 | |
| 	result = renderClient->lpVtbl->ReleaseBuffer(renderClient, frames, 0);
 | |
| 	if (FAILED(result)) {
 | |
| 		if (result == AUDCLNT_E_DEVICE_INVALIDATED) {
 | |
| 			initDefaultDevice();
 | |
| 			submitEmptyBuffer(bufferFrames);
 | |
| 			audioClient->lpVtbl->Start(audioClient);
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| static DWORD WINAPI audioThread(LPVOID ignored) {
 | |
| 	submitBuffer(bufferFrames);
 | |
| 	audioClient->lpVtbl->Start(audioClient);
 | |
| 	while (WAIT_OBJECT_0 != WaitForSingleObject(audioProcessingDoneEvent, 0)) {
 | |
| 		WaitForSingleObject(bufferEndEvent, INFINITE);
 | |
| 		UINT32 padding = 0;
 | |
| 		HRESULT result = audioClient->lpVtbl->GetCurrentPadding(audioClient, &padding);
 | |
| 		if (FAILED(result)) {
 | |
| 			if (result == AUDCLNT_E_DEVICE_INVALIDATED) {
 | |
| 				initDefaultDevice();
 | |
| 				submitEmptyBuffer(bufferFrames);
 | |
| 				audioClient->lpVtbl->Start(audioClient);
 | |
| 			}
 | |
| 			continue;
 | |
| 		}
 | |
| 		UINT32 frames = bufferFrames - padding;
 | |
| 		submitBuffer(frames);
 | |
| 	}
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| void kinc_windows_co_initialize(void);
 | |
| 
 | |
| static bool initialized = false;
 | |
| 
 | |
| void kinc_a2_init() {
 | |
| 	if (initialized) {
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	kinc_a2_internal_init();
 | |
| 	initialized = true;
 | |
| 
 | |
| 	a2_buffer.read_location = 0;
 | |
| 	a2_buffer.write_location = 0;
 | |
| 	a2_buffer.data_size = 128 * 1024;
 | |
| 	a2_buffer.channel_count = 2;
 | |
| 	a2_buffer.channels[0] = (float *)malloc(a2_buffer.data_size * sizeof(float));
 | |
| 	a2_buffer.channels[1] = (float *)malloc(a2_buffer.data_size * sizeof(float));
 | |
| 
 | |
| 	audioProcessingDoneEvent = CreateEvent(0, FALSE, FALSE, 0);
 | |
| 	kinc_affirm(audioProcessingDoneEvent != 0);
 | |
| 
 | |
| 	kinc_windows_co_initialize();
 | |
| 	kinc_microsoft_affirm(CoCreateInstance(&CLSID_MMDeviceEnumerator, NULL, CLSCTX_ALL, &IID_IMMDeviceEnumerator, (void **)&deviceEnumerator));
 | |
| 
 | |
| 	if (initDefaultDevice()) {
 | |
| 		CreateThread(0, 65536, audioThread, NULL, 0, 0);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void kinc_a2_update() {}
 | |
| 
 | |
| #define SAFE_RELEASE(punk)                                                                                                                                     \
 | |
| 	if ((punk) != NULL) {                                                                                                                                      \
 | |
| 		(punk)->Release();                                                                                                                                     \
 | |
| 		(punk) = NULL;                                                                                                                                         \
 | |
| 	}
 | |
| 
 | |
| void kinc_a2_shutdown() {
 | |
| 	// Wait for last data in buffer to play before stopping.
 | |
| 	// Sleep((DWORD)(hnsActualDuration/REFTIMES_PER_MILLISEC/2));
 | |
| 
 | |
| 	//	affirm(pAudioClient->Stop());  // Stop playing.
 | |
| 
 | |
| 	//	CoTaskMemFree(pwfx);
 | |
| 	//	SAFE_RELEASE(pEnumerator)
 | |
| 	//	SAFE_RELEASE(pDevice)
 | |
| 	//	SAFE_RELEASE(pAudioClient)
 | |
| 	//	SAFE_RELEASE(pRenderClient)
 | |
| }
 |