工控编程吧

标题: 上位机MFC实现树形菜单资源管理器源代码 [打印本页]

---

作者: qq263946146    时间: 2019-8-28 15:56
标题: 上位机MFC实现树形菜单资源管理器源代码


效果如图：
(, 下载次数: 1)

上传

点击文件名下载附件
(, 下载次数: 0)

上传

点击文件名下载附件
关键代码为

1. void CDirTreeView::AddNode(const char * path, HTREEITEM node, int type, int mode)
   
2. {
   
3. WIN32_FIND_DATA fd;
   
4. HANDLE                        hFind;
   
5. char                        buff [_MAX_PATH];        //        temporary storage
   
6. HTREEITEM                newNode = node;                //        May be used to build pathname without
   
7. int                                image1, image2;                                                                                //        adding any new nodes.
   
8. CString                        dirPath;
   
9. 
   
10. //        If a node name was passed to us then we will add the node.
    
11. //        Otherwise, we need to get a directory name first and then recurse.
    
12. //
    
13.         if (path)
    
14.         {
    
15. 
    
16. //        if the mode indicates that we are adding a drive node
    
17. //        then we need to:
    
18. //                add the drive label in parens.
    
19. //                reset the mode to tmShort (actually, on fast machines this
    
20. //                        doesn't really speed things up).
    
21.                 if (type >= DRIVE_REMOVABLE)
    
22.                 {
    
23.                 char        VolName[24];
    
24.                 char        RootName[10];
    
25.                 DWORD        dwCompLen, dwFlags;
    
26. 
    
27.                         memset (&fd, '\0', sizeof (WIN32_FIND_DATA));
    
28.                         fd.dwFileAttributes = FILE_ATTRIBUTE_SYSTEM;
    
29.                         sprintf (RootName, "%s\", path);
    
30.                         memset (VolName, '\0', sizeof (VolName));
    
31.                         strcpy (fd.cFileName, RootName);
    
32.                         GetVolumeInformation (RootName, VolName, 24, NULL, &dwCompLen, &dwFlags, NULL, 0);
    
33.                         if (strlen (VolName))
    
34.                         {
    
35.                                 wsprintf(buff, "%s (%s)", path, VolName);
    
36.                         }
    
37.                         else
    
38.                                 wsprintf (buff, "%s", path);
    
39.                         image1 = image2 = GetIconIndex (fd);
    
40.                         newNode = InsertChild (node, buff, image1, image2, TVI_LAST);
    
41.                         mode = CDirTreeView::tmShort;
    
42.                 }
    
43. //        Otherwise just use the node name passed
    
44.                 else
    
45.                 {
    
46.                         strcpy(buff, path);
    
47. 
    
48. //        Add the node as a child to the current node using the folder images.
    
49. //        Use the sort flag to sort the tree as we go.so the list is sorted
    
50. //        as we go.
    
51.                         memset (&fd, '\0', sizeof (WIN32_FIND_DATA));
    
52.                         strcpy (fd.cFileName, buff);
    
53.                         fd.dwFileAttributes = FILE_ATTRIBUTE_DIRECTORY;
    
54.                         image1 = GetIconIndex (fd);
    
55.                         fd.dwFileAttributes = FILE_ATTRIBUTE_DIRECTORY | SHGFI_OPENICON;
    
56.                         image2 = GetIconIndex (fd);
    
57.                         newNode = InsertChild (node, buff, image1, image2, TVI_SORT);
    
58.                 }
    
59.         }
    
60. 
    
61. //        Build a path name based on the node.
    
62.         BuildPath(dirPath, newNode);
    
63. //        Add wildcards
    
64.         dirPath += "*.*";
    
65. //
    
66. //        Look for the first match. Return if none.
    
67.         if ((hFind = FindFirstFile(dirPath, &fd)) == INVALID_HANDLE_VALUE)
    
68.         {
    
69.                 return;
    
70.         }
    
71. // add one to the level we are on
    
72.         m_nLevel++;
    
73.         do
    
74.         {
    
75.                 if ((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
    
76.                                 && (!(fd.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN)))
    
77.                 {
    
78.                         if (!strcmp (fd.cFileName, "."))
    
79.                         {
    
80.                                 ;                // Current directory. Do nothing.
    
81.                         }
    
82.                         else if (!strcmp (fd.cFileName, ".."))
    
83.                         {
    
84.                                 ;                // Parent directory. Do nothing.
    
85.                         }
    
86.                         else
    
87.                         {
    
88. 
    
89. //        If the are building the intial tree structure (tmShort) or we are
    
90. //        expanding a level (tmDetail), add the node by recursion.
    
91.                                 if (((mode == CDirTreeView::tmShort)
    
92.                                                 && (m_nLevel < 2))
    
93.                                                 || (mode == CDirTreeView::tmDetail))
    
94.                                         AddNode(fd.cFileName, newNode, 0, mode);        // type, 0);        // mode);
    
95. 
    
96. //        If we're building the initial structure, we want to add only one
    
97. //        subnode to make the plus symbold appear.
    
98.                                 if (mode == CDirTreeView::tmShort)
    
99.                                         break;
    
100. 
     
101. //        In the detail mode we need to fill this branch completely
     
102. //        but only one sub-node per node under this branch. Again,
     
103. //        we have to do this so the + sign shows up next to the node.
     
104.                                 if (mode == CDirTreeView::tmDetail && m_nLevel > 1)
     
105.                                 {
     
106.                                         break;
     
107.                                 }
     
108.                         }
     
109.                 }
     
110.         } while (::FindNextFile (hFind, &fd));                // Look for the next match
     
111.         FindClose (hFind);
     
112. 
     
113. // decrement the level counter
     
114.         m_nLevel--;
     
115. }

复制代码

---

欢迎光临 工控编程吧 (https://www.gkbc8.com/)

Powered by Discuz! X3.4