Adds 2 convenience methods to ShibokenGenerator to check for refcount mods.

The new expressively named methods hasMethodsWithReferenceCountModifications
and needsReferenceCountControl returns boolean values to help generation
of code for reference counting support.

2 files changed, 26 insertions, 0 deletions

diff --git a/shibokengenerator.cpp b/shibokengenerator.cpp
index 4bcb038f8..82eff597b 100644
--- a/shibokengenerator.cpp
+++ b/shibokengenerator.cpp
@@ -1091,6 +1091,26 @@ bool ShibokenGenerator::hasMultipleInheritanceInAncestry(const AbstractMetaClass
     return hasMultipleInheritanceInAncestry(metaClass->baseClass());
 }
 
+bool ShibokenGenerator::needsReferenceCountControl(const AbstractMetaClass* metaClass)
+{
+    if (!metaClass->fields().isEmpty())
+        return true;
+    return hasMethodsWithReferenceCountModifications(metaClass);
+}
+
+bool ShibokenGenerator::hasMethodsWithReferenceCountModifications(const AbstractMetaClass* metaClass)
+{
+    foreach (const AbstractMetaFunction* func, metaClass->functions()) {
+        foreach (FunctionModification func_mod, func->modifications()) {
+            foreach (ArgumentModification arg_mod, func_mod.argument_mods) {
+                if (!arg_mod.referenceCounts.isEmpty())
+                    return true;
+            }
+        }
+    }
+    return false;
+}
+
 bool ShibokenGenerator::classNeedsGetattroFunction(const AbstractMetaClass* metaClass)
 {
     if (!metaClass)
diff --git a/shibokengenerator.h b/shibokengenerator.h
index a675aa749..cae9943b1 100644
--- a/shibokengenerator.h
+++ b/shibokengenerator.h
@@ -177,6 +177,12 @@ public:
     /// Returns true if there are cases of multiple inheritance in any of its ancestors.
     bool hasMultipleInheritanceInAncestry(const AbstractMetaClass* metaClass);
 
+    /// Returns true if the class needs reference counting control.
+    bool needsReferenceCountControl(const AbstractMetaClass* metaClass);
+
+    /// Returns true if the class has any method that modifies the reference counting of any of its arguments.
+    bool hasMethodsWithReferenceCountModifications(const AbstractMetaClass* metaClass);
+
     /// Returns true if the class needs to have a getattro function.
     bool classNeedsGetattroFunction(const AbstractMetaClass* metaClass);