/****************************************************************************
**
** 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 "barchartmodel_p.h"
#include "qbarset.h"
#include <limits.h>
#include <QVector>
#include <QDebug>

QTCOMMERCIALCHART_BEGIN_NAMESPACE

BarChartModel::BarChartModel(QStringList categories, QObject *parent) : QObject(parent),
    m_category(categories)
{
}

QStringList BarChartModel::category()
{
    return m_category;
}

void BarChartModel::appendBarSet(QBarSet *set)
{
    m_dataModel.append(set);
}

void BarChartModel::removeBarSet(QBarSet *set)
{
    if (m_dataModel.contains(set)) {
        m_dataModel.removeOne(set);
    }
}

void BarChartModel::insertBarSet(int i, QBarSet *set)
{
    m_dataModel.insert(i, set);
}

void BarChartModel::insertCategory(int i, QString category)
{
    m_category.insert(i, category);
}

void BarChartModel::removeCategory(int i)
{
    m_category.removeAt(i);
}

QBarSet* BarChartModel::barsetAt(int index) const
{
    return m_dataModel.at(index);
}

QList<QBarSet*> BarChartModel::barSets() const
{
    return m_dataModel;
}

int BarChartModel::barsetCount() const
{
    return m_dataModel.count();
}

int BarChartModel::categoryCount() const
{
    return m_category.count();
}

qreal BarChartModel::min() const
{
    Q_ASSERT(m_dataModel.count() > 0);
    // TODO: make min and max members and update them when data changes.
    // This is slower since they are checked every time, even if data is same since previous call.
    qreal min = INT_MAX;

    for (int i = 0; i < m_dataModel.count(); i++) {
        int itemCount = m_dataModel.at(i)->count();
        for (int j = 0; j < itemCount; j++) {
            qreal temp = m_dataModel.at(i)->at(j);
            if (temp < min)
                min = temp;
        }
    }
    return min;
}

qreal BarChartModel::max() const
{
    if (m_dataModel.count() == 0) return 0;
    // TODO: make min and max members and update them when data changes.
    // This is slower since they are checked every time, even if data is same since previous call.
    qreal max = INT_MIN;

    for (int i = 0; i < m_dataModel.count(); i++) {
        int itemCount = m_dataModel.at(i)->count();
        for (int j = 0; j < itemCount; j++) {
            qreal temp = m_dataModel.at(i)->at(j);
            if (temp > max)
                max = temp;
        }
    }

    return max;
}

qreal BarChartModel::valueAt(int set, int category) const
{
    if ((set < 0) || (set >= m_dataModel.count())) {
        // No set, no value.
        return 0;
    } else if ((category < 0) || (category >= m_dataModel.at(set)->count())) {
        // No category, no value.
        return 0;
    }

    return m_dataModel.at(set)->at(category);
}

qreal BarChartModel::percentageAt(int set, int category) const
{
    if ((set < 0) || (set >= m_dataModel.count())) {
        // No set, no value.
        return 0;
    } else if ((category < 0) || (category >= m_dataModel.at(set)->count())) {
        // No category, no value.
        return 0;
    }

    qreal value = m_dataModel.at(set)->at(category);
    qreal total = categorySum(category);
    if ( qFuzzyCompare(total, 0) )
        return 0;

    return value / total;
}

qreal BarChartModel::categorySum(int category) const
{
    qreal sum(0);
    int count = m_dataModel.count(); // Count sets

    for (int set = 0; set < count; set++) {
        if (category < m_dataModel.at(set)->count())
            sum += m_dataModel.at(set)->at(category);
    }
    return sum;
}

qreal BarChartModel::absoluteCategorySum(int category) const
{
    qreal sum(0);
    int count = m_dataModel.count(); // Count sets

    for (int set = 0; set < count; set++) {
        if (category < m_dataModel.at(set)->count())
            sum += qAbs(m_dataModel.at(set)->at(category));
    }
    return sum;
}

qreal BarChartModel::maxCategorySum() const
{
    qreal max = INT_MIN;
    int count = categoryCount();

    for (int col = 0; col < count; col++) {
        qreal sum = categorySum(col);
        if (sum > max)
            max = sum;
    }
    return max;
}

QString BarChartModel::categoryName(int category)
{
    return m_category.at(category);
}

#include "moc_barchartmodel_p.cpp"

QTCOMMERCIALCHART_END_NAMESPACE