#include "markermodel.h"

#include "markergroup.h"
#include "CustomWidgets/siunitedit.h"
#include "unit.h"

#include <QComboBox>
#include <QApplication>
#include <QDebug>


using namespace std;

static constexpr int rowHeight = 21;

MarkerModel::MarkerModel(TraceModel &model, QObject *parent)
    : QAbstractItemModel(parent),
      model(model)
{
    model.setMarkerModel(this);
    markers.clear();
    root = new Marker(this);
}

MarkerModel::~MarkerModel()
{
    delete root;
}

QModelIndex MarkerModel::index(int row, int column, const QModelIndex &parent) const
{
    if (!hasIndex(row, column, parent)) {
        return QModelIndex();
    }
    if(parent.isValid()) {
        auto parentItem = markerFromIndex(parent);
        auto child = parentItem->helperMarker(row);
        if(child) {
            return createIndex(row, column, parentItem);
        }
    }
    return createIndex(row, column, root);
}

QModelIndex MarkerModel::parent(const QModelIndex &index) const
{
    if (!index.isValid()) {
        return QModelIndex();
    }

    auto childItem = markerFromIndex(index);
    auto *parentItem = childItem->getParent();
    if(parentItem) {
        // find out the number of the child
        auto it = find(markers.begin(), markers.end(), parentItem);
        auto row = it - markers.begin();
        return createIndex(row, 0, root);
    } else {
        // no parent
        return QModelIndex();
    }
}

Marker *MarkerModel::createDefaultMarker()
{
    // find lowest free number
    int number = 0;
    bool used;
    do {
        number++;
        used = false;
        for(auto m : markers) {
            if(m->getNumber() == number) {
                used = true;
                break;
            }
        }
    } while (used);
    auto marker = new Marker(this, number);
    marker->setPosition(2150000000);
    marker->assignTrace(model.trace(0));
    return marker;
}

void MarkerModel::addMarker(Marker *t)
{
    beginInsertRows(QModelIndex(), markers.size(), markers.size());
    markers.push_back(t);
    endInsertRows();
    connect(t, &Marker::dataChanged, this, &MarkerModel::markerDataChanged);
    connect(t, &Marker::typeChanged, this, &MarkerModel::markerDataChanged);
    connect(t, &Marker::traceChanged, this, &MarkerModel::markerDataChanged);
    connect(t, &Marker::beginRemoveHelperMarkers, [=](Marker *m) {
         auto row = find(markers.begin(), markers.end(), m) - markers.begin();
         auto modelIndex = createIndex(row, 0, root);
         beginRemoveRows(modelIndex, 0, m->getHelperMarkers().size() - 1);
    });
    connect(t, &Marker::visibilityChanged, [=](Marker *m) {
        auto row = find(markers.begin(), markers.end(), m) - markers.begin();
        auto index = createIndex(row, (int) ColIndexVisible, root);
        emit dataChanged(index, index);
    });
    connect(t, &Marker::endRemoveHelperMarkers, [=](Marker *m) {
        endRemoveRows();
        markerDataChanged(m);
    });
    connect(t, &Marker::deleted, this, qOverload<Marker*>(&MarkerModel::removeMarker));
    emit markerAdded(t);
}

void MarkerModel::removeMarker(unsigned int index)
{
    if (index < markers.size()) {
        beginRemoveRows(QModelIndex(), index, index);
        markers.erase(markers.begin() + index);
        endRemoveRows();
    }
}

void MarkerModel::removeMarker(Marker *m)
{
    auto it = std::find(markers.begin(), markers.end(), m);
    if(it != markers.end()) {
        removeMarker(it - markers.begin());
    }
}

void MarkerModel::markerDataChanged(Marker *m)
{
    auto row = find(markers.begin(), markers.end(), m) - markers.begin();
    if(m->editingFrequency) {
        // only update the other columns, do not override editor data
        emit dataChanged(index(row, ColIndexData), index(row, ColIndexData));
    } else {
        emit dataChanged(index(row, ColIndexNumber), index(row, ColIndexData));
        // also update any potential helper markers
        for(unsigned int i=0;i<m->getHelperMarkers().size();i++) {
            auto modelIndex = createIndex(i, 0, m);
            emit dataChanged(index(i, ColIndexNumber, modelIndex), index(i, ColIndexData, modelIndex));
        }
    }
}

void MarkerModel::groupEmptied(MarkerGroup *g)
{
    groups.erase(g);
    delete g;
    // notify markers of deleted group
    for(auto m : markers) {
        m->updateContextmenu();
    }
}

std::set<MarkerGroup *> MarkerModel::getGroups() const
{
    return groups;
}

MarkerGroup *MarkerModel::createMarkerGroup()
{
    // find lowest available number
    unsigned int number = 0;
    bool used;
    do {
        number++;
        used = false;
        for(auto g : groups) {
            if(g->getNumber() == number) {
                used = true;
                break;
            }
        }
    } while (used);
    auto group = createMarkerGroup(number);
    return group;
}

MarkerGroup *MarkerModel::createMarkerGroup(unsigned int number)
{
    auto group = new MarkerGroup(number);
    groups.insert(group);
    connect(group, &MarkerGroup::emptied, this, &MarkerModel::groupEmptied);

    // notify markers of additional group
    for(auto m : markers) {
        m->updateContextmenu();
    }
    return group;
}

void MarkerModel::addToGroupCreateIfNotExisting(Marker *m, unsigned int number)
{
    MarkerGroup *group = nullptr;
    // find group
    for(auto g : groups) {
        if(g->getNumber() == number) {
            group = g;
            break;
        }
    }
    if(!group) {
        // group does not exist yet
        group = createMarkerGroup(number);
    }
    group->add(m);
}

Marker *MarkerModel::marker(int index)
{
    return markers.at(index);
}

int MarkerModel::rowCount(const QModelIndex &index) const
{
    if(!index.isValid()) {
        return markers.size();
    }
    auto marker = markerFromIndex(index);
    return marker->getHelperMarkers().size();
}

int MarkerModel::columnCount(const QModelIndex &) const
{
    return ColIndexLast;
}

QVariant MarkerModel::data(const QModelIndex &index, int role) const
{
    auto marker = markerFromIndex(index);
    switch(role) {
    case Qt::DisplayRole:
        switch(index.column()) {
        case ColIndexNumber:
            return QString::number(marker->getNumber()) + marker->getSuffix();
        case ColIndexGroup:
            if(marker->getGroup()) {
                return QString::number(marker->getGroup()->getNumber());
            }
            break;
        case ColIndexTrace:
            if(marker->getTrace()) {
                return marker->getTrace()->name();
            }
            break;
        case ColIndexType:
            return marker->readableType();
        case ColIndexSettings:
            return marker->readableSettings();
        case ColIndexData:
            return marker->readableData();
        }
        break;
    case Qt::DecorationRole:
        switch(index.column()) {
        case ColIndexVisible:
            if(marker->isVisible()) {
                return QIcon(":/icons/visible.svg");
            } else {
                return QIcon(":/icons/invisible.svg");
            }
        }
    }
    return QVariant();
}

QVariant MarkerModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if(orientation == Qt::Horizontal) {
        switch(role) {
        case Qt::DecorationRole:
            switch(section) {
            case ColIndexGroup: return QIcon(":/icons/chainlink.png");
            case ColIndexVisible: return QIcon(":/icons/visible.svg");
            }
            break;
        case Qt::DisplayRole:
            switch(section) {
            case ColIndexNumber: return "Marker #"; // How to center-align this column, if only # char is used?
            case ColIndexTrace: return "Trace";
            case ColIndexType: return "Type";
            case ColIndexSettings: return "Settings";
            case ColIndexRestrict: return "Restrict";
            case ColIndexData: return "Data";
            }
            break;
        case Qt::ToolTipRole:
        case Qt::WhatsThisRole:
            switch(section) {
            case ColIndexNumber: return "Marker number, used to identify markers on the graphs";
            case ColIndexVisible: return "Temporarily hide/show marker";
            case ColIndexGroup: return "Ctrl-click to select marker(s), right-click to group. Linked markers move together.";
            case ColIndexTrace: return "The trace from which the marker gets its data";
            case ColIndexType: return "Depending on marker type, it can be positioned by the user or will be set automatically";
            case ColIndexSettings: return "Configurable marker parameter, depends on the marker type";
            case ColIndexRestrict: return "Restrict the range available for this marker";
            case ColIndexData: return "Tracedata at the marker position";
            }
            break;
        }
    }
    return QVariant();
}

bool MarkerModel::setData(const QModelIndex &index, const QVariant &value, int)
{
    if((unsigned int) index.row() >= markers.size()) {
        return false;
    }
    auto m = markerFromIndex(index);
    switch(index.column()) {
    case ColIndexNumber: {
        m->setNumber(value.toInt());
    }
        break;
    case ColIndexTrace: {
        auto info = qvariant_cast<MarkerWidgetTraceInfo>(value);
        m->assignTrace(info.trace);
    }
        break;
    case ColIndexSettings: {
        m->adjustSettings(value.toDouble());
    }
        break;
    }

    return false;
}

Qt::ItemFlags MarkerModel::flags(const QModelIndex &index) const
{
    int flags = Qt::ItemIsSelectable;
    switch(index.column()) {
    case ColIndexNumber: flags |= Qt::ItemIsEnabled | Qt::ItemIsEditable; break;
    case ColIndexVisible: flags |= Qt::ItemIsEnabled; break;
    case ColIndexGroup: flags |= Qt::ItemIsEnabled; break;
    case ColIndexTrace: flags |= Qt::ItemIsEnabled | Qt::ItemIsEditable; break;
    case ColIndexType: flags |= Qt::ItemIsEnabled | Qt::ItemIsEditable; break;
    case ColIndexSettings: flags |= Qt::ItemIsEnabled | Qt::ItemIsEditable; break;
    case ColIndexRestrict: flags |= Qt::ItemIsEnabled | Qt::ItemIsEditable; break;
    case ColIndexData: flags |= Qt::ItemIsEnabled; break;
    }
    auto marker = markerFromIndex(index);
    if(marker->getParent()) {
        // this is a helper marker -> nothing is editable
        flags &= ~Qt::ItemIsEditable;
    }
    return (Qt::ItemFlags) flags;
}

std::vector<Marker *> MarkerModel::getMarkers()
{
    return markers;
}

std::vector<Marker *> MarkerModel::getMarkers(Trace *t)
{
    std::vector<Marker*> attachedMarkers;
    for(auto m : markers) {
        if(m->getTrace() == t) {
            attachedMarkers.push_back(m);
        }
    }
    return attachedMarkers;
}

TraceModel &MarkerModel::getModel()
{
    return model;
}

void MarkerModel::updateMarkers()
{
    for(auto m : markers) {
        m->update();
    }
}

Marker *MarkerModel::markerFromIndex(const QModelIndex &index) const
{
    auto m = static_cast<Marker*>(index.internalPointer());
    if(m == root) {
        return markers[index.row()];
    } else {
        return m->helperMarker(index.row());
    }
}

nlohmann::json MarkerModel::toJSON()
{
    nlohmann::json j;
    for(auto m : markers) {
        j.push_back(m->toJSON());
    }
    return j;
}

void MarkerModel::fromJSON(nlohmann::json j)
{
    // remove old markers
    while(markers.size() > 0) {
        removeMarker((unsigned int) 0);
    }
    for(auto jm : j) {
        auto m = new Marker(this);
        try {
            m->fromJSON(jm);
            addMarker(m);
        } catch (const exception &e) {
            qWarning() << "Failed to creat marker from JSON:" << e.what();
            delete m;
        }
    }
    // second pass to assign delta markers
    for(unsigned int i=0;i<markers.size();i++) {
        if(markers[i]->getType() == Marker::Type::Delta) {
            if(!j[i].contains("delta_marker")) {
                qWarning() << "JSON data does not contain assigned delta marker";
                continue;
            }
            unsigned int hash = j[i]["delta_marker"];
            // attempt to find correct marker
            unsigned int m_delta = 0;
            for(;m_delta < markers.size();m_delta++) {
                auto m = markers[m_delta];
                if(m->toHash() == hash) {
                    markers[i]->assignDeltaMarker(m);
                    break;
                }
            }
            if(m_delta >= markers.size()) {
                qWarning() << "Unable to find assigned delta marker:" << hash;
            }
        }
    }
    // All done loading the markers, trigger update of persistent editors
    emit setupLoadComplete();
}

QSize MarkerTraceDelegate::sizeHint(const QStyleOptionViewItem &, const QModelIndex &) const
{
    return QSize(0, rowHeight);
}

QWidget *MarkerTraceDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem &, const QModelIndex &index) const
{
    auto model = (MarkerModel*) index.model();
    auto c = new QComboBox(parent);
    c->setMaximumHeight(rowHeight);
    connect(c, qOverload<int>(&QComboBox::currentIndexChanged), [c](int) {
        c->clearFocus();
    });
    auto traces = model->getModel().getTraces();
    for(auto t : traces) {
        MarkerWidgetTraceInfo info;
        info.trace = t;
        c->addItem(t->name(), QVariant::fromValue(info));
    }
    return c;
}

void MarkerTraceDelegate::setEditorData(QWidget *editor, const QModelIndex &index) const
{
    auto marker = static_cast<const MarkerModel*>(index.model())->markerFromIndex(index);
    auto c = (QComboBox*) editor;
    MarkerWidgetTraceInfo markerInfo;
    markerInfo.trace = marker->trace();
    for(int i=0;i<c->count();i++) {
        auto info = qvariant_cast<MarkerWidgetTraceInfo>(c->itemData(i));
        if(info == markerInfo) {
            c->setCurrentIndex(i);
            return;
        }
    }
}

void MarkerTraceDelegate::setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const
{
    auto markerModel = (MarkerModel*) model;
    auto c = (QComboBox*) editor;
    markerModel->setData(index, c->itemData(c->currentIndex()));
}

QSize MarkerSettingsDelegate::sizeHint(const QStyleOptionViewItem &, const QModelIndex &) const
{
    return QSize(0, rowHeight);
}

QWidget *MarkerSettingsDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem &, const QModelIndex &index) const
{
    auto marker = static_cast<const MarkerModel*>(index.model())->markerFromIndex(index);
    marker->editingFrequency = true;
    auto e = marker->getSettingsEditor();
    if(e) {
        e->setMaximumHeight(rowHeight);
        e->setParent(parent);
        connect(e, &SIUnitEdit::valueUpdated, this, &MarkerSettingsDelegate::commitData);
        connect(e, &SIUnitEdit::focusLost, [=](){
            marker->editingFrequency = false;
        });
    }
    return e;
}

void MarkerSettingsDelegate::setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const
{
    auto markerModel = (MarkerModel*) model;
    auto si = (SIUnitEdit*) editor;
    markerModel->setData(index, si->value());
}

QSize MarkerTypeDelegate::sizeHint(const QStyleOptionViewItem &, const QModelIndex &) const
{
    return QSize(0, rowHeight);
}

QWidget *MarkerTypeDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem &, const QModelIndex &index) const
{
    auto marker = static_cast<const MarkerModel*>(index.model())->markerFromIndex(index);
    auto editor = marker->getTypeEditor(const_cast<MarkerTypeDelegate*>(this));
    editor->setMaximumHeight(rowHeight);
    editor->setParent(parent);
    return editor;
}

void MarkerTypeDelegate::setModelData(QWidget *editor, QAbstractItemModel *, const QModelIndex &index) const
{
    auto marker = static_cast<const MarkerModel*>(index.model())->markerFromIndex(index);
    marker->updateTypeFromEditor(editor);
}

QSize MarkerRestrictDelegate::sizeHint(const QStyleOptionViewItem &, const QModelIndex &) const
{
    return QSize(0, rowHeight);
}

QWidget *MarkerRestrictDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem &, const QModelIndex &index) const
{
    auto marker = static_cast<const MarkerModel*>(index.model())->markerFromIndex(index);
    auto editor = marker->getRestrictEditor();
    editor->setMaximumHeight(rowHeight);
    editor->setParent(parent);
    return editor;
}