' 递归函数,为当前节点及其同层节点增加各自的子节点,到nLastLevel级节点结束
Private Sub AddDir(ByVal nCurIndex As Integer, ByVal nLevel As Integer, ByVal nLastLevel As Integer)
Dim nodIndex As Integer
' 已经到达结束层
If nLevel >= nLastLevel Then
Exit Sub
End If
' 为该层节点逐个增加子节点
Do While nodIndex <> TVW.Nodes(nCurIndex).LastSibling.Index
' 当前节点不是该层的最后一个节点时
' 为指定节点增加子节点
AddNodeForOne nodIndex
If TVW.Nodes(nodIndex).Children > 0 Then
' 如果有子节点,则为所有子节点加子节点
AddDir TVW.Nodes(nodIndex).Child.Index, nLevel + 1, nLastLevel
End If
' 为同层最末节点增加子节点
AddNodeForOne nodIndex
If TVW.Nodes(nodIndex).Children > 0 Then
' 如果有子节点,则为所有子节点加子节点
AddDir TVW.Nodes(nodIndex).Child.Index, nLevel + 1, nLastLevel
End If
End Sub
' 为指定节点增加子节点
Private Sub AddNodeForOne(ByVal nodIndex As Integer)
Dim nodItem As Node
Dim i As Integer
Set nodItem = TVW.Nodes(nodIndex) ' 当前节点
Dir1.Path = TVW.Nodes(nodIndex).Text ' 当前目录
i = 0
Do While i < Dir1.ListCount ' 增加子目录做子节点,以父节点的Key+i做新节点的Key
TVW.Nodes.Add nodItem.Key, tvwChild, nodItem.Key & "-" & Format(i), Dir1.List(i)
i = i + 1
Loop
nodItem.Expanded = True
End Sub
'''''以下是通用的模块
' 选取节点
Private Sub TVW_NodeCheck(ByVal Node As MSComctlLib.Node)
gCheckChildrenBySelf TVW, Node.Index, Node.Checked
gCheckParentBySibling TVW, Node.Index
End Sub
' 根据自身选取情况,确定全选或取消其下所有子项
Private Sub gCheckChildrenBySelf(TVW As TreeView, ByVal curIndex As Integer, ByVal bCh As Integer)
Dim n As Integer
If TVW.Nodes(curIndex).Children <= 0 Then
Exit Sub
Else
n = TVW.Nodes(curIndex).Child.Index
Do While n <> TVW.Nodes(curIndex).Child.LastSibling.Index
TVW.Nodes(n).Checked = bCh
gCheckChildrenBySelf TVW, n, bCh
n = TVW.Nodes(n).Next.Index
Loop
TVW.Nodes(n).Checked = bCh
gCheckChildrenBySelf TVW, n, bCh
End If
End Sub
' 根据同层、同父节点的选取情况,确定是否选取其父(乃至更上层),直至根节点
Private Sub gCheckParentBySibling(TVW As TreeView, ByVal curIndex As Integer)
Dim n As Integer
Dim bHaveChecked As Boolean
If TVW.Nodes(curIndex).FirstSibling.Index = 1 Then
Exit Sub
Else
bHaveChecked = False
n = TVW.Nodes(curIndex).FirstSibling.Index
Do While n <> TVW.Nodes(curIndex).LastSibling.Index
If TVW.Nodes(n).Checked = True Then
bHaveChecked = True
Exit Do
End If
n = TVW.Nodes(n).Next.Index
Loop
If TVW.Nodes(n).Checked = True Then
bHaveChecked = True
End If
If bHaveChecked = True Then
TVW.Nodes(curIndex).Parent.Checked = vbChecked
Else
TVW.Nodes(curIndex).Parent.Checked = vbUnchecked
End If
gCheckParentBySibling TVW, TVW.Nodes(curIndex).Parent.Index
End If
End Sub