Fix relayout system.

1 files changed, 12 insertions, 0 deletions

diff --git a/src/ui/layout_base.h b/src/ui/layout_base.h
index 163b99b2..126437cd 100644
--- a/src/ui/layout_base.h
+++ b/src/ui/layout_base.h
@@ -11,6 +11,7 @@ namespace cru
     namespace ui
     {
         class Control;
+        class Window;
 
         enum class Alignment
         {
@@ -114,6 +115,9 @@ namespace cru
             LayoutManager& operator=(LayoutManager&& other) = delete;
             ~LayoutManager() override = default;
 
+
+            //*************** region: position cache ***************
+
             //Mark position cache of the control and its descendants invalid,
             //(which is saved as an auto-managed list internal)
             //and send a message to refresh them.
@@ -126,11 +130,19 @@ namespace cru
             //Refresh position cache of the control and its descendants immediately.
             static void RefreshControlPositionCache(Control* control);
 
+
+            //*************** region: layout ***************
+
+            void InvalidateWindowLayout(Window* window);
+
+            void RefreshInvalidWindowLayout();
+
         private:
             static void RefreshControlPositionCacheInternal(Control* control, const Point& parent_lefttop_absolute);
 
         private:
             std::unordered_set<Control*> cache_invalid_controls_;
+            std::unordered_set<Window*> layout_invalid_windows_;
         };
     }
 }