Merge pull request #1652 from gioman/processing_count_points

result field in Processing 'count points in polygon' must be integer
qgis · Oct 23, 2014 · 88e5cde · 88e5cde
2 parents aeb9d93 + 6f30ab9
commit 88e5cde
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/python/plugins/processing/algs/qgis/PointsInPolygon.py b/python/plugins/processing/algs/qgis/PointsInPolygon.py
@@ -66,12 +66,14 @@ def processAlgorithm(self, progress):
         fieldName = self.getParameterValue(self.FIELD)
 
         polyProvider = polyLayer.dataProvider()
+        fields = polyProvider.fields()
+        fields.append(QgsField(fieldName, QVariant.Int))        
 
         (idxCount, fieldList) = vector.findOrCreateField(polyLayer,
                 polyLayer.pendingFields(), fieldName)
 
         writer = self.getOutputFromName(
-                self.OUTPUT).getVectorWriter(fieldList.toList(),
+                self.OUTPUT).getVectorWriter(fields.toList(),
                                              polyProvider.geometryType(),
                                              polyProvider.crs())
 

diff --git a/python/plugins/processing/algs/qgis/PointsInPolygonUnique.py b/python/plugins/processing/algs/qgis/PointsInPolygonUnique.py
@@ -70,13 +70,15 @@ def processAlgorithm(self, progress):
         classFieldName = self.getParameterValue(self.CLASSFIELD)
 
         polyProvider = polyLayer.dataProvider()
+        fields = polyProvider.fields()
+        fields.append(QgsField(fieldName, QVariant.Int))        
 
         classFieldIndex = pointLayer.fieldNameIndex(classFieldName)
         (idxCount, fieldList) = vector.findOrCreateField(polyLayer,
                 polyLayer.pendingFields(), fieldName)
 
         writer = self.getOutputFromName(
-                self.OUTPUT).getVectorWriter(fieldList.toList(),
+                self.OUTPUT).getVectorWriter(fields.toList(),
                                              polyProvider.geometryType(),
                                              polyProvider.crs())
 

diff --git a/python/plugins/processing/algs/qgis/PointsInPolygonWeighted.py b/python/plugins/processing/algs/qgis/PointsInPolygonWeighted.py
@@ -73,12 +73,14 @@ def processAlgorithm(self, progress):
                 self.getParameterValue(self.WEIGHT))
 
         polyProvider = polyLayer.dataProvider()
+        fields = polyProvider.fields()
+        fields.append(QgsField(fieldName, QVariant.Int))        
 
         (idxCount, fieldList) = vector.findOrCreateField(polyLayer,
                 polyLayer.pendingFields(), fieldName)
 
         writer = self.getOutputFromName(
-                self.OUTPUT).getVectorWriter(fieldList.toList(),
+                self.OUTPUT).getVectorWriter(fields.toList(),
                                              polyProvider.geometryType(),
                                              polyProvider.crs())