using System;
using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using System.Windows.Forms;
using System.Windows.Forms.DataVisualization.Charting;

namespace HeuristicLab.Problems.Scheduling.Views {
  public partial class GanttChart : UserControl {

    private IDictionary<string, Color> categories = new Dictionary<string, Color>();
    private IDictionary<string, int> rowNames = new Dictionary<string, int>();

    public GanttChart() {
      InitializeComponent();
    }

    public void AddCategory(string name, Color color) {
      if (!categories.ContainsKey(name)) {
        categories[name] = color;
        chart.Legends[0].CustomItems.Add(color, name);
      }
    }

    public void AddData(string rowName, string category, double start, double end, string tooltip, bool showLabel = true) {
      AddRowName(rowName);
      var point = CreateDataPoint(rowNames[rowName], rowName, start, end, showLabel ? category : string.Empty, categories[category]);
      point.ToolTip = tooltip;
      chart.Series[0].Points.Add(point);
    }

    private void AddRowName(string rowName) {
      if (!rowNames.ContainsKey(rowName)) {
        int nextId = rowNames.Count == 0 ? 1 : rowNames.Values.Max() + 1;
        rowNames.Add(rowName, nextId);
      }
    }

    private static DataPoint CreateDataPoint(double x, string axisLabel, double start, double end, string text, Color color) {
      var point = new DataPoint(x, new double[] { start, end });
      point.Color = color;
      point.Label = text;
      point.AxisLabel = axisLabel;
      return point;
    }

    public void Reset() {
      chart.Series[0].Points.Clear();
      categories.Clear();
      chart.Legends[0].CustomItems.Clear();
      rowNames.Clear();
    }
  }
}