支持選擇文件和目錄（可多選）的文件對話框CSelectDialog

MFC自帶的CFileDialog不支持選擇目錄，而且多選文件配置也不方便。有一些實現比較好的目錄選擇對話框，但是既可以支持選擇文件，又可以支持選擇目錄的非常難得。非常感謝Hojjat Bohlooli（[email protected]）的工作，給我們提供了一個非常方便的文件選擇對話框CSelectDialog。國人自己也能有些拿得出手的東西，而不是藏藏捏捏的。

其使用也是非常簡單的。

CSelectDialog ofd(TRUE, _T("*.*"), NULL, OFN_HIDEREADONLY | OFN_ALLOWMULTISELECT, _T("All files and folders(*.*)|*.*||") );

if( ofd.DoModal() != IDOK )
	return;

for( int i=0; i<ofd.m_SelectedItemList.GetCount(); i++ ){
	// 
}

詳細的實現如下：

#pragma once

#include <dlgs.h> // for (MULTI)FILEOPENORD

// CSelectDialog
class CSelectDialog : public CFileDialog
{
	DECLARE_DYNAMIC(CSelectDialog)

public:
	CSelectDialog(BOOL bOpenFileDialog, // TRUE for FileOpen, FALSE for FileSaveAs
		LPCTSTR lpszDefExt = NULL,
		LPCTSTR lpszFileName = NULL,
		DWORD dwFlags = OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT |
						OFN_EXPLORER & (~OFN_SHOWHELP),
		LPCTSTR lpszFilter = NULL,
		CWnd* pParentWnd = NULL);
	virtual ~CSelectDialog();
	
protected:
	virtual void OnInitDone();
	virtual void OnFolderChange();
	virtual BOOL OnFileNameOK();
	static LRESULT CALLBACK WindowProcNew(HWND hwnd,UINT message, WPARAM wParam, LPARAM lParam);
	DECLARE_MESSAGE_MAP()

public:
	static CString m_strCurrendDirectory;
	static CStringArray m_SelectedItemList;	/*this list includes files and folders
											are selected by user. */
	static WNDPROC m_wndProc;
};

#include "stdafx.h"
#include "SelectDialog.h"

#pragma warning( push )
#pragma warning( disable : 4311 4312 )
// CSelectDialog
CString CSelectDialog::m_strCurrendDirectory;
CStringArray CSelectDialog::m_SelectedItemList;
WNDPROC CSelectDialog::m_wndProc = NULL;

IMPLEMENT_DYNAMIC(CSelectDialog, CFileDialog)

CSelectDialog::CSelectDialog(BOOL bOpenFileDialog,
							 LPCTSTR lpszDefExt,
							 LPCTSTR lpszFileName,
							 DWORD dwFlags,
							 LPCTSTR lpszFilter,
							 CWnd* pParentWnd)
							 :CFileDialog(
							 bOpenFileDialog,
							 lpszDefExt,
							 lpszFileName,
							 dwFlags | OFN_EXPLORER | OFN_HIDEREADONLY & (~OFN_SHOWHELP),
							 lpszFilter,
							 pParentWnd)
{
	dwFlags |= (OFN_EXPLORER | OFN_HIDEREADONLY & (~OFN_SHOWHELP));
};

CSelectDialog::~CSelectDialog()
{
};

BEGIN_MESSAGE_MAP(CSelectDialog, CFileDialog)
END_MESSAGE_MAP()

// CSelectDialog message handlers
BOOL CSelectDialog::OnFileNameOK()
{
	if (CFileDialog* pDlg = (CFileDialog*)CWnd::FromHandle(GetParent()->m_hWnd))
	{
		CWnd* pWnd = pDlg->GetDlgItem(lst2);	//getting list
		if (pWnd == NULL)
			return FALSE;

		m_SelectedItemList.RemoveAll();			// emptying list
		
		CListCtrl* wndLst1 = (CListCtrl*)(pWnd->GetDlgItem(1));

		int nSelected = wndLst1->GetSelectedCount();
		if (!nSelected)		// nothing selected -- don't retrieve list
			return FALSE;
		CString strItemText, strDirectory = m_strCurrendDirectory;
		if (strDirectory.Right(1) != _T("\\"))
			strDirectory += _T("\\");

		CString fileslist = _T("");
		pDlg->SendMessage(CDM_GETSPEC, (WPARAM)MAX_PATH,
			(LPARAM)fileslist.GetBuffer(MAX_PATH));
		fileslist.ReleaseBuffer();

		strItemText = strDirectory + fileslist;
		if(nSelected == 1 && fileslist != _T(""))
		{
			m_SelectedItemList.Add(strItemText);
			return CFileDialog::OnFileNameOK();
		}
	}
	::MessageBeep( MB_ICONQUESTION );
	return 1; //don't let the dialog to close
};

void CSelectDialog::OnFolderChange()
{
	m_strCurrendDirectory = GetFolderPath();
	CFileDialog::OnFolderChange();
};

void CSelectDialog::OnInitDone()
{
	m_strCurrendDirectory = GetFolderPath();
	CWnd* pFD = GetParent();

	HideControl(edt1);
	HideControl(cmb1);
	HideControl(stc2);

	//HideControl(cmb13);
	//HideControl(stc3);

	CRect rectCancel; pFD->GetDlgItem(IDCANCEL)->GetWindowRect(&rectCancel);
	pFD->ScreenToClient(&rectCancel);

	CRect rectOK; pFD->GetDlgItem(IDOK)->GetWindowRect(&rectOK);
	pFD->ScreenToClient(&rectOK);
	pFD->GetDlgItem(IDOK)->SetWindowPos(0,rectCancel.left - rectOK.Width() - 5, rectCancel.top, 0,0, SWP_NOZORDER | SWP_NOSIZE);

	CRect rectList2; pFD->GetDlgItem(lst1)->GetWindowRect(&rectList2);
	pFD->ScreenToClient(&rectList2);
	pFD->GetDlgItem(lst1)->SetWindowPos(0,0,0,rectList2.Width(), abs(rectList2.top - (rectCancel.top - 5)), SWP_NOMOVE | SWP_NOZORDER);

	CRect rectStatic;pFD->GetDlgItem(stc3)->GetWindowRect(&rectStatic);
	pFD->ScreenToClient(&rectStatic);
	pFD->GetDlgItem(stc3)->SetWindowPos(0,rectCancel.left - 375,rectCancel.top + 5, rectStatic.Width(), rectStatic.Height(), SWP_NOZORDER);

	CRect rectEdit1;pFD->GetDlgItem(cmb13)->GetWindowRect(&rectEdit1);
	pFD->ScreenToClient(&rectEdit1);
	pFD->GetDlgItem(cmb13)->SetWindowPos(0,rectCancel.left - 320,rectCancel.top, rectEdit1.Width() - 15, rectEdit1.Height(), SWP_NOZORDER);

	SetControlText(stc3, _T("Item name:"));
	SetControlText(IDOK, _T("Select"));

	m_wndProc = (WNDPROC)::SetWindowLong(pFD->m_hWnd, GWL_WNDPROC, (long)WindowProcNew);
	pFD->CenterWindow();
};

LRESULT CALLBACK CSelectDialog::WindowProcNew(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
{
	if (message ==  WM_COMMAND)
	{
		if (HIWORD(wParam) == BN_CLICKED)
		{
			if (LOWORD(wParam) == IDOK)
			{
				if (CFileDialog* pDlg = (CFileDialog*)CWnd::FromHandle(hwnd))
				{
					m_SelectedItemList.RemoveAll();			// emptying list
					CWnd* pWnd = pDlg->GetDlgItem(lst2);	//getting list
					if (pWnd == NULL)
						return FALSE;

					CListCtrl* wndLst1 = (CListCtrl*)(pWnd->GetDlgItem(1));

					int nSelected = wndLst1->GetSelectedCount();
					if (!nSelected)		// nothing selected -- don't retrieve list
						return FALSE;
					CString strItemText, strDirectory = m_strCurrendDirectory;
					if (strDirectory.Right(1) != _T("\\"))
						strDirectory += _T("\\");

					int nItem = wndLst1->GetNextItem(-1,LVNI_SELECTED);
					CString fileslist = _T("");
					pDlg->SendMessage(CDM_GETSPEC, (WPARAM)MAX_PATH,
						(LPARAM)fileslist.GetBuffer(MAX_PATH));
					fileslist.ReleaseBuffer();
					//////////////////   Add directory names to list
					while((nSelected--) > 0)
					{
						strItemText = wndLst1->GetItemText(nItem,0);
						strItemText = strDirectory + strItemText;
						DWORD attr = GetFileAttributes(strItemText);
						if((attr != 0xFFFFFFFF) && (attr & FILE_ATTRIBUTE_DIRECTORY))
							m_SelectedItemList.Add(strItemText);							
						nItem = wndLst1->GetNextItem(nItem, LVNI_SELECTED);
					}
					//////////////////   Add FILE names to list
					strItemText = _T("");
					nSelected = wndLst1->GetSelectedCount();
					if(nSelected > m_SelectedItemList.GetCount())
					{
						int MoreThanOnFile = fileslist.Find(_T("\""));
						if(MoreThanOnFile != -1)
						{
							for(int i=0; i<fileslist.GetLength(); i++)
								if(fileslist[i] != '\"')
								{
									strItemText.AppendFormat(_T("%c"),fileslist[i]);
									if(fileslist[i-1] == '\"' && fileslist[i] == ' ')
										strItemText.Delete(strItemText.GetLength()-1);
								}
								else if(!strItemText.IsEmpty())
								{
									m_SelectedItemList.Add((strDirectory+strItemText));
									strItemText.Empty();
								}
						}
						else
							m_SelectedItemList.Add(strDirectory+fileslist);
					}
					pDlg->EndDialog(IDOK);
					return NULL;
				} // if IDOK
			}
		} // if BN_CLICKED
	}// if WM_COMMAND
	return CallWindowProc(m_wndProc, hwnd, message, wParam, lParam);
}

#pragma warning( pop )