#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2016 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.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Windows.Forms;
using System.Windows.Forms.DataVisualization.Charting;
using HeuristicLab.Core.Views;
using HeuristicLab.MainForm;
using System.Drawing.Imaging;

namespace HeuristicLab.Problems.MultiObjectiveTestFunctions
{
    [View("Scatter Plot")]
    [Content(typeof(IMOFrontModel))]
    public partial class MOQualitiesScatterPlotView : ItemView
    {
        private const string QUALITIES = "Qualities";
        private const string PARETO_FRONT = "Best Known Pareto Front";
        private Series qualitySeries;
        private Series paretoSeries;
        private int xDim = 0;
        private int yDim = 1;
        int objectives = -1;

        public new IMOFrontModel Content
        {
            get { return (IMOFrontModel)base.Content; }
            set { base.Content = value; }
        }

        public MOQualitiesScatterPlotView()
            : base()
        {
            InitializeComponent();

            BuildEmptySeries();

            //start with qualities toggled ON
            qualitySeries.Points.AddXY(0, 0);

            this.chart.TextAntiAliasingQuality = TextAntiAliasingQuality.High;
            this.chart.AxisViewChanged += new EventHandler<System.Windows.Forms.DataVisualization.Charting.ViewEventArgs>(chart_AxisViewChanged);

            //configure axis 
            this.chart.CustomizeAllChartAreas();
            this.chart.ChartAreas[0].AxisX.Title = "Objective " + xDim;
            this.chart.ChartAreas[0].CursorX.IsUserSelectionEnabled = true;
            this.chart.ChartAreas[0].AxisX.ScaleView.Zoomable = true;
            this.chart.ChartAreas[0].CursorX.Interval = 1;
            this.chart.ChartAreas[0].CursorY.Interval = 1;

            this.chart.ChartAreas[0].AxisY.Title = "Objective " + yDim;
            this.chart.ChartAreas[0].CursorY.IsUserSelectionEnabled = true;
            this.chart.ChartAreas[0].AxisY.ScaleView.Zoomable = true;
            this.chart.ChartAreas[0].AxisY.IsStartedFromZero = true;
        }

        protected override void RegisterContentEvents()
        {
            base.RegisterContentEvents();
            this.chart.GetToolTipText += new System.EventHandler<ToolTipEventArgs>(this.Chart_GetToolTipText);
        }

        protected override void DeregisterContentEvents()
        {
            base.DeregisterContentEvents();
            this.chart.GetToolTipText -= new System.EventHandler<ToolTipEventArgs>(this.Chart_GetToolTipText);
        }

        private void Chart_GetToolTipText(object sender, ToolTipEventArgs e)
        {
            if (e.HitTestResult.ChartElementType == ChartElementType.LegendItem)
            {
                if (e.HitTestResult.Series == paretoSeries && (Content.ParetoFront == null || Content.ParetoFront.Length == 0))
                {
                    e.Text = "No optimal pareto front is available for this problem with this number of objectives";
                }
                if (e.HitTestResult.Series == paretoSeries && (xDim >= Content.Objectives || yDim >= Content.Objectives))
                {
                    e.Text = "The optimal pareto front can only be displayed in  Objective Space";
                }
            }

            // Check selected chart element and set tooltip text
            if (e.HitTestResult.ChartElementType == ChartElementType.DataPoint)
            {
                int i = e.HitTestResult.PointIndex;
                StringBuilder toolTippText = new StringBuilder();
                DataPoint qp = e.HitTestResult.Series.Points[i];
                toolTippText.Append("Objective " + xDim + " = " + qp.XValue + "\n");
                toolTippText.Append("Objective " + yDim + " = " + qp.YValues[0]);

                Series s = e.HitTestResult.Series;
                if (s.Equals(this.chart.Series[QUALITIES]))
                {
                    double[] dp = Content.Solutions[i];
                    toolTippText.Append("\nSolution: {");
                    for (int j = 0; j < dp.Length; j++)
                    {
                        toolTippText.Append(dp[j]);
                        toolTippText.Append(";");
                    }
                    toolTippText.Remove(toolTippText.Length - 1, 1);
                    toolTippText.Append("}");
                    e.Text = toolTippText.ToString();
                }


            }
        }

        protected override void OnContentChanged()
        {
            base.OnContentChanged();
            if (Content == null) return;
            if (objectives != Content.Objectives)
            {
                AddMenuItems();
                objectives = Content.Objectives;
            }
            if (Content.ParetoFront == null && chart.Series.Contains(paretoSeries))
            {
                Series s = this.chart.Series[PARETO_FRONT];
                paretoSeries = null;
                this.chart.Series.Remove(s);

            }
            else if (Content.ParetoFront != null && !chart.Series.Contains(paretoSeries))
            {
                this.chart.Series.Add(PARETO_FRONT);
                paretoSeries = this.chart.Series[PARETO_FRONT];
                this.chart.Series[PARETO_FRONT].LegendText = PARETO_FRONT;
                this.chart.Series[PARETO_FRONT].ChartType = SeriesChartType.FastPoint;
            }
            UpdateChart();
        }

        private void UpdateChart()
        {
            if (InvokeRequired) Invoke((Action)UpdateChart);
            else
            {
                if (Content != null)
                {
                    this.UpdateSeries();
                    if (!this.chart.Series.Any(s => s.Points.Count > 0))
                        this.ClearChart();
                }
            }
        }

        private void UpdateCursorInterval()
        {
            var estimatedValues = this.chart.Series[QUALITIES].Points.Select(x => x.XValue).DefaultIfEmpty(1.0);
            var targetValues = this.chart.Series[QUALITIES].Points.Select(x => x.YValues[0]).DefaultIfEmpty(1.0);
            double estimatedValuesRange = estimatedValues.Max() - estimatedValues.Min();
            double targetValuesRange = targetValues.Max() - targetValues.Min();
            double interestingValuesRange = Math.Min(Math.Max(targetValuesRange, 1.0), Math.Max(estimatedValuesRange, 1.0));
            double digits = (int)Math.Log10(interestingValuesRange) - 3;
            double zoomInterval = Math.Max(Math.Pow(10, digits), 10E-5);
            this.chart.ChartAreas[0].CursorX.Interval = zoomInterval;
            this.chart.ChartAreas[0].CursorY.Interval = zoomInterval;

            this.chart.ChartAreas[0].AxisX.ScaleView.SmallScrollSize = zoomInterval;
            this.chart.ChartAreas[0].AxisY.ScaleView.SmallScrollSize = zoomInterval;

            this.chart.ChartAreas[0].AxisX.ScaleView.SmallScrollMinSizeType = DateTimeIntervalType.Number;
            this.chart.ChartAreas[0].AxisX.ScaleView.SmallScrollMinSize = zoomInterval;
            this.chart.ChartAreas[0].AxisY.ScaleView.SmallScrollMinSizeType = DateTimeIntervalType.Number;
            this.chart.ChartAreas[0].AxisY.ScaleView.SmallScrollMinSize = zoomInterval;

            if (digits < 0)
            {
                this.chart.ChartAreas[0].AxisX.LabelStyle.Format = "F" + (int)Math.Abs(digits);
                this.chart.ChartAreas[0].AxisY.LabelStyle.Format = "F" + (int)Math.Abs(digits);
            }
            else
            {
                this.chart.ChartAreas[0].AxisX.LabelStyle.Format = "F0";
                this.chart.ChartAreas[0].AxisY.LabelStyle.Format = "F0";
            }
        }

        private void UpdateSeries()
        {
            if (InvokeRequired) Invoke((Action)UpdateSeries);
            else
            {

                if (this.chart.Series.Contains(qualitySeries) && qualitySeries.Points.Count() != 0)
                {
                    fillSeries(Content.Qualities, Content.Solutions, qualitySeries);
                }
                if (this.chart.Series.Contains(paretoSeries) && paretoSeries.Points.Count() != 0)
                {
                    fillSeries(Content.ParetoFront, null, paretoSeries);
                }


                double minX = Double.MaxValue;
                double maxX = Double.MinValue;
                double minY = Double.MaxValue;
                double maxY = Double.MinValue;
                foreach (Series s in this.chart.Series)
                {
                    if (s.Points.Count == 0) continue;
                    minX = Math.Min(minX, s.Points.Select(p => p.XValue).Min());
                    maxX = Math.Max(maxX, s.Points.Select(p => p.XValue).Max());
                    minY = Math.Min(minY, s.Points.Select(p => p.YValues.Min()).Min());
                    maxY = Math.Max(maxY, s.Points.Select(p => p.YValues.Max()).Max());
                }

                maxX = maxX + 0.2 * Math.Abs(maxX);
                minX = minX - 0.2 * Math.Abs(minX);
                maxY = maxY + 0.2 * Math.Abs(maxY);
                minY = minY - 0.2 * Math.Abs(minY);

                double interestingValuesRangeX = maxX - minX;
                double interestingValuesRangeY = maxY - minY;

                int digitsX = Math.Max(0, 3 - (int)Math.Log10(interestingValuesRangeX));
                int digitsY = Math.Max(0, 3 - (int)Math.Log10(interestingValuesRangeY));


                maxX = Math.Round(maxX, digitsX);
                minX = Math.Round(minX, digitsX);
                maxY = Math.Round(maxY, digitsY);
                minY = Math.Round(minY, digitsY);
                if (minX > maxX)
                {
                    minX = 0;
                    maxX = 1;
                }
                if (minY > maxY)
                {
                    minY = 0;
                    maxY = 1;
                }


                this.chart.ChartAreas[0].AxisX.Maximum = maxX;
                this.chart.ChartAreas[0].AxisX.Minimum = minX;
                this.chart.ChartAreas[0].AxisY.Maximum = maxY;
                this.chart.ChartAreas[0].AxisY.Minimum = minY;
                UpdateCursorInterval();
            }
        }

        private void ClearChart()
        {
            if (chart.Series.Contains(qualitySeries)) chart.Series.Remove(qualitySeries);
            if (chart.Series.Contains(paretoSeries)) chart.Series.Remove(paretoSeries);
            BuildEmptySeries();
        }

        private void ToggleSeriesData(Series series)
        {
            if (series.Points.Count > 0)
            {  //checks if series is shown
                series.Points.Clear();
            }
            else if (Content != null)
            {
                switch (series.Name)
                {
                    case PARETO_FRONT:
                        fillSeries(Content.ParetoFront, null, this.chart.Series[PARETO_FRONT]);
                        break;
                    case QUALITIES:
                        fillSeries(Content.Qualities, Content.Solutions, this.chart.Series[QUALITIES]);
                        break;
                }
            }
        }

        private void chart_MouseDown(object sender, MouseEventArgs e)
        {
            HitTestResult result = chart.HitTest(e.X, e.Y);
            if (result.ChartElementType == ChartElementType.LegendItem)
            {
                this.ToggleSeriesData(result.Series);
            }

        }

        private void chart_MouseMove(object sender, MouseEventArgs e)
        {
            HitTestResult result = chart.HitTest(e.X, e.Y);
            if (result.ChartElementType == ChartElementType.LegendItem)
                this.Cursor = Cursors.Hand;
            else
                this.Cursor = Cursors.Default;
        }

        private void chart_AxisViewChanged(object sender, System.Windows.Forms.DataVisualization.Charting.ViewEventArgs e)
        {
            this.chart.ChartAreas[0].AxisX.ScaleView.Size = e.NewSize;
            this.chart.ChartAreas[0].AxisY.ScaleView.Size = e.NewSize;
        }

        private void chart_CustomizeLegend(object sender, CustomizeLegendEventArgs e)
        {
            if (this.chart.Series.Contains(qualitySeries)) e.LegendItems[0].Cells[1].ForeColor = this.chart.Series[QUALITIES].Points.Count == 0 ? Color.Gray : Color.Black;
            if (this.chart.Series.Contains(paretoSeries)) e.LegendItems[1].Cells[1].ForeColor = this.chart.Series[PARETO_FRONT].Points.Count == 0 ? Color.Gray : Color.Black;
        }

        private void AddMenuItems()
        {
            chooseDimensionToolStripMenuItem.DropDownItems.Clear();
            chooseYDimensionToolStripMenuItem.DropDownItems.Clear();
            if (Content == null) { return; }
            int i = 0;
            for (; i < Content.Objectives; i++)
            {
                //add Menu Points
                ToolStripMenuItem xItem = makeMenuItem("X", "Objective " + i, i);
                ToolStripMenuItem yItem = makeMenuItem("Y", "Objective " + i, i);
                xItem.Click += new System.EventHandler(this.XMenu_Click);
                yItem.Click += new System.EventHandler(this.YMenu_Click);
                chooseDimensionToolStripMenuItem.DropDownItems.Add(xItem);
                chooseYDimensionToolStripMenuItem.DropDownItems.Add(yItem);
            }

            for (; i < Content.Solutions[0].Length + Content.Objectives; i++)
            {
                ToolStripMenuItem xItem = makeMenuItem("X", "ProblemDimension " + (i - Content.Objectives), i);
                ToolStripMenuItem yItem = makeMenuItem("Y", "ProblemDimension " + (i - Content.Objectives), i); ;
                xItem.Click += new System.EventHandler(this.XMenu_Click);
                yItem.Click += new System.EventHandler(this.YMenu_Click);
                chooseDimensionToolStripMenuItem.DropDownItems.Add(xItem);
                chooseYDimensionToolStripMenuItem.DropDownItems.Add(yItem);
            }
        }

        private ToolStripMenuItem makeMenuItem(String axis, String label, int i)
        {
            ToolStripMenuItem xItem = new ToolStripMenuItem();
            xItem.Name = "obj" + i;
            xItem.Size = new System.Drawing.Size(269, 38);
            xItem.Text = label;
            return xItem;
        }

        private void YMenu_Click(object sender, EventArgs e)
        {
            ToolStripMenuItem item = (ToolStripMenuItem)sender;
            yDim = Int32.Parse(item.Name.Remove(0, 3));
            String label = item.Text;
            this.chooseYDimensionToolStripMenuItem.Text = label;
            this.chart.ChartAreas[0].AxisY.Title = label;
            UpdateChart();
        }

        private void XMenu_Click(object sender, EventArgs e)
        {
            ToolStripMenuItem item = (ToolStripMenuItem)sender;
            xDim = Int32.Parse(item.Name.Remove(0, 3));
            String label = item.Text;
            this.chooseDimensionToolStripMenuItem.Text = label;
            this.chart.ChartAreas[0].AxisX.Title = label;
            UpdateChart();
        }

        private void fillSeries(double[][] qualities, double[][] solutions, Series series)
        {
            series.Points.Clear();
            if (qualities == null || qualities.Length == 0) return;
            int jx = xDim - qualities[0].Length;
            int jy = yDim - qualities[0].Length;
            if ((jx >= 0 || jy >= 0) && solutions == null)
            {
                return;
            }
            for (int i = 0; i < qualities.Length; i++)
            {   //Assumtion: Columnwise
                double[] d = qualities[i];
                double[] q = null;
                if (jx >= 0 || jy >= 0) { q = solutions[i]; }
                series.Points.AddXY(jx < 0 ? d[xDim] : q[jx], jy < 0 ? d[yDim] : q[jy]);
            }
        }

        private void BuildEmptySeries()
        {

            this.chart.Series.Add(QUALITIES);
            qualitySeries = this.chart.Series[QUALITIES];

            this.chart.Series[QUALITIES].LegendText = QUALITIES;
            this.chart.Series[QUALITIES].ChartType = SeriesChartType.FastPoint;

            this.chart.Series.Add(PARETO_FRONT);
            paretoSeries = this.chart.Series[PARETO_FRONT];
            paretoSeries.Color = Color.FromArgb(100, Color.Orange);
            this.chart.Series[PARETO_FRONT].LegendText = PARETO_FRONT;
            this.chart.Series[PARETO_FRONT].ChartType = SeriesChartType.FastPoint;
        }
    }
}