// Copyright (C) Stichting Deltares 2016. All rights preserved.
//
// This file is part of Ringtoets.
//
// Ringtoets is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// 
// This program 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
//
// All names, logos, and references to "Deltares" are registered trademarks of 
// Stichting Deltares and remain full property of Stichting Deltares at all times. 
// All rights preserved.

using System.Linq;
using System.Windows.Forms;
using Core.Common.Base;
using Core.Common.Controls.TreeView;
using Core.Components.Charting.Data;
using Core.Components.OxyPlot.Forms;
using OxyPlotResources = Core.Plugins.OxyPlot.Properties.Resources;
using GuiResources = Core.Common.Gui.Properties.Resources;
using TreeView = Core.Common.Controls.TreeView.TreeView;

namespace Core.Plugins.OxyPlot.Legend
{
    /// <summary>
    /// This class describes the view for showing and configuring the data of a <see cref="BaseChart"/>.
    /// </summary>
    public class LegendTreeView : TreeView
    {
        /// <summary>
        /// Creates a new instance of <see cref="LegendTreeView"/>.
        /// </summary>
        public LegendTreeView()
        {
            TreeViewController.RegisterTreeNodeInfo(new TreeNodeInfo<PointData>
            {
                Text = pointData => OxyPlotResources.ChartDataNodePresenter_Point_data_label,
                Image = pointData => OxyPlotResources.PointsIcon,
                CanDrag = (pointData, sourceNode) => DragOperations.Move,
                CanCheck = pointData => true,
                IsChecked = pointData => pointData.IsVisible,
                OnNodeChecked = PointBasedChartDataOnNodeChecked
            });

            TreeViewController.RegisterTreeNodeInfo(new TreeNodeInfo<LineData>
            {
                Text = lineData => OxyPlotResources.ChartDataNodePresenter_Line_data_label,
                Image = lineData => OxyPlotResources.LineIcon,
                CanDrag = (lineData, sourceNode) => DragOperations.Move,
                CanCheck = lineData => true,
                IsChecked = lineData => lineData.IsVisible,
                OnNodeChecked = PointBasedChartDataOnNodeChecked
            });

            TreeViewController.RegisterTreeNodeInfo(new TreeNodeInfo<AreaData>
            {
                Text = areaData => OxyPlotResources.ChartDataNodePresenter_Area_data_label,
                Image = areaData => OxyPlotResources.AreaIcon,
                CanDrag = (areaData, sourceNode) => DragOperations.Move,
                CanCheck = areaData => true,
                IsChecked = areaData => areaData.IsVisible,
                OnNodeChecked = PointBasedChartDataOnNodeChecked
            });

            TreeViewController.RegisterTreeNodeInfo(new TreeNodeInfo<ChartDataCollection>
            {
                Text = baseChart => OxyPlotResources.General_Chart,
                Image = baseChart => GuiResources.folder,
                ChildNodeObjects = baseChart => baseChart.List.Reverse().Cast<object>().ToArray(),
                CanDrop = BaseChartCanDrop,
                CanInsert = BaseChartCanInsert,
                OnDrop = BaseChartOnDrop
            });
        }

        /// <summary>
        /// Gets or sets the <see cref="BaseChart"/> that is used as the source of the <see cref="LegendTreeView"/>.
        /// </summary>
        public ChartData ChartData
        {
            get
            {
                return (ChartData) TreeViewController.Data;
            }
            set
            {
                TreeViewController.Data = value;

                if (value == null)
                {
                    Nodes.Clear();
                }
            }
        }

        protected override void Dispose(bool disposing)
        {
            ChartData = null;
            base.Dispose(disposing);
        }

        # region ChartData

        private void PointBasedChartDataOnNodeChecked(TreeNode node)
        {
            var pointBasedChartData = node.Tag as PointBasedChartData;
            if (pointBasedChartData != null)
            {
                pointBasedChartData.IsVisible = node.Checked;
                pointBasedChartData.NotifyObservers();

                var parentData = node.Parent != null ? node.Parent.Tag as IObservable : null;
                if (parentData != null)
                {
                    parentData.NotifyObservers();
                }
            }
        }

        # endregion

        # region ChartDataCollection

        private DragOperations BaseChartCanDrop(TreeNode sourceNode, TreeNode targetNode, DragOperations validOperations)
        {
            if (sourceNode.Tag is ChartData)
            {
                return validOperations;
            }

            return DragOperations.None;
        }

        private bool BaseChartCanInsert(TreeNode sourceNode, TreeNode targetNode)
        {
            return sourceNode.Tag is ChartData;
        }

        private void BaseChartOnDrop(TreeNode sourceNode, TreeNode targetNode, DragOperations operation, int position)
        {
            var draggedData = (ChartData) sourceNode.Tag;
            var target = (ChartDataCollection) targetNode.Tag;
            target.List.Remove(draggedData);
            target.List.Insert(target.List.Count - position, draggedData);
            target.NotifyObservers();
        }

        # endregion
    }
}