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.

1 files changed, 20 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)