aboutsummaryrefslogtreecommitdiff
path: root/examples/obj_sticher/obj_sticher.cc
diff options
context:
space:
mode:
Diffstat (limited to 'examples/obj_sticher/obj_sticher.cc')
-rw-r--r--examples/obj_sticher/obj_sticher.cc110
1 files changed, 110 insertions, 0 deletions
diff --git a/examples/obj_sticher/obj_sticher.cc b/examples/obj_sticher/obj_sticher.cc
new file mode 100644
index 0000000..f59fee4
--- /dev/null
+++ b/examples/obj_sticher/obj_sticher.cc
@@ -0,0 +1,110 @@
+//
+// Stiches multiple .obj files into one .obj.
+//
+#include "../../tiny_obj_loader.h"
+#include "obj_writer.h"
+
+#include <cassert>
+#include <iostream>
+#include <cstdlib>
+#include <cstdio>
+
+typedef std::vector<tinyobj::shape_t> Shape;
+typedef std::vector<tinyobj::material_t> Material;
+
+void
+StichObjs(
+ std::vector<tinyobj::shape_t>& out_shape,
+ std::vector<tinyobj::material_t>& out_material,
+ const std::vector<Shape>& shapes,
+ const std::vector<Material>& materials)
+{
+ int numShapes = 0;
+ for (size_t i = 0; i < shapes.size(); i++) {
+ numShapes += (int)shapes[i].size();
+ }
+
+ printf("Total # of shapes = %d\n", numShapes);
+ int materialIdOffset = 0;
+
+ size_t face_offset = 0;
+ for (size_t i = 0; i < shapes.size(); i++) {
+
+ for (size_t k = 0; k < shapes[i].size(); k++) {
+
+ std::string new_name = shapes[i][k].name;
+ // Add suffix
+ char buf[1024];
+ sprintf(buf, "_%04d", (int)i);
+ new_name += std::string(buf);
+
+ printf("shape[%ld][%ld].name = %s\n", i, k, shapes[i][k].name.c_str());
+ assert((shapes[i][k].mesh.indices.size() % 3) == 0);
+ assert((shapes[i][k].mesh.positions.size() % 3) == 0);
+
+ tinyobj::shape_t new_shape = shapes[i][k];
+ // Add offset.
+ for (size_t f = 0; f < new_shape.mesh.material_ids.size(); f++) {
+ new_shape.mesh.material_ids[f] += materialIdOffset;
+ }
+
+ new_shape.name = new_name;
+ printf("shape[%ld][%ld].new_name = %s\n", i, k, new_shape.name.c_str());
+
+ out_shape.push_back(new_shape);
+ }
+
+ materialIdOffset += materials[i].size();
+ }
+
+ for (size_t i = 0; i < materials.size(); i++) {
+ for (size_t k = 0; k < materials[i].size(); k++) {
+ out_material.push_back(materials[i][k]);
+ }
+ }
+
+}
+
+int
+main(
+ int argc,
+ char **argv)
+{
+ if (argc < 3) {
+ printf("Usage: obj_sticher input0.obj input1.obj ... output.obj\n");
+ exit(1);
+ }
+
+ int num_objfiles = argc - 2;
+ std::string out_filename = std::string(argv[argc-1]); // last element
+
+ std::vector<Shape> shapes;
+ std::vector<Material> materials;
+ shapes.resize(num_objfiles);
+ materials.resize(num_objfiles);
+
+ for (int i = 0; i < num_objfiles; i++) {
+ std::cout << "Loading " << argv[i+1] << " ... " << std::flush;
+
+ std::string err;
+ bool ret = tinyobj::LoadObj(shapes[i], materials[i], err, argv[i+1]);
+ if (!err.empty()) {
+ std::cerr << err << std::endl;
+ }
+ if (!ret) {
+ exit(1);
+ }
+
+ std::cout << "DONE." << std::endl;
+ }
+
+ std::vector<tinyobj::shape_t> out_shape;
+ std::vector<tinyobj::material_t> out_material;
+ StichObjs(out_shape, out_material, shapes, materials);
+
+ bool coordTransform = true;
+ bool ret = WriteObj(out_filename, out_shape, out_material, coordTransform);
+ assert(ret);
+
+ return 0;
+}