renderer: respect viewporter dest on base surfaces

2 files changed, 12 insertions, 2 deletions

diff --git a/src/helpers/WLSurface.cpp b/src/helpers/WLSurface.cpp
index a34c77a1..3d3800cb 100644
--- a/src/helpers/WLSurface.cpp
+++ b/src/helpers/WLSurface.cpp
@@ -38,11 +38,20 @@ Vector2D CWLSurface::correctSmallVec() const {
     if (!m_pOwner || !exists() || !small() || m_bFillIgnoreSmall)
         return {};
 
-    return Vector2D{(m_pOwner->m_vReportedSize.x - m_pWLRSurface->current.buffer_width) / 2, (m_pOwner->m_vReportedSize.y - m_pWLRSurface->current.buffer_height) / 2}.clamp(
-               {}, {INFINITY, INFINITY}) *
+    const auto SIZE = getViewporterCorrectedSize();
+
+    return Vector2D{(m_pOwner->m_vReportedSize.x - SIZE.x) / 2, (m_pOwner->m_vReportedSize.y - SIZE.y) / 2}.clamp({}, {INFINITY, INFINITY}) *
         (m_pOwner->m_vRealSize.vec() / m_pOwner->m_vReportedSize);
 }
 
+Vector2D CWLSurface::getViewporterCorrectedSize() const {
+    if (!exists())
+        return {};
+
+    return m_pWLRSurface->current.viewport.has_dst ? Vector2D{m_pWLRSurface->current.viewport.dst_width, m_pWLRSurface->current.viewport.dst_height} :
+                                                     Vector2D{m_pWLRSurface->current.buffer_width, m_pWLRSurface->current.buffer_height};
+}
+
 void CWLSurface::destroy() {
     if (!m_pWLRSurface)
         return;
diff --git a/src/helpers/WLSurface.hpp b/src/helpers/WLSurface.hpp
index ee6574de..8b4ef835 100644
--- a/src/helpers/WLSurface.hpp
+++ b/src/helpers/WLSurface.hpp
@@ -22,6 +22,7 @@ class CWLSurface {
     bool         exists() const;
     bool         small() const;           // means surface is smaller than the requested size
     Vector2D     correctSmallVec() const; // returns a corrective vector for small() surfaces
+    Vector2D     getViewporterCorrectedSize() const;
 
     // allow stretching. Useful for plugins.
     bool m_bFillIgnoreSmall = false;