From 4e0a5338f15a247c33df863c4c619ffdb0a98aae Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Wed, 28 Oct 2020 17:51:41 +0800
Subject: ...

---
 src/ui/UiHost.cpp | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'src/ui/UiHost.cpp')

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;
-- 
cgit v1.2.3