pyqgis: add conversion to subclasses for QgsAbstractVectorLayerLabeling

qgis · Feb 3, 2018 · 4b365a8 · 4b365a8
1 parent 57d38cd
commit 4b365a8
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/python/core/qgsvectorlayerlabeling.sip.in b/python/core/qgsvectorlayerlabeling.sip.in
@@ -21,6 +21,14 @@ Abstract base class - its implementations define different approaches to the lab
 
 %TypeHeaderCode
 #include "qgsvectorlayerlabeling.h"
+%End
+%ConvertToSubClassCode
+    if ( sipCpp->type() == "simple" )
+      sipType = sipType_QgsVectorLayerSimpleLabeling;
+    else if ( sipCpp->type() == "rule-based" )
+      sipType = sipType_QgsRuleBasedLabeling;
+    else
+      sipType = 0;
 %End
   public:
     QgsAbstractVectorLayerLabeling();

diff --git a/src/core/qgsvectorlayerlabeling.h b/src/core/qgsvectorlayerlabeling.h
@@ -39,6 +39,18 @@ class QgsVectorLayerLabelProvider;
  */
 class CORE_EXPORT QgsAbstractVectorLayerLabeling
 {
+
+#ifdef SIP_RUN
+    SIP_CONVERT_TO_SUBCLASS_CODE
+    if ( sipCpp->type() == "simple" )
+      sipType = sipType_QgsVectorLayerSimpleLabeling;
+    else if ( sipCpp->type() == "rule-based" )
+      sipType = sipType_QgsRuleBasedLabeling;
+    else
+      sipType = 0;
+    SIP_END
+#endif
+
   public:
     //! Default constructor
     QgsAbstractVectorLayerLabeling() = default;