下面是根据http://blog.csdn.net/shuilan0066/article/details/6789204算法 写的资源管理器实现类, 其中代码借鉴了网上资源,根据需要 有所改进
实现效果如下:
准备阶段:
建立一个SDI 采用拆分窗口, 视图基类为CTreeView
头文件:TreePathView.h
// TreePathView.h : CTreePathView 类的接口 // #include "TreePathDoc.h" #pragma once class CTreePathView : public CTreeView { protected: // 仅从序列化创建 CTreePathView(); DECLARE_DYNCREATE(CTreePathView) // 属性 public: CTreePathDoc* GetDocument() const; // 操作 public: // 重写 public: virtual BOOL PreCreateWindow(CREATESTRUCT& cs); // 更改树目录的形式 protected: virtual void OnInitialUpdate(); // 构造后第一次调用 初始化树节点 // 实现 public: virtual ~CTreePathView(); #ifdef _DEBUG virtual void AssertValid() const; virtual void Dump(CDumpContext& dc) const; #endif protected: // 生成的消息映射函数 protected: DECLARE_MESSAGE_MAP() //--------------------- 需手动添加的变量函数 protected: CImageList m_imageList; HTREEITEM m_root; protected: void FindFileInDir(TCHAR * rootDir, HTREEITEM hroot, int depth); //在指定目录下寻找文件夹和文件 并将其添加到hroot节点下 CString GetFullPath(HTREEITEM hcurrent); //获得TREE控件中某节点的全路径 void ShowImage(CString imagePath); // 在Image视图中显示图像 void InitTree(); // 初始化tree 控件 void ModifyIcon(HTREEITEM htree); // 更改树节点的图标 public: afx_msg void OnTvnItemexpanding(NMHDR *pNMHDR, LRESULT *pResult); public: afx_msg void OnNMClick(NMHDR *pNMHDR, LRESULT *pResult); public: afx_msg void OnNMDblclk(NMHDR *pNMHDR, LRESULT *pResult); }; #ifndef _DEBUG // TreePathView.cpp 中的调试版本 inline CTreePathDoc* CTreePathView::GetDocument() const { return reinterpret_cast<CTreePathDoc*>(m_pDocument); } #endif
CPP 实现文件 TreePathView.CPP
// TreePathView.cpp : CTreePathView 类的实现 // #include "stdafx.h" #include "TreePath.h" #include "TreePathDoc.h" #include "TreePathView.h" #include "vector" //----------vector 需添加 #ifdef _DEBUG #define new DEBUG_NEW #endif // CTreePathView IMPLEMENT_DYNCREATE(CTreePathView, CTreeView) BEGIN_MESSAGE_MAP(CTreePathView, CTreeView) ON_NOTIFY_REFLECT(TVN_ITEMEXPANDING, &CTreePathView::OnTvnItemexpanding) ON_NOTIFY_REFLECT(NM_CLICK, &CTreePathView::OnNMClick) ON_NOTIFY_REFLECT(NM_DBLCLK, &CTreePathView::OnNMDblclk) END_MESSAGE_MAP() // CTreePathView 构造/析构 CTreePathView::CTreePathView() { // TODO: 在此处添加构造代码 } CTreePathView::~CTreePathView() { } BOOL CTreePathView::PreCreateWindow(CREATESTRUCT& cs) { // TODO: 在此处通过修改 // CREATESTRUCT cs 来修改窗口类或样式 cs.style|=TVS_HASLINES|TVS_HASBUTTONS|TVS_LINESATROOT; //若是想用CImageList的图标 ,则不要设置为TVS_HASBUTTONS形式 return CTreeView::PreCreateWindow(cs); } void CTreePathView::OnInitialUpdate() { CTreeView::OnInitialUpdate(); // CMagazineEditDoc * pDoc=this->GetDocument(); /* m_imageList.Create(12,12,ILC_COLORDDB | ILC_MASK, 3, 1); HICON hAdd=::LoadIcon(::AfxGetInstanceHandle(), (LPCTSTR)IDI_ADD); HICON hRemove=::LoadIcon(::AfxGetInstanceHandle(), (LPCTSTR)IDI_REMOVE); HICON hLeaf=::LoadIcon(::AfxGetInstanceHandle(), (LPCTSTR)IDI_LEAF); m_imageList.Add(hAdd); m_imageList.Add(hRemove); m_imageList.Add(hLeaf); GetTreeCtrl().SetImageList(&m_imageList,TVSIL_NORMAL); */ //设置节点图片 若默认,则不需要这一段语句 InitTree(); GetTreeCtrl().SetLineColor(RGB(0,0,255)); GetTreeCtrl().SetBkColor(RGB(196,226,216)); } // CTreePathView 诊断 #ifdef _DEBUG void CTreePathView::AssertValid() const { CTreeView::AssertValid(); } void CTreePathView::Dump(CDumpContext& dc) const { CTreeView::Dump(dc); } CTreePathDoc* CTreePathView::GetDocument() const // 非调试版本是内联的 { ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CTreePathDoc))); return (CTreePathDoc*)m_pDocument; } #endif //_DEBUG // CTreePathView 消息处理程序 void CTreePathView::FindFileInDir(TCHAR * rootDir, HTREEITEM hroot, int depth) { CTreeCtrl & m_treeCtrl=GetTreeCtrl(); struct Node{ TCHAR fname[256]; TCHAR filePathName[256]; }; Node directoryTemp,imageFileTemp; std::vector <Node> directory,imageFile; CString imageName; TCHAR fname[256]; WIN32_FIND_DATA fd; HANDLE hSearch; TCHAR filePathName[256]; TCHAR tmpPath[256]; BOOL bSearchFinished = FALSE; ZeroMemory(fname,256); ZeroMemory(&fd, sizeof(WIN32_FIND_DATA)); ZeroMemory(filePathName, 256); ZeroMemory(tmpPath, 256); _tcscpy(filePathName, rootDir); //兼容UNICODE ASIC copy函数 if( filePathName[_tcslen(filePathName) -1] != L'\\' ) { _tcscat(filePathName, L"\\"); _tcscat(rootDir,L"\\"); } _tcscpy(tmpPath,filePathName); _tcscat(filePathName, L"*"); hSearch = FindFirstFile(filePathName, &fd); if((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&!(fd.dwFileAttributes&FILE_ATTRIBUTE_SYSTEM)&&!(fd.dwFileAttributes&FILE_ATTRIBUTE_HIDDEN) && _tcscmp((const TCHAR *)&fd.cFileName, L".") && _tcscmp((const TCHAR *)&fd.cFileName, L"..")&&_tcscmp((const TCHAR *)&fd.cFileName,L"RECYCLER")&&_tcscmp((const TCHAR *)&fd.cFileName,L"System Volume Information")) //文件夹 { _tcscat(tmpPath, (const TCHAR *)&fd.cFileName); _tcscpy(directoryTemp.fname,reinterpret_cast<const TCHAR *>(fd.cFileName)); _tcscpy(directoryTemp.filePathName,tmpPath); directory.push_back(directoryTemp); _tcscpy(tmpPath,rootDir); if (depth==0)// 判断子目录下是否有文件夹或图像 如果有 则将其中一个加入tree控件中,目的是为了子目录前面可以显示"+" { HTREEITEM child=m_treeCtrl.GetChildItem(hroot); if (NULL==child) // 判断是否已经存在 已经存在的话就不用再添加了 ,否则Parent'Parent 扩展时 会出现连续往树控件中添加此节点的错误 m_treeCtrl.InsertItem(directoryTemp.fname,hroot,TVI_LAST); return; } } else if( _tcscmp((const TCHAR *)&fd.cFileName, L".") && _tcscmp((const TCHAR *)&fd.cFileName, L"..") ) { imageName=fd.cFileName; int dotPosition=imageName.ReverseFind(L'.'); // 反向查找 int num=imageName.GetLength()-dotPosition-1; CString temp=imageName.Right(num); if(!temp.CompareNoCase(L"jpg")||!temp.CompareNoCase(L"jpeg")||!temp.CompareNoCase(L"bmp")||!temp.CompareNoCase(L"tif")||!temp.CompareNoCase(L"tiff")) { _tcscat(tmpPath, (const TCHAR *)&fd.cFileName); _tcscpy(imageFileTemp.fname,reinterpret_cast<const TCHAR *>(fd.cFileName)); _tcscpy(imageFileTemp.filePathName,tmpPath); imageFile.push_back(imageFileTemp); _tcscpy(tmpPath,rootDir); if (depth==0)// 判断子目录下是否有文件夹或图像 如果有 则将其中一个加入tree控件中,目的是为了子目录前面可以显示"+" { HTREEITEM child=m_treeCtrl.GetChildItem(hroot); // imageName.Format("%.8s",imageFileTemp.fname); if (child==NULL) // 判断是否已经存在 已经存在的话就不用再添加了 ,否则Parent'Parent 扩展时 会出现连续往树控件中添加此节点的错误 child=m_treeCtrl.InsertItem(imageFileTemp.fname,hroot,TVI_LAST); return; } } } while( !bSearchFinished ) { if( FindNextFile(hSearch, &fd) ) { if( (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&!(fd.dwFileAttributes&FILE_ATTRIBUTE_SYSTEM) &&!(fd.dwFileAttributes&FILE_ATTRIBUTE_HIDDEN) && _tcscmp((const TCHAR *)&fd.cFileName, L".") && _tcscmp((const TCHAR *)&fd.cFileName, L"..") &&_tcscmp((const TCHAR *)&fd.cFileName,L"RECYCLER")&&_tcscmp((const TCHAR *)&fd.cFileName,L"System Volume Information")) { _tcscat(tmpPath, (const TCHAR *)&fd.cFileName); _tcscpy(directoryTemp.fname,reinterpret_cast<const TCHAR *>(fd.cFileName)); _tcscpy(directoryTemp.filePathName,tmpPath); directory.push_back(directoryTemp); _tcscpy(tmpPath,rootDir); if (depth==0)// 判断子目录下是否有文件夹或图像 如果有 则将其中一个加入tree控件中,目的是为了子目录前面可以显示"+" { HTREEITEM child=m_treeCtrl.GetChildItem(hroot); if (NULL==child) // 判断是否已经存在 已经存在的话就不用再添加了 ,否则Parent'Parent 扩展时 会出现连续往树控件中添加此节点的错误 m_treeCtrl.InsertItem(directoryTemp.fname,hroot,TVI_LAST); return; } } else if( _tcscmp((const TCHAR *)&fd.cFileName, L".") && _tcscmp((const TCHAR *)&fd.cFileName, L"..") ) { imageName=fd.cFileName; int dotPosition=imageName.ReverseFind(L'.'); int num=imageName.GetLength()-dotPosition-1; CString temp=imageName.Right(num); if(!temp.CompareNoCase(L"jpg")||!temp.CompareNoCase(L"jpeg")||!temp.CompareNoCase(L"bmp")||!temp.CompareNoCase(L"tif")||!temp.CompareNoCase(L"tiff")) { _tcscat(tmpPath, (const TCHAR *)&fd.cFileName); _tcscpy(imageFileTemp.fname,reinterpret_cast<const TCHAR *>(fd.cFileName)); _tcscpy(imageFileTemp.filePathName,tmpPath); imageFile.push_back(imageFileTemp); _tcscpy(tmpPath,rootDir); if (depth==0)// 判断子目录下是否有文件夹或图像 如果有 则将其中一个加入tree控件中,目的是为了子目录前面可以显示"+" { HTREEITEM child=m_treeCtrl.GetChildItem(hroot); if (child==NULL) // 判断是否已经存在 已经存在的话就不用再添加了 ,否则Parent'Parent 扩展时 会出现连续往树控件中添加此节点的错误 child=m_treeCtrl.InsertItem(imageFileTemp.fname,hroot,TVI_LAST); return; } } } } else { if( GetLastError() == ERROR_NO_MORE_FILES ) //Normal Finished bSearchFinished = TRUE; else bSearchFinished = TRUE; //Terminate Search } } FindClose(hSearch); HTREEITEM sub; for(unsigned int i=0;i<directory.size();i++) { CString directoryName=directory.at(i).fname; BOOL flag=FALSE; sub=m_treeCtrl.GetChildItem(hroot); while (sub!=NULL) { CString tem=m_treeCtrl.GetItemText(sub); if (m_treeCtrl.GetItemText(sub)==directoryName) { flag=TRUE; break; } sub=m_treeCtrl.GetNextItem(sub,TVGN_NEXT); } if(!flag) sub=m_treeCtrl.InsertItem(directory.at(i).fname,hroot,TVI_LAST); if (depth>0) FindFileInDir(directory.at(i).filePathName,sub,depth-1); } std::vector<Node>().swap(directory); //释放vector资源 for( unsigned int i=0;i<imageFile.size();i++) { CString fileName; fileName=imageFile.at(i).fname; // 使用CA2CT 将字符数组转换为CString串 BOOL flag=FALSE; sub=m_treeCtrl.GetChildItem(hroot); while (sub!=NULL) { if (m_treeCtrl.GetItemText(sub)==fileName) { flag=TRUE; break; } sub=m_treeCtrl.GetNextItem(sub,TVGN_NEXT); } if(!flag) { HTREEITEM hsub=m_treeCtrl.InsertItem(fileName,hroot,TVI_LAST); } } std::vector<Node>().swap(imageFile); } CString CTreePathView::GetFullPath(HTREEITEM hcurrent) { CTreeCtrl & m_treeCtrl=GetTreeCtrl(); CString fullPathName,tmpPathName; HTREEITEM hparent=hcurrent; fullPathName=m_treeCtrl.GetItemText(hcurrent); while (hparent!=m_root) { hcurrent=hparent; fullPathName=tmpPathName+fullPathName; hparent=m_treeCtrl.GetParentItem(hcurrent); tmpPathName=m_treeCtrl.GetItemText(hparent); if (tmpPathName.Right(1)!=L'\\') { tmpPathName+=L"\\"; } } tmpPathName=m_treeCtrl.GetItemText(hcurrent); if (tmpPathName==L"桌面") { TCHAR myDesktop[_MAX_PATH]; SHGetSpecialFolderPath(NULL,myDesktop,CSIDL_DESKTOPDIRECTORY,0); fullPathName.Delete(0,2); //删除根目录 桌面 fullPathName=myDesktop + fullPathName; return fullPathName; } if (tmpPathName==L"我的文档") { TCHAR myDocument[_MAX_PATH]; SHGetSpecialFolderPath(NULL,myDocument,CSIDL_PERSONAL,0); fullPathName.Delete(0,4); //删除根目录 桌面 fullPathName=myDocument + fullPathName; return fullPathName; } return fullPathName; } void CTreePathView::ShowImage(CString imagePath) //在另一个视图中显示图像 需根据需要手动更改 { /* CImageView * pView=(CImageView* )(((CMainFrame *)AfxGetMainWnd())->m_wndSplitter.GetPane(0,1)); CRect rect; pView->GetClientRect(&rect); Graphics graphics(pView->GetDC()->m_hDC); m_pImage=Image::FromFile(imagePath); graphics.DrawImage(m_pImage,0,0,rect.Width(),rect.Height()); */ } void CTreePathView::InitTree() { CTreeCtrl & treeCtrl=GetTreeCtrl(); HTREEITEM root; m_root=treeCtrl.InsertItem(reinterpret_cast<LPCTSTR>(L"我的电脑"),TVI_ROOT); TCHAR buf[100]; GetLogicalDriveStrings(sizeof(buf)/sizeof(TCHAR),buf); //获得逻辑盘信息 并将其保存在buf中 CString strDisks; for (TCHAR *s=buf;*s;s+=_tcslen(s)+1) { strDisks=s; // s为获得的每一个逻辑盘 root=treeCtrl.InsertItem(strDisks,m_root,TVI_LAST); treeCtrl.SetItemImage(root,0,0); FindFileInDir(reinterpret_cast<TCHAR *>(s),root,1); ModifyIcon(root); } // 获得桌面目录 TCHAR myDesktop[_MAX_PATH],myDocument[_MAX_PATH]; SHGetSpecialFolderPath(NULL,myDesktop,CSIDL_DESKTOPDIRECTORY,0); HTREEITEM hDesktop=treeCtrl.InsertItem(reinterpret_cast<LPCTSTR>(L"桌面"),m_root,TVI_FIRST); FindFileInDir(reinterpret_cast<TCHAR *>(myDesktop),hDesktop,1); ModifyIcon(hDesktop); //获得我的文档目录 SHGetSpecialFolderPath(NULL,myDocument,CSIDL_PERSONAL,0); HTREEITEM hmyDocument=treeCtrl.InsertItem(reinterpret_cast<LPCTSTR>(L"我的文档"),m_root,hDesktop); FindFileInDir(reinterpret_cast<TCHAR *>(myDocument),hmyDocument,1); ModifyIcon(hmyDocument); treeCtrl.Expand(m_root,TVE_COLLAPSE); } void CTreePathView::ModifyIcon(HTREEITEM htree) // 设置节点图片 若是希望默认时,则不需要这个函数 { CTreeCtrl & treeCtrl=GetTreeCtrl(); if (treeCtrl.ItemHasChildren(htree)) { if ((treeCtrl.GetItemState(htree,TVIS_EXPANDED)&TVIS_EXPANDED)==TVIS_EXPANDED) { treeCtrl.SetItemImage(htree,1,1); return; }else{ treeCtrl.SetItemImage(htree,0,0); return; } } CString strName=treeCtrl.GetItemText(htree); if(strName.Find(L'.')!=-1) treeCtrl.SetItemImage(htree,2,2); else treeCtrl.SetItemImage(htree,4,4); } void CTreePathView::OnTvnItemexpanding(NMHDR *pNMHDR, LRESULT *pResult) { LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR); // TODO: 在此添加控件通知处理程序代码 CTreeCtrl &m_treeCtrl=GetTreeCtrl(); HTREEITEM htree=pNMTreeView->itemNew.hItem; // 获得即将被扩展节点的句柄 if (htree!=m_root) { CString directorPath=GetFullPath(htree); TCHAR *path; path=reinterpret_cast<TCHAR *>(directorPath.GetBuffer(directorPath.GetLength()+1)); // getlength +1 否则内存出错 FindFileInDir(path,htree,1); directorPath.ReleaseBuffer(); } HTREEITEM hChild=m_treeCtrl.GetChildItem(htree); while (hChild!=NULL) { ModifyIcon(hChild); hChild=m_treeCtrl.GetNextItem(hChild,TVGN_NEXT); } if ((m_treeCtrl.GetItemState(htree,TVIS_EXPANDED)&TVIS_EXPANDED)==TVIS_EXPANDED) { m_treeCtrl.SetItemImage(htree,0,0); //+ }else{ m_treeCtrl.SetItemImage(htree,1,1); //- } *pResult = 0; } void CTreePathView::OnNMClick(NMHDR *pNMHDR, LRESULT *pResult) //若是改变默认+-号 则需重载此函数;若使用默认的+-号,则不要重载此函数 否则出错 { // TODO: 在此添加控件通知处理程序代码 /* CPoint pt; GetCursorPos(&pt); GetTreeCtrl().ScreenToClient(&pt); UINT uFlag = 0; HTREEITEM hCurSel = GetTreeCtrl().GetSelectedItem(); HTREEITEM hItem = GetTreeCtrl().HitTest(pt, &uFlag); //返回鼠标单击的那个节点句柄 if(hItem) { if(hCurSel != hItem) { GetTreeCtrl().SelectItem(hItem); } GetTreeCtrl().Expand(hItem,TVE_TOGGLE); ModifyIcon(hItem); } */ *pResult = 0; } void CTreePathView::OnNMDblclk(NMHDR *pNMHDR, LRESULT *pResult) // 扩展操作 ,双击时,用于显示图片等 需自己根据需要添加 { // TODO: 在此添加控件通知处理程序代码 CString tempPath; HTREEITEM htree=GetTreeCtrl().GetSelectedItem(); tempPath=GetFullPath(htree); int dotPosition=tempPath.ReverseFind(L'.'); // 反向查找 if(dotPosition!=-1) { int num=tempPath.GetLength()-dotPosition-1; CString temp=tempPath.Right(num); //temp 保存文件扩展名 if(!temp.CompareNoCase(L"jpg")||!temp.CompareNoCase(L"jpeg")||!temp.CompareNoCase(L"bmp")||!temp.CompareNoCase(L"tif")||!temp.CompareNoCase(L"tiff")) { /* m_imagePath=tempPath; ShowImage(m_imagePath); */ } } *pResult = 0; }
注: 上面的程序FindFileInDir 不严谨, 没有及时释放资源 ,导致在程序运行时,修改系统文件夹名称时 ,出现文件或文件夹不可修改 问题
一直,没有找到原因,今天终于弄清楚了,原来没有释放系统资源。
FindFirstFile、FindNextFile 查找结束后 一定要调用FindClose释放资源
程序中虽然也有调用FindClose, 但是,程序中有好几个 return ,使得程序提前退出,但提前退出时,未能FindClose释放资源,由此导致了程序运行时,一直占用着资源,而不释放,从而导致问题。
解决方案: 提前return退出时,同样需要释放系统资源
在每个return之前,添加 FindClose(hSearch)释放资源
FindClose(hSearch); // 提前退出时,需要先释放系统资源 return;
这样,在程序运行时,就可以继续对文件和文件夹进行操作了
修改后的FindFileInDir 程序如下:
void FindFileInDir(TCHAR * rootDir, HTREEITEM hroot, int depth) { CTreeCtrl & m_treeCtrl=GetTreeCtrl(); struct Node{ TCHAR fname[256]; TCHAR filePathName[256]; }; Node directoryTemp,imageFileTemp; std::vector <Node> directory,imageFile; CString imageName; TCHAR fname[256]; WIN32_FIND_DATA fd; HANDLE hSearch; TCHAR filePathName[256]; TCHAR tmpPath[256]; BOOL bSearchFinished = FALSE; ZeroMemory(fname,256); ZeroMemory(&fd, sizeof(WIN32_FIND_DATA)); ZeroMemory(filePathName, 256); ZeroMemory(tmpPath, 256); _tcscpy(filePathName, rootDir); //兼容UNICODE ASIC copy函数 if( filePathName[_tcslen(filePathName) -1] != L'\\' ) { _tcscat(filePathName, L"\\"); _tcscat(rootDir,L"\\"); } _tcscpy(tmpPath,filePathName); _tcscat(filePathName, L"*"); hSearch = FindFirstFile(filePathName, &fd); if((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&!(fd.dwFileAttributes&FILE_ATTRIBUTE_SYSTEM)&&!(fd.dwFileAttributes&FILE_ATTRIBUTE_HIDDEN) && _tcscmp((const TCHAR *)&fd.cFileName, L".") && _tcscmp((const TCHAR *)&fd.cFileName, L"..")&&_tcscmp((const TCHAR *)&fd.cFileName,L"RECYCLER")&&_tcscmp((const TCHAR *)&fd.cFileName,L"System Volume Information")) //文件夹 { _tcscat(tmpPath, (const TCHAR *)&fd.cFileName); _tcscpy(directoryTemp.fname,reinterpret_cast<const TCHAR *>(fd.cFileName)); _tcscpy(directoryTemp.filePathName,tmpPath); directory.push_back(directoryTemp); _tcscpy(tmpPath,rootDir); if (depth==0)// 判断子目录下是否有文件夹或图像 如果有 则将其中一个加入tree控件中,目的是为了子目录前面可以显示"+" { HTREEITEM child=m_treeCtrl.GetChildItem(hroot); if (NULL==child) // 判断是否已经存在 已经存在的话就不用再添加了 ,否则Parent'Parent 扩展时 会出现连续往树控件中添加此节点的错误 m_treeCtrl.InsertItem(directoryTemp.fname,hroot,TVI_LAST); FindClose(hSearch); // 提前退出时,需要先释放系统资源 return; } } else if( _tcscmp((const TCHAR *)&fd.cFileName, L".") && _tcscmp((const TCHAR *)&fd.cFileName, L"..") ) { imageName=fd.cFileName; int dotPosition=imageName.ReverseFind(L'.'); // 反向查找 int num=imageName.GetLength()-dotPosition-1; CString temp=imageName.Right(num); if(!temp.CompareNoCase(L"jpg")||!temp.CompareNoCase(L"jpeg")||!temp.CompareNoCase(L"bmp")||!temp.CompareNoCase(L"tif")||!temp.CompareNoCase(L"tiff")) { _tcscat(tmpPath, (const TCHAR *)&fd.cFileName); _tcscpy(imageFileTemp.fname,reinterpret_cast<const TCHAR *>(fd.cFileName)); _tcscpy(imageFileTemp.filePathName,tmpPath); imageFile.push_back(imageFileTemp); _tcscpy(tmpPath,rootDir); if (depth==0)// 判断子目录下是否有文件夹或图像 如果有 则将其中一个加入tree控件中,目的是为了子目录前面可以显示"+" { HTREEITEM child=m_treeCtrl.GetChildItem(hroot); // imageName.Format("%.8s",imageFileTemp.fname); if (child==NULL) // 判断是否已经存在 已经存在的话就不用再添加了 ,否则Parent'Parent 扩展时 会出现连续往树控件中添加此节点的错误 child=m_treeCtrl.InsertItem(imageFileTemp.fname,hroot,TVI_LAST); FindClose(hSearch); // 提前退出时,需要先释放系统资源 return; } } } while( !bSearchFinished ) { if( FindNextFile(hSearch, &fd) ) { if( (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&!(fd.dwFileAttributes&FILE_ATTRIBUTE_SYSTEM) &&!(fd.dwFileAttributes&FILE_ATTRIBUTE_HIDDEN) && _tcscmp((const TCHAR *)&fd.cFileName, L".") && _tcscmp((const TCHAR *)&fd.cFileName, L"..") &&_tcscmp((const TCHAR *)&fd.cFileName,L"RECYCLER")&&_tcscmp((const TCHAR *)&fd.cFileName,L"System Volume Information")) { _tcscat(tmpPath, (const TCHAR *)&fd.cFileName); _tcscpy(directoryTemp.fname,reinterpret_cast<const TCHAR *>(fd.cFileName)); _tcscpy(directoryTemp.filePathName,tmpPath); directory.push_back(directoryTemp); _tcscpy(tmpPath,rootDir); if (depth==0)// 判断子目录下是否有文件夹或图像 如果有 则将其中一个加入tree控件中,目的是为了子目录前面可以显示"+" { HTREEITEM child=m_treeCtrl.GetChildItem(hroot); if (NULL==child) // 判断是否已经存在 已经存在的话就不用再添加了 ,否则Parent'Parent 扩展时 会出现连续往树控件中添加此节点的错误 m_treeCtrl.InsertItem(directoryTemp.fname,hroot,TVI_LAST); FindClose(hSearch); // 提前退出时,需要先释放系统资源 return; } } else if( _tcscmp((const TCHAR *)&fd.cFileName, L".") && _tcscmp((const TCHAR *)&fd.cFileName, L"..") ) { imageName=fd.cFileName; int dotPosition=imageName.ReverseFind(L'.'); int num=imageName.GetLength()-dotPosition-1; CString temp=imageName.Right(num); if(!temp.CompareNoCase(L"jpg")||!temp.CompareNoCase(L"jpeg")||!temp.CompareNoCase(L"bmp")||!temp.CompareNoCase(L"tif")||!temp.CompareNoCase(L"tiff")) { _tcscat(tmpPath, (const TCHAR *)&fd.cFileName); _tcscpy(imageFileTemp.fname,reinterpret_cast<const TCHAR *>(fd.cFileName)); _tcscpy(imageFileTemp.filePathName,tmpPath); imageFile.push_back(imageFileTemp); _tcscpy(tmpPath,rootDir); if (depth==0)// 判断子目录下是否有文件夹或图像 如果有 则将其中一个加入tree控件中,目的是为了子目录前面可以显示"+" { HTREEITEM child=m_treeCtrl.GetChildItem(hroot); if (child==NULL) // 判断是否已经存在 已经存在的话就不用再添加了 ,否则Parent'Parent 扩展时 会出现连续往树控件中添加此节点的错误 child=m_treeCtrl.InsertItem(imageFileTemp.fname,hroot,TVI_LAST); FindClose(hSearch); // 提前退出时,需要先释放系统资源 return; } } } } else { if( GetLastError() == ERROR_NO_MORE_FILES ) //Normal Finished bSearchFinished = TRUE; else bSearchFinished = TRUE; //Terminate Search } } FindClose(hSearch); HTREEITEM sub; for(unsigned int i=0;i<directory.size();i++) { CString directoryName=directory.at(i).fname; BOOL flag=FALSE; sub=m_treeCtrl.GetChildItem(hroot); while (sub!=NULL) { CString tem=m_treeCtrl.GetItemText(sub); if (m_treeCtrl.GetItemText(sub)==directoryName) { flag=TRUE; break; } sub=m_treeCtrl.GetNextItem(sub,TVGN_NEXT); } if(!flag) sub=m_treeCtrl.InsertItem(directory.at(i).fname,hroot,TVI_LAST); if (depth>0) FindFileInDir(directory.at(i).filePathName,sub,depth-1); } for( unsigned int i=0;i<imageFile.size();i++) { CString fileName; fileName=imageFile.at(i).fname; // 使用CA2CT 将字符数组转换为CString串 BOOL flag=FALSE; sub=m_treeCtrl.GetChildItem(hroot); while (sub!=NULL) { if (m_treeCtrl.GetItemText(sub)==fileName) { flag=TRUE; break; } sub=m_treeCtrl.GetNextItem(sub,TVGN_NEXT); } if(!flag) { HTREEITEM hsub=m_treeCtrl.InsertItem(fileName,hroot,TVI_LAST); } } }