From d5a8064202c8f5bbd7e890a51a6d63351267c442 Mon Sep 17 00:00:00 2001
From: rabbitism <dongbin@irihi.tech>
Date: Thu, 22 Feb 2024 22:52:26 +0800
Subject: [PATCH] feat: implement dynamic measure.

---
 demo/Ursa.Demo/Pages/ToolBarDemo.axaml    |  2 +-
 src/Ursa/Controls/ToolBar/ToolBarPanel.cs | 51 ++++++++++++++++++++++-
 2 files changed, 50 insertions(+), 3 deletions(-)

diff --git a/demo/Ursa.Demo/Pages/ToolBarDemo.axaml b/demo/Ursa.Demo/Pages/ToolBarDemo.axaml
index f6fab35..e9e4367 100644
--- a/demo/Ursa.Demo/Pages/ToolBarDemo.axaml
+++ b/demo/Ursa.Demo/Pages/ToolBarDemo.axaml
@@ -12,7 +12,7 @@
     <StackPanel>
         <u:ToolBar HorizontalAlignment="Left" Header="Hello World">
             <Button Content="Button 1" />
-            <Button Content="Button 2" />
+            <Button Content="Button 2" u:ToolBar.OverflowMode="AsNeeded" />
             <Button Content="Button 3" u:ToolBar.OverflowMode="Always" />
         </u:ToolBar>
         <u:ToolBar ItemsSource="{Binding Items}" HorizontalAlignment="Left">
diff --git a/src/Ursa/Controls/ToolBar/ToolBarPanel.cs b/src/Ursa/Controls/ToolBar/ToolBarPanel.cs
index 3c25ed9..03fe1bb 100644
--- a/src/Ursa/Controls/ToolBar/ToolBarPanel.cs
+++ b/src/Ursa/Controls/ToolBar/ToolBarPanel.cs
@@ -44,9 +44,55 @@ public class ToolBarPanel: StackPanel
         Size measureSize = availableSize;
         bool horizontal = Orientation == Orientation.Horizontal;
         bool hasVisibleChildren = false;
-        measureSize = new Size(double.PositiveInfinity, double.PositiveInfinity);
         int index = 0;
         if (logicalChildren is null) return size;
+        for (int i = 0; i < logicalChildren.Count; i++)
+        {
+            Control control = logicalChildren[i];
+            var mode = ToolBar.GetOverflowMode(control);
+            control.Measure(measureSize);
+            if (mode == OverflowMode.Always)
+            {
+                ToolBar.SetIsOverflowItem(control, true);
+                continue;
+            }
+            else if (mode == OverflowMode.Never)
+            {
+                if (control.IsVisible)
+                {
+                    hasVisibleChildren = true;
+                    size = horizontal
+                        ? size.WithWidth(size.Width + control.DesiredSize.Width + spacing)
+                            .WithHeight(Math.Max(size.Height, control.DesiredSize.Height))
+                        : size.WithHeight(size.Height + control.DesiredSize.Height + spacing)
+                            .WithWidth(Math.Max(size.Width, control.DesiredSize.Width));
+                    ToolBar.SetIsOverflowItem(control, false);
+                }
+            }
+        }
+        for(int i = 0; i < logicalChildren.Count; i++)
+        {
+            Control control = logicalChildren[i];
+            var mode = ToolBar.GetOverflowMode(control);
+            if (mode != OverflowMode.AsNeeded) continue;
+            bool isFit = horizontal
+                ? (size.Width + control.DesiredSize.Width <= availableSize.Width)
+                : (size.Height + control.DesiredSize.Height <= availableSize.Height);
+            if (isFit)
+            {
+                ToolBar.SetIsOverflowItem(control, false);
+                size = horizontal
+                    ? size.WithWidth(size.Width + control.DesiredSize.Width + spacing)
+                        .WithHeight(Math.Max(size.Height, control.DesiredSize.Height))
+                    : size.WithHeight(size.Height + control.DesiredSize.Height + spacing)
+                        .WithWidth(Math.Max(size.Width, control.DesiredSize.Width));
+            }
+            else
+            {
+                ToolBar.SetIsOverflowItem(control, true);
+            }
+        }
+        /*
         for (int count = logicalChildren.Count; index < count; ++index)
         {
             Control control = logicalChildren[index];
@@ -74,11 +120,12 @@ public class ToolBarPanel: StackPanel
                 size = size.WithWidth(Math.Max(size.Width, desiredSize.Width));
             }
         }
-
+        */
         if (hasVisibleChildren)
         {
             size = horizontal ? size.WithWidth(size.Width - spacing) : size.WithHeight(size.Height - spacing);
         }
+
         return size;
     }