1 files changed, 34 insertions, 0 deletions
diff --git a/libs/plugin_python/src/mobase/wrappers/pyfiletree.h b/libs/plugin_python/src/mobase/wrappers/pyfiletree.h
new file mode 100644
index 0000000..0e94665
--- /dev/null
+++ b/libs/plugin_python/src/mobase/wrappers/pyfiletree.h
@@ -0,0 +1,34 @@
+#ifndef MO2_PYTHON_FILETREE_H
+#define MO2_PYTHON_FILETREE_H
+
+#include "../pybind11_all.h"
+
+#include <uibase/ifiletree.h>
+
+namespace pybind11 {
+    template <>
+    struct polymorphic_type_hook<MOBase::FileTreeEntry> {
+        static const void* get(const MOBase::FileTreeEntry* src,
+                               const std::type_info*& type);
+    };
+}  // namespace pybind11
+
+namespace mo2::python {
+
+    /**
+     * @brief Add bindings for FileTreeEntry andIFileTree to the given module.
+     *
+     * @param mobase Module to add the bindings to.
+     */
+    void add_ifiletree_bindings(pybind11::module_& m);
+
+    /**
+     * @brief Add makeTree() function to the given module, useful for debugging.
+     *
+     * @param mobase Module to add the function to.
+     */
+    void add_make_tree_function(pybind11::module_& m);
+
+}  // namespace mo2::python
+
+#endif