fix: add filter in python file

Author: lachlangrose

loopstructural/gui/interpolation_dialog.py

@@ -14,6 +14,7 @@
     QgsPointXY,
     QgsField,
     QgsFields,
+    QgsMapLayerProxyModel
 )
 from qgis.PyQt.QtCore import QVariant
 
@@ -88,8 +89,14 @@ def _initialize_ui(self):
                 self.grid3DExtentWidget.setMapCanvas(canvas)
         except:
             pass
-        
-        # Set field combos to follow layer combos
+        self.valueLayerCombo.setFilters(QgsMapLayerProxyModel.PointLayer | QgsMapLayerProxyModel.LineLayer)
+        self.gradientLayerCombo.setFilters(QgsMapLayerProxyModel.PointLayer)
+        self.pointLayerCombo.setFilters(QgsMapLayerProxyModel.PointLayer)
+        # Allow empty selection for layer combos
+        self.valueLayerCombo.setAllowEmptyLayer(True)
+        self.gradientLayerCombo.setAllowEmptyLayer(True)
+        self.pointLayerCombo.setAllowEmptyLayer(True)
+        # Allow empty selection for field combos
         self.valueFieldCombo.setLayer(None)
         self.strikeFieldCombo.setLayer(None)
         self.dipFieldCombo.setLayer(None)
@@ -154,13 +161,14 @@ def _on_build_interpolator(self):
                 QMessageBox.warning(self, "Invalid Input", "Please select a value field.")
                 return
 
-            if gradient_layer and (not strike_field or not dip_field):
-                QMessageBox.warning(
-                    self, 
-                    "Invalid Input", 
-                    "Please select both strike and dip fields for gradient data."
-                )
-                return
+            if gradient_layer:
+                if not strike_field or not dip_field:
+                    QMessageBox.warning(
+                        self, 
+                        "Invalid Input", 
+                        "Please select both strike and dip fields for gradient data."
+                    )
+                    return
 
             # Get extent and pixel size
             extent = self.extentWidget.outputExtent()

loopstructural/gui/interpolation_dialog.ui

@@ -54,11 +54,7 @@
            </widget>
           </item>
           <item row="0" column="1">
-           <widget class="QgsMapLayerComboBox" name="valueLayerCombo">
-            <property name="filters">
-             <set>QgsMapLayerProxyModel::PointLayer|QgsMapLayerProxyModel::LineLayer</set>
-            </property>
-           </widget>
+           <widget class="QgsMapLayerComboBox" name="valueLayerCombo"/>
           </item>
           <item row="1" column="0">
            <widget class="QLabel" name="label_2">
@@ -78,11 +74,7 @@
            </widget>
           </item>
           <item row="2" column="1">
-           <widget class="QgsMapLayerComboBox" name="gradientLayerCombo">
-            <property name="filters">
-             <set>QgsMapLayerProxyModel::PointLayer</set>
-            </property>
-           </widget>
+           <widget class="QgsMapLayerComboBox" name="gradientLayerCombo"/>
           </item>
           <item row="3" column="0">
            <widget class="QLabel" name="label_4">
@@ -384,11 +376,7 @@
             </widget>
            </item>
            <item row="0" column="1">
-            <widget class="QgsMapLayerComboBox" name="pointLayerCombo">
-             <property name="filters">
-              <set>QgsMapLayerProxyModel::PointLayer</set>
-             </property>
-            </widget>
+            <widget class="QgsMapLayerComboBox" name="pointLayerCombo"/>
            </item>
           </layout>
          </widget>