source: addons/HeuristicLab.DataImporter/HeuristicLab.DataImporter.Data/Command/SortCommand.cs @ 17607

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2013 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
 *
 * This file is part of HeuristicLab.
 *
 * HeuristicLab is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * HeuristicLab is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
 */
#endregion

using System;
using System.Collections.Generic;
using System.Linq;
using System.Windows.Forms;
using HeuristicLab.DataImporter.Data.CommandBase;
using HeuristicLab.DataImporter.Data.Model;
using HEAL.Attic;

namespace HeuristicLab.DataImporter.Data.Command {
  [StorableType("316503E6-8A33-498A-9D76-93F9CF016D55")]
  public class SortCommand : ColumnGroupCommandBase {
    [Storable]
    private bool addToSortedColumns;
    [Storable]
    private int sortColumnIndex;

    private ICollection<int> oldSortedColumnIndices;
    private IEnumerable<SortOrder> oldSortOrdersForColumns;
    private int[] indices;

    [StorableConstructor]
    protected SortCommand(StorableConstructorFlag _) : base(_) { }

    public SortCommand(DataSet dataSet, string columnGroupName, int sortColumnIndex) :
      this(dataSet, columnGroupName, sortColumnIndex, false) {
    }

    public SortCommand(DataSet dataSet, string columnGroupName, int sortColumnIndex, bool addToSortedColumns)
      : base(dataSet, columnGroupName) {
      this.addToSortedColumns = addToSortedColumns;
      this.sortColumnIndex = sortColumnIndex;
    }

    public override void Execute() {
      base.Execute();
      oldSortedColumnIndices = new List<int>(ColumnGroup.SortedColumnIndexes);
      oldSortOrdersForColumns = ColumnGroup.SortOrdersForColumns.ToList();

      if (!addToSortedColumns) {
        ColumnGroup.SortedColumnIndexes.Clear();
        for (int i = 0; i < ColumnGroup.Columns.Count(); i++)
          if (i != sortColumnIndex)
            ColumnGroup.Columns.ElementAt(i).SortOrder = SortOrder.None;
      }
      if (!ColumnGroup.SortedColumnIndexes.Contains(sortColumnIndex))
        ColumnGroup.SortedColumnIndexes.Add(sortColumnIndex);
      ICollection<int> tempSortedColumnIndices = new List<int>(ColumnGroup.SortedColumnIndexes);

      RowComparer rowComparer = new RowComparer(ColumnGroup, sortColumnIndex);
      indices = Enumerable.Range(0, ColumnGroup.RowCount).OrderBy(k => k, rowComparer).ToArray();

      List<ColumnBase> newColumns = Sort();
      ColumnGroup.ClearColumns();
      ColumnGroup.AddColumns(newColumns);
      ColumnGroup.SortedColumnIndexes = tempSortedColumnIndices;
      ColumnGroup.FireChanged();
      this.ColumnGroup = null;
    }

    public override void UndoExecute() {
      base.UndoExecute();
      indices = Enumerable.Range(0, ColumnGroup.RowCount).OrderBy<int, int>(i => indices[i]).ToArray();
      List<ColumnBase> newColumns = Sort();
      ColumnGroup.ClearColumns();
      ColumnGroup.AddColumns(newColumns);
      ColumnGroup.SortedColumnIndexes = this.oldSortedColumnIndices;
      ColumnGroup.SortOrdersForColumns = oldSortOrdersForColumns;
      indices = null;
      oldSortedColumnIndices = null;
      oldSortOrdersForColumns = null;
      ColumnGroup.FireChanged();
      this.ColumnGroup = null;
    }

    public override string Description {
      get { return "Sort ColumnGroup"; }
    }

    private List<ColumnBase> Sort() {
      List<ColumnBase> newColumns = new List<ColumnBase>();
      ColumnBase newColumn;
      ColumnBase oldColumn;

      for (int col = 0; col < ColumnGroup.Columns.Count(); col++) {
        oldColumn = ColumnGroup.GetColumn(col);
        newColumn = oldColumn.CreateCopyOfColumnWithoutValues();
        newColumn.SortOrder = oldColumn.SortOrder;
        if (col == sortColumnIndex) {
          if (newColumn.SortOrder == SortOrder.None || newColumn.SortOrder == SortOrder.Descending)
            newColumn.SortOrder = SortOrder.Ascending;
          else
            newColumn.SortOrder = SortOrder.Descending;
        }
        newColumns.Add(newColumn);
      }

      for (int j = 0; j < indices.Length; j++) {
        for (int col = 0; col < ColumnGroup.Columns.Count(); col++)
          newColumns[col].AddValue(ColumnGroup.Columns.ElementAt(col).GetValue(indices[j]));
      }
      return newColumns;
    }

    private class RowComparer : IComparer<int> {
      private ColumnGroup columnGroup;
      private int sortColumnIndex;
      public RowComparer(ColumnGroup columnGroup, int sortColumnIndex) {
        this.columnGroup = columnGroup;
        this.sortColumnIndex = sortColumnIndex;
      }

      #region IComparer<int> Members
      public int Compare(int i, int j) {
        IEnumerator<int> e = columnGroup.SortedColumnIndexes.GetEnumerator();
        int result = 0;
        while (result == 0 && e.MoveNext()) {
          IComparable x = columnGroup.GetColumn(e.Current).GetValue(i);
          IComparable y = columnGroup.GetColumn(e.Current).GetValue(j);
          if (x == null) {
            result = -1;
            if (y == null)
              result = 0;
          } else
            result = x.CompareTo(y);
          if (columnGroup.SortOrdersForColumns.ElementAt(e.Current) == SortOrder.Descending)
            result *= -1;
          if (e.Current == sortColumnIndex && columnGroup.Columns.ElementAt(e.Current).SortOrder != SortOrder.None)
            result *= -1;
        }
        return result;
      }
      #endregion
    }
  }
}