Fix subclassing of QgsDiagramRenderer in Python

qgis · Jun 22, 2017 · e6863be · e6863be
1 parent 78a990b
commit e6863be
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/python/core/qgsdiagramrenderer.sip b/python/core/qgsdiagramrenderer.sip
@@ -450,6 +450,15 @@ class QgsDiagramRenderer
 
 %TypeHeaderCode
 #include "qgsdiagramrenderer.h"
+%End
+
+%ConvertToSubClassCode
+    if ( sipCpp->rendererName() == QStringLiteral( "SingleCategory" ) )
+      sipType = sipType_QgsSingleCategoryDiagramRenderer;
+    else if ( sipCpp->rendererName() == QStringLiteral( "LinearlyInterpolated" ) )
+      sipType = sipType_QgsLinearlyInterpolatedDiagramRenderer;
+    else
+      sipType = NULL;
 %End
   public:
 

diff --git a/src/core/qgsdiagramrenderer.h b/src/core/qgsdiagramrenderer.h
@@ -485,6 +485,18 @@ class CORE_EXPORT QgsDiagramInterpolationSettings
 
 class CORE_EXPORT QgsDiagramRenderer
 {
+
+#ifdef SIP_RUN
+    SIP_CONVERT_TO_SUBCLASS_CODE
+    if ( sipCpp->rendererName() == QStringLiteral( "SingleCategory" ) )
+      sipType = sipType_QgsSingleCategoryDiagramRenderer;
+    else if ( sipCpp->rendererName() == QStringLiteral( "LinearlyInterpolated" ) )
+      sipType = sipType_QgsLinearlyInterpolatedDiagramRenderer;
+    else
+      sipType = NULL;
+    SIP_END
+#endif
+
   public:
 
     QgsDiagramRenderer();