decorations: recalc layout and positioner on add/remove

2 files changed, 11 insertions, 0 deletions

diff --git a/src/render/decorations/DecorationPositioner.cpp b/src/render/decorations/DecorationPositioner.cpp
index 937eaa9a..b41b92a9 100644
--- a/src/render/decorations/DecorationPositioner.cpp
+++ b/src/render/decorations/DecorationPositioner.cpp
@@ -95,6 +95,16 @@ void CDecorationPositioner::sanitizeDatas() {
     });
 }
 
+void CDecorationPositioner::forceRecalcFor(CWindow* pWindow) {
+    const auto WIT = std::find_if(m_mWindowDatas.begin(), m_mWindowDatas.end(), [&](const auto& other) { return other.first == pWindow; });
+    if (WIT == m_mWindowDatas.end())
+        return;
+
+    const auto WINDOWDATA = &WIT->second;
+
+    WINDOWDATA->needsRecalc = true;
+}
+
 void CDecorationPositioner::onWindowUpdate(CWindow* pWindow) {
     if (!g_pCompositor->windowExists(pWindow) || !pWindow->m_bIsMapped)
         return;
diff --git a/src/render/decorations/DecorationPositioner.hpp b/src/render/decorations/DecorationPositioner.hpp
index ae4a814a..740b4efe 100644
--- a/src/render/decorations/DecorationPositioner.hpp
+++ b/src/render/decorations/DecorationPositioner.hpp
@@ -71,6 +71,7 @@ class CDecorationPositioner {
     CBox                     getBoxWithIncludedDecos(CWindow* pWindow);
     void                     repositionDeco(IHyprWindowDecoration* deco);
     CBox                     getWindowDecorationBox(IHyprWindowDecoration* deco);
+    void                     forceRecalcFor(CWindow* pWindow);
 
   private:
     struct SWindowPositioningData {