dirtree.h

在VC编程环境下

// DirTree.h: interface for the CDirTree class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_DIRTREE_H__3C19DFC0_C3B2_11D3_B431_812F2AD2CE4E__INCLUDED_)
#define AFX_DIRTREE_H__3C19DFC0_C3B2_11D3_B431_812F2AD2CE4E__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#include "CtrlView.h"
#include <shlobj.h>

/////////////////////////////////////////////////////////////////////////
//class CDirTree
class CDirTree : public CFBTreeView
{
public:
	CDirTree();
	virtual ~CDirTree();

	enum FindAttribs{SHELL_DRIVE = 1,SHELL_FOLDER = 2};

	typedef struct tagLVID
	{
		LPSHELLFOLDER lpsfParent;
		LPITEMIDLIST  lpi;
		ULONG         ulAttribs;
	} LVITEMDATA, *LPLVITEMDATA;

	typedef struct tagID
	{
		LPSHELLFOLDER lpsfParent;
		LPITEMIDLIST  lpi;
		LPITEMIDLIST  lpifq;
	} TVITEMDATA, *LPTVITEMDATA;

	BOOL	Create(CFBWnd* pParent,UINT nID);

	void    TunnelTree(LPCTSTR szFindPath);
	bool	SearchTree(HTREEITEM treeNode,
						LPTSTR szSearchName,
						FindAttribs attr);
	void	OnItemexpanding(LPNMHDR pNMHDR);
	void	OnSelchanged(LPNMHDR pNMHDR);
	BOOL	OnFolderSelected(NMHDR* pNMHDR, LPTSTR szFolderPath);

protected:

	char	m_strCurDir[MAX_PATH];

	void			FillDirTree(LPSHELLFOLDER lpsf, LPITEMIDLIST lpifq,HTREEITEM hParent);
    LPITEMIDLIST	CreatePidl(UINT cbSize);
    LPITEMIDLIST	ConcatPidls(LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2);
    UINT			GetSize(LPCITEMIDLIST pidl);
	BOOL			GetName(LPSHELLFOLDER lpsf,
							LPITEMIDLIST  lpi,
							DWORD         dwFlags,
							LPTSTR         lpFriendlyName);
	LPITEMIDLIST	CopyITEMID(LPMALLOC lpMalloc, LPITEMIDLIST lpi);
	int				GetItemIcon(LPITEMIDLIST lpi, UINT uFlags);

	virtual BOOL	OnCreate();
	virtual void	OnNotify(LPNMHDR pNMHDR,LRESULT &lResult);

	static int CALLBACK SortCBProc(LPARAM, LPARAM, LPARAM);
};

#endif // !defined(AFX_DIRTREE_H__3C19DFC0_C3B2_11D3_B431_812F2AD2CE4E__INCLUDED_)