waveformbufferdlg.cpp

来自「《Visual C++视频技术方案宝典》配套光盘」· C++ 代码 · 共 425 行

// WaveFormBufferDlg.cpp : implementation file
//

#include "stdafx.h"
#include "WaveFormBuffer.h"
#include "WaveFormBufferDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

#include <MMSystem.h>
#pragma comment(lib, "winmm.lib")

WAVEFORMATEX waveformat;
PWAVEHDR hdr[2];
DWORD size;
DWORD savesize;
UINT wDeviceID;
PBYTE buf[2];
LPSTR lptr;
HANDLE hData;
HWAVEIN in;
int nBuf=1;
int count=0;

LONG CALLBACK WaveIOProc(LPMMIOINFO lpmmioInfo, UINT uMsg, 
                         LPARAM lParam1, LPARAM lParam2);

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CWaveFormBufferDlg dialog

CWaveFormBufferDlg::CWaveFormBufferDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CWaveFormBufferDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CWaveFormBufferDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CWaveFormBufferDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CWaveFormBufferDlg)
		// NOTE: the ClassWizard will add DDX and DDV calls here
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CWaveFormBufferDlg, CDialog)
	//{{AFX_MSG_MAP(CWaveFormBufferDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_ADDPATH, OnAddpath)
	ON_BN_CLICKED(IDC_RECORD, OnRecord)
	ON_BN_CLICKED(IDC_STOP, OnStop)
	ON_MESSAGE(MM_WIM_DATA,OnData)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CWaveFormBufferDlg message handlers
FILE *f;
BOOL CWaveFormBufferDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon

	GetDlgItem(IDC_PATH)->EnableWindow(false);
	GetDlgItem(IDC_RECORD)->EnableWindow(false);
	GetDlgItem(IDC_STOP)->EnableWindow(false);

	GetDlgItem(IDC_SAMPSIZE)->SetWindowText("8");
	GetDlgItem(IDC_FREQ)->SetWindowText("11052");
	mmioInstallIOProc((FOURCC)mmioStringToFOURCC("WAV",0),
			(LPMMIOPROC)WaveIOProc, MMIO_INSTALLPROC);

	return TRUE;
}

void CWaveFormBufferDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CWaveFormBufferDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CWaveFormBufferDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CWaveFormBufferDlg::OnAddpath() 
{
	CFileDialog file(false,NULL,NULL,NULL,"文件(*.wav)|*.wav||");
	if(file.DoModal()==IDOK)
	{
		CString strname=file.GetPathName();
		strname+=".wav";
		GetDlgItem(IDC_PATH)->SetWindowText(strname);
		GetDlgItem(IDC_RECORD)->EnableWindow(true);
	}
}

void CWaveFormBufferDlg::OnRecord() 
{
	CString sampsize,freq;
	GetDlgItem(IDC_SAMPSIZE)->GetWindowText(sampsize);
	GetDlgItem(IDC_FREQ)->GetWindowText(freq);
	if(atoi(sampsize)!=8 && atoi(sampsize)!=16)
	{
		MessageBox("采样大小应为8或16");
		GetDlgItem(IDC_SAMPSIZE)->SetWindowText("8");
		return;
	}
	if(atoi(freq)>65500)
	{
		MessageBox("输入的数值太大");
		GetDlgItem(IDC_FREQ)->SetWindowText("11052");
		return;
	}
	size=atoi(freq);
	hData=GlobalAlloc(GMEM_MOVEABLE, size);
	lptr=(char*)GlobalLock(hData);

	waveformat.nChannels=2;
	waveformat.wFormatTag=WAVE_FORMAT_PCM;
	waveformat.cbSize=0;
	waveformat.wBitsPerSample=atoi(sampsize);
	waveformat.nSamplesPerSec=atoi(freq);
	waveformat.nBlockAlign=waveformat.nChannels*(waveformat.wBitsPerSample/8);
	waveformat.nAvgBytesPerSec=waveformat.nSamplesPerSec*waveformat.nBlockAlign;
	//初始化
	for(int i=0;i<=1;i++)
	{
		buf[i]=(PBYTE)malloc(size);
		memset(buf[i],0,sizeof(BYTE));
	
		hdr[i]=new WAVEHDR;
		hdr[i]->dwBufferLength=size;
		hdr[i]->lpData=(char*)buf[i];
		hdr[i]->dwFlags=WHDR_BEGINLOOP | WHDR_ENDLOOP;
		hdr[i]->dwLoops=1;
		hdr[i]->lpNext=NULL;
		hdr[i]->dwUser=0;
		hdr[i]->dwBytesRecorded=0;
		hdr[i]->reserved=0;
	}
	waveInOpen(&in,0,&waveformat,(DWORD)this->m_hWnd,0,                   
                       CALLBACK_WINDOW);
	MMRESULT result;
	for(int j=0;j<=1;j++)
	{
		result=waveInPrepareHeader(in,hdr[j],sizeof(WAVEHDR));
		if(result != MMSYSERR_NOERROR)
		{
			MessageBox("文件头准备失败");
			return;
		}
	}
	OnChange();
	GetDlgItem(IDC_RECORD)->EnableWindow(false);
	GetDlgItem(IDC_STOP)->EnableWindow(TRUE);
}

void CWaveFormBufferDlg::OnChange()
{
	waveInAddBuffer(in,hdr[count],sizeof(WAVEHDR));
	waveInStart(in);
}
LONG CALLBACK WaveIOProc(LPMMIOINFO lpmmioInfo, UINT uMsg, 
                         LPARAM lParam1, LPARAM lParam2)
{
	static int  file = 0;
           int  nStatus;
           LONG lStatus;
	switch(uMsg)
    {
		case MMIO_CREATE:
			file=_lcreat((LPSTR)lParam1, 0);
			if(file==-1)
				return(MMIOERR_CANNOTOPEN);
			return(0);
		case MMIOM_OPEN:
			file=_lopen((LPSTR)lParam1, MMIO_WRITE);
			if(file==-1)
			{
				file=_lcreat((LPSTR)lParam1, 0);
			}
			if(file == -1)
				return(MMIOERR_CANNOTOPEN);
			else
			{
				lpmmioInfo->lDiskOffset = 0;
				return(0);
			}

       case MMIOM_CLOSE:
              return(_lclose(file));
       case MMIOM_READ:
              nStatus=_lread(file, (LPSTR)lParam1, (int)lParam2);
              lpmmioInfo->lDiskOffset += (int)lParam2;
              return((LONG)nStatus);
       case MMIOM_WRITEFLUSH:
		   {
              nStatus=_lwrite(file, (LPSTR)lParam1, (int)lParam2);
              lpmmioInfo->lDiskOffset += (int)lParam2;
              return((LONG)nStatus);
		   }
       case MMIOM_SEEK:
              lStatus=_llseek(file, (LONG)lParam1, (int)lParam2);
              lpmmioInfo->lDiskOffset=lStatus;
              return(lStatus);
    }
	return(0);
}

void CWaveFormBufferDlg::OnStop() 
{
	waveInStop(in);
	waveInClose(in);
	//保存
	CString tmp;
	HMMIO       hmmio; 
    MMCKINFO    ciRiffChunk; 
    MMCKINFO    ciSubChunk;
    MMIOINFO    mmioInfo;
	TCHAR file[255];
	GetDlgItem(IDC_PATH)->GetWindowText(tmp);

	CString freq;
	GetDlgItem(IDC_FREQ)->GetWindowText(freq);

	mmioInfo.dwFlags     = 0;
    mmioInfo.fccIOProc   = mmioStringToFOURCC("WAV ", 0);
    mmioInfo.pIOProc     = (LPMMIOPROC)WaveIOProc;
    mmioInfo.wErrorRet   = 0;
    mmioInfo.htask       = 0;
    mmioInfo.cchBuffer   = 0;
    mmioInfo.pchBuffer   = 0;
    mmioInfo.pchNext     = 0;
    mmioInfo.pchEndRead  = 0;
    mmioInfo.pchEndWrite = 0;
    mmioInfo.lBufOffset  = 0;
    mmioInfo.lDiskOffset = 0;
    mmioInfo.adwInfo[0]  = 0;
    mmioInfo.adwInfo[1]  = 0;
    mmioInfo.adwInfo[2]  = 0;
    mmioInfo.adwInfo[3]  = 0;
    mmioInfo.dwReserved1 = 0;
    mmioInfo.dwReserved2 = 0;
    mmioInfo.hmmio       = 0;
	
	wsprintf(file,_T("%s"),tmp);

	hmmio = mmioOpen(file, &mmioInfo, 
                     MMIO_CREATE | MMIO_WRITE | MMIO_ALLOCBUF);
	
	mmioSetBuffer(hmmio, NULL, nBuf*size, 0); //设置大小

	mmioSeek(hmmio, 0, SEEK_SET);
	

    ciRiffChunk.fccType   = mmioFOURCC('W', 'A', 'V', 'E');
    ciRiffChunk.cksize    = 0L;
    
	mmioCreateChunk(hmmio, &ciRiffChunk, MMIO_CREATERIFF);
	ciSubChunk.ckid   = mmioStringToFOURCC("fmt", 0);
    ciSubChunk.cksize = sizeof(WAVEFORMATEX)-2;
    
	mmioCreateChunk(hmmio, &ciSubChunk, 0);
    mmioWrite(hmmio, (HPSTR)&waveformat, sizeof(WAVEFORMATEX));
	mmioAscend(hmmio, &ciSubChunk, 0);

	ciSubChunk.ckid   = mmioStringToFOURCC("data", 0);
    ciSubChunk.cksize = nBuf*size;
	mmioCreateChunk(hmmio, &ciSubChunk, 0);
	mmioWrite(hmmio, (HPSTR)lptr,(LONG)nBuf*size);

	mmioAscend(hmmio, &ciSubChunk, 0);
	mmioAscend(hmmio, &ciRiffChunk, 0);
	mmioFlush(hmmio, 0);
    mmioClose(hmmio, 0);
	
	MessageBox("已保存");
	GetDlgItem(IDC_STOP)->EnableWindow(false);
}

void CWaveFormBufferDlg::OnData()
{
	Onsave();
	count=1-count;
	OnChange();
}

void CWaveFormBufferDlg::Onsave()
{
	memcpy(lptr+(nBuf-1)*size,hdr[count]->lpData,size*sizeof(BYTE));
	GlobalUnlock(hData);
	hData = GlobalReAlloc(hData, (++nBuf)*size, GMEM_MOVEABLE);
	lptr=(char*)GlobalLock(hData);

}
BOOL CWaveFormBufferDlg::DestroyWindow() 
{
	free(buf[0]);
	free(buf[1]);
	GlobalLock(hData);
	GlobalFree(hData);
	waveInUnprepareHeader(in,hdr[0],sizeof(WAVEHDR));
	waveInUnprepareHeader(in,hdr[1],sizeof(WAVEHDR));
	in=NULL;
	waveInStop(in);
	waveInClose(in);
	return CDialog::DestroyWindow();
}