doubango/tinyDAV/src/audio/directsound/tdav_producer_dsound.c
c732d49e
 #if HAVE_CRT
 #define _CRTDBG_MAP_ALLOC 
 #include <stdlib.h> 
 #include <crtdbg.h>
 #endif //HAVE_CRT
 /*
74ca6d11
 * Copyright (C) 2020, University of the Basque Country (UPV/EHU)
c732d49e
 * Contact for licensing options: <licensing-mcpttclient(at)mcopenplatform(dot)com>
 *
 * The original file was part of Open Source Doubango Framework
 * Copyright (C) 2010-2011 Mamadou Diop.
 * Copyright (C) 2012 Doubango Telecom <http://doubango.org>
 *
 * This file is part of Open Source Doubango Framework.
 *
 * DOUBANGO is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * DOUBANGO is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with DOUBANGO.
 *
 */
 
 
 /**@file tdav_producer_dsound.c
  * @brief Microsoft DirectSound producer.
  *
  */
 #include "tinydav/audio/directsound/tdav_producer_dsound.h"
 
 #if HAVE_DSOUND_H
 
 #if defined(_MSC_VER)
 #	pragma comment(lib, "dsound.lib")
 // #	pragma comment(lib, "dxguid.lib")
 #endif
 
 #if !defined(SEND_SILENCE_ON_MUTE)
 #	if METROPOLIS
 #		define SEND_SILENCE_ON_MUTE	1
 #	else
 #		define SEND_SILENCE_ON_MUTE	0
 #	endif
 #endif /* SEND_SILENCE_ON_MUTE */
 
 #include "tsk_string.h"
 #include "tsk_thread.h"
 #include "tsk_memory.h"
 #include "tsk_debug.h"
 
 #include <initguid.h>
 #include <dsound.h>
 
 extern void tdav_win32_print_error(const char* func, HRESULT hr);
 
 #if !defined(TDAV_DSOUND_PRODUCER_NOTIF_POS_COUNT)
 #	define TDAV_DSOUND_PRODUCER_NOTIF_POS_COUNT		10
 #endif /* TDAV_DSOUND_PRODUCER_NOTIF_POS_COUNT */
 
 typedef struct tdav_producer_dsound_s
 {
 	TDAV_DECLARE_PRODUCER_AUDIO;
 
 	tsk_bool_t started;
 	tsk_bool_t mute;
 	tsk_size_t bytes_per_notif_size;
 	tsk_thread_handle_t* tid[1];
 
 	LPDIRECTSOUNDCAPTURE device;
 	LPDIRECTSOUNDCAPTUREBUFFER captureBuffer;
 	HANDLE notifEvents[TDAV_DSOUND_PRODUCER_NOTIF_POS_COUNT];
 }
 tdav_producer_dsound_t;
 
 static void* TSK_STDCALL _tdav_producer_dsound_record_thread(void *param)
 {
 	tdav_producer_dsound_t* dsound = (tdav_producer_dsound_t*)param;
 
 	HRESULT hr;
 	LPVOID lpvAudio1, lpvAudio2;
 	DWORD dwBytesAudio1, dwBytesAudio2, dwEvent, dwIndex;
 
 	TSK_DEBUG_INFO("_tdav_producer_dsound_record_thread -- START");
 
 	SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_TIME_CRITICAL);
 
 	while (dsound->started) {
 		dwEvent = WaitForMultipleObjects(TDAV_DSOUND_PRODUCER_NOTIF_POS_COUNT, dsound->notifEvents, FALSE, INFINITE);
 		if (!dsound->started) {
 			break;
 		}
 		if (dwEvent < WAIT_OBJECT_0 || dwEvent >(WAIT_OBJECT_0 + TDAV_DSOUND_PRODUCER_NOTIF_POS_COUNT)) {
 			TSK_DEBUG_ERROR("Invalid dwEvent(%d)", dwEvent);
 			break;
 		}
 		dwIndex = (dwEvent - WAIT_OBJECT_0);
 
 		// lock
 		if ((hr = IDirectSoundCaptureBuffer_Lock(dsound->captureBuffer, (DWORD)(dwIndex * dsound->bytes_per_notif_size), (DWORD)dsound->bytes_per_notif_size, &lpvAudio1, &dwBytesAudio1, &lpvAudio2, &dwBytesAudio2, 0)) != DS_OK) {
 			tdav_win32_print_error("IDirectSoundCaptureBuffer_Lock", hr);
 			continue;
 		}
 
 		if (TMEDIA_PRODUCER(dsound)->enc_cb.callback) {
 #if SEND_SILENCE_ON_MUTE
 			if (dsound->mute) {
 				memset(lpvAudio1, 0, dwBytesAudio1);
 				if(lpvAudio2){
 					memset(lpvAudio2, 0, dwBytesAudio2);
 				}
 			}
 #endif
 			TMEDIA_PRODUCER(dsound)->enc_cb.callback(TMEDIA_PRODUCER(dsound)->enc_cb.callback_data, lpvAudio1, dwBytesAudio1);
 			if (lpvAudio2) {
 				TMEDIA_PRODUCER(dsound)->enc_cb.callback(TMEDIA_PRODUCER(dsound)->enc_cb.callback_data, lpvAudio2, dwBytesAudio2);
 			}
 		}
 
 		// unlock
 		if ((hr = IDirectSoundCaptureBuffer_Unlock(dsound->captureBuffer, lpvAudio1, dwBytesAudio1, lpvAudio2, dwBytesAudio2)) != DS_OK) {
 			tdav_win32_print_error("IDirectSoundCaptureBuffer_Unlock", hr);
 			continue;
 		}
 	}
 
 	TSK_DEBUG_INFO("_tdav_producer_dsound_record_thread -- STOP");
 
 
 	return tsk_null;
 }
 
 static int _tdav_producer_dsound_unprepare(tdav_producer_dsound_t* dsound)
 {
 	if (dsound) {
 		tsk_size_t i;
 		if (dsound->captureBuffer) {
 			IDirectSoundCaptureBuffer_Release(dsound->captureBuffer);
 			dsound->captureBuffer = NULL;
 		}
 		if (dsound->device) {
 			IDirectSoundCapture_Release(dsound->device);
 			dsound->device = NULL;
 		}
 		for (i = 0; i < (sizeof(dsound->notifEvents) / sizeof(dsound->notifEvents[0])); i++){
 			if (dsound->notifEvents[i]) {
 				CloseHandle(dsound->notifEvents[i]);
 				dsound->notifEvents[i] = NULL;
 			}
 		}
 	}
 	return 0;
 }
 
 
 
 
 /* ============ Media Producer Interface ================= */
 static int tdav_producer_dsound_set(tmedia_producer_t* self, const tmedia_param_t* param)
 {
 	tdav_producer_dsound_t* dsound = (tdav_producer_dsound_t*)self;
 	if (param->plugin_type == tmedia_ppt_producer) {
 		if (param->value_type == tmedia_pvt_int32) {
 			if (tsk_striequals(param->key, "volume")) {
 				return 0;
 			}
 			else if (tsk_striequals(param->key, "mute")) {
 				dsound->mute = (TSK_TO_INT32((uint8_t*)param->value) != 0);
 #if !SEND_SILENCE_ON_MUTE
 				if (dsound->started) {
 					if (dsound->mute) {
 						IDirectSoundCaptureBuffer_Stop(dsound->captureBuffer);
 					}
 					else {
 						IDirectSoundCaptureBuffer_Start(dsound->captureBuffer, DSBPLAY_LOOPING);
 					}
 				}
 #endif
 				return 0;
 			}
 		}
 	}
 	return tdav_producer_audio_set(TDAV_PRODUCER_AUDIO(self), param);
 }
 static int tdav_producer_dsound_prepare(tmedia_producer_t* self, const tmedia_codec_t* codec)
 {
 	HRESULT hr;
 
 	WAVEFORMATEX wfx = { 0 };
 	DSCBUFFERDESC dsbd = { 0 };
 
 	tdav_producer_dsound_t* dsound = (tdav_producer_dsound_t*)self;
 
 	if (!dsound || !codec) {
 		TSK_DEBUG_ERROR("Invalid parameter");
 		return -1;
 	}
 
 	if (dsound->device || dsound->captureBuffer) {
 		TSK_DEBUG_ERROR("Producer already prepared");
 		return -2;
 	}
 
 	TMEDIA_PRODUCER(dsound)->audio.channels = TMEDIA_CODEC_CHANNELS_AUDIO_ENCODING(codec);
 	TMEDIA_PRODUCER(dsound)->audio.rate = TMEDIA_CODEC_RATE_ENCODING(codec);
 	TMEDIA_PRODUCER(dsound)->audio.ptime = TMEDIA_CODEC_PTIME_AUDIO_ENCODING(codec);
 
 #if 0
 	TMEDIA_PRODUCER(dsound)->audio.rate = 48000;
 	TMEDIA_PRODUCER(dsound)->audio.channels = 1;
 #endif
 
 	/* Create capture device */
 	if ((hr = DirectSoundCaptureCreate(NULL, &dsound->device, NULL) != DS_OK)) {
 		tdav_win32_print_error("DirectSoundCaptureCreate", hr);
 		return -3;
 	}
 
 	/* Creates the capture buffer */
 	wfx.wFormatTag = WAVE_FORMAT_PCM;
 	wfx.nChannels = TMEDIA_PRODUCER(dsound)->audio.channels;
 	wfx.nSamplesPerSec = TMEDIA_PRODUCER(dsound)->audio.rate;
 	wfx.wBitsPerSample = TMEDIA_PRODUCER(dsound)->audio.bits_per_sample;
 	wfx.nBlockAlign = (wfx.nChannels * wfx.wBitsPerSample / 8);
 	wfx.nAvgBytesPerSec = (wfx.nSamplesPerSec * wfx.nBlockAlign);
 
 	/* Average bytes (count) for each notification */
 	dsound->bytes_per_notif_size = ((wfx.nAvgBytesPerSec * TMEDIA_PRODUCER(dsound)->audio.ptime) / 1000);
 
 	dsbd.dwSize = sizeof(DSCBUFFERDESC);
 	dsbd.dwBufferBytes = (DWORD)(TDAV_DSOUND_PRODUCER_NOTIF_POS_COUNT * dsound->bytes_per_notif_size);
 	dsbd.lpwfxFormat = &wfx;
 
 	if ((hr = IDirectSoundCapture_CreateCaptureBuffer(dsound->device, &dsbd, &dsound->captureBuffer, NULL)) != DS_OK) {
 		tdav_win32_print_error("IDirectSoundCapture_CreateCaptureBuffer", hr);
 		return -4;
 	}
 
 	return 0;
 }
 
 static int tdav_producer_dsound_start(tmedia_producer_t* self)
 {
 	tdav_producer_dsound_t* dsound = (tdav_producer_dsound_t*)self;
 
 	tsk_size_t i;
 	DWORD dwOffset;
 	HRESULT hr;
 	LPDIRECTSOUNDNOTIFY lpDSBNotify;
 	DSBPOSITIONNOTIFY pPosNotify[TDAV_DSOUND_PRODUCER_NOTIF_POS_COUNT] = { 0 };
 
 	if (!dsound) {
 		TSK_DEBUG_ERROR("Invalid parameter");
 		return -1;
 	}
 
 	if (!dsound->device || !dsound->captureBuffer) {
 		TSK_DEBUG_ERROR("Producer not prepared");
 		return -2;
 	}
 
 	if (dsound->started) {
 		return 0;
 	}
 
 	if ((hr = IDirectSoundCaptureBuffer_QueryInterface(dsound->captureBuffer, &IID_IDirectSoundNotify, (LPVOID*)&lpDSBNotify)) != DS_OK) {
 		tdav_win32_print_error("IDirectSoundCaptureBuffer_QueryInterface", hr);
 		return -3;
 	}
 
 	/* Events associated to notification points */
 	dwOffset = (DWORD)(dsound->bytes_per_notif_size - 1);
 	for (i = 0; i < (sizeof(dsound->notifEvents) / sizeof(dsound->notifEvents[0])); i++){
 		dsound->notifEvents[i] = CreateEvent(NULL, FALSE, FALSE, NULL);
 		pPosNotify[i].dwOffset = dwOffset;
 		pPosNotify[i].hEventNotify = dsound->notifEvents[i];
 		dwOffset += (DWORD)dsound->bytes_per_notif_size;
 	}
 	if ((hr = IDirectSoundNotify_SetNotificationPositions(lpDSBNotify, TDAV_DSOUND_PRODUCER_NOTIF_POS_COUNT, pPosNotify)) != DS_OK) {
 		IDirectSoundNotify_Release(lpDSBNotify);
 		tdav_win32_print_error("IDirectSoundBuffer_QueryInterface", hr);
 		return -4;
 	}
 
 	if ((hr = IDirectSoundNotify_Release(lpDSBNotify))) {
 		tdav_win32_print_error("IDirectSoundNotify_Release", hr);
 	}
 
 	/* Start the buffer */
 	if ((hr = IDirectSoundCaptureBuffer_Start(dsound->captureBuffer, DSBPLAY_LOOPING)) != DS_OK) {
 		tdav_win32_print_error("IDirectSoundCaptureBuffer_Start", hr);
 		return -5;
 	}
 
 	/* start the reader thread */
 	dsound->started = tsk_true;
 	tsk_thread_create(&dsound->tid[0], _tdav_producer_dsound_record_thread, dsound);
 
 	return 0;
 }
 
 static int tdav_producer_dsound_pause(tmedia_producer_t* self)
 {
 	return 0;
 }
 
 static int tdav_producer_dsound_stop(tmedia_producer_t* self)
 {
 	tdav_producer_dsound_t* dsound = (tdav_producer_dsound_t*)self;
 
 	HRESULT hr;
 
 	if (!self) {
 		TSK_DEBUG_ERROR("Invalid parameter");
 		return -1;
 	}
 
 	if (!dsound->started) {
 		return 0;
 	}
 
 	// should be done here
 	dsound->started = tsk_false;
 
 #if !SEND_SILENCE_ON_MUTE
 	if (dsound->mute && dsound->notifEvents[0]) {
 		// thread is paused -> raise event now that "started" is equal to false
 		SetEvent(dsound->notifEvents[0]);
 	}
 #endif
 
 	// stop thread
 	if (dsound->tid[0]) {
 		tsk_thread_join(&(dsound->tid[0]));
 	}
 
 	if ((hr = IDirectSoundCaptureBuffer_Stop(dsound->captureBuffer)) != DS_OK) {
 		tdav_win32_print_error("IDirectSoundCaptureBuffer_Stop", hr);
 	}
 
 	// unprepare
 	// will be prepared again before next start()
 	_tdav_producer_dsound_unprepare(dsound);
 
 	return 0;
 }
 
 
 //
 //	WaveAPI producer object definition
 //
 /* constructor */
 static tsk_object_t* tdav_producer_dsound_ctor(tsk_object_t * self, va_list * app)
 {
 	tdav_producer_dsound_t *producer = self;
 	if (producer) {
 		/* init base */
 		tdav_producer_audio_init(TDAV_PRODUCER_AUDIO(producer));
 		/* init self */
 
 	}
 	return self;
 }
 /* destructor */
 static tsk_object_t* tdav_producer_dsound_dtor(tsk_object_t * self)
 {
 	tdav_producer_dsound_t *dsound = self;
 	if (dsound) {
 		/* stop */
 		if (dsound->started) {
 			tdav_producer_dsound_stop(self);
 		}
 
 		/* deinit base */
 		tdav_producer_audio_deinit(TDAV_PRODUCER_AUDIO(dsound));
 		/* deinit self */
 		_tdav_producer_dsound_unprepare(dsound);
 	}
 
 	return self;
 }
 /* object definition */
 static const tsk_object_def_t tdav_producer_dsound_def_s =
 {
 	sizeof(tdav_producer_dsound_t),
 	tdav_producer_dsound_ctor,
 	tdav_producer_dsound_dtor,
 	tdav_producer_audio_cmp,
 };
 /* plugin definition*/
 static const tmedia_producer_plugin_def_t tdav_producer_dsound_plugin_def_s =
 {
 	&tdav_producer_dsound_def_s,
 
 	tmedia_audio,
 	"Microsoft DirectSound producer",
 
 	tdav_producer_dsound_set,
 	tdav_producer_dsound_prepare,
 	tdav_producer_dsound_start,
 	tdav_producer_dsound_pause,
 	tdav_producer_dsound_stop
 };
 const tmedia_producer_plugin_def_t *tdav_producer_dsound_plugin_def_t = &tdav_producer_dsound_plugin_def_s;
 
 
 #endif /* HAVE_DSOUND_H */