HYTOS

/*---------------------------------------------------------------------+\

|                                                                       |

|        Copyright 2016 Intergraph Corporation                          |

|        All Rights Reserved                                            |

|                                                                       |

|        Including software, file formats, and audio-visual displays;   |

|        may only be used pursuant to applicable software license       |

|        agreement; contains confidential and proprietary information of|

|        Intergraph and/or third parties which is protected by copyright|

|        and trade secret law and may not be provided or otherwise made |

|        available without proper authorization.                        |

|                                                                       |

|        Unpublished -- rights reserved under the Copyright Laws of the |

|        United States.                                                 |

|                                                                       |

|        Intergraph Corporation                                         |

|        Huntsville, Alabama         35894-0001                         |

|                                                                       |

\+---------------------------------------------------------------------*/

using System;

using System.Collections.Generic;

using System.Windows.Forms;

using Ingr.RAD2D;

using System.Runtime.InteropServices;

using System.Drawing;

using SPPID.Modeling;

using SPPID.Utill;

using SPPID.Model;

using System.Threading;

using Microsoft.Win32;

namespace CustomCommand

{

    /// <summary>

    /// This is the primary form.  The form is shown as modal.

    /// This command may be built as an executable (EXE) or as a dynamic link 

    /// library (DLL).

    /// 

    /// When you implement the code that dismisses your form, e.g., an OK

    /// button or a Stop Processing button, you should include the following:

    /// 

    ///     Hide the form, for example

    ///         converterForm.Hide

    ///     

    ///     Set the Application's Interactive property to True, for example

    ///         commandControl.Application.Interactive = True

    ///

    ///     Set Intergraph Command Control Done property to True, for example

    ///         commandControl.Done = True

    /// </summary>

    public partial class ConverterForm : Form

    {

        #region Default CustomCommand

        readonly string menuControlName = "ID2 to SPPID Converter";

        readonly string toolTipText = "ID2 to SPPID Converter";

        readonly Image image = (Image)Properties.Resources.ResourceManager.GetObject("ToolbarImage");

        /// <summary>

        /// Constructs and initializes the form.

        /// </summary>

        public ConverterForm()

        {

            InitializeComponent();

        }

        /// <summary>

        /// The OK Click Event is used to dismiss the form.

        /// </summary>

        /// <param name="sender">The sender of the event</param>

        /// <param name="e">Arguments passed during event</param>

        //private void cmdOK_Click(object sender, EventArgs e)

        //{

        //    commandControl.Application.RADApplication.Interactive = true;

        //    Hide();

        //    commandControl.Done = true;

        //}

        /// <summary>

        /// The Activate Event is where the command should show its form

        /// if it should be displayed.

        /// </summary>

        /// <param name="sender">The sender of the event</param>

        /// <param name="e">Arguments passed during event</param>

        private void commandControl_Activate(object sender, EventArgs e)

        {

            ShowDialog();

        }

        /// <summary>

        /// The Deactivate event is where the command should hide its form if it

        /// was displayed.  The command should not unload the form here.

        /// The command's form should be unloaded in the Class Module Terminate event.

        /// </summary>

        /// <param name="sender">The sender of the event</param>

        /// <param name="e">Arguments passed during event</param>

        private void commandControl_Deactivate(object sender, EventArgs e)

        {

            Hide();

        }

        /// <summary>

        /// The Initialize event is where the command should perform 1 time 

        /// initialization, for example it might save a reference to the 

        /// active document in private global data.

        /// </summary>

        /// <param name="sender">The sender of the event</param>

        /// <param name="e">Arguments passed during event</param>

        private void commandControl_Initialize(object sender, EventArgs e)

        {

            try

            {

                RegistryKey key = Registry.LocalMachine;

                RegistryKey software = key.OpenSubKey("SOFTWARE");

                RegistryKey DOFTECH = software.OpenSubKey("DOFTECH");

                if (DOFTECH != null)

                {

                    RegistryKey ID2 = DOFTECH.OpenSubKey("ID2");

                    if (ID2 != null)

                    {

                        SPPIDUtill.defaultPath = ID2.GetValue("Path").ToString();

                        Log.logPath = SPPIDUtill.defaultPath + @"Converter\SPPID Converter.log";

                        SPPIDUtill.mappingFilePath = SPPIDUtill.defaultPath + @"Converter\mapping.xml";

                        string dllPath = SPPIDUtill.defaultPath + @"Converter\SPPIDConverter_CustomCommand.dll";

                        SPPIDUtill.LoadMapping(mapping);

                        Ingr.RAD2D.Application application = commandControl.Application.RADApplication;

                        ToolBars toolBars = application.ToolBars;

                        bool find = true;

                        while (find)

                        {

                            find = false;

                            foreach (ToolbarControl item in toolBars["Main"].Controls)

                            {

                                if (item.DLLName.Contains("SPPIDConverter_CustomCommand.dll"))

                                {

                                    item.Delete();

                                    find = true;

                                    break;

                                }

                            }

                        }

                        ToolbarControl toolBarControl = toolBars["Main"].Controls.AppendMacroCommand(dllPath);

                        toolBarControl.ToolTipText = toolTipText;

                        //toolBarControl.Picture = image;

                        Menus menu = application.Menus;

                        find = true;

                        while (find)

                        {

                            find = false;

                            foreach (MenuControl item in menu["&Tools"].Controls)

                            {

                                if (item.Caption == menuControlName)

                                {

                                    item.Delete();

                                    find = true;

                                    break;

                                }

                            }

                        }

                        MenuControl menuControl = application.Menus["&Tools"].Controls.AppendMacroCommand(menuControlName, dllPath);

                        menuControl.ToolTipText = toolTipText;

                    }

                }

            }

            catch (Exception ex)

            {

                Log.WriteLine(ex);

            }

        }

        /// <summary>

        /// The Terminate event is where the command can clean up any command

        /// specific allocated resources.

        /// </summary>

        /// <param name="sender">The sender of the event</param>

        /// <param name="e">Arguments passed during event</param>

        public void commandControl_Terminate(object sender, EventArgs e)

        {

        }

        /// <summary>

        /// The primary form should not simply be unloaded. You should set the

        /// Intergraph Command Control Done property to True when you want the 

        /// form to be unloaded. Then unload the form in the dispose method.

        /// </summary>

        /// <param name="sender">The sender of the event</param>

        /// <param name="e">Arguments passed during event</param>

        private void ConverterForm_FormClosing(object sender, FormClosingEventArgs e)

        {

            commandControl.Application.RADApplication.Interactive = true; // Add by kyouho

            Hide();

            commandControl.Done = true;

            e.Cancel = true; //Do not let C# close out the form ... Let RAD close it.

        }

        #endregion

        #region ?? CustomCommand Source ????

        private Dictionary<string, string> mapping = new Dictionary<string, string>();

        Thread autoModelingThread;

        #region Botton Event

        private void openDocumentsToolStripMenuItem_Click(object sender, EventArgs e)

        {

            OpenFileDialog dia = new OpenFileDialog();

            dia.Multiselect = true;

            dia.Filter = "Xml Files(*.xml)|*.xml";

            if (dia.ShowDialog() == DialogResult.OK)

            {

                SPPIDUtill.LoadMapping(mapping);

                foreach (string fileName in dia.FileNames)

                    SetDrawingTreeNode(fileName);

                treeViewDrawing.Nodes[0].Expand();

                RefreshItemNode();

            }

        }

        private void itemMappingToolStripMenuItem_Click(object sender, EventArgs e)

        {

            List<string> needName = GetNeedMapping();

            MappingForm form = new MappingForm(mapping, needName);

            if (form.ShowDialog() == DialogResult.OK)

            {

                SPPIDUtill.SaveMapping(mapping);

                RefreshItemNode();

            }

        }

        private void importToolStripMenuItem_Click(object sender, EventArgs e)

        {

            OpenFileDialog dia = new OpenFileDialog();

            dia.Multiselect = false;

            dia.Filter = "Excel File(*.xlsx;*.xls)|*.xlsx;*.xls";

            if (dia.ShowDialog() == DialogResult.OK)

            {

                ExcelUtill.LoadMappingExcelFile(dia.FileName, mapping);

                try

                {

                    SPPIDUtill.SaveMapping(mapping);

                    RefreshItemNode();

                    MessageBox.Show("Success Load Excel File");

                }

                catch (Exception ex)

                {

                    Log.WriteLine(ex);

                    MessageBox.Show("Fail Save Mapping File");

                }

            }

        }

        private void convertToolStripMenuItem_Click(object sender, EventArgs e)

        {

            Ingr.RAD2D.Application application = commandControl.Application.RADApplication;

            SPPIDUtill.LoadMapping(mapping);

            List<SPPID.Model.Document> documents = new List<SPPID.Model.Document>();

            foreach (TreeNode node in treeViewDrawing.Nodes[0].Nodes)

            {

                if (node.Checked)

                {

                    SPPID.Model.Document document = node.Tag as SPPID.Model.Document;

                    if (document != null && document.Init())

                        documents.Add(document);

                }

            }

            //Ingr.RAD2D.Document SPPID_Doc = application.Documents.Add();

            //SPPID_Doc.SaveOnClose = false;

            //SPPID_Doc.SaveAs(@"C:\dasd.pid");

            //SPPID_Doc.Close(false);

            // Save ??? ????

            string folderPath = "";

            if (documents.Count > 0)

            {

                FolderBrowserDialog dia = new FolderBrowserDialog();

                dia.ShowNewFolderButton = true;

                if (dia.ShowDialog() == DialogResult.OK)

                    folderPath = dia.SelectedPath + @"\";

                else

                    return;

            }

            autoModelingThread = new Thread(Func =>

            {

                try

                {

                    foreach (SPPID.Model.Document document in documents)

                    {

                        Ingr.RAD2D.Document SPPID_Doc = application.Documents.Add();

                        SPPID_Doc.SaveOnClose = false;

                        AutoModeling auto = new AutoModeling(document);

                        auto.SetListBoxLog(listBoxLog);

                        auto.SetProgressBar(toolStripProgressBar);

                        auto.Run();

                        SPPID_Doc.SaveAs(@"C:\" + document.DWGNAME);

                        SPPID_Doc.Close(false);

                    }

                }

                catch (Exception ex)

                {

                    Log.WriteLine(ex);

                }

            });

            autoModelingThread.Start();

        }

        private void stopToolStripMenuItem_Click(object sender, EventArgs e)

        {

            try

            {

                if (autoModelingThread != null)

                    autoModelingThread.Abort();

            }

            catch (Exception ex)

            {

                Log.WriteLine(ex);

            }

        }

        #endregion

        private void listBoxLog_DrawItem(object sender, DrawItemEventArgs e)

        {

            if (e.Index >= 0)

            {

                ListBoxItem item = listBoxLog.Items[e.Index] as ListBoxItem;

                if (item != null)

                {

                    e.Graphics.DrawString(item.Message,

                        listBoxLog.Font,

                        new SolidBrush(item.Color),

                        0,

                        e.Index * listBoxLog.ItemHeight);

                }

            }

        }

        #region TreeNode

        #region Hide TreeNode CheckBox

        private const int TVIF_STATE = 0x8;

        private const int TVIS_STATEIMAGEMASK = 0xF000;

        private const int TV_FIRST = 0x1100;

        private const int TVM_SETITEM = TV_FIRST + 63;

        [StructLayout(LayoutKind.Sequential, Pack = 8, CharSet = CharSet.Auto)]

        private struct TVITEM

        {

            public int mask;

            public IntPtr hItem;

            public int state;

            public int stateMask;

            [MarshalAs(UnmanagedType.LPTStr)]

            public string lpszText;

            public int cchTextMax;

            public int iImage;

            public int iSelectedImage;

            public int cChildren;

            public IntPtr lParam;

        }

        [DllImport("user32.dll", CharSet = CharSet.Auto)]

        private static extern IntPtr SendMessage(IntPtr hWnd, int Msg, IntPtr wParam,

                                                 ref TVITEM lParam);

        /// <summary>

        /// Hides the checkbox for the specified node on a TreeView control.

        /// </summary>

        private void HideCheckBox(TreeView tvw, TreeNode node)

        {

            TVITEM tvi = new TVITEM();

            tvi.hItem = node.Handle;

            tvi.mask = TVIF_STATE;

            tvi.stateMask = TVIS_STATEIMAGEMASK;

            tvi.state = 0;

            SendMessage(tvw.Handle, TVM_SETITEM, IntPtr.Zero, ref tvi);

        }

        #endregion

        private void SetDrawingTreeNode(string fileName)

        {

            SPPID.Model.Document document = SPPID.Model.Document.Load(fileName, mapping);

            TreeNode drawingNode = new TreeNode();

            if (document == null)

            {

                drawingNode.Text = fileName;

                drawingNode.Name = fileName;

                drawingNode.ForeColor = Color.Red;

                treeViewDrawing.Nodes[0].Nodes.Add(drawingNode);

            }

            else

            {

                foreach (TreeNode treeNode in treeViewDrawing.Nodes[0].Nodes)

                {

                    if (treeNode.Text == document.DWGNAME)

                    {

                        treeViewDrawing.Nodes[0].Nodes.Remove(treeNode);

                        break;

                    }

                }

                drawingNode.Text = document.DWGNAME;

                drawingNode.Name = document.DWGNAME;

                drawingNode.Tag = document;

                treeViewDrawing.Nodes[0].Nodes.Add(drawingNode);

            }

            SetItemTreeNode(drawingNode);

        }

        private void SetItemTreeNode(TreeNode drawingNode)

        {

            SPPID.Model.Document document = drawingNode.Tag as SPPID.Model.Document;

            if (document != null)

            {

                foreach (LineNumber item in document.LINENUMBERS)

                {

                    TreeNode node = new TreeNode();

                    node.Name = LineNumber.LineNumberSymbolPath;

                    node.Text = item.TEXT;

                    node.Tag = item;

                    drawingNode.Nodes.Add(node);

                    HideCheckBox(treeViewDrawing, node);

                    foreach (ItemAttribute attribute in item.ATTRIBUTES)

                    {

                        TreeNode attrNode = new TreeNode();

                        attrNode.Name = attribute.Attribute;

                        attrNode.Text = attribute.Attribute;

                        attrNode.Tag = attribute;

                        node.Nodes.Add(attrNode);

                        HideCheckBox(treeViewDrawing, attrNode);

                    }

                }

                foreach (Symbol item in document.SYMBOLS)

                {

                    TreeNode node = new TreeNode();

                    node.Name = item.PARENT;

                    node.Text = item.NAME;

                    node.Tag = item;

                    drawingNode.Nodes.Add(node);

                    HideCheckBox(treeViewDrawing, node);

                    foreach (ItemAttribute attribute in item.ATTRIBUTES)

                    {

                        TreeNode attrNode = new TreeNode();

                        attrNode.Name = attribute.Attribute;

                        attrNode.Text = attribute.Attribute;

                        attrNode.Tag = attribute;

                        node.Nodes.Add(attrNode);

                        HideCheckBox(treeViewDrawing, attrNode);

                    }

                }

                foreach (Line item in document.LINES)

                {

                    TreeNode node = new TreeNode();

                    node.Name = item.TYPE;

                    node.Text = item.TYPE;

                    node.Tag = item;

                    drawingNode.Nodes.Add(node);

                    HideCheckBox(treeViewDrawing, node);

                    foreach (ItemAttribute attribute in item.ATTRIBUTES)

                    {

                        TreeNode attrNode = new TreeNode();

                        attrNode.Name = attribute.Attribute;

                        attrNode.Text = attribute.Attribute;

                        attrNode.Tag = attribute;

                        node.Nodes.Add(attrNode);

                        HideCheckBox(treeViewDrawing, attrNode);

                    }

                }

                foreach (Text item in document.TEXTS)

                {

                    TreeNode node = new TreeNode();

                    node.Name = item.NAME;

                    node.Text = item.NAME;

                    node.Tag = item;

                    drawingNode.Nodes.Add(node);

                    HideCheckBox(treeViewDrawing, node);

                    foreach (ItemAttribute attribute in item.ATTRIBUTES)

                    {

                        TreeNode attrNode = new TreeNode();

                        attrNode.Name = attribute.Attribute;

                        attrNode.Text = attribute.Attribute;

                        attrNode.Tag = attribute;

                        node.Nodes.Add(attrNode);

                        HideCheckBox(treeViewDrawing, attrNode);

                    }

                }

            }

        }

        private void RefreshItemNode()

        {

            foreach (TreeNode docNode in treeViewDrawing.Nodes[0].Nodes)

            {

                docNode.ForeColor = Color.Red;

                bool result = true;

                foreach (TreeNode itemNode in docNode.Nodes)

                {

                    string name = itemNode.Name;

                    if (name == "SIZE")

                    {

                        string sppidName = SPPIDUtill.GetSPPIDMappingName(SPPID.Model.Text.pipingCompSize, mapping);

                        if (!string.IsNullOrEmpty(sppidName))

                        {

                            itemNode.ForeColor = Color.Gray;

                        }

                        else

                        {

                            itemNode.ForeColor = Color.Red;

                            result = false;

                        }

                        sppidName = SPPIDUtill.GetSPPIDMappingName(SPPID.Model.Text.instrumentSize, mapping);

                        if (!string.IsNullOrEmpty(sppidName))

                        {

                            itemNode.ForeColor = Color.Gray;

                        }

                        else

                        {

                            itemNode.ForeColor = Color.Red;

                            result = false;

                        }

                    }

                    else

                    {

                        string sppidName = SPPIDUtill.GetSPPIDMappingName(name, mapping);

                        if (!string.IsNullOrEmpty(sppidName))

                        {

                            itemNode.ForeColor = Color.Gray;

                        }

                        else

                        {

                            itemNode.ForeColor = Color.Red;

                            result = false;

                        }

                    }

                    foreach (TreeNode attrNode in itemNode.Nodes)

                    {

                        name = attrNode.Name;

                        string sppidName = SPPIDUtill.GetSPPIDMappingName(name, mapping);

                        if (!string.IsNullOrEmpty(sppidName))

                        {

                            attrNode.ForeColor = Color.Gray;

                        }

                        else

                        {

                            attrNode.ForeColor = Color.Red;

                        }

                    }

                }

                if (result)

                {

                    docNode.ForeColor = Color.Black;

                }

            }

        }

        private List<string> GetNeedMapping()

        {

            List<string> nameList = new List<string>();

            foreach (TreeNode dwgNode in treeViewDrawing.Nodes[0].Nodes)

            {

                foreach (TreeNode docNode in dwgNode.Nodes)

                {

                    string name = docNode.Name;

                    if (name == "SIZE")

                    {

                        string sppidName = SPPIDUtill.GetSPPIDMappingName(SPPID.Model.Text.pipingCompSize, mapping);

                        if (string.IsNullOrEmpty(sppidName) && !nameList.Contains(name))

                            nameList.Add(SPPID.Model.Text.pipingCompSize);

                        sppidName = SPPIDUtill.GetSPPIDMappingName(SPPID.Model.Text.instrumentSize, mapping);

                        if (string.IsNullOrEmpty(sppidName) && !nameList.Contains(name))

                            nameList.Add(SPPID.Model.Text.instrumentSize);

                    }

                    else

                    {

                        string sppidName = SPPIDUtill.GetSPPIDMappingName(name, mapping);

                        if (string.IsNullOrEmpty(sppidName) && !nameList.Contains(name))

                            nameList.Add(name);

                    }

                    // Attribute

                    foreach (TreeNode attrNode in docNode.Nodes)

                    {

                        name = attrNode.Name;

                        string sppidName = SPPIDUtill.GetSPPIDMappingName(name, mapping);

                        if (string.IsNullOrEmpty(sppidName) && !nameList.Contains(name))

                            nameList.Add(name);

                    }

                }

            }

            return nameList;

        }

        private void treeViewDrawing_AfterCheck(object sender, TreeViewEventArgs e)

        {

            treeViewDrawing.AfterCheck -= new TreeViewEventHandler(this.treeViewDrawing_AfterCheck);

            if (e.Node.Level == 0)

            {

                foreach (TreeNode node in e.Node.Nodes)

                    node.Checked = e.Node.Checked;

            }

            else if (e.Node.Level == 1)

            {

                bool result = false;

                foreach (TreeNode node in e.Node.Parent.Nodes)

                {

                    if (node.Checked)

                    {

                        result = true;

                        break;

                    }

                }

                if (result)

                    e.Node.Parent.Checked = true;

                else

                    e.Node.Parent.Checked = false;

            }

            treeViewDrawing.AfterCheck += new TreeViewEventHandler(this.treeViewDrawing_AfterCheck);

        }

        #endregion

        #endregion

    }

}