#include "ConnectionStyle.hpp"

#include <iostream>

#include <QtCore/QFile>
#include <QtCore/QJsonDocument>
#include <QtCore/QJsonObject>
#include <QtCore/QJsonValueRef>
#include <QtCore/QJsonArray>

#include <QDebug>

#include "StyleCollection.hpp"

using QtNodes::ConnectionStyle;

inline void initResources()
{
    Q_INIT_RESOURCE(QNodeEditor_resources);
}

ConnectionStyle::
ConnectionStyle()
{
    // Explicit resources inialization for preventing the static initialization
    // order fiasco: https://isocpp.org/wiki/faq/ctors#static-init-order
    initResources();
    // This configuration is stored inside the compiled unit and is loaded statically
    loadJsonFile(":DefaultStyle.json");
}


ConnectionStyle::
ConnectionStyle(QString jsonText)
{
    loadJsonFile(":DefaultStyle.json");
    loadJsonText(jsonText);
}


void
ConnectionStyle::
setConnectionStyle(QString jsonText)
{
    ConnectionStyle style(jsonText);
    StyleCollection::setConnectionStyle(style);
}

#ifdef STYLE_DEBUG
#define CONNECTION_STYLE_CHECK_UNDEFINED_VALUE(v, variable) { \
        if (v.type() == QJsonValue::Undefined || \
            v.type() == QJsonValue::Null) \
            qWarning() << "Undefined value for parameter:" << #variable; \
    }
#else
#define CONNECTION_STYLE_CHECK_UNDEFINED_VALUE(v, variable)
#endif


#define CONNECTION_VALUE_EXISTS(v) \
    (v.type() != QJsonValue::Undefined && \
     v.type() != QJsonValue::Null)

#define CONNECTION_STYLE_READ_COLOR(values, variable)  { \
        auto valueRef = values[#variable]; \
        CONNECTION_STYLE_CHECK_UNDEFINED_VALUE(valueRef, variable) \
        if (CONNECTION_VALUE_EXISTS(valueRef)) {\
            if (valueRef.isArray()) { \
                auto colorArray = valueRef.toArray(); \
                std::vector<int> rgb; rgb.reserve(3); \
                for (auto it = colorArray.begin(); it != colorArray.end(); ++it) { \
                    rgb.push_back((*it).toInt()); \
                } \
                variable = QColor(rgb[0], rgb[1], rgb[2]); \
            } else { \
                variable = QColor(valueRef.toString()); \
            } \
        } \
    }

#define CONNECTION_STYLE_READ_FLOAT(values, variable)  { \
        auto valueRef = values[#variable]; \
        CONNECTION_STYLE_CHECK_UNDEFINED_VALUE(valueRef, variable) \
        if (CONNECTION_VALUE_EXISTS(valueRef)) \
            variable = valueRef.toDouble(); \
    }

#define CONNECTION_STYLE_READ_BOOL(values, variable)  { \
        auto valueRef = values[#variable]; \
        CONNECTION_STYLE_CHECK_UNDEFINED_VALUE(valueRef, variable) \
        if (CONNECTION_VALUE_EXISTS(valueRef)) \
            variable = valueRef.toBool(); \
    }

void
ConnectionStyle::
loadJsonFile(QString styleFile)
{
    QFile file(styleFile);

    if (!file.open(QIODevice::ReadOnly)) {
        qWarning() << "Couldn't open file " << styleFile;
        return;
    }

    loadJsonFromByteArray(file.readAll());
}


void
ConnectionStyle::
loadJsonText(QString jsonText)
{
    loadJsonFromByteArray(jsonText.toUtf8());
}


void
ConnectionStyle::
loadJsonFromByteArray(QByteArray const &byteArray)
{
    QJsonDocument json(QJsonDocument::fromJson(byteArray));
    QJsonObject topLevelObject = json.object();
    QJsonValueRef nodeStyleValues = topLevelObject["ConnectionStyle"];
    QJsonObject obj = nodeStyleValues.toObject();
    CONNECTION_STYLE_READ_COLOR(obj, ConstructionColor);
    CONNECTION_STYLE_READ_COLOR(obj, NormalColor);
    CONNECTION_STYLE_READ_COLOR(obj, SelectedColor);
    CONNECTION_STYLE_READ_COLOR(obj, SelectedHaloColor);
    CONNECTION_STYLE_READ_COLOR(obj, HoveredColor);
    CONNECTION_STYLE_READ_FLOAT(obj, LineWidth);
    CONNECTION_STYLE_READ_FLOAT(obj, ConstructionLineWidth);
    CONNECTION_STYLE_READ_FLOAT(obj, PointDiameter);
    CONNECTION_STYLE_READ_BOOL(obj, UseDataDefinedColors);
}


QColor
ConnectionStyle::
constructionColor() const
{
    return ConstructionColor;
}


QColor
ConnectionStyle::
normalColor() const
{
    return NormalColor;
}


QColor
ConnectionStyle::
normalColor(QString typeId) const
{
    std::size_t hash = qHash(typeId);
    std::size_t const hue_range = 0xFF;
    qsrand(hash);
    std::size_t hue = qrand() % hue_range;
    std::size_t sat = 120 + hash % 129;
    return QColor::fromHsl(hue,
                           sat,
                           160);
}


QColor
ConnectionStyle::
selectedColor() const
{
    return SelectedColor;
}


QColor
ConnectionStyle::
selectedHaloColor() const
{
    return SelectedHaloColor;
}


QColor
ConnectionStyle::
hoveredColor() const
{
    return HoveredColor;
}


float
ConnectionStyle::
lineWidth() const
{
    return LineWidth;
}


float
ConnectionStyle::
constructionLineWidth() const
{
    return ConstructionLineWidth;
}


float
ConnectionStyle::
pointDiameter() const
{
    return PointDiameter;
}


bool
ConnectionStyle::
useDataDefinedColors() const
{
    return UseDataDefinedColors;
}
