splitfiledlg.cpp

VC.net 的一些有用的技巧

// SplitFileDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "SplitFile.h"
#include "SplitFileDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

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

// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

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

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

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// CSplitFileDlg 对话框



CSplitFileDlg::CSplitFileDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CSplitFileDlg::IDD, pParent)
	, m_path(_T(""))
	, Size(1.2)
	, m_targetpath(_T(""))
	, m_filename(_T(""))
	, m_parts(_T("单个文件大小"))
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CSplitFileDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_EDIT, m_edit);
	DDX_Text(pDX, IDC_SPLITSIZE, Size);
	DDX_Text(pDX, IDC_EDIT, m_path);
	DDX_Control(pDX, IDC_PROGRESS, pProgress);
	DDX_Text(pDX, IDC_PARTS, m_parts);
}

BEGIN_MESSAGE_MAP(CSplitFileDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_BROWSEOPEN, OnBrowseopen)
	ON_BN_CLICKED(ID_SPLITFILE, OnSplitfile)
END_MESSAGE_MAP()


// CSplitFileDlg 消息处理程序

BOOL CSplitFileDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 将\“关于...\”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	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);
		}
	}

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO：在此添加额外的初始化代码

	return TRUE;  // 除非设置了控件的焦点，否则返回 TRUE
}

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

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CSplitFileDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作矩形中居中
		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;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
HCURSOR CSplitFileDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CSplitFileDlg::OnBrowseopen()
{
	// TODO: 在此添加控件通知处理程序代码
	CString filter=_T("待分割文件(*.*)|*.*||");
	CFileDialog dlg(TRUE,
					NULL, 
					NULL, 
					OFN_HIDEREADONLY	| 
					OFN_OVERWRITEPROMPT | 
					OFN_FILEMUSTEXIST,
					filter,
					0);
	//获取源文件路径、名称及所在文件夹
	if (dlg.DoModal() == IDOK) {
		m_path = dlg.GetPathName();
		m_filename = dlg.GetFileName();
		int r_pos=m_path.ReverseFind('\\');
		m_targetpath=m_path.Left(r_pos);
		UpdateData(FALSE);
	}
}

void CSplitFileDlg::OnSplitfile()
{
	// TODO: 在此添加控件通知处理程序代码
	CWaitCursor wait;
	CFile destFile;
	// 构造文件异常对象以获取错误信息
	CFileException ex;

	DWORD dwRead;
	UINT newlen;
	char buff [20];
	char b [20];
	long l = 1;
	CString name;
	UINT len = 0;

	UpdateData(TRUE);

	//获取文件分割后的大小，定义相对应变量数值
    newlen=(UINT)(Size*1024*1024);
	UINT nCount = newlen/10;
    BYTE buffer[140000];

	//打开文件读取
	if (!m_path.IsEmpty()) 
	{
		if (!m_SourceFile.Open(m_path, CFile::modeRead | CFile::shareDenyNone | CFile::typeBinary, &ex)) 
		{
			TCHAR szError[1024];
			ex.GetErrorMessage(szError, 1024);
      		::AfxMessageBox(szError);
			m_edit.SetFocus();
			m_edit.SetSel(0, -1);
			return;
		}
		//获取文件大小
		len = m_SourceFile.GetLength();
		//设置进度条范围
		pProgress.SetRange(0,len/newlen);
	}
	else 
	{
		MessageBox(_T("请选择待分割的源文件."), _T("文件分割器"), MB_ICONEXCLAMATION);
		return;
	}
	//检查文件大小是否符合分割要求
	if (len < newlen)
	{
		CString length = _itoa(len, b, 10);
		MessageBox(_T("文件长度为 " + length + " 字节，不够指定的分割大小， 没有必要再进行分割."), _T("文件分割器"), MB_ICONEXCLAMATION);
		m_SourceFile.Close();
		m_path = _T("");
		m_filename = _T("");
		UpdateData(FALSE);
		return;
	}

	//分割文件
	do {
		//通过加前缀动态生成分隔小文件的文件名
		name = _ltoa(l, buff, 10);
		name += _T("_");
		CString newpath; 

		//判断选择目录未尾是否已有"\"符
		if(m_targetpath.Right(1)=='\\')
			newpath = m_targetpath;
		else
			newpath = m_targetpath + _T("\\");
		if (!destFile.Open(newpath + name + m_SourceFile.GetFileName(),  
			CFile::modeWrite		| 
			CFile::shareExclusive	| 
			CFile::typeBinary		|
			CFile::modeCreate, &ex)) 
		{
			TCHAR szError[1024];
			ex.GetErrorMessage(szError, 1024);
			::AfxMessageBox(szError);
			m_SourceFile.Close();
			return;
		}
		do
		{
			dwRead = m_SourceFile.Read(buffer, nCount);
			destFile.Write(buffer, dwRead);
		}
		while (dwRead > 0 && destFile.GetLength() < newlen); 
		destFile.Close();
	
		pProgress.SetPos(l);
		m_parts = _ltoa(l , buff, 10);
		m_parts += _T("个文件生成");
		UpdateData(FALSE);
		l++;
		UpdateWindow();
	}
	while (dwRead > 0);
	m_SourceFile.Close();

	pProgress.SetPos(len/newlen);
	MessageBox(_T("分割文件完成！"), _T("文件分割器"), MB_ICONEXCLAMATION);
	m_path = _T("");
	m_filename = _T("");
    pProgress.SetPos(0);
    m_parts = _T("单个文件大小");
	UpdateData(FALSE);
	return;
}