LibGUI: Allow TabWidget tabs to have icons and custom text alignment

2 files changed, 36 insertions, 5 deletions

diff --git a/Libraries/LibGUI/TabWidget.cpp b/Libraries/LibGUI/TabWidget.cpp
index 1be9c6f9d9..36bc3d91f0 100644
--- a/Libraries/LibGUI/TabWidget.cpp
+++ b/Libraries/LibGUI/TabWidget.cpp
@@ -27,6 +27,7 @@
 #include <LibGUI/BoxLayout.h>
 #include <LibGUI/Painter.h>
 #include <LibGUI/TabWidget.h>
+#include <LibGfx/Bitmap.h>
 #include <LibGfx/Font.h>
 #include <LibGfx/Palette.h>
 #include <LibGfx/StylePainter.h>
@@ -43,7 +44,7 @@ TabWidget::~TabWidget()
 
 void TabWidget::add_widget(const StringView& title, Widget& widget)
 {
-    m_tabs.append({ title, &widget });
+    m_tabs.append({ title, nullptr, &widget });
     add_child(widget);
 }
 
@@ -159,13 +160,25 @@ void TabWidget::paint_event(PaintEvent& event)
     if (container_padding() > 0)
         Gfx::StylePainter::paint_frame(painter, container_rect, palette(), Gfx::FrameShape::Container, Gfx::FrameShadow::Raised, 2);
 
+    auto paint_tab_icon_if_needed = [&](auto& icon, auto& button_rect, auto& text_rect) {
+        if (!icon)
+            return;
+        Gfx::Rect icon_rect { button_rect.x(), button_rect.y(), 16, 16 };
+        icon_rect.move_by(3, 3);
+        painter.draw_scaled_bitmap(icon_rect, *icon, icon->rect());
+        text_rect.set_x(icon_rect.right() + 1 + 4);
+        text_rect.intersect(button_rect);
+    };
+
     for (size_t i = 0; i < m_tabs.size(); ++i) {
         if (m_tabs[i].widget == m_active_widget)
             continue;
         bool hovered = static_cast<int>(i) == m_hovered_tab_index;
         auto button_rect = this->button_rect(i);
         Gfx::StylePainter::paint_tab_button(painter, button_rect, palette(), false, hovered, m_tabs[i].widget->is_enabled());
-        painter.draw_text(button_rect.translated(0, 1), m_tabs[i].title, Gfx::TextAlignment::Center, palette().button_text());
+        auto text_rect = button_rect.translated(0, 1);
+        paint_tab_icon_if_needed(m_tabs[i].icon, button_rect, text_rect);
+        painter.draw_text(text_rect, m_tabs[i].title, m_text_alignment, palette().button_text());
     }
 
     for (size_t i = 0; i < m_tabs.size(); ++i) {
@@ -174,7 +187,9 @@ void TabWidget::paint_event(PaintEvent& event)
         bool hovered = static_cast<int>(i) == m_hovered_tab_index;
         auto button_rect = this->button_rect(i);
         Gfx::StylePainter::paint_tab_button(painter, button_rect, palette(), true, hovered, m_tabs[i].widget->is_enabled());
-        painter.draw_text(button_rect.translated(0, 1), m_tabs[i].title, Gfx::TextAlignment::Center, palette().button_text());
+        auto text_rect = button_rect.translated(0, 1);
+        paint_tab_icon_if_needed(m_tabs[i].icon, button_rect, text_rect);
+        painter.draw_text(text_rect, m_tabs[i].title, m_text_alignment, palette().button_text());
         painter.draw_line(button_rect.bottom_left().translated(1, 1), button_rect.bottom_right().translated(-1, 1), palette().button());
         break;
     }
@@ -199,7 +214,7 @@ Gfx::Rect TabWidget::button_rect(int index) const
 
 int TabWidget::TabData::width(const Gfx::Font& font) const
 {
-    return 16 + font.width(title);
+    return 16 + font.width(title) + (icon ? (16 + 4) : 0);
 }
 
 void TabWidget::mousedown_event(MouseEvent& event)
@@ -277,6 +292,17 @@ void TabWidget::set_tab_title(Widget& tab, const StringView& title)
     }
 }
 
+void TabWidget::set_tab_icon(Widget& tab, const Gfx::Bitmap* icon)
+{
+    for (auto& t : m_tabs) {
+        if (t.widget == &tab) {
+            t.icon = icon;
+            update();
+            return;
+        }
+    }
+}
+
 void TabWidget::activate_next_tab()
 {
     if (m_tabs.size() <= 1)
diff --git a/Libraries/LibGUI/TabWidget.h b/Libraries/LibGUI/TabWidget.h
index 6a8f28b733..0bdffad270 100644
--- a/Libraries/LibGUI/TabWidget.h
+++ b/Libraries/LibGUI/TabWidget.h
@@ -68,10 +68,14 @@ public:
     void remove_tab(Widget& tab) { remove_widget(tab); }
 
     void set_tab_title(Widget& tab, const StringView& title);
+    void set_tab_icon(Widget& tab, const Gfx::Bitmap*);
 
     void activate_next_tab();
     void activate_previous_tab();
 
+    void set_text_alignment(Gfx::TextAlignment alignment) { m_text_alignment = alignment; }
+    Gfx::TextAlignment text_alignment() const { return m_text_alignment; }
+
     Function<void(Widget&)> on_change;
 
 protected:
@@ -94,15 +98,16 @@ private:
     RefPtr<Widget> m_active_widget;
 
     struct TabData {
-        Gfx::Rect rect(const Gfx::Font&) const;
         int width(const Gfx::Font&) const;
         String title;
+        RefPtr<Gfx::Bitmap> icon;
         Widget* widget { nullptr };
     };
     Vector<TabData> m_tabs;
     TabPosition m_tab_position { TabPosition::Top };
     int m_hovered_tab_index { -1 };
     int m_container_padding { 2 };
+    Gfx::TextAlignment m_text_alignment { Gfx::TextAlignment::Center };
 };
 
 }