Set widget context for QgsProcessingMapLayerComboBox

qgis · Mar 26, 2020 · 9d0efd3 · 9d0efd3
1 parent 6cf9bee
commit 9d0efd3
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 0 deletions.
diff --git a/python/gui/auto_generated/processing/qgsprocessingmaplayercombobox.sip.in b/python/gui/auto_generated/processing/qgsprocessingmaplayercombobox.sip.in
@@ -78,6 +78,13 @@ Sets the ``value`` shown in the widget.
 Returns the current value of the widget.
 
 .. seealso:: :py:func:`setValue`
+%End
+
+    void setWidgetContext( QgsProcessingParameterWidgetContext *context );
+%Docstring
+Sets the ``context`` in which the widget is shown.
+
+.. versionadded:: 3.14
 %End
 
   signals:

diff --git a/python/plugins/processing/gui/wrappers.py b/python/plugins/processing/gui/wrappers.py
@@ -1186,6 +1186,11 @@ def createWidget(self):
             widget.setLayout(layout)
             return widget
 
+    def setWidgetContext(self, context):
+        if isinstance(self.combo, QgsProcessingMapLayerComboBox):
+            self.combo.setWidgetContext(context)
+        super().setWidgetContext(context)
+
     def selectFile(self):
         filename, selected_filter = self.getFileName(self.combo.currentText())
         if filename:

diff --git a/src/gui/processing/qgsprocessingmaplayercombobox.cpp b/src/gui/processing/qgsprocessingmaplayercombobox.cpp
@@ -329,6 +329,10 @@ QVariant QgsProcessingMapLayerComboBox::value() const
   return QVariant();
 }
 
+void QgsProcessingMapLayerComboBox::setWidgetContext( QgsProcessingParameterWidgetContext *context )
+{
+  mBrowserModel = context->browserModel();
+}
 
 QgsMapLayer *QgsProcessingMapLayerComboBox::compatibleMapLayerFromMimeData( const QMimeData *data, bool &incompatibleLayerSelected ) const
 {

diff --git a/src/gui/processing/qgsprocessingmaplayercombobox.h b/src/gui/processing/qgsprocessingmaplayercombobox.h
@@ -29,6 +29,8 @@ class QgsMapLayerComboBox;
 class QToolButton;
 class QCheckBox;
 class QgsProcessingParameterDefinition;
+class QgsBrowserGuiModel;
+class QgsProcessingParameterWidgetContext;
 
 ///@cond PRIVATE
 
@@ -92,6 +94,12 @@ class GUI_EXPORT QgsProcessingMapLayerComboBox : public QWidget
      */
     QVariant value() const;
 
+    /**
+     * Sets the \a context in which the widget is shown.
+     * \since QGIS 3.14
+     */
+    void setWidgetContext( QgsProcessingParameterWidgetContext *context );
+
   signals:
 
     /**
@@ -130,6 +138,9 @@ class GUI_EXPORT QgsProcessingMapLayerComboBox : public QWidget
     QgsFeatureRequest::InvalidGeometryCheck mGeometryCheck = QgsFeatureRequest::GeometryAbortOnInvalid;
     QPointer< QgsMapLayer> mPrevLayer;
     int mBlockChangedSignal = 0;
+
+    QgsBrowserGuiModel *mBrowserModel = nullptr;
+
     QgsMapLayer *compatibleMapLayerFromMimeData( const QMimeData *data, bool &incompatibleLayerSelected ) const;
     QString compatibleUriFromMimeData( const QMimeData *data ) const;
 };