#include "stackedbarpresenter.h"
#include "bar_p.h"
#include "barlabel_p.h"
#include "barvalue_p.h"
#include "separator_p.h"
#include "qbarset.h"
#include <QDebug>

QTCOMMERCIALCHART_BEGIN_NAMESPACE

StackedBarPresenter::StackedBarPresenter(QBarChartSeries *series, QGraphicsItem *parent) :
    BarPresenterBase(series,parent)
{
}

void StackedBarPresenter::layoutChanged()
{
    // Scale bars to new layout
    // Layout for bars:
    if (mSeries->countSets() <= 0) {
        qDebug() << "No sets in model!";
        // Nothing to do.
        return;
    }

    if (mSeries->countCategories() == 0) {
        qDebug() << "No categories in model!";
        // Nothing to do
        return;
    }

    if (childItems().count() == 0) {
        qDebug() << "WARNING: StackedBarPresenter::layoutChanged called before graphics items are created!";
        return;
    }

    // TODO: better way to auto-layout
    // Use reals for accurancy (we might get some compiler warnings... :)
    // TODO: use temp variable for category count...
    qreal maxSum = mSeries->maxCategorySum();
    qreal h = mHeight;
    qreal scale = (h / maxSum);

    int itemIndex(0);
    int labelIndex(0);
    qreal tW = mWidth;
    qreal tC = mSeries->countCategories() + 1;
    qreal xStep = (tW/tC);
    qreal xPos = ((tW/tC) - mBarDefaultWidth / 2);

    for (int category = 0; category < mSeries->countCategories(); category++) {
        qreal yPos = h;
        for (int set=0; set < mSeries->countSets(); set++) {
            qreal barHeight = mSeries->valueAt(set, category) * scale;
            Bar* bar = mBars.at(itemIndex);

            bar->resize(mBarDefaultWidth, barHeight);
            bar->setBrush(mSeries->setAt(set)->brush());
            bar->setPos(xPos, yPos-barHeight);
            itemIndex++;
            yPos -= barHeight;
        }

        // TODO: Layout for labels, remove magic number
        BarLabel* label = mLabels.at(labelIndex);
        label->setPos(xPos, mHeight + 20);
        labelIndex++;
        xPos += xStep;
    }

    // Position separators
    xPos = xStep + xStep/2;
    for (int s=0; s < mSeries->countCategories() - 1; s++) {
        Separator* sep = mSeparators.at(s);
        sep->setPos(xPos,0);
        sep->setSize(QSizeF(1,mHeight));
        xPos += xStep;
    }

    // Position floating values
    itemIndex = 0;
    xPos = ((tW/tC) - mBarDefaultWidth / 2);
    for (int category=0; category < mSeries->countCategories(); category++) {
        qreal yPos = h;
        for (int set=0; set < mSeries->countSets(); set++) {
            qreal barHeight = mSeries->valueAt(set,category) * scale;
            BarValue* value = mFloatingValues.at(itemIndex);

            // TODO: remove hard coding, apply layout
            value->resize(100,50);
            value->setPos(xPos, yPos-barHeight/2);
            value->setPen(QPen(QColor(255,255,255,255)));

            if (mSeries->valueAt(set,category) != 0) {
                value->setValueString(QString::number(mSeries->valueAt(set,category)));
            } else {
                value->setValueString(QString(""));
            }

            itemIndex++;
            yPos -= barHeight;
        }
        xPos += xStep;
    }

    mLayoutDirty = true;
}

#include "moc_stackedbarpresenter.cpp"

QTCOMMERCIALCHART_END_NAMESPACE