source: branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.3/Implementations/LineChartLogic.cs @ 10558

#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 HeuristicLab.Analysis;

namespace HeuristicLab.DataPreprocessing {
  public class LineChartLogic : ILineChartLogic {
    private IPreprocessingData preprocessingData;
    private DataTable dataTable;

    public LineChartLogic(IPreprocessingData preprocessingData) {
      this.preprocessingData = preprocessingData;
      dataTable = new DataTable("LineChart");
      FillDataTable();
      preprocessingData.Changed += PreprocessingData_Changed;
    }

    private void FillDataTable() {
      IEnumerable<string> variableNames = preprocessingData.VariableNames;

      foreach (string variableName in variableNames) {
        IList<double> values = preprocessingData.GetValues<double>(variableName);
        DataRow row = new DataRow(variableName, "", values);
        dataTable.Rows.Add(row);
      }

    }

    public IEnumerable<object> GetVariableNames() {
      List<string> doubleVariableNames = new List<string>();

      //only return variable names from type double
      foreach (string variableName in preprocessingData.VariableNames) {
        if (preprocessingData.IsType<double>(preprocessingData.GetColumnIndex(variableName)))
          doubleVariableNames.Add(variableName);
      }

      return doubleVariableNames;
    }

    public DataTable GetDataTable() {
      return dataTable;
    }

    public void RemoveVariable(string name) {
      dataTable.Rows.Remove(name);
    }

    public void AddVariable(string name) {
      IList<double> values = preprocessingData.GetValues<double>(name);
      DataRow row = new DataRow(name, "", values);
      dataTable.Rows.Add(row);
    }

    public bool VariableIsDisplayed(string name) {

      foreach (var item in dataTable.Rows) {
        if (item.Name == name)
          return true;
      }
      return false;
    }

    void PreprocessingData_Changed(object sender, DataPreprocessingChangedEventArgs e) {
      var variableName = preprocessingData.GetVariableName(e.Column);
      switch (e.Type) {
        case DataPreprocessingChangedEventType.DeleteColumn:
          dataTable.Rows.Remove(variableName);
          break;
        case DataPreprocessingChangedEventType.AddColumn:
          dataTable.Rows.Add(new DataRow(variableName, String.Empty, preprocessingData.GetValues<double>(e.Column)));
          break;
        case DataPreprocessingChangedEventType.ChangeColumn:
        case DataPreprocessingChangedEventType.ChangeItem:
          dataTable.Rows.Remove(variableName);
          dataTable.Rows.Add(new DataRow(variableName, String.Empty, preprocessingData.GetValues<double>(e.Column)));
          break;
        case DataPreprocessingChangedEventType.DeleteRow:
        case DataPreprocessingChangedEventType.AddRow:
          dataTable.Rows.Clear();
          FillDataTable();
          break;
      }
    }
  }
}