Program Listing for File namespace.hpp

↰ Return to documentation for file (include/pinocchio/bindings/python/utils/namespace.hpp)

//
// Copyright (c) 2019-2021 INRIA
//

#ifndef __pinocchio_python_utils_namespace_hpp__
#define __pinocchio_python_utils_namespace_hpp__

#include <boost/python.hpp>

namespace pinocchio
{
  namespace python
  {

    inline boost::python::object getOrCreatePythonNamespace(const std::string & submodule_name)
    {
      namespace bp = boost::python;

      bp::scope current_scope;
      std::string current_scope_name(bp::extract<const char*>(current_scope.attr("__name__")));
      std::string complete_submodule_name = current_scope_name + "." + submodule_name;

      bp::object submodule(bp::borrowed(PyImport_AddModule(complete_submodule_name.c_str())));
      current_scope.attr(submodule_name.c_str()) = submodule;

      return submodule;
    }
  } // namespace python
} // namespace pinocchio

#endif // ifndef __pinocchio_python_utils_namespace_hpp__