...

1 files changed, 10 insertions, 0 deletions

diff --git a/src/ui/UiHost.cpp b/src/ui/UiHost.cpp
index d3564619..ccc5fea0 100644
--- a/src/ui/UiHost.cpp
+++ b/src/ui/UiHost.cpp
@@ -193,6 +193,8 @@ void UiHost::Relayout() {
                                      : render::MeasureSize::NotSpecified()},
       render::MeasureSize::NotSpecified());
   root_render_object_->Layout(Point{});
+  for (auto& action : after_layout_stable_action_) action();
+  after_layout_stable_action_.clear();
   after_layout_event_.Raise(AfterLayoutEventArgs{});
   log::TagDebug(log_tag, u"A relayout is finished.");
 }
@@ -249,6 +251,14 @@ bool UiHost::CaptureMouseFor(Control* control) {
 
 Control* UiHost::GetMouseCaptureControl() { return mouse_captured_control_; }
 
+void UiHost::RunAfterLayoutStable(std::function<void()> action) {
+  if (need_layout_) {
+    after_layout_stable_action_.push_back(std::move(action));
+  } else {
+    action();
+  }
+}
+
 void UiHost::OnNativeDestroy(INativeWindow* window, std::nullptr_t) {
   CRU_UNUSED(window)
   native_window_destroyed_ = true;