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.ComponentModel;

using System.Windows.Forms;

using System.Collections.Generic;

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>

    /// Used to create a modeless dialog form command to be used in  

    /// SmartSketch.

    /// </summary>

    public partial class SPPIDConverter : UserControl

    {

        /// Expose my Caption property

        /// </summary>

        public string Caption

        {

            get; set;

        }

        /// <summary>

        /// Expose my PreferredHeight property

        /// </summary>

        public int PreferredHeight

        {

            get; set;

        }

        /// <summary>

        /// Expose my PreferredWidth property

        /// </summary>

        public int PreferredWidth

        {

            get; set;

        }

        /// <summary>

        /// Expose my PreferredLeft property

        /// </summary>

        public int PreferredLeft

        {

            get; set;

        }

        /// <summary>

        /// Expose my PreferredTop property

        /// </summary>

        public int PreferredTop

        {

            get; set;

        }

        /// <summary>

        /// Expose a Dialog property letting the application know that this

        /// control should be shown as a dialog box.

        /// </summary>

        public bool Dialog

        {

            get; set;

        }

        /// <summary>

        /// Gets the description.

        /// </summary>

        /// <value>

        /// The description of the command as assigned in the resources file.

        /// </value>

        public string Description

        {

            get

            {

                return Properties.Resources.ResourceManager.GetString("Description", System.Globalization.CultureInfo.CurrentCulture);

            }

        }

        /// <summary>

        /// Gets the tooltip of the command.

        /// </summary>

        /// <value>

        /// The tooltip of the command as assigned in the resources file.

        /// </value>

        public string GetTooltip

        {

            get

            {

                return Properties.Resources.ResourceManager.GetString("Tooltip", System.Globalization.CultureInfo.CurrentCulture);

            }

        }

        /// <summary>

        /// Gets the toolbar image.

        /// </summary>

        /// <value>

        /// The toolbar image of the command as assigned in the resources file.

        /// </value>

        public System.Drawing.Image ToolbarImage

        {

            get

            {

                return (System.Drawing.Image)Properties.Resources.ResourceManager.GetObject("ToolbarImage", System.Globalization.CultureInfo.CurrentCulture);

            }

        }

        /// <summary>

        /// Initialize this control's member data.  Be sure to modify the

        /// preferred height and width if you change the size of the user

        /// control.

        /// </summary>

        public SPPIDConverter()

        {

            InitializeComponent();

            //My caption property value

            Caption = "SPPID Converter";

            //My preferred height value.

            PreferredHeight = this.Height;

            // My preferred width.

            PreferredWidth = this.Width + 15;

            // My preferred left and top positions.

            PreferredLeft = 100;

            PreferredTop = 100;

        }

        /// <summary>

        /// This function lets the Intergraph application communicate with

        /// the command.

        /// </summary>

        /// <param name="basicCommand">The basic command</param>

        public void Command_Initialize(object basicCommand)

        {

            commandControl.Command = basicCommand;

        }

        /// <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">The event's arguments</param>

        private void commandControl_Initialize(object sender, EventArgs e)

        {

            /* Set mouseControl.WantsWrappers to True if you want the Mouse Control

             * to return your applications objects as opposed to RAD2D objects.

             * This is required if you are programming on a typelibrary other

             * than RAD2D.

             */

            mouseControl.WantsWrappers(false);

            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.dll";

                        SPPIDUtill.LoadMapping(symbolMapping, attributeMapping);

                        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.dll"))

                                {

                                    item.Delete();

                                    find = true;

                                    break;

                                }

                            }

                        }

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

                        toolBarControl.ToolTipText = GetTooltip;

                        toolBarControl.Picture = ToolbarImage;

                        Menus menu = application.Menus;

                        find = true;

                        while (find)

                        {

                            find = false;

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

                            {

                                if (item.Caption == GetTooltip)

                                {

                                    item.Delete();

                                    find = true;

                                    break;

                                }

                            }

                        }

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

                        menuControl.ToolTipText = GetTooltip;

                    }

                }

            }

            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">The event's arguments</param>

        public void commandControl_Terminate(object sender, EventArgs e)

        {

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

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

            //application.RunMacro(dllPath);

        }

        #region ?? CustomCommand Source ????

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

        private Dictionary<string, string> attributeMapping = 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(symbolMapping, attributeMapping);

                foreach (string fileName in dia.FileNames)

                    SetDrawingTreeNode(fileName);

                treeViewDrawing.Nodes[0].Expand();

                RefreshItemNode();

            }

        }

        private void itemMappingToolStripMenuItem_Click(object sender, EventArgs e)

        {

            List<string> needSymbolName = GetNeedSymbolMapping();

            List<string> needAttributeName = GetNeedAttributeMapping();

            MappingForm form = new MappingForm(symbolMapping, attributeMapping, needSymbolName, needAttributeName);

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

            {

                SPPIDUtill.SaveMapping(symbolMapping, attributeMapping);

                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, symbolMapping);

                try

                {

                    SPPIDUtill.SaveMapping(symbolMapping, attributeMapping);

                    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)

        {

            try

            {

                SPPIDUtill.LoadMapping(symbolMapping, attributeMapping);

                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);

                        else

                            Log.WriteLine("Fail Init");

                    }

                }

                autoModelingThread = new Thread(Func =>

                {

                    try

                    {

                        foreach (SPPID.Model.Document document in documents)

                        {

                            AutoModeling auto = new AutoModeling(document);

                            auto.SetListBoxLog(listBoxLog);

                            auto.SetProgressBar(toolStripProgressBar);

                            auto.Run();

                        }

                    }

                    catch (Exception ex)

                    {

                        Log.WriteLine(ex);

                    }

                });

                autoModelingThread.Start();

            }

            catch (Exception ex)

            {

                Log.WriteLine(ex);

            }

        }

        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, symbolMapping, attributeMapping);

            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 (SymbolChild child in item.CHILD_LIST)

                    {

                        TreeNode attrNode = new TreeNode();

                        attrNode.Name = child.Name;

                        attrNode.Text = child.Name;

                        attrNode.Tag = child;

                        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 (SpecBreak item in document.SPECBREAK)

                {

                    foreach (ItemAttribute attribute in item.ATTRIBUTES)

                    {

                        TreeNode node = new TreeNode();

                        node.Name = attribute.Attribute;

                        node.Text = attribute.Attribute;

                        node.Tag = item;

                        drawingNode.Nodes.Add(node);

                        HideCheckBox(treeViewDrawing, node);

                        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.GetSPPIDSymbolMappingName(SPPID.Model.Text.pipingCompSize, symbolMapping);

                        if (!string.IsNullOrEmpty(sppidName))

                        {

                            itemNode.ForeColor = Color.Gray;

                        }

                        else

                        {

                            itemNode.ForeColor = Color.Red;

                            result = false;

                        }

                        sppidName = SPPIDUtill.GetSPPIDSymbolMappingName(SPPID.Model.Text.instrumentSize, symbolMapping);

                        if (!string.IsNullOrEmpty(sppidName))

                        {

                            itemNode.ForeColor = Color.Gray;

                        }

                        else

                        {

                            itemNode.ForeColor = Color.Red;

                            result = false;

                        }

                    }

                    else

                    {

                        string sppidName = SPPIDUtill.GetSPPIDSymbolMappingName(name, symbolMapping);

                        if (!string.IsNullOrEmpty(sppidName))

                        {

                            itemNode.ForeColor = Color.Gray;

                        }

                        else

                        {

                            itemNode.ForeColor = Color.Red;

                            result = false;

                        }

                    }

                    foreach (TreeNode childNode in itemNode.Nodes)

                    {

                        name = childNode.Name;

                        string sppidName = "";

                        if (childNode.Tag.GetType() == typeof(ItemAttribute))

                            sppidName = SPPIDUtill.GetSPPIDAttributeMappingName(name, attributeMapping);

                        else

                            sppidName = SPPIDUtill.GetSPPIDSymbolMappingName(name, symbolMapping);

                        if (!string.IsNullOrEmpty(sppidName))

                        {

                            childNode.ForeColor = Color.Gray;

                        }

                        else

                        {

                            childNode.ForeColor = Color.Red;

                            if (typeof(SymbolChild) == childNode.Tag.GetType())

                                childNode.Parent.ForeColor = Color.Red;

                        }

                    }

                }

                if (result)

                {

                    docNode.ForeColor = Color.Black;

                }

            }

        }

        private List<string> GetNeedSymbolMapping()

        {

            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.GetSPPIDSymbolMappingName(SPPID.Model.Text.pipingCompSize, symbolMapping);

                        if (string.IsNullOrEmpty(sppidName) && !nameList.Contains(SPPID.Model.Text.pipingCompSize))

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

                        sppidName = SPPIDUtill.GetSPPIDSymbolMappingName(SPPID.Model.Text.instrumentSize, symbolMapping);

                        if (string.IsNullOrEmpty(sppidName) && !nameList.Contains(SPPID.Model.Text.instrumentSize))

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

                    }

                    else

                    {

                        string sppidName = SPPIDUtill.GetSPPIDSymbolMappingName(name, symbolMapping);

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

                            nameList.Add(name);

                    }

                    foreach (TreeNode childNode in docNode.Nodes)

                    {

                        name = childNode.Name;

                        if (childNode.Tag.GetType() == typeof(SymbolChild))

                        {

                            string sppidName = SPPIDUtill.GetSPPIDSymbolMappingName(name, symbolMapping);

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

                                nameList.Add(name);

                        }

                    }

                }

            }

            return nameList;

        }

        private List<string> GetNeedAttributeMapping()

        {

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

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

            {

                foreach (TreeNode docNode in dwgNode.Nodes)

                {

                    foreach (TreeNode childNode in docNode.Nodes)

                    {

                        string name = childNode.Name;

                        if (childNode.Tag.GetType() == typeof(ItemAttribute))

                        {

                            string sppidName = SPPIDUtill.GetSPPIDAttributeMappingName(name, attributeMapping);

                            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);

        }

        private void exitToolStripMenuItem_Click(object sender, EventArgs e)

        {

            commandControl.Application.RADApplication.Interactive = true;

            Hide();

            commandControl.Done = true;

        }

        #endregion

        #endregion

    }

}