30#include <QApplication>
33#include <QFutureWatcher>
34#include <QStorageInfo>
39#include "moc_qgsbrowsermodel.cpp"
41using namespace Qt::StringLiterals;
43#define PROJECT_HOME_PREFIX "project:"
44#define HOME_PREFIX "home:"
47class QgsBrowserWatcher :
public QFutureWatcher<QVector<QgsDataItem *> >
53 : QFutureWatcher(
nullptr )
60 void finished(
QgsDataItem *item,
const QVector<QgsDataItem *> &items );
70 return QString::localeAwareCompare( a->
name(), b->
name() ) < 0;
74 : QAbstractItemModel(
parent )
97 beginRemoveRows( QModelIndex(), idx, idx );
107 beginInsertRows( QModelIndex(), 0, 0 );
120 setupItemConnections( item );
132 const auto drives { QDir::drives() };
133 for (
const QFileInfo &drive : drives )
135 const QString path = drive.absolutePath();
140 const QString driveName = QStorageInfo( path ).displayName();
141 const QString name = driveName.isEmpty() || driveName == path ? path : u
"%1 (%2)"_s.arg( path, driveName );
145 mDriveItems.insert( path, item );
147 setupItemConnections( item );
152 QString path = QString(
"/Volumes" );
154 setupItemConnections( vols );
159 QMultiMap<int, QgsDataItem *> providerMap;
164 if (
QgsDataItem *item = addProviderRootItem( pr ) )
166 providerMap.insert( pr->capabilities(), item );
171 const auto constUniqueKeys = providerMap.uniqueKeys();
172 for (
int key : constUniqueKeys )
174 QList<QgsDataItem *> providerGroup = providerMap.values( key );
175 if ( providerGroup.size() > 1 )
177 std::sort( providerGroup.begin(), providerGroup.end(), cmpByDataItemName_ );
180 const auto constProviderGroup = providerGroup;
205 if (
QgsDataItem *item = addProviderRootItem( provider ) )
216 for ( QgsDataItem *item : constMRootItems )
218 if ( item->providerKey() == provider->
name() )
220 removeRootItem( item );
226void QgsBrowserModel::onConnectionsChanged(
const QString &providerKey )
229 for ( QgsDataItem *item : std::as_const(
mRootItems ) )
231 if ( item->providerKey() == providerKey )
259 if ( !
index.isValid() )
260 return Qt::ItemFlags();
262 Qt::ItemFlags
flags = Qt::ItemIsEnabled | Qt::ItemIsSelectable;
269 return Qt::ItemFlags();
273 flags |= Qt::ItemIsDragEnabled;
277 flags |= Qt::ItemIsDropEnabled;
281 flags |= Qt::ItemIsEditable;
288 if ( !
index.isValid() )
296 else if ( role == Qt::DisplayRole || role == Qt::EditRole )
304 else if ( role == Qt::ToolTipRole )
308 else if ( role == Qt::DecorationRole &&
index.column() == 0 )
320 QgsLayerItem *lyrItem = qobject_cast<QgsLayerItem *>( item );
329 QgsLayerItem *lyrItem = qobject_cast<QgsLayerItem *>( item );
347 if ( !
index.isValid() )
365 return item->
rename( value.toString() );
375 if ( orientation == Qt::Horizontal && role == Qt::DisplayRole )
377 return QVariant(
"header" );
418 return findPath(
this, path, matchFlag );
424 return QModelIndex();
427 bool foundChild =
true;
433 for (
int i = 0; i < model->rowCount(
index ); i++ )
435 QModelIndex idx = model->index( i, 0,
index );
438 if ( itemPath == path )
445 if ( path.startsWith( itemPath +
'/' ) )
454 if ( matchFlag == Qt::MatchStartsWith )
458 return QModelIndex();
463 for (
int i = 0; i < this->
rowCount( index ); i++ )
467 if ( qobject_cast<QgsLayerItem *>(
dataItem( idx ) ) )
469 QString itemUri = qobject_cast<QgsLayerItem *>(
dataItem( idx ) )->uri();
471 if ( itemUri == uri )
478 QModelIndex childIdx =
findUri( uri, idx );
479 if ( childIdx.isValid() )
482 return QModelIndex();
501 const QList< QFileInfo > drives = QDir::drives();
503 const QStringList existingDrives = mDriveItems.keys();
504 for (
const QString &drivePath : existingDrives )
506 bool stillExists =
false;
507 for (
const QFileInfo &drive : drives )
509 if ( drivePath == drive.absolutePath() )
521 removeRootItem( driveItem );
524 for (
const QFileInfo &drive : drives )
526 const QString path = drive.absolutePath();
532 if ( !mDriveItems.contains( path ) )
534 const QString driveName = QStorageInfo( path ).displayName();
535 const QString name = driveName.isEmpty() || driveName == path ? path : u
"%1 (%2)"_s.arg( path, driveName );
540 mDriveItems.insert( path, item );
541 setupItemConnections( item );
552 if ( column < 0 || column >=
columnCount() || row < 0 )
553 return QModelIndex();
558 return item ? createIndex( row, column, item ) : QModelIndex();
565 return QModelIndex();
574 for (
int i = 0; i < items.size(); i++ )
576 if ( items[i] == item )
577 return createIndex( i, 0, item );
579 QModelIndex childIndex =
findItem( item, items[i] );
580 if ( childIndex.isValid() )
583 return QModelIndex();
590 if ( !idx.isValid() )
593 beginInsertRows( idx, first, last );
605 if ( !idx.isValid() )
607 beginRemoveRows( idx, first, last );
618 if ( !idx.isValid() )
620 emit dataChanged( idx, idx );
627 if ( !idx.isValid() )
633void QgsBrowserModel::setupItemConnections(
QgsDataItem *item )
644 if ( collectionItem )
653 types << u
"application/x-vnd.qgis.qgis.uri"_s;
660 const auto constIndexes = indexes;
661 for (
const QModelIndex &
index : constIndexes )
663 if (
index.isValid() )
667 if ( uris.isEmpty() )
681 uri.filePath = ptr->
path();
706 void *v = idx.internalPointer();
777 QStringList hiddenItems = settings.
value( u
"browser/hiddenPaths"_s, QStringList() ).toStringList();
778 int idx = hiddenItems.indexOf( item->
path() );
781 hiddenItems.removeAt( idx );
785 hiddenItems << item->
path();
787 settings.
setValue( u
"browser/hiddenPaths"_s, hiddenItems );
794 removeRootItem( item );
799void QgsBrowserModel::removeRootItem(
QgsDataItem *item )
802 beginRemoveRows( QModelIndex(), i, i );
805 if ( !mDriveItems.key( dirItem ).isEmpty() )
807 mDriveItems.remove( mDriveItems.key( dirItem ) );
830 setupItemConnections( item );
836#include "qgsbrowsermodel.moc"
@ NoCapabilities
No capabilities.
BrowserItemState
Browser item states.
@ NotPopulated
Children not yet created.
@ Populating
Creating children in separate thread (populating or refreshing).
@ Populated
Children created.
@ Rename
Item can be renamed.
@ ItemRepresentsFile
Item's path() directly represents a file on disk.
QFlags< DataItemProviderCapability > DataItemProviderCapabilities
Capabilities for data item providers.
@ Layer
Represents a map layer.
static QgsDataItemProviderRegistry * dataItemProviderRegistry()
Returns the application's data item provider registry, which keeps a list of data item providers that...
Q_DECL_DEPRECATED void connectItem(QgsDataItem *item)
void connectionsChanged(const QString &providerKey)
Emitted when connections for the specified providerKey have changed in the browser.
bool hasChildren(const QModelIndex &parent=QModelIndex()) const override
QgsFavoritesItem * mFavorites
QgsDataItem * dataItem(const QModelIndex &idx) const
Returns the data item at the specified index, or nullptr if no item exists at the index.
bool canFetchMore(const QModelIndex &parent) const override
void hidePath(QgsDataItem *item)
Hide the given path in the browser model.
void refreshDrives()
Refreshes the list of drive items, removing any corresponding to removed drives and adding newly adde...
void reload()
Reload the whole model.
Qt::ItemFlags flags(const QModelIndex &index) const override
void itemStateChanged(QgsDataItem *item, Qgis::BrowserItemState oldState)
Emitted when an item's state is changed.
void beginInsertItems(QgsDataItem *parent, int first, int last)
QModelIndex findPath(const QString &path, Qt::MatchFlag matchFlag=Qt::MatchExactly)
Returns index of item with given path.
QVariant headerData(int section, Qt::Orientation orientation, int role=Qt::DisplayRole) const override
void refresh(const QString &path)
Refresh item specified by path.
bool setData(const QModelIndex &index, const QVariant &value, int role=Qt::EditRole) override
int columnCount(const QModelIndex &parent=QModelIndex()) const override
QVariant data(const QModelIndex &index, int role=Qt::DisplayRole) const override
QVector< QgsDataItem * > mRootItems
QMap< QString, QgsDirectoryItem * > driveItems() const
Returns a map of the root drive items shown in the browser.
~QgsBrowserModel() override
int rowCount(const QModelIndex &parent=QModelIndex()) const override
void beginRemoveItems(QgsDataItem *parent, int first, int last)
QStringList mimeTypes() const override
QModelIndex parent(const QModelIndex &index) const override
QModelIndex findItem(QgsDataItem *item, QgsDataItem *parent=nullptr) const
Returns the model index corresponding to the specified data item.
void initialize()
Delayed initialization, needed because the provider registry must be already populated.
QMimeData * mimeData(const QModelIndexList &indexes) const override
void addFavoriteDirectory(const QString &directory, const QString &name=QString())
Adds a directory to the favorites group.
QgsBrowserModel(QObject *parent=nullptr)
Constructor for QgsBrowserModel, with the specified parent object.
QgsDirectoryItem * mProjectHome
QModelIndex index(int row, int column, const QModelIndex &parent=QModelIndex()) const override
void stateChanged(const QModelIndex &index, Qgis::BrowserItemState oldState)
Emitted when item children fetch was finished.
void itemDataChanged(QgsDataItem *item)
void addRootItems()
Populates the model.
QModelIndex findUri(const QString &uri, QModelIndex index=QModelIndex())
Returns index of layer item with given uri.
@ Sort
Custom sort role, see QgsDataItem::sortKey().
@ Path
Item path used to access path in the tree, see QgsDataItem::mPath.
@ ProviderKey
Data item provider key that created the item, see QgsDataItem::providerKey().
bool dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) override
void removeFavorite(const QModelIndex &index)
Removes a favorite directory from its corresponding model index.
void fetchMore(const QModelIndex &parent) override
A browser item for collections of data.
QList< QgsDataItemProvider * > providers() const
Returns the list of available providers.
void providerWillBeRemoved(QgsDataItemProvider *provider)
Emitted when a data item provider is about to be removed.
void providerAdded(QgsDataItemProvider *provider)
Emitted when a new data item provider has been added.
Interface for providers that add custom data items to the browser tree.
virtual QString name()=0
Human-readable name of the provider name.
virtual QgsDataItem * createDataItem(const QString &path, QgsDataItem *parentItem)=0
Create a new instance of QgsDataItem (or nullptr) for given path and parent item.
virtual Qgis::DataItemProviderCapabilities capabilities() const =0
Returns combination of flags from QgsDataProvider::DataCapabilities.
Base class for all items in the model.
void stateChanged(QgsDataItem *item, Qgis::BrowserItemState oldState)
Emitted when an item's state is changed.
void setSortKey(const QVariant &key)
Sets a custom sorting key for the item.
bool hasChildren() const
Returns whether this item has children.
int rowCount() const
Returns the number of rows of this item.
virtual void deleteChildItem(QgsDataItem *child)
Removes and deletes a child item, emitting relevant signals to the model.
virtual QVariant sortKey() const
Returns the sorting key for the item.
virtual Q_DECL_DEPRECATED bool handleDrop(const QMimeData *, Qt::DropAction)
Attempts to process the mime data dropped on this item.
virtual void refresh(const QVector< QgsDataItem * > &children)
Refresh the items from a specified list of child items.
void dataChanged(QgsDataItem *item)
Emitted when data changes for an item.
static void deleteLater(QVector< QgsDataItem * > &items)
void endRemoveItems()
Emitted after child items have been removed from this data item.
QVector< QgsDataItem * > children() const
void beginRemoveItems(QgsDataItem *parent, int first, int last)
Emitted before child items are removed from this data item.
virtual Q_DECL_DEPRECATED bool acceptDrop()
Returns whether the item accepts drag and dropped layers - e.g.
Qgis::BrowserItemType type() const
Qgis::BrowserItemState state() const
virtual bool hasDragEnabled() const
Returns true if the item may be dragged.
virtual Q_DECL_DEPRECATED bool rename(const QString &name)
Sets a new name for the item, and returns true if the item was successfully renamed.
void connectionsChanged(const QString &providerKey=QString())
Emitted when the connections of the provider with the specified providerKey have changed.
virtual Q_DECL_DEPRECATED QgsMimeDataUtils::Uri mimeUri() const
Returns mime URI for the data item.
QString name() const
Returns the name of the item (the displayed text for the item).
void beginInsertItems(QgsDataItem *parent, int first, int last)
Emitted before child items are added to this data item.
QgsDataItem * parent() const
Gets item parent.
virtual QgsMimeDataUtils::UriList mimeUris() const
Returns mime URIs for the data item, most data providers will only return a single URI but some data ...
QString providerKey() const
Returns the provider key that created this item (e.g.
void endInsertItems()
Emitted after child items have been added to this data item.
virtual Qgis::BrowserItemCapabilities capabilities2() const
Returns the capabilities for the data item.
void setProviderKey(const QString &value)
Sets the provider key that created this item (e.g.
virtual void populate(const QVector< QgsDataItem * > &children)
A browser item for directories: contains subdirectories and layers.
static bool hiddenPath(const QString &path)
Check if the given path is hidden from the browser model.
A directory item showing a single favorite directory.
A browser item which contains various Favorites directories.
A browser item that represents a layer that can be opened with one of the providers.
virtual QString comments() const
Returns comments of the layer.
const QgsLayerMetadata & layerMetadata() const
Returns layer's metadata, it may be a default constructed metadata if metadata is not explicitly set.
QList< QgsMimeDataUtils::Uri > UriList
static QMimeData * encodeUriList(const UriList &layers)
Encodes a URI list to a new QMimeData object.
A directory item showing the current project directory.
static QgsProject * instance()
Returns the QgsProject singleton instance.
void homePathChanged()
Emitted when the home path of the project changes.
Stores settings for use within QGIS.
QVariant value(const QString &key, const QVariant &defaultValue=QVariant(), Section section=NoSection) const
Returns the value for setting key.
void setValue(const QString &key, const QVariant &value, QgsSettings::Section section=QgsSettings::NoSection)
Sets the value of setting key to value.
#define Q_NOWARN_DEPRECATED_POP
QString qgsEnumValueToKey(const T &value, bool *returnOk=nullptr)
Returns the value for the given key of an enum.
#define Q_NOWARN_DEPRECATED_PUSH
#define PROJECT_HOME_PREFIX
#define QgsDebugMsgLevel(str, level)