在WPF应用程序开发中，TreeView控件是一种常见的UI组件，用于展示具有层级关系的数据。然而，标准的TreeView控件可能无法满足所有需求，因此自定义TreeView控件成为了一种解决方案。本文将介绍如何使用WPF中的ControlTemplate自定义TreeView控件，并实现节点的展开和折叠功能。
自定义TreeView控件的核心在于重写其ControlTemplate。通过这种方式，可以控制TreeView的外观和行为。以下是实现节点展开和折叠功能的基本步骤：
首先，需要定义TreeViewItem的ControlTemplate。在这个模板中，将定义TreeViewItem的布局和样式。
<ControlTemplate TargetType="TreeViewItem">
<Grid>
<Grid.RowDefinitions>
<RowDefinition Height="Auto"/>
<RowDefinition Height="Auto"/>
<RowDefinition Height="*"/>
</Grid.RowDefinitions>
<Grid Grid.Row="0">
<Grid.ColumnDefinitions>
<ColumnDefinition Width="1*"/>
<ColumnDefinition Width="1*"/>
</Grid.ColumnDefinitions>
<Line Grid.Column="0" ... />
<Line Grid.Column="1" ... />
</Grid>
<StackPanel Grid.Row="1" ... />
<ItemsPresenter Grid.Row="2" />
</Grid>
</ControlTemplate>
在这个模板中，定义了三行：第一行用于绘制水平线，第二行用于显示用户控件，最后一行用于显示子节点。
为了在节点之间绘制水平线，需要在第一行中放置一个新的Grid，并为这个Grid定义两个等宽的列。这样，可以在每个节点之间绘制两条半长度的水平线。
根据节点的位置，需要隐藏或显示这些线。如果节点是最左边的一个，隐藏左边的线；如果节点是最右边的一个，隐藏右边的线；如果节点既不是左边也不是右边，显示两条线。
需要在数据库中设置每个节点的首尾标志。这可以通过递归函数实现，每次添加、删除、复制或移动项目时都需要执行这个函数。
private void RecursiveSetFirstLast(Database.FlowElements flowelement)
{
List<Database.FlowElements> toList = flowelement.FlowElementsTo.ToList();
for (int i = 0; i < toList.Count; i++)
{
toList[i].TreeviewItem_IsFirst = (i == 0);
toList[i].TreeviewItem_IsLast = (i == toList.Count - 1);
if (i != 0 && i != toList.Count - 1)
{
toList[i].TreeviewItem_IsFirst = false;
toList[i].TreeviewItem_IsLast = false;
}
RecursiveSetFirstLast(toList[i]);
}
}
这个函数递归地为每个节点设置首尾标志，以便正确绘制树形线。
可以通过转换器来设置线的可见性。需要两个转换器，一个用于左边的线，另一个用于右边的线。
public class IsFlowElementFrom_ToLeftLineVisiblility_Converter : System.Windows.Markup.MarkupExtension, IValueConverter
{
public object Convert(object value, Type targetType, object parameter, System.Globalization.CultureInfo culture)
{
Database.FlowElements flowelement = (Database.FlowElements)value;
if (flowelement == null || flowelement.FlowElementFrom == null)
return Visibility.Hidden;
if (flowelement.TreeviewItem_IsFirst)
return Visibility.Hidden;
return Visibility.Visible;
}
public object ConvertBack(object value, Type targetType, object parameter, System.Globalization.CultureInfo culture)
{
return true;
}
}
这个转换器根据节点的位置来决定线的可见性。
最后，需要在用户控件中绘制两条短的垂直线（顶部和底部），并根据上面的或下面的元素来显示或隐藏它们。
<Menu Background="Transparent" HorizontalAlignment="Center" VerticalAlignment="Center">
<MenuItem x:Name="ButtonExpand" ... Click="ButtonExpand_Click">
<MenuItem.Header>
<TextBlock FontSize="12" ... >
</TextBlock>
</MenuItem.Header>
</MenuItem>
</Menu>