#include "toonzqt/tabbar.h"

#include "toonzqt/dvdialog.h"

#include <QPainter>

using namespace DVGui;

//=============================================================================

// TabBar

//-----------------------------------------------------------------------------

TabBar::TabBar(QWidget *parent)

	: QTabBar(parent)

{

}

//-----------------------------------------------------------------------------

TabBar::~TabBar()

{

	m_pixmaps.clear();

}

//-----------------------------------------------------------------------------

void TabBar::paintEvent(QPaintEvent *event)

{

	QTabBar::paintEvent(event);

	QPainter p(this);

	// Paint tab Icons

	int tabCount = count();

	int selectedIndex = currentIndex();

	for (int i = 0; i < tabCount; i++) {

		QRect tabRectangle = tabRect(i).adjusted(2, 1, -4, 0);

		if (selectedIndex == i) {

			tabRectangle = tabRectangle.adjusted(0, -2, 0, 0);

			if (!m_pixmaps[(i * 2) + 1])

				continue;

			p.drawPixmap(tabRectangle.x(), tabRectangle.y(), m_pixmaps[(i * 2) + 1]);

		} else {

			if (!m_pixmaps[(i * 2)])

				continue;

			p.drawPixmap(tabRectangle.x(), tabRectangle.y(), m_pixmaps[(i * 2)]);

		}

	}

}

//-----------------------------------------------------------------------------

void TabBar::addIconTab(const char *iconPrefixName, const QString &tooltip)

{

	QString normal = QString(":Resources/") + iconPrefixName + "_off.svg";

	QString over = QString(":Resources/") + iconPrefixName + "_on.svg";

	int indexTab = addTab("");

	setTabToolTip(indexTab, tooltip);

	m_pixmaps.push_back(QPixmap(normal));

	m_pixmaps.push_back(QPixmap(over));

}

//-----------------------------------------------------------------------------

void TabBar::addSimpleTab(const QString &text)

{

	addTab(text);

	m_pixmaps.push_back(QPixmap(""));

	m_pixmaps.push_back(QPixmap(""));

}

//-----------------------------------------------------------------------------

void TabBar::clearTabBar()

{

	int i, n = count();

	for (i = 0; i < n; i++)

		removeTab(0);

	if (!m_pixmaps.empty())

		m_pixmaps.clear();

}