StatisticsPlot.cpp

#include "StatisticsPlot.h"

StatisticsPlot::StatisticsPlot( QWidget *parent )
    : QCustomPlot( parent )
{
    init();
    replot();
}

void StatisticsPlot::init()
{
    setBackground( QBrush( QColor( 0, 0, 0, 0 ) ) );
    axisRect(0)->setBackground( QBrush( QColor( 255, 255, 255 ) ) );
    axisRect(0)->setMargins( QMargins( 0, 0, 0, 0 ) );
    xAxis->setPadding( 0 );
    xAxis2->setPadding( 0 );
    yAxis->setPadding( 0 );
    yAxis2->setPadding( 0 );

    //Font
    QFont font = xAxis->labelFont();
    font.setPointSize( 10 );
    //Darkmode font color
    if( palette().background().color().redF() < 0.5 )
    {
        xAxis->setLabelColor( Qt::white );
        yAxis->setLabelColor( Qt::white );
        yAxis2->setLabelColor( Qt::white );
        xAxis->setTickLabelColor( Qt::white );
        yAxis->setTickLabelColor( Qt::white );
        yAxis2->setTickLabelColor( Qt::white );
        xAxis->setTickPen( QPen( Qt::gray ) );
        yAxis->setTickPen( QPen( Qt::gray ) );
        yAxis2->setTickPen( QPen( Qt::gray ) );
        xAxis->setSubTickPen( QPen( Qt::gray ) );
        yAxis->setSubTickPen( QPen( Qt::gray ) );
        yAxis2->setSubTickPen( QPen( Qt::gray ) );
        legend->setTextColor( Qt::white );
        axisRect(0)->setBackground( QColor( 64, 64, 64 ) );
    }

    //X Axis and grid
    xAxis->setLabelFont( font );
    xAxis->setTickLabelFont( font );
    xAxis->grid()->setSubGridVisible( false );
    xAxis->grid()->setPen( QPen( Qt::lightGray, 0, Qt::SolidLine ) );
    xAxis->setTickLength( 0, 8 );
    xAxis->setLabelPadding( 0 );
    xAxis->setTickLabelPadding( 0 );
    xAxis->setOffset( 2 );

    //Y Axis left and grid
    yAxis->setLabelFont( font );
    yAxis->setTickLabelFont( font );
    yAxis->grid()->setSubGridVisible( true );
    yAxis->grid()->setPen( QPen( Qt::lightGray, 0, Qt::SolidLine ) );
    yAxis->grid()->setSubGridPen( QPen( Qt::lightGray, 0, Qt::DotLine ) );
    yAxis->setTickLength( 0, 8 );
    yAxis->setSubTickLength( 0, 4 );
    yAxis->setLabelPadding( 5 );
    yAxis->setTickLabelPadding( 0 );
    yAxis->setOffset( 2 );
    yAxis->setLabel( tr( "Distance [km]" ) );

    QCPLayoutGrid *subLayout = new QCPLayoutGrid;
    plotLayout()->addElement(0, 1, subLayout);
    subLayout->addElement(0, 0, new QCPLayoutElement);
    subLayout->addElement(1, 0, legend);
    subLayout->addElement(2, 0, new QCPLayoutElement);
    plotLayout()->setColumnStretchFactor(1, 0.001);

    legend->setVisible(true);
    legend->setBorderPen( QColor( 0, 0, 0, 0 ) );
    legend->setBrush( QBrush( QColor( 0, 0, 0, 0 ) ) );
    legend->setMargins( QMargins( 0, 0, 0, 0 ) );
    legend->setFont( font );
}

void StatisticsPlot::setDayTicker()
{
    QVector<double> vectorDays = QVector<double>() << 0 << 1 << 2 << 3 << 4 << 5 << 6;
    QVector<QString> labels = QVector<QString>() << tr( "Monday" ) << tr( "Tuesday" ) << tr( "Wednesday" ) << tr( "Thursday" ) << tr( "Friday" ) << tr( "Saturday" ) << tr( "Sunday" );
    QSharedPointer<QCPAxisTickerText> textTicker( new QCPAxisTickerText );
    textTicker->addTicks( vectorDays, labels );
    xAxis->setTicker( textTicker );
    xAxis->setRange( -0.5, 6.5 );
    xAxis->setLabel( tr( "Days of Week" ) );
}

void StatisticsPlot::setMonthTicker()
{
    QVector<double> vectorMonths = QVector<double>() << 0 << 1 << 2 << 3 << 4 << 5 << 6 << 7 << 8 << 9 << 10 << 11;
    QVector<QString> labels = QVector<QString>() << tr( "January" ) << tr( "February" ) << tr( "March" ) << tr( "April" )
                                                 << tr( "May" ) << tr( "June" ) << tr( "Juli" ) << tr( "August" )
                                                 << tr( "September" ) << tr( "October" ) << tr( "November" ) << tr( "December" );
    QSharedPointer<QCPAxisTickerText> textTicker( new QCPAxisTickerText );
    textTicker->addTicks( vectorMonths, labels );
    xAxis->setTicker( textTicker );
    xAxis->setRange( -0.5, 11.5 );
    xAxis->setLabel( tr( "Month" ) );
}

void StatisticsPlot::setYearTicker()
{
    QSharedPointer<QCPAxisTicker> stdTicker( new QCPAxisTicker );
    xAxis->setTicker( stdTicker );
    xAxis->setLabel( tr( "Year" ) );
}

void StatisticsPlot::groupBars(QVector<QCPBars *> bikeBars)
{
    QCPBarsGroup *group = new QCPBarsGroup( this );
    foreach( QCPBars *bikeBar, bikeBars )
    {
        bikeBar->setBarsGroup(group);
        if( bikeBars.size() ) bikeBar->setWidth( 0.6 / bikeBars.size() );
    }
}

void StatisticsPlot::stackBars(QVector<QCPBars *> bikeBars)
{
    for( int i = 1; i < bikeBars.size(); i++ )
    {
        bikeBars.at(i)->setStackingGap(0);
        bikeBars.at(i)->moveAbove( bikeBars.at(i-1) );
        bikeBars.at(i)->setWidth( 0.7 );
    }
    bikeBars.at(0)->setWidth( 0.7 );
}

void StatisticsPlot::setLegendTitles(QVector<QCPBars *> bikeBars, QStringList names)
{
    if( bikeBars.size() != names.size() ) return;

    for( int i = 0; i < bikeBars.size(); i++ )
    {
        bikeBars.at(i)->setName( names.at(i) );
    }
}