doubango/branches/2.0/doubango/tinyDAV/src/audio/wasapi/tdav_producer_wasapi.cxx

670 lines
18 KiB
C++

/*Copyright (C) 2013 Mamadou Diop
* Copyright (C) 2013 Doubango Telecom <http://www.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_wasapi.h
* @brief Microsoft Windows Audio Session API (WASAPI) producer.
* http://msdn.microsoft.com/en-us/library/windows/desktop/dd316551(v=vs.85).aspx
*/
#include "tinydav/audio/wasapi/tdav_producer_wasapi.h"
#if HAVE_WASAPI
#include "tinydav/audio/tdav_producer_audio.h"
#include "tsk_memory.h"
#include "tsk_string.h"
#include "tsk_debug.h"
#include <windows.h>
#include <audioclient.h>
#include <phoneaudioclient.h>
#include <speex/speex_buffer.h>
#if !defined(TDAV_WASAPI_PRODUCER_NOTIF_POS_COUNT)
# define TDAV_WASAPI_PRODUCER_NOTIF_POS_COUNT 10
#endif
#define WASAPI_MILLIS_TO_100NS(MILLIS) (((LONGLONG)(MILLIS)) * 10000ui64)
#define WASAPI_100NS_TO_MILLIS(NANOS) (((LONGLONG)(NANOS)) / 10000ui64)
struct tdav_producer_wasapi_s;
namespace Doubango
{
namespace VoIP
{
ref class AudioCapture sealed
{
public:
virtual ~AudioCapture();
internal:
AudioCapture();
int Prepare(struct tdav_producer_wasapi_s* wasapi, const tmedia_codec_t* codec);
int UnPrepare();
int Start();
int Stop();
int Pause();
private:
void AsyncThread(Windows::Foundation::IAsyncAction^ operation);
private:
tsk_mutex_handle_t* m_hMutex;
IAudioClient2* m_pDevice;
IAudioCaptureClient* m_pClient;
HANDLE m_hCaptureEvent;
HANDLE m_hShutdownEvent;
Windows::Foundation::IAsyncAction^ m_pAsyncThread;
INT32 m_nBytesPerNotif;
INT32 m_nSourceFrameSizeInBytes;
struct{
tmedia_producer_enc_cb_f fn;
const void* pcData;
} m_callback;
struct {
struct {
void* buffer;
tsk_size_t size;
} chunck;
SpeexBuffer* buffer;
tsk_size_t size;
} m_ring;
bool m_bStarted;
bool m_bPrepared;
bool m_bPaused;
};
}
}
typedef struct tdav_producer_wasapi_s
{
TDAV_DECLARE_PRODUCER_AUDIO;
Doubango::VoIP::AudioCapture ^audioCapture;
}
tdav_producer_wasapi_t;
extern "C" void tdav_win32_print_error(const char* func, HRESULT hr);
/* ============ Media Producer Interface ================= */
static int tdav_producer_wasapi_set(tmedia_producer_t* self, const tmedia_param_t* param)
{
tdav_producer_wasapi_t* wasapi = (tdav_producer_wasapi_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")){
//wasapi->mute = (TSK_TO_INT32((uint8_t*)param->value) != 0);
#if !FIXME_SEND_SILENCE_ON_MUTE
//if(wasapi->started){
// if(wasapi->mute){
//IDirectSoundCaptureBuffer_Stop(wasapi->captureBuffer);
// }
// else{
//IDirectSoundCaptureBuffer_Start(wasapi->captureBuffer, DSBPLAY_LOOPING);
// }
//}
#endif
return 0;
}
}
}
return tdav_producer_audio_set(TDAV_PRODUCER_AUDIO(self), param);
}
static int tdav_producer_wasapi_prepare(tmedia_producer_t* self, const tmedia_codec_t* codec)
{
tdav_producer_wasapi_t* wasapi = (tdav_producer_wasapi_t*)self;
if(!wasapi || !codec || !wasapi->audioCapture){
TSK_DEBUG_ERROR("Invalid parameter");
return -1;
}
TMEDIA_PRODUCER(wasapi)->audio.channels = codec->plugin->audio.channels;
TMEDIA_PRODUCER(wasapi)->audio.rate = codec->plugin->rate;
TMEDIA_PRODUCER(wasapi)->audio.ptime = codec->plugin->audio.ptime;
return wasapi->audioCapture->Prepare(wasapi, codec);
}
static int tdav_producer_wasapi_start(tmedia_producer_t* self)
{
tdav_producer_wasapi_t* wasapi = (tdav_producer_wasapi_t*)self;
TSK_DEBUG_INFO("tdav_producer_wasapi_start()");
if(!wasapi || !wasapi->audioCapture){
TSK_DEBUG_ERROR("Invalid parameter");
return -1;
}
return wasapi->audioCapture->Start();
}
static int tdav_producer_wasapi_pause(tmedia_producer_t* self)
{
tdav_producer_wasapi_t* wasapi = (tdav_producer_wasapi_t*)self;
if(!wasapi || !wasapi->audioCapture){
TSK_DEBUG_ERROR("Invalid parameter");
return -1;
}
return wasapi->audioCapture->Pause();
}
static int tdav_producer_wasapi_stop(tmedia_producer_t* self)
{
tdav_producer_wasapi_t* wasapi = (tdav_producer_wasapi_t*)self;
TSK_DEBUG_INFO("tdav_producer_wasapi_stop()");
if(!wasapi || !wasapi->audioCapture){
TSK_DEBUG_ERROR("Invalid parameter");
return -1;
}
return wasapi->audioCapture->Stop();
}
Doubango::VoIP::AudioCapture::AudioCapture()
: m_pDevice(nullptr)
, m_hMutex(nullptr)
, m_pClient(nullptr)
, m_hCaptureEvent(nullptr)
, m_hShutdownEvent(nullptr)
, m_pAsyncThread(nullptr)
, m_nBytesPerNotif(0)
, m_nSourceFrameSizeInBytes(0)
, m_bStarted(false)
, m_bPrepared(false)
, m_bPaused(false)
{
m_callback.fn = nullptr, m_callback.pcData = nullptr;
if(!(m_hMutex = tsk_mutex_create())){
throw ref new Platform::FailureException(L"Failed to create mutex");
}
}
Doubango::VoIP::AudioCapture::~AudioCapture()
{
Stop();
UnPrepare();
tsk_mutex_destroy(&m_hMutex);
}
int Doubango::VoIP::AudioCapture::Prepare(tdav_producer_wasapi_t* wasapi, const tmedia_codec_t* codec)
{
HRESULT hr = E_FAIL;
int ret = 0;
WAVEFORMATEX wfx = {0};
AudioClientProperties properties = {0};
LPCWSTR pwstrCaptureId = nullptr;
#define WASAPI_SET_ERROR(code) ret = (code); goto bail;
tsk_mutex_lock(m_hMutex);
if(m_bPrepared)
{
TSK_DEBUG_INFO("#WASAPI: Audio producer already prepared");
goto bail;
}
if(!wasapi || !codec)
{
TSK_DEBUG_ERROR("Invalid parameter");
WASAPI_SET_ERROR(-1);
}
if(m_pDevice || m_pClient){
TSK_DEBUG_ERROR("Producer already prepared");
WASAPI_SET_ERROR(-2);
}
pwstrCaptureId = GetDefaultAudioCaptureId(AudioDeviceRole::Communications);
if (NULL == pwstrCaptureId){
tdav_win32_print_error("GetDefaultAudioCaptureId", HRESULT_FROM_WIN32(GetLastError()));
WASAPI_SET_ERROR(-3);
}
hr = ActivateAudioInterface(pwstrCaptureId, __uuidof(IAudioClient2), (void**)&m_pDevice);
if(!SUCCEEDED(hr)){
tdav_win32_print_error("ActivateAudioInterface", HRESULT_FROM_WIN32(GetLastError()));
WASAPI_SET_ERROR(-4);
}
if (SUCCEEDED(hr)){
properties.cbSize = sizeof AudioClientProperties;
properties.eCategory = AudioCategory_Communications;
hr = m_pDevice->SetClientProperties(&properties);
if (!SUCCEEDED(hr)){
tdav_win32_print_error("SetClientProperties", HRESULT_FROM_WIN32(GetLastError()));
WASAPI_SET_ERROR(-5);
}
}
else{
tdav_win32_print_error("ActivateAudioInterface", HRESULT_FROM_WIN32(GetLastError()));
WASAPI_SET_ERROR(-6);
}
/* Set best format */
{
wfx.wFormatTag = WAVE_FORMAT_PCM;
wfx.nChannels = TMEDIA_PRODUCER(wasapi)->audio.channels;
wfx.nSamplesPerSec = TMEDIA_PRODUCER(wasapi)->audio.rate;
wfx.wBitsPerSample = TMEDIA_PRODUCER(wasapi)->audio.bits_per_sample;
wfx.nBlockAlign = (wfx.nChannels * wfx.wBitsPerSample/8);
wfx.nAvgBytesPerSec = (wfx.nSamplesPerSec * wfx.nBlockAlign);
PWAVEFORMATEX pwfxClosestMatch = NULL;
hr = m_pDevice->IsFormatSupported(AUDCLNT_SHAREMODE_SHARED, &wfx, &pwfxClosestMatch);
if(hr != S_OK && hr != S_FALSE)
{
tdav_win32_print_error("IsFormatSupported", HRESULT_FROM_WIN32(GetLastError()));
WASAPI_SET_ERROR(-8);
}
if(hr == S_FALSE)
{
if(!pwfxClosestMatch)
{
TSK_DEBUG_ERROR("malloc(%d) failed", sizeof(WAVEFORMATEX));
WASAPI_SET_ERROR(-7);
}
wfx.nChannels = pwfxClosestMatch->nChannels;
wfx.nSamplesPerSec = pwfxClosestMatch->nSamplesPerSec;
#if 0
wfx.wBitsPerSample = pwfxClosestMatch->wBitsPerSample;
#endif
wfx.nBlockAlign = wfx.nChannels * (wfx.wBitsPerSample / 8);
wfx.nAvgBytesPerSec = wfx.nSamplesPerSec * wfx.nBlockAlign;
// Request resampler
TMEDIA_PRODUCER(wasapi)->audio.rate = (uint32_t)wfx.nSamplesPerSec;
TMEDIA_PRODUCER(wasapi)->audio.bits_per_sample = (uint8_t)wfx.wBitsPerSample;
TMEDIA_PRODUCER(wasapi)->audio.channels = (uint8_t)wfx.nChannels;
TSK_DEBUG_INFO("Audio device format fallback: rate=%d, bps=%d, channels=%d", wfx.nSamplesPerSec, wfx.wBitsPerSample, wfx.nChannels);
}
if(pwfxClosestMatch)
{
CoTaskMemFree(pwfxClosestMatch);
}
}
m_nSourceFrameSizeInBytes = (wfx.wBitsPerSample >> 3) * wfx.nChannels;
m_nBytesPerNotif = ((wfx.nAvgBytesPerSec * TMEDIA_PRODUCER(wasapi)->audio.ptime)/1000);
// Initialize
hr = m_pDevice->Initialize(
AUDCLNT_SHAREMODE_SHARED,
AUDCLNT_STREAMFLAGS_EVENTCALLBACK,
(TDAV_WASAPI_PRODUCER_NOTIF_POS_COUNT * WASAPI_MILLIS_TO_100NS(TMEDIA_PRODUCER(wasapi)->audio.ptime)),
0,
&wfx,
NULL);
if (!SUCCEEDED(hr)){
tdav_win32_print_error("#WASAPI: Capture::SetClientProperties", hr);
WASAPI_SET_ERROR(-9);
}
REFERENCE_TIME DefaultDevicePeriod, MinimumDevicePeriod;
hr = m_pDevice->GetDevicePeriod(&DefaultDevicePeriod, &MinimumDevicePeriod);
if (!SUCCEEDED(hr)){
tdav_win32_print_error("GetDevicePeriod", hr);
WASAPI_SET_ERROR(-10);
}
TSK_DEBUG_INFO("#WASAPI(Capture): DefaultDevicePeriod=%lld ms, MinimumDevicePeriod=%lldms", WASAPI_100NS_TO_MILLIS(DefaultDevicePeriod), WASAPI_100NS_TO_MILLIS(MinimumDevicePeriod));
if(!m_hCaptureEvent){
if(!(m_hCaptureEvent = CreateEventEx(NULL, NULL, 0, EVENT_ALL_ACCESS))){
tdav_win32_print_error("CreateEventEx(Capture)", HRESULT_FROM_WIN32(GetLastError()));
WASAPI_SET_ERROR(-11);
}
}
if(!m_hShutdownEvent){
if(!(m_hShutdownEvent = CreateEventEx(NULL, NULL, CREATE_EVENT_MANUAL_RESET, EVENT_ALL_ACCESS))){
tdav_win32_print_error("CreateEventEx(Shutdown)", HRESULT_FROM_WIN32(GetLastError()));
WASAPI_SET_ERROR(-12);
}
}
hr = m_pDevice->SetEventHandle(m_hCaptureEvent);
if (!SUCCEEDED(hr)){
tdav_win32_print_error("SetEventHandle", hr);
WASAPI_SET_ERROR(-13);
}
hr = m_pDevice->GetService(__uuidof(IAudioCaptureClient), (void**)&m_pClient);
if (!SUCCEEDED(hr)){
tdav_win32_print_error("GetService", hr);
WASAPI_SET_ERROR(-14);
}
int packetperbuffer = (1000 / codec->plugin->audio.ptime);
m_ring.chunck.size = wfx.nSamplesPerSec * (wfx.wBitsPerSample >> 3) / packetperbuffer;
TSK_DEBUG_INFO("#WASAPI: Audio producer ring chunk size = %u", m_ring.chunck.size);
// allocate our chunck buffer
if(!(m_ring.chunck.buffer = tsk_realloc(m_ring.chunck.buffer, m_ring.chunck.size))){
TSK_DEBUG_ERROR("Failed to allocate new buffer");
WASAPI_SET_ERROR(-15);
}
// create ringbuffer
m_ring.size = TDAV_WASAPI_PRODUCER_NOTIF_POS_COUNT * m_ring.chunck.size;
TSK_DEBUG_INFO("#WASAPI: Audio producer ring size = %u", m_ring.size);
if(!m_ring.buffer){
m_ring.buffer = speex_buffer_init(m_ring.size);
}
else {
int sret;
if((sret = speex_buffer_resize(m_ring.buffer, m_ring.size)) < 0){
TSK_DEBUG_ERROR("speex_buffer_resize(%d) failed with error code=%d", m_ring.size, sret);
WASAPI_SET_ERROR(-16);
}
}
if(!m_ring.buffer){
TSK_DEBUG_ERROR("Failed to create a new ring buffer with size = %d", m_ring.size);
WASAPI_SET_ERROR(-17);
}
m_callback.fn = TMEDIA_PRODUCER(wasapi)->enc_cb.callback;
m_callback.pcData = TMEDIA_PRODUCER(wasapi)->enc_cb.callback_data;
bail:
if (pwstrCaptureId){
CoTaskMemFree((LPVOID)pwstrCaptureId);
}
if(ret != 0){
UnPrepare();
}
m_bPrepared = (ret == 0);
tsk_mutex_unlock(m_hMutex);
return ret;
}
int Doubango::VoIP::AudioCapture::UnPrepare()
{
tsk_mutex_lock(m_hMutex);
if(m_hCaptureEvent)
{
CloseHandle(m_hCaptureEvent), m_hCaptureEvent = nullptr;
}
if(m_hShutdownEvent)
{
CloseHandle(m_hShutdownEvent), m_hShutdownEvent = nullptr;
}
if(m_pDevice)
{
m_pDevice->Release(), m_pDevice = nullptr;
}
if(m_pClient)
{
m_pClient->Release(), m_pClient = nullptr;
}
TSK_FREE(m_ring.chunck.buffer);
if(m_ring.buffer){
speex_buffer_destroy(m_ring.buffer);
m_ring.buffer = nullptr;
}
m_callback.fn = nullptr;
m_callback.pcData = nullptr;
m_bPrepared = false;
tsk_mutex_unlock(m_hMutex);
return 0;
}
int Doubango::VoIP::AudioCapture::Start()
{
tsk_mutex_lock(m_hMutex);
if(m_bStarted)
{
TSK_DEBUG_INFO("#WASAPI: Audio producer already started");
goto bail;
}
if(!m_bPrepared)
{
TSK_DEBUG_ERROR("Audio producer not prepared");
goto bail;
}
m_pAsyncThread = Windows::System::Threading::ThreadPool::RunAsync(ref new Windows::System::Threading::WorkItemHandler(this, &Doubango::VoIP::AudioCapture::AsyncThread),
Windows::System::Threading::WorkItemPriority::High,
Windows::System::Threading::WorkItemOptions::TimeSliced);
if((m_bStarted = (m_pAsyncThread != nullptr)))
{
HRESULT hr = m_pDevice->Start();
if(!SUCCEEDED(hr))
{
tdav_win32_print_error("Device::Start", hr);
Stop();
}
m_bPaused = false;
}
bail:
tsk_mutex_unlock(m_hMutex);
return (m_bStarted ? 0 : -2);
}
int Doubango::VoIP::AudioCapture::Stop()
{
m_bStarted = false;
tsk_mutex_lock(m_hMutex);
if (m_hShutdownEvent)
{
SetEvent(m_hShutdownEvent);
}
if (m_pAsyncThread)
{
m_pAsyncThread->Cancel();
m_pAsyncThread->Close();
m_pAsyncThread = nullptr;
}
if(m_pDevice)
{
m_pDevice->Stop();
}
// will be prepared again before next start()
UnPrepare();
tsk_mutex_unlock(m_hMutex);
return 0;
}
int Doubango::VoIP::AudioCapture::Pause()
{
tsk_mutex_lock(m_hMutex);
m_bPaused = true;
tsk_mutex_unlock(m_hMutex);
return 0;
}
void Doubango::VoIP::AudioCapture::AsyncThread(Windows::Foundation::IAsyncAction^ operation)
{
HRESULT hr = S_OK;
BYTE* pbData = nullptr;
UINT32 nFrames = 0;
DWORD dwFlags = 0;
UINT32 incomingBufferSize;
INT32 avail;
UINT32 nNextPacketSize;
HANDLE eventHandles[] = {
m_hCaptureEvent, // WAIT_OBJECT0
m_hShutdownEvent // WAIT_OBJECT1
};
TSK_DEBUG_INFO("#WASAPI: __record_thread -- START");
#define BREAK_WHILE tsk_mutex_unlock(m_hMutex); break;
while(m_bStarted && SUCCEEDED(hr)){
DWORD waitResult = WaitForMultipleObjectsEx(SIZEOF_ARRAY(eventHandles), eventHandles, FALSE, INFINITE, FALSE);
tsk_mutex_lock(m_hMutex);
if(!m_bStarted){
BREAK_WHILE;
}
if(waitResult == WAIT_OBJECT_0 && m_callback.fn) {
hr = m_pClient->GetNextPacketSize(&nNextPacketSize);
while(SUCCEEDED(hr) && nNextPacketSize >0){
hr = m_pClient->GetBuffer(&pbData, &nFrames, &dwFlags, nullptr, nullptr);
if(SUCCEEDED(hr) && pbData && nFrames){
if((dwFlags & AUDCLNT_BUFFERFLAGS_SILENT) != AUDCLNT_BUFFERFLAGS_SILENT){
incomingBufferSize = nFrames * m_nSourceFrameSizeInBytes;
speex_buffer_write(m_ring.buffer, pbData, incomingBufferSize);
avail = speex_buffer_get_available(m_ring.buffer);
while (m_bStarted && avail >= (INT32)m_ring.chunck.size) {
avail -= speex_buffer_read(m_ring.buffer, m_ring.chunck.buffer, m_ring.chunck.size);
m_callback.fn(m_callback.pcData, m_ring.chunck.buffer, m_ring.chunck.size);
}
}
if (SUCCEEDED(hr)){
hr = m_pClient->ReleaseBuffer(nFrames);
}
if (SUCCEEDED(hr)){
hr = m_pClient->GetNextPacketSize(&nNextPacketSize);
}
}
}
}
else if(waitResult != WAIT_OBJECT_0){
BREAK_WHILE;
}
tsk_mutex_unlock(m_hMutex);
}// end-of-while
if (!SUCCEEDED(hr)){
tdav_win32_print_error("AsyncThread: ", hr);
}
TSK_DEBUG_INFO("WASAPI: __record_thread(%s) -- STOP", SUCCEEDED(hr) ? "OK": "NOK");
}
//
// WaveAPI producer object definition
//
/* constructor */
static tsk_object_t* tdav_producer_wasapi_ctor(tsk_object_t * self, va_list * app)
{
tdav_producer_wasapi_t *wasapi = (tdav_producer_wasapi_t*)self;
if(wasapi){
/* init base */
tdav_producer_audio_init(TDAV_PRODUCER_AUDIO(wasapi));
/* init self */
wasapi->audioCapture = ref new Doubango::VoIP::AudioCapture();
}
return self;
}
/* destructor */
static tsk_object_t* tdav_producer_wasapi_dtor(tsk_object_t * self)
{
tdav_producer_wasapi_t *wasapi = (tdav_producer_wasapi_t*)self;
if(wasapi){
/* stop */
tdav_producer_wasapi_stop((tmedia_producer_t*)self);
/* deinit base */
tdav_producer_audio_deinit(TDAV_PRODUCER_AUDIO(wasapi));
/* deinit self */
if(wasapi->audioCapture){
delete wasapi->audioCapture;
wasapi->audioCapture = nullptr;
}
}
return self;
}
/* object definition */
static const tsk_object_def_t tdav_producer_wasapi_def_s =
{
sizeof(tdav_producer_wasapi_t),
tdav_producer_wasapi_ctor,
tdav_producer_wasapi_dtor,
tdav_producer_audio_cmp,
};
/* plugin definition*/
static const tmedia_producer_plugin_def_t tdav_producer_wasapi_plugin_def_s =
{
&tdav_producer_wasapi_def_s,
tmedia_audio,
"Microsoft Windows Audio Session API (WASAPI) producer",
tdav_producer_wasapi_set,
tdav_producer_wasapi_prepare,
tdav_producer_wasapi_start,
tdav_producer_wasapi_pause,
tdav_producer_wasapi_stop
};
const tmedia_producer_plugin_def_t *tdav_producer_wasapi_plugin_def_t = &tdav_producer_wasapi_plugin_def_s;
#endif /* HAVE_WASAPI */