source: branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.4/Implementations/FilteredPreprocessingData.cs @ 10939

using System;
using System.Collections.Generic;
using HeuristicLab.Common;
using HeuristicLab.Core;
using HeuristicLab.Data;
using HeuristicLab.DataPreprocessing.Interfaces;
using HeuristicLab.Problems.DataAnalysis;
using HeuristicLab.Problems.DataAnalysis.Transformations;

namespace HeuristicLab.DataPreprocessing.Implementations {
  public class FilteredPreprocessingData : NamedItem, IFilteredPreprocessingData {
    protected ITransactionalPreprocessingData originalData;
    protected ITransactionalPreprocessingData filteredData;

    protected FilteredPreprocessingData(FilteredPreprocessingData original, Cloner cloner)
      : base(original, cloner) {
      originalData = (ITransactionalPreprocessingData)original.originalData;
      filteredData = (ITransactionalPreprocessingData)original.filteredData;
    }

    public FilteredPreprocessingData(ITransactionalPreprocessingData preporcessingData)
      : base() {
      originalData = preporcessingData;
      filteredData = null;
    }

    public override IDeepCloneable Clone(Cloner cloner) {
      return new FilteredPreprocessingData(this, cloner);
    }

    public T GetCell<T>(int columnIndex, int rowIndex) {
      return ActiveData.GetCell<T>(columnIndex, rowIndex);
    }

    public void SetCell<T>(int columnIndex, int rowIndex, T value) {
      ReadOnlyOnFilterData.SetCell<T>(columnIndex, rowIndex, value);
    }

    public string GetCellAsString(int columnIndex, int rowIndex) {
      return ActiveData.GetCellAsString(columnIndex, rowIndex);
    }

    public IList<T> GetValues<T>(string variableName, bool considerSelection) {
      return ActiveData.GetValues<T>(variableName, considerSelection);
    }

    public IList<T> GetValues<T>(int columnIndex, bool considerSelection) {
      return ActiveData.GetValues<T>(columnIndex, considerSelection);
    }

    public void SetValues<T>(int columnIndex, IList<T> values) {
      ReadOnlyOnFilterData.SetValues<T>(columnIndex, values);
    }

    public void InsertRow(int rowIndex) {
      ReadOnlyOnFilterData.InsertRow(rowIndex);
    }

    public void DeleteRow(int rowIndex) {
      ReadOnlyOnFilterData.DeleteRow(rowIndex);
    }

    public void InsertColumn<T>(string variableName, int columnIndex) {
      ReadOnlyOnFilterData.InsertColumn<T>(variableName, columnIndex);
    }

    public void DeleteColumn(int columnIndex) {
      ReadOnlyOnFilterData.DeleteColumn(columnIndex);
    }

    public IntRange TrainingPartition {
      get { return originalData.TrainingPartition; }
    }

    public IntRange TestPartition {
      get { return originalData.TestPartition; }
    }

    public IList<ITransformation> Transformations {
      get { return originalData.Transformations; }
    }

    public IEnumerable<string> VariableNames {
      get { return ActiveData.VariableNames; }
    }

    public string GetVariableName(int columnIndex) {
      return ActiveData.GetVariableName(columnIndex);
    }

    public int GetColumnIndex(string variableName) {
      return ActiveData.GetColumnIndex(variableName);
    }

    public bool IsType<T>(int columnIndex) {
      return originalData.IsType<T>(columnIndex);
    }

    public int Columns {
      get { return ActiveData.Columns; }
    }

    public int Rows {
      get { return ActiveData.Rows; }
    }

    public Dataset ExportToDataset() {
      return originalData.ExportToDataset();
    }

    public void SetFilter(bool[] rowFilters) {
      filteredData = (ITransactionalPreprocessingData)originalData.Clone();
      filteredData.InTransaction(() => {
        for (int row = (rowFilters.Length - 1); row >= 0; --row) {
          if (rowFilters[row]) {
            filteredData.DeleteRow(row);
          }
        }
      });
      OnFilterChanged();
    }

    public void PersistFilter() {
      originalData.InTransaction(() => {
        for (int i = 0; i < filteredData.Columns; ++i) {
          if (filteredData.IsType<double>(i)) {
            originalData.SetValues<double>(i, filteredData.GetValues<double>(i));
          } else if (filteredData.IsType<string>(i)) {
            originalData.SetValues<string>(i, filteredData.GetValues<string>(i));
          } else if (filteredData.IsType<DateTime>(i)) {
            originalData.SetValues<DateTime>(i, filteredData.GetValues<DateTime>(i));
          } else {
            throw new ArgumentException("Data types of columns do not match");
          }
        }
      });
      ResetFilter();
    }

    public void ResetFilter() {
      filteredData = null;
      OnFilterChanged();
    }

    private void OnFilterChanged() {
      if (FilterChanged != null) {
        FilterChanged(this, new EventArgs());
      }
    }

    public ITransactionalPreprocessingData ActiveData {
      get { return IsFiltered ? filteredData : originalData; }
    }

    public ITransactionalPreprocessingData ReadOnlyOnFilterData {
      get {
        if (IsFiltered)
          throw new InvalidOperationException();

        return originalData;
      }
    }

    public bool IsFiltered {
      get { return filteredData != null; }
    }

    public event DataPreprocessingChangedEventHandler Changed {
      add { originalData.Changed += value; }
      remove { originalData.Changed -= value; }
    }

    public bool IsUndoAvailable {
      get { return IsFiltered ? false : originalData.IsUndoAvailable; }
    }

    public void Undo() {
      ReadOnlyOnFilterData.Undo();
    }

    public void InTransaction(Action action, DataPreprocessingChangedEventType type = DataPreprocessingChangedEventType.Any) {
      ReadOnlyOnFilterData.InTransaction(action, type);
    }

    public void BeginTransaction(DataPreprocessingChangedEventType type) {
      ReadOnlyOnFilterData.BeginTransaction(type);
    }

    public void EndTransaction() {
      originalData.EndTransaction();
    }

    public event EventHandler FilterChanged;

    #region IPreprocessingData Members


    public void SetSelection(IDictionary<int, IList<int>> selection) {
      originalData.SetSelection(selection);
    }

    public IDictionary<int, IList<int>> GetSelection() {
      return originalData.GetSelection();
    }

    public void ClearSelection() {
      originalData.ClearSelection();
    }

    public event EventHandler SelectionChanged {
      add { originalData.SelectionChanged += value; }
      remove { originalData.SelectionChanged -= value; }
    }

    #endregion
  }
}