qgsbrowsermodel.h

Go to the documentation of this file.

/***************************************************************************
    qgsbrowsermodel.h
    ---------------------
    begin                : July 2011
    copyright            : (C) 2011 by Martin Dobias
    email                : wonder dot sk at gmail dot com
 ***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
#ifndef QGSBROWSERMODEL_H
#define QGSBROWSERMODEL_H

#include <QAbstractItemModel>
#include <QIcon>
#include <QMimeData>
#include <QMovie>
#include <QFuture>
#include <QFutureWatcher>

#include "qgsdataitem.h"

class CORE_EXPORT QgsBrowserWatcher : public QFutureWatcher<QVector <QgsDataItem*> >
{
    Q_OBJECT

  public:
    QgsBrowserWatcher( QgsDataItem * item );
    ~QgsBrowserWatcher();

    QgsDataItem* item() const { return mItem; }

  signals:
    void finished( QgsDataItem* item, const QVector <QgsDataItem*>& items );

  private:
    QgsDataItem *mItem;
};

class CORE_EXPORT QgsBrowserModel : public QAbstractItemModel
{
    Q_OBJECT

  public:

    explicit QgsBrowserModel( QObject *parent = nullptr , bool initialize = true );
    ~QgsBrowserModel();

    enum ItemDataRole
    {
      PathRole = Qt::UserRole, 
      CommentRole = Qt::UserRole + 1, 
    };
    // implemented methods from QAbstractItemModel for read-only access

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

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

    virtual QVariant headerData( int section, Qt::Orientation orientation, int role = Qt::DisplayRole ) const override;

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

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

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

    QModelIndex findItem( QgsDataItem *item, QgsDataItem *parent = nullptr ) const;

    virtual QModelIndex parent( const QModelIndex &index ) const override;

    virtual QStringList mimeTypes() const override;

    virtual QMimeData * mimeData( const QModelIndexList &indexes ) const override;

    virtual bool dropMimeData( const QMimeData * data, Qt::DropAction action, int row, int column, const QModelIndex & parent ) override;

    QgsDataItem *dataItem( const QModelIndex &idx ) const;

    bool hasChildren( const QModelIndex &parent = QModelIndex() ) const override;

    void refresh( const QString& path );

    void refresh( const QModelIndex &index = QModelIndex() );

    QModelIndex findPath( const QString& path, Qt::MatchFlag matchFlag = Qt::MatchExactly );

    static QModelIndex findPath( QAbstractItemModel *model, const QString& path, Qt::MatchFlag matchFlag = Qt::MatchExactly );

    void connectItem( QgsDataItem *item );

    bool canFetchMore( const QModelIndex & parent ) const override;
    void fetchMore( const QModelIndex & parent ) override;

    bool initialized( ) const { return mInitialized; }

    void init();

  signals:
    void stateChanged( const QModelIndex & index, QgsDataItem::State oldState );

  public slots:
    void reload();
    void beginInsertItems( QgsDataItem *parent, int first, int last );
    void endInsertItems();
    void beginRemoveItems( QgsDataItem *parent, int first, int last );
    void endRemoveItems();
    void itemDataChanged( QgsDataItem * item );
    void itemStateChanged( QgsDataItem * item, QgsDataItem::State oldState );

    void addFavouriteDirectory( const QString& favDir );
    void removeFavourite( const QModelIndex &index );
    void updateProjectHome();

    void hidePath( QgsDataItem *item );

  protected:
    void addRootItems();
    void removeRootItems();

    QVector<QgsDataItem*> mRootItems;
    QgsFavouritesItem *mFavourites;
    QgsDirectoryItem *mProjectHome;

  private:
    bool mInitialized;
};

#endif // QGSBROWSERMODEL_H