开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 界面编程 > TreeView控件 > 查看源码
DirTreeCtrl.cpp
资源名称：CDirTreeCtrl_demo.zip　[点击查看]
上传用户：httpyym
上传日期：2007-01-02
资源大小：38k
文件大小：11k
源码类别：

TreeView控件

开发平台：

Visual C++

DirTreeCtrl.cpp：源码内容
  1. // DirTreeCtrl.cpp: 
  2. // 
  3. // wrapped CTreeCtrl to select and or display folders and files (optional )
  4. // 
  6. #include "stdafx.h"
  8. #include "DirTreeCtrl.h"
  9. #include "SortStringArray.h"
  12. #ifdef _DEBUG
  13. #define new DEBUG_NEW
  14. #undef THIS_FILE
  15. static char THIS_FILE[] = __FILE__;
  16. #endif
  18. /////////////////////////////////////////////////////////////////////////////
  19. // CDirTreeCtrl
  21. CDirTreeCtrl::CDirTreeCtrl()
  22. {
  23. m_strRoot = "";      // Auf Leer setzen
  24. }
  26. CDirTreeCtrl::~CDirTreeCtrl()
  27. {
  28. m_imgList.Detach();
  29. }
  32. BEGIN_MESSAGE_MAP(CDirTreeCtrl, CTreeCtrl)
  33. //{{AFX_MSG_MAP(CDirTreeCtrl)
  34. ON_NOTIFY_REFLECT(TVN_ITEMEXPANDED, OnItemexpanded)
  35. //}}AFX_MSG_MAP
  36. END_MESSAGE_MAP()
  38. /////////////////////////////////////////////////////////////////////////////
  39. // Behandlungsroutinen f黵 Nachrichten CDirTreeCtrl 
  41. BOOL CDirTreeCtrl::DisplayTree(LPCTSTR strRoot, BOOL bFiles)
  42. {
  43. DWORD dwStyle = GetStyle();   // read the windowstyle
  44. if ( dwStyle & TVS_EDITLABELS ) 
  45. {
  46. // Don't allow the user to edit ItemLabels
  47. ModifyStyle( TVS_EDITLABELS , 0 );
  48. }
  50. // Display the DirTree with the Rootname e.g. C:
  51. // if Rootname == NULL then Display all Drives on this PC
  52.     // First, we need the system-ImageList
  54. DeleteAllItems();
  56. if ( !GetSysImgList() )
  57. return FALSE;
  58.     m_bFiles = bFiles;  // if TRUE, Display Path- and Filenames 
  59. if ( strRoot == NULL || strRoot[0] == '' )
  60. {
  61. if ( !DisplayDrives() )
  62. return FALSE;
  63. m_strRoot = "";
  64. }
  65.     else
  66. {
  67. m_strRoot = strRoot;
  68. if ( m_strRoot.Right(1) != '\' )
  69. m_strRoot += "\";
  70. HTREEITEM hParent = AddItem( TVI_ROOT, m_strRoot );
  71. DisplayPath( hParent, strRoot );
  72. }
  73. return TRUE;
  74. }
  75. /////////////////////////////////////////////////
  76. BOOL CDirTreeCtrl::GetSysImgList()
  77. /////////////////////////////////////////////////
  78. {
  79. SHFILEINFO shFinfo;
  80. HIMAGELIST hImgList = NULL;
  82. if ( GetImageList( TVSIL_NORMAL ) )
  83. m_imgList.Detach();
  85. hImgList = (HIMAGELIST)SHGetFileInfo( "C:\",
  86.   0,
  87.   &shFinfo,
  88.   sizeof( shFinfo ),
  89.   SHGFI_SYSICONINDEX | 
  90.   SHGFI_SMALLICON );
  91. if ( !hImgList )
  92. {
  93. m_strError = "Cannot retrieve the Handle of SystemImageList!";
  94. return FALSE;
  95. }
  96.     
  97. if ( !m_imgList.Attach( hImgList ) )
  98. {
  99. m_strError = "Cannot Attach SystemImageList-Handle";
  100. return FALSE;
  101. }
  102.     
  103. SetImageList( &m_imgList, TVSIL_NORMAL );
  104. return TRUE;   // OK
  105. }
  107. BOOL CDirTreeCtrl::DisplayDrives()
  108. {
  109. //
  110. // Displaying the Availible Drives on this PC
  111. // This are the First Items in the TreeCtrl
  112. //
  113. DeleteAllItems();
  114. char  szDrives[128];
  115. char* pDrive;
  117. if ( !GetLogicalDriveStrings( sizeof(szDrives), szDrives ) )
  118. {
  119. m_strError = "Error Getting Logical DriveStrings!";
  120. return FALSE;
  121. }
  123. pDrive = szDrives;
  124. while( *pDrive )
  125. {
  126. HTREEITEM hParent = AddItem( TVI_ROOT, pDrive );
  127. if ( FindSubDir( pDrive ) )
  128. InsertItem( "", 0, 0, hParent );
  129. pDrive += strlen( pDrive ) + 1;
  130. }
  133. return TRUE;
  135. }
  137. void CDirTreeCtrl::DisplayPath(HTREEITEM hParent, LPCTSTR strPath)
  138. {
  139. //
  140. // Displaying the Path in the TreeCtrl
  141. //
  142. CFileFind find;
  143. CString   strPathFiles = strPath;
  144. BOOL      bFind;
  145. CSortStringArray strDirArray;
  146. CSortStringArray strFileArray;
  148. if ( strPathFiles.Right(1) != "\" )
  149. strPathFiles += "\";
  150. strPathFiles += "*.*";
  152. bFind = find.FindFile( strPathFiles );
  154. while ( bFind )
  155. {
  156. bFind = find.FindNextFile();
  157. if ( find.IsDirectory() && !find.IsDots() )
  158. {
  159. strDirArray.Add( find.GetFilePath() );
  160. }
  161. if ( !find.IsDirectory() && m_bFiles )
  162. strFileArray.Add( find.GetFilePath() );
  164. }
  165.     
  166. strDirArray.Sort();
  167. SetRedraw( FALSE );
  168. CWaitCursor wait;
  169.     
  170. for ( int i = 0; i < strDirArray.GetSize(); i++ )
  171. {
  172. HTREEITEM hItem = AddItem( hParent, strDirArray.GetAt(i) );
  173. if ( FindSubDir( strDirArray.GetAt(i) ) )
  174. InsertItem( "", 0, 0, hItem );
  175. }
  177. if ( m_bFiles )
  178. {
  179. strFileArray.Sort();
  180. for ( i = 0; i < strFileArray.GetSize(); i++ )
  181. {
  182. HTREEITEM hItem = AddItem( hParent, strFileArray.GetAt(i) );
  184. }
  185. }
  186.     
  187. SetRedraw( TRUE );
  190. }
  192. HTREEITEM CDirTreeCtrl::AddItem(HTREEITEM hParent, LPCTSTR strPath)
  193. {
  194. // Adding the Item to the TreeCtrl with the current Icons
  195. SHFILEINFO shFinfo;
  196. int iIcon, iIconSel;
  197.     CString    strTemp = strPath;
  198.     
  199. if ( strTemp.Right(1) != '\' )
  200.  strTemp += "\";
  201. if ( !SHGetFileInfo( strTemp,
  202. 0,
  203. &shFinfo,
  204. sizeof( shFinfo ),
  205. SHGFI_ICON | 
  206.     SHGFI_SMALLICON ) )
  207. {
  208. m_strError = "Error Gettting SystemFileInfo!";
  209. return NULL;
  210. }
  212. iIcon = shFinfo.iIcon;
  214. if ( !SHGetFileInfo( strTemp,
  215. 0,
  216. &shFinfo,
  217. sizeof( shFinfo ),
  218. SHGFI_ICON | SHGFI_OPENICON |
  219.     SHGFI_SMALLICON ) )
  220. {
  221. m_strError = "Error Gettting SystemFileInfo!";
  222. return NULL;
  223. }
  225. iIconSel = shFinfo.iIcon;
  227. if ( strTemp.Right(1) == "\" )
  228. strTemp.SetAt( strTemp.GetLength() - 1, '' );
  230. if ( hParent == TVI_ROOT )
  231. return InsertItem( strTemp, iIcon, iIconSel, hParent );
  233. return InsertItem( GetSubPath( strTemp ), iIcon, iIconSel, hParent );
  234. }
  236. LPCTSTR CDirTreeCtrl::GetSubPath(LPCTSTR strPath)
  237. {
  238. //
  239. // getting the last SubPath from a PathString
  240. // e.g. C:tempreadme.txt
  241. // the result = readme.txt
  242. static CString strTemp;
  243. int     iPos;
  245. strTemp = strPath;
  246. if ( strTemp.Right(1) == '\' )
  247.  strTemp.SetAt( strTemp.GetLength() - 1, '' );
  248. iPos = strTemp.ReverseFind( '\' );
  249. if ( iPos != -1 )
  250.     strTemp = strTemp.Mid( iPos + 1);
  252. return (LPCTSTR)strTemp;
  253. }
  255. BOOL CDirTreeCtrl::FindSubDir( LPCTSTR strPath)
  256. {
  257. //
  258. // Are there subDirs ?
  259. //
  260. CFileFind find;
  261. CString   strTemp = strPath;
  262. BOOL      bFind;
  264. if (_tcslen(strPath) == 3 )
  265. {
  266. if (strPath[1] == ':' && strPath[2] =='\')
  267. strTemp += "*.*";
  268. else
  269. strTemp += "\*.*";
  270. }
  271. else strTemp += "\*.*";
  273. bFind = find.FindFile( strTemp );
  276. while ( bFind )
  277. {
  278. bFind = find.FindNextFile();
  280. if ( find.IsDirectory() && !find.IsDots() )
  281. {
  282. return TRUE;
  283. }
  284. if ( !find.IsDirectory() && m_bFiles && !find.IsHidden() )
  285. return TRUE;
  287. }
  289. return FALSE;
  291. }
  293. void CDirTreeCtrl::OnItemexpanded(NMHDR* pNMHDR, LRESULT* pResult) 
  294. {
  295. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  296. CString strPath;
  297.  
  298. if ( pNMTreeView->itemNew.state & TVIS_EXPANDED )
  299. {
  300. //UINT uTest = TVIS_EXPANDEDONCE;
  301. ExpandItem( pNMTreeView->itemNew.hItem, TVE_EXPAND );
  302.     /*
  303. //
  304. // Delete All items
  305. // And display the subpath
  306. //
  307. HTREEITEM hChild = GetChildItem( pNMTreeView->itemNew.hItem );
  308. while ( hChild )
  309. {
  310. DeleteItem( hChild );
  311. hChild = GetChildItem( pNMTreeView->itemNew.hItem );
  312. }
  313.         
  314. strPath = GetFullPath( pNMTreeView->itemNew.hItem );
  315. DisplayPath( pNMTreeView->itemNew.hItem, strPath );
  316. */
  317. }
  318. else
  319. {
  320. //
  321. // Delete the Items, but leave one there, for 
  322. // expanding the item next time
  323. //
  324. HTREEITEM hChild = GetChildItem( pNMTreeView->itemNew.hItem );
  326. while ( hChild ) 
  327. {
  328. DeleteItem( hChild );
  329. hChild = GetChildItem( pNMTreeView->itemNew.hItem );
  330. }
  331. InsertItem( "", pNMTreeView->itemNew.hItem );
  332. }
  334. *pResult = 0;
  335. }
  337. CString CDirTreeCtrl::GetFullPath(HTREEITEM hItem)
  338. {
  339. // get the Full Path of the item
  340. CString strReturn;
  341. CString strTemp;
  342. HTREEITEM hParent = hItem;
  344. strReturn = "";
  346. while ( hParent )
  347. {
  349. strTemp  = GetItemText( hParent );
  350. strTemp += "\";
  351. strReturn = strTemp + strReturn;
  352. hParent = GetParentItem( hParent );
  353. }
  354.     if ( strReturn.Right(1) == '\' )
  355. strReturn.SetAt( strReturn.GetLength() - 1, '' );
  357.     return strReturn;
  359. }
  361. BOOL CDirTreeCtrl::SetSelPath(LPCTSTR strPath)
  362. {
  363. // Setting the Selection in the Tree
  364. HTREEITEM hParent  = TVI_ROOT;
  365. int       iLen    = strlen(strPath) + 2;
  366. char*     pszPath = new char[iLen];
  367. char*     pPath   = pszPath;
  368. BOOL      bRet    = FALSE;
  369.     
  370. if ( !IsValidPath( strPath ) )
  371. {
  372. delete [] pszPath; // this must be added 29.03.99
  373. return FALSE;
  374. }
  376. strcpy( pszPath, strPath );
  377. strupr( pszPath );
  379. if ( pszPath[strlen(pszPath)-1] != '\' )
  380. strcat( pszPath, "\" );
  381.     
  382. int iLen2 = strlen( pszPath );
  384. for (WORD i = 0; i < iLen2; i++ )
  385. {
  386. if ( pszPath[i] == '\' )
  387. {
  388. SetRedraw( FALSE );
  389. pszPath[i] = '';
  390. hParent = SearchSiblingItem( hParent, pPath );
  391. if ( !hParent )  // Not found!
  392. break;
  393. else
  394. {
  395. // Info:
  396. // the notification OnItemExpanded 
  397. // will not called every time 
  398. // after the call Expand. 
  399. // You must call Expand with TVE_COLLAPSE | TVE_COLLAPSERESET
  400. // to Reset the TVIS_EXPANDEDONCE Flag
  402. UINT uState;
  403. uState = GetItemState( hParent, TVIS_EXPANDEDONCE );
  404. if ( uState )
  405. {
  406. Expand( hParent, TVE_EXPAND );
  407. Expand( hParent, TVE_COLLAPSE | TVE_COLLAPSERESET );
  408. InsertItem("", hParent ); // insert a blank child-item
  409. Expand( hParent, TVE_EXPAND ); // now, expand send a notification
  410. }
  411. else
  412. Expand( hParent, TVE_EXPAND );
  413. }
  414. pPath += strlen(pPath) + 1;
  415. }
  416. }
  418. delete [] pszPath;
  420. if ( hParent ) // Ok the last subpath was found
  421. {
  422. SelectItem( hParent ); // select the last expanded item
  423. bRet = TRUE;
  424. }
  425. else
  426. {
  427. bRet = FALSE;
  428. }
  430. SetRedraw( TRUE );
  432.     return bRet;
  433. }
  435. HTREEITEM CDirTreeCtrl::SearchSiblingItem( HTREEITEM hItem, LPCTSTR strText)
  436. {
  437. HTREEITEM hFound = GetChildItem( hItem );
  438. CString   strTemp;
  439. while ( hFound )
  440. {
  441. strTemp = GetItemText( hFound );
  442.         strTemp.MakeUpper();
  443. if ( strTemp == strText )
  444. return hFound;
  445. hFound = GetNextItem( hFound, TVGN_NEXT );
  446. }
  448. return NULL;
  449. }
  452. void CDirTreeCtrl::ExpandItem(HTREEITEM hItem, UINT nCode)
  453. {
  454. CString strPath;
  456. if ( nCode == TVE_EXPAND )
  457. {
  458. HTREEITEM hChild = GetChildItem( hItem );
  459. while ( hChild )
  460. {
  461. DeleteItem( hChild );
  462. hChild = GetChildItem( hItem );
  463. }
  464.         
  465. strPath = GetFullPath( hItem );
  466. DisplayPath( hItem, strPath );
  467. }
  468. }
  470. BOOL CDirTreeCtrl::IsValidPath(LPCTSTR strPath)
  471. {
  472. // This function check the Pathname
  474. HTREEITEM hChild;
  475. CString   strItem;
  476. CString   strTempPath = strPath;
  477. BOOL      bFound = FALSE;
  478. CFileFind find;
  480. hChild = GetChildItem( TVI_ROOT );
  481. strTempPath.MakeUpper();
  483. while ( hChild )
  484. {
  485. strItem = GetItemText( hChild );
  486. strItem.MakeUpper();
  487. if ( strItem == strTempPath.Mid( 0, strItem.GetLength() ) )
  488. {
  489. bFound = TRUE;
  490. break;
  491. }
  492. hChild = GetNextItem( hChild, TVGN_NEXT );
  493. }
  494.     
  495. if ( !bFound )
  496. return FALSE;
  498. if ( find.FindFile( strPath ) )
  499. return TRUE;
  500.      
  501. return FALSE;
  502. }