/*
http://sdr-reu.wikispaces.com/file/view/SoundIn.c
File to read sound from sound card and record it to a file in PCM format.
*/
#include <windows.h>
#include <mmsystem.h>
#include <stdio.h>
#include <conio.h>
#pragma comment(lib, "winmm.lib")
#define IN_BUFFERS_NUMBER 4 //开辟三个数据缓冲区
#define REC_STATUS_STARTING_RECORD 1
#define REC_STATUS_RECORDING 2
#define REC_STATUS_ALL_DONE -2 //录音完成
#define REC_STATUS_FINISHING -1 //即将完成
#define REC_STATUS_FAIL -3
#define SIZE_OF_WAVEHDR 32 //wave头大小
//函数
void FreeBuffers(WAVEHDR * buffers);
DWORD WINAPI waveInProc(LPVOID threadControls);
typedef struct soundInDataAndControls
{
WAVEHDR buffers[IN_BUFFERS_NUMBER]; //开辟三个数据缓冲区
int recordingStatus; //录音状态
HWAVEIN inWaveDevice;
int buffers_left; //可用内存数目
}SoundThreadInfo;
WAVEFORMATEX sampleFormat; //全局变量
char * outfilename;
HMMIO m_hFile;
MMCKINFO m_MMCKInfoData;
MMCKINFO m_MMCKInfoParent;
MMCKINFO m_MMCKInfoChild;
int main()
{
MMRESULT errorResult = 0 ;
SoundThreadInfo soundIn;
HANDLE inWaveThreadHandle; //线程
int x = 0; //计数器
soundIn.recordingStatus = 0; //设置录音状态
soundIn.buffers_left = IN_BUFFERS_NUMBER; //设置预期缓冲数量
outfilename = ".\\1.wav";
inWaveThreadHandle = CreateThread(0 ,0,(LPTHREAD_START_ROUTINE)waveInProc,/*参数*/ &soundIn, 0, &errorResult); //创建线程
if(!inWaveThreadHandle)
{
printf("Can't create sound in thread! -- %08X\n",GetLastError());
return 0;
}
CloseHandle(inWaveThreadHandle); //关闭线程
for (x = 0; x<IN_BUFFERS_NUMBER; x++) //清空缓存
{
ZeroMemory(&soundIn.buffers[x], SIZE_OF_WAVEHDR);
}
//设置 waveFormat
sampleFormat.wFormatTag = WAVE_FORMAT_PCM;
sampleFormat.nChannels = 2; //声道
sampleFormat.nSamplesPerSec = 44100;
sampleFormat.wBitsPerSample = 16;
sampleFormat.nBlockAlign = sampleFormat.nChannels * (sampleFormat.wBitsPerSample >> 3);
sampleFormat.nAvgBytesPerSec = sampleFormat.nSamplesPerSec * sampleFormat.nBlockAlign;
if(!CreateWaveFile())
{
return getchar();
}
//分配 memory sound in buffers.
for(x = 0; x < IN_BUFFERS_NUMBER; x++) //this loop is not need after testing
{ //can be replaced with hard coded buffer allocation
//clear waveheaders
ZeroMemory(&soundIn.buffers[x],SIZE_OF_WAVEHDR);
soundIn.buffers[x].dwBufferLength = sampleFormat.nAvgBytesPerSec << 2; //内存大小估计一块
soundIn.buffers[x].dwFlags = 0;
soundIn.buffers[x].lpData = (char *) VirtualAlloc(NULL, soundIn.buffers[0].dwBufferLength, MEM_COMMIT, PAGE_READWRITE); //申请内存
if(soundIn.buffers[x].lpData == NULL)
{
printf("Sound In Buffer number %d failed to get allocate memory.\n",x);
for(--x;x;x--) // start 1 down, and while x!=0 x--
{
VirtualFree(soundIn.buffers[x].lpData, 0, MEM_RELEASE); //释放内存
}
return 0;
}
else
{
printf(" %d Buffer Application sucsess\n",x);
}
}
//open wave device
errorResult = waveInOpen(&soundIn.inWaveDevice, WAVE_MAPPER, &sampleFormat, (DWORD)errorResult, 0, CALLBACK_THREAD);
if(errorResult != MMSYSERR_NOERROR)
{
FreeBuffers(soundIn.buffers);
return getchar();
}
// buffers and queue them
for(x=0;x<IN_BUFFERS_NUMBER;x++)
{
errorResult = waveInPrepareHeader(soundIn.inWaveDevice, &soundIn.buffers[x],SIZE_OF_WAVEHDR);
if(errorResult != MMSYSERR_NOERROR)
{
printf("Failed to prepare header %d. Error %d\n",x,errorResult);
//fprintf(std,"Last was %s
FreeBuffers(soundIn.buffers);
errorResult= waveInClose(soundIn.inWaveDevice);
if(errorResult)
{
printf("Can't Close Wave Device!\n");
getchar();
}
return 0;
}
errorResult = waveInAddBuffer(soundIn.inWaveDevice, &soundIn.buffers[x],SIZE_OF_WAVEHDR);
if(errorResult != MMSYSERR_NOERROR )
{
printf("Failed to queue wave buffer %d.\n",x);
soundIn.buffers_left--;
if(soundIn.buffers_left < 2)
{
FreeBuffers(soundIn.buffers);
errorResult= waveInClose(soundIn.inWaveDevice);
if(errorResult)
{
printf("Can't Close Wave Device!\n");
getchar();
}
return 0;
}
}
}
//set record flag
printf("Hit any key to begin recording.\n");
getchar();
soundIn.recordingStatus = 1; //1 == starting recording
//start recording
errorResult = waveInStart(soundIn.inWaveDevice);
if(errorResult == MMSYSERR_NOERROR)
{
//wait for user to stop recording
printf("Hit any key to stop recording.\n");
getchar();
soundIn.recordingStatus = REC_STATUS_FINISHING;
}
else
{
//Recording never started.
soundIn.recordingStatus = REC_STATUS_ALL_DONE;
}
//reset headers
waveInReset(soundIn.inWaveDevice); //this should send a MM_WIM_DONE for each of the buffers/headers
//wait thread
while( (soundIn.recordingStatus != REC_STATUS_ALL_DONE) ) //status -2 is finished
{
Sleep(100); //tenth of a second
printf("Waiting for all done.\n");
printf("Recording status = %d\n",soundIn.recordingStatus);
}
//un prepare the buffers
for(x=0;x<IN_BUFFERS_NUMBER;x++)
{
errorResult = waveInUnprepareHeader(soundIn.inWaveDevice, &soundIn.buffers[x], SIZE_OF_WAVEHDR);
if(errorResult != 0)
{
printf("Failed to unprepare wave header.\n");
}
}
//deallocate memory
FreeBuffers(soundIn.buffers);
mmioAscend(m_hFile, &m_MMCKInfoChild, 0);
mmioAscend(m_hFile, &m_MMCKInfoParent, 0);
//close wave device
errorResult= waveInClose(soundIn.inWaveDevice);
if(errorResult)
{
printf("Can't Close Wave Device!\n");
getchar();
}
return getchar();
}
void FreeBuffers(WAVEHDR * buffers)
{
int x;
for(x=0;x<IN_BUFFERS_NUMBER;x++)
{
VirtualFree(buffers[x].lpData, 0, MEM_RELEASE); //zero meams all memory
}
}
int CreateWaveFile()
{
MMRESULT mmResult = 0;
ZeroMemory(&m_MMCKInfoParent,sizeof(MMCKINFO));
ZeroMemory(&m_MMCKInfoChild,sizeof(MMCKINFO));
ZeroMemory(&m_MMCKInfoData,sizeof(MMCKINFO));
m_hFile = mmioOpen(outfilename,NULL, MMIO_CREATE|MMIO_WRITE|MMIO_EXCLUSIVE | MMIO_ALLOCBUF);
if(m_hFile == NULL)
{
return getchar();
}
m_MMCKInfoParent.fccType = mmioFOURCC('W','A','V','E');
mmResult = mmioCreateChunk( m_hFile,&m_MMCKInfoParent, MMIO_CREATERIFF);
m_MMCKInfoChild.ckid = mmioFOURCC('f','m','t',' ');
m_MMCKInfoChild.cksize = sizeof(WAVEFORMATEX) + sampleFormat.cbSize;
mmResult = mmioCreateChunk(m_hFile, &m_MMCKInfoChild, 0);
mmResult = mmioWrite(m_hFile, (char*)&sampleFormat, sizeof(WAVEFORMATEX) + sampleFormat.cbSize);
mmResult = mmioAscend(m_hFile, &m_MMCKInfoChild, 0);
m_MMCKInfoChild.ckid = mmioFOURCC('d', 'a', 't', 'a');
mmResult = mmioCreateChunk(m_hFile, &m_MMCKInfoChild, 0);
return 1;
}
DWORD WINAPI waveInProc(void *threadControls)
{
MSG msg;
SoundThreadInfo *inFORMATION = (SoundThreadInfo *) threadControls;
MMRESULT errorResult;
int outputlenth = 0;
while(GetMessage(&msg,0,0,0)== 1) //得到一消息
{
if (msg.message == MM_WIM_DATA )
{
printf("Data amount %d\n",((WAVEHDR *)msg.lParam)->dwBytesRecorded);
//collect data
if(inFORMATION->recordingStatus != 0) //recording not began or stopped只有0 ，-1 -2 三个值
{
if(inFORMATION->recordingStatus == REC_STATUS_ALL_DONE)
{
printf("Possible error recieving WIM_DATA during condition ALL_DONE\n");
break; //time for thread to end
}
outputlenth = mmioWrite(m_hFile, ((WAVEHDR *) msg.lParam)->lpData,((WAVEHDR *)msg.lParam)->dwBytesRecorded);
if(outputlenth == ((WAVEHDR *)msg.lParam)->dwBytesRecorded)
{
printf("sucsess save data_dwBytesRecorded %d\n",outputlenth);
printf("sucsess save data_dwBufferLength %d\n",((WAVEHDR *)msg.lParam)->dwBufferLength);
}
//was this that last block the user wanted.
if(inFORMATION->recordingStatus == REC_STATUS_FINISHING)
{
//the buffers are each sent back, but not requeue
if(--inFORMATION->buffers_left == 0)
{
inFORMATION->recordingStatus = REC_STATUS_ALL_DONE;
}
printf("Buffer finished.\n");
continue;
}
}
//if stopped it only requeue s the buffer and continues.
//requeue buffer
if(!waveInAddBuffer(inFORMATION->inWaveDevice, (WAVEHDR *) msg.lParam, SIZE_OF_WAVEHDR))
{
if(!waveInAddBuffer(inFORMATION->inWaveDevice, (WAVEHDR *) msg.lParam, SIZE_OF_WAVEHDR))
{
printf("Buffer requeue error.\n");
if(--inFORMATION->buffers_left < 2)
{
FreeBuffers(inFORMATION->buffers);
printf("in requeue error device.");
errorResult = waveInClose(inFORMATION->inWaveDevice);
if(errorResult)
{
printf("Can't Close Wave Device!\n");
getchar();
}
return 0;
inFORMATION->recordingStatus = REC_STATUS_FINISHING;
//finishing means takes the queued blocks and no more
//in thoery the next block (the last one) that was queue
//previously
mmioAscend(m_hFile, &m_MMCKInfoChild, 0);
mmioAscend(m_hFile, &m_MMCKInfoParent, 0);
return -1;
}
}
}
}
}
return 1;
}
