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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 绘图程序 > 查看源码
ChooseDirDlg.cpp
资源名称：MiniCAD_ALL.zip　[点击查看]
上传用户：netltd
上传日期：2013-02-12
资源大小：7234k
文件大小：7k
源码类别：

绘图程序

开发平台：

Visual C++

ChooseDirDlg.cpp：源码内容
  1. // ChooseDirDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "resource.h"
  6. #include "ChooseDirDlg.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CChooseDirDlg dialog
  18. CChooseDirDlg::CChooseDirDlg(CWnd* pParent /*=NULL*/)
  19. : CDialog(CChooseDirDlg::IDD, pParent)
  20. {
  21. //{{AFX_DATA_INIT(CChooseDirDlg)
  22. m_Path = _T("");
  23. //}}AFX_DATA_INIT
  24. }
  27. void CChooseDirDlg::DoDataExchange(CDataExchange* pDX)
  28. {
  29. CDialog::DoDataExchange(pDX);
  30. //{{AFX_DATA_MAP(CChooseDirDlg)
  31. DDX_Control(pDX, IDC_DIRECTORY, m_Directory);
  32. DDX_Text(pDX, IDC_PATH, m_Path);
  33. //}}AFX_DATA_MAP
  34. }
  37. BEGIN_MESSAGE_MAP(CChooseDirDlg, CDialog)
  38. //{{AFX_MSG_MAP(CChooseDirDlg)
  39. ON_NOTIFY(TVN_ITEMEXPANDING, IDC_DIRECTORY, OnItemexpandingDirectory)
  40. ON_NOTIFY(TVN_SELCHANGED, IDC_DIRECTORY, OnSelchangedDirectory)
  41. //}}AFX_MSG_MAP
  42. END_MESSAGE_MAP()
  44. /////////////////////////////////////////////////////////////////////////////
  45. // CChooseDirDlg message handlers
  47. //获得选定的路径
  48. CString CChooseDirDlg::GetDirectory(CString path)
  49. {
  50. m_Path = path;
  51. DoModal();
  52. return m_Path;
  53. }
  55. int CChooseDirDlg::DoModal() 
  56. {
  57. // TODO: Add your specialized code here and/or call the base class
  58. //加入图标
  59. m_ImageList.Create(16, 16, ILC_MASK, 2, 2);
  60. m_ImageList.Add(AfxGetApp()->LoadIcon(IDI_MYCOMPUTER));
  61. m_ImageList.Add(AfxGetApp()->LoadIcon(IDI_FLOPPY));
  62.     m_ImageList.Add(AfxGetApp()->LoadIcon(IDI_HARDDISK));
  63.     m_ImageList.Add(AfxGetApp()->LoadIcon(IDI_CDROM));
  64.     m_ImageList.Add(AfxGetApp()->LoadIcon(IDI_CLSFOLD));
  65.     m_ImageList.Add(AfxGetApp()->LoadIcon(IDI_OPENFOLD));
  67. return CDialog::DoModal();
  68. }
  71. BOOL CChooseDirDlg::OnInitDialog() 
  72. {
  73. CDialog::OnInitDialog();
  75. // TODO: Add extra initialization here
  77. HTREEITEM hLevel;
  79. //初始化目录列表框
  80. m_Directory.SetImageList(&m_ImageList, TVSIL_NORMAL);
  81. hLevelRoot = m_Directory.InsertItem("我的电脑",
  82. 0, 0, TVI_ROOT, TVI_SORT);
  84.     char strDriver[_MAX_PATH];   
  85. char temp[_MAX_PATH];
  86. int index =0 ;
  87. int IconIndex;
  88. CString DriverName;
  90. if(GetLogicalDriveStrings(MAX_PATH, strDriver)==0)
  91. return TRUE;
  93. for(;;)
  94. {
  95. strncpy(temp, strDriver+index, 2);
  96. //加入字符串结束标志
  97. temp[2] = NULL;
  98. index = index + 4;
  100. switch( GetDriveType(temp) )
  101. {
  102. case DRIVE_REMOVABLE:
  103. IconIndex = 1;
  104. break;
  106. case DRIVE_CDROM:
  107.             IconIndex = 3;
  108. break;
  110. default:
  111. IconIndex = 2;
  113. }
  115. DriverName = temp;
  116. DriverName.MakeUpper();
  117. //不显示 A 驱
  118. if(DriverName != "A:")
  119. {
  120. hLevel= m_Directory.InsertItem((LPCTSTR)DriverName, 
  121. IconIndex, IconIndex, hLevelRoot);
  122. FindDir(hLevel, DriverName);
  123. }
  124.         if(strDriver[index] == NULL)
  125. break;
  126. }
  127. m_Directory.Expand(hLevelRoot, TVE_EXPAND);
  129. ExpandPath(m_Path);
  130. SetDlgItemText(IDC_PATH, m_Path);
  131. return TRUE;  // return TRUE unless you set the focus to a control
  132.               // EXCEPTION: OCX Property Pages should return FALSE
  133. }
  135. void CChooseDirDlg::FindDir(HTREEITEM level, CString strDirect)
  136. {
  137. HANDLE hFind;
  138. WIN32_FIND_DATA fd;
  139. HTREEITEM Childlevel;
  140. CString direct;
  141. CString filename;
  142.     strDirect +=  "\*.*";
  143. //查找指定目录下的的所有文件,包括子目录
  144. if( ( hFind = FindFirstFile(strDirect  , &fd) ) == INVALID_HANDLE_VALUE )
  145. return;
  147. while(TRUE)
  148. {
  149. if( (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0)
  150. {
  151. //不显示".",".."这两个目录
  152. if(fd.cFileName[0] == '.')
  153. ;
  154. else
  155. {
  156. filename = fd.cFileName;
  157. Childlevel = m_Directory.InsertItem((LPCSTR)filename, 
  158. 4, 5, level);
  159. }
  160. }
  161.         //查找下一个文件 
  162. if(FindNextFile(hFind, &fd) == 0)
  163. {
  164. FindClose(hFind);
  165. //对level的子项排序
  166. m_Directory.SortChildren(level);
  167. break;
  168. }
  169. }
  170. }
  172. void CChooseDirDlg::OnItemexpandingDirectory(NMHDR* pNMHDR, LRESULT* pResult) 
  173. {
  174. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  175. // TODO: Add your control notification handler code here
  177. HTREEITEM hLevel;
  178. hLevel = pNMTreeView->itemNew.hItem;
  179. if( pNMTreeView->action == TVE_EXPAND)
  180. {
  181. if((hLevel = m_Directory.GetChildItem(hLevel)) == NULL)
  182. return;
  183. else
  184. if(!m_Directory.ItemHasChildren(hLevel))
  185. FindDir(hLevel, GetstrDirect(hLevel));
  186. while((hLevel = m_Directory.GetNextSiblingItem(hLevel))
  187. !=NULL)
  188. {
  189. if(!m_Directory.ItemHasChildren(hLevel))
  190. FindDir(hLevel, GetstrDirect(hLevel));
  191. }
  192. }
  194. *pResult = 0;
  195. }
  197. /*保留
  198. void CChooseDirDlg::FindAllDir(HTREEITEM level,  CString strDirect)
  199. {
  200. HTREEITEM parentlevel;
  201. HTREEITEM friendlevel;
  202. CString str;
  203. CString filename;
  205. parentlevel = level;
  206. str = strDirect;
  207. while(TRUE)
  208. {
  209.   FindDir(parentlevel, str);
  210.   if(!m_Directory.ItemHasChildren(parentlevel))
  211.   {
  212.   if(parentlevel==level)
  213.   return;
  214.   filename = m_Directory.GetItemText(parentlevel);
  215.   //have problem,use "Delete" function, such as
  216.   //CString s1 = "abcaabd", s2 = "abc";
  217.   //s1.TrimLeft((LPCTSTR)s2);
  218.   //s1={"bd"};
  220.   str.TrimRight((LPCTSTR)filename);
  221.   str.TrimRight("\");
  223.   while((friendlevel = m_Directory.GetNextSiblingItem(parentlevel))
  224.  ==NULL)
  225.   {
  226.   parentlevel = m_Directory.GetParentItem(parentlevel);
  227.   filename = m_Directory.GetItemText(parentlevel);
  228.       str.TrimRight((LPCTSTR)filename);
  229.       str.TrimRight("\");
  230.   if(parentlevel==level)
  231.  return;
  232.   }
  234.           parentlevel = friendlevel;
  235.   filename = m_Directory.GetItemText(parentlevel);
  236.   str = str + "\" + filename;
  237.   continue;
  238.   }
  240.   parentlevel = m_Directory.GetChildItem(parentlevel);
  241.   str += "\" + m_Directory.GetItemText(parentlevel);
  242. }
  243. }
  244. */
  246. CString CChooseDirDlg::GetstrDirect(HTREEITEM level)
  247. {
  248. CString str;
  249. str = m_Directory.GetItemText(level);
  250. while((level = m_Directory.GetParentItem(level)) != hLevelRoot)
  251. {
  252. str = m_Directory.GetItemText(level) + "\" + str;
  253. }
  254. return str;
  255. }
  257. void CChooseDirDlg::ExpandPath(CString path)
  258. {
  259. CString str;
  260. CString s;
  261. int index;
  262. HTREEITEM level;
  263. level = m_Directory.GetChildItem(hLevelRoot);
  265. while((index = path.Find("\")) != -1)
  266. {
  267. str = path.Left(index);
  268. path.Delete(0, index+1);
  269. if(str == m_Directory.GetItemText(level))
  270. m_Directory.Expand(level, TVE_EXPAND);
  271. else
  272. while(TRUE)
  273. {
  274. if((level =  m_Directory.GetNextSiblingItem(level))
  275. ==NULL)
  276. return;
  277. s = m_Directory.GetItemText(level);
  278. if(str.CompareNoCase((LPCTSTR)m_Directory.GetItemText(level))
  279. == 0)
  280. {
  282. m_Directory.Expand(level, TVE_EXPAND);
  283. level = m_Directory.GetChildItem(level);
  284. break;
  285. }
  286. }
  287. }
  288. }
  290. void CChooseDirDlg::OnSelchangedDirectory(NMHDR* pNMHDR, LRESULT* pResult) 
  291. {
  292. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  293. // TODO: Add your control notification handler code here
  295. //当被选择的目录改变后,改变EDIT 控件的内容
  296. HTREEITEM hLevel;
  297. CString str;
  298. hLevel = pNMTreeView->itemNew.hItem;
  299. if(hLevel == hLevelRoot)
  300. return;
  301. str = GetstrDirect(hLevel);
  302. if(str.Find("\") == -1)
  303. str += "\";
  304. m_Path = str;
  305. UpdateData(FALSE);
  307. *pResult = 0;
  308. }