waveformrecorddlg.cpp

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

// WaveFormRecordDlg.cpp : implementation file
//

#include "stdafx.h"
#include "WaveFormRecord.h"
#include "WaveFormRecordDlg.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;
DWORD size;
UINT wDeviceID;
PBYTE buf;
PBYTE oldptr;
HWAVEIN in;
BOOL m_bClose;
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()

/////////////////////////////////////////////////////////////////////////////
// CWaveFormRecordDlg dialog

CWaveFormRecordDlg::CWaveFormRecordDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CWaveFormRecordDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CWaveFormRecordDlg)
		// 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 CWaveFormRecordDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CWaveFormRecordDlg)
		// NOTE: the ClassWizard will add DDX and DDV calls here
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CWaveFormRecordDlg, CDialog)
	//{{AFX_MSG_MAP(CWaveFormRecordDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BTPATH, OnAddPath)
	ON_BN_CLICKED(IDC_PLAY, OnPlay)
	ON_BN_CLICKED(IDC_SAVE, OnSave)
	ON_MESSAGE(MM_WIM_DATA,OnData)
	ON_BN_CLICKED(IDC_RECORD, OnRecord)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CWaveFormRecordDlg message handlers

BOOL CWaveFormRecordDlg::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
	
	m_bClose=FALSE;
	GetDlgItem(IDC_PATH)->EnableWindow(false);
	GetDlgItem(IDC_RECORD)->EnableWindow(false);
	GetDlgItem(IDC_PLAY)->EnableWindow(false);
	GetDlgItem(IDC_SAVE)->EnableWindow(false);
	GetDlgItem(IDC_SAMPSIZE)->SetWindowText("8");
	GetDlgItem(IDC_TIME)->SetWindowText("6");
	GetDlgItem(IDC_FREQ)->SetWindowText("11052");
	mmioInstallIOProc( (FOURCC)mmioStringToFOURCC("WAV", 0),
										 (LPMMIOPROC)WaveIOProc, MMIO_INSTALLPROC);
	return TRUE;
}

void CWaveFormRecordDlg::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 CWaveFormRecordDlg::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 CWaveFormRecordDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}


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 CWaveFormRecordDlg::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 CWaveFormRecordDlg::OnRecord() 
{
	CString sampsize,freq,time;
	GetDlgItem(IDC_SAMPSIZE)->GetWindowText(sampsize);
	GetDlgItem(IDC_TIME)->GetWindowText(time);
	GetDlgItem(IDC_FREQ)->GetWindowText(freq);
	if(atoi(time)>60)
	{
		MessageBox("可以录制60秒以内的内容");
		GetDlgItem(IDC_TIME)->SetWindowText("6");
		return;
	}
	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)*atoi(time)*2;
	//初始化
	buf=(PBYTE)malloc(size);
	oldptr=buf;
	memset(buf,0,sizeof(BYTE));
	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;
	
	hdr=new WAVEHDR;
	hdr->dwBufferLength=size;
	hdr->lpData=(char*)buf;
	hdr->dwFlags=WHDR_BEGINLOOP | WHDR_ENDLOOP;
	hdr->dwLoops=1;
	hdr->lpNext=NULL;
	hdr->dwUser=0;
	hdr->dwBytesRecorded=0;
	hdr->reserved=0;

	waveInOpen(&in,0,&waveformat,(DWORD)this->m_hWnd,0,                   
                       CALLBACK_WINDOW);
	m_bClose=TRUE;
	waveInPrepareHeader(in,hdr,sizeof(WAVEHDR));
	waveInAddBuffer(in,hdr,sizeof(WAVEHDR));
	waveInStart(in);
	GetDlgItem(IDC_RECORD)->EnableWindow(false);

}

void CWaveFormRecordDlg::OnPlay() 
{
	HWAVEOUT out;
	WAVEFORMATEX waveformat;
	PWAVEHDR hdr;
	CString sampsize,freq,time;
	GetDlgItem(IDC_SAMPSIZE)->GetWindowText(sampsize);
	GetDlgItem(IDC_TIME)->GetWindowText(time);
	GetDlgItem(IDC_FREQ)->GetWindowText(freq);

	waveInStop(in);
	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;

	hdr=new WAVEHDR;
	hdr->dwBufferLength=size;
	hdr->lpData=(char*)oldptr;
	hdr->dwFlags=WHDR_BEGINLOOP|WHDR_ENDLOOP;
	hdr->dwLoops=1;
	hdr->lpNext=NULL;
	hdr->dwUser=0;
	hdr->dwBytesRecorded=0;
	hdr->reserved=0;
	MMRESULT result;
	result=waveOutOpen(&out,WAVE_MAPPER,&waveformat,(DWORD)this->m_hWnd,0,CALLBACK_WINDOW);

	waveOutPrepareHeader(out,hdr,sizeof(WAVEHDR));
	waveOutWrite(out,hdr,sizeof(WAVEHDR));
	
}
void CWaveFormRecordDlg::OnData()
{
	MessageBox("缓冲区已满");
	GetDlgItem(IDC_PLAY)->EnableWindow(true);
	GetDlgItem(IDC_SAVE)->EnableWindow(true);
	waveInStop(in);
}

void CWaveFormRecordDlg::OnSave() 
{
	CString tmp;

	HMMIO       hmmio; 
    MMCKINFO    ciRiffChunk; 
    MMCKINFO    ciSubChunk;
    MMIOINFO    mmioInfo;
	TCHAR file[255];
	GetDlgItem(IDC_PATH)->GetWindowText(tmp);
	if(m_bClose)
	{
		waveInStop(in);
		waveInClose(in);
		m_bClose=false;
	}
	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, 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 = size;//
    
	mmioCreateChunk(hmmio, &ciSubChunk, 0);
	mmioWrite(hmmio, (HPSTR)hdr->lpData,(LONG)hdr->dwBytesRecorded);
	mmioAscend(hmmio, &ciSubChunk, 0);
	mmioAscend(hmmio, &ciRiffChunk, 0);
	mmioFlush(hmmio, 0);
    mmioClose(hmmio, 0);
	if(m_bClose)
	{
		waveInStop(in);
		waveInClose(in);
		m_bClose=false;
	}
	GetDlgItem(IDC_PLAY)->EnableWindow(false);
	GetDlgItem(IDC_SAVE)->EnableWindow(false);
	MessageBox("已保存");
	
}

BOOL CWaveFormRecordDlg::DestroyWindow() 
{
	if(m_bClose)
	{
		waveInStop(in);
		waveInClose(in);
		m_bClose=false;
		free(buf);
	}
		waveInUnprepareHeader(in,hdr,sizeof(WAVEHDR));
		in=NULL;
	return CDialog::DestroyWindow();
}