def __getStyleOption(self, row, col):
tab = self.__tabAt(row, col)
if not tab: return None
opt = QStyleOptionTab()
opt.initFrom(self)
opt.state &= ~(QStyle.State_HasFocus | QStyle.State_MouseOver)
opt.rect = self.tabRect(row, col)
isCurrent = (row, col) == self.__selectedIndex
opt.row = 0
if (row, col) == self.__pressedIndex:
opt.state |= QStyle.State_Sunken
if isCurrent:
opt.state |= QStyle.State_Selected
if isCurrent and self.hasFocus():
opt.state |= QStyle.State_HasFocus
if not tab.enabled:
opt.state &= ~QStyle.State_Enabled
if self.isActiveWindow():
opt.state |= QStyle.State_Active
if opt.rect == self.__hoverRect:
opt.state |= QStyle.State_MouseOver
opt.shape = QTabBar.RoundedNorth
opt.text = tab.text
if tab.textColor.isValid():
opt.palette.setColor(self.foregroundRole(), tab.textColor)
opt.icon = tab.icon
opt.iconSize = self.iconSize()
if row == self.__selectedIndex[0]:
if col > 0 and col - 1 == self.__selectedIndex[1]:
opt.selectedPosition = QStyleOptionTab.PreviousIsSelected
elif col < self.numTabsInRow(row) - 1 and \
col + 1 == self.__selectedIndex[1]:
opt.selectedPosition = QStyleOptionTab.NextIsSelected
else:
opt.selectedPosition = QStyleOptionTab.NotAdjacent
else:
opt.selectedPosition = QStyleOptionTab.NotAdjacent
if col == 0:
if self.numTabsInRow(row) > 1:
opt.position = QStyleOptionTab.Beginning
else:
opt.position = QStyleOptionTab.OnlyOneTab
elif col == self.numTabsInRow(row) - 1:
opt.position = QStyleOptionTab.End
else:
opt.position = QStyleOptionTab.Middle
return opt
def paintEvent(self, event):
painter = QStylePainter(self)
opt = QStyleOptionTab()
for i in range(self.count()):
self.initStyleOption(opt, i)
key, name = opt.text.split(':')
if key in self.mColors:
opt.text = name
opt.palette.setColor(QPalette.Button, self.mColors[key])
painter.drawControl(QStyle.CE_TabBarTabShape, opt)
painter.drawControl(QStyle.CE_TabBarTabLabel, opt)
