namespace.hpp

//
// Copyright (c) 2019-2021 INRIA
//
 
#ifndef __pinocchio_python_utils_namespace_hpp__
#define __pinocchio_python_utils_namespace_hpp__
 
#include <boost/python.hpp>
#include <string>
 
namespace pinocchio
{
  namespace python
  {
 
    inline std::string getCurrentScopeName()
    {
      namespace bp = boost::python;
      bp::scope current_scope;
 
      return std::string(bp::extract<const char *>(current_scope.attr("__name__")));
    }
 
    inline boost::python::object getOrCreatePythonNamespace(const std::string & submodule_name)
    {
      namespace bp = boost::python;
 
      const std::string complete_submodule_name = getCurrentScopeName() + "." + submodule_name;
 
      bp::object submodule(bp::borrowed(PyImport_AddModule(complete_submodule_name.c_str())));
      bp::scope current_scope;
      current_scope.attr(submodule_name.c_str()) = submodule;
 
      return submodule;
    }
  } // namespace python
} // namespace pinocchio
 
#endif // ifndef __pinocchio_python_utils_namespace_hpp__