#ifndef CSM_WOLRD_IDTREE_H
#define CSM_WOLRD_IDTREE_H

#include "columns.hpp"
#include "idtable.hpp"

#include <QModelIndex>
#include <QString>
#include <QVariant>

#include <string>
#include <utility>

/*! \brief
 * Class for holding the model. Uses typical qt table abstraction/interface for granting access
 * to the individiual fields of the records, Some records are holding nested data (for instance
 * inventory list of the npc). In cases like this, table model offers interface to access
 * nested data in the qt way - that is specify parent. Since some of those nested data require
 * multiple columns to represent information, single int (default way to index model in the
 * qmodelindex) is not sufficiant. Therefore tablemodelindex class can hold two ints for the
 * sake of indexing two dimensions of the table. This model does not support multiple levels of
 * the nested data. Vast majority of methods makes sense only for the top level data.
 */

namespace CSMWorld
{
    class CollectionBase;
    class NestedCollection;
    struct NestedTableWrapperBase;

    class IdTree : public IdTable
    {
        Q_OBJECT

    private:
        NestedCollection* mNestedCollection;

        unsigned int foldIndexAddress(const QModelIndex& index) const;
        std::pair<int, int> unfoldIndexAddress(unsigned int id) const;

    public:
        IdTree(NestedCollection* nestedCollection, CollectionBase* idCollection, unsigned int features = 0);
        ///< The ownerships of \a nestedCollecton and \a idCollection are not transferred.
        IdTree(const IdTree&) = delete;
        IdTree& operator=(const IdTree&) = delete;
        ~IdTree() override = default;

        int rowCount(const QModelIndex& parent = QModelIndex()) const override;

        int columnCount(const QModelIndex& parent = QModelIndex()) const override;

        QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const override;

        bool setData(const QModelIndex& index, const QVariant& value, int role = Qt::EditRole) override;

        Qt::ItemFlags flags(const QModelIndex& index) const override;

        bool removeRows(int row, int count, const QModelIndex& parent = QModelIndex()) override;

        QModelIndex index(int row, int column, const QModelIndex& parent = QModelIndex()) const override;

        QModelIndex parent(const QModelIndex& index) const override;

        QModelIndex getNestedModelIndex(const std::string& id, int column) const;

        QVariant nestedHeaderData(
            int section, int subSection, Qt::Orientation orientation, int role = Qt::DisplayRole) const;

        NestedTableWrapperBase* nestedTable(const QModelIndex& index) const;

        void setNestedTable(const QModelIndex& index, const NestedTableWrapperBase& nestedTable);

        void addNestedRow(const QModelIndex& parent, int position);

        bool hasChildren(const QModelIndex& index) const override;

        virtual int searchNestedColumnIndex(int parentColumn, Columns::ColumnId id);
        ///< \return the column index or -1 if the requested column wasn't found.

        virtual int findNestedColumnIndex(int parentColumn, Columns::ColumnId id);
        ///< \return the column index or throws an exception if the requested column wasn't found.

    signals:

        void resetStart(const QString& id);

        void resetEnd(const QString& id);
    };
}

#endif
