QgsVectorLayerImport is a QgsFeatureSink

qgis · May 6, 2017 · 46ff652 · 46ff652
1 parent 8d03642
commit 46ff652
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 5 deletions.
diff --git a/python/core/qgsvectorlayerimport.sip b/python/core/qgsvectorlayerimport.sip
@@ -4,7 +4,7 @@
  1. static call to QgsVectorFileWriter::writeAsShapefile(...) which saves the whole vector layer
  2. create an instance of the class and issue calls to addFeature(...)
  */
-class QgsVectorLayerImport
+class QgsVectorLayerImport : QgsFeatureSink
 {
 %TypeHeaderCode
 #include <qgsvectorlayerimport.h>
@@ -85,7 +85,7 @@ class QProgressDialog;
 
     int errorCount() const;
 
-    /** Add feature to the new created layer */
+    bool addFeatures( QgsFeatureList &features );
     bool addFeature( QgsFeature &feature );
 
     /** Close the new created layer */

diff --git a/src/core/qgsvectorlayerimport.cpp b/src/core/qgsvectorlayerimport.cpp
@@ -145,6 +145,17 @@ QString QgsVectorLayerImport::errorMessage()
   return mErrorMessage;
 }
 
+bool QgsVectorLayerImport::addFeatures( QgsFeatureList &features )
+{
+  QgsFeatureList::iterator fIt = features.begin();
+  bool result = true;
+  for ( ; fIt != features.end(); ++fIt )
+  {
+    result = result && addFeature( *fIt );
+  }
+  return result;
+}
+
 bool QgsVectorLayerImport::addFeature( QgsFeature &feat )
 {
   QgsAttributes attrs = feat.attributes();

diff --git a/src/core/qgsvectorlayerimport.h b/src/core/qgsvectorlayerimport.h
@@ -22,6 +22,7 @@
 #include "qgis_core.h"
 #include "qgis.h"
 #include "qgsfeature.h"
+#include "qgsfeaturesink.h"
 
 class QProgressDialog;
 class QgsVectorDataProvider;
@@ -34,7 +35,7 @@ class QgsFields;
  1. static call to QgsVectorFileWriter::writeAsShapefile(...) which saves the whole vector layer
  2. create an instance of the class and issue calls to addFeature(...)
  */
-class CORE_EXPORT QgsVectorLayerImport
+class CORE_EXPORT QgsVectorLayerImport : public QgsFeatureSink
 {
   public:
 
@@ -114,8 +115,8 @@ class CORE_EXPORT QgsVectorLayerImport
 
     int errorCount() const { return mErrorCount; }
 
-    //! Add feature to the new created layer
-    bool addFeature( QgsFeature &feature );
+    bool addFeatures( QgsFeatureList &features ) override;
+    bool addFeature( QgsFeature &feature ) override;
 
     //! Close the new created layer
     ~QgsVectorLayerImport();