LibDraw: Add orientation-based size helpers to Size as well.

Now you can ask for e.g Size::primary_size_for_orientation(Orientation).

1 files changed, 27 insertions, 0 deletions

diff --git a/Libraries/LibDraw/Size.h b/Libraries/LibDraw/Size.h
index a499dbc2fd..a9209ab448 100644
--- a/Libraries/LibDraw/Size.h
+++ b/Libraries/LibDraw/Size.h
@@ -2,6 +2,7 @@
 
 #include <AK/AKString.h>
 #include <AK/LogStream.h>
+#include <LibDraw/Orientation.h>
 
 struct WSAPI_Size;
 
@@ -50,6 +51,32 @@ public:
         return *this;
     }
 
+    int primary_size_for_orientation(Orientation orientation) const
+    {
+        return orientation == Orientation::Vertical ? height() : width();
+    }
+
+    void set_primary_size_for_orientation(Orientation orientation, int value)
+    {
+        if (orientation == Orientation::Vertical)
+            set_height(value);
+        else
+            set_width(value);
+    }
+
+    int secondary_size_for_orientation(Orientation orientation) const
+    {
+        return orientation == Orientation::Vertical ? width() : height();
+    }
+
+    void set_secondary_size_for_orientation(Orientation orientation, int value)
+    {
+        if (orientation == Orientation::Vertical)
+            set_width(value);
+        else
+            set_height(value);
+    }
+
     operator WSAPI_Size() const;
 
     String to_string() const { return String::format("[%dx%d]", m_width, m_height); }