ResInsight/ApplicationLibCode/UserInterface/RiuNightchartsWidget.cpp

/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2017     Statoil ASA
//
//  ResInsight is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  ResInsight is distributed in the hope that it will be useful, but WITHOUT ANY
//  WARRANTY; without even the implied warranty of MERCHANTABILITY or
//  FITNESS FOR A PARTICULAR PURPOSE.
//
//  See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html>
//  for more details.
//
/////////////////////////////////////////////////////////////////////////////////

#include "RiuNightchartsWidget.h"
#include "RiuGuiTheme.h"

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RiuNightchartsWidget::RiuNightchartsWidget( QWidget* parent )
    : QWidget( parent )
{
    m_showLegend = true;
    m_showPie    = true;
    updateSizePolicy();
    clear();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiuNightchartsWidget::setType( Nightcharts::type t )
{
    m_chart.setType( t );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiuNightchartsWidget::showLegend( bool doShow )
{
    m_showLegend = doShow;
    updateSizePolicy();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiuNightchartsWidget::showPie( bool doShow )
{
    m_showPie = doShow;
    updateSizePolicy();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiuNightchartsWidget::updateSizePolicy()
{
    if ( m_showPie && m_showLegend )
    {
        setSizePolicy( QSizePolicy::Preferred, QSizePolicy::Preferred );
    }
    else if ( m_showPie && !m_showLegend )
    {
        setSizePolicy( QSizePolicy::Maximum, QSizePolicy::Maximum );
    }
    else if ( !m_showPie && m_showLegend )
    {
        setSizePolicy( QSizePolicy::Minimum, QSizePolicy::Minimum );
    }
    else
    {
        setSizePolicy( QSizePolicy::Minimum, QSizePolicy::Minimum );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiuNightchartsWidget::clear()
{
    m_chart = Nightcharts();
    m_chart.setType( Nightcharts::Pie );
    m_chart.setLegendType( Nightcharts::Vertical );
    m_chart.setShadows( false );

    auto col = RiuGuiTheme::getColorByVariableName( "textColor" );
    m_chart.setTextColor( col );

    m_marginLeft   = 10;
    m_marginTop    = 10;
    m_maxNameWidth = 0;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QSize RiuNightchartsWidget::sizeHint() const
{
    int widthHint  = 0;
    int heightHint = 0;

    if ( m_showLegend )
    {
        int lineHeight = fontMetrics().height();
        int lineCount  = m_chart.pieceCount();

        int exactLegendHeight = ( lineCount + lineCount - 1 ) * lineHeight;

        widthHint  = m_maxNameWidth + 5 + lineHeight;
        heightHint = exactLegendHeight;
    }

    if ( m_showPie )
    {
        int maxPieSize = 300;

        widthHint  = widthHint + maxPieSize;
        heightHint = heightHint > maxPieSize ? heightHint : maxPieSize;
    }

    if ( m_showPie || m_showLegend )
    {
        widthHint += 2 * m_marginLeft;
        heightHint += 2 * m_marginTop;
        return QSize( widthHint, heightHint );
    }
    else
    {
        return QSize();
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiuNightchartsWidget::paintEvent( QPaintEvent* e )
{
    QWidget::paintEvent( e );

    if ( !m_chart.pieceCount() ) return;

    QPainter painter;
    painter.begin( this );

    int legendWidth  = 170;
    int legendMargin = 20;

    if ( !m_showLegend )
    {
        legendWidth  = 0;
        legendMargin = 0;
    }

    int w = ( width() - 2 * m_marginLeft - legendWidth - legendMargin );
    int h = ( height() - 2 * m_marginTop );

    int size = ( w < h ) ? w : h;

    if ( m_showPie )
    {
        m_chart.setCords( m_marginLeft, m_marginTop, size, size );
        m_chart.setLegendCords( m_marginLeft + size + legendMargin, m_marginTop );
    }
    else
    {
        m_chart.setLegendCords( m_marginLeft, m_marginTop );
    }

    if ( m_showPie ) m_chart.draw( &painter );

    if ( m_showLegend ) m_chart.drawLegend( &painter );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiuNightchartsWidget::addItem( const QString& name, const QColor& color, float value )
{
    m_chart.addPiece( name, color, value );
    int textWidth = fontMetrics().boundingRect( name + " (00 %)" ).width();

    m_maxNameWidth = textWidth > m_maxNameWidth ? textWidth : m_maxNameWidth;
}