3rdpartylibs/fastreports/Fast-Report/FastReport.DesignExt/Controls/DataTreeView.cs

using System.Collections.Generic;
using System.Windows.Forms;
using System.ComponentModel;
using System.Reflection;
using FastReport.Data;
using FastReport.Utils;
using FastReport.Dialog;

namespace FastReport.Controls
{
    internal interface ICustomDataTreeItemsProvider
    {
        void ProvideCustomItems(DataTreeView tree);
    }

    /// <summary>
    /// This is an internal enum that represents the selected datatree item type.
    /// </summary>
    public enum DataTreeSelectedItemType
    {
        /// <summary>
        /// Nothing selected.
        /// </summary>
        None,

        /// <summary>
        /// Data column selected.
        /// </summary>
        DataColumn,

        /// <summary>
        /// Total selected.
        /// </summary>
        Total,

        /// <summary>
        /// Parameter selected.
        /// </summary>
        Parameter,

        /// <summary>
        /// Function selected.
        /// </summary>
        Function,

        /// <summary>
        /// Custom item selected.
        /// </summary>
        CustomItem
    }

    /// <summary>
    /// This is an internal control that represents the data tree.
    /// </summary>
    /// <remarks>
    /// This control is for internal use only.
    /// </remarks>
    [ToolboxItem(false)]
    public class DataTreeView : TreeView
    {
        private string selectedItem;
        private bool showDataSources;
        private bool showColumns;
        private bool showParameters;
        private bool showVariables;
        private bool showNone;
        private bool showRelations;
        private bool showEnabledOnly;
        private bool showTotals;
        private bool showFunctions;
        private bool showCustomItems;
        private bool showCubeSources;
        private DataSourceBase dataSource;

        /// <summary>
        /// DataSource property
        /// </summary>
        [Browsable(false)]
        [DesignerSerializationVisibility(DesignerSerializationVisibility.Hidden)]
        public DataSourceBase DataSource
        {
            get { return dataSource; }
            set
            {
                dataSource = value;
                if (value != null)
                    RemoveDataSourcesExcept(Nodes, value);
            }
        }

        private void RemoveDataSourcesExcept(TreeNodeCollection nodes, DataSourceBase value)
        {
            for (int i = 0; i < nodes.Count; i++)
            {
                TreeNode node = nodes[i];
                if (node.Tag == value || value.HasParent(node.Tag as Base))
                {
                    node.Expand();
                    continue;
                }
                if (node.Tag is DataConnectionBase)
                    RemoveDataSourcesExcept(node.Nodes, value);
                else
                {
                    node.Remove();
                    i--;
                }
            }
        }

        /// <summary>
        /// SelectedItem property
        /// </summary>
        [Browsable(false), DesignerSerializationVisibility(DesignerSerializationVisibility.Hidden)]
        public string SelectedItem
        {
            get
            {
                selectedItem = "";
                TreeNode node = SelectedNode;
                if (node != null)
                {
                    if (node.Tag is Column && !(node.Tag is DataSourceBase))
                    {
                        while (true)
                        {
                            if (node.Tag is DataSourceBase)
                            {
                                selectedItem = (node.Tag as DataSourceBase).FullName + "." + selectedItem;
                                break;
                            }
                            selectedItem = node.Text + (selectedItem == "" ? "" : ".") + selectedItem;
                            node = node.Parent;
                        }
                    }
                    else if (node.Tag is Parameter || node.Tag is Total)
                    {
                        while (node != null && node.Tag != null)
                        {
                            selectedItem = node.Text + (selectedItem == "" ? "" : ".") + selectedItem;
                            node = node.Parent;
                        }
                    }
                    else if (node.Tag is MethodInfo)
                    {
                        MethodInfo info = node.Tag as MethodInfo;
                        ParameterInfo[] pars = info.GetParameters();

                        selectedItem = info.Name + "(" + (pars.Length > 1 ? "".PadRight(pars.Length - 1, ',') : "") + ")";
                    }
                    else if (node.Tag == null && node.Nodes.Count == 0)
                    {
                        selectedItem = node.Text;
                    }
                }
                return selectedItem;
            }
            set
            {
                selectedItem = value;
                SetSelectedItem(Nodes, "");
            }
        }

        /// <summary>
        /// SelectedItemType property
        /// </summary>
        public DataTreeSelectedItemType SelectedItemType
        {
            get
            {
                TreeNode node = SelectedNode;
                if (node != null)
                {
                    if (node.Tag is Column && !(node.Tag is DataSourceBase))
                        return DataTreeSelectedItemType.DataColumn;
                    else if (node.Tag is Parameter)
                        return DataTreeSelectedItemType.Parameter;
                    else if (node.Tag is Total)
                        return DataTreeSelectedItemType.Total;
                    else if (node.Tag is MethodInfo)
                        return DataTreeSelectedItemType.Function;
                    else if (node.Tag == null && node.Nodes.Count == 0)
                        return DataTreeSelectedItemType.CustomItem;
                }
                return DataTreeSelectedItemType.None;
            }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowDataSources
        {
            get { return showDataSources; }
            set { showDataSources = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowColumns
        {
            get { return showColumns; }
            set { showColumns = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowParameters
        {
            get { return showParameters; }
            set { showParameters = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowVariables
        {
            get { return showVariables; }
            set { showVariables = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowNone
        {
            get { return showNone; }
            set { showNone = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowRelations
        {
            get { return showRelations; }
            set { showRelations = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowEnabledOnly
        {
            get { return showEnabledOnly; }
            set { showEnabledOnly = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowTotals
        {
            get { return showTotals; }
            set { showTotals = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowFunctions
        {
            get { return showFunctions; }
            set { showFunctions = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowCustomItems
        {
            get { return showCustomItems; }
            set { showCustomItems = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ShowCubeSources
        {
            get { return showCubeSources; }
            set { showCubeSources = value; }
        }

        /// <summary>
        /// 
        /// </summary>
        public List<string> ExpandedNodes
        {
            get
            {
                List<string> result = new List<string>();
                GetExpandedNodes(Nodes, result);
                return result;
            }
            set
            {
                EnumNodes(Nodes, value);
            }
        }

        private void GetExpandedNodes(TreeNodeCollection nodes, List<string> list)
        {
            foreach (TreeNode node in nodes)
            {
                if (node.IsExpanded)
                    list.Add(node.FullPath);
                GetExpandedNodes(node.Nodes, list);
            }
        }

        private void EnumNodes(TreeNodeCollection nodes, List<string> list)
        {
            foreach (TreeNode node in nodes)
            {
                if (list.Contains(node.FullPath))
                    node.Expand();
                EnumNodes(node.Nodes, list);
            }
        }

        private void SetSelectedItem(TreeNodeCollection nodes, string prefix)
        {
            foreach (TreeNode node in nodes)
            {
                string nodeName = "";
                if (node.Tag is DataSourceBase || node.Tag is Column || node.Tag is CubeSourceBase)
                    nodeName = prefix + (node.Tag as DataComponentBase).Alias;
                else if (node.Tag is Relation)
                    nodeName = prefix + (node.Tag as Relation).ParentDataSource.Alias;
                else if (node.Tag is Parameter)
                    nodeName = prefix + (node.Tag as Parameter).Name;

                if (nodeName == selectedItem)
                {
                    SelectedNode = node;
                    break;
                }
                SetSelectedItem(node.Nodes, nodeName == "" ? prefix : nodeName + ".");
            }
        }

        /// <summary>
        /// 
        /// </summary>
        /// <param name="dictionary"></param>
        public void CreateNodes(Dictionary dictionary)
        {
            Nodes.Clear();
            if (ShowNone)
            {
                TreeNode noneNode = Nodes.Add(Res.Get("Misc,None"));
                noneNode.ImageIndex = 76;
                noneNode.SelectedImageIndex = noneNode.ImageIndex;
            }
            if (ShowVariables || ShowParameters || ShowTotals)
            {
                if (ShowDataSources)
                {
                    TreeNode dataNode = Nodes.Add(Res.Get("Designer,ToolWindow,Dictionary,DataSources"));
                    dataNode.ImageIndex = 53;
                    dataNode.SelectedImageIndex = dataNode.ImageIndex;
                    DataTreeHelper.CreateDataTree(dictionary, dataNode.Nodes, ShowEnabledOnly, ShowRelations, ShowDataSources,
                      ShowColumns);
                }

                if (ShowParameters && dictionary.Parameters.Count > 0)
                {
                    TreeNode parNode = Nodes.Add(Res.Get("Designer,ToolWindow,Dictionary,Parameters"));
                    parNode.ImageIndex = 234;
                    parNode.SelectedImageIndex = parNode.ImageIndex;
                    DataTreeHelper.CreateParametersTree(dictionary.Parameters, parNode.Nodes);
                }

                if (ShowVariables)
                {
                    TreeNode sysNode = Nodes.Add(Res.Get("Designer,ToolWindow,Dictionary,SystemVariables"));
                    sysNode.ImageIndex = 60;
                    sysNode.SelectedImageIndex = sysNode.ImageIndex;
                    DataTreeHelper.CreateVariablesTree(dictionary.SystemVariables, sysNode.Nodes);
                }

                if (ShowFunctions)
                {
                    TreeNode funcNode = Nodes.Add(Res.Get("Designer,ToolWindow,Dictionary,Functions"));
                    funcNode.ImageIndex = 52;
                    funcNode.SelectedImageIndex = funcNode.ImageIndex;
                    DataTreeHelper.CreateFunctionsTree(dictionary.Report, funcNode.Nodes);
                }

                if (ShowTotals && dictionary.Totals.Count > 0)
                {
                    TreeNode sumNode = Nodes.Add(Res.Get("Designer,ToolWindow,Dictionary,Totals"));
                    sumNode.ImageIndex = 132;
                    sumNode.SelectedImageIndex = sumNode.ImageIndex;
                    DataTreeHelper.CreateTotalsTree(dictionary.Totals, sumNode.Nodes);
                }

                if (ShowCustomItems && dictionary.Report.IsDesigning)
                {
                    SelectedObjectCollection sel = dictionary.Report.Designer.SelectedObjects;
                    if (sel.Count > 0 && sel[0] is ICustomDataTreeItemsProvider)
                        (sel[0] as ICustomDataTreeItemsProvider).ProvideCustomItems(this);
                }

                if (ShowCubeSources && dictionary.CubeSources.Count > 0)
                {
                    TreeNode dataNode = Nodes.Add(Res.Get("Designer,ToolWindow,Dictionary,CubeSources"));
                    dataNode.ImageIndex = 248;
                    dataNode.SelectedImageIndex = dataNode.ImageIndex;
                    DataTreeHelper.CreateCubeTree(dictionary, dataNode.Nodes, ShowEnabledOnly);
                }
            }
            else if (ShowCubeSources)
            {
                DataTreeHelper.CreateCubeTree(dictionary, Nodes, ShowEnabledOnly);
            }
            else
            {
                DataTreeHelper.CreateDataTree(dictionary, Nodes, ShowEnabledOnly, ShowRelations, ShowDataSources, ShowColumns);
            }
        }

        /// <summary>
        /// Creates a new instance of the DataTreeView control.
        /// </summary>
        public DataTreeView()
        {
            ImageList = Res.GetImages(this.Dpi());
            selectedItem = "";
            ShowDataSources = true;
            ShowColumns = true;
            ShowCubeSources = false;
        }
    }
}