-
Notifications
You must be signed in to change notification settings - Fork 0
/
ObjectLoader.h
67 lines (46 loc) · 1.5 KB
/
ObjectLoader.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#pragma once
#include <unordered_map>
#include <vector>
#include <objloader/objloader.h>
#include <string>
#include "VKVertex.h"
class ObjectLoader {
public:
tinyobj::attrib_t attrib;
std::vector<tinyobj::shape_t> shapes;
std::vector<tinyobj::material_t> materials;
std::string err;
std::vector<uint32_t> indices;
std::vector<VKVertex> vertices;
ObjectLoader::ObjectLoader( const std::string obj_file ) {
if ( ! tinyobj::LoadObj(&attrib, &shapes, &materials, &err, obj_file.c_str(), NULL, true ) ) {
throw std::runtime_error(err);
}
std::unordered_map<VKVertex, uint32_t> uniqueVertices = {};
for (const auto& shape : shapes) {
for (const auto& index : shape.mesh.indices) {
VKVertex vertex = {};
vertex.pos = {
attrib.vertices[3 * index.vertex_index + 0],
attrib.vertices[3 * index.vertex_index + 1],
attrib.vertices[3 * index.vertex_index + 2]
};
vertex.normal = {
attrib.normals[3 * index.normal_index + 0],
attrib.normals[3 * index.normal_index + 1],
attrib.normals[3 * index.normal_index + 2]
};
vertex.texCoord = {
attrib.texcoords[2 * index.texcoord_index + 0],
1.0f - attrib.texcoords[2 * index.texcoord_index + 1]
};
vertex.color = {1.0f, 1.0f, 1.0f};
if (uniqueVertices.count(vertex) == 0) {
uniqueVertices[vertex] = static_cast<uint32_t>(vertices.size());
vertices.push_back(vertex);
}
indices.push_back(uniqueVertices[vertex]);
}
}
}
};