...

1 files changed, 7 insertions, 1 deletions

diff --git a/include/cru/ui/host/WindowHost.hpp b/include/cru/ui/host/WindowHost.hpp
index 77ed937f..81eabb52 100644
--- a/include/cru/ui/host/WindowHost.hpp
+++ b/include/cru/ui/host/WindowHost.hpp
@@ -7,8 +7,11 @@
 #include "../render/Base.hpp"
 
 #include <functional>
+#include <memory>
 
 namespace cru::ui::host {
+class LayoutPaintCycler;
+
 struct AfterLayoutEventArgs {};
 
 // The bridge between control tree and native window.
@@ -43,6 +46,8 @@ class WindowHost : public Object {
   void Relayout();
   void Relayout(const Size& available_size);
 
+  void Repaint();
+
   // Is layout is invalid, wait for relayout and then run the action. Otherwist
   // run it right now.
   void RunAfterLayoutStable(std::function<void()> action);
@@ -125,7 +130,8 @@ class WindowHost : public Object {
 
   platform::gui::INativeWindow* native_window_ = nullptr;
 
-  bool need_layout_ = false;
+  std::unique_ptr<LayoutPaintCycler> layout_paint_cycler_;
+
   Event<AfterLayoutEventArgs> after_layout_event_;
   std::vector<std::function<void()> > after_layout_stable_action_;