#include "toonzqt/tabbar.h"

#include "toonzqt/dvdialog.h"

#include <qpainter></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();

}