树状结构类的封装CDirTreeCtrl
Posted Jane_bai
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了树状结构类的封装CDirTreeCtrl相关的知识，希望对你有一定的参考价值。
1 // DirTreeCtrl.cpp : implementation file 2 // 3 4 #include "stdafx.h" 5 #include "DirTreeCtrl.h" 6 7 #ifdef _DEBUG 8 #define new DEBUG_NEW 9 #undef THIS_FILE 10 static char THIS_FILE[] = __FILE__; 11 #endif 12 13 ///////////////////////////////////////////////////////////////////////////// 14 // CDirTreeCtrl 15 16 CDirTreeCtrl::CDirTreeCtrl() 17 { 18 } 19 20 CDirTreeCtrl::~CDirTreeCtrl() 21 { 22 } 23 24 25 BEGIN_MESSAGE_MAP(CDirTreeCtrl, CTreeCtrl) 26 //{{AFX_MSG_MAP(CDirTreeCtrl) 27 ON_NOTIFY_REFLECT(TVN_ITEMEXPANDED, OnItemexpanded) 28 //}}AFX_MSG_MAP 29 END_MESSAGE_MAP() 30 31 ///////////////////////////////////////////////////////////////////////////// 32 // CDirTreeCtrl message handlers 33 HTREEITEM CDirTreeCtrl::AddItem( HTREEITEM hParent, CString strPath) 34 { 35 SHFILEINFO shFinfo; 36 int nIcon, nSelIcon; 37 CString strTemp = strPath; 38 39 if( strTemp.Right(1) != ‘\\‘ ) 40 strTemp += ‘\\‘; 41 42 if( !SHGetFileInfo( strTemp, 43 0, 44 &shFinfo, 45 sizeof(shFinfo), 46 SHGFI_ICON | SHGFI_SMALLICON ) ) 47 { 48 m_strError = "Error Gettting SystemFileInfo!"; 49 return NULL; 50 } 51 nIcon = shFinfo.iIcon; 52 53 DestroyIcon( shFinfo.hIcon ); 54 55 if( !SHGetFileInfo( strTemp, 56 0, 57 &shFinfo, 58 sizeof(shFinfo), 59 SHGFI_ICON | SHGFI_OPENICON | 60 SHGFI_SMALLICON ) ) 61 { 62 m_strError = "Error Getting SystemFileError!"; 63 return NULL; 64 } 65 nSelIcon = shFinfo.iIcon; 66 DestroyIcon( shFinfo.hIcon ); 67 68 strTemp.TrimRight(‘\\‘); 69 m_strGloble = strTemp; 70 71 if( hParent == TVI_ROOT ) 72 return InsertItem( strTemp, nIcon, nSelIcon, hParent ); 73 74 return InsertItem( GetSubPath( strTemp ), nIcon, nSelIcon, hParent ); 75 } 76 //HTREEITEM CDirTreeCtrl::AddItem( HTREEITEM hParent, LPCTSTR strPath) 77 //{ 78 // SHFILEINFO shFinfo; 79 // int nIcon, nSelIcon; 80 // CString strTemp = strPath; 81 // 82 // if( strTemp.Right(1) != ‘\\‘ ) 83 // strTemp += ‘\\‘; 84 // 85 // if( !SHGetFileInfo( strTemp, 86 // 0, 87 // &shFinfo, 88 // sizeof(shFinfo), 89 // SHGFI_ICON | SHGFI_SMALLICON ) ) 90 // { 91 // m_strError = "Error Gettting SystemFileInfo!"; 92 // return NULL; 93 // } 94 // nIcon = shFinfo.iIcon; 95 // 96 // DestroyIcon( shFinfo.hIcon ); 97 // 98 // if( !SHGetFileInfo( strTemp, 99 // 0, 100 // &shFinfo, 101 // sizeof(shFinfo), 102 // SHGFI_ICON | SHGFI_OPENICON | 103 // SHGFI_SMALLICON ) ) 104 // { 105 // m_strError = "Error Getting SystemFileError!"; 106 // return NULL; 107 // } 108 // nSelIcon = shFinfo.iIcon; 109 // DestroyIcon( shFinfo.hIcon ); 110 // 111 // strTemp.TrimRight(‘\\‘); 112 // 113 // if( hParent == TVI_ROOT ) 114 // return InsertItem( strTemp, nIcon, nSelIcon, hParent ); 115 // 116 // return InsertItem( GetSubPath( strTemp ), nIcon, nSelIcon, hParent ); 117 //} 118 BOOL CDirTreeCtrl::DisplayDrives() 119 { 120 DeleteAllItems(); 121 122 char szDrives[128]; 123 char *pDrive; 124 //LPWSTR LPDriveStr = CA2W(szDrives); 125 if( !GetLogicalDriveStringsA( sizeof(szDrives), szDrives)) //byy 126 { 127 m_strError = " Error Getting Logical DriveStrings!"; 128 return FALSE; 129 } 130 pDrive = szDrives; 131 while( *pDrive ) 132 { 133 ////LPCTSTR lptstr = (LPCTSTR)pDrive; 134 //int charlength = strlen(pDrive); 135 //int len = MultiByteToWideChar(CP_ACP, 0, pDrive, charlength, NULL, 0); 136 //wchar_t *pWChar = new wchar_t[len + 1]; //为宽字节字符数申请空间， 137 //MultiByteToWideChar(CP_ACP, 0, pDrive, charlength, pWChar, len); //多字节编码转换成宽字节编码 138 //pWChar[len] = ‘\0‘; 139 ////将wchar_t数组转换为CString 140 CString str; 141 str.Append(pDrive); 142 143 HTREEITEM hParent = AddItem( TVI_ROOT, str); //-byy 144 if (FindSubDir(str)) 145 InsertItem(_T(""), 0, 0, hParent); 146 pDrive += strlen( pDrive ) + 1; 147 // [7/11/2016 B01974] 148 //delete[] pWChar; 149 } 150 151 return TRUE; 152 153 } 154 155 // 156 157 BOOL CDirTreeCtrl::FindSubDir(CString strPath) 158 { 159 CFileFind finder; 160 //CString strTemp = strPath; 161 CString strTemp = m_strGloble; 162 BOOL bContinue; 163 164 if (strTemp[strTemp.GetLength()-1] == ‘\\‘) 165 strTemp += "*.*"; 166 else 167 strTemp += "\\*.*"; 168 169 bContinue = finder.FindFile(strTemp); 170 171 while(bContinue) 172 { 173 bContinue = finder.FindNextFile(); 174 175 if ( finder.IsDirectory() && !finder.IsDots() ) 176 return TRUE; 177 178 179 if ( !finder.IsDirectory() && m_bFiles && !finder.IsDots() ) 180 return TRUE; 181 } 182 183 finder.Close(); 184 return FALSE; 185 186 } 187 188 // 189 190 BOOL CDirTreeCtrl::DisplayTree(LPCTSTR strRoot, BOOL bFiles) 191 { 192 DWORD dwStyle = GetStyle(); 193 if ( dwStyle & TVS_EDITLABELS ) 194 { 195 ModifyStyle( TVS_EDITLABELS, 0 ); 196 } 197 198 DeleteAllItems(); 199 200 if ( !GetSysImgList() ) 201 return FALSE; 202 203 m_bFiles = bFiles; 204 205 if ( strRoot == NULL || strRoot[0] == ‘\0‘ ) 206 { 207 if ( !DisplayDrives() ) 208 return FALSE; 209 m_strRoot = ""; 210 } 211 else 212 { 213 m_strRoot = strRoot; 214 215 if ( m_strRoot.Right(1) != ‘\\‘ ) 216 m_strRoot += "\\"; 217 // [4/18/2018 baiyangyang] 218 HTREEITEM hParent = AddItem( TVI_ROOT, m_strRoot ); 219 //HTREEITEM hParent; 220 //SetCurrentPath(hParent, m_strRoot); 221 DisplayPath( hParent, strRoot ); 222 } 223 return TRUE; 224 } 225 void CDirTreeCtrl::DisplayPath( HTREEITEM hParent, LPCTSTR strPath) 226 { 227 CFileFind finder; 228 CString strTempPath = strPath; 229 BOOL bContinue; 230 231 CStringArray strDirArray; 232 CStringArray strFileArray; 233 234 if ( strTempPath.Right(1) != ‘\\‘ ) 235 strTempPath += "\\*.*"; 236 else 237 strTempPath += "*.*"; 238 239 bContinue = finder.FindFile( strTempPath ); 240 241 while ( bContinue ) 242 { 243 bContinue = finder.FindNextFile(); 244 245 if ( finder.IsDirectory() && !finder.IsDots() ) 246 { 247 strDirArray.Add( finder.GetFilePath() ); 248 TRACE(finder.GetFilePath()+"\n"); 249 } 250 251 if ( !finder.IsDirectory() && m_bFiles && !finder.IsDots()) 252 strFileArray.Add( finder.GetFilePath() ); 253 } 254 255 SetRedraw( FALSE ); 256 CWaitCursor wait; 257 258 for ( int i = 0; i < strDirArray.GetSize(); i++ ) 259 { 260 HTREEITEM hItem = AddItem( hParent, strDirArray.GetAt(i) ); 261 262 if ( FindSubDir( strDirArray.GetAt(i))) 263 InsertItem(_T(""), 0, 0, hItem ); 264 } 265 266 if ( m_bFiles ) 267 { 268 for (int i = 0; i < strFileArray.GetSize(); i++ ) 269 { 270 AddItem( hParent, strFileArray.GetAt(i) ) ; 271 } 272 } 273 274 SetRedraw( TRUE ); 275 276 finder.Close(); 277 } 278 279 LPCTSTR CDirTreeCtrl::GetSubPath(LPCTSTR strPath) 280 { 281 static CString strTemp; 282 int nPos; 283 284 strTemp = strPath; 285 286 if ( strTemp.GetAt( strTemp.GetLength() - 1) == ‘\\‘ ) 287 strTemp.SetAt( strTemp.GetLength() - 1,‘\0‘ ); 288 289 nPos = strTemp.ReverseFind(‘\\‘); 290 291 if ( nPos != -1 ) 292 strTemp = strTemp.Mid( nPos + 1 ); 293 294 return (LPCTSTR)strTemp; 295 } 296 297 // 298 299 BOOL CDirTreeCtrl::GetSysImgList() 300 { 301 SHFILEINFO shFinfo; 302 HIMAGELIST hImgList = NULL; 303 304 if ( GetImageList( TVSIL_NORMAL ) ) 305 m_imgList.Detach(); 306 307 hImgList = (HIMAGELIST)SHGetFileInfo(_T("C:\\"), 308 0, 309 &shFinfo, 310 sizeof( shFinfo ), 311 SHGFI_SYSICONINDEX | 312 SHGFI_SMALLICON ); 313 314 if ( !hImgList ) 315 { 316 m_strError = "Cannot retrieve the Handle of SystemImageList"; 317 return FALSE; 318 } 319 320 321 m_imgList.m_hImageList = hImgList; 322 323 SetImageList( &m_imgList, TVSIL_NORMAL ); 324 return TRUE; 325 326 } 327 328 CString CDirTreeCtrl::GetFullPath(HTREEITEM hItem) 329 { 330 CString strReturn = _T(""), strTemp; 331 332 HTREEITEM hParent = hItem; 333 334 while( hParent ) 335 { 336 strTemp = GetItemText( hParent ); 337 strTemp += "\\"; 338 strReturn = strTemp + strReturn; 339 hParent = GetParentItem( hParent ); 340 } 341 342 strReturn.TrimRight( ‘\\‘ ); 343 344 return strReturn; 345 } 346 347 // 348 BOOL CDirTreeCtrl::SetSelPath(LPCTSTR strPath) 349 { 350 HTREEITEM hParent = TVI_ROOT; 351 int iLen = strlen(strPath) + 2; 352 char* pszPath = new char[iLen]; 353 char* pPath = pszPath; 354 BOOL bRet = FALSE; 355 356 if ( !IsValidPath( strPath ) ) 357 { 358 delete [] pszPath; // this must be added 29.03.99 359 return FALSE; 360 } 361 strcpy( pszPath, strPath ); 362 _strupr( pszPath); 363 364 if ( pszPath[strlen(pszPath)-1] != ‘\\‘ ) 365 strcat( pszPath, "\\" ); 366 367 int iLen2 = strlen( pszPath ); 368 369 for (WORD i = 0; i < iLen2; i++ ) 370 { 371 if ( pszPath[i] == ‘\\‘ ) 372 { 373 SetRedraw( FALSE ); 374 pszPath[i] = ‘\0‘; 375 hParent = SearchSiblingItem( hParent, pPath ); 376 if ( !hParent ) // Not found! 377 break; 378 else 379 { 380 // Info: 381 // the notification OnItemExpanded 382 // will not called every time 383 // after the call Expand. 384 // You must call Expand with TVE_COLLAPSE | TVE_COLLAPSERESET 385 // to Reset the TVIS_EXPANDEDONCE Flag 386 387 UINT uState; 388 uState = GetItemState( hParent, TVIS_EXPANDEDONCE ); 389 if ( uState ) 390 { 391 Expand( hParent, TVE_EXPAND ); 392 Expand( hParent, TVE_COLLAPSE | TVE_COLLAPSERESET ); 393 InsertItem("", hParent ); // insert a blank child-item 394 Expand( hParent, TVE_EXPAND ); // now, expand send a notification 395 } 396 else 397 Expand( hParent, TVE_EXPAND ); 398 } 399 pPath += strlen(pPath) + 1; 400 } 401 } 402 403 delete [] pszPath; 404 405 if ( hParent ) // Ok the last subpath was found 406 { 407 SelectItem( hParent ); // select the last expanded item 408 bRet = TRUE; 409 } 410 else 411 { 412 bRet = FALSE; 413 } 414 415 SetRedraw( TRUE ); 416 417 return bRet; 418 } 419 420 void CDirTreeCtrl::ExpandItem( HTREEITEM hItem, UINT nCode) 421 { 422 CString strPath; 423 424 if ( nCode == TVE_EXPAND ) 425 { 426 HTREEITEM hChild = GetChildItem( hItem ); 427 while ( hChild ) 428 { 429 DeleteItem( hChild ); 430 hChild = GetChildItem( hItem ); 431 } 432 433 strPath = GetFullPath( hItem ); 434 DisplayPath( hItem, strPath ); 435 } 436 } 437 438 void CDirTreeCtrl::OnItemexpanded(NMHDR* pNMHDR, LRESULT* pResult) 439 { 440 NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR; 441 // TODO: Add your control notification handler code here 442 CString strPath; 443 444 if ( pNMTreeView->itemNew.state & TVIS_EXPANDED ) 445 { 446 ExpandItem( pNMTreeView->itemNew.hItem, TVE_EXPAND ); 447 } 448 else 449 { 450 HTREEITEM hChild = GetChildItem( pNMTreeView->itemNew.hItem ); 451 452 while ( hChild ) 453 { 454 DeleteItem( hChild ); 455 hChild = GetChildItem( pNMTreeView->itemNew.hItem ); 456 } 457 458 InsertItem(_T(""), pNMTreeView->itemNew.hItem ); 459 } 460 *pResult = 0; 461 } 462 HTREEITEM CDirTreeCtrl::SearchSiblingItem( HTREEITEM hItem, LPCTSTR strText) 463 { 464 HTREEITEM hFound = GetChildItem( hItem ); 465 CString strTemp; 466 while ( hFound ) 467 { 468 strTemp = GetItemText( hFound ); 469 strTemp.MakeUpper(); 470 if ( strTemp == strText ) 471 return hFound; 472 hFound = GetNextItem( hFound, TVGN_NEXT ); 473 } 474 475 return NULL; 476 } 477 BOOL CDirTreeCtrl::IsValidPath(LPCTSTR strPath) 478 { 479 // This function check the Pathname 480 481 HTREEITEM hChild; 482 CString strItem; 483 CString strTempPath = strPath; 484 BOOL bFound = FALSE; 485 CFileFind find; 486 487 hChild = GetChildItem( TVI_ROOT ); 488 strTempPath.MakeUpper(); 489 strTempPath.TrimRight(‘\\‘); 490 491 while ( hChild ) 492 { 493 strItem = GetItemText( hChild ); 494 strItem.MakeUpper(); 495 if ( strItem == strTempPath.Mid( 0, strItem.GetLength() ) ) 496 { 497 bFound = TRUE; 498 break; 499 } 500 hChild = GetNextItem( hChild, TVGN_NEXT ); 501 } 502 503 if ( !bFound ) 504 return FALSE; 505 506 strTempPath += "\\nul"; 507 if ( find.FindFile( strTempPath ) ) 508 return TRUE; 509 510 return FALSE; 511 }
1 #if !defined(AFX_DIRTREECTRL_H__0F2536D9_CBE2_4DF2_83F4_621FFD9593C9__INCLUDED_) 2 #define AFX_DIRTREECTRL_H__0F2536D9_CBE2_4DF2_83F4_621FFD9593C9__INCLUDED_ 3 4 #if _MSC_VER > 1000 5 #pragma once 6 #endif // _MSC_VER > 1000 7 // DirTreeCtrl.h : header file 8 // 9 10 ///////////////////////////////////////////////////////////////////////////// 11 // CDirTreeCtrl window 12 13 class CDirTreeCtrl : public CTreeCtrl 14 { 15 // Construction 16 public: 17 CDirTreeCtrl(); 18 19 // Attributes 20 private: 21 LPCTSTR GetSubPath(LPCTSTR strPath); 22 BOOL m_bFiles; 23 CString m_strError; 24 CString m_strRoot; 25 CImageList m_imgList; 26 void ExpandItem( HTREEITEM hItem, UINT nCode); 27 void DisplayPath( HTREEITEM hParent, LPCTSTR strPath); 28 BOOL DisplayDrives(); 29 void GetFormatList(CString csPath, CStringArray &aryDwBitmaps); 30 //HTREEITEM AddItem( HTREEITEM hParent, LPCTSTR strPath ); 31 HTREEITEM AddItem( HTREEITEM hParent, CString strPath); 32 //BOOL FindSubDir( LPCTSTR strPath ); 33 BOOL FindSubDir( CString strPath ); 34 BOOL GetSysImgList(); 35 CString m_strGloble; 36 37 38 // Operations 39 public: 40 41 // Overrides 42 // ClassWizard generated virtual function overrides 43 //{{AFX_VIRTUAL(CDirTreeCtrl) 44 //}}AFX_VIRTUAL 45 46 // Implementation 47 public: 48 BOOL SetSelPath(LPCTSTR strPath); 49 BOOL DisplayTree(LPCTSTR strRoot, BOOL bFiles); 50 CString GetFullPath(HTREEITEM hItem); 51 virtual ~CDirTreeCtrl(); 52 53 // Generated message map functions 54 protected: 55 BOOL IsValidPath( LPCTSTR strPath ); 56 HTREEITEM SearchSiblingItem( HTREEITEM hItem, LPCTSTR strText ); 57 //{{AFX_MSG(CDirTreeCtrl) 58 afx_msg void OnItemexpanded(NMHDR* pNMHDR, LRESULT* pResult); 59 //}}AFX_MSG 60 61 DECLARE_MESSAGE_MAP() 62 }; 63 64 ///////////////////////////////////////////////////////////////////////////// 65 66 //{{AFX_INSERT_LOCATION}} 67 // Microsoft Visual C++ will insert additional declarations immediately before the previous line. 68 69 #endif // !defined(AFX_DIRTREECTRL_H__0F2536D9_CBE2_4DF2_83F4_621FFD9593C9__INCLUDED_)
目前很多项目用到的基础类代码，供大家学习
以上是关于树状结构类的封装CDirTreeCtrl的主要内容，如果未能解决你的问题，请参考以下文章
一个队asp.net session进行了再次封装的C#类的代码
VSCode自定义代码片段14——Vue的axios网络请求封装
VSCode自定义代码片段14——Vue的axios网络请求封装