make QgsRelation::RelationStrength a Q_ENUM and simplify code

qgis · Sep 10, 2020 · e87ba45 · e87ba45
1 parent 27422fd
commit e87ba45
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 10 deletions.
diff --git a/python/core/auto_additions/qgsrelation.py b/python/core/auto_additions/qgsrelation.py
@@ -0,0 +1,2 @@
+# The following has been generated automatically from src/core/qgsrelation.h
+QgsRelation.RelationStrength.baseClass = QgsRelation
diff --git a/python/core/auto_generated/qgsrelation.sip.in b/python/core/auto_generated/qgsrelation.sip.in
@@ -26,7 +26,6 @@ class QgsRelation
     {
       Association,
       Composition
-
     };
 
 

diff --git a/src/core/qgsrelation.cpp b/src/core/qgsrelation.cpp
@@ -126,14 +126,7 @@ void QgsRelation::writeXml( QDomNode &node, QDomDocument &doc ) const
   elem.setAttribute( QStringLiteral( "name" ), d->mRelationName );
   elem.setAttribute( QStringLiteral( "referencingLayer" ), d->mReferencingLayerId );
   elem.setAttribute( QStringLiteral( "referencedLayer" ), d->mReferencedLayerId );
-  if ( d->mRelationStrength == RelationStrength::Composition )
-  {
-    elem.setAttribute( QStringLiteral( "strength" ), QStringLiteral( "Composition" ) );
-  }
-  else
-  {
-    elem.setAttribute( QStringLiteral( "strength" ), QStringLiteral( "Association" ) );
-  }
+  elem.setAttribute( QStringLiteral( "strength" ), qgsEnumValueToKey<RelationStrength>( d->mRelationStrength ) );
 
   for ( const FieldPair &pair : qgis::as_const( d->mFieldPairs ) )
   {

diff --git a/src/core/qgsrelation.h b/src/core/qgsrelation.h
@@ -58,8 +58,8 @@ class CORE_EXPORT QgsRelation
     {
       Association, //!< Loose relation, related elements are not part of the parent and a parent copy will not copy any children.
       Composition  //!< Fix relation, related elements are part of the parent and a parent copy will copy any children or delete of parent will delete children
-
     };
+    Q_ENUM( RelationStrength )
 
 #ifndef SIP_RUN