#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();
}