#include "qchart.h"
#include "qchartseries.h"
#include "qscatterseries.h"
#include "qscatterseries_p.h"
#include "qpieseries.h"
#include "qxychartseries.h"

#include "barchartseries.h"
#include "bargroup.h"

#include "xylinechartitem_p.h"
#include "plotdomain_p.h"
#include "axisitem_p.h"
#include <QGraphicsScene>
#include <QDebug>

QTCOMMERCIALCHART_BEGIN_NAMESPACE

QChart::QChart(QGraphicsObject* parent) : QGraphicsObject(parent),
    m_background(new QGraphicsRectItem(this)),
    m_title(new QGraphicsTextItem(this)),
    m_axisX(new AxisItem(AxisItem::X_AXIS,this)),
    m_axisY(new AxisItem(AxisItem::Y_AXIS,this)),
    m_plotDataIndex(0),
    m_marginSize(0)
{
    // TODO: the default theme?
    setTheme(QChart::ChartThemeVanilla);
    //  setFlags(QGraphicsItem::ItemClipsChildrenToShape);
    PlotDomain domain;
    m_plotDomainList<<domain;

    m_chartItems<<m_axisX;
    m_chartItems<<m_axisY;
}

QChart::~QChart(){}

QRectF QChart::boundingRect() const
{
    return m_rect;
}

void QChart::addSeries(QChartSeries* series)
{
    // TODO: we should check the series not already added

    m_series<<series;

    switch(series->type())
    {
    case QChartSeries::SeriesTypeLine: {

        QXYChartSeries* xyseries = static_cast<QXYChartSeries*>(series);
        // Use color defined by theme in case the series does not define a custom color
        if (!xyseries->color().isValid() && m_themeColors.count())
            xyseries->setColor(m_themeColors.takeFirst());

        m_plotDataIndex = 0 ;
        m_plotDomainList.resize(1);

        PlotDomain& domain = m_plotDomainList[m_plotDataIndex];

        for (int i = 0 ; i < xyseries->count() ; i++)
        {
            qreal x = xyseries->x(i);
            qreal y = xyseries->y(i);
            domain.m_minX = qMin(domain.m_minX,x);
            domain.m_minY = qMin(domain.m_minY,y);
            domain.m_maxX = qMax(domain.m_maxX,x);
            domain.m_maxY = qMax(domain.m_maxY,y);
        }

        XYLineChartItem* item = new XYLineChartItem(xyseries,this);

        m_chartItems<<item;

        foreach(ChartItem* i ,m_chartItems)
            i->setPlotDomain(m_plotDomainList.at(m_plotDataIndex));

        break;
    }
    case QChartSeries::SeriesTypeBar: {

        qDebug() << "barSeries added";
        BarChartSeries* barSeries = static_cast<BarChartSeries*>(series);
        BarGroup* barGroup = new BarGroup(*barSeries,this);
        m_chartItems<<barGroup;
        childItems().append(barGroup);
        break;
        }
    case QChartSeries::SeriesTypeScatter: {
        QScatterSeries *scatterSeries = qobject_cast<QScatterSeries *>(series);
        connect(this, SIGNAL(sizeChanged(QRectF)),
                scatterSeries, SLOT(chartSizeChanged(QRectF)));
        scatterSeries->d->setParentItem(this);
        QColor nextColor = m_themeColors.takeFirst();
        nextColor.setAlpha(150); // TODO: default opacity?
        scatterSeries->setMarkerColor(nextColor);
        }
    case QChartSeries::SeriesTypePie: {
        // TODO: we now have also a list of y values as a parameter, it is ignored
        // we should use a generic data class instead of list of x and y values
        QPieSeries *pieSeries = qobject_cast<QPieSeries *>(series);
        connect(this, SIGNAL(sizeChanged(QRectF)),
                pieSeries, SLOT(chartSizeChanged(QRectF)));
        // TODO: how to define the color for all the slices of a pie?
        }
    }
}

QChartSeries* QChart::createSeries(QChartSeries::QChartSeriesType type)
{
    // TODO: support also other types; not only scatter and pie

    QChartSeries *series(0);

    switch (type) {
    case QChartSeries::SeriesTypeLine: {
        series = QXYChartSeries::create();
        break;
    }
    case QChartSeries::SeriesTypeBar: {
        series = new BarChartSeries(this);
        break;
    }
    case QChartSeries::SeriesTypeScatter: {
        series = new QScatterSeries(this);
        break;
    }
    case QChartSeries::SeriesTypePie: {
        series = new QPieSeries(this);
        break;
    }
    default:
        Q_ASSERT(false);
        break;
    }

    addSeries(series);
    return series;
}

void QChart::setSize(const QSize& size)
{
    m_rect = QRect(QPoint(0,0),size);
    QRect rect = m_rect.adjusted(margin(),margin(), -margin(), -margin());


    //recalculate background gradient
    m_background->setRect(rect);
    m_backgroundGradient.setFinalStop(0,m_background->rect().height());
    m_background->setBrush(m_backgroundGradient);

    //resize elements
    foreach (ChartItem* item ,m_chartItems) {
        item->setPos(rect.topLeft());
        item->setSize(rect.size());

    }
    // TODO: TTD for setting scale
    //emit scaleChanged(100, 100);
    // TODO: calculate the origo
    // TODO: not sure if emitting a signal here is the best from performance point of view
    emit sizeChanged(QRectF(0, 0, size.width(), size.height()));

    update();
}

void QChart::setBackgroundColor(const QColor& color)
{
    m_backgroundGradient.setColorAt( 0.0, Qt::white);
    m_backgroundGradient.setColorAt( 1.0, color);
    m_background->setBrush(m_backgroundGradient);
    m_background->setPen(Qt::NoPen);
    m_background->update();
}

void QChart::setTitle(const QString& title)
{
    m_title->setPlainText(title);
}

int QChart::margin() const
{
    return m_marginSize;
}

void QChart::setMargin(int margin)
{
    m_marginSize = margin;
}

void QChart::setTheme(QChart::ChartTheme theme)
{
    // TODO: define color themes
    switch (theme) {
    case ChartThemeVanilla:
        m_themeColors.append(QColor(255, 238, 174));
        m_themeColors.append(QColor(228, 228, 160));
        m_themeColors.append(QColor(228, 179, 160));
        m_themeColors.append(QColor(180, 151, 18));
        m_themeColors.append(QColor(252, 252, 37));
        break;
    case ChartThemeIcy:
        m_themeColors.append(QColor(255, 238, 174));
        m_themeColors.append(QColor(228, 228, 160));
        m_themeColors.append(QColor(228, 179, 160));
        m_themeColors.append(QColor(180, 151, 18));
        m_themeColors.append(QColor(252, 252, 37));
        break;
    case ChartThemeGrayscale:
        m_themeColors.append(QColor(255, 238, 174));
        m_themeColors.append(QColor(228, 228, 160));
        m_themeColors.append(QColor(228, 179, 160));
        m_themeColors.append(QColor(180, 151, 18));
        m_themeColors.append(QColor(252, 252, 37));
        break;
    default:
        Q_ASSERT(false);
        break;
    }

    // TODO: update coloring of different elements to match the selected theme
}

void QChart::zoomInToRect(const QRect& rectangle)
{

    if(!rectangle.isValid()) return;

    qreal margin = this->margin();

    QRect rect = rectangle.normalized();
    rect.translate(-margin, -margin);

    PlotDomain& oldDomain = m_plotDomainList[m_plotDataIndex];
    PlotDomain newDomain;

    qreal dx = oldDomain.spanX() / (m_rect.width() - 2 * margin);
    qreal dy = oldDomain.spanY() / (m_rect.height() - 2 * margin);

    newDomain.m_minX = oldDomain.m_minX + dx * rect.left();
    newDomain.m_maxX = oldDomain.m_minX + dx * rect.right();
    newDomain.m_minY = oldDomain.m_maxY - dy * rect.bottom();
    newDomain.m_maxY = oldDomain.m_maxY - dy * rect.top();

    m_plotDomainList.resize(m_plotDataIndex + 1);
    m_plotDomainList<<newDomain;
    m_plotDataIndex++;

    foreach (ChartItem* item ,m_chartItems)
        item->setPlotDomain(m_plotDomainList[m_plotDataIndex]);
    update();
}

void QChart::zoomIn()
{
    if (m_plotDataIndex < m_plotDomainList.count() - 1) {
        m_plotDataIndex++;
        foreach (ChartItem* item ,m_chartItems)
            item->setPlotDomain(m_plotDomainList[m_plotDataIndex]);
        update();
    }else{
        QRect rect = m_rect.adjusted(margin(),margin(), -margin(), -margin());
        rect.setWidth(rect.width()/2);
        rect.setHeight(rect.height()/2);
        rect.moveCenter(m_rect.center());
        zoomInToRect(rect);
    }
}

void QChart::zoomOut()
{
    if (m_plotDataIndex > 0) {
        m_plotDataIndex--;
        foreach (ChartItem* item ,m_chartItems)
            item->setPlotDomain(m_plotDomainList[m_plotDataIndex]);
        update();
    }
}

#include "moc_qchart.cpp"

QTCOMMERCIALCHART_END_NAMESPACE