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 ++++++++++
 src/ui/controls/TextControlService.hpp | 12 +++---------
 2 files changed, 13 insertions(+), 9 deletions(-)

(limited to 'src/ui')

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;
diff --git a/src/ui/controls/TextControlService.hpp b/src/ui/controls/TextControlService.hpp
index 04807c30..376f9177 100644
--- a/src/ui/controls/TextControlService.hpp
+++ b/src/ui/controls/TextControlService.hpp
@@ -206,16 +206,12 @@ class TextControlService : public Object {
     SetSelection(GetSelection().GetStart() + text.size());
   }
 
-  void ScrollToCaret(bool next_tick = true) {
-    if (next_tick) {
-      scroll_to_caret_timer_canceler_.Reset(
-          GetUiApplication()->GetInstance()->SetImmediate(
-              [this]() { this->ScrollToCaret(false); }));
-    } else {
+  void ScrollToCaret() {
+    this->control_->GetUiHost()->RunAfterLayoutStable([this]() {
       const auto caret_rect = this->GetTextRenderObject()->GetCaretRect();
       this->GetScrollRenderObject()->ScrollToContain(caret_rect,
                                                      Thickness{5.f});
-    }
+    });
   }
 
  private:
@@ -453,8 +449,6 @@ class TextControlService : public Object {
 
   ShortcutHub shortcut_hub_;
 
-  platform::native::TimerAutoCanceler scroll_to_caret_timer_canceler_;
-
   // nullopt means not selecting
   std::optional<MouseButton> select_down_button_;
 
-- 
cgit v1.2.3