' This is the C# implementation of Data Binding TreeView Control converted and modified 
' from VB .NET by LZF of www.codeproject.com. The original code and article, 
' writen by Duncan Mackenzie of MSDN, can be found at 
' "http://msdn.microsoft.com/library/default.asp?url=/library/en-us/dnwinforms/html/custcntrlsamp3.asp" 
' 
' Following lines are the copyright from the original code 
 
' 
' * Copyright (C) 2002 Microsoft Corporation 
' * All rights reserved. 
' * 
' * THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER 
' * EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED WARRANTIES OF 
' * MERCHANTIBILITY AND/OR FITNESS FOR A PARTICULAR PURPOSE. 
' * 
' * Date: May 2002 
' * Author: Duncan Mackenzie 
' * 
' * Requires the release version of .NET Framework 
' 
 
Imports System
Imports System.Collections
Imports System.ComponentModel
Imports System.Drawing
Imports System.Data
Imports System.Windows.Forms
 
Namespace ControlLib
  Public Class dbTreeViewCtrl
    Inherits System.Windows.Forms.TreeView
 
    Protected Overloads Overrides Sub OnPaint(ByVal pe As PaintEventArgs)
      ' TODO: Add custom paint code here 
 
      ' Calling the base class OnPaint 
      MyBase.OnPaint(pe)
    End Sub
 
    Private m_autoBuild As Boolean = True
 
    Public Property AutoBuildTree() As Boolean
      Get
        Return Me.m_autoBuild
      End Get
      Set(ByVal value As Boolean)
        Me.m_autoBuild = value
      End Set
    End Property
 
#Region "Data Binding"
    Private m_currencyManager As CurrencyManager = Nothing
    Private m_ValueMember As String
    Private m_DisplayMember As String
    Private m_oDataSource As Object
 
    <Category("Data")> _
    Public Property DataSource() As Object
      Get
        Return m_oDataSource
      End Get
      Set(ByVal value As Object)
        If value Is Nothing Then
          Me.m_currencyManager = Nothing
          Me.Nodes.Clear()
        Else
          If Not (TypeOf value Is IList OrElse TypeOf m_oDataSource Is IListSource) Then
            Throw (New System.Exception("Invalid DataSource"))
          Else
            If TypeOf value Is IListSource Then
              Dim myListSource As IListSource = DirectCast(value, IListSource)
              If myListSource.ContainsListCollection = True Then
                Throw (New System.Exception("Invalid DataSource"))
              End If
            End If
            Me.m_oDataSource = value
            Me.m_currencyManager = DirectCast(Me.BindingContext(value), CurrencyManager)
            If Me.AutoBuildTree Then
              BuildTree()
            End If
          End If
        End If
      End Set
    End Property
    ' end of DataSource property 
    <Category("Data")> _
    Public Property ValueMember() As String
      Get
        Return Me.m_ValueMember
      End Get
      Set(ByVal value As String)
        Me.m_ValueMember = value
      End Set
    End Property
 
    <Category("Data")> _
    Public Property DisplayMember() As String
      Get
        Return Me.m_DisplayMember
      End Get
      Set(ByVal value As String)
        Me.m_DisplayMember = value
      End Set
    End Property
 
    Public Function GetValue(ByVal index As Integer) As Object
      Dim innerList As IList = Me.m_currencyManager.List
      If innerList IsNot Nothing Then
        If (Me.ValueMember <> "") AndAlso (index >= 0 AndAlso 0 < innerList.Count) Then
          Dim pdValueMember As PropertyDescriptor
          pdValueMember = Me.m_currencyManager.GetItemProperties()(Me.ValueMember)
          Return pdValueMember.GetValue(innerList(index))
        End If
      End If
      Return Nothing
    End Function
 
    Public Function GetDisplay(ByVal index As Integer) As Object
      Dim innerList As IList = Me.m_currencyManager.List
      If innerList IsNot Nothing Then
        If (Me.DisplayMember <> "") AndAlso (index >= 0 AndAlso 0 < innerList.Count) Then
          Dim pdDisplayMember As PropertyDescriptor
          pdDisplayMember = Me.m_currencyManager.GetItemProperties()(Me.ValueMember)
          Return pdDisplayMember.GetValue(innerList(index))
        End If
      End If
      Return Nothing
    End Function
 
#End Region
 
#Region "Building the Tree"
 
    Private treeGroups As New ArrayList()
 
    Public Sub BuildTree()
      Me.Nodes.Clear()
      If (Me.m_currencyManager IsNot Nothing) AndAlso (Me.m_currencyManager.List IsNot Nothing) Then
        Dim innerList As IList = Me.m_currencyManager.List
        Dim currNode As TreeNodeCollection = Me.Nodes
        Dim currGroupIndex As Integer = 0
        Dim currListIndex As Integer = 0
 
 
        If Me.treeGroups.Count > currGroupIndex Then
          Dim currGroup As Group = DirectCast(treeGroups(currGroupIndex), Group)
          Dim myFirstNode As dbTreeNode = Nothing
          Dim pdGroupBy As PropertyDescriptor
          Dim pdValue As PropertyDescriptor
          Dim pdDisplay As PropertyDescriptor
 
          pdGroupBy = Me.m_currencyManager.GetItemProperties()(currGroup.GroupBy)
          pdValue = Me.m_currencyManager.GetItemProperties()(currGroup.ValueMember)
          pdDisplay = Me.m_currencyManager.GetItemProperties()(currGroup.DisplayMember)
 
          Dim currGroupBy As String = Nothing
          If innerList.Count > currListIndex Then
            Dim currObject As Object
            While currListIndex < innerList.Count
              currObject = innerList(currListIndex)
              If pdGroupBy.GetValue(currObject).ToString() <> currGroupBy Then
                currGroupBy = pdGroupBy.GetValue(currObject).ToString()
 
                myFirstNode = New dbTreeNode(currGroup.Name, pdDisplay.GetValue(currObject).ToString(), currObject, pdValue.GetValue(innerList(currListIndex)), currGroup.ImageIndex, currGroup.SelectedImageIndex, _
                currListIndex)
 
                currNode.Add(DirectCast(myFirstNode, TreeNode))
              Else
                AddNodes(currGroupIndex, currListIndex, myFirstNode.Nodes, currGroup.GroupBy)
              End If
              ' end while 
            End While
            ' end if 
          End If
        Else
          ' end if 
          While currListIndex < innerList.Count
            AddNodes(currGroupIndex, currListIndex, Me.Nodes, "")
          End While
        End If
        ' end else 
        If Me.Nodes.Count > 0 Then
          Me.SelectedNode = Me.Nodes(0)
 
        End If
      End If
      ' end if 
    End Sub
 
    Private Sub AddNodes(ByVal currGroupIndex As Integer, ByRef currentListIndex As Integer, ByVal currNodes As TreeNodeCollection, ByVal prevGroupByField As String)
      Dim innerList As IList = Me.m_currencyManager.List
      Dim pdPrevGroupBy As System.ComponentModel.PropertyDescriptor = Nothing
      Dim prevGroupByValue As String = Nothing
 
 
      Dim currGroup As Group
 
      If prevGroupByField <> "" Then
        pdPrevGroupBy = Me.m_currencyManager.GetItemProperties()(prevGroupByField)
      End If
 
      currGroupIndex += 1
 
      If treeGroups.Count > currGroupIndex Then
        currGroup = DirectCast(treeGroups(currGroupIndex), Group)
        Dim pdGroupBy As PropertyDescriptor = Nothing
        Dim pdValue As PropertyDescriptor = Nothing
        Dim pdDisplay As PropertyDescriptor = Nothing
 
        pdGroupBy = Me.m_currencyManager.GetItemProperties()(currGroup.GroupBy)
        pdValue = Me.m_currencyManager.GetItemProperties()(currGroup.ValueMember)
        pdDisplay = Me.m_currencyManager.GetItemProperties()(currGroup.DisplayMember)
 
        Dim currGroupBy As String = Nothing
 
        If innerList.Count > currentListIndex Then
          If pdPrevGroupBy IsNot Nothing Then
            prevGroupByValue = pdPrevGroupBy.GetValue(innerList(currentListIndex)).ToString()
          End If
 
          Dim myFirstNode As dbTreeNode = Nothing
          Dim currObject As Object = Nothing
 
          While (currentListIndex < innerList.Count) AndAlso (pdPrevGroupBy IsNot Nothing) AndAlso (pdPrevGroupBy.GetValue(innerList(currentListIndex)).ToString() = prevGroupByValue)
            currObject = innerList(currentListIndex)
            If pdGroupBy.GetValue(currObject).ToString() <> currGroupBy Then
              currGroupBy = pdGroupBy.GetValue(currObject).ToString()
 
              myFirstNode = New dbTreeNode(currGroup.Name, pdDisplay.GetValue(currObject).ToString(), currObject, pdValue.GetValue(innerList(currentListIndex)), currGroup.ImageIndex, currGroup.SelectedImageIndex, _
              currentListIndex)
 
              currNodes.Add(DirectCast(myFirstNode, TreeNode))
            Else
              AddNodes(currGroupIndex, currentListIndex, myFirstNode.Nodes, currGroup.GroupBy)
            End If
          End While
        End If
      Else
        Dim myNewLeafNode As dbTreeNode
        Dim currObject As Object = Me.m_currencyManager.List(currentListIndex)
 
        If (Me.DisplayMember IsNot Nothing) AndAlso (Me.ValueMember IsNot Nothing) AndAlso (Me.DisplayMember <> "") AndAlso (Me.ValueMember <> "") Then
          Dim pdDisplayloc As PropertyDescriptor = Me.m_currencyManager.GetItemProperties()(Me.DisplayMember)
          Dim pdValueloc As PropertyDescriptor = Me.m_currencyManager.GetItemProperties()(Me.ValueMember)
 
          myNewLeafNode = New dbTreeNode(IIf(Me.Tag Is Nothing, "", Me.Tag.ToString()), pdDisplayloc.GetValue(currObject).ToString(), currObject, pdValueloc.GetValue(currObject), currentListIndex)
        Else
          myNewLeafNode = New dbTreeNode("", currentListIndex.ToString(), currObject, currObject, Me.ImageIndex, Me.SelectedImageIndex, _
          currentListIndex)
        End If
 
        currNodes.Add(DirectCast(myNewLeafNode, TreeNode))
        currentListIndex += 1
      End If
    End Sub
#End Region
 
#Region "Groups"
 
    Public Sub RemoveGroup(ByVal group As Group)
      If treeGroups.Contains(group) Then
        treeGroups.Remove(group)
        If Me.AutoBuildTree Then
          BuildTree()
        End If
      End If
    End Sub
 
    Public Sub RemoveGroup(ByVal groupName As String)
      For Each group As Group In Me.treeGroups
        If group.Name = groupName Then
          RemoveGroup(group)
          Return
        End If
      Next
    End Sub
 
    Public Sub RemoveAllGroups()
      Me.treeGroups.Clear()
      If Me.AutoBuildTree Then
        Me.BuildTree()
      End If
    End Sub
 
    Public Sub AddGroup(ByVal group As Group)
      Try
        treeGroups.Add(group)
        If Me.AutoBuildTree Then
          Me.BuildTree()
        End If
      Catch e As NotSupportedException
        MessageBox.Show(e.Message)
      Catch e As System.Exception
        Throw e
      End Try
    End Sub
 
    Public Sub AddGroup(ByVal name As String, ByVal groupBy As String, ByVal displayMember As String, ByVal valueMember As String, ByVal imageIndex As Integer, ByVal selectedImageIndex As Integer)
      Dim myNewGroup As New Group(name, groupBy, displayMember, valueMember, imageIndex, selectedImageIndex)
      Me.AddGroup(myNewGroup)
    End Sub
 
    Public Function GetGroups() As Group()
      Return DirectCast(treeGroups.ToArray(Type.[GetType]("Group")), Group())
    End Function
 
#End Region
 
    Public Sub SetLeafData(ByVal name As String, ByVal displayMember As String, ByVal valueMember As String, ByVal imageIndex As Integer, ByVal selectedImageIndex As Integer)
      Me.Tag = name
      Me.DisplayMember = displayMember
      Me.ValueMember = valueMember
      Me.ImageIndex = imageIndex
      Me.SelectedImageIndex = selectedImageIndex
    End Sub
 
    Public Function IsLeafNode(ByVal node As TreeNode) As Boolean
      Return (node.Nodes.Count = 0)
    End Function
 
#Region "Keeping Everything In Sync"
 
    Public Function FindNodeByValue(ByVal value As Object) As TreeNode
      Return FindNodeByValue(value, Me.Nodes)
    End Function
 
    Public Function FindNodeByValue(ByVal Value As Object, ByVal nodesToSearch As TreeNodeCollection) As TreeNode
      Dim i As Integer = 0
      Dim currNode As TreeNode
      Dim leafNode As dbTreeNode
 
      While i < nodesToSearch.Count
        currNode = nodesToSearch(i)
        i += 1
        If currNode.LastNode Is Nothing Then
          leafNode = DirectCast(currNode, dbTreeNode)
          If leafNode.Value.ToString() = Value.ToString() Then
            Return currNode
          End If
        Else
          currNode = FindNodeByValue(Value, currNode.Nodes)
          If currNode IsNot Nothing Then
            Return currNode
          End If
        End If
      End While
 
      Return Nothing
    End Function
 
    Private Function FindNodeByPosition(ByVal posIndex As Integer) As TreeNode
      Return FindNodeByPosition(posIndex, Me.Nodes)
    End Function
 
    Private Function FindNodeByPosition(ByVal posIndex As Integer, ByVal nodesToSearch As TreeNodeCollection) As TreeNode
      Dim i As Integer = 0
      Dim currNode As TreeNode
      Dim leafNode As dbTreeNode
 
      While i < nodesToSearch.Count
        currNode = nodesToSearch(i)
        i += 1
        If currNode.Nodes.Count = 0 Then
          leafNode = DirectCast(currNode, dbTreeNode)
          If leafNode.Position = posIndex Then
            Return currNode
          Else
            currNode = FindNodeByPosition(posIndex, currNode.Nodes)
            If currNode IsNot Nothing Then
              Return currNode
            End If
          End If
        End If
      End While
      Return Nothing
    End Function
 
    Protected Overloads Overrides Sub OnAfterSelect(ByVal e As TreeViewEventArgs)
      Dim leafNode As dbTreeNode = DirectCast(e.Node, dbTreeNode)
 
      If leafNode IsNot Nothing Then
        If Me.m_currencyManager.Position <> leafNode.Position Then
          Me.m_currencyManager.Position = leafNode.Position
        End If
      End If
      ' TODO: Add MyTreeViewCtrl.OnAfterSelect implementation 
      MyBase.OnAfterSelect(e)
    End Sub
 
#End Region
 
  End Class
 
  Public Class Group
    Private groupName As String
    Private groupByMember As String
    Private groupByDisplayMember As String
    Private groupByValueMember As String
 
    Private groupImageIndex As Integer
    Private groupSelectedImageIndex As Integer
 
    Public Sub New(ByVal name As String, ByVal groupBy As String, ByVal displayMember As String, ByVal valueMember As String, ByVal imageIndex As Integer, ByVal selectedImageIndex As Integer)
      Me.ImageIndex = imageIndex
      Me.Name = name
      Me.GroupBy = groupBy
      Me.DisplayMember = displayMember
      Me.ValueMember = valueMember
      Me.SelectedImageIndex = selectedImageIndex
    End Sub
 
    Public Sub New(ByVal name As String, ByVal groupBy As String, ByVal displayMember As String, ByVal valueMember As String, ByVal imageIndex As Integer)
      Me.New(name, groupBy, displayMember, valueMember, imageIndex, imageIndex)
    End Sub
 
    Public Sub New(ByVal name As String, ByVal groupBy As String)
      Me.New(name, groupBy, groupBy, groupBy, -1, -1)
    End Sub
 
    Public Property SelectedImageIndex() As Integer
      Get
        Return groupSelectedImageIndex
      End Get
      Set(ByVal value As Integer)
        groupSelectedImageIndex = value
      End Set
    End Property
    Public Property ImageIndex() As Integer
      Get
        Return groupImageIndex
      End Get
      Set(ByVal value As Integer)
        groupImageIndex = value
      End Set
    End Property
 
    Public Property Name() As String
      Get
        Return groupName
      End Get
      Set(ByVal value As String)
        groupName = value
      End Set
    End Property
 
    Public Property GroupBy() As String
      Get
        Return groupByMember
      End Get
      Set(ByVal value As String)
        groupByMember = value
      End Set
    End Property
 
    Public Property DisplayMember() As String
      Get
        Return groupByDisplayMember
      End Get
      Set(ByVal value As String)
        groupByDisplayMember = value
      End Set
    End Property
 
    Public Property ValueMember() As String
      Get
        Return groupByValueMember
      End Get
      Set(ByVal value As String)
        groupByValueMember = value
      End Set
    End Property
  End Class
 
  Public Class dbTreeNode
    Inherits TreeNode
    Private m_groupName As String
    Private m_value As Object
    Private m_item As Object
    Private m_position As Integer
 
    Public Sub New()
    End Sub
 
    Public Sub New(ByVal groupName As String, ByVal text As String, ByVal item As Object, ByVal value As Object, ByVal imageIndex As Integer, ByVal selectedImgIndex As Integer, _
    ByVal position As Integer)
      Me.GroupName = groupName
      Me.Text = text
      Me.Item = item
      Me.Value = value
      Me.ImageIndex = imageIndex
      Me.SelectedImageIndex = selectedImgIndex
      Me.m_position = position
    End Sub
 
    Public Sub New(ByVal groupName As String, ByVal text As String, ByVal item As Object, ByVal value As Object, ByVal position As Integer)
      Me.GroupName = groupName
      Me.Text = text
      Me.Item = item
      Me.Value = value
      Me.m_position = position
    End Sub
 
    Public Property GroupName() As String
      Get
        Return m_groupName
      End Get
      Set(ByVal value As String)
        Me.m_groupName = value
      End Set
    End Property
 
    Public Property Item() As Object
      Get
        Return m_item
      End Get
      Set(ByVal value As Object)
        m_item = value
      End Set
    End Property
 
    Public Property Value() As Object
      Get
        Return m_value
      End Get
      Set(ByVal value As Object)
        m_value = value
      End Set
    End Property
 
    Public ReadOnly Property Position() As Integer
      Get
        Return m_position
      End Get
    End Property
  End Class
 
End Namespace