qt-material-widgets/components/iconbutton.cpp

#include <QEvent>
#include <QPainter>
#include <QDebug>
#include "iconbutton.h"
#include "../lib/rippleoverlay.h"
#include "style.h"

IconButton::IconButton(const QIcon &icon, QWidget *parent)
    : QAbstractButton(parent),
      _overlay(new RippleOverlay(parent)),
      _geometryWidget(0)
{
    setIcon(icon);

    QSizePolicy policy;
    policy.setWidthForHeight(true);
    setSizePolicy(policy);

    setGeometryWidget(this);
}

IconButton::~IconButton()
{
}

QSize IconButton::sizeHint() const
{
    return iconSize();
}

void IconButton::setGeometryWidget(QWidget *widget)
{
    if (_geometryWidget) {
        _geometryWidget->removeEventFilter(this);
    }
    _geometryWidget = widget;
    widget->installEventFilter(this);
    updateOverlayGeometry();
}

void IconButton::paintEvent(QPaintEvent *event)
{
    Q_UNUSED(event)

    QPainter painter(this);
    const QSize &size = iconSize();
    QPoint pos(width()/2-size.width()/2, height()/2-size.height()/2);
    icon().paint(&painter, QRect(pos, size));
}

void IconButton::mousePressEvent(QMouseEvent *event)
{
    Q_UNUSED(event)

    if (!_overlay)
        return;

    const QPoint p(_overlay->width(), _overlay->height());
    _overlay->addRipple(p/2, iconSize().width());

    emit clicked();
}

bool IconButton::event(QEvent *event)
{
    if (QEvent::ParentChange == event->type() && parentWidget()) {
        _overlay->setParent(parentWidget());
    }
    return QAbstractButton::event(event);
}

bool IconButton::eventFilter(QObject *obj, QEvent *event)
{
    const QEvent::Type type = event->type();
    if (QEvent::Resize == type || QEvent::Move == type) {
        updateOverlayGeometry();
    }
    return QAbstractButton::eventFilter(obj, event);
}

void IconButton::updateOverlayGeometry()
{
    if (!_overlay || !_geometryWidget)
        return;

    const int s = iconSize().width()/2;
    _overlay->setGeometry(_geometryWidget->geometry().adjusted(-s, -s, s, s));
}