1 files changed, 147 insertions, 0 deletions
diff --git a/src/ui/Avatar.cpp b/src/ui/Avatar.cpp
new file mode 100644
index 00000000..4b4cd272
--- /dev/null
+++ b/src/ui/Avatar.cpp
@@ -0,0 +1,147 @@
+#include <QPainter>
+
+#include "Utils.h"
+#include "ui/Avatar.h"
+
+Avatar::Avatar(QWidget *parent)
+  : QWidget(parent)
+{
+        size_   = ui::AvatarSize;
+        type_   = ui::AvatarType::Letter;
+        letter_ = "A";
+
+        QFont _font(font());
+        _font.setPointSizeF(ui::FontSize);
+        setFont(_font);
+
+        QSizePolicy policy(QSizePolicy::MinimumExpanding, QSizePolicy::MinimumExpanding);
+        setSizePolicy(policy);
+}
+
+QColor
+Avatar::textColor() const
+{
+        if (!text_color_.isValid())
+                return QColor("black");
+
+        return text_color_;
+}
+
+QColor
+Avatar::backgroundColor() const
+{
+        if (!text_color_.isValid())
+                return QColor("white");
+
+        return background_color_;
+}
+
+int
+Avatar::size() const
+{
+        return size_;
+}
+
+QSize
+Avatar::sizeHint() const
+{
+        return QSize(size_ + 2, size_ + 2);
+}
+
+void
+Avatar::setTextColor(const QColor &color)
+{
+        text_color_ = color;
+}
+
+void
+Avatar::setBackgroundColor(const QColor &color)
+{
+        background_color_ = color;
+}
+
+void
+Avatar::setSize(int size)
+{
+        size_ = size;
+
+        if (!image_.isNull())
+                pixmap_ = utils::scaleImageToPixmap(image_, size_);
+
+        QFont _font(font());
+        _font.setPointSizeF(size_ * (ui::FontSize) / 40);
+
+        setFont(_font);
+        update();
+}
+
+void
+Avatar::setLetter(const QString &letter)
+{
+        letter_ = letter;
+        type_   = ui::AvatarType::Letter;
+        update();
+}
+
+void
+Avatar::setImage(const QImage &image)
+{
+        image_  = image;
+        type_   = ui::AvatarType::Image;
+        pixmap_ = utils::scaleImageToPixmap(image_, size_);
+        update();
+}
+
+void
+Avatar::setIcon(const QIcon &icon)
+{
+        icon_ = icon;
+        type_ = ui::AvatarType::Icon;
+        update();
+}
+
+void
+Avatar::paintEvent(QPaintEvent *)
+{
+        QPainter painter(this);
+        painter.setRenderHint(QPainter::Antialiasing);
+
+        QRect r      = rect();
+        const int hs = size_ / 2;
+
+        if (type_ != ui::AvatarType::Image) {
+                QBrush brush;
+                brush.setStyle(Qt::SolidPattern);
+                brush.setColor(backgroundColor());
+
+                painter.setPen(Qt::NoPen);
+                painter.setBrush(brush);
+                painter.drawEllipse(r.center(), hs, hs);
+        }
+
+        switch (type_) {
+        case ui::AvatarType::Icon: {
+                icon_.paint(&painter,
+                            QRect((width() - hs) / 2, (height() - hs) / 2, hs, hs),
+                            Qt::AlignCenter,
+                            QIcon::Normal);
+                break;
+        }
+        case ui::AvatarType::Image: {
+                QPainterPath ppath;
+                ppath.addEllipse(width() / 2 - hs, height() / 2 - hs, size_, size_);
+                painter.setClipPath(ppath);
+                painter.drawPixmap(QRect(width() / 2 - hs, height() / 2 - hs, size_, size_),
+                                   pixmap_);
+                break;
+        }
+        case ui::AvatarType::Letter: {
+                painter.setPen(textColor());
+                painter.setBrush(Qt::NoBrush);
+                painter.drawText(r.translated(0, -1), Qt::AlignCenter, letter_);
+                break;
+        }
+        default:
+                break;
+        }
+}