#include "IconLabel.h"

#include <QStyle>
#include <QStyleOption>
#include <QLayout>
#include <QPainter>
#include <QRect>

IconLabel::IconLabel(QWidget *parent, QIcon icon, QSize size)
	: QWidget(parent), m_size(size), m_icon(icon)
{
	setSizePolicy(QSizePolicy::Minimum, QSizePolicy::Minimum);
}

QSize IconLabel::sizeHint() const
{
	return m_size;
}

void IconLabel::setIcon(QIcon icon)
{
	m_icon = icon;
	update();
}

void IconLabel::paintEvent(QPaintEvent *)
{
	QPainter p(this);
	QRect rect = contentsRect();
	int width = rect.width();
	int height = rect.height();
	if(width < height)
	{
		rect.setHeight(width);
		rect.translate(0, (height - width) / 2);
	}
	else if (width > height)
	{
		rect.setWidth(height);
		rect.translate((width - height) / 2, 0);
	}
	m_icon.paint(&p, rect);
}