Add support for multiple inheritance involving more than one C++ object.

1 files changed, 139 insertions, 0 deletions

diff --git a/libshiboken/basewrapper_p.h b/libshiboken/basewrapper_p.h
new file mode 100644
index 000000000..555f40d49
--- /dev/null
+++ b/libshiboken/basewrapper_p.h
@@ -0,0 +1,139 @@
+/*
+* This file is part of the Shiboken Python Bindings Generator project.
+*
+* Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).
+*
+* Contact: PySide team <contact@pyside.org>
+*
+* This program is free software; you can redistribute it and/or
+* modify it under the terms of the GNU Lesser General Public License
+* version 2.1 as published by the Free Software Foundation. Please
+* review the following information to ensure the GNU Lesser General
+* Public License version 2.1 requirements will be met:
+* http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
+*
+* As a special exception to the GNU Lesser General Public License
+* version 2.1, the object code form of a "work that uses the Library"
+* may incorporate material from a header file that is part of the
+* Library.  You may distribute such object code under terms of your
+* choice, provided that the incorporated material (i) does not exceed
+* more than 5% of the total size of the Library; and (ii) is limited to
+* numerical parameters, data structure layouts, accessors, macros,
+* inline functions and templates.
+*
+* This program is distributed in the hope that it will be useful, but
+* WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+* Lesser General Public License for more details.
+*
+* You should have received a copy of the GNU Lesser General Public
+* License along with this program; if not, write to the Free Software
+* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+* 02110-1301 USA
+*/
+
+#ifndef BASEWRAPPER_P_H
+#define BASEWRAPPER_P_H
+
+#include <Python.h>
+#include <list>
+
+namespace Shiboken
+{
+
+struct SbkBaseWrapperType;
+
+/**
+*   Visitor class used by walkOnClassHierarchy function.
+*/
+class HierarchyVisitor
+{
+public:
+    HierarchyVisitor() : m_wasFinished(false) {}
+    virtual ~HierarchyVisitor() {}
+    virtual void visit(SbkBaseWrapperType* node) = 0;
+    void finish() { m_wasFinished = true; };
+    bool wasFinished() const { return m_wasFinished; }
+private:
+    bool m_wasFinished;
+};
+
+class BaseCountVisitor : public HierarchyVisitor
+{
+public:
+    BaseCountVisitor() : m_count(0) {}
+
+    void visit(SbkBaseWrapperType*)
+    {
+        m_count++;
+    }
+
+    int count() const { return m_count; }
+private:
+    int m_count;
+};
+
+class BaseAccumulatorVisitor : public HierarchyVisitor
+{
+public:
+    BaseAccumulatorVisitor() {}
+
+    void visit(SbkBaseWrapperType* node)
+    {
+        m_bases.push_back(node);
+    }
+
+    std::list<SbkBaseWrapperType*> bases() const { return m_bases; }
+private:
+    std::list<SbkBaseWrapperType*> m_bases;
+};
+
+class GetIndexVisitor : public HierarchyVisitor
+{
+public:
+    GetIndexVisitor(PyTypeObject* desiredType) : m_index(-1), m_desiredType(desiredType) {}
+    virtual void visit(SbkBaseWrapperType* node)
+    {
+        m_index++;
+        if (PyType_IsSubtype(reinterpret_cast<PyTypeObject*>(node), m_desiredType))
+            finish();
+    }
+    int index() const { return m_index; }
+
+private:
+    int m_index;
+    PyTypeObject* m_desiredType;
+};
+
+/// \internal Internal function used to walk on classes inheritance trees.
+/**
+*   Walk on class hierarchy using a DFS algorithm.
+*   For each pure Shiboken type found, HiearchyVisitor::visit is called and the algorithm consider
+*   all children of this type as visited.
+*/
+void walkThroughClassHierarchy(PyTypeObject* currentType, HierarchyVisitor* visitor);
+
+inline int getTypeIndexOnHierarchy(PyTypeObject* baseType, PyTypeObject* desiredType)
+{
+    GetIndexVisitor visitor(desiredType);
+    walkThroughClassHierarchy(baseType, &visitor);
+    return visitor.index();
+}
+
+inline int getNumberOfCppBaseClasses(PyTypeObject* baseType)
+{
+    BaseCountVisitor visitor;
+    walkThroughClassHierarchy(baseType, &visitor);
+    return visitor.count();
+}
+
+inline std::list<SbkBaseWrapperType*> getCppBaseClasses(PyTypeObject* baseType)
+{
+    BaseAccumulatorVisitor visitor;
+    walkThroughClassHierarchy(baseType, &visitor);
+    return visitor.bases();
+}
+
+}
+
+#endif