仿Explorer的目录树视图类

在网上找了很长时间也没有关于这方面的VC代码.倒是找到几个控件,用起来却很不爽.很奇怪WINDOWS里为什么没有这样的控件.没办法.自己实现一个呗. 下面这个类从CTreeView拜生.参考了DELPHI的相应控件源代码写的,本人对DELPHI不会,所以看得很勉强,好多地方也没看明白.不过这个类使用起来的效果,嘿嘿.已经和那个控件差不多了:).只是比起Explorer里的来,还差着一些.具体差在什么地方,大家用一用就知道了. 将你的类从此类拜生,就能做你想做的了. 该类只导出两个函数: 1.SetCurPath : 设置选中一个路径.如果需要的话,它会展开相应有文件夹 2.GetSelectedDir : 返回当前选中的文件夹,没有选中的话,返回自然就是空串 如果你有兴趣,自己对这个类进行扩充吧.欢迎和我交流. ////////////////////////////////////////////////////////////////////////// // ShellFolderView.h ---- 头文件 #if !defined(AFX_SHELLFOLDERVIEW_H__D30D10ED_3D11_4873_9909_146B8E18BEB1__INCLUDED_) #define AFX_SHELLFOLDERVIEW_H__D30D10ED_3D11_4873_9909_146B8E18BEB1__INCLUDED_ #if _MSC_VER > 1000 #pragma once #endif // _MSC_VER > 1000 // ShellFolderView.h : header file //********************************************************************** // 该类根据DELPHI相应控件源代码改编 /*********************************************************************** 简介: CShellFolderView从CTreeView拜生,它显示整个WINDOWS的文件目录树.在WINDOWS中 又叫外壳名字空间(Shell Name Space).外壳名字空间是Windows下的标准文件系统， 它大大扩展了Dos文件系统，形成了以“桌面”（Desktop）为根的单一的文件系统 树，原有的Ｃ盘、Ｄ盘等目录树变成了“我的电脑”这一外壳名字空间子树的下一级 子树，而像“控制面板”、“回收站”、“网上邻居”等应用程序及“打印机”等设 备也被虚拟成了外壳名字空间中的节点。另外，与ＤＯＳ中物理存储只能和文件系统 项一一对应这一点不同的是，一个实际目录在外壳名字空间中可以表现为不同的项。 例如“我的文档”与“C:/My Documents”其实都指向“C:/My Documents”目录，但 它们在外壳名字空间中是不同的项。 外壳名字空间下的路径: PIDL PIDL是一个元素类型为ITEMIDLIST结构的数组，数组中元素的个数是未知的，但紧接 着数组末尾的必是一个双字节的零。每个数组元素代表了外壳名字空间树中的一层（ 即一个文件夹或文件），数组中的前一元素代表的是后一元素的父文件夹。由此可见， PIDL实际上就是指向一块由若干个顺序排列的ITEMIDLIST结构组成、并在最后有一个 双字节零的空间的指针。所以PIDL的类型就被Windows定义为ITEMIDLIST结构的指针。 PIDL亦有“绝对路径”与“相对路径”的概念。表示“相对路径”的PIDL只有一个 ITEMIDLIST结构的元素，用于标识相对于父文件夹的“路径”；表示“绝对路径”的 PIDL（简称为“绝对PIDL”）有若干个ITEMIDLIST结构的元素，第一个元素表示外壳 名字空间根文件夹（“桌面”）下的某一子文件夹A，第二个元素则表示文件夹A下的 某一子文件夹B，其余依此类推。这样绝对PIDL就通过保存一条从“桌面”下的直接子 文件夹或文件的绝对PIDL与相对PIDL是相同的，而其他的文件夹或文件的相对PIDL就 只是其绝对PIDL的最后一部分了。由于所有的PIDL都是从桌面下的某一个子文件夹开 始的,所以对于桌面本身来说,它的PIDL数组显然一个元素都没有。这样就只剩下PIDL 数组最后的那个双字节的零了。所以，“桌面”的PIDL就是一个16位的零。 **********************************************************************/ // 监视指定的文件夹的变化的类. 该类由CShellFolderView专用,所以定义在源文件中 // 不对外部公开,在这里进行提前声明,因为下面要用到 class CFolderChangeMonitor; // 外壳名字空间下结点的串表示 typedef struct _FILEPATHLIST { char cNormal[MAX_PATH]; // 绝对名称(如,"我的电脑") char cForParsing[MAX_PATH]; // 用来分析解释的名称, 可以根据它来产生PIDL char cInFolder[MAX_PATH]; // 相对名称 } FILEPATHLIST, *LPFILEPATHLIST; // 文件夹的分类 typedef enum emFolderType{ ftFolder, // 文件夹 ftNetworkNeighborhood, // 网上邻居 ftRecycleBin, // 回收站 ftMyComputer, // 我的电脑 ftDesktop, // 桌面 ftNetNode, // 网络结点 ftNone // 空 }FOLDERTYPE; // 路径信息 typedef struct _PATHINFO{ char cPath[MAX_PATH]; // 路径 char cText[MAX_PATH]; // 名称 UINT uIconIndex; // 图标索引 UINT uSelectedIndex; // 选中时的图标索引 LPITEMIDLIST pidl; // PIDL }PATHINFO,*LPPATHINFO; // 结点数据 typedef struct _FOLDERNODE{ char cPath[MAX_PATH]; // 路径 BOOL bHasParent; // 是否有父结点 LONG Vol_Ser; // 卷标 ULONG dwAttributes; // 属性 FOLDERTYPE fType; // 类型 IShellFolder* pShellFolder; // 指向该文件夹的IShellFolder接口 LPITEMIDLIST pidl; // 文件夹的PIDL WORD widlen; // PIDL的长度 CFolderChangeMonitor* pMonitorThread; // 监视文件夹变化的线程 }FOLDERNODE, *LPFOLDERNODE; ///////////////////////////////////////////////////////////////////////////// // CShellFolderView view class CShellFolderView : public CTreeView { protected: CShellFolderView(); // protected constructor used by dynamic creation DECLARE_DYNCREATE(CShellFolderView) // Attributes public: // Operations public: BOOL SetCurPath(LPCTSTR lpszPath); // 设置当前选中的文件夹, 成功返回TRUE CString GetSelectedDir(); // 取得当前选中的文件夹,没有则返回一个空串 // Overrides // ClassWizard generated virtual function overrides //{{AFX_VIRTUAL(CShellFolderView) protected: //virtual void OnDraw(CDC* pDC); // overridden to draw this view //}}AFX_VIRTUAL // Implementation protected: virtual ~CShellFolderView(); #ifdef _DEBUG virtual void AssertValid() const; virtual void Dump(CDumpContext& dc) const; #endif // Generated message map functions protected: //{{AFX_MSG(CShellFolderView) afx_msg int OnCreate(LPCREATESTRUCT lpCreateStruct); afx_msg void OnItemexpanding(NMHDR* pNMHDR, LRESULT* pResult); afx_msg void OnDestroy(); //}}AFX_MSG afx_msg LRESULT OnFolderChanged(WPARAM wParam, LPARAM); // 处理文件夹发生变化的消息 DECLARE_MESSAGE_MAP() private: IMalloc* m_pMalloc; // COM内存分配接口用来分配和回收PIDL使用的空间 protected: void GetSpecialFolder(int nFolder, LPPATHINFO lppi); // 取得指定的特殊文件夹的信息 void GetNetHood(LPPATHINFO lpPI); // 取得网上邻居结点 // 创建一个结点 HTREEITEM CreateFolderNode(LPITEMIDLIST lpidl, HTREEITEM hParent, BOOL bVirtual); CString GetPath(HTREEITEM hNode); // 取得结点的对应的路径 BOOL AttachFolders(HTREEITEM hNode); // 展开指定的结点下的子结点 LONG GetVol_Ser(LPCTSTR strDriver); // 获取卷标 void ShowDesktop(); // 显示桌面 void RetrieveSysImageList(); // 获取系统图像列表 void FreeNode(HTREEITEM hNode); // 释放结点 void FreeNodeData(LPFOLDERNODE lpfn); // 释放结点数据 int Expand(HTREEITEM hNode, UINT nCode); // 展开结点 }; ///////////////////////////////////////////////////////////////////////////// //{{AFX_INSERT_LOCATION}} // Microsoft Visual C++ will insert additional declarations immediately before the previous line. #endif // !defined(AFX_SHELLFOLDERVIEW_H__D30D10ED_3D11_4873_9909_146B8E18BEB1__INCLUDED_) /////////////////////////////////////////////////////////////////////////// // ShellFolderView.cpp : implementation file /////////////////////////////////////////////////////////////////////////// #include "stdafx.h" #include "ShellFolderView.h" #ifdef _DEBUG #define new DEBUG_NEW #undef THIS_FILE static char THIS_FILE[] = __FILE__; #endif #define WM_U_FOLDERCHANGED (WM_USER+215) // 消息 // 一些固定的名字: static const char c_cMyComputer[] = "::{20D04FE0-3AEA-1069-A2D8-08002B30309D}"; // Normal = 我的电脑 static const char c_cNetworkNeighborhood[] = "::{208D2C60-3AEA-1069-A2D7-08002B30309D}"; // Normal = 网上邻居 static const char c_cRecycleBin[] = "::{645FF040-5081-101B-9F08-00AA002F954E}"; // Normal = 回收站 static const char c_cController[] = "::{20D04FE0-3AEA-1069-A2D8-08002B30309D}//::{21EC2020-3AEA-1069-A2DD-08002B30309D}"; // Normal = 控制面板 static const char c_cPrinter[] = "::{20D04FE0-3AEA-1069-A2D8-08002B30309D}//::{21EC2020-3AEA-1069-A2DD-08002B30309D}//::{2227A280-3AEA-1069-A2DE-08002B30309D}"; // Normal = 打印机 static const char c_cDial[] = "::{20D04FE0-3AEA-1069-A2D8-08002B30309D}//::{21EC2020-3AEA-1069-A2DD-08002B30309D}//::{7007ACC7-3202-11D1-AAD2-00805FC1270E}"; // Normal = 网络和拨号连接 static const char c_cFonts[] = "::{20D04FE0-3AEA-1069-A2D8-08002B30309D}//::{21EC2020-3AEA-1069-A2DD-08002B30309D}//::{D20EA4E1-3957-11D2-A40B-0C5020524152}"; // Normal = 字体 static const char c_cManager[] = "::{20D04FE0-3AEA-1069-A2D8-08002B30309D}//::{21EC2020-3AEA-1069-A2DD-08002B30309D}//::{D20EA4E1-3957-11D2-A40B-0C5020524153}"; // Normal = 管理工具 static const char c_cPlans[] = "::{20D04FE0-3AEA-1069-A2D8-08002B30309D}//::{21EC2020-3AEA-1069-A2DD-08002B30309D}//::{D6277990-4C6A-11CF-8D87-00AA0060F5BF}"; // Normal = 任务计划 // 文件夹监视线程类 class CFolderChangeMonitor { public: CFolderChangeMonitor(LPCTSTR lpszPath, HWND hWnd, WPARAM wParam); ~CFolderChangeMonitor(); private: CWinThread* m_pThread; CString m_strPath; HWND m_hWnd; WPARAM m_wParam; BOOL m_bStop; friend UINT MonitorProc(void* lp); void Run(); }; ///////////////////////////////////////////////////////////////////////////// // CFolderChangeMonitor class ///////////////////////////////////////////////////////////////////////////// // 文件夹监视线程类的构造函数 // lpszPath : 被监视的文件夹 // hWnd : 文件夹已经改变的消息将发往该窗口 // wParam : 消息的参数,在这里是树上的一结点句柄 CFolderChangeMonitor::CFolderChangeMonitor(LPCTSTR lpszPath, HWND hWnd, WPARAM wParam) :m_hWnd(hWnd), m_wParam(wParam), m_strPath(lpszPath) { m_bStop = FALSE; // 开始监视线程 m_pThread = AfxBeginThread(MonitorProc, (void*)this); } CFolderChangeMonitor::~CFolderChangeMonitor() { // 停止监视线程 m_bStop = TRUE; // 原来可以等待线程结束后才返回,后来不知道怎么搞的,居然会死锁,只好先不等待了,嘻嘻 // ::WaitForSingleObject(m_pThread->m_hThread, INFINITE); } // 监视线程函数 UINT MonitorProc(void*lp) { if(lp) { // 将线程反调回类中 ((CFolderChangeMonitor*)lp)->Run(); } return 0; } // 监视函数 void CFolderChangeMonitor::Run() { // 检测结束标志, 没有就一直循环 while(!m_bStop) { // 创建事件句柄,当文件夹发生了指定的变化时,该事件被触发 HANDLE handle = ::FindFirstChangeNotification(m_strPath, FALSE, FILE_NOTIFY_CHANGE_DIR_NAME); // 如果事件创建不成功则结束 if(handle == INVALID_HANDLE_VALUE) break; // 开始监视 if(::FindNextChangeNotification(handle)) { int i = WAIT_TIMEOUT; // 除非指明结束监视或者发生了一次事件,否则一直循环等待 while(i == WAIT_TIMEOUT && !m_bStop) { // 等待事件发生(每次等待20mS)因为线程需要不断检测m_bStop // 标志,以便及时退出,所以不能无限制地等待下去 i = ::WaitForSingleObject(handle, 20); } // 循环结束,如果不是指定结束,则一定是发事件被触发了 if(!m_bStop) { // 发送消息通知窗口来更新目录树 ::SendMessage(m_hWnd, WM_U_FOLDERCHANGED, m_wParam,0); } } else { // 监视不能开始,则退出 ::FindCloseChangeNotification(handle); break; } // 关闭事件 ::FindCloseChangeNotification(handle); } } /*************************************************************** FUNCTION : GetName 描述：lpio lpsf所指的IshellFolder接口代表的文件夹下的相对PIDL， 本函数获得lpi所指项的显示名称，dwFlags表明欲得到的显示名称类型， lpFriendlyName为存放显示名称的缓冲区。 ***************************************************************/ BOOL GetName(LPSHELLFOLDER lpsf,LPITEMIDLIST lpi,DWORD dwFlags,LPSTR lpFriendlyName) { STRRET str; AGAIN: //得到显示名称 if(NOERROR!=lpsf->GetDisplayNameOf(lpi,dwFlags,&str)) return FALSE; //根据返回值进行转换 switch(str.uType) { case STRRET_WSTR: //如为Unicode字符串，则转成Ansi字符集的字符串case STRRET_WSTR: if(str.pOleStr != NULL) WideCharToMultiByte(CP_ACP,0,str.pOleStr,-1,lpFriendlyName,MAX_PATH,NULL,NULL); else if(dwFlags == SHGDN_NORMAL) { dwFlags = SHGDN_FORPARSING; goto AGAIN; } break; case STRRET_OFFSET: //如为偏移量，则去除偏移量 lstrcpy(lpFriendlyName,(LPSTR)lpi+str.uOffset); break; case STRRET_CSTR: // 如为Ansi字符串，则直接拷贝 lstrcpy(lpFriendlyName,(LPSTR)str.cStr); break; default: //非法情况 return FALSE; } return TRUE; } // 获取路径 void GetLongFilePath(IShellFolder* pShellFolder, LPITEMIDLIST lpidl, LPFILEPATHLIST lpfpl) { BOOL bRelease = FALSE; // 如果IShellFolder接口为空,则使用桌面的IShellFolder接口 if(pShellFolder == NULL) { if(SHGetDesktopFolder(&pShellFolder) != NOERROR) return; bRelease = TRUE; } // 取得显示, PIDL及相对名称 GetName(pShellFolder, lpidl, SHGDN_NORMAL, lpfpl->cNormal); GetName(pShellFolder, lpidl, SHGDN_FORPARSING, lpfpl->cForParsing); GetName(pShellFolder, lpidl, SHGDN_INFOLDER, lpfpl->cInFolder); // 取得的COM接口一定要释放 if(bRelease) pShellFolder->Release(); } ///////////////////////////////////////////////////////////////////////////// // CShellFolderView IMPLEMENT_DYNCREATE(CShellFolderView, CTreeView) CShellFolderView::CShellFolderView() { // 取得内存分配的COM接口, 这里一定要成功,否则接下来的程序中无法分配空间 VERIFY(SHGetMalloc(&m_pMalloc)==NOERROR); } CShellFolderView::~CShellFolderView() { // 释放IMalloc接口 m_pMalloc->Release(); } BEGIN_MESSAGE_MAP(CShellFolderView, CTreeView) //{{AFX_MSG_MAP(CShellFolderView) ON_WM_CREATE() ON_NOTIFY_REFLECT(TVN_ITEMEXPANDING, OnItemexpanding) ON_WM_DESTROY() //}}AFX_MSG_MAP ON_MESSAGE(WM_U_FOLDERCHANGED, OnFolderChanged) END_MESSAGE_MAP() ///////////////////////////////////////////////////////////////////////////// // CShellFolderView drawing //void CShellFolderView::OnDraw(CDC* pDC) //{ // CDocument* pDoc = GetDocument(); // TODO: add draw code here //} ///////////////////////////////////////////////////////////////////////////// // CShellFolderView diagnostics #ifdef _DEBUG void CShellFolderView::AssertValid() const { CTreeView::AssertValid(); } void CShellFolderView::Dump(CDumpContext& dc) const { CTreeView::Dump(dc); } #endif //_DEBUG ///////////////////////////////////////////////////////////////////////////// // CShellFolderView message handlers int CShellFolderView::OnCreate(LPCREATESTRUCT lpCreateStruct) { if (CTreeView::OnCreate(lpCreateStruct) == -1) return -1; // TODO: Add your specialized creation code here // 设置树控件的风格:带有展开按钮,结点间有连线 GetTreeCtrl().ModifyStyle(0, TVS_HASLINES|TVS_HASBUTTONS|TVS_LINESATROOT); RetrieveSysImageList(); // 取得系统图标列表 // 显示桌面虚拟文件夹 ShowDesktop(); return 0; } // 当某个结点将要被展开前 void CShellFolderView::OnItemexpanding(NMHDR* pNMHDR, LRESULT* pResult) { NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR; // TODO: Add your control notification handler code here // 如果是展开操作 if(pNMTreeView->action == TVE_EXPAND) { // 如果被展开的结点还没有子结点则为该结点附加它的子结点 // 我们不能一开始就将所有的结点添加上去,这将是一个漫长的过程,且浪费 // 大量的系统资源,只有那些被展开的结点才需要显示其子结点 if(GetTreeCtrl().GetChildItem(pNMTreeView->itemNew.hItem) == NULL) { AttachFolders(pNMTreeView->itemNew.hItem); } } *pResult = 0; *pResult = 0; } void CShellFolderView::OnDestroy() { // 释放所有的结点 FreeNode(NULL); CTreeView::OnDestroy(); } // 当被监视的文件夹发生变化时,该消息被监视线程发送过来 LRESULT CShellFolderView::OnFolderChanged(WPARAM wParam, LPARAM) { // 变化对应的结点句柄 HTREEITEM hNode = (HTREEITEM)wParam; CTreeCtrl& tree = GetTreeCtrl(); // 视图上的树控件 // 保存该结点的状态(是否是已经展开的) BOOL bExpanded = tree.GetItemState(hNode, TVIS_EXPANDED) == TVIS_EXPANDED; // 它的子结点 HTREEITEM hChild = tree.GetChildItem(hNode); // 删除该结点下的所有子结点 while(hChild != NULL) { HTREEITEM hCur = hChild; hChild = tree.GetNextSiblingItem(hChild); // 下一个子结点 FreeNode(hCur); tree.DeleteItem(hCur); } // 重新生成子结点. AttachFolders(hNode); // 恢复展开状态(如果以前是展开的) if(bExpanded) tree.Expand(hNode, TVE_EXPAND); return 0; } ////////////////////////////////////////////////////////////////// // protected member function // 获取系统图标列表 void CShellFolderView::RetrieveSysImageList() { PATHINFO DesktopInfo; SHFILEINFO FileInfo; CString strDesktop; GetSpecialFolder(CSIDL_DESKTOPDIRECTORY, &DesktopInfo); // SHGetFileInfo函数的返回值就是系统图标列表的句柄(可参考MSDN) HIMAGELIST hSysImageList = (HIMAGELIST)::SHGetFileInfo( (LPCTSTR)DesktopInfo.pidl, 0, &FileInfo, sizeof(FileInfo), SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON); // 设置重叠图标 // 设置1号重叠图标对应图标列表中索引为0的图标,这是一个手托,象征被共享的文件夹 // 设置2号重叠图标对应图标列表中索引为1的图标,这是一个箭头,象征快捷方式 // 设置3号重叠图标对应图标列表中索引为2的图标,这个图标我研究了好长时间也没看出来是个什么东东 ImageList_SetOverlayImage(hSysImageList, 0,1); ImageList_SetOverlayImage(hSysImageList, 1,2); ImageList_SetOverlayImage(hSysImageList, 2,3); // 树控件就使用这个图标列表 GetTreeCtrl().SetImageList(CImageList::FromHandle(hSysImageList),TVSIL_NORMAL); } // 获取特殊的文件夹信息 void CShellFolderView::GetSpecialFolder(int nFolder, LPPATHINFO lppi) { LPITEMIDLIST lpidl = NULL; // 取得该文件夹的PIDL if(SUCCEEDED(SHGetSpecialFolderLocation(m_hWnd, nFolder, &lpidl))) { lppi->pidl = lpidl; // Get the actual path of the directory from the PItemIDList SHGetPathFromIDList(lpidl, lppi->cPath); // Do it // Get the Normal Image index SHFILEINFO shfi; SHGetFileInfo((LPCTSTR)lpidl, SFGAO_SHARE, &shfi, sizeof(shfi), SHGFI_PIDL | SHGFI_SYSICONINDEX); lppi->uIconIndex = shfi.iIcon; // Get the selected image index SHGetFileInfo((LPCTSTR)lpidl, SFGAO_SHARE, &shfi, sizeof(shfi), SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_OPENICON); lppi->uSelectedIndex = shfi.iIcon; } } void CShellFolderView::GetNetHood(LPPATHINFO lpPI) { FILEPATHLIST fpl; GetSpecialFolder(CSIDL_NETWORK, lpPI); GetLongFilePath(NULL, lpPI->pidl, &fpl); strcpy(lpPI->cText, fpl.cNormal); } // CreateFolderNode creates a Folder_Node and inserts it under // the "parent" node (if any), using the last of the path // string as the Name, and setting the new node's text // property to match. HTREEITEM CShellFolderView::CreateFolderNode( LPITEMIDLIST lpidl, // 文件夹的PIDL HTREEITEM hParent, // 父结点 BOOL bVirtual) // 是否包含虚拟文件夹 { CTreeCtrl& tree = GetTreeCtrl(); LPFOLDERNODE lpfnParent = NULL; FILEPATHLIST fpl; // 父结点的信息 if(hParent != NULL) { lpfnParent = (LPFOLDERNODE)tree.GetItemData(hParent); GetLongFilePath(lpfnParent->pShellFolder, lpidl, &fpl); } else { GetLongFilePath(NULL, lpidl, &fpl); } CString strFileName(fpl.cInFolder); // 相对名字 CString strPath(fpl.cForParsing); // 法定名字 // 以"::"开头的,是一些特殊的文件夹,普通的目录或文件的名称中是不能包含":"字符的 // 打印出来看看都是些什么东东 if(strncmp(fpl.cForParsing, "::", 2) ==0) { TRACE3("