rev |
line source |
nuclear@6
|
1 #include <stdio.h>
|
nuclear@6
|
2 #include <stdlib.h>
|
nuclear@6
|
3 #include <string.h>
|
John@15
|
4 #include <math.h>
|
nuclear@6
|
5 #include <errno.h>
|
nuclear@8
|
6 #include <limits.h>
|
nuclear@6
|
7 #include <string>
|
nuclear@6
|
8 #include <vector>
|
nuclear@6
|
9 #include <map>
|
nuclear@6
|
10 #include "mesh.h"
|
nuclear@6
|
11
|
John@11
|
12 #ifndef PATH_MAX
|
John@11
|
13 #define PATH_MAX 512
|
John@11
|
14 #endif
|
John@11
|
15
|
nuclear@6
|
16 using namespace std;
|
nuclear@6
|
17
|
nuclear@8
|
18 #define COMMANDS \
|
nuclear@6
|
19 CMD(V), \
|
nuclear@6
|
20 CMD(VN), \
|
nuclear@6
|
21 CMD(VT), \
|
nuclear@6
|
22 CMD(F), \
|
nuclear@6
|
23 CMD(O), \
|
nuclear@6
|
24 CMD(G), \
|
nuclear@6
|
25 CMD(MTLLIB), \
|
nuclear@6
|
26 CMD(USEMTL), \
|
nuclear@6
|
27 CMD(NEWMTL), \
|
nuclear@6
|
28 CMD(KA), \
|
nuclear@6
|
29 CMD(KD), \
|
nuclear@6
|
30 CMD(KS), \
|
nuclear@16
|
31 CMD(KR), \
|
nuclear@6
|
32 CMD(NS), \
|
nuclear@6
|
33 CMD(NI), \
|
nuclear@6
|
34 CMD(D), \
|
nuclear@6
|
35 CMD(TR), \
|
nuclear@6
|
36 CMD(MAP_KD), \
|
nuclear@6
|
37 CMD(MAP_KS), \
|
nuclear@6
|
38 CMD(MAP_NS), \
|
nuclear@6
|
39 CMD(MAP_D), \
|
nuclear@6
|
40 CMD(REFL), \
|
nuclear@6
|
41 CMD(BUMP)
|
nuclear@6
|
42
|
nuclear@6
|
43 #define CMD(x) CMD_##x
|
nuclear@6
|
44 enum {
|
nuclear@6
|
45 COMMANDS,
|
nuclear@6
|
46 CMD_UNK
|
nuclear@6
|
47 };
|
nuclear@6
|
48 #undef CMD
|
nuclear@6
|
49
|
nuclear@6
|
50 #define CMD(x) #x
|
nuclear@6
|
51 static const char *cmd_names[] = {
|
nuclear@6
|
52 COMMANDS,
|
nuclear@6
|
53 0
|
nuclear@6
|
54 };
|
nuclear@6
|
55 #undef CMD
|
nuclear@6
|
56
|
nuclear@6
|
57 struct Vector3 {
|
nuclear@6
|
58 float x, y, z;
|
nuclear@6
|
59
|
nuclear@6
|
60 Vector3() { x = y = z = 0.0; }
|
nuclear@6
|
61 Vector3(float a, float b, float c) { x = a; y = b; z = c; }
|
nuclear@6
|
62 };
|
nuclear@6
|
63
|
nuclear@6
|
64 struct Vector2 {
|
nuclear@6
|
65 float x, y;
|
nuclear@6
|
66
|
nuclear@6
|
67 Vector2() { x = y = 0.0; }
|
nuclear@6
|
68 Vector2(float a, float b) { x = a; y = b; }
|
nuclear@6
|
69 };
|
nuclear@6
|
70
|
nuclear@6
|
71 struct obj_face {
|
nuclear@6
|
72 int elem;
|
nuclear@6
|
73 int v[4], n[4], t[4];
|
nuclear@6
|
74 };
|
nuclear@6
|
75
|
nuclear@6
|
76 struct obj_file {
|
nuclear@6
|
77 string cur_obj, cur_mat;
|
nuclear@6
|
78 vector<Vector3> v, vn, vt;
|
nuclear@6
|
79 vector<obj_face> f;
|
nuclear@6
|
80 };
|
nuclear@6
|
81
|
nuclear@6
|
82 struct obj_mat {
|
nuclear@6
|
83 string name; // newmtl <name>
|
nuclear@6
|
84 Vector3 ambient, diffuse, specular; // Ka, Kd, Ks
|
nuclear@6
|
85 float shininess; // Ns
|
nuclear@6
|
86 float ior; // Ni
|
nuclear@6
|
87 float alpha; // d, Tr
|
nuclear@16
|
88 float refl; // Kr (my extesnsion)
|
nuclear@6
|
89
|
nuclear@6
|
90 string tex_dif, tex_spec, tex_shin, tex_alpha; // map_Kd, map_Ks, map_Ns, map_d
|
nuclear@6
|
91 string tex_refl; // refl -type sphere|cube file
|
nuclear@6
|
92 string tex_bump; // bump
|
nuclear@6
|
93
|
nuclear@6
|
94 obj_mat() { reset(); }
|
nuclear@6
|
95
|
nuclear@6
|
96 void reset() {
|
nuclear@6
|
97 ambient = diffuse = Vector3(0.5, 0.5, 0.5);
|
nuclear@6
|
98 specular = Vector3(0.0, 0.0, 0.0);
|
nuclear@6
|
99 name = tex_dif = tex_spec = tex_shin = tex_alpha = tex_refl = tex_bump = "";
|
nuclear@6
|
100 shininess = 0;
|
nuclear@6
|
101 ior = alpha = 1;
|
nuclear@6
|
102 }
|
nuclear@6
|
103 };
|
nuclear@6
|
104
|
nuclear@6
|
105 static bool read_materials(FILE *fp, vector<obj_mat> *vmtl);
|
nuclear@6
|
106 static Mesh *cons_mesh(obj_file *obj);
|
nuclear@6
|
107
|
nuclear@6
|
108 static int get_cmd(char *str);
|
nuclear@6
|
109 static bool is_int(const char *str);
|
nuclear@6
|
110 static bool is_float(const char *str);
|
nuclear@6
|
111 static bool parse_vec(Vector3 *vec);
|
nuclear@6
|
112 static bool parse_color(Vector3 *col);
|
nuclear@6
|
113 static bool parse_face(obj_face *face);
|
nuclear@6
|
114 static const char *parse_map();
|
nuclear@6
|
115
|
nuclear@6
|
116 static bool find_file(char *res, int sz, const char *fname, const char *path = ".", const char *mode = "rb");
|
nuclear@6
|
117 static const char *dirname(const char *str);
|
nuclear@6
|
118
|
nuclear@6
|
119 static map<string, int> matnames;
|
nuclear@6
|
120
|
John@15
|
121
|
John@15
|
122 #define FEQ(a, b) (fabs((a) - (b)) < 1e-8)
|
John@15
|
123 bool Face::operator ==(const Face &f) const
|
John@15
|
124 {
|
John@15
|
125 for(int i=0; i<3; i++) {
|
John@15
|
126 for(int j=0; j<3; j++) {
|
John@15
|
127 if(!FEQ(v[i].pos[j], f.v[i].pos[j])) {
|
John@15
|
128 return false;
|
John@15
|
129 }
|
John@15
|
130 if(!FEQ(v[i].normal[j], f.v[i].normal[j])) {
|
John@15
|
131 return false;
|
John@15
|
132 }
|
John@15
|
133 }
|
John@15
|
134 if(!FEQ(normal[i], f.normal[i])) {
|
John@15
|
135 return false;
|
John@15
|
136 }
|
John@15
|
137 }
|
John@15
|
138 return true;
|
John@15
|
139 }
|
John@15
|
140
|
nuclear@13
|
141 bool Scene::add_mesh(Mesh *m)
|
nuclear@13
|
142 {
|
nuclear@13
|
143 // make sure triangles have material ids
|
nuclear@13
|
144 for(size_t i=0; i<m->faces.size(); i++) {
|
nuclear@13
|
145 m->faces[i].matid = m->matid;
|
nuclear@13
|
146 }
|
nuclear@13
|
147 meshes.push_back(m);
|
nuclear@13
|
148 return true;
|
nuclear@13
|
149 }
|
nuclear@13
|
150
|
John@14
|
151 int Scene::get_num_meshes() const
|
John@14
|
152 {
|
John@14
|
153 return (int)meshes.size();
|
John@14
|
154 }
|
John@14
|
155
|
nuclear@13
|
156 int Scene::get_num_faces() const
|
nuclear@13
|
157 {
|
nuclear@13
|
158 int num_faces = 0;
|
nuclear@13
|
159 for(size_t i=0; i<meshes.size(); i++) {
|
nuclear@13
|
160 num_faces += meshes[i]->faces.size();
|
nuclear@13
|
161 }
|
nuclear@13
|
162 printf("get_num_faces() = %d\n", num_faces);
|
nuclear@13
|
163 return num_faces;
|
nuclear@13
|
164 }
|
nuclear@13
|
165
|
John@14
|
166 int Scene::get_num_materials() const
|
John@14
|
167 {
|
John@14
|
168 return (int)matlib.size();
|
John@14
|
169 }
|
John@14
|
170
|
John@14
|
171 Material *Scene::get_materials()
|
John@14
|
172 {
|
John@14
|
173 if(matlib.empty()) {
|
John@14
|
174 return 0;
|
John@14
|
175 }
|
John@14
|
176 return &matlib[0];
|
John@14
|
177 }
|
John@14
|
178
|
John@14
|
179 const Material *Scene::get_materials() const
|
John@14
|
180 {
|
John@14
|
181 if(matlib.empty()) {
|
John@14
|
182 return 0;
|
John@14
|
183 }
|
John@14
|
184 return &matlib[0];
|
John@14
|
185 }
|
John@14
|
186
|
nuclear@6
|
187
|
nuclear@6
|
188 #define INVALID_IDX INT_MIN
|
nuclear@6
|
189
|
nuclear@6
|
190 #define SEP " \t\n\r\v"
|
nuclear@6
|
191 #define BUF_SZ 512
|
nuclear@6
|
192
|
nuclear@6
|
193 bool Scene::load(const char *fname)
|
nuclear@6
|
194 {
|
nuclear@6
|
195 FILE *fp;
|
nuclear@6
|
196
|
nuclear@6
|
197 if(!(fp = fopen(fname, "rb"))) {
|
nuclear@6
|
198 fprintf(stderr, "failed to open %s: %s\n", fname, strerror(errno));
|
nuclear@6
|
199 return false;
|
nuclear@6
|
200 }
|
nuclear@6
|
201
|
nuclear@6
|
202 bool res = load(fp);
|
nuclear@6
|
203 fclose(fp);
|
nuclear@6
|
204 return res;
|
nuclear@6
|
205 }
|
nuclear@6
|
206
|
nuclear@6
|
207 bool Scene::load(FILE *fp)
|
nuclear@6
|
208 {
|
nuclear@6
|
209 static int seq;
|
nuclear@6
|
210 char cur_name[16];
|
nuclear@6
|
211
|
nuclear@6
|
212 obj_file obj;
|
nuclear@8
|
213
|
nuclear@6
|
214 sprintf(cur_name, "default%02d.obj", seq++);
|
nuclear@6
|
215 obj.cur_obj = cur_name;
|
nuclear@6
|
216
|
nuclear@6
|
217 int prev_cmd = 0, obj_added = 0;
|
nuclear@6
|
218 for(;;) {
|
nuclear@6
|
219 Vector3 vec;
|
nuclear@6
|
220 obj_face face;
|
nuclear@8
|
221
|
nuclear@6
|
222 char line[BUF_SZ];
|
nuclear@6
|
223 fgets(line, sizeof line, fp);
|
nuclear@6
|
224 if(feof(fp)) {
|
nuclear@6
|
225 break;
|
nuclear@6
|
226 }
|
nuclear@6
|
227
|
nuclear@6
|
228 char *tok;
|
nuclear@6
|
229 if(!(tok = strtok(line, SEP))) {
|
nuclear@6
|
230 continue; // ignore empty lines
|
nuclear@6
|
231 }
|
nuclear@6
|
232
|
nuclear@6
|
233 int cmd;
|
nuclear@6
|
234 if((cmd = get_cmd(tok)) == -1) {
|
nuclear@6
|
235 continue; // ignore unknown commands ...
|
nuclear@6
|
236 }
|
nuclear@6
|
237
|
nuclear@6
|
238 switch(cmd) {
|
nuclear@6
|
239 case CMD_V:
|
nuclear@6
|
240 if(!parse_vec(&vec)) {
|
nuclear@6
|
241 continue;
|
nuclear@6
|
242 }
|
nuclear@6
|
243 obj.v.push_back(vec);
|
nuclear@6
|
244 break;
|
nuclear@6
|
245
|
nuclear@6
|
246 case CMD_VN:
|
nuclear@6
|
247 if(!parse_vec(&vec)) {
|
nuclear@6
|
248 continue;
|
nuclear@6
|
249 }
|
nuclear@6
|
250 obj.vn.push_back(vec);
|
nuclear@6
|
251 break;
|
nuclear@6
|
252
|
nuclear@6
|
253 case CMD_VT:
|
nuclear@6
|
254 if(!parse_vec(&vec)) {
|
nuclear@6
|
255 continue;
|
nuclear@6
|
256 }
|
nuclear@6
|
257 vec.y = 1.0 - vec.y;
|
nuclear@6
|
258 obj.vt.push_back(vec);
|
nuclear@6
|
259 break;
|
nuclear@6
|
260
|
nuclear@6
|
261 case CMD_O:
|
nuclear@6
|
262 case CMD_G:
|
nuclear@6
|
263 if(prev_cmd == CMD_O || prev_cmd == CMD_G) {
|
nuclear@6
|
264 break; // just in case we've got both of them in a row
|
nuclear@6
|
265 }
|
nuclear@6
|
266 /* if we have any previous data, group them up, add the object
|
nuclear@6
|
267 * and continue with the new one...
|
nuclear@6
|
268 */
|
nuclear@6
|
269 if(!obj.f.empty()) {
|
nuclear@6
|
270 Mesh *mesh = cons_mesh(&obj);
|
nuclear@6
|
271 mesh->matid = matnames[obj.cur_mat];
|
nuclear@13
|
272 add_mesh(mesh);
|
nuclear@6
|
273 obj_added++;
|
nuclear@6
|
274
|
nuclear@6
|
275 obj.f.clear(); // clean the face list
|
nuclear@6
|
276 }
|
nuclear@6
|
277 if((tok = strtok(0, SEP))) {
|
nuclear@6
|
278 obj.cur_obj = tok;
|
nuclear@6
|
279 } else {
|
nuclear@6
|
280 sprintf(cur_name, "default%02d.obj", seq++);
|
nuclear@6
|
281 obj.cur_obj = cur_name;
|
nuclear@6
|
282 }
|
nuclear@6
|
283 break;
|
nuclear@6
|
284
|
nuclear@6
|
285 case CMD_MTLLIB:
|
nuclear@6
|
286 if((tok = strtok(0, SEP))) {
|
nuclear@6
|
287 char path[PATH_MAX];
|
nuclear@6
|
288
|
nuclear@6
|
289 sprintf(path, ".:%s", dirname(tok));
|
nuclear@6
|
290 if(!find_file(path, PATH_MAX, tok, path)) {
|
nuclear@6
|
291 fprintf(stderr, "material library not found: %s\n", tok);
|
nuclear@6
|
292 continue;
|
nuclear@6
|
293 }
|
nuclear@6
|
294
|
nuclear@6
|
295 FILE *mfile;
|
nuclear@6
|
296 if(!(mfile = fopen(path, "rb"))) {
|
nuclear@6
|
297 fprintf(stderr, "failed to open material library: %s\n", path);
|
nuclear@6
|
298 continue;
|
nuclear@6
|
299 }
|
nuclear@6
|
300
|
nuclear@6
|
301 // load all materials of the mtl file into a vector
|
nuclear@6
|
302 vector<obj_mat> vmtl;
|
nuclear@6
|
303 if(!read_materials(mfile, &vmtl)) {
|
nuclear@6
|
304 continue;
|
nuclear@6
|
305 }
|
nuclear@6
|
306 fclose(mfile);
|
nuclear@6
|
307
|
nuclear@6
|
308 // and add them all to the scene
|
nuclear@6
|
309 for(size_t i=0; i<vmtl.size(); i++) {
|
nuclear@6
|
310 Material mat;
|
nuclear@6
|
311
|
nuclear@6
|
312 mat.kd[0] = vmtl[i].diffuse.x;
|
nuclear@6
|
313 mat.kd[1] = vmtl[i].diffuse.y;
|
nuclear@6
|
314 mat.kd[2] = vmtl[i].diffuse.z;
|
nuclear@6
|
315
|
nuclear@6
|
316 mat.ks[0] = vmtl[i].specular.x;
|
nuclear@6
|
317 mat.ks[1] = vmtl[i].specular.y;
|
nuclear@6
|
318 mat.ks[2] = vmtl[i].specular.z;
|
nuclear@6
|
319
|
nuclear@6
|
320 mat.kt = 1.0 - vmtl[i].alpha;
|
nuclear@16
|
321 mat.kr = vmtl[i].refl;
|
nuclear@6
|
322 mat.spow = vmtl[i].shininess;
|
nuclear@8
|
323
|
nuclear@13
|
324 matlib.push_back(mat);
|
nuclear@6
|
325 matnames[vmtl[i].name] = i;
|
nuclear@6
|
326 }
|
nuclear@6
|
327 }
|
nuclear@6
|
328 break;
|
nuclear@6
|
329
|
nuclear@6
|
330 case CMD_USEMTL:
|
nuclear@6
|
331 if((tok = strtok(0, SEP))) {
|
nuclear@6
|
332 obj.cur_mat = tok;
|
nuclear@6
|
333 } else {
|
nuclear@6
|
334 obj.cur_mat = "";
|
nuclear@6
|
335 }
|
nuclear@6
|
336 break;
|
nuclear@6
|
337
|
nuclear@6
|
338 case CMD_F:
|
nuclear@6
|
339 if(!parse_face(&face)) {
|
nuclear@6
|
340 continue;
|
nuclear@6
|
341 }
|
nuclear@6
|
342
|
nuclear@6
|
343 // convert negative indices to regular indices
|
nuclear@6
|
344 for(int i=0; i<4; i++) {
|
nuclear@6
|
345 if(face.v[i] < 0 && face.v[i] != INVALID_IDX) {
|
nuclear@6
|
346 face.v[i] = obj.v.size() + face.v[i];
|
nuclear@6
|
347 }
|
nuclear@6
|
348 if(face.n[i] < 0 && face.n[i] != INVALID_IDX) {
|
nuclear@6
|
349 face.n[i] = obj.vn.size() + face.n[i];
|
nuclear@6
|
350 }
|
nuclear@6
|
351 if(face.t[i] < 0 && face.t[i] != INVALID_IDX) {
|
nuclear@6
|
352 face.t[i] = obj.vt.size() + face.t[i];
|
nuclear@6
|
353 }
|
nuclear@6
|
354 }
|
nuclear@6
|
355
|
nuclear@6
|
356 // break quads into triangles if needed
|
nuclear@6
|
357 obj.f.push_back(face);
|
nuclear@6
|
358 if(face.elem == 4) {
|
nuclear@6
|
359 face.v[1] = face.v[2];
|
nuclear@6
|
360 face.n[1] = face.n[2];
|
nuclear@6
|
361 face.t[1] = face.t[2];
|
nuclear@6
|
362
|
nuclear@6
|
363 face.v[2] = face.v[3];
|
nuclear@6
|
364 face.n[2] = face.n[3];
|
nuclear@6
|
365 face.t[2] = face.t[3];
|
nuclear@6
|
366
|
nuclear@6
|
367 obj.f.push_back(face);
|
nuclear@6
|
368 }
|
nuclear@6
|
369 break;
|
nuclear@6
|
370
|
nuclear@6
|
371 default:
|
nuclear@6
|
372 break; // ignore unknown commands
|
nuclear@6
|
373 }
|
nuclear@6
|
374
|
nuclear@6
|
375 prev_cmd = cmd;
|
nuclear@6
|
376 }
|
nuclear@6
|
377
|
nuclear@6
|
378 // reached end of file...
|
nuclear@6
|
379 if(!obj.f.empty()) {
|
nuclear@6
|
380 Mesh *mesh = cons_mesh(&obj);
|
nuclear@6
|
381 mesh->matid = matnames[obj.cur_mat];
|
nuclear@13
|
382 add_mesh(mesh);
|
nuclear@6
|
383 obj_added++;
|
nuclear@6
|
384 }
|
nuclear@6
|
385
|
nuclear@6
|
386 return obj_added > 0;
|
nuclear@6
|
387 }
|
nuclear@6
|
388
|
nuclear@6
|
389 static Mesh *cons_mesh(obj_file *obj)
|
nuclear@6
|
390 {
|
nuclear@6
|
391 Mesh *mesh;
|
nuclear@6
|
392
|
nuclear@6
|
393 // need at least one of each element
|
nuclear@6
|
394 bool added_norm = false, added_tc = false;
|
nuclear@6
|
395 if(obj->vn.empty()) {
|
nuclear@6
|
396 obj->vn.push_back(Vector3(0, 0, 0));
|
nuclear@6
|
397 added_norm = true;
|
nuclear@6
|
398 }
|
nuclear@6
|
399 if(obj->vt.empty()) {
|
nuclear@6
|
400 obj->vt.push_back(Vector3(0, 0, 0));
|
nuclear@6
|
401 added_tc = true;
|
nuclear@6
|
402 }
|
nuclear@6
|
403
|
nuclear@6
|
404 mesh = new Mesh;
|
nuclear@6
|
405
|
nuclear@6
|
406 for(size_t i=0; i<obj->f.size(); i++) {
|
nuclear@6
|
407 Face face;
|
nuclear@6
|
408
|
nuclear@6
|
409 for(int j=0; j<3; j++) {
|
nuclear@6
|
410 obj_face *f = &obj->f[i];
|
nuclear@6
|
411
|
nuclear@6
|
412 face.v[j].pos[0] = obj->v[f->v[j]].x;
|
nuclear@6
|
413 face.v[j].pos[1] = obj->v[f->v[j]].y;
|
nuclear@6
|
414 face.v[j].pos[2] = obj->v[f->v[j]].z;
|
nuclear@13
|
415 face.v[j].pos[3] = 0.0;
|
nuclear@6
|
416
|
nuclear@6
|
417 int nidx = f->n[j] < 0 ? 0 : f->n[j];
|
nuclear@6
|
418 face.v[j].normal[0] = obj->vn[nidx].x;
|
nuclear@6
|
419 face.v[j].normal[1] = obj->vn[nidx].y;
|
nuclear@6
|
420 face.v[j].normal[2] = obj->vn[nidx].z;
|
nuclear@13
|
421 face.v[j].normal[3] = 0.0;
|
nuclear@6
|
422
|
nuclear@6
|
423 int tidx = f->t[j] < 0 ? 0 : f->t[j];
|
nuclear@6
|
424 face.v[j].tex[0] = obj->vt[tidx].x;
|
nuclear@6
|
425 face.v[j].tex[1] = obj->vt[tidx].y;
|
nuclear@6
|
426 }
|
nuclear@13
|
427
|
nuclear@13
|
428 face.normal[0] = face.v[0].normal[0];
|
nuclear@13
|
429 face.normal[1] = face.v[1].normal[1];
|
nuclear@13
|
430 face.normal[2] = face.v[2].normal[2];
|
nuclear@13
|
431 face.normal[3] = 0.0;
|
nuclear@13
|
432
|
nuclear@6
|
433 mesh->faces.push_back(face);
|
nuclear@6
|
434 }
|
nuclear@6
|
435
|
nuclear@6
|
436 if(added_norm) {
|
nuclear@6
|
437 obj->vn.pop_back();
|
nuclear@6
|
438 }
|
nuclear@6
|
439 if(added_tc) {
|
nuclear@6
|
440 obj->vt.pop_back();
|
nuclear@6
|
441 }
|
nuclear@8
|
442
|
nuclear@6
|
443 return mesh;
|
nuclear@6
|
444 }
|
nuclear@6
|
445
|
nuclear@6
|
446 static bool read_materials(FILE *fp, vector<obj_mat> *vmtl)
|
nuclear@6
|
447 {
|
nuclear@6
|
448 obj_mat mat;
|
nuclear@6
|
449
|
nuclear@6
|
450 for(;;) {
|
nuclear@6
|
451 char line[BUF_SZ];
|
nuclear@6
|
452 fgets(line, sizeof line, fp);
|
nuclear@6
|
453 if(feof(fp)) {
|
nuclear@6
|
454 break;
|
nuclear@6
|
455 }
|
nuclear@6
|
456
|
nuclear@6
|
457 char *tok;
|
nuclear@6
|
458 if(!(tok = strtok(line, SEP))) {
|
nuclear@6
|
459 continue;
|
nuclear@6
|
460 }
|
nuclear@6
|
461
|
nuclear@6
|
462 int cmd;
|
nuclear@6
|
463 if((cmd = get_cmd(tok)) == -1) {
|
nuclear@6
|
464 continue;
|
nuclear@6
|
465 }
|
nuclear@6
|
466
|
nuclear@6
|
467 switch(cmd) {
|
nuclear@6
|
468 case CMD_NEWMTL:
|
nuclear@6
|
469 // add the previous material, and start a new one
|
nuclear@6
|
470 if(mat.name.length() > 0) {
|
nuclear@13
|
471 printf("Adding material: %s\n", mat.name.c_str());
|
nuclear@6
|
472 vmtl->push_back(mat);
|
nuclear@6
|
473 mat.reset();
|
nuclear@6
|
474 }
|
nuclear@6
|
475 if((tok = strtok(0, SEP))) {
|
nuclear@6
|
476 mat.name = tok;
|
nuclear@6
|
477 }
|
nuclear@6
|
478 break;
|
nuclear@6
|
479
|
nuclear@6
|
480 case CMD_KA:
|
nuclear@6
|
481 parse_color(&mat.ambient);
|
nuclear@6
|
482 break;
|
nuclear@6
|
483
|
nuclear@6
|
484 case CMD_KD:
|
nuclear@6
|
485 parse_color(&mat.diffuse);
|
nuclear@6
|
486 break;
|
nuclear@6
|
487
|
nuclear@6
|
488 case CMD_KS:
|
nuclear@6
|
489 parse_color(&mat.specular);
|
nuclear@6
|
490 break;
|
nuclear@6
|
491
|
nuclear@16
|
492 case CMD_KR:
|
nuclear@16
|
493 if((tok = strtok(0, SEP)) && is_float(tok)) {
|
nuclear@16
|
494 mat.refl = atof(tok);
|
nuclear@16
|
495 }
|
nuclear@16
|
496 break;
|
nuclear@16
|
497
|
nuclear@6
|
498 case CMD_NS:
|
nuclear@6
|
499 if((tok = strtok(0, SEP)) && is_float(tok)) {
|
nuclear@6
|
500 mat.shininess = atof(tok);
|
nuclear@6
|
501 }
|
nuclear@6
|
502 break;
|
nuclear@6
|
503
|
nuclear@6
|
504 case CMD_NI:
|
nuclear@6
|
505 if((tok = strtok(0, SEP)) && is_float(tok)) {
|
nuclear@6
|
506 mat.ior = atof(tok);
|
nuclear@6
|
507 }
|
nuclear@6
|
508 break;
|
nuclear@6
|
509
|
nuclear@6
|
510 case CMD_D:
|
nuclear@6
|
511 case CMD_TR:
|
nuclear@6
|
512 {
|
nuclear@6
|
513 Vector3 c;
|
nuclear@6
|
514 if(parse_color(&c)) {
|
nuclear@6
|
515 mat.alpha = cmd == CMD_D ? c.x : 1.0 - c.x;
|
nuclear@6
|
516 }
|
nuclear@6
|
517 }
|
nuclear@6
|
518 break;
|
nuclear@6
|
519
|
nuclear@6
|
520 case CMD_MAP_KD:
|
nuclear@6
|
521 mat.tex_dif = parse_map();
|
nuclear@6
|
522 break;
|
nuclear@6
|
523
|
nuclear@6
|
524 default:
|
nuclear@6
|
525 break;
|
nuclear@6
|
526 }
|
nuclear@6
|
527 }
|
nuclear@6
|
528
|
nuclear@6
|
529 if(mat.name.length() > 0) {
|
nuclear@13
|
530 printf("Adding material: %s\n", mat.name.c_str());
|
nuclear@6
|
531 vmtl->push_back(mat);
|
nuclear@6
|
532 }
|
nuclear@6
|
533 return true;
|
nuclear@6
|
534 }
|
nuclear@6
|
535
|
nuclear@6
|
536 static int get_cmd(char *str)
|
nuclear@6
|
537 {
|
nuclear@6
|
538 char *s = str;
|
nuclear@8
|
539 while((*s = toupper(*s))) s++;
|
nuclear@6
|
540
|
nuclear@6
|
541 for(int i=0; cmd_names[i]; i++) {
|
nuclear@6
|
542 if(strcmp(str, cmd_names[i]) == 0) {
|
nuclear@6
|
543 return i;
|
nuclear@6
|
544 }
|
nuclear@6
|
545 }
|
nuclear@6
|
546 return CMD_UNK;
|
nuclear@6
|
547 }
|
nuclear@6
|
548
|
nuclear@6
|
549 static bool is_int(const char *str)
|
nuclear@6
|
550 {
|
nuclear@6
|
551 char *tmp;
|
nuclear@6
|
552 strtol(str, &tmp, 10);
|
nuclear@6
|
553 return tmp != str;
|
nuclear@6
|
554 }
|
nuclear@6
|
555
|
nuclear@6
|
556 static bool is_float(const char *str)
|
nuclear@6
|
557 {
|
nuclear@6
|
558 char *tmp;
|
nuclear@6
|
559 strtod(str, &tmp);
|
nuclear@6
|
560 return tmp != str;
|
nuclear@6
|
561 }
|
nuclear@6
|
562
|
nuclear@6
|
563 static bool parse_vec(Vector3 *vec)
|
nuclear@6
|
564 {
|
nuclear@6
|
565 for(int i=0; i<3; i++) {
|
nuclear@6
|
566 char *tok;
|
nuclear@6
|
567
|
nuclear@6
|
568 if(!(tok = strtok(0, SEP)) || !is_float(tok)) {
|
nuclear@6
|
569 if(i < 2) {
|
nuclear@6
|
570 return false;
|
nuclear@6
|
571 }
|
nuclear@6
|
572 vec->z = 0.0;
|
nuclear@6
|
573 } else {
|
nuclear@6
|
574 float v = atof(tok);
|
nuclear@6
|
575
|
nuclear@6
|
576 switch(i) {
|
nuclear@6
|
577 case 0:
|
nuclear@6
|
578 vec->x = v;
|
nuclear@6
|
579 break;
|
nuclear@6
|
580 case 1:
|
nuclear@6
|
581 vec->y = v;
|
nuclear@6
|
582 break;
|
nuclear@6
|
583 case 2:
|
nuclear@6
|
584 vec->z = v;
|
nuclear@6
|
585 break;
|
nuclear@6
|
586 }
|
nuclear@6
|
587 }
|
nuclear@6
|
588 }
|
nuclear@6
|
589 return true;
|
nuclear@6
|
590 }
|
nuclear@6
|
591
|
nuclear@6
|
592 static bool parse_color(Vector3 *col)
|
nuclear@6
|
593 {
|
nuclear@6
|
594 for(int i=0; i<3; i++) {
|
nuclear@6
|
595 char *tok;
|
nuclear@6
|
596
|
nuclear@6
|
597 if(!(tok = strtok(0, SEP)) || !is_float(tok)) {
|
nuclear@6
|
598 col->y = col->z = col->x;
|
nuclear@6
|
599 return i > 0 ? true : false;
|
nuclear@6
|
600 }
|
nuclear@6
|
601
|
nuclear@6
|
602 float v = atof(tok);
|
nuclear@6
|
603 switch(i) {
|
nuclear@6
|
604 case 0:
|
nuclear@6
|
605 col->x = v;
|
nuclear@6
|
606 break;
|
nuclear@6
|
607 case 1:
|
nuclear@6
|
608 col->y = v;
|
nuclear@6
|
609 break;
|
nuclear@6
|
610 case 2:
|
nuclear@6
|
611 col->z = v;
|
nuclear@6
|
612 break;
|
nuclear@6
|
613 }
|
nuclear@6
|
614 }
|
nuclear@6
|
615 return true;
|
nuclear@6
|
616 }
|
nuclear@6
|
617
|
nuclear@6
|
618 static bool parse_face(obj_face *face)
|
nuclear@6
|
619 {
|
nuclear@6
|
620 char *tok[] = {0, 0, 0, 0};
|
nuclear@6
|
621 face->elem = 0;
|
nuclear@6
|
622
|
nuclear@6
|
623 for(int i=0; i<4; i++) {
|
nuclear@6
|
624 if((!(tok[i] = strtok(0, SEP)) || !is_int(tok[i]))) {
|
nuclear@6
|
625 if(i < 3) return false; // less than 3 verts? not a polygon
|
nuclear@6
|
626 } else {
|
nuclear@6
|
627 face->elem++;
|
nuclear@6
|
628 }
|
nuclear@6
|
629 }
|
nuclear@6
|
630
|
nuclear@6
|
631 for(int i=0; i<4; i++) {
|
nuclear@6
|
632 char *subtok = tok[i];
|
nuclear@6
|
633
|
nuclear@6
|
634 if(!subtok || !*subtok || !is_int(subtok)) {
|
nuclear@6
|
635 if(i < 3) {
|
nuclear@6
|
636 return false;
|
nuclear@6
|
637 }
|
nuclear@6
|
638 face->v[i] = INVALID_IDX;
|
nuclear@6
|
639 } else {
|
nuclear@6
|
640 face->v[i] = atoi(subtok);
|
nuclear@6
|
641 if(face->v[i] > 0) face->v[i]--; /* convert to 0-based */
|
nuclear@6
|
642 }
|
nuclear@6
|
643
|
nuclear@6
|
644 while(subtok && *subtok && *subtok != '/') {
|
nuclear@6
|
645 subtok++;
|
nuclear@6
|
646 }
|
nuclear@6
|
647 if(subtok && *subtok && *++subtok && is_int(subtok)) {
|
nuclear@6
|
648 face->t[i] = atoi(subtok);
|
nuclear@6
|
649 if(face->t[i] > 0) face->t[i]--; /* convert to 0-based */
|
nuclear@6
|
650 } else {
|
nuclear@6
|
651 face->t[i] = INVALID_IDX;
|
nuclear@6
|
652 }
|
nuclear@6
|
653
|
nuclear@6
|
654 while(subtok && *subtok && *subtok != '/') {
|
nuclear@6
|
655 subtok++;
|
nuclear@6
|
656 }
|
nuclear@6
|
657 if(subtok && *subtok && *++subtok && is_int(subtok)) {
|
nuclear@6
|
658 face->n[i] = atoi(subtok);
|
nuclear@6
|
659 if(face->n[i] > 0) face->n[i]--; /* convert to 0-based */
|
nuclear@6
|
660 } else {
|
nuclear@6
|
661 face->n[i] = INVALID_IDX;
|
nuclear@6
|
662 }
|
nuclear@6
|
663 }
|
nuclear@6
|
664
|
nuclear@6
|
665 return true;
|
nuclear@6
|
666 }
|
nuclear@6
|
667
|
nuclear@6
|
668 static const char *parse_map()
|
nuclear@6
|
669 {
|
nuclear@6
|
670 char *tok, *prev = 0;
|
nuclear@6
|
671
|
nuclear@6
|
672 while((tok = strtok(0, SEP))) {
|
nuclear@6
|
673 prev = tok;
|
nuclear@6
|
674 }
|
nuclear@6
|
675
|
nuclear@6
|
676 return prev ? prev : "";
|
nuclear@6
|
677 }
|
nuclear@6
|
678
|
nuclear@6
|
679 static bool find_file(char *res, int sz, const char *fname, const char *path, const char *mode)
|
nuclear@6
|
680 {
|
nuclear@6
|
681 FILE *fp;
|
nuclear@6
|
682 const char *beg, *end;
|
nuclear@6
|
683 int fnamelen = strlen(fname);
|
nuclear@6
|
684
|
nuclear@6
|
685 beg = path;
|
nuclear@6
|
686 while(beg && *beg) {
|
nuclear@6
|
687 end = beg;
|
nuclear@6
|
688 while(*end && *end != ':') {
|
nuclear@6
|
689 end++;
|
nuclear@6
|
690 }
|
nuclear@6
|
691
|
nuclear@13
|
692 int res_len = end - beg;
|
nuclear@13
|
693 char *pathname = (char*)alloca(res_len + fnamelen + 2);
|
nuclear@13
|
694 memcpy(pathname, beg, res_len);
|
nuclear@13
|
695 pathname[res_len] = 0;
|
John@14
|
696 if(res_len) {
|
John@14
|
697 strcat(pathname, "/");
|
John@14
|
698 }
|
nuclear@6
|
699 strcat(pathname, fname);
|
nuclear@6
|
700
|
nuclear@6
|
701 if((fp = fopen(pathname, mode))) {
|
nuclear@6
|
702 fclose(fp);
|
nuclear@6
|
703 strncpy(res, pathname, sz);
|
nuclear@6
|
704 return true;
|
nuclear@6
|
705 }
|
nuclear@6
|
706
|
nuclear@13
|
707 beg += res_len;
|
John@14
|
708 if(*beg == ':') beg++;
|
nuclear@6
|
709 }
|
nuclear@6
|
710 return false;
|
nuclear@6
|
711 }
|
nuclear@6
|
712
|
nuclear@6
|
713 static const char *dirname(const char *str)
|
nuclear@6
|
714 {
|
nuclear@6
|
715 static char buf[PATH_MAX];
|
nuclear@6
|
716
|
nuclear@6
|
717 if(!str || !*str) {
|
nuclear@6
|
718 strcpy(buf, ".");
|
nuclear@6
|
719 } else {
|
nuclear@6
|
720 strncpy(buf, str, PATH_MAX);
|
nuclear@6
|
721 char *ptr = strrchr(buf, '/');
|
nuclear@6
|
722
|
John@14
|
723 if(ptr && *ptr) {
|
John@14
|
724 *ptr = 0;
|
John@14
|
725 } else {
|
John@14
|
726 strcpy(buf, ".");
|
John@14
|
727 }
|
nuclear@6
|
728 }
|
nuclear@6
|
729 return buf;
|
nuclear@6
|
730 }
|