文件夹浏览对话框类

使用方法：

1，包含头文件：#include "pathdialog.h"；

2，使用CPathDialog pathDlg(”标题“, ”说明“, ”默认路径“, 父窗口指针)进行初始，并调用pathDlg.DoModal()显示对话框;

 

例子：

 

 CString strTemp = _T("日志目录");
 CString strTemp1 = _T("请选择或输入存放日志的目录");

 GetDlgItem(IDC_EDIT_LOGPATH)->GetWindowText(m_strLogPath);

 CPathDialog pathDlg(strTemp, strTemp1, m_strLogPath, this);

 if (pathDlg.DoModal() == IDOK)
 {
  m_strLogPath = pathDlg.GetPathName();
  GetDlgItem(IDC_EDIT_LOGPATH)->SetWindowText(m_strLogPath);
 } 

 

类源代码：

//////////////////////////////////////////////////////////////////////////
//PathDialog.h file
//
//Written by Nguyen Tan Hung <[e-mail]>
//////////////////////////////////////////////////////////////////////////

#if !defined(AFX_PATHDIALOG_H__0F70BC86_11DB_11D4_B012_0000E8DD8DAA__INCLUDED_)
#define AFX_PATHDIALOG_H__0F70BC86_11DB_11D4_B012_0000E8DD8DAA__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
// PathDialog.h : header file
//
#include "shlobj.h"

class CPathDialog;

// CPathDialogSub - intercepts messages from child controls
class CPathDialogSub : public CWnd
{
 friend CPathDialog;
public:
 CPathDialog* m_pPathDialog;
protected:
    afx_msg void OnOK();              // OK button clicked
 afx_msg void OnChangeEditPath();
    DECLARE_MESSAGE_MAP()
private:
};

/////////////////////////////////////////////////////////////////////////////
// CPathDialog dialog

class CPathDialog
{
 friend CPathDialogSub;
// Construction
public:
 CPathDialog(LPCTSTR lpszCaption=NULL,
  LPCTSTR lpszTitle=NULL,
  LPCTSTR lpszInitialPath=NULL,
  CWnd* pParent = NULL);
 ~CPathDialog();

 CString GetPathName();
 virtual int DoModal();

 static Touch(LPCTSTR lpPath, BOOL bValidate=TRUE);
 static int MakeSurePathExists(LPCTSTR lpPath);
 static BOOL IsFileNameValid(LPCTSTR lpFileName);
 static int ConcatPath(LPTSTR lpRoot, LPCTSTR lpMorePath);

private:
 static int CALLBACK BrowseCallbackProc(HWND hwnd,UINT uMsg,LPARAM lParam, LPARAM pData);

 LPCTSTR m_lpszCaption;
 LPTSTR m_lpszInitialPath;

 TCHAR m_szPathName[MAX_PATH];

 BROWSEINFO m_bi;
 HWND m_hWnd;
 CWnd* m_pParentWnd;
 BOOL m_bParentDisabled;
 BOOL m_bGetSuccess;

 CPathDialogSub m_PathDialogSub;

};

//{{AFX_INSERT_LOCATION}}
// Microsoft Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_PATHDIALOG_H__0F70BC86_11DB_11D4_B012_0000E8DD8DAA__INCLUDED_)

 

//////////////////////////////////////////////////////////////////////////
//PathDialog.h file
//
//Written by Nguyen Tan Hung <[e-mail]>
//////////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "PathDialog.h"
#include <io.h>

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

#define IDC_FOLDERTREE  0x3741
#define IDC_TITLE   0x3742
#define IDC_STATUSTEXT  0x3743

#define IDC_NEW_EDIT_PATH 0x3744

// Class CDlgWnd
BEGIN_MESSAGE_MAP(CPathDialogSub, CWnd)
        ON_BN_CLICKED(IDOK, OnOK)
  ON_EN_CHANGE(IDC_NEW_EDIT_PATH, OnChangeEditPath)
END_MESSAGE_MAP()

void CPathDialogSub::OnOK()
{
 ::GetWindowText(::GetDlgItem(m_hWnd, IDC_NEW_EDIT_PATH),
  m_pPathDialog->m_szPathName, MAX_PATH);

 if(CPathDialog::MakeSurePathExists(m_pPathDialog->m_szPathName)==0)
 {
  m_pPathDialog->m_bGetSuccess=TRUE;
  ::EndDialog(m_pPathDialog->m_hWnd, IDOK);
 }
 else
 {
  ::SetFocus(::GetDlgItem(m_hWnd, IDC_NEW_EDIT_PATH));
 }
}

void CPathDialogSub::OnChangeEditPath()
{
 ::GetWindowText(::GetDlgItem(m_hWnd, IDC_NEW_EDIT_PATH),
  m_pPathDialog->m_szPathName, MAX_PATH);
 BOOL bEnableOKButton = (_tcslen(m_pPathDialog->m_szPathName)>0);
 SendMessage(BFFM_ENABLEOK, 0, bEnableOKButton);
}
/////////////////////////////////////////////////////////////////////////////
// CPathDialog dialog

CPathDialog::CPathDialog(LPCTSTR lpszCaption,
       LPCTSTR lpszTitle,
       LPCTSTR lpszInitialPath,
       CWnd* pParent)
{
 m_hWnd=NULL;
 m_PathDialogSub.m_pPathDialog= this;
    m_bParentDisabled = FALSE;

    // Get the true parent of the dialog
    m_pParentWnd = CWnd::GetSafeOwner(pParent);

 m_lpszCaption = lpszCaption;
 size_t nPathLen = 0;
 if (lpszInitialPath != NULL)
  nPathLen = _tcslen(lpszInitialPath);
 if (nPathLen > 0)
 {
  m_lpszInitialPath = new TCHAR[nPathLen + 1];
  _tcscpy(m_lpszInitialPath, lpszInitialPath);
  if (m_lpszInitialPath[nPathLen - 1] == _T('//'))
   m_lpszInitialPath[nPathLen - 1] = _T('/0');

 }
 else
  m_lpszInitialPath = NULL;
 

 memset(&m_bi, 0, sizeof(BROWSEINFO) );
 m_bi.hwndOwner = (m_pParentWnd==NULL) ? NULL : m_pParentWnd->GetSafeHwnd();
 m_bi.pszDisplayName = 0;
 m_bi.pidlRoot = 0;
 m_bi.ulFlags = BIF_RETURNONLYFSDIRS | BIF_STATUSTEXT;
 m_bi.lpfn = BrowseCallbackProc;
 m_bi.lpszTitle = lpszTitle;
}

CPathDialog::~CPathDialog()
{
 if (m_lpszInitialPath != NULL)
  delete []m_lpszInitialPath;
}
/////////////////////////////////////////////////////////////////////////////
// CPathDialog message handlers

CString CPathDialog::GetPathName()
{
 return CString(m_szPathName);
}

int CALLBACK CPathDialog::BrowseCallbackProc(HWND hwnd,UINT uMsg,LPARAM lParam, LPARAM pData)
{
 CPathDialog* pDlg = (CPathDialog*)pData;

 switch(uMsg)
 {
 case BFFM_INITIALIZED:
  {
   RECT rc;
   HWND hEdit;
   HFONT hFont;

   pDlg->m_hWnd = hwnd;

   ::SendMessage(::GetDlgItem(hwnd, IDC_FOLDERTREE), WM_SETFONT, NULL, MAKELPARAM(TRUE, 0));
   if(pDlg->m_lpszCaption!=NULL)
   {
    ::SetWindowText(hwnd, pDlg->m_lpszCaption);
   }

   VERIFY(pDlg->m_PathDialogSub.SubclassWindow(hwnd));
   ::ShowWindow(::GetDlgItem(hwnd, IDC_STATUSTEXT), SW_HIDE);
   ::GetWindowRect(::GetDlgItem(hwnd, IDC_FOLDERTREE), &rc);
   rc.bottom = rc.top - 10;
   rc.top = rc.bottom - 20;
   ::ScreenToClient(hwnd, (LPPOINT)&rc);
   ::ScreenToClient(hwnd, ((LPPOINT)&rc)+1);
   hEdit = ::CreateWindowEx(WS_EX_CLIENTEDGE, _T("EDIT"), _T(""),
    WS_CHILD|WS_TABSTOP|WS_VISIBLE|ES_AUTOHSCROLL,
    rc.left, rc.top,
    rc.right-rc.left, rc.bottom-rc.top,
    hwnd, NULL, NULL, NULL);
   ::SetWindowLong(hEdit, GWL_ID, IDC_NEW_EDIT_PATH);
   ::ShowWindow(hEdit, SW_SHOW);

   hFont = (HFONT)::SendMessage(::GetDlgItem(hwnd, IDC_FOLDERTREE),
    WM_GETFONT, 0, 0);
   ::SendMessage(hEdit, WM_SETFONT, (WPARAM)hFont, MAKELPARAM(TRUE, 0));

   LPCTSTR lpszPath = pDlg->m_lpszInitialPath;
   TCHAR szTemp[MAX_PATH];
   if(lpszPath==NULL)
   {
    ::GetCurrentDirectory(MAX_PATH, szTemp );
    lpszPath = szTemp;
   }
   // WParam is TRUE since you are passing a path.
   // It would be FALSE if you were passing a pidl.
   ::SendMessage(hwnd,BFFM_SETSELECTION,TRUE,
    (LPARAM)lpszPath);
   break;
  }
 case BFFM_SELCHANGED:
  {
   char szSelection[MAX_PATH];
   if(!::SHGetPathFromIDList((LPITEMIDLIST)lParam, szSelection) ||
    szSelection[1]!=':')
   {
    szSelection[0] = '/0';
    ::SendMessage(hwnd, BFFM_ENABLEOK, 0, FALSE);
   }
   else
   {
    ::SendMessage(hwnd, BFFM_ENABLEOK, 0, TRUE);
   }
   ::SendMessage(hwnd,BFFM_SETSTATUSTEXT,0,(LPARAM)szSelection);
   ::SetWindowText(::GetDlgItem(hwnd, IDC_NEW_EDIT_PATH), szSelection);
   break;
  }
 default:
  break;
 }
 return 0;
}

 

int CPathDialog::DoModal()
{

 /////////////////////////////////////////////////////////
 TCHAR szPathTemp[MAX_PATH];
    m_bi.lpfn = BrowseCallbackProc;  // address of callback function
    m_bi.lParam = (LPARAM)this;      // pass address of object to callback function
 m_bi.pszDisplayName = szPathTemp;

 LPITEMIDLIST pidl;
 LPMALLOC pMalloc;

 int iResult=-1;
 if(SUCCEEDED(SHGetMalloc(&pMalloc)))
 {
//  if((m_pParentWnd!=NULL) && m_pParentWnd->IsWindowEnabled())
//  {
//    m_pParentWnd->EnableWindow(FALSE);
//    m_bParentDisabled = TRUE;
//  }
  m_bGetSuccess = FALSE;
  pidl = SHBrowseForFolder(&m_bi);
  if (pidl!=NULL)
  {
   //not need do this because OnOK function did
   //bSucceeded = SHGetPathFromIDList(pidl, m_szPathName);
   // In C++:
   pMalloc->Free(pidl);
   //In C:
   //pMalloc->lpVtbl->Free(pMalloc,pidl);
   //pMalloc->lpVtbl->Release(pMalloc);
  }
  if(m_bGetSuccess)
  {
   iResult = IDOK;
  }
  pMalloc->Release();
 }

    if(m_bParentDisabled && (m_pParentWnd!=NULL))
 {
  m_pParentWnd->EnableWindow(TRUE);
 }
    m_bParentDisabled=FALSE;

 return iResult;
}

BOOL CPathDialog::IsFileNameValid(LPCTSTR lpFileName)
{
 if(lpFileName==NULL)
 {
  return FALSE;
 }

 int nLen = _tcslen(lpFileName);
 if(nLen<=0)
 {
  return FALSE;
 }

 //check first char
 switch(lpFileName[0])
 {
 case _T('.'):
 case _T(' '):
 case _T('/t'):
  return FALSE;
 }

 //check last char
 switch(lpFileName[nLen-1])
 {
 case _T('.'):
 case _T(' '):
 case _T('/t'):
  return FALSE;
 }

 //check all
 int i=0;
 while(lpFileName[i]!=0)
 {
  switch(lpFileName[i])
  {
  case _T('//'):
  case _T('/'):
  case _T(':'):
  case _T('*'):
  case _T('?'):
  case _T('/"'):
  case _T('<'):
  case _T('>'):
  case _T('|'):
   return FALSE;
  }
  i++;
 }

 return TRUE;
}

const TCHAR c_FolderDoesNotExist[] = _T(
  "文件夹:/n"
  "'%s'/n"
  "不存在. 是否新建?");
const TCHAR c_szErrInvalidPath[] = _T(
  "文件夹:/n"
  "'%s'/n"
  "非法. 请重新输入.");
const TCHAR c_szErrCreatePath[] = _T(
  "文件夹:/n"
  "'%s'"
  "/n创建失败. 请双击检测.");

//return -1: user break;
//return 0: no error
//return 1: lpPath is invalid
//return 2: can not create lpPath
int CPathDialog::MakeSurePathExists(LPCTSTR lpPath)
{
 CString strMsg;
 int iRet;
 try
 {
/*  int nLen = _tcslen(lpPath);
  LPTSTR lpTempPath = new TCHAR[nLen + 1];
  _tcsncpy(lpTempPath, lpPath, nLen + 1);
  if (lpPath[nLen - 1] == _T('//'))
   lpTempPath[nLen - 1] = _T('/0');
  //validate path
*/  iRet=Touch(lpPath, TRUE);
//  delete []lpTempPath;
  if(iRet!=0)
  {
   throw iRet;
  }

  if(_taccess(lpPath, 0)==0)
  {
   return (int)0;
  }

  strMsg.Format(c_FolderDoesNotExist, lpPath);
  if(AfxMessageBox(strMsg, MB_YESNO|MB_ICONQUESTION) != IDYES)
  {
   return (int)-1;
  }

  //create path
  iRet=Touch(lpPath, FALSE);
  if(iRet!=0)
  {
   throw iRet;
  }

  return 0;
 }
 catch(int nErrCode)
 {
  switch(nErrCode)
  {
  case 1:
   strMsg.Format(c_szErrInvalidPath, lpPath);
   break;
  case 2:
  default:
   strMsg.Format(c_szErrCreatePath, lpPath);
   break;
  }

  AfxMessageBox(strMsg, MB_OK|MB_ICONEXCLAMATION);

 }

 return iRet;
}

//return 0: no error
//return 1: lpPath is invalid
//return 2: lpPath can not be created(bValidate==FALSE)
int CPathDialog::Touch(LPCTSTR lpPath, BOOL bValidate)
{
 if(lpPath==NULL)
 {
  return 1;
 }

 TCHAR szPath[MAX_PATH];
 _tcscpy(szPath, lpPath);
 int nLen = _tcslen(szPath);

 //path must be "x:/..."
 if( ( nLen<3 ) ||
  ( ( szPath[0]<_T('A') || _T('Z')<szPath[0] ) &&
    ( szPath[0]<_T('a') || _T('z')<szPath[0] ) ||
  ( szPath[1]!=_T(':') )||
  ( szPath[2]!=_T('//') )
  )
   )
 {
  return 1;
 }

 int i;
 if(nLen==3)
 {
  if(!bValidate)
  {
   if(_access(szPath, 0)!=0)
   {
    return 2;
   }
  }
  return 0;
 }

 i = 3;
 BOOL bLastOne=TRUE;
 LPTSTR lpCurrentName;
 while(szPath[i]!=0)
 {
  lpCurrentName = &szPath[i];
  while( (szPath[i]!=0) && (szPath[i]!=_T('//')) )
  {
   i++;
  }

  bLastOne =(szPath[i]==0);
  szPath[i] = 0;

  if( !IsFileNameValid(lpCurrentName) )
  {
   return 1;
  }

  if(!bValidate)
  {
   CreateDirectory(szPath, NULL);
   if(_taccess(szPath, 0)!=0)
   {
    return 2;
   }
  }

  if(bLastOne)
  {
   break; //it's done
  }
  else
  {
   szPath[i] = _T('//');
  }

  i++;
 }

 return 0;
}

//return 0: ok
//return 1: error
int CPathDialog::ConcatPath(LPTSTR lpRoot, LPCTSTR lpMorePath)
{
 if(lpRoot==NULL)
 {
  return 1;
 }

 int nLen = _tcslen(lpRoot);

 if(nLen<3)
 {
  return 1;
 }

 if(lpMorePath==NULL)
 {
  return 0;
 }

 if(nLen==3)
 {
  _tcscat(lpRoot, lpMorePath);
  return 0;
 }

 _tcscat(lpRoot, _T("//"));
 _tcscat(lpRoot, lpMorePath);

 return 0;
}