/* dtkComposerSceneModel.cpp ---
 *
 * Author: Julien Wintz
 * Copyright (C) 2008-2011 - Julien Wintz, Inria.
 * Created: Sun Feb  5 15:30:18 2012 (+0100)
 * Version: $Id: b4403fdfb1a3955311cd3ec4f10b83161e0aa19f $
 * Last-Updated: Thu Jul 12 10:46:56 2012 (+0200)
 *           By: Julien Wintz
 *     Update #: 915
 */

/* Commentary:
 *
 */

/* Change log:
 *
 */

#include "dtkComposerNode.h"
#include "dtkComposerScene.h"
#include "dtkComposerSceneEdge.h"
#include "dtkComposerSceneNote.h"
#include "dtkComposerSceneNode.h"
#include "dtkComposerSceneNodeComposite.h"
#include "dtkComposerSceneNodeControl.h"
#include "dtkComposerSceneModel.h"
#include "dtkComposerScenePort.h"

// /////////////////////////////////////////////////////////////////
// dtkComposerSceneModelPrivate
// /////////////////////////////////////////////////////////////////

class dtkComposerSceneModelPrivate
{
public:
    dtkComposerScene *scene;

public:
    dtkComposerSceneModel *q;
};

// /////////////////////////////////////////////////////////////////
// dtkComposerSceneModel
// /////////////////////////////////////////////////////////////////

dtkComposerSceneModel::dtkComposerSceneModel(QObject *parent) : QAbstractItemModel(parent), d(new dtkComposerSceneModelPrivate)
{
    d->q = this;
    d->scene = NULL;
}

dtkComposerSceneModel::~dtkComposerSceneModel(void)
{
    delete d;

    d = NULL;
}

void dtkComposerSceneModel::setScene(dtkComposerScene *scene)
{
    d->scene = scene;

    connect(d->scene, SIGNAL(modified(bool)), this, SIGNAL(modelReset()));
}

Qt::ItemFlags dtkComposerSceneModel::flags(const QModelIndex& index) const
{
    if (!index.isValid())
        return 0;

    return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
}

QVariant dtkComposerSceneModel::data(const QModelIndex& index, int role) const
{
    if (!index.isValid())
        return QVariant();

    if (role != Qt::DisplayRole)
        return QVariant();

    dtkComposerSceneNodeComposite *composite = NULL;

    if (dtkComposerSceneNode *node = dynamic_cast<dtkComposerSceneNode *>((QGraphicsItem *)(index.internalPointer())))
        composite = dynamic_cast<dtkComposerSceneNodeComposite *>(node->parent());
    else if (dtkComposerSceneNote *note = dynamic_cast<dtkComposerSceneNote *>((QGraphicsItem *)(index.internalPointer())))
        composite = dynamic_cast<dtkComposerSceneNodeComposite *>(note->parent());
    else if (dtkComposerSceneEdge *edge = dynamic_cast<dtkComposerSceneEdge *>((QGraphicsItem *)(index.internalPointer())))
        composite = dynamic_cast<dtkComposerSceneNodeComposite *>(edge->parent());

    if (composite) {

        int c_notes = composite->notes().count();
        int c_nodes = composite->nodes().count();
        int c_edges = composite->edges().count();

        if (c_notes && index.row() < c_notes)
            return QString("Note");
        else if (c_nodes && index.row() < c_notes + c_nodes)
            return QString("Node (%1 | %2)").arg(dynamic_cast<dtkComposerSceneNode *>((QGraphicsItem *)(index.internalPointer()))->title()).arg(dynamic_cast<dtkComposerSceneNode *>((QGraphicsItem *)(index.internalPointer()))->wrapee()->type());
        else if (c_edges && index.row() < c_notes + c_nodes + c_edges)
            return QString("Edge (%1 -> %2)").arg(dynamic_cast<dtkComposerSceneEdge *>((QGraphicsItem *)(index.internalPointer()))->source()->node()->title()).arg(dynamic_cast<dtkComposerSceneEdge *>((QGraphicsItem *)(index.internalPointer()))->destination()->node()->title());
    }

// --

    dtkComposerSceneNodeControl *control = NULL;

    if (dtkComposerSceneNode *node = dynamic_cast<dtkComposerSceneNode *>((QGraphicsItem *)(index.internalPointer())))
        control = dynamic_cast<dtkComposerSceneNodeControl *>(node->parent());
    else if (dtkComposerSceneNote *note = dynamic_cast<dtkComposerSceneNote *>((QGraphicsItem *)(index.internalPointer())))
        control = dynamic_cast<dtkComposerSceneNodeControl *>(note->parent());
    else if (dtkComposerSceneEdge *edge = dynamic_cast<dtkComposerSceneEdge *>((QGraphicsItem *)(index.internalPointer())))
        control = dynamic_cast<dtkComposerSceneNodeControl *>(edge->parent());

    if (control) {

        return QString("Block (%1)").arg(dynamic_cast<dtkComposerSceneNodeComposite *>((QGraphicsItem *)(index.internalPointer()))->title());
    }

// --

    return QVariant();
}

QVariant dtkComposerSceneModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    Q_UNUSED(section);

    if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
        return QString("Composition");

    if (orientation == Qt::Vertical && role == Qt::DisplayRole)
        return QString("Composition");

    return QVariant();
}

QModelIndex dtkComposerSceneModel::index(int row, int column, const QModelIndex& parent) const
{
    if (!hasIndex(row, column, parent))
        return QModelIndex();

    dtkComposerSceneNodeComposite *composite = NULL;

    if (!parent.isValid())
        composite = d->scene->root();
    else
        composite = dynamic_cast<dtkComposerSceneNodeComposite *>((QGraphicsItem *)(parent.internalPointer()));

    if (composite) {

        int c_notes = composite->notes().count();
        int c_nodes = composite->nodes().count();
        int c_edges = composite->edges().count();

        if (c_notes && row < c_notes)
            return this->createIndex(row, column, composite->notes().at(row));
        else if (c_nodes && row < c_notes + c_nodes)
            return this->createIndex(row, column, composite->nodes().at(row - c_notes));
        else if (c_edges && row < c_notes + c_nodes + c_edges)
            return this->createIndex(row, column, composite->edges().at(row - c_notes - c_nodes));
    }

// --
    else if (dtkComposerSceneNodeControl *control = dynamic_cast<dtkComposerSceneNodeControl *>((QGraphicsItem *)(parent.internalPointer()))) {

        return this->createIndex(row, column, control->blocks().at(row));
    }

// --

    return QModelIndex();
}

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

    dtkComposerSceneNodeComposite *composite = NULL;
    dtkComposerSceneNodeControl *control = NULL;

    if (dtkComposerSceneNodeComposite *block = dynamic_cast<dtkComposerSceneNodeComposite *>((QGraphicsItem *)(index.internalPointer())))
        control = dynamic_cast<dtkComposerSceneNodeControl *>(block->parent());

    if (dtkComposerSceneNode *node = dynamic_cast<dtkComposerSceneNode *>((QGraphicsItem *)(index.internalPointer())))
        composite = dynamic_cast<dtkComposerSceneNodeComposite *>(node->parent());
    else if (dtkComposerSceneNote *note = dynamic_cast<dtkComposerSceneNote *>((QGraphicsItem *)(index.internalPointer())))
        composite = dynamic_cast<dtkComposerSceneNodeComposite *>(note->parent());
    else if (dtkComposerSceneEdge *edge = dynamic_cast<dtkComposerSceneEdge *>((QGraphicsItem *)(index.internalPointer())))
        composite = dynamic_cast<dtkComposerSceneNodeComposite *>(edge->parent());

    if (control) {

        dtkComposerSceneNodeComposite *parent = dynamic_cast<dtkComposerSceneNodeComposite *>(control->parent());

        if (parent)
            return this->createIndex(parent->notes().count() + parent->nodes().indexOf(control), 0, control);
    }

    if (composite) {

        if (composite == d->scene->root())
            return QModelIndex();

        dtkComposerSceneNodeComposite *parent = dynamic_cast<dtkComposerSceneNodeComposite *>(composite->parent());

        if (parent)
            return this->createIndex(parent->notes().count() + parent->nodes().indexOf(composite), 0, composite);

        dtkComposerSceneNodeControl *ctrl = dynamic_cast<dtkComposerSceneNodeControl *>(composite->parent());

        if (ctrl)
            return this->createIndex(ctrl->blocks().indexOf(composite), 0, composite);
    }

    return QModelIndex();
}

int dtkComposerSceneModel::rowCount(const QModelIndex& parent) const
{
    if (!d->scene)
        return 0;

    if (!parent.isValid())
        return d->scene->root()->notes().count() + d->scene->root()->nodes().count() + d->scene->root()->edges().count();

    if (dtkComposerSceneNodeComposite *composite = dynamic_cast<dtkComposerSceneNodeComposite *>((QGraphicsItem *)(parent.internalPointer())))
        return composite->notes().count() + composite->nodes().count() + composite->edges().count();

// --

    if (dtkComposerSceneNodeControl *control = dynamic_cast<dtkComposerSceneNodeControl *>((QGraphicsItem *)(parent.internalPointer())))
        return control->blocks().count();

// --

    return 0;
}

int dtkComposerSceneModel::columnCount(const QModelIndex& parent) const
{
    return 1;
}