/****************************************************************************
 *
 *   (c) 2009-2016 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
 *
 * QGroundControl is licensed according to the terms in the file
 * COPYING.md in the root of the source code directory.
 *
 ****************************************************************************/

#ifndef QmlObjectListModel_H
#define QmlObjectListModel_H

#include <QAbstractListModel>

class QmlObjectListModel : public QAbstractListModel {
  Q_OBJECT

public:
  QmlObjectListModel(QObject *parent = NULL);
  ~QmlObjectListModel();

  Q_PROPERTY(int count READ count NOTIFY countChanged)

  /// Returns true if any of the items in the list are dirty. Requires each
  /// object to have a dirty property and dirtyChanged signal.
  Q_PROPERTY(bool dirty READ dirty WRITE setDirty NOTIFY dirtyChanged)

  Q_INVOKABLE QObject *get(int index) { return _objectList[index]; }
  // No Q_INVOKABLE here, as QML seems to have a problem with const QObject*.
  const QObject *get(int index) const { return _objectList[index]; }

  // Property accessors

  int count() const;
  bool dirty() const { return _dirty; }
  void setDirty(bool dirty);
  void append(QObject *object);
  void append(QList<QObject *> objects);
  QObjectList swapObjectList(const QObjectList &newlist);
  void clear();
  QObject *removeAt(int i);
  QObject *removeOne(QObject *object) { return removeAt(indexOf(object)); }
  void insert(int i, QObject *object);
  void insert(int i, QList<QObject *> objects);
  bool contains(QObject *object) { return _objectList.indexOf(object) != -1; }
  int indexOf(QObject *object) { return _objectList.indexOf(object); }

  QObject *operator[](int i);
  const QObject *operator[](int i) const;
  template <class T> T value(int index) const {
    return qobject_cast<T>(_objectList[index]);
  }
  QList<QObject *> *objectList() { return &_objectList; }

  bool operator==(const QmlObjectListModel &other) const;
  bool operator!=(const QmlObjectListModel &other) const;

  /// Calls deleteLater on all items and this itself.
  void deleteListAndContents();

  /// Clears the list and calls deleteLater on each entry
  void clearAndDeleteContents();

  void beginReset() { beginResetModel(); }
  void endReset() { endResetModel(); }

  // Friends
  friend void swap(QmlObjectListModel &list1, QmlObjectListModel &list2);

signals:
  void countChanged(int count);
  void dirtyChanged(bool dirtyChanged);

private slots:
  void _childDirtyChanged(bool dirty);

private:
  // Overrides from QAbstractListModel
  int rowCount(const QModelIndex &parent = QModelIndex()) const override;
  QVariant data(const QModelIndex &index,
                int role = Qt::DisplayRole) const override;
  bool insertRows(int position, int rows,
                  const QModelIndex &index = QModelIndex()) override;
  bool removeRows(int position, int rows,
                  const QModelIndex &index = QModelIndex()) override;
  bool setData(const QModelIndex &index, const QVariant &value,
               int role = Qt::EditRole) override;
  QHash<int, QByteArray> roleNames(void) const override;

private:
  QList<QObject *> _objectList;

  bool _dirty;
  bool _skipDirtyFirstItem;

  static const int ObjectRole;
  static const int TextRole;
};

#endif