#include "toonzqt/tabbar.h"
#include "toonzqt/dvdialog.h"

#include <QPainter>

using namespace DVGui;

//=============================================================================
// TabBar
//-----------------------------------------------------------------------------

TabBar::TabBar(QWidget *parent) : QTabBar(parent) { setMouseTracking(true); }

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

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