Implement some 2.0 API.

syoyo · syoyo · commit 8b3d2439ec7d · 2019-03-24T19:30:15.000+09:00
Write some python bindings using pybind11.
diff --git a/python/Makefile b/python/Makefile
@@ -0,0 +1,2 @@
+all:
+	python setup.py build
diff --git a/python/bindings.cc b/python/bindings.cc
@@ -8,19 +8,6 @@ namespace py = pybind11;
 
 using namespace tinyobj;
 
-// Simple wrapper for LoadObj.
-// TODO(syoyo): Make it more pythonic.
-bool load_obj(attrib_t *attrib, std::vector<shape_t> *shapes,
-              std::vector<material_t> *materials,
-              std::string *warn, std::string *err,
-              const std::string &filename,
-              const std::string &mtl_basedir = "./",
-              bool triangulate = true,
-              bool default_vcols_fallback = true)
-{
-  return LoadObj(attrib, shapes, materials, warn, err, filename.c_str(), mtl_basedir.c_str(), triangulate, default_vcols_fallback);
-}
-
 // Simple wrapper for LoadMtl.
 // TODO(syoyo): Make it more pythonic.
 void load_mtl(std::map<std::string, int> *material_map,
@@ -36,14 +23,28 @@ PYBIND11_MODULE(tinyobjloader, tobj_module)
 
   // register struct
   // py::init<>() for default constructor
-  py::class_<attrib_t>(tobj_module, "Attrib")
+  py::class_<ObjLoader>(tobj_module, "ObjLoader")
+    .def(py::init<>())
+    .def("Load", &ObjLoader::Load)
+    .def("Valid", &ObjLoader::Valid)
+    .def("GetAttrib", &ObjLoader::GetAttrib)
+    .def("GetShapes", &ObjLoader::GetShapes)
+    .def("GetMaterials", &ObjLoader::GetMaterials)
+    .def("Warning", &ObjLoader::Warning)
+    .def("Error", &ObjLoader::Error);
+
+  py::class_<attrib_t>(tobj_module, "attrib_t")
+    .def(py::init<>())
+    .def_property_readonly("vertices", &attrib_t::GetVertices);
+
+  py::class_<shape_t>(tobj_module, "shape_t")
     .def(py::init<>());
-  py::class_<shape_t>(tobj_module, "Shape")
+
+  py::class_<material_t>(tobj_module, "material_t")
     .def(py::init<>());
-  py::class_<material_t>(tobj_module, "Material")
+
+  py::class_<ObjLoaderConfig>(tobj_module, "ObjLoaderConfig")
     .def(py::init<>());
 
-  tobj_module.def("load_obj", &load_obj, "Load wavefront .obj file.");
-  tobj_module.def("load_mtl", &load_mtl, "Load wavefront material file.");
 }
 
diff --git a/python/sample.py b/python/sample.py
@@ -0,0 +1,19 @@
+import sys
+import tinyobjloader
+
+filename = "../models/cornell_box.obj";
+
+config = tinyobjloader.ObjLoaderConfig()
+
+loader = tinyobjloader.ObjLoader()
+
+ret = loader.Load(filename, config)
+
+if ret == False:
+    print("Failed to load : ", filename)
+    sys.exit(-1)
+
+attrib = loader.GetAttrib()
+print("attrib.vertices = ", len(attrib.vertices))
+for v in attrib.vertices:
+    print(v)
diff --git a/tiny_obj_loader.h b/tiny_obj_loader.h
