Adding the binding of the getCategories() method

Author: samFabrice

bindings/SofaRuntime/src/SofaPython3/SofaRuntime/Module_SofaRuntime.cpp

@@ -48,6 +48,15 @@ using sofa::helper::system::PluginRepository;
 using sofa::simulation::SceneLoaderFactory;
 using sofa::simulation::SceneLoader;
 
+//
+#include <pybind11/stl.h>
+#include <sofa/core/ObjectFactory.h>
+using sofa::core::ObjectFactory;
+#include <sofa/core/CategoryLibrary.h>
+using sofa::core::CategoryLibrary;
+//
+
+
 #include <SofaPython3/SceneLoaderPY3.h>
 using sofapython3::SceneLoaderPY3;
 
@@ -68,6 +77,7 @@ using sofapython3::SceneLoaderPY3;
 namespace sofapython3
 {
 
+
 class SofaInitializer
 {
 public:
@@ -85,7 +95,25 @@ class SofaInitializer
     }
 };
 
+static std::vector<std::string>  getCategories(std::string className)
+{
+    std::vector<std::string> categories;
+    ObjectFactory* factory = ObjectFactory::getInstance();
 
+    if (factory->hasCreator(className))
+    {
+        ObjectFactory::ClassEntry entry = factory->getEntry(className);
+        ObjectFactory::CreatorMap::iterator it2 = entry.creatorMap.begin();
+
+        if( it2 != entry.creatorMap.end())
+        {
+                ObjectFactory::Creator::SPtr c = it2->second;
+                const sofa::core::objectmodel::BaseClass* objClass = c->getClass();
+                CategoryLibrary::getCategories(objClass,categories);
+        }
+    }
+    return categories ;
+}
 
 static SofaInitializer s;
 
@@ -132,6 +160,7 @@ PYBIND11_MODULE(SofaRuntime, m) {
 
     m.add_object("DataRepository", py::cast(&sofa::helper::system::DataRepository));
     m.add_object("PluginRepository", py::cast(&sofa::helper::system::PluginRepository));
+    m.def("getCategories", &getCategories);
 
     addSubmoduleTimer(m);
 }