allow saving single geometries as multi geometries

qgis · Nov 5, 2012 · 368146c · 368146c
1 parent 214b593
commit 368146c
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 1 deletion.
diff --git a/src/core/qgis.h b/src/core/qgis.h
@@ -66,6 +66,27 @@ class CORE_EXPORT QGis
       WKBMultiPolygon25D,
     };
 
+    static WkbType flatType( WkbType type )
+    {
+      switch ( type )
+      {
+        case WKBMultiPoint:
+          return WKBPoint;
+        case WKBMultiLineString:
+          return WKBLineString;
+        case WKBMultiPolygon:
+          return WKBPolygon;
+        case WKBMultiPoint25D:
+          return WKBPoint25D;
+        case WKBMultiLineString25D:
+          return WKBLineString25D;
+        case WKBMultiPolygon25D:
+          return WKBPolygon25D;
+        default:
+          return type;
+      }
+    }
+
     enum GeometryType
     {
       Point,

diff --git a/src/core/qgsvectorfilewriter.cpp b/src/core/qgsvectorfilewriter.cpp
@@ -482,6 +482,13 @@ bool QgsVectorFileWriter::addFeature( QgsFeature& feature )
   {
     // build geometry from WKB
     QgsGeometry *geom = feature.geometry();
+
+    // turn single geoemetry to multi geometry if needed
+    if ( geom && geom->wkbType() != mWkbType && geom->wkbType() == QGis::flatType( mWkbType ) )
+    {
+      geom->convertToMultiType();
+    }
+
     if ( geom && geom->wkbType() != mWkbType )
     {
       // there's a problem when layer type is set as wkbtype Polygon

diff --git a/src/providers/postgres/qgspostgresprovider.cpp b/src/providers/postgres/qgspostgresprovider.cpp
@@ -683,7 +683,7 @@ QString QgsPostgresProvider::pkParamWhereClause( int offset, const char *alias )
   QString whereClause;
 
   QString aliased;
-  if ( alias ) aliased = QString("%1.").arg( alias );
+  if ( alias ) aliased = QString( "%1." ).arg( alias );
 
   switch ( mPrimaryKeyType )
   {