'
' (C) Copyright 2003-2019 by Autodesk, Inc. 
'
' Permission to use, copy, modify, and distribute this software in
' object code form for any purpose and without fee is hereby granted,
' provided that the above copyright notice appears in all copies and
' that both that copyright notice and the limited warranty and
' restricted rights notice below appear in all supporting
' documentation.
'
' AUTODESK PROVIDES THIS PROGRAM "AS IS" AND WITH ALL FAULTS.
' AUTODESK SPECIFICALLY DISCLAIMS ANY IMPLIED WARRANTY OF
' MERCHANTABILITY OR FITNESS FOR A PARTICULAR USE. AUTODESK, INC.
' DOES NOT WARRANT THAT THE OPERATION OF THE PROGRAM WILL BE
' UNINTERRUPTED OR ERROR FREE.
'
' Use, duplication, or disclosure by the U.S. Government is subject to
' restrictions set forth in FAR 52.227-19 (Commercial Computer
' Software - Restricted Rights) and DFAR 252.227-7013(c)(1)(ii)
' (Rights in Technical Data and Computer Software), as applicable. 
'

Imports System
Imports System.Data
Imports System.Drawing
Imports System.Collections
Imports System.ComponentModel
Imports System.Windows.Forms
Imports Autodesk.Revit.DB

'Summary description for MaterialPropFrm.
Public Class MaterialPropertiesForm
    Inherits System.Windows.Forms.Form

#Region " Windows Form Designer generated code "

    Public Sub New(ByVal dataBuffer As MaterialProperties)
        MyBase.New()

        'This call is required by the Windows Form Designer.
        InitializeComponent()

        m_dataBuffer = dataBuffer

    End Sub

    'Form overrides dispose to clean up the component list.
    Protected Overloads Overrides Sub Dispose(ByVal disposing As Boolean)
        If disposing Then
            If Not (components Is Nothing) Then
                components.Dispose()
            End If
        End If
        MyBase.Dispose(disposing)
    End Sub

    'Required by the Windows Form Designer
    Private components As System.ComponentModel.IContainer

    'NOTE: The following procedure is required by the Windows Form Designer
    'It can be modified using the Windows Form Designer.  
    'Do not modify it using the code editor.
    Friend WithEvents changeButton As System.Windows.Forms.Button
    Friend WithEvents applyButton As System.Windows.Forms.Button
    Friend WithEvents okButton As System.Windows.Forms.Button
   Friend WithEvents parameterDataGridView As System.Windows.Forms.DataGridView
   Friend WithEvents subTypeComboBox As System.Windows.Forms.ComboBox
   Friend WithEvents typeComboBox As System.Windows.Forms.ComboBox
   Friend WithEvents typeLable As System.Windows.Forms.Label
   Friend WithEvents exitButton As System.Windows.Forms.Button
   <System.Diagnostics.DebuggerStepThrough()> Private Sub InitializeComponent()
      Me.changeButton = New System.Windows.Forms.Button
      Me.applyButton = New System.Windows.Forms.Button
      Me.exitButton = New System.Windows.Forms.Button
      Me.okButton = New System.Windows.Forms.Button
      Me.parameterDataGridView = New System.Windows.Forms.DataGridView
      Me.subTypeComboBox = New System.Windows.Forms.ComboBox
      Me.typeComboBox = New System.Windows.Forms.ComboBox
      Me.typeLable = New System.Windows.Forms.Label
      CType(Me.parameterDataGridView, System.ComponentModel.ISupportInitialize).BeginInit()
      Me.SuspendLayout()
      '
      'changeButton
      '
      Me.changeButton.Location = New System.Drawing.Point(353, 455)
      Me.changeButton.Name = "changeButton"
      Me.changeButton.Size = New System.Drawing.Size(128, 23)
      Me.changeButton.TabIndex = 16
      Me.changeButton.Text = "Change &Unit Weight"
      '
      'applyButton
      '
      Me.applyButton.Location = New System.Drawing.Point(267, 455)
      Me.applyButton.Name = "applyButton"
      Me.applyButton.Size = New System.Drawing.Size(75, 23)
      Me.applyButton.TabIndex = 15
      Me.applyButton.Text = "&Apply"
      '
      'exitButton
      '
      Me.exitButton.DialogResult = System.Windows.Forms.DialogResult.Cancel
      Me.exitButton.Location = New System.Drawing.Point(173, 455)
      Me.exitButton.Name = "exitButton"
      Me.exitButton.Size = New System.Drawing.Size(75, 23)
      Me.exitButton.TabIndex = 14
      Me.exitButton.Text = "&Cancel"
      '
      'okButton
      '
      Me.okButton.Location = New System.Drawing.Point(87, 455)
      Me.okButton.Name = "okButton"
      Me.okButton.Size = New System.Drawing.Size(75, 23)
      Me.okButton.TabIndex = 13
      Me.okButton.Text = "&OK"
      '
      'parameterDataGridView
      '
      'Me.parameterDataGridView.CaptionVisible = False
      Me.parameterDataGridView.DataMember = ""
      'Me.parameterDataGridView.HeaderForeColor = System.Drawing.SystemColors.ControlText
      Me.parameterDataGridView.Location = New System.Drawing.Point(12, 69)
      Me.parameterDataGridView.Name = "parameterDataGridView"
      Me.parameterDataGridView.ReadOnly = True
      Me.parameterDataGridView.RowHeadersVisible = False
      Me.parameterDataGridView.Size = New System.Drawing.Size(480, 380)
      Me.parameterDataGridView.TabIndex = 12
      '
      'subTypeComboBox
      '
      Me.subTypeComboBox.DropDownStyle = System.Windows.Forms.ComboBoxStyle.DropDownList
      Me.subTypeComboBox.Location = New System.Drawing.Point(100, 42)
      Me.subTypeComboBox.Name = "subTypeComboBox"
      Me.subTypeComboBox.Size = New System.Drawing.Size(264, 21)
      Me.subTypeComboBox.TabIndex = 11
      '
      'typeComboBox
      '
      Me.typeComboBox.DropDownStyle = System.Windows.Forms.ComboBoxStyle.DropDownList
      Me.typeComboBox.Location = New System.Drawing.Point(100, 7)
      Me.typeComboBox.Name = "typeComboBox"
      Me.typeComboBox.Size = New System.Drawing.Size(264, 21)
      Me.typeComboBox.TabIndex = 10
      '
      'typeLable
      '
      Me.typeLable.Location = New System.Drawing.Point(13, 7)
      Me.typeLable.Name = "typeLable"
      Me.typeLable.Size = New System.Drawing.Size(80, 23)
      Me.typeLable.TabIndex = 9
      Me.typeLable.Text = "Material Type:"
      '
      'MaterialPropertiesForm
      '
      Me.AcceptButton = Me.okButton
      Me.AutoScaleBaseSize = New System.Drawing.Size(5, 13)
      Me.CancelButton = Me.exitButton
      Me.ClientSize = New System.Drawing.Size(502, 481)
      Me.Controls.Add(Me.changeButton)
      Me.Controls.Add(Me.applyButton)
      Me.Controls.Add(Me.exitButton)
      Me.Controls.Add(Me.okButton)
      Me.Controls.Add(Me.parameterDataGridView)
      Me.Controls.Add(Me.subTypeComboBox)
      Me.Controls.Add(Me.typeComboBox)
      Me.Controls.Add(Me.typeLable)
      Me.FormBorderStyle = System.Windows.Forms.FormBorderStyle.FixedDialog
      Me.MaximizeBox = False
      Me.MinimizeBox = False
      Me.Name = "MaterialPropertiesForm"
      Me.ShowInTaskbar = False
      Me.Text = "Material Properties"
      CType(Me.parameterDataGridView, System.ComponentModel.ISupportInitialize).EndInit()
      Me.ResumeLayout(False)

   End Sub

#End Region
   Dim m_dataBuffer As MaterialProperties = Nothing

   'set selected element's material to current selection and close form
   Private Sub okButton_Click(ByVal sender As System.Object, ByVal e As System.EventArgs) Handles okButton.Click
      If Not (subTypeComboBox.SelectedValue Is Nothing) Then
         m_dataBuffer.SetMaterial(subTypeComboBox.SelectedValue)
      End If
      Me.Close()
   End Sub

   'close form
   Private Sub cancelButton_Click(ByVal sender As System.Object, ByVal e As System.EventArgs) Handles exitButton.Click
      Me.Close()
   End Sub

   'set selected element's material to current selection
   Private Sub applyButton_Click(ByVal sender As System.Object, ByVal e As System.EventArgs) Handles applyButton.Click
      If Not (subTypeComboBox.SelectedValue Is Nothing) Then
         m_dataBuffer.SetMaterial(subTypeComboBox.SelectedValue)
      End If
   End Sub

   'change unit weight all instances of the elements that use this material
   Private Sub changeButton_Click(ByVal sender As System.Object, ByVal e As System.EventArgs) Handles changeButton.Click
      Autodesk.Revit.UI.TaskDialog.Show("Revit", "This will change the unit weight of all instances that use this material in current document.")

      If Not (m_dataBuffer.ChangeUnitWeight()) Then
         Autodesk.Revit.UI.TaskDialog.Show("Revit", "Failed to change the unit weight.")
         Return
      End If
      LoadCurrentMaterial()
   End Sub

   ' when typeComboBox changed, then update the subTypeCombobox 
   Private Sub typeComboBox_SelectedIndexChanged(ByVal sender As System.Object, ByVal e As System.EventArgs) Handles typeComboBox.SelectedIndexChanged
      If (CType(typeComboBox.SelectedIndex, StructuralAssetClass) = StructuralAssetClass.Metal) Then
         applyButton.Enabled = True
         changeButton.Enabled = True
         subTypeComboBox.Enabled = True
         subTypeComboBox.DataSource = m_dataBuffer.SteelCollection
         subTypeComboBox.DisplayMember = "MaterialName"
         subTypeComboBox.ValueMember = "Material"
         parameterDataGridView.DataSource = m_dataBuffer.GetParameterTable(subTypeComboBox.SelectedValue, CType(typeComboBox.SelectedIndex, StructuralAssetClass))
      ElseIf (CType(typeComboBox.SelectedIndex, StructuralAssetClass) = StructuralAssetClass.Concrete) Then
         applyButton.Enabled = True
         changeButton.Enabled = True
         subTypeComboBox.Enabled = True
         subTypeComboBox.DataSource = m_dataBuffer.ConcreteCollection
         subTypeComboBox.DisplayMember = "MaterialName"
         subTypeComboBox.ValueMember = "Material"
         parameterDataGridView.DataSource = m_dataBuffer.GetParameterTable(subTypeComboBox.SelectedValue, CType(typeComboBox.SelectedIndex, StructuralAssetClass))
      Else
         applyButton.Enabled = False
         changeButton.Enabled = False
         subTypeComboBox.DataSource = New ArrayList
         subTypeComboBox.Enabled = False
         parameterDataGridView.DataSource = New DataTable
      End If

      If typeComboBox.SelectedIndex = CInt(m_dataBuffer.CurrentType) Then
         If (m_dataBuffer.CurrentMaterial Is Nothing Or (m_dataBuffer.CurrentType <> StructuralAssetClass.Metal And m_dataBuffer.CurrentType <> StructuralAssetClass.Concrete)) Then
            Return
         End If

         Dim tmp As Autodesk.Revit.DB.Material
         tmp = m_dataBuffer.CurrentMaterial
         If (tmp Is Nothing) Then
            Return
         End If
         subTypeComboBox.SelectedValue = tmp
         parameterDataGridView.DataSource = m_dataBuffer.GetParameterTable(subTypeComboBox.SelectedValue,
            CType(typeComboBox.SelectedIndex, StructuralAssetClass))
      ElseIf subTypeComboBox.Items.Count = 0 Then
         parameterDataGridView.DataSource = New DataTable
      End If
   End Sub

   'change the content in datagrid according to selected material type
   Private Sub subTypeComboBox_SelectedIndexChanged(ByVal sender As System.Object, ByVal e As System.EventArgs) Handles subTypeComboBox.SelectedIndexChanged
      If subTypeComboBox.SelectedValue Is Nothing Then
         parameterDataGridView.DataSource = New DataTable
      End If
      parameterDataGridView.DataSource = m_dataBuffer.GetParameterTable(subTypeComboBox.SelectedValue, CType(typeComboBox.SelectedIndex, StructuralAssetClass))
   End Sub

   'update display data to selected element's material
   Private Sub LoadCurrentMaterial()
      typeComboBox.DataSource = m_dataBuffer.MaterialTypes
      typeComboBox.SelectedIndex = CInt(m_dataBuffer.CurrentType)
      If (m_dataBuffer.CurrentMaterial Is Nothing Or (m_dataBuffer.CurrentType <> StructuralAssetClass.Metal And m_dataBuffer.CurrentType <> StructuralAssetClass.Concrete)) Then
         Return
      End If


      Dim tmp As autodesk.Revit.DB.Material
      tmp = m_dataBuffer.CurrentMaterial
      If (tmp Is Nothing) Then
         Return
      End If
      subTypeComboBox.SelectedValue = tmp
      parameterDataGridView.DataSource = m_dataBuffer.GetParameterTable(subTypeComboBox.SelectedValue,
        CType(typeComboBox.SelectedIndex, StructuralAssetClass))
   End Sub

   ' when the form loading, then load the current material of your selected beam, column or brace
   Private Sub MaterialPropertiesForm_Load(ByVal sender As Object, ByVal e As System.EventArgs) Handles MyBase.Load
      'parameterDataGridView.PreferredColumnWidth = parameterDataGridView.Width / 2 - 2
      LoadCurrentMaterial()
    End Sub
End Class