Allow for fetching the mimetypes that can be dropped onto an index

2 files changed, 7 insertions, 0 deletions

diff --git a/sigmodr/tree/TreeModel.cpp b/sigmodr/tree/TreeModel.cpp
index 2f23a0c1..c22020d3 100644
--- a/sigmodr/tree/TreeModel.cpp
+++ b/sigmodr/tree/TreeModel.cpp
@@ -70,6 +70,8 @@ QVariant TreeModel::data(const QModelIndex& index, const int role) const
     }
     else if (role == Qt::EditRole)
         return QVariant::fromValue<QWidget*>(item->editWidget());
+    else if (role == AcceptedMimeTypesRole)
+        return item->acceptedMimeTypes();
     return QVariant();
 }
 
diff --git a/sigmodr/tree/TreeModel.h b/sigmodr/tree/TreeModel.h
index 034e9d17..269faaf0 100644
--- a/sigmodr/tree/TreeModel.h
+++ b/sigmodr/tree/TreeModel.h
@@ -41,6 +41,11 @@ class SIGMODRTREE_NO_EXPORT TreeModel : public QAbstractItemModel
     Q_OBJECT
     
     public:
+        enum
+        {
+            AcceptedMimeTypesRole = Qt::UserRole
+        };
+        
         TreeModel(QObject* parent);
         ~TreeModel();