Save and Load TreeView to a File in VB.NET

[Ensonix]

I found a post about this, and it had some C# code that wouldn't convert, and when I finally got it converted it didn't work in .NET 1.1. So I made a few changes to the converted VB version and now it works perfect, however I could find a post replay on that form, so I thought I'd put it here for anyone else searching.

'ENSONIX ENTERPRISES, INCORPORATED
'WEB:            http://www.ensonix.com
'============================================================
'CONVERTED FROM: Original C# jhTreeViewTools
'CREATED BY:     j-hannemann
'WEB:            http://www.jhmedia.de
'------------------------------------------------------------
'CONVERTED BY:   Jason Martin of EEI
'CONVERTED ON:   January 1st, 2008
'CODE VERSION:   VB.NET 2003, .NET 1.1
 
#REGION " >> IMPORTS                                       "
 
  Imports System
  Imports System.Collections
  Imports System.Windows.Forms
  Imports System.IO
  Imports System.Runtime.Serialization.Formatters.Binary
 
#END REGION
 
NameSpace Utilities
Public Class Tree
 
  #REGION " >> ENUMERATORS                                   "
 
    Public Enum ErrorCodeTypes As Integer
      None          =  0
      Serialization = -1
      File          = -2
      Exception     = -3
    End Enum
 
  #END REGION
 
  #REGION " >> SHARED METHODS                                "
 
        Public Shared Function Save(ByVal TreeViewToSave As TreeView, ByVal FileName As String) As Tree.ErrorCodeTypes
 
      Dim errorCode   As Tree.ErrorCodeTypes  = ErrorCodeTypes.None
            Dim rootNodes   As ArrayList            = Nothing
      Dim fileStream  As FileStream           = Nothing
            Dim formatter   As BinaryFormatter      = Nothing
 
      Try
 
        rootNodes   = New ArrayList()
        fileStream  = New FileStream(FileName,FileMode.Create)
        formatter   = New BinaryFormatter()
 
              For Each node As TreeNode In TreeViewToSave.Nodes
                  rootNodes.Add(node)
              Next
 
              Try
          formatter.Serialize(fileStream, rootNodes)
              Catch ex As System.Runtime.Serialization.SerializationException
                  errorCode = ErrorCodeTypes.Serialization
                  Throw(ex)
              End Try
 
      Catch ex As Exception
        errorCode = ErrorCodeTypes.Exception
        Throw(ex)
      Finally
        If Not(fileStream Is Nothing) Then fileStream.Close()
      End Try
 
             Return errorCode
 
        End Function
 
        Public Shared Function Load(ByVal TreeViewToLoad As TreeView, ByVal FileName As String) As Tree.ErrorCodeTypes
 
      Dim errorCode   As Tree.ErrorCodeTypes  = ErrorCodeTypes.None              
      Dim nodeList    As ArrayList            = Nothing
      Dim fileStream  As FileStream           = Nothing
      Dim formatter   As BinaryFormatter      = Nothing
      Dim rootNodes   As Object               = Nothing
 
      Try
 
        If File.Exists(FileName) Then
 
          fileStream  = New FileStream(FileName,FileMode.Open)
                  formatter   = New BinaryFormatter() 
 
                  Try
                      rootNodes = formatter.Deserialize(fileStream)
                  Catch ex As System.Runtime.Serialization.SerializationException
                      errorCode = ErrorCodeTypes.Serialization
            Throw(ex)
                  End Try
 
          For Each node As TreeNode In CType(rootNodes, ArrayList)
            TreeViewToLoad.Nodes.Add(node) 
          Next 
 
        Else 
          errorCode = ErrorCodeTypes.File
        End If 
 
      Catch ex As Exception
        errorCode = ErrorCodeTypes.Exception
        Throw(ex)
      Finally
        If Not(fileStream Is Nothing) Then fileStream.Close()
      End Try
 
      Return errorCode        
 
    End Function
 
  #END REGION
 
End Class
End NameSpace

Also make sure that if you are including a tag (ie TreeNode.Tag) and your Tag Object is you own class that you make it Serializable for that information to save also.

EXAMPLE

<Serializable()> _
Public Class Person
  Public NameFirst As String
  Public NameLast As String
End Class

Thanks,
Jason Martin