UniformGrid移除子项时重建所有子控件的问题咨询

问题场景 #

开发WPF应用时，使用UniformGrid作为ItemsControl的面板容器，承载最多48个自定义WellView控件，XAML代码如下：

<ItemsControl
    ItemsSource="{Binding WellsCollection, IsAsync=True}">
    <ItemsControl.ItemsPanel>
        <ItemsPanelTemplate>
            <UniformGrid Columns="4" />
        </ItemsPanelTemplate>
    </ItemsControl.ItemsPanel>
    <ItemsControl.ItemTemplate>
        <DataTemplate DataType="{x:Type controls:WellViewModel}">
            <controls:WellView/>
        </DataTemplate>
    </ItemsControl.ItemTemplate>
</ItemsControl>

其中WellsCollection是ObservableCollection<WellViewModel>类型，WellView继承自System.Windows.Controls.Control。

异常现象 #

从WellsCollection中移除一项时，所有现有WellView控件都会被重建。例如容器中有48个控件时，移除一个会触发47个新控件的构造函数执行。通过断点捕获的堆栈跟踪如下：

WellAnalysis.dll!WellAnalysis.Controls.WellView.WellView() Line 11  C#
    System.Private.CoreLib.dll!System.RuntimeType.CreateInstanceImpl(System.Reflection.BindingFlags bindingAttr, System.Reflection.Binder binder, object[] args, System.Globalization.CultureInfo culture)  Unknown
    System.Xaml.dll!System.Xaml.Schema.XamlTypeInvoker.CreateInstance(object[] arguments)   Unknown
    System.Xaml.dll!MS.Internal.Xaml.Runtime.ClrObjectRuntime.CreateInstance(System.Xaml.XamlType xamlType, object[] args)  Unknown
    System.Xaml.dll!MS.Internal.Xaml.Runtime.PartialTrustTolerantRuntime.CreateInstance(System.Xaml.XamlType xamlType, object[] args)   Unknown
    System.Xaml.dll!System.Xaml.XamlObjectWriter.Logic_CreateAndAssignToParentStart(MS.Internal.Xaml.Context.ObjectWriterContext ctx)   Unknown
    System.Xaml.dll!System.Xaml.XamlObjectWriter.WriteEndObject()   Unknown
    PresentationFramework.dll!System.Windows.FrameworkTemplate.LoadTemplateXaml(System.Xaml.XamlReader templateReader, System.Xaml.XamlObjectWriter currentWriter)  Unknown
    PresentationFramework.dll!System.Windows.FrameworkTemplate.LoadOptimizedTemplateContent(System.Windows.DependencyObject container, System.Windows.Markup.IComponentConnector componentConnector, System.Windows.Markup.IStyleConnector styleConnector, System.Collections.Generic.List<System.Windows.DependencyObject> affectedChildren, System.Windows.UncommonField<System.Collections.Hashtable> templatedNonFeChildrenField)   Unknown
    PresentationFramework.dll!System.Windows.FrameworkTemplate.LoadContent(System.Windows.DependencyObject container, System.Collections.Generic.List<System.Windows.DependencyObject> affectedChildren)    Unknown
    PresentationFramework.dll!System.Windows.StyleHelper.ApplyTemplateContent(System.Windows.UncommonField<System.Collections.Specialized.HybridDictionary[]> dataField, System.Windows.DependencyObject container, System.Windows.FrameworkElementFactory templateRoot, int lastChildIndex, System.Collections.Specialized.HybridDictionary childIndexFromChildID, System.Windows.FrameworkTemplate frameworkTemplate) Unknown
    PresentationFramework.dll!System.Windows.FrameworkElement.ApplyTemplate()   Unknown
    PresentationFramework.dll!System.Windows.FrameworkElement.MeasureCore(System.Windows.Size availableSize)    Unknown
    PresentationCore.dll!System.Windows.UIElement.Measure(System.Windows.Size availableSize)    Unknown
    PresentationFramework.dll!System.Windows.Controls.Primitives.UniformGrid.MeasureOverride(System.Windows.Size constraint)    Unknown

控件全量重建会导致明显视觉闪烁，用户体验极差。

问题原因 #

为什么UIElement.Measure()会触发控件重建？ #

1. UniformGrid的布局特性：UniformGrid在子元素数量变化时，会重新计算所有单元格的尺寸，触发MeasureOverride执行全量布局测量。
2. 模板重新加载连锁反应：布局测量会触发父容器（ItemsControl的ContentPresenter）调用ApplyTemplate()，而DataTemplate在默认情况下，会因布局上下文变化重新加载内容，最终导致所有WellView实例被重新构造。
3. 容器复用机制失效：ItemsControl默认的容器复用逻辑在UniformGrid的全量布局变更下无法生效，因为布局系统认为所有元素的位置和尺寸都可能发生变化，需要重新创建控件以保证布局正确性。

解决方案 #

1. 替换UniformGrid为WrapPanel #

WrapPanel的布局变更仅影响局部区域，不会触发全量控件重建，适合固定列数的网格场景：

<ItemsControl.ItemsPanel>
    <ItemsPanelTemplate>
        <WrapPanel Orientation="Horizontal" Width="800"/> <!-- 根据实际需求设置宽度，确保每行显示4个元素 -->
    </ItemsPanelTemplate>
</ItemsControl.ItemsPanel>

2. 优化DataTemplate的复用逻辑 #

为DataTemplate添加x:Shared="False"标记，确保控件实例可以被正确复用，避免模板重复加载：

<DataTemplate DataType="{x:Type controls:WellViewModel}" x:Shared="False">
    <controls:WellView/>
</DataTemplate>

3. 自定义优化版UniformGrid #

如果必须使用UniformGrid，可以继承并重写布局方法，避免全量测量触发的模板重建：

public class OptimizedUniformGrid : UniformGrid
{
    private Size _lastCellSize;

    protected override Size MeasureOverride(Size constraint)
    {
        if (Children.Count == 0)
            return base.MeasureOverride(constraint);

        // 计算当前单元格尺寸，仅当尺寸变化时才触发全量测量
        int rows = (int)Math.Ceiling((double)Children.Count / Columns);
        var currentCellSize = new Size(
            constraint.Width / Columns,
            constraint.Height / rows
        );

        if (_lastCellSize.Equals(currentCellSize))
        {
            // 尺寸未变，直接复用之前的测量结果
            foreach (UIElement child in Children)
            {
                child.Measure(currentCellSize);
            }
            return new Size(Columns * currentCellSize.Width, rows * currentCellSize.Height);
        }

        _lastCellSize = currentCellSize;
        return base.MeasureOverride(constraint);
    }
}

使用时替换原UniformGrid为local:OptimizedUniformGrid即可。

4. 检查WellView的内部实现 #

确保WellView的ControlTemplate没有包含会触发频繁更新的绑定或动态资源，同时重写OnApplyTemplate方法，避免不必要的初始化逻辑：

public class WellView : Control
{
    protected override void OnApplyTemplate()
    {
        // 仅当模板首次加载或关键元素变更时执行初始化
        if (Template == null)
            return;
            
        // 你的初始化逻辑
        base.OnApplyTemplate();
    }
}

内容的提问来源于stack exchange，提问作者Alex Gdalevich