cdsnatchdlg.cpp

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

// CDSnatchDlg.cpp : implementation file
//

#include "stdafx.h"
#include "CDSnatch.h"
#include "CDSnatchDlg.h"

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

#define  NSECTORS  13
#define  UNDERSAMPLING 1
#define  CB_CDDASECTOR 2368
#define  CB_QSUBCHANNEL 16
#define  CB_CDROMSECTOR 2048
#define  CB_AUDIO  (CB_CDDASECTOR-CB_QSUBCHANNEL)

#include "winioctl.h"
#include "ntddcdrm.h"

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

UINT m_wDeviceID;
CDROM_TOC CdromTOC;
HANDLE m_hDevice;
LPSTR lptr;
HANDLE hData;
WAVEFORMATEX waveformat;
/////////////////////////////////////////////////////////////////////////////
// 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()

/////////////////////////////////////////////////////////////////////////////
// CCDSnatchDlg dialog

CCDSnatchDlg::CCDSnatchDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CCDSnatchDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CCDSnatchDlg)
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CCDSnatchDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CCDSnatchDlg)
	DDX_Control(pDX, IDC_CD, m_cd);
	DDX_Control(pDX, IDC_CDLIST, m_cdlist);
	DDX_Control(pDX, IDC_SNATCHPRO, m_snatchpro);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CCDSnatchDlg, CDialog)
	//{{AFX_MSG_MAP(CCDSnatchDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_OPEN, OnOpen)
	ON_BN_CLICKED(IDC_SNATCH, OnSnatch)
	ON_BN_CLICKED(IDC_EXIT, OnExit)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CCDSnatchDlg message handlers

BOOL CCDSnatchDlg::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_snatchpro.SetRange(0,100);
	//获取光驱数量
	for(int i=0; i<26; i++)
	{
		char DrvStr[]="A:\\";
		DrvStr[0] = i + 'A';
		if(GetDriveType(DrvStr) == DRIVE_CDROM)
			m_cd.AddString(DrvStr);
	}
	//初始化轨道列表
	m_cdlist.SetExtendedStyle(LVS_EX_GRIDLINES);
	m_cdlist.InsertColumn(0,"曲目",LVCFMT_LEFT,50);
	m_cdlist.InsertColumn(1,"起始扇区号",LVCFMT_LEFT,130);
	m_cdlist.InsertColumn(2,"结束扇区号",LVCFMT_LEFT,130);
	return TRUE;
}

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

void CCDSnatchDlg::OnOpen() 
{
	CString FileName;
	DWORD trackstart;
	DWORD trackend;
	CString tmp;
	int index=m_cd.GetCurSel();
	if(index>=0)
	{
		m_cd.GetLBText(index,FileName);
		FileName=FileName.Left(2);

		m_hDevice =CreateFile("\\\\.\\"+FileName,GENERIC_READ,
		FILE_SHARE_READ | FILE_SHARE_WRITE,	NULL,OPEN_EXISTING,
		0, NULL);
	
		BOOL bResult;
		DWORD dwOutBytes;
		bResult=DeviceIoControl(m_hDevice,IOCTL_CDROM_READ_TOC,NULL,
			0,&CdromTOC,sizeof(CdromTOC),&dwOutBytes, 
			(LPOVERLAPPED)NULL);
	}
	int cdnum=CdromTOC.LastTrack;
	//获得曲目数量
	for(int i=1;i<=cdnum;i++)
	{
		CString cdstr;
		cdstr.Format("%d",i);
		m_cdlist.InsertItem(i-1,"");
		//获得音轨开始地址
		trackstart=(CdromTOC.TrackData[i-1].Address[1]*60*75 + 
			CdromTOC.TrackData[i-1].Address[2]*75 +
			CdromTOC.TrackData[i-1].Address[3])-150;
		//获得音轨结束地址
		trackend=(CdromTOC.TrackData[i].Address[1]*60*75 + 
			CdromTOC.TrackData[i].Address[2]*75 + 
			CdromTOC.TrackData[i].Address[3])-151;
		m_cdlist.SetItemText(i-1,0,cdstr);
		tmp.Format("%d",trackstart);
		m_cdlist.SetItemText(i-1,1,tmp);
		tmp.Format("%d",trackend);
		m_cdlist.SetItemText(i-1,2,tmp);
	}
}

void CCDSnatchDlg::OnSnatch() 
{
	int track;
	CString tmp;
	TCHAR filename[255];
	int index=m_cdlist.GetSelectionMark();
	if(index>=0)
	{
		tmp=m_cdlist.GetItemText(index,0);
		track=atoi(tmp);
	}
	else
	return;

	DWORD trackstart=(CdromTOC.TrackData[track-1].Address[1]*60*75 + 
	CdromTOC.TrackData[track-1].Address[2]*75 +
	CdromTOC.TrackData[track-1].Address[3])-150;
	//获得音轨结束地址
	DWORD trackend=(CdromTOC.TrackData[track].Address[1]*60*75 + 
	CdromTOC.TrackData[track].Address[2]*75 + 
	CdromTOC.TrackData[track].Address[3])-151;
	//根据音轨所有的扇区计算出音频数据的大小，并分配内存
	hData=GlobalAlloc(GMEM_MOVEABLE,(trackend-trackstart)*CB_AUDIO);
	lptr=(char*)GlobalLock(hData);
	//设置一次读取的缓存
	BYTE buf[CB_AUDIO*NSECTORS];
	//步进为NSECTORS的循环
	int pos=0;
	int poslen=(trackend-trackstart)/100;
	int count=0;
	int i=0;
	for(int sector=trackstart;(sector<trackend);sector+=NSECTORS)
	{
		//当剩余sector不足NSECTORS时
		int read=((sector+NSECTORS)<trackend)?NSECTORS:(trackend-sector);

		DWORD dwOutBytes;
		RAW_READ_INFO rri; 
		rri.TrackMode =(TRACK_MODE_TYPE)2;
		rri.SectorCount = (DWORD)read;
		rri.DiskOffset.QuadPart =(DWORD64)(sector*CB_CDROMSECTOR);
		if(DeviceIoControl(m_hDevice,IOCTL_CDROM_RAW_READ,&rri,sizeof(rri),
		buf,(DWORD)read*CB_AUDIO,&dwOutBytes,(LPOVERLAPPED)NULL)) 
		
		memcpy(lptr+CB_AUDIO*read*(i++),buf,CB_AUDIO*read);
		
		memset(buf,0,sizeof(BYTE));
		//计算滚动条位置
		count+=NSECTORS;
		if(count>poslen)
		{
		pos++;
		m_snatchpro.SetPos(pos);
		count=0;
		}
	}
	
	waveformat.nChannels=2;
	waveformat.wFormatTag=WAVE_FORMAT_PCM;
	waveformat.cbSize=0;
	waveformat.nSamplesPerSec=44100;
	waveformat.wBitsPerSample=16;
	waveformat.nBlockAlign=waveformat.nChannels*(waveformat.wBitsPerSample/8);
	waveformat.nAvgBytesPerSec=waveformat.nSamplesPerSec*waveformat.nBlockAlign;
	HMMIO       hmmio; 
    MMCKINFO    ciRiffChunk; 
    MMCKINFO    ciSubChunk;
    MMIOINFO    mmioInfo;

	memset(&mmioInfo,0,sizeof(mmioInfo));
    mmioInfo.fccIOProc   = mmioStringToFOURCC("WAV ", 0);
    mmioInfo.pIOProc     = NULL;

	wsprintf(filename,_T("trace%d.wav"),track);
	hmmio = mmioOpen(filename, &mmioInfo, 
                     MMIO_CREATE | MMIO_WRITE | MMIO_ALLOCBUF);

	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);
	//将waveformat.cbSize去除
    ciSubChunk.cksize = sizeof(WAVEFORMATEX)-2;
    
	mmioCreateChunk(hmmio, &ciSubChunk, 0);
	mmioWrite(hmmio, (HPSTR)&waveformat, sizeof(WAVEFORMATEX)-2);
	mmioAscend(hmmio, &ciSubChunk, 0);
	ciSubChunk.ckid   = mmioStringToFOURCC("data", 0);
    ciSubChunk.cksize = (trackend-trackstart)*CB_AUDIO;

	mmioCreateChunk(hmmio, &ciSubChunk, 0);
	mmioWrite(hmmio, (HPSTR)lptr,(LONG)(trackend-trackstart)*CB_AUDIO);
	mmioAscend(hmmio, &ciSubChunk, 0);
	mmioAscend(hmmio, &ciRiffChunk, 0);
	mmioFlush(hmmio, 0);
    mmioClose(hmmio, 0);
	MessageBox("抓取完成");
}

BOOL CCDSnatchDlg::DestroyWindow() 
{
	GlobalLock(hData);
	GlobalFree(hData);
	return CDialog::DestroyWindow();
}

void CCDSnatchDlg::OnExit() 
{
	OnCancel();	
}