source: trunk/sources/HeuristicLab.ExtLibs/HeuristicLab.Netron/3.0.2672.12446/Netron.Diagramming.Core-3.0.2672.12446/UndoRedo/Commands/GroupCommand.cs @ 2768

using System;
using System.Collections.Generic;
using System.Text;
using System.Drawing;
namespace Netron.Diagramming.Core
{
    /// <summary>
    /// Group command
    /// </summary>
    class GroupCommand : Command
    {
        #region Fields
        IBundle bundle;
        IController controller;
        IGroup mGroup;
        #endregion

        #region Properties
        /// <summary>
        /// Gets the newly created group after <see cref="Redo"/> was called
        /// </summary>
        public IGroup Group
        {
            get { return mGroup; }
        }

        /// <summary>
        /// Gets the entities.
        /// </summary>
        /// <value>The shape.</value>
        public CollectionBase<IDiagramEntity> Entities
        {
            get { return bundle.Entities; }
        }

        #endregion

        #region Constructor
        /// <summary>
        /// Initializes a new instance of the <see cref="T:GroupCommand"/> class.
        /// </summary>
        /// <param name="controller">The controller.</param>
        /// <param name="bundle">The bundle.</param>
        public GroupCommand(IController controller, IBundle bundle)
            : base(controller)
        {
            this.Text = "Group";
            this.controller = controller;
            this.bundle = bundle;//the bundle should contain only IShape and IConnection entities!
        }
        #endregion

        #region Methods

        /// <summary>
        /// Perform redo of this command.
        /// </summary>
        public override void Redo()
        {
            //create a new group; use the standard GroupShape or the CollapsibleGroupShape for a painted group with collapse/expand features.
            //GroupShape group = new GroupShape(this.Controller.Model);
            //CollapsibleGroupShape group = new CollapsibleGroupShape(this.controller.Model);
            GroupShape group = new GroupShape(this.controller.Model);
            //asign the entities to the group
            group.Entities.Clear();
            
            foreach (IDiagramEntity entity in bundle.Entities)
            {
                //this will be recursive if an entity is itself an IGroup
                entity.Group = group;
                group.Entities.Add(entity);
            }
            //add the new group to the layer
            this.Controller.Model.AddEntity(group);

            mGroup = group;

            //select the newly created group
            CollectionBase<IDiagramEntity> col = new CollectionBase<IDiagramEntity>();
            col.Add(mGroup);                  
            Selection.SelectedItems = col;
            mGroup.Invalidate();
        }

        /// <summary>
        /// Perform undo of this command.
        /// </summary>
        public override void Undo()
        {
            if (mGroup.CanUnGroup == false)
            {
                return;
            }

            //remove the group from the layer
            this.Controller.Model.DefaultPage.DefaultLayer.Entities.Remove(mGroup);

            // keep track of the entities removed.
            CollectionBase<IDiagramEntity> removedItems = 
                new CollectionBase<IDiagramEntity>();

            int numberOfItems = mGroup.Entities.Count;
          //detach the entities from the group
            for (int i = 0; i < numberOfItems; numberOfItems--)
            {
                IDiagramEntity entity = mGroup.Entities[0];
                //this will be recursive if an entity is itself an IGroup
                entity.Group = null;
                mGroup.Entities.Remove(entity);
                Controller.Model.AddEntity(entity);
                entity.Invalidate();
                removedItems.Add(entity);
            }
            //change the visuals such that the entities in the group are selected
            Selection.SelectedItems = removedItems;
            //mGroup.Entities.Clear();

            //mGroup.Invalidate();

            mGroup = null;

            //note that the entities have never been disconnected from the layer
            //so they don't have to be re-attached to the anything.
            //The insertion of the Group simply got pushed in the scene-graph.


        }


        #endregion
    }

}