学习心得：控件之Tree Control （仿系统目录树视图）

一、几个概念（摘）

1.外壳名字空间：

在WINDOWS中又叫外壳名字空间(Shell Name Space).外壳名字空间是Windows下的标准文件系统，它大大扩展了Dos文件系统，形成了以“桌面”（Desktop）为根的单一的文件系统树，原有的C盘、D盘等目录树变成“我的电脑”这一外壳名字空间子树的下一级子树，而像“控制面板”、“回收站”、“网上邻居”等应用程序及“打印机”等设备也被虚拟成了外壳名字空间中的节点。另外，与ＤＯＳ中物理存储只能和文件系统项一一对应这一点不同的是，一个实际目录在外壳名字空间中可以表现为不同的项。例如“我的文档”与“C:/MyDocuments”其实都指向“C:/My Documents”目录，但它们在外壳名字空间中是不同的项。

2. 外壳名字空间下的路径: 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位的零。

二、几个API

1.HRESULT SHGetSpecialFolderLocation(
HWND hwndOwner,
int nFolder, //CSIDL
LPITEMIDLIST *ppidl //返回CSIDL所对应的绝对PIDL（输出）
);

2.DWORD_PTR SHGetFileInfo(
LPCTSTR pszPath, //uFlags含SHGFI_PIDL时为绝对PIDL（输入）
DWORD dwFileAttributes, //绝对PIDL所对应文件的 file attribute flags （输出）
SHFILEINFO *psfi, //返回file information
UINT cbFileInfo, //SHFILEINFO结构大 UINT uFlags //指定你所要获取的信息
); //该函数返回系统HIMAGELIST

说明：

typedef struct _SHFILEINFO {
HICON hIcon;
int iIcon; //图标索引
DWORD dwAttributes; //文件属性
TCHAR szDisplayName[MAX_PATH]; // 显示名称
TCHAR szTypeName[80]; //文件类型：一个值对应一个二进制位
} SHFILEINFO;

uFlags ：SHGFI_DISPLAYNAME | SHGFI_TYPENAME ：psfi返回中包含显示名称和文件类型 (其他类推)

3.HRESULT SHGetDesktopFolder(
IShellFolder** ppshf //f返回桌面IShellFolder接口
);

IShellFolder的几个方法：

（1）.BingToObject 获取子文件夹的IShellFolder接口

（2）.EnumObject获取IEnumIDList ，IEnumIDList －>Next方法获取子文件的相对PIDL（详见MSDN）

（3）GetAttributesOf获取文件属性

注意事项：参数中是绝对PIDL还是相对PIDL。SHGet绝对，IShellFolder相对。PIDL用IMalloc接口开辟空间

三、仿系统目录树

注：本程序在VS2005编译通过（存在小Bug：树节点的加号要展开后才显示，待修改）。

下面是原码


// BrowseSysTreeDlg.h : 头文件


//




#pragma once


#include "afxcmn.h"






// CBrowseSysTreeDlg 对话框


class CBrowseSysTreeDlg : public CDialog




...{


// 构造


public:


CBrowseSysTreeDlg(CWnd* pParent = NULL); // 标准构造函数






// 对话框数据




enum ...{ IDD = IDD_BROWSESYSTREE_DIALOG };




protected:


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




//添加


protected:


HTREEITEM CreateFolderNode(LPITEMIDLIST lpPidl,HTREEITEM hParent);


BOOL AttachFolders(HTREEITEM hNode);


void FreeNode(HTREEITEM hNode);






// 实现


protected:


HICON m_hIcon;


CTreeCtrl m_ctrlTree;


CImageList m_imageList;


IMalloc* m_pMalloc;




// 生成的消息映射函数


virtual BOOL OnInitDialog();


afx_msg void OnSysCommand(UINT nID, LPARAM lParam);


afx_msg void OnPaint();


afx_msg HCURSOR OnQueryDragIcon();


DECLARE_MESSAGE_MAP()




public:


afx_msg void OnNMDblclkSysTree(NMHDR *pNMHDR, LRESULT *pResult);


public:


afx_msg void OnTvnItemexpandingSysTree(NMHDR *pNMHDR, LRESULT *pResult);


public:


afx_msg void OnDestroy();


};




// BrowseSysTreeDlg.cpp : 实现文件


//




#include "stdafx.h"


#include "BrowseSysTree.h"


#include "BrowseSysTreeDlg.h"


#include "shlobj.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()






// CBrowseSysTreeDlg 对话框




typedef struct _NodeInfo //节点信息




...{


TCHAR szName[MAX_PATH]; //显示的名称


UINT uIcon; //图标


ULONG dwAttributes; //属性


TCHAR szPath[MAX_PATH]; //路径


LPITEMIDLIST lpPidl; //PIDL


WORD wPidlLen; //PLID的长度


IShellFolder* pShellFolder; //指向该节点的IShellFolder接口


BOOL bHasParent; //是否有父节点


}NODEINFO,*LPNODEINFO;




HTREEITEM CBrowseSysTreeDlg::CreateFolderNode(LPITEMIDLIST lpPidl, HTREEITEM hParent)




...{


LPNODEINFO lpParentNodeInfo = NULL;


IShellFolder* pShellFolder = NULL;


TCHAR szName[MAX_PATH];


BOOL bRelease = FALSE;




if (NULL != hParent) //获取父IShellFolder接口




...{


lpParentNodeInfo = (LPNODEINFO) m_ctrlTree.GetItemData(hParent);


pShellFolder = lpParentNodeInfo->pShellFolder;


}


else //没有父节点,则取桌面IShellFolder




...{


::SHGetDesktopFolder(&pShellFolder);


bRelease = TRUE;


if (NULL == pShellFolder)


return NULL;


}




//获取属性


ULONG Attributes = SFGAO_SHARE | SFGAO_FILESYSTEM |


SFGAO_LINK | SFGAO_HASSUBFOLDER;


if(lpParentNodeInfo == NULL || pShellFolder->GetAttributesOf(1,(LPCITEMIDLIST*)&lpPidl, &Attributes) != NOERROR)




...{


Attributes = 0;


}




//长度


WORD wParentPidlLen = 0;


LPITEMIDLIST lpPidlParent;


if (NULL != lpParentNodeInfo)




...{


wParentPidlLen = lpParentNodeInfo->wPidlLen;


lpPidlParent = lpParentNodeInfo->lpPidl;


}


// 使用IMalloc接口分配新PIDL需要的空间.


LPITEMIDLIST lpPidlNew = (LPITEMIDLIST)m_pMalloc->Alloc(lpPidl->mkid.cb + wParentPidlLen + 2);


if(wParentPidlLen != 0)




...{


memcpy(lpPidlNew, lpPidlParent, wParentPidlLen);


memcpy((char*)lpPidlNew+wParentPidlLen, lpPidl,lpPidl->mkid.cb);


}


else




...{


memcpy(lpPidlNew, lpPidl, lpPidl->mkid.cb);


}


*(WORD*)((char*)lpPidlNew + wParentPidlLen + lpPidl->mkid.cb) = 0;


LPITEMIDLIST lpPidlTemp = lpPidlNew;




//获取显示图标


SHFILEINFO shif;


UINT uIcon;


UINT uSelectedIcon;


::SHGetFileInfo((LPCWSTR)lpPidlTemp, 0, &shif, sizeof(shif),


SHGFI_PIDL | SHGFI_SYSICONINDEX);


uIcon = shif.iIcon;


::SHGetFileInfo((LPCWSTR)lpPidlTemp, 0, &shif, sizeof(shif),


SHGFI_PIDL | SHGFI_SYSICONINDEX|SHGFI_OPENICON|SHGFI_DISPLAYNAME);


uSelectedIcon = shif.iIcon;




//获取路径


TCHAR szPath[MAX_PATH];


::SHGetPathFromIDList(lpPidlTemp, szPath);




lstrcpy(szName, shif.szDisplayName); //取得显示名称




//获取节点IShellFolder


IShellFolder* pShellFolderNew = NULL;


pShellFolder->BindToObject(lpPidl, NULL, IID_IShellFolder,(void**) &pShellFolderNew);




//建立新节点


LPNODEINFO lpNodeInfoNew = new NODEINFO;


lpNodeInfoNew->bHasParent = (hParent != NULL);


lpNodeInfoNew->dwAttributes = Attributes;


lpNodeInfoNew->lpPidl = lpPidlNew;


if (NULL == hParent)


lpNodeInfoNew->pShellFolder = pShellFolder;


else


lpNodeInfoNew->pShellFolder = pShellFolderNew;


memcpy(lpNodeInfoNew->szName, szName, sizeof(szName));


memcpy(lpNodeInfoNew->szPath, szPath, sizeof(szPath));


lpNodeInfoNew->uIcon = uIcon;


lpNodeInfoNew->wPidlLen = wParentPidlLen + lpPidl->mkid.cb;




//建立树视图插入结构


TVINSERTSTRUCT tvis;


tvis.item.mask = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE |


TVIF_STATE | TVIF_PARAM;


if(Attributes & SFGAO_HASSUBFOLDER)




...{


tvis.item.mask |= TVIF_CHILDREN;


tvis.item.cChildren = I_CHILDRENCALLBACK;


// 使用 I_CHILDRENCALLBACK 值告诉控件,该结点有子结点,但具体的结点还没给出


// 当该结点被展开时就会通知父窗口.这时你应该为该结点添加子结点


}


else


tvis.item.cChildren = 0;


tvis.item.stateMask = TVIS_OVERLAYMASK; // 指明状态标志包含覆盖图标


tvis.hInsertAfter = TVI_LAST;


tvis.hParent = hParent;


tvis.item.iImage =uIcon;


tvis.item.iSelectedImage = uSelectedIcon;


tvis.item.lParam = (DWORD)lpNodeInfoNew;


tvis.item.cchTextMax =MAX_PATH;


tvis.item.pszText = szName;


tvis.item.stateMask = TVIS_OVERLAYMASK;


// 设置覆盖图标


if(Attributes & SFGAO_SHARE) // 共享的


tvis.item.state = INDEXTOOVERLAYMASK(1);


else if(Attributes & SFGAO_LINK) // 快捷方式


tvis.item.state = INDEXTOOVERLAYMASK(2);


else // 其它的


tvis.item.state = INDEXTOOVERLAYMASK(0);


HTREEITEM hIns = m_ctrlTree.InsertItem(&tvis); // 插入该结点




if(bRelease) //释放桌面IShellFolder


pShellFolder->Release();




return hIns;


}




BOOL CBrowseSysTreeDlg::AttachFolders(HTREEITEM hNode)




...{


CWaitCursor cur; // 显示等待光标


BOOL bRet = FALSE;


BOOL bChildren = FALSE;


m_ctrlTree.SetRedraw(FALSE); // 禁止控件更新窗口,以免插入时闪烁




LPNODEINFO lpfn = (LPNODEINFO)m_ctrlTree.GetItemData(hNode);




IEnumIDList* pEnum = NULL;


if(lpfn->pShellFolder->EnumObjects(m_hWnd, SHCONTF_FOLDERS | SHCONTF_INCLUDEHIDDEN,


&pEnum) == NOERROR)




...{


bChildren = FALSE;


pEnum->Reset();


ULONG u = 1;


LPITEMIDLIST lpidlChild = NULL;


while(pEnum->Next(1, &lpidlChild, &u) == NOERROR)




...{


// 为每个PIDL创建对应的树结点(包含其中的虚拟文件夹)


HTREEITEM hChild = CreateFolderNode(lpidlChild, hNode);


if(hChild != NULL)


bChildren = TRUE;


}


// 释放枚举接口


pEnum->Release();


// 调整父结点的属性


if (TRUE == bChildren)




...{


TVITEM tvi;


tvi.mask = TVIF_CHILDREN;


tvi.hItem = hNode;


m_ctrlTree.SetItem(&tvi);


}


}


// 可以更新窗口了


m_ctrlTree.SetRedraw(TRUE);


return TRUE;


}




void CBrowseSysTreeDlg::FreeNode(HTREEITEM hNode)




...{


if (NULL == hNode)


hNode = m_ctrlTree.GetRootItem();


else




...{


LPNODEINFO lpNodeInfo = (LPNODEINFO)m_ctrlTree.GetItemData(hNode);


if (NULL != lpNodeInfo->pShellFolder)


lpNodeInfo->pShellFolder->Release();


m_pMalloc->Free(lpNodeInfo->lpPidl);


delete lpNodeInfo;


hNode = m_ctrlTree.GetChildItem(hNode);


}


while (NULL != hNode)




...{


FreeNode(hNode);


hNode = m_ctrlTree.GetNextSiblingItem(hNode);


}


}






CBrowseSysTreeDlg::CBrowseSysTreeDlg(CWnd* pParent /**//*=NULL*/)


: CDialog(CBrowseSysTreeDlg::IDD, pParent)




...{


m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);


VERIFY(SHGetMalloc(&m_pMalloc)==NOERROR);


}




void CBrowseSysTreeDlg::DoDataExchange(CDataExchange* pDX)




...{


CDialog::DoDataExchange(pDX);


DDX_Control(pDX, IDC_SYS_TREE, m_ctrlTree);


}




BEGIN_MESSAGE_MAP(CBrowseSysTreeDlg, CDialog)


ON_WM_SYSCOMMAND()


ON_WM_PAINT()


ON_WM_QUERYDRAGICON()


//}}AFX_MSG_MAP


ON_NOTIFY(NM_DBLCLK, IDC_SYS_TREE, &CBrowseSysTreeDlg::OnNMDblclkSysTree)


ON_NOTIFY(TVN_ITEMEXPANDING, IDC_SYS_TREE, &CBrowseSysTreeDlg::OnTvnItemexpandingSysTree)


ON_WM_DESTROY()


END_MESSAGE_MAP()






// CBrowseSysTreeDlg 消息处理程序




BOOL CBrowseSysTreeDlg::OnInitDialog()




...{


CDialog::OnInitDialog();




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




// IDM_ABOUTBOX 必须在系统命令范围内。


ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);


ASSERT(IDM_ABOUTBOX < 0xF000);




CMenu* pSysMenu = GetSystemMenu(NULL);


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, NULL); // 设置小图标




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


LPITEMIDLIST lpPidl;


HIMAGELIST hSysImageList = NULL;


SHFILEINFO shif;




if (NOERROR != SHGetSpecialFolderLocation(m_hWnd, CSIDL_DESKTOP, &lpPidl))


return FALSE;


hSysImageList =(HIMAGELIST) ::SHGetFileInfo((LPCWSTR)lpPidl, 0, &shif, sizeof(shif),


SHGFI_PIDL | SHGFI_SMALLICON | SHGFI_SYSICONINDEX);


// 设置重叠图标


// 设置1号重叠图标对应图标列表中索引为0的图标,这是一个手托,象征被共享的文件夹


// 设置2号重叠图标对应图标列表中索引为1的图标,这是一个箭头,象征快捷方式


// 设置3号重叠图标对应图标列表中索引为2的图标,这个图标??


ImageList_SetOverlayImage(hSysImageList, 0,1);


ImageList_SetOverlayImage(hSysImageList, 1,2);


ImageList_SetOverlayImage(hSysImageList, 2,3);


m_ctrlTree.SetImageList(CImageList::FromHandle(hSysImageList),TVSIL_NORMAL);




HTREEITEM hItem = CreateFolderNode(lpPidl, NULL);


AttachFolders(hItem);


m_ctrlTree.Expand(hItem, TVE_EXPAND);




return TRUE; // 除非将焦点设置到控件，否则返回 TRUE


}




void CBrowseSysTreeDlg::OnSysCommand(UINT nID, LPARAM lParam)




...{


if ((nID & 0xFFF0) == IDM_ABOUTBOX)




...{


CAboutDlg dlgAbout;


dlgAbout.DoModal();


}


else




...{


CDialog::OnSysCommand(nID, lParam);


}


}




// 如果向对话框添加最小化按钮，则需要下面的代码


// 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，


// 这将由框架自动完成。




void CBrowseSysTreeDlg::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 CBrowseSysTreeDlg::OnQueryDragIcon()




...{


return static_cast<HCURSOR>(m_hIcon);


}






void CBrowseSysTreeDlg::OnNMDblclkSysTree(NMHDR *pNMHDR, LRESULT *pResult)




...{


// TODO: Add your control notification handler code here


HTREEITEM hItem = m_ctrlTree.GetSelectedItem();


if(m_ctrlTree.GetChildItem(hItem) == NULL)


AttachFolders(hItem);


*pResult = 0;


}




void CBrowseSysTreeDlg::OnTvnItemexpandingSysTree(NMHDR *pNMHDR, LRESULT *pResult)




...{


LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);


// TODO: Add your control notification handler code here


if (pNMTreeView->action == TVE_EXPAND )


if(m_ctrlTree.GetChildItem(pNMTreeView->itemNew.hItem) == NULL)


AttachFolders(pNMTreeView->itemNew.hItem);


*pResult = 0;


}








void CBrowseSysTreeDlg::OnDestroy()




...{


FreeNode(NULL);


m_pMalloc->Release();


delete m_pToolTipCtrl;


CDialog::OnDestroy();


// TODO: 在此处添加消息处理程序代码


}