using Autodesk.Revit.DB;
using Autodesk.Revit.DB.Plumbing;
using Autodesk.Revit.UI;
using Glodon.Revit.Utility;
using Hydro.Core.Model;
using Hydro.Revit;
using Hydro.Revit.Model;
using HStation.RevitDev.RevitDataExport.Common;
using HStation.RevitDev.RevitDataExport.Service;
using HStation.RevitDev.RevitDataExport.Utility;
using HStation.RevitDev.RevitDataExport.Utils;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
using System.Windows.Controls;
using System.Windows.Forms;
using static Hydro.Core.ObjectEnum;

namespace HStation.RevitDev.RevitDataExport.Forms
{
    public partial class Form_SystemSelect : System.Windows.Forms.Form
    {
        private UIDocument _uiDocument;
        private Document _doc;
        private ExternalCommandData _data;
        private ItemCheckEventArgs _itemCheck;
        public RevitNetWork _revitNetWork;

        public Form_SystemSelect(ExternalCommandData data)
        {
            _data = data;
            _uiDocument = data.Application.ActiveUIDocument;
            _doc = _uiDocument.Document;
            InitializeComponent();
        }

        private void SystemSelect_Load(object sender, System.EventArgs e)
        {
            InitControlsData();
        }

        private void InitControlsData()
        {
            InitMepSystemComboBox();
            InitMepSystemCheckListBox();
            InitDataGridViewStyle();
            InitDataGridViewEvent();
        }

        private void InitMepSystemCheckListBox()
        {
            this.checkedListBox_systemType.Items.Clear();
            var systemTypes = RevitMepSystemService.GetSystemTypeList(_doc).ToList();
            systemTypes = systemTypes.FindAll(x => x.GetDependentElements(new ElementClassFilter(typeof(PipingSystem), false)).Count != 0);
            if (systemTypes != null && systemTypes.Count > 0)
            {
                this.checkedListBox_systemType.DataSource = systemTypes;
                this.checkedListBox_systemType.DisplayMember = "Name";
                this.checkedListBox_systemType.SelectedIndex = 0;
            }
            this.checkedListBox_systemType.SelectedIndex = -1;
        }

        private void InitDataGridViewStyle()
        {
            var pages = this.pageContainer.TabPages;
            foreach (TabPage page in pages)
            {
                foreach (var subControl in page.Controls)
                {
                    if (subControl is DataGridView dgv)
                    {
                        dgv.AllowUserToResizeRows = false;
                    }
                }
            }
        }

        private void InitDataGridViewEvent()
        {
            var dgvs = new List<DataGridView>();
            var pages = this.pageContainer.TabPages;
            foreach (TabPage page in pages)
            {
                foreach (var subControl in page.Controls)
                {
                    if (subControl is DataGridView dgv)
                    {
                        dgvs.Add(dgv);
                        dgv.CellClick += Dgv_CellClick;
                        //dgv.CellContentDoubleClick += Dgv_DoubleCellClick;
                    }
                }
            }
        }

        private void Dgv_DoubleCellClick(object sender, DataGridViewCellEventArgs e)
        {
            var dgv = sender as DataGridView;
            var rowIndex = e.RowIndex;
            var columnIndex = e.ColumnIndex;
            var cell = dgv[columnIndex, rowIndex];

            if (cell.Value == null) { return; }
            var view = Set3DView();

            var cellText = cell.Value.ToString();
            if (dgv.Columns[columnIndex].HeaderText == "ID" ||
                dgv.Columns[columnIndex].HeaderText.Contains("连接点"))
            {
                var idContent = cellText.Split(':')?.Last();
                if (string.IsNullOrEmpty(idContent)) { return; }
                if (int.TryParse(idContent, out int id))
                {
                    var elementId = new ElementId(id);
                    var ids = new List<ElementId> { elementId };
                    view.Select(_uiDocument, ids);
                    view.OnlyShow(ids);
                    view.Zoom(_uiDocument, ids);
                }
            }
        }

        private void Dgv_CellClick(object sender, DataGridViewCellEventArgs e)
        {
            var dgv = sender as DataGridView;
            var rowIndex = e.RowIndex;
            var columnIndex = e.ColumnIndex;
            var cell = dgv[columnIndex, rowIndex];

            if (cell.Value == null) { return; }
            var view = Set3DView();

            var cellText = cell.Value.ToString();
            if (dgv.Columns[columnIndex].HeaderText == "ID" ||
                dgv.Columns[columnIndex].HeaderText.Contains("连接点"))
            {
                var idContent = cellText.Split(':')?.Last();
                if (string.IsNullOrEmpty(idContent)) { return; }
                if (int.TryParse(idContent, out int id))
                {
                    var elementId = new ElementId(id);
                    var ids = new List<ElementId> { elementId };
                    view.Select(_uiDocument, ids);
                }
            }
        }

        private void InitMepSystemComboBox()
        {
            //this.comboBox1.Items.Clear();
            //var systems = RevitMepSystemService.GetSystemList(_doc).ToList();
            //systems = systems.FindAll(x => x is PipingSystem);
            //if (systems != null && systems.Count > 0)
            //{
            //    this.comboBox1.DataSource = systems;
            //    this.comboBox1.DisplayMember = "Name";
            //    this.comboBox1.SelectedIndex = 0;
            //}
            //this.comboBox1.SelectedIndex = -1;
        }

        private View3D Set3DView()
        {
            var view3d = ViewHelper.Get3dView(_doc);
            _uiDocument.ActiveView = view3d;
            return view3d;
        }

        private List<Tuple<string, string>> PostProcessing(Dictionary<string, List<Element>> groupElems, List<Tuple<string, string>> properties)
        {
            var result = new List<Tuple<string, string>>();
            foreach (var prop in properties)
            {
                if (!prop.Item1.Contains("连接点"))
                {
                    result.Add(prop);
                    continue;
                }
                bool findGroupName = false;
                foreach (var group in groupElems)
                {
                    if (group.Value.Any(x => x.Id.IntegerValue.ToString() == prop.Item2))
                    {
                        result.Add(new Tuple<string, string>(prop.Item1, /*group.Key + ":" +*/ prop.Item2));
                        findGroupName = true;
                        break;
                    }
                }
                if (!findGroupName)
                {
                    result.Add(new Tuple<string, string>(prop.Item1, prop.Item2));
                }
            }

            return result;
        }

        private void FillRow(DataGridView dgv, List<Tuple<string, string>> properties, ref int elemIndex)
        {
            properties.Insert(0, new Tuple<string, string>("序号", (++elemIndex).ToString()));
            var newRowIndex = dgv.Rows.Add();
            for (var i = 0; i < dgv.ColumnCount; i++)
            {
                var cell = dgv.Rows[newRowIndex].Cells[i];
                var hd = dgv.Columns[cell.ColumnIndex].HeaderText;
                var p = properties.Where(c => c.Item1.Equals(hd)).FirstOrDefault();
                if (p != null)
                {
                    cell.Value = p.Item2?.ToString();
                }
            }
        }


        private void FillDataGridView(Dictionary<string, List<Element>> groupElems)
        {
            foreach (TabPage page in this.pageContainer.TabPages)
            //for (var i = 0; i < this.pageContainer.TabPages.Count - 1; i++)
            {
                var t = new Task(() =>
                 {
                     //var page = this.pageContainer.TabPages[i];

                     string str = "";

                     Stopwatch sw = Stopwatch.StartNew();
                     sw.Restart();
                     var pageTitle = page.Text;
                     var dgv = page.Controls[0] as DataGridView;
                     dgv.Rows.Clear();
                     if (groupElems.ContainsKey(pageTitle))
                     {
                         //continue;
                         var elems = groupElems[pageTitle];
                         var parser = ParserManager.Instance.GetParser(pageTitle);
                         var elemIndex = 0;
                         foreach (var element in elems)
                         {
                             List<Tuple<string, string>> properties = parser.PropertyParse(element);
                             var newProps = PostProcessing(groupElems, properties);
                             FillRow(dgv, newProps, ref elemIndex);
                         }
                         sw.Stop();
                         str += pageTitle + " 数据加载耗时：" + sw.Elapsed.TotalSeconds + " 秒，数量：" + elems.Count + "，时间：" + DateTime.Now.ToString();
                         textBox1.Text += str;
                     }

                 });
                t.Start();

            }
        }


        private void button1_Click2(object sender, EventArgs e)
        {
            var excelFolder = DialogHelper.UserSelectFolderPath();
            if (string.IsNullOrEmpty(excelFolder)) { return; }
            var excelPath = Path.Combine(excelFolder, $@"管道系统数据.xlsx");

            ExcelHelper excelHelper = new ExcelHelper();
            var indexes = new List<int>();
            var items = this.checkedListBox_systemType.CheckedIndices;
            for (int i = 0; i < items.Count; i++)
            {
                int index = items[i];
                indexes.Add(index);
            }
            Dictionary<string, List<Element>> dict = GetGroupElements(indexes);


            foreach (TabPage page in this.pageContainer.TabPages)
            {
                var pageTitle = page.Text;
                if (!dict.ContainsKey(pageTitle))
                {
                    continue;
                }
                var elems = dict[pageTitle];
                var parser = ParserManager.Instance.GetParser(pageTitle);

                excelHelper.SetCurrentSheet(pageTitle);

                for (int i = 1; i <= elems.Count; i++)
                {
                    var element = elems[i - 1];
                    List<Tuple<string, string>> properties = parser.PropertyParse(element);
                    var newProps = PostProcessing(dict, properties);
                    if (i == 1)//先写表头
                    {
                        var headers = newProps.Select(x => x.Item1).ToList();
                        for (int j = 1; j <= headers.Count; j++)
                        {
                            excelHelper.Write(i, j, headers[j - 1]);
                        }
                    }
                    var contents = newProps.Select(x => x.Item2).ToList();
                    for (int j = 1; j <= contents.Count; j++)
                    {
                        excelHelper.Write(i + 1, j, contents[j - 1]);
                    }
                }
            }

            excelHelper.SaveAs(excelPath);
            MessageBox.Show("导出完成！");
        }

        private void checkedListBox_systemType_ItemCheck(object sender, ItemCheckEventArgs e)
        {
            this._itemCheck = e;
            //SetCheckIndexs();
        }

        public void SetCheckIndexs()
        {
            Stopwatch stopwatch = new Stopwatch();
            stopwatch.Start();
            textBox1.Text += "开始时间：" + DateTime.Now.ToString();
            List<int> indexes = GetCheckedIndexes(this._itemCheck);
            Dictionary<string, List<Element>> dict = GetGroupElements(indexes);
            var elemIds = dict.Values.SelectMany(x => x).Select(y => y.Id).ToList();
            var view = Set3DView();
            if (elemIds.Count != 0)
            {
                view.OnlyShow(elemIds);
                view.Zoom(_uiDocument, elemIds);
            }
            stopwatch.Stop();
            textBox1.Text += "，获取所有构件耗时：" + stopwatch.Elapsed.TotalSeconds + "秒，时间：" + DateTime.Now.ToString();
            //stopwatch.Restart();
            FillDataGridView(dict);
        }

        private List<int> GetCheckedIndexes(ItemCheckEventArgs e)
        {
            var indexes = new List<int>();
            var items = this.checkedListBox_systemType.CheckedIndices;
            for (int i = 0; i < items.Count; i++)
            {
                int index = items[i];
                indexes.Add(index);
            }
            //if (e.NewValue == CheckState.Checked)
            //{
            //    indexes.Add(e.Index);
            //}
            //else
            //{
            //    if (indexes.Contains(e.Index))
            //    {
            //        indexes.Remove(e.Index);
            //    }
            //}

            return indexes;
        }


        private void SetProp(int eleId, string propName, string propValue)
        {
            var eid = new ElementId(eleId);
            var ele = _doc.GetElement(eid);
            if (ele == null) { return; }

            var prop = ele.LookupParameter(propName);
            if (propName.Equals("流量"))
                propValue = Math.Abs(decimal.Parse(propValue)).ToString();
            if (prop != null && !prop.IsReadOnly && prop.StorageType == StorageType.String)
            {
                prop.Set(propValue);
            }

        }

        private Dictionary<string, List<Element>> GetGroupElements(List<int> indexes)
        {
            Dictionary<string, List<Element>> dict = new Dictionary<string, List<Element>>();
            foreach (var index in indexes)
            {
                var systemType = checkedListBox_systemType.Items[index] as PipingSystemType;
                var typeName = systemType.Name;
                var elems = RevitMepElementService.GetElements(_doc, typeName);
                var subDict = RevitMepCategoryService.GroupElements(elems);
                foreach (var pairs in subDict)
                {
                    if (!dict.ContainsKey(pairs.Key))
                    {
                        dict.Add(pairs.Key, new List<Element>());
                    }
                    dict[pairs.Key].AddRange(pairs.Value);
                }
            }

            return dict;
        }

        public CheckModel GetNetWrok()
        {
            this._revitNetWork = NetWorkHelper.GetModelList(this.pageContainer);
            this._revitNetWork.BuildToInp(inpPath);
            return NetWorkHelper.Check(this._revitNetWork);
        }
        private string inpPath = "d:\\revitinp.inp";
        private void button2_Click(object sender, EventArgs e)
        {
           
            //SetProp(577273, "标记", "注释测试");
            ////var json = JsonHelper.ToJson(list);
            var result = GetNetWrok();
            //nw.BuildToInp(inpPath);

            if (result.Result)
            {
                if (MessageBox.Show("校验通过，是否执行计算？", "校验结果", MessageBoxButtons.OKCancel, MessageBoxIcon.Question) == DialogResult.OK)
                {
                    var r = this._revitNetWork.Calc(inpPath);
                    SetEleProp(r);
                    var json = JsonHelper.ToJson(r);
                    JsonHelper.WriteJson("d:\\revittest.json", json);
                    MessageBox.Show("计算成功!");
                }
            }
            else
            {
                var form = new Form_CheckResult(result, this.pageContainer, _data, this);
                var revitHandle = Process.GetCurrentProcess().MainWindowHandle;
                form.Show(new WindowHandle(revitHandle));
            }

        }

        private void SetEleProp(List<TimePoint> eleProp)
        {
            //Transaction trans = new Transaction(_doc, "修改参数");
            //trans.Start();
            eleProp.ForEach(d =>
            {
                var id = int.Parse(d.Key.Split('_')[0]);
                var prop = d.Key.Split('_')[1];
                var vu = d.Value.ToString();
                SetProp(id, prop, vu);
            });
            //trans.Commit();
        }

        public void Test()
        {
            MessageBox.Show("OK");
        }

        private void button5_Click(object sender, EventArgs e)
        {
            SetCheckIndexs();
        }

        private void button6_Click(object sender, EventArgs e)
        {
            //var refernce = _uiDocument.Selection.PickObject(Autodesk.Revit.UI.Selection.ObjectType.Element);
            var pipe1 = _doc.GetElement(new ElementId(714683)) as MEPCurve;

            //var refernce2 = _uiDocument.Selection.PickObject(Autodesk.Revit.UI.Selection.ObjectType.Element);
            var pipe2 = _doc.GetElement(new ElementId(732605)) as MEPCurve;
            JoinGeometryUtils.JoinGeometry(_doc, pipe1, pipe2);
        }
    }
}