//Windows音频编程:Win32 Wave API 的使用

//下面是一个Win32控制台程序的源码，环境VS2008，新建一个空的Win32控制台项目，加入此文件编译即可，功能为录制5秒并回放。注意不要选择Unicode字符集，不然设备名会出现乱码。
//  
//  WaveAPI.cpp  
//  
// by: 知秋一叶  
//  
// 转载:http://blog.csdn.net/zhi_qiu_yi_ye/article/details/6663366

#include <windows.h>  
#include <stdio.h>  
#include <mmsystem.h>  
#pragma comment(lib, "winmm.lib")

#define BUFFER_SIZE (44100*16*2/8*5)    // 录制声音长度  
#define FRAGMENT_SIZE 1024              // 缓存区大小  
#define FRAGMENT_NUM 4                  // 缓存区个数

static unsigned char buffer[BUFFER_SIZE] = {0};  
static int buf_count = 0;

// 函数定义  
void CALLBACK waveInProc(HWAVEIN hwi,        
                        UINT uMsg,          
                        DWORD_PTR dwInstance,    
                        DWORD_PTR dwParam1,    
                        DWORD_PTR dwParam2     );  
void CALLBACK waveOutProc(  HWAVEOUT hwo,    
                         UINT uMsg,          
                         DWORD_PTR dwInstance,    
                         DWORD_PTR dwParam1,      
                         DWORD_PTR dwParam2     );

// 入口  
int main()  
{  
   /* 录音 */

// Device  
   int nReturn = waveInGetNumDevs();  
   printf("输入设备数目：%d\n", nReturn);  
   for (int i=0; i<nReturn; i++)  
   {  
       WAVEINCAPS wic;  
       waveInGetDevCaps(i, &wic, sizeof(WAVEINCAPS));  
       printf("#%d\t设备名：%s\n", i, wic.szPname);  
   }

// open  
   HWAVEIN hWaveIn;  
   WAVEFORMATEX wavform;  
   wavform.wFormatTag = WAVE_FORMAT_PCM;  
   wavform.nChannels = 2;  
   wavform.nSamplesPerSec = 44100;  
   wavform.nAvgBytesPerSec = 44100*16*2/8;  
   wavform.nBlockAlign = 4;  
   wavform.wBitsPerSample = 16;  
   wavform.cbSize = 0;

waveInOpen(&hWaveIn, WAVE_MAPPER, &wavform, (DWORD_PTR)waveInProc, 0, CALLBACK_FUNCTION);

WAVEINCAPS wic;  
   waveInGetDevCaps((UINT_PTR)hWaveIn, &wic, sizeof(WAVEINCAPS));  
   printf("打开的输入设备：%s\n", wic.szPname);

// prepare buffer  
   static WAVEHDR wh[FRAGMENT_NUM];  
   for (int i=0; i<FRAGMENT_NUM; i++)  
   {  
       wh[i].lpData = new char[FRAGMENT_SIZE];  
       wh[i].dwBufferLength = FRAGMENT_SIZE;  
       wh[i].dwBytesRecorded = 0;  
       wh[i].dwUser = NULL;  
       wh[i].dwFlags = 0;  
       wh[i].dwLoops = 1;  
       wh[i].lpNext = NULL;  
       wh[i].reserved = 0;

waveInPrepareHeader(hWaveIn, &wh[i], sizeof(WAVEHDR));  
       waveInAddBuffer(hWaveIn, &wh[i], sizeof(WAVEHDR));  
   }

// record  
   printf("Start to Record...\n");

buf_count = 0;  
   waveInStart(hWaveIn);

while (buf_count < BUFFER_SIZE)  
   {  
       Sleep(1);  
   }

printf("Record Over!\n\n");

// clean  
   waveInStop(hWaveIn);  
   waveInReset(hWaveIn);  
   for (int i=0; i<FRAGMENT_NUM; i++)  
   {  
       waveInUnprepareHeader(hWaveIn, &wh[i], sizeof(WAVEHDR));  
       delete wh[i].lpData;  
   }  
   waveInClose(hWaveIn);

system("pause");  
   printf("\n");

/* 放音 */

// Device  
   nReturn = waveOutGetNumDevs();  
   printf("\n输出设备数目：%d\n", nReturn);  
   for (int i=0; i<nReturn; i++)  
   {  
       WAVEOUTCAPS woc;  
       waveOutGetDevCaps(i, &woc, sizeof(WAVEOUTCAPS));  
       printf("#%d\t设备名：%s\n", i, wic.szPname);  
   }

// open  
   HWAVEOUT hWaveOut;  
   waveOutOpen(&hWaveOut, WAVE_MAPPER, &wavform, (DWORD_PTR)waveOutProc, 0, CALLBACK_FUNCTION);

WAVEOUTCAPS woc;  
   waveOutGetDevCaps((UINT_PTR)hWaveOut, &woc, sizeof(WAVEOUTCAPS));  
   printf("打开的输出设备：%s\n", wic.szPname);

// prepare buffer  
   WAVEHDR wavhdr;  
   wavhdr.lpData = (LPSTR)buffer;  
   wavhdr.dwBufferLength = BUFFER_SIZE;  
   wavhdr.dwFlags = 0;  
   wavhdr.dwLoops = 0;

waveOutPrepareHeader(hWaveOut, &wavhdr, sizeof(WAVEHDR));

// play  
   printf("Start to Play...\n");

buf_count = 0;  
   waveOutWrite(hWaveOut, &wavhdr, sizeof(WAVEHDR));  
   while (buf_count < BUFFER_SIZE)  
   {  
       Sleep(1);  
   }

// clean  
   waveOutReset(hWaveOut);  
   waveOutUnprepareHeader(hWaveOut, &wavhdr, sizeof(WAVEHDR));  
   waveOutClose(hWaveOut);

printf("Play Over!\n\n");

return 0;  
}

// 录音回调函数  
void CALLBACK waveInProc(HWAVEIN hwi,        
                        UINT uMsg,          
                        DWORD_PTR dwInstance,    
                        DWORD_PTR dwParam1,    
                        DWORD_PTR dwParam2     )  
{  
   LPWAVEHDR pwh = (LPWAVEHDR)dwParam1;

if ((WIM_DATA==uMsg) && (buf_count<BUFFER_SIZE))  
   {  
       int temp = BUFFER_SIZE - buf_count;  
       temp = (temp>pwh->dwBytesRecorded) ? pwh->dwBytesRecorded : temp;  
       memcpy(buffer+buf_count, pwh->lpData, temp);  
       buf_count += temp;

waveInAddBuffer(hwi, pwh, sizeof(WAVEHDR));  
   }  
}

// 放音回调函数  
void CALLBACK waveOutProc(  HWAVEOUT hwo,    
                         UINT uMsg,          
                         DWORD_PTR dwInstance,    
                         DWORD_PTR dwParam1,      
                         DWORD_PTR dwParam2     )  
{  
   if (WOM_DONE == uMsg)  
   {  
       buf_count = BUFFER_SIZE;  
   }  
}