source: branches/PersistenceOverhaul/HeuristicLab.ExtLibs/HeuristicLab.Netron/3.0.2672.12446/Netron.Diagramming.Core-3.0.2672.12446/UndoRedo/Commands/GroupCommand.cs

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);
      this.Controller.Model.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
      this.Controller.Model.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
  }

}