wiki:SampleCodes/ObjLoader

OBJ 파일과 BVH 를 로딩하는 샘플 코드

Line 
1#pragma mark Mesh Loading
2ModelBvh LoadObjModel(const DKString& obj, const DKString& tex, DKMaterial* material, bool genBvh, bool useCachedData)
3{
4    ModelBvh result = { "", NULL, NULL };
5    DKResourcePool pool;
6    DKObject<DKTexture2D> texture = pool.LoadResource(tex).SafeCast<DKTexture2D>();
7    if (texture == NULL)
8    {
9        DKLog("Failed to load texture: %ls\n", (const wchar_t*)tex);
10        return result;
11    }
12    DKString dkMeshFile = obj + ".dkmesh";
13    DKString dkBvhFile = obj + ".dktbvh";
14
15    DKObject<DKStaticMesh> mesh = NULL;
16    DKObject<DKTriangleMeshBvh> bvh = NULL;
17
18    if (useCachedData)
19    {
20        mesh = pool.LoadResource(dkMeshFile).SafeCast<DKStaticMesh>();
21        if (mesh && genBvh)
22        {
23            DKObject<DKData> triData = pool.LoadResourceData(dkBvhFile);
24            if (triData && triData->Length() > 0)
25            {
26                DKObject<BvhMesh> bvhMesh = DKOBJECT_NEW BvhMesh();
27
28                const DKTriangle* triangles = reinterpret_cast<const DKTriangle*>(triData->LockShared());
29                int numTri = (int)(triData->Length() / sizeof(DKTriangle));
30                bvhMesh->triangles.Add(triangles, numTri);
31                triData->UnlockShared();
32
33                bvh = DKOBJECT_NEW DKTriangleMeshBvh();
34                bvh->Build(bvhMesh);
35            }
36        }
37    }
38    if (mesh == NULL || (genBvh && bvh == NULL))
39    {
40        DKObject<DKData> objData = pool.LoadResourceData(obj);
41        if (objData && objData->Length() > 0)
42        {
43            DKLog("Loading %ls... (%lu bytes)\n", (const wchar_t*)obj, objData->Length());
44            DKTimer timer;
45            timer.Reset();
46
47            DKArray<DKVector3> positions;
48            DKArray<DKVector2> texcoords;
49            DKArray<DKVector3> normals;
50            struct Face
51            {
52                int pos, uv, normal;
53            };
54            DKArray<Face> faces;
55            DKString lineString;
56
57            positions.Reserve(0x10000);
58            texcoords.Reserve(0x10000);
59            normals.Reserve(0x10000);
60            faces.Reserve(0x10000);
61
62            DKUniChar8* p = (DKUniChar8*)objData->LockShared();
63            size_t length = objData->Length();
64            size_t pos = 0;
65            for (size_t i = 0; i < length; ++i)
66            {
67                if (p[i] == '\n')
68                {
69                    if (pos < i)
70                    {
71                        DKString::StringArray words = DKString(&p[pos], (i - pos)).SplitByWhitespace();
72                        if (words.Count() > 0)
73                        {
74                            if (words.Value(0).CompareNoCase("v") == 0) // vertex
75                            {
76                                if (words.Count() > 3)
77                                {
78                                    DKVector3 vector(words.Value(1).ToRealNumber(),
79                                                     words.Value(2).ToRealNumber(),
80                                                     words.Value(3).ToRealNumber());
81                                    positions.Add(vector);
82                                }
83                            }
84                            else if (words.Value(0).CompareNoCase("vn") == 0)   // normal
85                            {
86                                if (words.Count() > 3)
87                                {
88                                    DKVector3 vector(words.Value(1).ToRealNumber(),
89                                                     words.Value(2).ToRealNumber(),
90                                                     words.Value(3).ToRealNumber());
91                                    normals.Add(vector);
92                                }
93                            }
94                            else if (words.Value(0).CompareNoCase("vt") == 0)   // tex-coords
95                            {
96                                if (words.Count() > 2)
97                                {
98                                    DKVector2 vector(words.Value(1).ToRealNumber(),
99                                                     words.Value(2).ToRealNumber());
100
101                                    texcoords.Add(vector);
102                                }
103                            }
104                            else if (words.Value(0).CompareNoCase("f") == 0) // face
105                            {
106                                size_t numFaces = words.Count() - 1;
107                                Face f[4];
108                                int idx = 0;
109                                auto getIndex = [](int idx, int count) -> int
110                                {
111                                    int v;
112                                    if (idx > 0)
113                                        v = idx - 1;
114                                    else
115                                        v = count + idx;
116                                    return Clamp(v, 0, count - 1);
117                                };
118                                while (idx < numFaces)
119                                {
120                                    DKString::IntegerArray indices = words.Value(idx + 1).ToIntegerArray("/");
121                                    if (indices.Count() > 2)    // pos, uv, normal
122                                    {
123                                        f[idx].pos = getIndex(indices.Value(0), positions.Count());
124                                        f[idx].uv = getIndex(indices.Value(1), texcoords.Count());
125                                        f[idx].normal = getIndex(indices.Value(2), normals.Count());
126                                        idx++;
127                                    }
128                                    else if (indices.Count() > 1)   // pos, uv
129                                    {
130                                        f[idx].pos = getIndex(indices.Value(0), positions.Count());
131                                        f[idx].uv = getIndex(indices.Value(1), texcoords.Count());
132                                        f[idx].normal = 0;
133                                        idx++;
134                                    }
135                                    else
136                                        break;
137                                }
138                                if (idx == 3)       // triangle
139                                {
140                                    faces.Add(f, 3);
141                                }
142                                else if (idx == 4)  // quad
143                                {
144                                    faces.Add({ f[1], f[2], f[0], f[0], f[2], f[3] });
145                                }
146                                else    // error??
147                                {
148                                }
149                            }
150                        }
151                    }
152                    pos = i + 1;
153                }
154            }
155            objData->UnlockShared();
156            objData = NULL;
157
158            DKLog("Num Positions: %lu\n", positions.Count());
159            DKLog("Num TexCoords: %lu\n", texcoords.Count());
160            DKLog("Num Normals: %lu\n", normals.Count());
161            DKLog("Num Faces: %lu\n", faces.Count());
162
163            if (faces.Count() > 0 && positions.Count() > 0 && texcoords.Count() > 0)
164            {
165                struct Vertex
166                {
167                    DKVector3 pos;
168                    DKVector3 normal;
169                    DKVector2 uv;
170                };
171                DKAabb aabb;
172                DKArray<Vertex> vertices;
173                vertices.Reserve(faces.Count());
174                if (normals.Count() > 0)
175                {
176                    for (int i = 0; (i + 2) < faces.Count(); i += 3)
177                    {
178                        const Face& f1 = faces.Value(i);
179                        const Face& f2 = faces.Value(i + 1);
180                        const Face& f3 = faces.Value(i + 2);
181
182                        Vertex v[3] = {
183                            { positions.Value(f1.pos), normals.Value(f1.normal), texcoords.Value(f1.uv) },
184                        { positions.Value(f2.pos), normals.Value(f2.normal), texcoords.Value(f2.uv) },
185                        { positions.Value(f3.pos), normals.Value(f2.normal), texcoords.Value(f3.uv) },
186                        };
187
188                        vertices.Add(v, 3);
189                        aabb.Expand(v[0].pos);
190                        aabb.Expand(v[1].pos);
191                        aabb.Expand(v[2].pos);
192                    }
193                }
194                else
195                {
196                    DKLog("Generating face normals...\n");
197                    for (int i = 0; (i + 2) < faces.Count(); i += 3)
198                    {
199                        const Face& f1 = faces.Value(i);
200                        const Face& f2 = faces.Value(i + 1);
201                        const Face& f3 = faces.Value(i + 2);
202
203                        const DKVector3& pos1 = positions.Value(f1.pos);
204                        const DKVector3& pos2 = positions.Value(f2.pos);
205                        const DKVector3& pos3 = positions.Value(f3.pos);
206
207                        DKVector3 faceNormal = DKVector3::Cross(pos2 - pos1, pos3 - pos1).Normalize();
208
209                        Vertex v[3] = {
210                            { pos1, faceNormal, texcoords.Value(f1.uv) },
211                        { pos2, faceNormal, texcoords.Value(f2.uv) },
212                        { pos3, faceNormal, texcoords.Value(f3.uv) },
213                        };
214
215                        vertices.Add(v, 3);
216                        aabb.Expand(v[0].pos);
217                        aabb.Expand(v[1].pos);
218                        aabb.Expand(v[2].pos);
219                    }
220                }
221                DKVertexBuffer::Decl decls[3] = {
222                    { DKVertexStream::StreamPosition, L"", DKVertexStream::TypeFloat3, false },
223                { DKVertexStream::StreamNormal , L"", DKVertexStream::TypeFloat3, false },
224                { DKVertexStream::StreamTexCoord, L"", DKVertexStream::TypeFloat2, false },
225                };
226                DKObject<DKVertexBuffer> vb = DKVertexBuffer::Create(decls, 3, vertices, sizeof(Vertex), vertices.Count(), DKVertexBuffer::MemoryLocationStatic, DKVertexBuffer::BufferUsageDraw);
227
228                mesh = DKObject<DKStaticMesh>::New();
229                mesh->SetDrawFace(DKMesh::DrawFaceBoth);
230                mesh->SetDefaultPrimitiveType(DKPrimitive::TypeTriangles);
231                mesh->AddVertexBuffer(vb);
232                mesh->SetAabb(aabb);
233                DKLog("Obj %ls file loaded. (%f sec)\n", (const wchar_t*)obj, timer.Elapsed());
234
235                DKObject<BvhMesh> bvhMesh = NULL;
236                if (genBvh)
237                {
238                    // Create bvh
239                    DKTimer bvhTimer;
240                    bvhTimer.Reset();
241
242                    bvhMesh = DKOBJECT_NEW BvhMesh();
243                    int numTri = (int)(vertices.Count() / 3);
244                    bvhMesh->triangles.Reserve(numTri);
245                    for (int i = 0; i < numTri; ++i)
246                    {
247                        const Vertex& v1 = vertices.Value(i * 3);
248                        const Vertex& v2 = vertices.Value(i * 3 + 1);
249                        const Vertex& v3 = vertices.Value(i * 3 + 2);
250
251                        DKTriangle t = { v1.pos, v2.pos, v3.pos };
252                        bvhMesh->triangles.Add(t);
253                    }
254                    bvh = DKOBJECT_NEW DKTriangleMeshBvh();
255                    bvh->Build(bvhMesh);
256
257                    double d = bvhTimer.Elapsed();
258                    DKLog("Generating BVH elapsed: %f\n", d);
259                }
260                if (useCachedData)
261                {
262                    DKObject<DKData> data = mesh->Serialize(DKSerializer::SerializeFormCompressedBinary);
263                    if (data)
264                    {
265                        DKLog("Writing DKMesh file: %ls\n", (const wchar_t*)dkMeshFile);
266                        data->WriteToFile(dkMeshFile, true);
267                    }
268                    if (bvhMesh && bvhMesh->triangles.Count() > 0)
269                    {
270                        data = DKData::StaticData(bvhMesh->triangles, sizeof(DKTriangle) * bvhMesh->triangles.Count());
271                        if (data)
272                        {
273                            DKLog("Writing DKBvh file: %ls\n", (const wchar_t*)dkBvhFile);
274                            data->WriteToFile(dkBvhFile, true);
275                        }
276                    }
277                }
278            }
279        }
280        else
281        {
282            // error.
283            DKLog("Failed to load obj file: %ls\n", (const wchar_t*)obj);
284        }
285    }
286    if (mesh)
287    {
288        DKAabb aabb = mesh->Aabb();
289        DKLog("AABB: (%.3f, %.3f, %.3f), (%.3f, %.3f, %.3f)\n",
290              aabb.positionMin.x, aabb.positionMin.y, aabb.positionMin.z,
291              aabb.positionMax.x, aabb.positionMax.y, aabb.positionMax.z);
292        DKLog("AABB width: %.3f, height: %.3f, depth: %.3f\n",
293              aabb.positionMax.x - aabb.positionMin.x,
294              aabb.positionMax.y - aabb.positionMin.y,
295              aabb.positionMax.z - aabb.positionMin.z);
296
297        //      DKObject<DKTextureSampler> sampler = DKOBJECT_NEW DKTextureSampler();
298        //      sampler->magFilter = DKTextureSampler::MagFilterNearest;
299        //      sampler->minFilter = DKTextureSampler::MinFilterNearest;
300        mesh->SetMaterial(material);
301        mesh->SetSampler(L"diffuseMap", DKMaterial::TextureArray(texture.SafeCast<DKTexture>(), 1), NULL);
302        result.mesh = mesh;
303        result.bvh = bvh;
304    }
305    else
306    {
307        DKLog("Load %ls failed.\n", (const wchar_t*)obj);
308    }
309    return result;
310}
  • DKGL develop 브랜치 버전용
Last modified 4 months ago Last modified on 07/03/18 20:08:58