#include "trianglesunits.h"

#include <QStringList>

TrianglesUnits::TrianglesUnits(QObject *parent):
        QAbstractListModel(parent),
        unitlist(availableUnits())
{
}

QList<TrianglesUnits::Unit> TrianglesUnits::availableUnits()
{
    QList<TrianglesUnits::Unit> unitlist;
    unitlist.append(TRI);
    unitlist.append(mTRI);
    unitlist.append(uTRI);
    return unitlist;
}

bool TrianglesUnits::valid(int unit)
{
    switch(unit)
    {
    case TRI:
    case mTRI:
    case uTRI:
        return true;
    default:
        return false;
    }
}

QString TrianglesUnits::name(int unit)
{
    switch(unit)
    {
    case TRI: return QString("TRI");
    case mTRI: return QString("mTRI");
    case uTRI: return QString::fromUtf8("μTRI");
    default: return QString("???");
    }
}

QString TrianglesUnits::description(int unit)
{
    switch(unit)
    {
    case TRI: return QString("Triangles");
    case mTRI: return QString("Milli-Triangles (1 / 1,000)");
    case uTRI: return QString("Micro-Triangles (1 / 1,000,000)");
    default: return QString("???");
    }
}

qint64 TrianglesUnits::factor(int unit)
{
    switch(unit)
    {
    case TRI:  return 1000000;
    case mTRI: return 1000;
    case uTRI: return 1;
    default:   return 1000000;
    }
}

int TrianglesUnits::amountDigits(int unit)
{
    switch(unit)
    {
    case TRI: return 6; // 120,000 (# digits, without commas)
    case mTRI: return 9; // 120,000,000
    case uTRI: return 12; //  120,000,000,000
    default: return 0;
    }
}

int TrianglesUnits::decimals(int unit)
{
    switch(unit)
    {
    case TRI: return 6;
    case mTRI: return 3;
    case uTRI: return 0;
    default: return 0;
    }
}

QString TrianglesUnits::format(int unit, qint64 n, bool fPlus)
{
    // Note: not using straight sprintf here because we do NOT want
    // localized number formatting.
    if(!valid(unit))
        return QString(); // Refuse to format invalid unit
    qint64 coin = factor(unit);
    int num_decimals = decimals(unit);
    qint64 n_abs = (n > 0 ? n : -n);
    qint64 quotient = n_abs / coin;
    qint64 remainder = n_abs % coin;
    QString quotient_str = QString::number(quotient);
    QString remainder_str = QString::number(remainder).rightJustified(num_decimals, '0');

    // Right-trim excess zeros after the decimal point
    int nTrim = 0;
    for (int i = remainder_str.size()-1; i>=2 && (remainder_str.at(i) == '0'); --i)
        ++nTrim;
    remainder_str.chop(nTrim);

    if (n < 0)
        quotient_str.insert(0, '-');
    else if (fPlus && n > 0)
        quotient_str.insert(0, '+');
    return quotient_str + QString(".") + remainder_str;
}

QString TrianglesUnits::formatWithUnit(int unit, qint64 amount, bool plussign)
{
    return format(unit, amount, plussign) + QString(" ") + name(unit);
}

bool TrianglesUnits::parse(int unit, const QString &value, qint64 *val_out)
{
    if(!valid(unit) || value.isEmpty())
        return false; // Refuse to parse invalid unit or empty string
    int num_decimals = decimals(unit);
    QStringList parts = value.split(".");

    if(parts.size() > 2)
    {
        return false; // More than one dot
    }
    QString whole = parts[0];
    QString decimals;

    if(parts.size() > 1)
    {
        decimals = parts[1];
    }
    if(decimals.size() > num_decimals)
    {
        return false; // Exceeds max precision
    }
    bool ok = false;
    QString str = whole + decimals.leftJustified(num_decimals, '0');

    if(str.size() > 18)
    {
        return false; // Longer numbers will exceed 63 bits
    }
    qint64 retvalue = str.toLongLong(&ok);
    if(val_out)
    {
        *val_out = retvalue;
    }
    return ok;
}

int TrianglesUnits::rowCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent);
    return unitlist.size();
}

QVariant TrianglesUnits::data(const QModelIndex &index, int role) const
{
    int row = index.row();
    if(row >= 0 && row < unitlist.size())
    {
        Unit unit = unitlist.at(row);
        switch(role)
        {
        case Qt::EditRole:
        case Qt::DisplayRole:
            return QVariant(name(unit));
        case Qt::ToolTipRole:
            return QVariant(description(unit));
        case UnitRole:
            return QVariant(static_cast<int>(unit));
        }
    }
    return QVariant();
}