126 lines
6.1 KiB
C++
126 lines
6.1 KiB
C++
#include "menubtnstyle.h"
|
||
#include <QStyleOptionButton>
|
||
#include <QPainter>
|
||
#include<QDebug>
|
||
|
||
/**
|
||
* 根据Python版改的，还是Python写起来比较快比较方便
|
||
*/
|
||
void MenuBtnStyle::drawControl(QStyle::ControlElement element, const QStyleOption *option,
|
||
QPainter *painter, const QWidget *widget) const
|
||
{
|
||
if (element == QStyle::CE_PushButtonLabel) {
|
||
const QStyleOptionButton *btnOption = qstyleoption_cast<const QStyleOptionButton *>(option);
|
||
QStyleOptionButton new_option = *btnOption;
|
||
QIcon icon = btnOption->icon;
|
||
new_option.icon = QIcon();
|
||
QProxyStyle::drawControl(element, &new_option, painter, widget);
|
||
|
||
if (icon.isNull()) return;
|
||
QIcon::Mode mode = btnOption->state & State_Enabled ? QIcon::Normal : QIcon::Disabled;
|
||
if (mode == QIcon::Normal && btnOption->state & State_HasFocus)
|
||
mode = QIcon::Active;
|
||
QIcon::State state = QIcon::Off;
|
||
if (btnOption->state & State_On)
|
||
state = QIcon::On;
|
||
QWindow *window = widget == nullptr ? nullptr : widget->window()->windowHandle();
|
||
QPixmap pixmap = icon.pixmap(window, btnOption->iconSize, mode, state);
|
||
int pixmapWidth = pixmap.width() / pixmap.devicePixelRatio();
|
||
int pixmapHeight = pixmap.height() / pixmap.devicePixelRatio();
|
||
|
||
QRect iconRect = QRect(QPoint(), QSize(pixmapWidth, pixmapHeight));
|
||
int iconSpacing = 4;
|
||
iconRect.moveCenter(btnOption->rect.center());
|
||
iconRect.moveLeft(btnOption->rect.left() + iconSpacing);
|
||
iconRect = visualRect(btnOption->direction, btnOption->rect, iconRect);
|
||
iconRect.translate(proxy()->pixelMetric(PM_ButtonShiftHorizontal, btnOption, widget),
|
||
proxy()->pixelMetric(PM_ButtonShiftVertical, btnOption, widget));
|
||
painter->drawPixmap(iconRect, pixmap);
|
||
} else {
|
||
// 其他的control，使用默认渲染机制
|
||
QProxyStyle::drawControl(element, option, painter, widget);
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* 这个是直接贴stackoverflow上的回答的，但是好复杂，没太看的懂。
|
||
*/
|
||
void MenuBtnStyle::drawControl2(QStyle::ControlElement element, const QStyleOption *option,
|
||
QPainter *painter, const QWidget *widget) const
|
||
{
|
||
if(element==CE_PushButtonLabel)
|
||
{
|
||
if (const QStyleOptionButton *button = qstyleoption_cast<const QStyleOptionButton *>(option))
|
||
{
|
||
QRect textRect = button->rect;
|
||
uint tf = Qt::AlignVCenter | Qt::TextShowMnemonic;
|
||
if (!proxy()->styleHint(SH_UnderlineShortcut, button, widget))
|
||
tf |= Qt::TextHideMnemonic;
|
||
|
||
if (!button->icon.isNull()) {
|
||
QRect iconRect;
|
||
QIcon::Mode mode = button->state & State_Enabled ? QIcon::Normal : QIcon::Disabled;
|
||
if (mode == QIcon::Normal && button->state & State_HasFocus)
|
||
mode = QIcon::Active;
|
||
QIcon::State state = QIcon::Off;
|
||
if (button->state & State_On)
|
||
state = QIcon::On;
|
||
|
||
QWindow *window = widget == nullptr ? nullptr : widget->window()->windowHandle();
|
||
QPixmap pixmap = button->icon.pixmap(window, button->iconSize, mode, state);
|
||
|
||
int pixmapWidth = pixmap.width() / pixmap.devicePixelRatio();
|
||
int pixmapHeight = pixmap.height() / pixmap.devicePixelRatio();
|
||
int labelWidth = pixmapWidth;
|
||
int labelHeight = pixmapHeight;
|
||
int iconSpacing = 4;//### 4 is currently hardcoded in QPushButton::sizeHint()
|
||
int textWidth = button->fontMetrics.boundingRect(option->rect, tf, button->text).width();
|
||
if (!button->text.isEmpty())
|
||
labelWidth += (textWidth + iconSpacing);
|
||
|
||
/*************************************************************/
|
||
// Make the icon rectangle always be 10px in from the left edge
|
||
/*************************************************************/
|
||
iconRect = QRect(10,
|
||
textRect.y() + (textRect.height() - labelHeight) / 2,
|
||
pixmapWidth, pixmapHeight);
|
||
|
||
iconRect = visualRect(button->direction, textRect, iconRect);
|
||
|
||
/***********************************/
|
||
// Always horizontal align the text
|
||
/***********************************/
|
||
tf |= Qt::AlignHCenter;
|
||
|
||
|
||
if (button->state & (State_On | State_Sunken))
|
||
iconRect.translate(proxy()->pixelMetric(PM_ButtonShiftHorizontal, option, widget),
|
||
proxy()->pixelMetric(PM_ButtonShiftVertical, option, widget));
|
||
painter->drawPixmap(iconRect, pixmap);
|
||
} else {
|
||
tf |= Qt::AlignHCenter;
|
||
}
|
||
if (button->state & (State_On | State_Sunken))
|
||
textRect.translate(proxy()->pixelMetric(PM_ButtonShiftHorizontal, option, widget),
|
||
proxy()->pixelMetric(PM_ButtonShiftVertical, option, widget));
|
||
|
||
if (button->features & QStyleOptionButton::HasMenu) {
|
||
int indicatorSize = proxy()->pixelMetric(PM_MenuButtonIndicator, button, widget);
|
||
if (button->direction == Qt::LeftToRight)
|
||
textRect = textRect.adjusted(0, 0, -indicatorSize, 0);
|
||
else
|
||
textRect = textRect.adjusted(indicatorSize, 0, 0, 0);
|
||
}
|
||
proxy()->drawItemText(painter, textRect, tf, button->palette, (button->state & State_Enabled),
|
||
button->text, QPalette::ButtonText);
|
||
}
|
||
|
||
qDebug() << "调用了吗？？？";
|
||
return;
|
||
}
|
||
|
||
// For all other controls, draw the default
|
||
QProxyStyle::drawControl(element, option, painter, widget);
|
||
}
|