wg_cpso/CPSO/Forms/26_ElementSet/FrmElementSet.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using CaeMesh;
using System.Windows.Forms;
using CaeGlobals;

namespace CPSO.Forms
{
    class FrmElementSet : UserControls.FrmProperties, IFormBase, IFormItemSetDataParent, IFormHighlight
    {
        // Variables                                                                                                                
        private HashSet<string> _elementSetNames;
        private string _elementSetToEditName;
        private ViewElementSet _viewElementSet;
        private List<SelectionNode> _prevSelectionNodes;
        private SelectionNodeIds _selectionNodeIds;
        private Controller _controller;


        // Properties                                                                                                               
        public FeElementSet ElementSet
        {
            get { return _viewElementSet.GetBase(); }
            set { _viewElementSet = new ViewElementSet(value.DeepClone()); }
        }


        // Constructors                                                                                                             
        public FrmElementSet(Controller controller)
        {
            InitializeComponent();
            //
            _controller = controller;
            _viewElementSet = null;
            _elementSetNames = new HashSet<string>();
        }
        private void InitializeComponent()
        {
            this.gbProperties.SuspendLayout();
            this.SuspendLayout();
            // 
            // FrmElementSet
            // 
            this.ClientSize = new System.Drawing.Size(334, 411);
            this.Name = "FrmElementSet";
            this.Text = "Edit Element Set";
            this.gbProperties.ResumeLayout(false);
            this.ResumeLayout(false);
        }


        // Overrides                                                                                                                
        protected override void OnApply(bool onOkAddNew)
        {
            _viewElementSet = (ViewElementSet)propertyGrid.SelectedObject;
            //
            CheckName(_elementSetToEditName, _viewElementSet.Name, _elementSetNames, "element set");
            //
            if (ElementSet.Labels == null || ElementSet.Labels.Length <= 0)
                throw new CaeException("The element set selection must contain at least one item.");
            //
            if (_elementSetToEditName == null)
            {
                // Create
                _controller.AddElementSetCommand(ElementSet);
            }
            else
            {
                // Replace
                if (_propertyItemChanged || !ElementSet.Valid)
                {
                    // replace the ids by the previous selection
                    Selection selection = ElementSet.CreationData;
                    if (selection.Nodes[0] is SelectionNodeIds sn && sn.Equals(_selectionNodeIds))
                    {
                        selection.Nodes.RemoveAt(0);
                        selection.Nodes.InsertRange(0, _prevSelectionNodes);
                    }
                    //
                    ElementSet.Valid = true;
                    _controller.ReplaceElementSetCommand(_elementSetToEditName, ElementSet);
                }
            }
            // If all is successful close the ItemSetSelectionForm - except for OKAddNew
            if (!onOkAddNew) ItemSetDataEditor.SelectionForm.Hide();
        }
        protected override void OnHideOrClose()
        {
            // Close the ItemSetSelectionForm
            ItemSetDataEditor.SelectionForm.Hide();
            // Deactivate selection limit
            _controller.Selection.LimitSelectionToFirstGeometryType = false;
            //
            base.OnHideOrClose();
        }
        protected override bool OnPrepareForm(string stepName, string elementSetToEditName)
        {
            this.btnOkAddNew.Visible = elementSetToEditName == null;
            //
            _propertyItemChanged = false;
            _elementSetNames.Clear();
            _elementSetToEditName = null;
            _viewElementSet = null;
            propertyGrid.SelectedObject = null;
            _prevSelectionNodes = null;
            _selectionNodeIds = null;
            //
            _elementSetNames.UnionWith(_controller.GetAllMeshEntityNames());
            _elementSetToEditName = elementSetToEditName;
            //
            if (_elementSetToEditName == null)
            {
                ElementSet = new FeElementSet(GetElementSetName(), null);
                _controller.Selection.Clear();
            }
            else
            {
                ElementSet = _controller.GetElementSet(_elementSetToEditName);  // to clone
                int[] ids = ElementSet.Labels;
                if (ElementSet.CreationData == null && ids != null)             // from .inp
                {
                    // Add creation data
                    ElementSet.CreationData = new Selection();
                    ElementSet.CreationData.SelectItem = vtkSelectItem.Element;
                    ElementSet.CreationData.Add(new SelectionNodeIds(vtkSelectOperation.None, false, ids));
                }
            }
            //
            propertyGrid.SelectedObject = _viewElementSet;
            propertyGrid.Select();
            // Show ItemSetDataForm
            ItemSetDataEditor.SelectionForm.ItemSetData = new ItemSetData(ElementSet.Labels);
            ItemSetDataEditor.SelectionForm.ShowIfHidden(this.Owner);
            //
            SetSelectItem();
            //
            _controller.Selection.LimitSelectionToFirstGeometryType = true;
            //
            HighlightElementSet();
            //
            propertyGrid.BuildAutocompleteMenu(_controller.GetAllParameterNames());
            //
            return true;
        }


        // Methods                                                                                                                  
        private string GetElementSetName()
        {
            return _elementSetNames.GetNextNumberedKey("Element_Set");
        }
        private void HighlightElementSet()
        {
            try
            {
                if (ElementSet.CreationData != null)
                {
                    _controller.Selection = ElementSet.CreationData.DeepClone();
                    _controller.HighlightSelection();
                }
            }
            catch { }
        }
        //
        private void SetSelectItem()
        {
            _controller.SetSelectItemToElement();
        }
        public void SelectionChanged(int[] ids)
        {
            ElementSet.Labels = ids;
            ElementSet.CreationData = _controller.Selection.DeepClone();
            //
            propertyGrid.Refresh();
            //
            _propertyItemChanged = true;
        }
        // IFormHighlight
        public void Highlight()
        {
            if (!_closing) HighlightElementSet();
        }
        // IFormItemSetDataParent
        public bool IsSelectionGeometryBased()
        {
            // Prepare ItemSetDataEditor - prepare Geometry or Mesh based selection
            FeElementSet elementSet = ElementSet;
            //
            if (elementSet.CreationData != null) return elementSet.CreationData.IsGeometryBased();
            else return true;
        }
        public bool IsGeometrySelectionIdBased()
        {
            bool defaultMode = _controller.Settings.Pre.GeometrySelectMode == GeometrySelectModeEnum.SelectId;
            // Prepare ItemSetDataEditor - prepare Geometry or Mesh based selection
            FeElementSet elementSet = ElementSet;
            //
            if (elementSet.CreationData != null && IsSelectionGeometryBased())
                return elementSet.CreationData.IsGeometryIdBased(defaultMode);
            else return defaultMode;
        }
    }
}