qwt_arrow_button.cpp

/******************************************************************************
 * Qwt Widget Library
 * Copyright (C) 1997   Josef Wilgen
 * Copyright (C) 2002   Uwe Rathmann
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the Qwt License, Version 1.0
 *****************************************************************************/
 
#include "qwt_arrow_button.h"
#include "qwt.h"
 
#include <qpainter.h>
#include <qstyle.h>
#include <qstyleoption.h>
#include <qevent.h>
 
static const int MaxNum = 3;
static const int Margin = 2;
static const int Spacing = 1;
 
class QwtArrowButton::PrivateData
{
  public:
    int num;
    Qt::ArrowType arrowType;
};
 
static QStyleOptionButton styleOpt( const QwtArrowButton* btn )
{
    QStyleOptionButton option;
    option.initFrom( btn );
    option.features = QStyleOptionButton::None;
    if ( btn->isFlat() )
        option.features |= QStyleOptionButton::Flat;
    if ( btn->menu() )
        option.features |= QStyleOptionButton::HasMenu;
    if ( btn->autoDefault() || btn->isDefault() )
        option.features |= QStyleOptionButton::AutoDefaultButton;
    if ( btn->isDefault() )
        option.features |= QStyleOptionButton::DefaultButton;
    if ( btn->isDown() )
        option.state |= QStyle::State_Sunken;
    if ( !btn->isFlat() && !btn->isDown() )
        option.state |= QStyle::State_Raised;
 
    return option;
}
 
QwtArrowButton::QwtArrowButton( int num, Qt::ArrowType arrowType, QWidget* parent )
    : QPushButton( parent )
{
    m_data = new PrivateData;
    m_data->num = qBound( 1, num, MaxNum );
    m_data->arrowType = arrowType;
 
    setAutoRepeat( true );
    setAutoDefault( false );
 
    switch ( m_data->arrowType )
    {
        case Qt::LeftArrow:
        case Qt::RightArrow:
            setSizePolicy( QSizePolicy::Expanding,
                QSizePolicy::Fixed );
            break;
        default:
            setSizePolicy( QSizePolicy::Fixed,
                QSizePolicy::Expanding );
    }
}
 
QwtArrowButton::~QwtArrowButton()
{
    delete m_data;
    m_data = NULL;
}
 
Qt::ArrowType QwtArrowButton::arrowType() const
{
    return m_data->arrowType;
}
 
int QwtArrowButton::num() const
{
    return m_data->num;
}
 
QRect QwtArrowButton::labelRect() const
{
    const int m = Margin;
 
    QRect r = rect();
    r.setRect( r.x() + m, r.y() + m,
        r.width() - 2 * m, r.height() - 2 * m );
 
    if ( isDown() )
    {
        QStyleOptionButton option = styleOpt( this );
        const int ph = style()->pixelMetric(
            QStyle::PM_ButtonShiftHorizontal, &option, this );
        const int pv = style()->pixelMetric(
            QStyle::PM_ButtonShiftVertical, &option, this );
 
        r.translate( ph, pv );
    }
 
    return r;
}
 
void QwtArrowButton::paintEvent( QPaintEvent* event )
{
    QPushButton::paintEvent( event );
    QPainter painter( this );
    drawButtonLabel( &painter );
}
 
void QwtArrowButton::drawButtonLabel( QPainter* painter )
{
    const bool isVertical = m_data->arrowType == Qt::UpArrow ||
        m_data->arrowType == Qt::DownArrow;
 
    const QRect r = labelRect();
    QSize boundingSize = labelRect().size();
    if ( isVertical )
        boundingSize.transpose();
 
    const int w =
        ( boundingSize.width() - ( MaxNum - 1 ) * Spacing ) / MaxNum;
 
    QSize arrow = arrowSize( Qt::RightArrow,
        QSize( w, boundingSize.height() ) );
 
    if ( isVertical )
        arrow.transpose();
 
    QRect contentsSize; // aligned rect where to paint all arrows
    if ( m_data->arrowType == Qt::LeftArrow || m_data->arrowType == Qt::RightArrow )
    {
        contentsSize.setWidth( m_data->num * arrow.width()
            + ( m_data->num - 1 ) * Spacing );
        contentsSize.setHeight( arrow.height() );
    }
    else
    {
        contentsSize.setWidth( arrow.width() );
        contentsSize.setHeight( m_data->num * arrow.height()
            + ( m_data->num - 1 ) * Spacing );
    }
 
    QRect arrowRect( contentsSize );
    arrowRect.moveCenter( r.center() );
    arrowRect.setSize( arrow );
 
    painter->save();
    for ( int i = 0; i < m_data->num; i++ )
    {
        drawArrow( painter, arrowRect, m_data->arrowType );
 
        int dx = 0;
        int dy = 0;
 
        if ( isVertical )
            dy = arrow.height() + Spacing;
        else
            dx = arrow.width() + Spacing;
 
        arrowRect.translate( dx, dy );
    }
    painter->restore();
 
    if ( hasFocus() )
    {
        QStyleOptionFocusRect option;
        option.initFrom( this );
        option.backgroundColor = palette().color( QPalette::Window );
 
        style()->drawPrimitive( QStyle::PE_FrameFocusRect,
            &option, painter, this );
    }
}
 
void QwtArrowButton::drawArrow( QPainter* painter,
    const QRect& r, Qt::ArrowType arrowType ) const
{
    QPolygon pa( 3 );
 
    switch ( arrowType )
    {
        case Qt::UpArrow:
            pa.setPoint( 0, r.bottomLeft() );
            pa.setPoint( 1, r.bottomRight() );
            pa.setPoint( 2, r.center().x(), r.top() );
            break;
        case Qt::DownArrow:
            pa.setPoint( 0, r.topLeft() );
            pa.setPoint( 1, r.topRight() );
            pa.setPoint( 2, r.center().x(), r.bottom() );
            break;
        case Qt::RightArrow:
            pa.setPoint( 0, r.topLeft() );
            pa.setPoint( 1, r.bottomLeft() );
            pa.setPoint( 2, r.right(), r.center().y() );
            break;
        case Qt::LeftArrow:
            pa.setPoint( 0, r.topRight() );
            pa.setPoint( 1, r.bottomRight() );
            pa.setPoint( 2, r.left(), r.center().y() );
            break;
        default:
            break;
    }
 
    painter->save();
 
    painter->setRenderHint( QPainter::Antialiasing, true );
    painter->setPen( Qt::NoPen );
    painter->setBrush( palette().brush( QPalette::ButtonText ) );
    painter->drawPolygon( pa );
 
    painter->restore();
}
 
QSize QwtArrowButton::sizeHint() const
{
    const QSize hint = minimumSizeHint();
    return qwtExpandedToGlobalStrut( hint );
}
 
QSize QwtArrowButton::minimumSizeHint() const
{
    const QSize asz = arrowSize( Qt::RightArrow, QSize() );
 
    QSize sz(
        2 * Margin + ( MaxNum - 1 ) * Spacing + MaxNum * asz.width(),
        2 * Margin + asz.height()
        );
 
    if ( m_data->arrowType == Qt::UpArrow || m_data->arrowType == Qt::DownArrow )
        sz.transpose();
 
    QStyleOption styleOption;
    styleOption.initFrom( this );
 
    sz = style()->sizeFromContents( QStyle::CT_PushButton,
        &styleOption, sz, this );
 
    return sz;
}
 
QSize QwtArrowButton::arrowSize( Qt::ArrowType arrowType,
    const QSize& boundingSize ) const
{
    QSize bs = boundingSize;
    if ( arrowType == Qt::UpArrow || arrowType == Qt::DownArrow )
        bs.transpose();
 
    const int MinLen = 2;
    const QSize sz = bs.expandedTo(
        QSize( MinLen, 2 * MinLen - 1 ) ); // minimum
 
    int w = sz.width();
    int h = 2 * w - 1;
 
    if ( h > sz.height() )
    {
        h = sz.height();
        w = ( h + 1 ) / 2;
    }
 
    QSize arrSize( w, h );
    if ( arrowType == Qt::UpArrow || arrowType == Qt::DownArrow )
        arrSize.transpose();
 
    return arrSize;
}
 
void QwtArrowButton::keyPressEvent( QKeyEvent* event )
{
    if ( event->isAutoRepeat() && event->key() == Qt::Key_Space )
        Q_EMIT clicked();
 
    QPushButton::keyPressEvent( event );
}