Changeset 10550 for branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.3/Implementations

Timestamp:

03/05/14 17:14:26 (11 years ago)

Author:

tsteinre

Message:

implemented Undo-feature of PreprocessingData

File:

• branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.3/Implementations/PreprocessingData.cs (modified) (11 diffs)

branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.3/Implementations/PreprocessingData.cs

@@ -30 +30 @@
 
 namespace HeuristicLab.DataPreprocessing {
+
+  internal class PDSnapshot {
+    public IDictionary<int, IList> VariableValues { get; set; }
+
+    public IList<string> VariableNames { get; set; }
+
+    public double TrainingToTestRatio { get; set; }
+
+    public DataPreprocessingChangedEventType ChangedType { get; set; }
+
+    public int ChangedColumn { get; set; }
+
+    public int ChangedRow { get; set; }
+  }
+
   [Item("PreprocessingData", "Represents data used for preprocessing.")]
   public class PreprocessingData : NamedItem, IPreprocessingData {
 
+    private const int MAX_UNDO_DEPTH = 5;
+
     private IDictionary<int, IList> variableValues;
 
@@ -38 +55 @@
 
     private double trainingToTestRatio;
+
+    private IList<PDSnapshot> undoHistory;
 
     private PreprocessingData(PreprocessingData original, Cloner cloner)
       : base(original, cloner) {
-      variableValues = new Dictionary<int, IList>(original.variableValues);
-      for (int i = 0; i < variableValues.Count; i++)
-        variableValues[i] = new ArrayList(original.variableValues[i]);
+      variableValues = CopyVariableValues(original.variableValues);
       variableNames = new List<string>(original.variableNames);
       trainingToTestRatio = original.trainingToTestRatio;
+      undoHistory = new List<PDSnapshot>();
     }
 
@@ -71 +89 @@
 
       trainingToTestRatio = (double)problemData.TrainingPartition.Size / Math.Max(problemData.Dataset.Rows, double.Epsilon);
+      undoHistory = new List<PDSnapshot>();
     }
 
@@ -81 +100 @@
     }
 
+    private IDictionary<int, IList> CopyVariableValues(IDictionary<int, IList> original) {
+      var copy = new Dictionary<int, IList>(variableValues);
+      for (int i = 0; i < original.Count; i++) {
+        if (variableValues[i] is IList<double>) {
+          copy[i] = new List<double>((IList<double>)variableValues[i]);
+        } else if (variableValues[i] is IList<DateTime>) {
+          copy[i] = new List<DateTime>((IList<DateTime>)variableValues[i]);
+        } else if (variableValues[i] is IList<string>) {
+          copy[i] = new List<string>((IList<string>)variableValues[i]);
+        } else {
+          throw new NotImplementedException("The Type is not Supported");
+        }
+      }
+      return copy;
+    }
+
+    private void SaveSnapshot(DataPreprocessingChangedEventType changedType, int column, int row) {
+      PDSnapshot currentSnapshot = new PDSnapshot();
+      currentSnapshot.VariableValues = CopyVariableValues(variableValues);
+      currentSnapshot.VariableNames = new List<string>(variableNames);
+      currentSnapshot.TrainingToTestRatio = trainingToTestRatio;
+      currentSnapshot.ChangedType = changedType;
+      currentSnapshot.ChangedColumn = column;
+      currentSnapshot.ChangedRow = row;
+
+      if (undoHistory.Count >= MAX_UNDO_DEPTH)
+        undoHistory.RemoveAt(0);
+
+      undoHistory.Add(currentSnapshot);
+    }
+
     #region NamedItem abstract Member Implementations
 
@@ -96 +146 @@
 
 
-    public void SetCell<T>(int columnIndex, int rowIndex, T value) {    // Undo-sensitive
+    public void SetCell<T>(int columnIndex, int rowIndex, T value) {
+      SaveSnapshot(DataPreprocessingChangedEventType.ChangeItem, columnIndex, rowIndex);
       variableValues[columnIndex][rowIndex] = value;
       OnChanged(DataPreprocessingChangedEventType.ChangeItem, columnIndex, rowIndex);
@@ -116 +167 @@
     }
 
-    public void SetValues<T>(int columnIndex, IList<T> values) {    // Undo-sensitive
+    public void SetValues<T>(int columnIndex, IList<T> values) {
       if (IsType<T>(columnIndex)) {
+        SaveSnapshot(DataPreprocessingChangedEventType.ChangeColumn, columnIndex, -1);
         variableValues[columnIndex] = (IList)values;
       } else {
@@ -125 +177 @@
     }
 
-    public void InsertRow(int rowIndex) {    // Undo-sensitive
+    public void InsertRow(int rowIndex) {
+      SaveSnapshot(DataPreprocessingChangedEventType.DeleteRow, -1, rowIndex);
       foreach (IList column in variableValues.Values) {
         Type type = column.GetType().GetGenericArguments()[0];
@@ -133 +186 @@
     }
 
-    public void DeleteRow(int rowIndex) {    // Undo-sensitive
+    public void DeleteRow(int rowIndex) {
+      SaveSnapshot(DataPreprocessingChangedEventType.AddRow, -1, rowIndex);
       foreach (IList column in variableValues.Values) {
         column.RemoveAt(rowIndex);
@@ -140 +194 @@
     }
 
-    public void InsertColumn<T>(string variableName, int columnIndex) {    // Undo-sensitive
+    public void InsertColumn<T>(string variableName, int columnIndex) {
+      SaveSnapshot(DataPreprocessingChangedEventType.DeleteColumn, columnIndex, -1);
       variableValues.Add(columnIndex, new List<T>(Rows));
       variableNames.Insert(columnIndex, variableName);
@@ -146 +201 @@
     }
 
-    public void DeleteColumn(int columnIndex) {    // Undo-sensitive
+    public void DeleteColumn(int columnIndex) {
+      SaveSnapshot(DataPreprocessingChangedEventType.AddColumn, columnIndex, -1);
       variableValues.Remove(columnIndex);
       variableNames.RemoveAt(columnIndex);
@@ -203 +259 @@
 
     public bool IsUndoAvailable {
-      get { throw new NotImplementedException(); }
+      get { return undoHistory.Count > 0; }
     }
 
     public void Undo() {
-      throw new NotImplementedException();
+      if (IsUndoAvailable) {
+        PDSnapshot previousSnapshot = undoHistory[undoHistory.Count - 1];
+        variableValues = previousSnapshot.VariableValues;
+        variableNames = previousSnapshot.VariableNames;
+        trainingToTestRatio = previousSnapshot.TrainingToTestRatio;
+        undoHistory.Remove(previousSnapshot);
+        OnChanged(previousSnapshot.ChangedType,
+          previousSnapshot.ChangedColumn,
+          previousSnapshot.ChangedRow);
+      }
     }