WPF 扩展TreeView 自定义图标和可隐藏checkbox

要扩展 WPF TreeView 控件以自定义图标和可隐藏复选框，你可以通过继承 TreeView 类来实现自定义控件。

以下是一个自定义 TreeView 控件的示例代码：

public class CustomTreeView : TreeView
{
    public static readonly DependencyProperty IconProperty =
        DependencyProperty.Register("Icon", typeof(ImageSource), typeof(CustomTreeView));

    public ImageSource Icon
    {
        get { return (ImageSource)GetValue(IconProperty); }
        set { SetValue(IconProperty, value); }
    }

    protected override DependencyObject GetContainerForItemOverride()
    {
        return new CustomTreeViewItem();
    }

    protected override bool IsItemItsOwnContainerOverride(object item)
    {
        return item is CustomTreeViewItem;
    }
}

public class CustomTreeViewItem : TreeViewItem
{
    public static readonly DependencyProperty ShowCheckBoxProperty =
        DependencyProperty.Register("ShowCheckBox", typeof(bool), typeof(CustomTreeViewItem), new PropertyMetadata(true));

    public bool ShowCheckBox
    {
        get { return (bool)GetValue(ShowCheckBoxProperty); }
        set { SetValue(ShowCheckBoxProperty, value); }
    }

    public static readonly DependencyProperty IconProperty =
        DependencyProperty.Register("Icon", typeof(ImageSource), typeof(CustomTreeViewItem));

    public ImageSource Icon
    {
        get { return (ImageSource)GetValue(IconProperty); }
        set { SetValue(IconProperty, value); }
    }

    public override void OnApplyTemplate()
    {
        base.OnApplyTemplate();

        var checkBox = GetTemplateChild("PART_CheckBox") as CheckBox;
        if (checkBox != null)
        {
            checkBox.Visibility = ShowCheckBox ? Visibility.Visible : Visibility.Collapsed;
        }

        var icon = GetTemplateChild("PART_Icon") as Image;
        if (icon != null)
        {
            icon.Source = Icon;
            icon.Visibility = Icon != null ? Visibility.Visible : Visibility.Collapsed;
        }
    }
}

在上面的代码中，自定义 TreeView 控件包含一个 Icon 属性，该属性可以设置 TreeView 的图标。自定义 TreeViewItem 控件包含一个 ShowCheckBox 属性，该属性可以控制复选框是否可见，以及一个 Icon 属性，该属性可以设置 TreeViewItem 的图标。在 OnApplyTemplate 方法中，我们可以获取 CheckBox 和 Image 控件并设置它们的可见性和图标源。

在 XAML 中使用自定义 TreeView 控件可以像这样：

<local:CustomTreeView>
    <local:CustomTreeViewItem Header="Item 1" Icon="/Images/Icon1.png">
        <local:CustomTreeViewItem Header="Subitem 1.1" Icon="/Images/Icon2.png"/>
        <local:CustomTreeViewItem Header="Subitem 1.2" Icon="/Images/Icon3.png"/>
    </local:CustomTreeViewItem>
    <local:CustomTreeViewItem Header="Item 2" Icon="/Images/Icon4.png">
        <local:CustomTreeViewItem Header="Subitem 2.1" Icon="/Images/Icon5.png"/>
        <local:CustomTreeViewItem Header="Subitem 2.2" Icon="/Images/Icon6.png"/>
    </local:CustomTreeViewItem>
</local:CustomTreeView>

在上面的 XAML 代码中，我们可以使用 Icon 属性来设置 TreeView 和 TreeViewItem 的图标，并使用 ShowCheckBox 属性来控制 TreeViewItem 的复选框是否可见。