Use SelectionModel in TreeView.

AvaloniaUI · Feb 7, 2020 · 24a0541 · 24a0541
1 parent 9b77a73
commit 24a0541
Show file tree

Hide file tree

Showing 4 changed files with 306 additions and 388 deletions.
diff --git a/src/Avalonia.Controls/SelectionModel.cs b/src/Avalonia.Controls/SelectionModel.cs
@@ -6,7 +6,6 @@
 using System;
 using System.Collections.Generic;
 using System.ComponentModel;
-using System.Linq;
 using Avalonia.Controls.Utils;
 
 #nullable enable
@@ -117,6 +116,8 @@ public bool AutoSelect
             }
         }
 
+        public bool SelectOnlyLeafNodes { get; set; }
+
         public IndexPath AnchorIndex
         {
             get
@@ -667,9 +668,16 @@ private void SelectRangeImpl(IndexPath start, IndexPath end, bool select)
                 winrtEnd,
                 info =>
                 {
-                    if (info.Node.DataCount == 0)
+                    if (SelectOnlyLeafNodes)
+                    {
+                        if (info.Node.DataCount == 0)
+                        {
+                            // Select only leaf nodes
+                            info.ParentNode!.Select(info.Path.GetAt(info.Path.GetSize() - 1), select);
+                        }
+                    }
+                    else
                     {
-                        // Select only leaf nodes
                         info.ParentNode!.Select(info.Path.GetAt(info.Path.GetSize() - 1), select);
                     }
                 });