WPF 獲得DataGridRow和 DataGridCell的方法

原文: WPF 獲得DataGridRow和 DataGridCell的方法 原文位址

簡介

　　在WPF中，DataGrid控件并沒有提供通路其DataGridRow或者DataGridCell的方法。

　　是以我們需要自己來編寫擷取的方法，這其中主要用到了ItemsControl類的一個執行個體方法：ItemContainerGenerator。

實作代碼

using System.Windows.Controls;
using System.Windows.Controls.Primitives;
using System.Windows.Media;

namespace Splash.WPF
{
    public static class DataGridPlus
    {
        /// <summary>
        /// 擷取DataGrid控件單元格
        /// </summary>
        /// <param name="dataGrid">DataGrid控件</param>
        /// <param name="rowIndex">單元格所在的行号</param>
        /// <param name="columnIndex">單元格所在的列号</param>
        /// <returns>指定的單元格</returns>
        public static DataGridCell GetCell(this DataGrid dataGrid, int rowIndex, int columnIndex)        
        {
            DataGridRow rowContainer = dataGrid.GetRow(rowIndex);
            if (rowContainer != null)
            {
                DataGridCellsPresenter presenter = GetVisualChild<DataGridCellsPresenter>(rowContainer);
                DataGridCell cell = (DataGridCell)presenter.ItemContainerGenerator.ContainerFromIndex(columnIndex);
                if (cell == null)
                {
                    dataGrid.ScrollIntoView(rowContainer, dataGrid.Columns[columnIndex]);
                    cell = (DataGridCell)presenter.ItemContainerGenerator.ContainerFromIndex(columnIndex);
                }
                return cell;
            }
            return null;
        }

        /// <summary>
        /// 擷取DataGrid的行
        /// </summary>
        /// <param name="dataGrid">DataGrid控件</param>
        /// <param name="rowIndex">DataGrid行号</param>
        /// <returns>指定的行号</returns>
        public static DataGridRow GetRow(this DataGrid dataGrid, int rowIndex)        
        {
            DataGridRow rowContainer = (DataGridRow)dataGrid.ItemContainerGenerator.ContainerFromIndex(rowIndex);
            if (rowContainer == null)
            {
                dataGrid.UpdateLayout();
                dataGrid.ScrollIntoView(dataGrid.Items[rowIndex]);
                rowContainer = (DataGridRow)dataGrid.ItemContainerGenerator.ContainerFromIndex(rowIndex);
            }
            return rowContainer;
        }

        /// <summary>
        /// 擷取父可視對象中第一個指定類型的子可視對象
        /// </summary>
        /// <typeparam name="T">可視對象類型</typeparam>
        /// <param name="parent">父可視對象</param>
        /// <returns>第一個指定類型的子可視對象</returns>
        public static T GetVisualChild<T>(Visual parent) where T : Visual
        {
            T child = default(T);
            int numVisuals = VisualTreeHelper.GetChildrenCount(parent);
            for (int i = 0; i < numVisuals; i++)
            {
                Visual v = (Visual)VisualTreeHelper.GetChild(parent, i);
                child = v as T;
                if (child == null)
                {
                    child = GetVisualChild<T>(v);
                }
                if (child != null)
                {
                    break;
                }
            }
            return child;
        }
    }
}