/****************************************************************************
**
** Copyright (C) 2012 Digia Plc
** All rights reserved.
** For any questions to Digia, please use contact form at http://qt.digia.com
**
** This file is part of the Qt Commercial Charts Add-on.
**
** $QT_BEGIN_LICENSE$
** Licensees holding valid Qt Commercial licenses may use this file in
** accordance with the Qt Commercial License Agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and Digia.
**
** If you have questions regarding the use of this file, please use
** contact form at http://qt.digia.com
** $QT_END_LICENSE$
**
****************************************************************************/

#include "chartaxis_p.h"
#include "qabstractaxis.h"
#include "qabstractaxis_p.h"
#include "chartpresenter_p.h"
#include "chartlayout_p.h"
#include "domain_p.h"
#include <qmath.h>
#include <QDateTime>
#include <QValueAxis>
#include <QGraphicsLayout>
#include <QFontMetrics>

QTCOMMERCIALCHART_BEGIN_NAMESPACE

ChartAxis::ChartAxis(QAbstractAxis *axis,ChartPresenter *presenter, bool intervalAxis) : ChartElement(presenter),
    m_chartAxis(axis),
    m_labelsAngle(0),
    m_grid(new QGraphicsItemGroup(presenter->rootItem())),
    m_shades(new QGraphicsItemGroup(presenter->rootItem())),
    m_labels(new QGraphicsItemGroup(presenter->rootItem())),
    m_arrow(new QGraphicsItemGroup(presenter->rootItem())),
    m_title(new QGraphicsSimpleTextItem(presenter->rootItem())),
    m_min(0),
    m_max(0),
    m_animation(0),
    m_labelPadding(5),
    m_intervalAxis(intervalAxis)
{
    //initial initialization
    m_arrow->setZValue(ChartPresenter::AxisZValue);
    m_arrow->setHandlesChildEvents(false);
    m_labels->setZValue(ChartPresenter::AxisZValue);
    m_shades->setZValue(ChartPresenter::ShadesZValue);
    m_grid->setZValue(ChartPresenter::GridZValue);

    QObject::connect(m_chartAxis->d_ptr.data(),SIGNAL(updated()),this,SLOT(handleAxisUpdated()));

    QGraphicsSimpleTextItem item;
    m_font = item.font();

}

ChartAxis::~ChartAxis()
{
}

void ChartAxis::setAnimation(AxisAnimation* animation)
{
    m_animation=animation;
}

void ChartAxis::setLayout(QVector<qreal> &layout)
{
    m_layoutVector=layout;
}

void ChartAxis::createItems(int count)
{
    if (m_arrow->children().size() == 0)
        m_arrow->addToGroup(new AxisItem(this,presenter()->rootItem()));
    if (m_intervalAxis && m_grid->children().size() == 0){
        for(int i = 0 ; i < 2 ;i  ++)
            m_grid->addToGroup(new QGraphicsLineItem(presenter()->rootItem()));
    }
    for (int i = 0; i < count; ++i) {
        m_grid->addToGroup(new QGraphicsLineItem(presenter()->rootItem()));
        m_labels->addToGroup(new QGraphicsSimpleTextItem(presenter()->rootItem()));
        m_arrow->addToGroup(new QGraphicsLineItem(presenter()->rootItem()));
        if ((m_grid->childItems().size())%2 && m_grid->childItems().size()>2)
            m_shades->addToGroup(new QGraphicsRectItem(presenter()->rootItem()));
    }
}

void ChartAxis::deleteItems(int count)
{
    QList<QGraphicsItem *> lines = m_grid->childItems();
    QList<QGraphicsItem *> labels = m_labels->childItems();
    QList<QGraphicsItem *> shades = m_shades->childItems();
    QList<QGraphicsItem *> axis = m_arrow->childItems();

    for (int i = 0; i < count; ++i) {
        if (lines.size()%2 && lines.size() > 1) delete(shades.takeLast());
        delete(lines.takeLast());
        delete(labels.takeLast());
        delete(axis.takeLast());
    }
}

void ChartAxis::updateLayout(QVector<qreal> &layout)
{
    int diff = m_layoutVector.size() - layout.size();

    if (diff>0) {
        deleteItems(diff);
    }
    else if (diff<0) {
        createItems(-diff);
    }

    if(diff<0) handleAxisUpdated();

    if (m_animation) {
          switch(presenter()->state()){
              case ChartPresenter::ZoomInState:
                  m_animation->setAnimationType(AxisAnimation::ZoomInAnimation);
                  m_animation->setAnimationPoint(presenter()->statePoint());
                  break;
              case ChartPresenter::ZoomOutState:
                  m_animation->setAnimationType(AxisAnimation::ZoomOutAnimation);
                  m_animation->setAnimationPoint(presenter()->statePoint());
                  break;
              case ChartPresenter::ScrollUpState:
              case ChartPresenter::ScrollLeftState:
                  m_animation->setAnimationType(AxisAnimation::MoveBackwordAnimation);
                  break;
              case ChartPresenter::ScrollDownState:
              case ChartPresenter::ScrollRightState:
                  m_animation->setAnimationType(AxisAnimation::MoveForwardAnimation);
                  break;
              case ChartPresenter::ShowState:
                  m_animation->setAnimationType(AxisAnimation::DefaultAnimation);
                  break;
          }
          m_animation->setValues(m_layoutVector,layout);
          presenter()->startAnimation(m_animation);
    }
    else {
        setLayout(layout);
        updateGeometry();
    }
}

void ChartAxis::setArrowOpacity(qreal opacity)
{
    m_arrow->setOpacity(opacity);
}

qreal ChartAxis::arrowOpacity() const
{
    return m_arrow->opacity();
}

void ChartAxis::setArrowVisibility(bool visible)
{
    m_arrow->setOpacity(visible);
}

void ChartAxis::setGridOpacity(qreal opacity)
{
    m_grid->setOpacity(opacity);
}

qreal ChartAxis::gridOpacity() const
{
    return m_grid->opacity();
}

void ChartAxis::setGridVisibility(bool visible)
{
    m_grid->setOpacity(visible);
}

void ChartAxis::setLabelsOpacity(qreal opacity)
{
    m_labels->setOpacity(opacity);
}

qreal ChartAxis::labelsOpacity() const
{
    return m_labels->opacity();
}

void ChartAxis::setLabelsVisibility(bool visible)
{
    m_labels->setOpacity(visible);
}

void ChartAxis::setShadesOpacity(qreal opacity)
{
    m_shades->setOpacity(opacity);
}

qreal ChartAxis::shadesOpacity() const
{
    return m_shades->opacity();
}

void ChartAxis::setShadesVisibility(bool visible)
{
    m_shades->setVisible(visible);
}

void ChartAxis::setLabelsAngle(int angle)
{
    foreach(QGraphicsItem* item , m_labels->childItems()) {
          item->setRotation(angle);
    }

    m_labelsAngle=angle;
}

void ChartAxis::setLabelsPen(const QPen &pen)
{
    foreach(QGraphicsItem* item , m_labels->childItems()) {
        static_cast<QGraphicsSimpleTextItem*>(item)->setPen(pen);
    }
}

void ChartAxis::setLabelsBrush(const QBrush &brush)
{
    foreach(QGraphicsItem* item , m_labels->childItems()) {
        static_cast<QGraphicsSimpleTextItem*>(item)->setBrush(brush);
    }
}

void ChartAxis::setLabelsFont(const QFont &font)
{
    foreach(QGraphicsItem* item , m_labels->childItems()) {
        static_cast<QGraphicsSimpleTextItem*>(item)->setFont(font);
    }
    if(m_font!=font) {
        m_font = font;
        foreach(QGraphicsItem* item , m_labels->childItems()) {
            static_cast<QGraphicsSimpleTextItem*>(item)->setFont(font);
        }
        QGraphicsLayoutItem::updateGeometry();
        presenter()->layout()->invalidate();

    }
}

void ChartAxis::setShadesBrush(const QBrush &brush)
{
    foreach(QGraphicsItem* item , m_shades->childItems()) {
        static_cast<QGraphicsRectItem*>(item)->setBrush(brush);
    }
}

void ChartAxis::setShadesPen(const QPen &pen)
{
    foreach(QGraphicsItem* item , m_shades->childItems()) {
        static_cast<QGraphicsRectItem*>(item)->setPen(pen);
    }
}

void ChartAxis::setArrowPen(const QPen &pen)
{
    foreach(QGraphicsItem* item , m_arrow->childItems()) {
           static_cast<QGraphicsLineItem*>(item)->setPen(pen);
    }
}

void ChartAxis::setGridPen(const QPen &pen)
{
    foreach(QGraphicsItem* item , m_grid->childItems()) {
        static_cast<QGraphicsLineItem*>(item)->setPen(pen);
    }
}

void ChartAxis::setLabelPadding(int padding)
{
    m_labelPadding=padding;
}

bool ChartAxis::isEmpty()
{
    return !m_axisRect.isValid() || m_gridRect.isEmpty() || qFuzzyIsNull(m_min - m_max);
}

void ChartAxis::handleDomainUpdated()
{
    Domain* domain = qobject_cast<Domain*>(sender());
    qreal min(0);
    qreal max(0);

    if(m_chartAxis->orientation()==Qt::Horizontal) {
        min = domain->minX();
        max = domain->maxX();
    }
    else if (m_chartAxis->orientation()==Qt::Vertical)
    {
        min = domain->minY();
        max = domain->maxY();
    }

    if (!qFuzzyIsNull(m_min - min) || !qFuzzyIsNull(m_max - max))
    {
        m_min = min;
        m_max = max;

        if (!isEmpty()) {

            QVector<qreal> layout = calculateLayout();
            updateLayout(layout);
            QSizeF before = effectiveSizeHint(Qt::MinimumSize);

            QSizeF after= sizeHint(Qt::MinimumSize);

            if(before!=after) {
                QGraphicsLayoutItem::updateGeometry();
                //we don't want to call invalidate on layout, since it will change minimum size of component,
                //which we would like to avoid since it causes nasty flips when scrolling or zooming,
                //instead recalculate layout and use plotArea for extra space.
                presenter()->layout()->setGeometry(presenter()->layout()->geometry());
            }
        }
    }
}

void ChartAxis::handleAxisUpdated()
{
    if(isEmpty()) return;


    bool visible = m_chartAxis->isVisible();

    setArrowVisibility(visible && m_chartAxis->isLineVisible());
    setGridVisibility(visible && m_chartAxis->isGridLineVisible());
    setLabelsVisibility(visible && m_chartAxis->labelsVisible());
    setShadesVisibility(visible && m_chartAxis->shadesVisible());
    setLabelsAngle(m_chartAxis->labelsAngle());
    setArrowPen(m_chartAxis->linePen());
    setLabelsPen(m_chartAxis->labelsPen());
    setLabelsBrush(m_chartAxis->labelsBrush());
    setLabelsFont(m_chartAxis->labelsFont());
    setGridPen(m_chartAxis->gridLinePen());
    setShadesPen(m_chartAxis->shadesPen());
    setShadesBrush(m_chartAxis->shadesBrush());
    setTitleText(m_chartAxis->title());
}

void ChartAxis::setTitleText(const QString& title)
{
    if(m_titleText!=title) {
        m_titleText = title;
        m_axisRect = QRect();
        QGraphicsLayoutItem::updateGeometry();
        presenter()->layout()->invalidate();
    }
}

void ChartAxis::hide()
{
    setArrowVisibility(false);
    setGridVisibility(false);
    setLabelsVisibility(false);
    setShadesVisibility(false);
}

void ChartAxis::setGeometry(const QRectF &axis, const QRectF &grid)
{
        m_gridRect = grid;
        m_axisRect = axis;

        if (isEmpty()) return;

        if(!m_titleText.isNull()) {
            QFontMetrics fn(m_title->font());

            int size(0);

            if(orientation()==Qt::Horizontal)
            size = grid.width();
            else if(orientation()==Qt::Vertical)
            size = grid.height();

            if (fn.boundingRect(m_titleText).width() > size)
            {
                QString string = m_titleText + "...";
                while (fn.boundingRect(string).width() > size && string.length() > 3)
                string.remove(string.length() - 4, 1);
                m_title->setText(string);
            }
            else
            m_title->setText(m_titleText);

            QPointF center = grid.center() - m_title->boundingRect().center();
            if(orientation()==Qt::Horizontal) {
                m_title->setPos(center.x(),m_axisRect.bottom()-m_title->boundingRect().height());
            }
            else if(orientation()==Qt::Vertical) {
                m_title->setTransformOriginPoint(m_title->boundingRect().center());
                m_title->setRotation(270);
                m_title->setPos(m_axisRect.left()- m_title->boundingRect().width()/2+m_title->boundingRect().height()/2,center.y());
            }
        }

        QVector<qreal> layout = calculateLayout();
        updateLayout(layout);

}

void ChartAxis::axisSelected()
{
    //TODO: axis clicked;
}

Qt::Orientation ChartAxis::orientation() const
{
    return m_chartAxis->orientation();
}

Qt::Alignment ChartAxis::alignment() const
{
    return m_chartAxis->alignment();
}

bool ChartAxis::isVisible()
{
    return m_chartAxis->isVisible();
}

void ChartAxis::setLabels(const QStringList& labels)
{
    m_labelsList=labels;
}

QSizeF ChartAxis::sizeHint(Qt::SizeHint which, const QSizeF& constraint) const
{

    Q_UNUSED(constraint);
    QFontMetrics fn(m_title->font());
    QSizeF sh;

    if(m_titleText.isNull()) return sh;

    switch(which) {
        case Qt::MinimumSize:
        if(orientation()==Qt::Horizontal) {
            sh = QSizeF(fn.boundingRect ("...").width(),fn.height());
        }
        else if(orientation()==Qt::Vertical) {
            sh = QSizeF(fn.height(),fn.boundingRect ("...").width());
        }

        break;
        case Qt::MaximumSize:
        case Qt::PreferredSize:
        if(orientation()==Qt::Horizontal) {
            sh = QSizeF(fn.boundingRect(m_chartAxis->title()).width(),fn.height());
        }
        else if(orientation()==Qt::Vertical) {
            sh = QSizeF(fn.height(),fn.boundingRect(m_chartAxis->title()).width());
        }

        break;
        default:
        break;
    }

    return sh;
}

QStringList ChartAxis::createValueLabels(int ticks) const
{
    Q_ASSERT(m_max>m_min);
    Q_ASSERT(ticks>1);

    QStringList labels;

    int n = qMax(int(-qFloor(log10((m_max-m_min)/(ticks-1)))),0);
    n++;

    QValueAxis *axis = qobject_cast<QValueAxis *>(m_chartAxis);

    QString format = axis->labelFormat();

    if(format.isNull()) {
        for (int i=0; i< ticks; i++) {
            qreal value = m_min + (i * (m_max - m_min)/ (ticks-1));
            labels << QString::number(value,'f',n);
        }
    }
    else {
        QByteArray array = format.toAscii();
        for (int i=0; i< ticks; i++) {
            qreal value = m_min + (i * (m_max - m_min)/ (ticks-1));
            labels << QString().sprintf(array, value);
        }
    }

    return labels;
}

QStringList ChartAxis::createDateTimeLabels(const QString& format,int ticks) const
{
    Q_ASSERT(m_max>m_min);
    Q_ASSERT(ticks>1);
    QStringList labels;
    int n = qMax(int(-floor(log10((m_max-m_min)/(ticks-1)))),0);
    n++;
    for (int i=0; i< ticks; i++) {
        qreal value = m_min + (i * (m_max - m_min)/ (ticks-1));
        labels << QDateTime::fromMSecsSinceEpoch(value).toString(format);
    }

    return labels;
}

#include "moc_chartaxis_p.cpp"

QTCOMMERCIALCHART_END_NAMESPACE