Imports INFITF

Imports KnowledgewareTypeLib

Imports MECMOD

Imports PARTITF

Imports ProductStructureTypeLib

Imports SPATypeLib

Imports System.IO



Public Class UserForm1

    Public CATIA As INFITF.Application



    Private Sub Form1_Load(sender As Object, e As EventArgs) Handles MyBase.Load



        On Error Resume Next



        Dim f As UserForm1

        'Dim Tree As TreeView



        f = New UserForm1

        Tree = f.Tree



        ' Some variable declarations

        Dim doc As Document

        Dim rootProduct As Product



        doc = CATIA.ActiveDocument

        If doc Is Nothing Then

            MsgBox("Please load a product - Then activate this function!")

            Return

        End If



        rootProduct = doc.Product



        visitProduct(rootProduct, "", Tree)



        f.Show()



    End Sub



    Sub visitProduct(prod As Product, parentKey As String, ByRef Tree As Treeview)



        On Error Resume Next



        Dim children As Products

        children = prod.Products



        Dim i As Integer

        Dim child As Product

        Dim key As String

        Dim label As String



        label = prod.ReferenceProduct.Parent.Name & " (" & prod.Name & ")"

        key = parentKey & "###" & prod.ReferenceProduct.Parent.Name & " (" & prod.Name & ")"

        'label = prod.name

        'key = parentKey & "###" & prod.name

        'ImageList1.Enabled = True

        'tree.ImageList = ImageList1

        ' MsgBox "key: " & key & "  parent: " & parentKey



        If parentKey = "" Then

            Tree.Nodes.Add(, , key, label).Expanded = True

        Else

            Tree.Nodes.Add(parentKey, tvwChild, key, label).Expanded = True

            'tree.Nodes.Item(1).Image = 1

        End If



        For i = 1 To children.Count

            child = children.Item(i)

            visitProduct(child, key, Tree)

        Next

    End Sub



End Class