#include "lrverticalbarchart.h"

namespace LimeReport {

void VerticalBarChart::paintChart(QPainter* painter, QRectF chartRect)
{
    updateMinAndMaxValues();

    const qreal valuesHMargin = this->valuesHMargin(painter);
    const qreal valuesVMargin = this->valuesVMargin(painter);

    QRectF calcRect = horizontalLabelsRect(
        painter,
        chartRect.adjusted(hPadding(chartRect) * 2 + valuesHMargin,
                           chartRect.height()
                               - (painter->fontMetrics().height() + vPadding(chartRect) * 2),
                           -(hPadding(chartRect) * 2), -vPadding(chartRect)));
    qreal barsShift = calcRect.height();
    paintVerticalGrid(painter,
                      chartRect.adjusted(hPadding(chartRect), vPadding(chartRect) + valuesVMargin,
                                         -hPadding(chartRect), -(vPadding(chartRect) + barsShift)));
    paintVerticalBars(painter,
                      chartRect.adjusted(hPadding(chartRect) * 2 + valuesHMargin,
                                         vPadding(chartRect) + valuesVMargin,
                                         -hPadding(chartRect) * 2,
                                         -(vPadding(chartRect) + barsShift)));
    paintSerialLines(painter,
                     chartRect.adjusted(hPadding(chartRect) * 2 + valuesHMargin,
                                        vPadding(chartRect) + valuesVMargin,
                                        -hPadding(chartRect) * 2,
                                        -(vPadding(chartRect) + barsShift)));
    paintHorizontalLabels(painter, calcRect);
}

void VerticalBarChart::paintVerticalBars(QPainter* painter, QRectF barsRect)
{

    if (valuesCount() == 0)
        return;

    const AxisData& yAxisData = this->yAxisData();
    const qreal delta = yAxisData.delta();

    int barSeriesCount = 0;
    foreach (SeriesItem* series, m_chartItem->series()) {
        if (series->preferredType() == SeriesItem::Bar)
            barSeriesCount++;
    }

    barSeriesCount = (m_chartItem->itemMode() == DesignMode) ? seriesCount() : barSeriesCount;
    if (barSeriesCount < 1)
        return;
    painter->save();
    painter->setRenderHint(QPainter::Antialiasing, false);

    qreal vStep = barsRect.height() / delta;
    qreal hStep = (barsRect.width() / valuesCount()) / (barSeriesCount == 0 ? 1 : barSeriesCount);
    qreal topShift = (delta - (maxValue() - minValue())) * vStep + barsRect.top();

    if (!m_chartItem->series().isEmpty() && (m_chartItem->itemMode() != DesignMode)) {
        int curSeries = 0;
        foreach (SeriesItem* series, m_chartItem->series()) {
            if (series->preferredType() == SeriesItem::Bar) {
                qreal curHOffset = curSeries * hStep + barsRect.left();
                painter->setBrush(series->color());
                foreach (qreal value, series->data()->values()) {
                    painter->drawRect(
                        QRectF(curHOffset, maxValue() * vStep + topShift, hStep, -value * vStep));
                    curHOffset += hStep * barSeriesCount;
                }
                curSeries++;
            }
        }
    } else {
        qreal curHOffset = barsRect.left();
        int curColor = 0;
        for (int i = 0; i < 9; ++i) {
            if (curColor == 3)
                curColor = 0;
            painter->setBrush(color_map[curColor]);
            painter->drawRect(QRectF(curHOffset, maxValue() * vStep + barsRect.top(), hStep,
                                     -designValues()[i] * vStep));
            curHOffset += hStep;
            curColor++;
        }
    }
    painter->restore();
}

void VerticalBarChart::paintSerialLines(QPainter* painter, QRectF barsRect)
{
    if (valuesCount() == 0 || m_chartItem->series().isEmpty())
        return;

    painter->save();
    painter->setRenderHint(QPainter::Antialiasing, true);

    for (SeriesItem* series : m_chartItem->series()) {
        if (series->preferredType() == SeriesItem::Line) {
            paintSeries(painter, series, barsRect);
        }
    }
    painter->restore();
}

} // namespace LimeReport