WPF的TreeView中实现右键选定

在wpf中，treeview默认情况是不支持右键选定的，也就是说，当右键点击某节点时，是无法选中该节点的。当我们想在treeviewitem中实现右键菜单时，往往希望在弹出菜单的同时选中该节点，以使得菜单针对选中的节点生效。

    

图1：虽然是在gnu节点上弹出的右键菜单，但选中的节点仍然是上次左键单击的doc节点。
图2：弹出的右键菜单同时选中gnu节点才是我们所需要的效果

实现这个功能并不是很难，我最开始的做法就是目前网上流行的版本：

1. 在treeviewitem中响应 previewmouserightbuttondown 事件
2. 在响应右键单击事件前选中sender(treeviewitem)节点。

private void treeviewitem_previewmouserightbuttondown(object sender, mousebuttoneventargs e)
{
    var treeviewitem = sender as treeviewitem;
    if (treeviewitem != null)
    {
        treeviewitem.focus();
        e.handled = true;
    }
}

但这样做还存在一个问题：当选择子节点时，sender并不是子节点的treeviewitem，而是其父节点。导致无法选择选择子节点。如下图所示：

  

要解决这个问题也不难，那就是根据mousebuttoneventargs的originalsource来获取节点。但是originalsource也不是treeviewitem，而是产生鼠标事件的子控件，因此还得往上查找，从而找到所属的treeviewitem。

最终的解决方案如下：

1. 在treeviewitem中响应previewmouserightbuttondown事件

<treeview.itemcontainerstyle>
    <style targettype="{x:type treeviewitem}">
        <eventsetter event="treeviewitem.previewmouserightbuttondown" handler="treeviewitem_previewmouserightbuttondown"/>
    </style>
</treeview.itemcontainerstyle>

2. 在响应右键单击事件前选中treeviewitem节点。

private void treeviewitem_previewmouserightbuttondown(object sender, mousebuttoneventargs e)
{
    var treeviewitem = visualupwardsearch<treeviewitem>(e.originalsource as dependencyobject) as treeviewitem;
    if (treeviewitem != null)
    {
        treeviewitem.focus();
        e.handled = true;
    }
}

static dependencyobject visualupwardsearch<t>(dependencyobject source)
{
    while (source != null && source.gettype() != typeof(t))
        source = visualtreehelper.getparent(source);

    return source;
}

现在才是一个比较完美的解决方案。