eqemu

annotate src/objfile.cc @ 3:f9274bebe55e

find changesets by author, revision, files, or words in the commit message
adding 3d graphics stuff
author John Tsiombikas <nuclear@member.fsf.org>
date Thu, 17 Jul 2014 02:35:19 +0300
parents
children 3d3656360a82
rev   line source
nuclear@3 1 #include <stdlib.h>
nuclear@3 2 #include <string.h>
nuclear@3 3 #include <ctype.h>
nuclear@3 4 #include <limits.h>
nuclear@3 5 #include <vector>
nuclear@3 6 #include <map>
nuclear@3 7 #include <string>
nuclear@3 8 #include <sstream>
nuclear@3 9 #include <iomanip>
nuclear@3 10 #include "scene.h"
nuclear@3 11
nuclear@3 12 using namespace std;
nuclear@3 13
nuclear@3 14 #define COMMANDS \
nuclear@3 15 CMD(V), \
nuclear@3 16 CMD(VN), \
nuclear@3 17 CMD(VT), \
nuclear@3 18 CMD(F), \
nuclear@3 19 CMD(O), \
nuclear@3 20 CMD(G), \
nuclear@3 21 CMD(MTLLIB), \
nuclear@3 22 CMD(USEMTL), \
nuclear@3 23 CMD(NEWMTL), \
nuclear@3 24 CMD(KA), \
nuclear@3 25 CMD(KD), \
nuclear@3 26 CMD(KS), \
nuclear@3 27 CMD(NS), \
nuclear@3 28 CMD(NI), \
nuclear@3 29 CMD(D), \
nuclear@3 30 CMD(TR), \
nuclear@3 31 CMD(MAP_KD), \
nuclear@3 32 CMD(MAP_KS), \
nuclear@3 33 CMD(MAP_NS), \
nuclear@3 34 CMD(MAP_D), \
nuclear@3 35 CMD(REFL), \
nuclear@3 36 CMD(BUMP)
nuclear@3 37
nuclear@3 38 #define CMD(x) CMD_##x
nuclear@3 39 enum {
nuclear@3 40 COMMANDS,
nuclear@3 41 CMD_UNK
nuclear@3 42 };
nuclear@3 43 #undef CMD
nuclear@3 44
nuclear@3 45 #define CMD(x) #x
nuclear@3 46 static const char *cmd_names[] = {
nuclear@3 47 COMMANDS,
nuclear@3 48 0
nuclear@3 49 };
nuclear@3 50 #undef CMD
nuclear@3 51
nuclear@3 52
nuclear@3 53 struct ObjFace {
nuclear@3 54 int elem;
nuclear@3 55 int v[4], n[4], t[4];
nuclear@3 56 };
nuclear@3 57
nuclear@3 58 struct ObjFile {
nuclear@3 59 string cur_obj, cur_mat;
nuclear@3 60 vector<Vector3> v, vn;
nuclear@3 61 vector<Vector2> vt;
nuclear@3 62 vector<ObjFace> f;
nuclear@3 63 };
nuclear@3 64
nuclear@3 65 typedef Vector3 Color;
nuclear@3 66
nuclear@3 67 struct ObjMat {
nuclear@3 68 string name; // newmtl <name>
nuclear@3 69 Color ambient, diffuse, specular; // Ka, Kd, Ks
nuclear@3 70 float shininess; // Ns
nuclear@3 71 float ior; // Ni
nuclear@3 72 float alpha; // d, Tr
nuclear@3 73
nuclear@3 74 string tex_dif, tex_spec, tex_shin, tex_alpha; // map_Kd, map_Ks, map_Ns, map_d
nuclear@3 75 string tex_refl; // refl -type sphere|cube file
nuclear@3 76 string tex_bump; // bump
nuclear@3 77
nuclear@3 78 ObjMat() { reset(); }
nuclear@3 79
nuclear@3 80 void reset() {
nuclear@3 81 ambient = diffuse = Color(0.5, 0.5, 0.5);
nuclear@3 82 specular = Color(0.0, 0.0, 0.0);
nuclear@3 83 name = tex_dif = tex_spec = tex_shin = tex_alpha = tex_refl = tex_bump = "";
nuclear@3 84 shininess = 0;
nuclear@3 85 ior = alpha = 1;
nuclear@3 86 }
nuclear@3 87 };
nuclear@3 88
nuclear@3 89 static bool read_materials(FILE *fp, vector<ObjMat> *vmtl);
nuclear@3 90 static Object *cons_object(ObjFile *obj);
nuclear@3 91
nuclear@3 92 static int get_cmd(char *str);
nuclear@3 93 static bool is_int(const char *str);
nuclear@3 94 static bool is_float(const char *str);
nuclear@3 95 static bool parse_vec(Vector3 *vec);
nuclear@3 96 static bool parse_color(Color *col);
nuclear@3 97 static bool parse_face(ObjFace *face);
nuclear@3 98 static const char *parse_map();
nuclear@3 99
nuclear@3 100
nuclear@3 101 static map<string, Material> matlib;
nuclear@3 102
nuclear@3 103
nuclear@3 104 #define INVALID_IDX INT_MIN
nuclear@3 105
nuclear@3 106 #define SEP " \t\n\r\v"
nuclear@3 107 #define BUF_SZ 512
nuclear@3 108 bool Scene::load_obj(FILE *fp)
nuclear@3 109 {
nuclear@3 110 static int seq;
nuclear@3 111 char cur_name[16];
nuclear@3 112 stringstream sstr;
nuclear@3 113
nuclear@3 114 ObjFile obj;
nuclear@3 115
nuclear@3 116 sprintf(cur_name, "default%02d.obj", seq++);
nuclear@3 117 obj.cur_obj = cur_name;
nuclear@3 118
nuclear@3 119 int prev_cmd = 0, obj_added = 0;
nuclear@3 120 for(;;) {
nuclear@3 121 Vector3 vec;
nuclear@3 122 ObjFace face;
nuclear@3 123
nuclear@3 124 char line[BUF_SZ];
nuclear@3 125 fgets(line, sizeof line, fp);
nuclear@3 126 if(feof(fp)) {
nuclear@3 127 break;
nuclear@3 128 }
nuclear@3 129
nuclear@3 130 char *tok;
nuclear@3 131 if(!(tok = strtok(line, SEP))) {
nuclear@3 132 continue; // ignore empty lines
nuclear@3 133 }
nuclear@3 134
nuclear@3 135 int cmd;
nuclear@3 136 if((cmd = get_cmd(tok)) == -1) {
nuclear@3 137 continue; // ignore unknown commands ...
nuclear@3 138 }
nuclear@3 139
nuclear@3 140 switch(cmd) {
nuclear@3 141 case CMD_V:
nuclear@3 142 if(!parse_vec(&vec)) {
nuclear@3 143 continue;
nuclear@3 144 }
nuclear@3 145 obj.v.push_back(vec);
nuclear@3 146 break;
nuclear@3 147
nuclear@3 148 case CMD_VN:
nuclear@3 149 if(!parse_vec(&vec)) {
nuclear@3 150 continue;
nuclear@3 151 }
nuclear@3 152 obj.vn.push_back(vec);
nuclear@3 153 break;
nuclear@3 154
nuclear@3 155 case CMD_VT:
nuclear@3 156 if(!parse_vec(&vec)) {
nuclear@3 157 continue;
nuclear@3 158 }
nuclear@3 159 vec.y = 1.0 - vec.y;
nuclear@3 160 obj.vt.push_back(Vector2(vec.x, vec.y));
nuclear@3 161 break;
nuclear@3 162
nuclear@3 163 case CMD_O:
nuclear@3 164 case CMD_G:
nuclear@3 165 if(prev_cmd == CMD_O || prev_cmd == CMD_G) {
nuclear@3 166 break; // just in case we've got both of them in a row
nuclear@3 167 }
nuclear@3 168 /* if we have any previous data, group them up, add the object
nuclear@3 169 * and continue with the new one...
nuclear@3 170 */
nuclear@3 171 if(!obj.f.empty()) {
nuclear@3 172 Object *robj = cons_object(&obj);
nuclear@3 173 robj->mtl = matlib[obj.cur_mat];
nuclear@3 174 add_object(robj);
nuclear@3 175 obj_added++;
nuclear@3 176
nuclear@3 177 obj.f.clear(); // clean the face list
nuclear@3 178 }
nuclear@3 179 if((tok = strtok(0, SEP))) {
nuclear@3 180 obj.cur_obj = tok;
nuclear@3 181 } else {
nuclear@3 182 sprintf(cur_name, "default%02d.obj", seq++);
nuclear@3 183 obj.cur_obj = cur_name;
nuclear@3 184 }
nuclear@3 185 break;
nuclear@3 186
nuclear@3 187 case CMD_MTLLIB:
nuclear@3 188 if((tok = strtok(0, SEP))) {
nuclear@3 189 FILE *mfile;
nuclear@3 190 if(!(mfile = fopen(tok, "rb"))) {
nuclear@3 191 fprintf(stderr, "failed to open material library: %s\n", tok);
nuclear@3 192 continue;
nuclear@3 193 }
nuclear@3 194
nuclear@3 195 // load all materials of the mtl file into a vector
nuclear@3 196 vector<ObjMat> vmtl;
nuclear@3 197 if(!read_materials(mfile, &vmtl)) {
nuclear@3 198 continue;
nuclear@3 199 }
nuclear@3 200 fclose(mfile);
nuclear@3 201
nuclear@3 202 // and add them all to the scene
nuclear@3 203 for(size_t i=0; i<vmtl.size(); i++) {
nuclear@3 204 Material mat;
nuclear@3 205 mat.ambient = vmtl[i].ambient;
nuclear@3 206 mat.diffuse = vmtl[i].diffuse;
nuclear@3 207 mat.specular = vmtl[i].specular;
nuclear@3 208 mat.shininess = vmtl[i].shininess;
nuclear@3 209 mat.alpha = vmtl[i].alpha;
nuclear@3 210
nuclear@3 211 /*if(vmtl[i].tex_dif.length()) {
nuclear@3 212 mat.set_texture(get_texture(vmtl[i].tex_dif.c_str()));
nuclear@3 213 }*/
nuclear@3 214
nuclear@3 215 matlib[vmtl[i].name] = mat;
nuclear@3 216 }
nuclear@3 217 }
nuclear@3 218 break;
nuclear@3 219
nuclear@3 220 case CMD_USEMTL:
nuclear@3 221 if((tok = strtok(0, SEP))) {
nuclear@3 222 obj.cur_mat = tok;
nuclear@3 223 } else {
nuclear@3 224 obj.cur_mat = "";
nuclear@3 225 }
nuclear@3 226 break;
nuclear@3 227
nuclear@3 228 case CMD_F:
nuclear@3 229 if(!parse_face(&face)) {
nuclear@3 230 continue;
nuclear@3 231 }
nuclear@3 232
nuclear@3 233 // convert negative indices to regular indices
nuclear@3 234 for(int i=0; i<4; i++) {
nuclear@3 235 if(face.v[i] < 0 && face.v[i] != INVALID_IDX) {
nuclear@3 236 face.v[i] = obj.v.size() + face.v[i];
nuclear@3 237 }
nuclear@3 238 if(face.n[i] < 0 && face.n[i] != INVALID_IDX) {
nuclear@3 239 face.n[i] = obj.vn.size() + face.n[i];
nuclear@3 240 }
nuclear@3 241 if(face.t[i] < 0 && face.t[i] != INVALID_IDX) {
nuclear@3 242 face.t[i] = obj.vt.size() + face.t[i];
nuclear@3 243 }
nuclear@3 244 }
nuclear@3 245
nuclear@3 246 // break quads into triangles if needed
nuclear@3 247 obj.f.push_back(face);
nuclear@3 248 if(face.elem == 4) {
nuclear@3 249 face.v[1] = face.v[2];
nuclear@3 250 face.n[1] = face.n[2];
nuclear@3 251 face.t[1] = face.t[2];
nuclear@3 252
nuclear@3 253 face.v[2] = face.v[3];
nuclear@3 254 face.n[2] = face.n[3];
nuclear@3 255 face.t[2] = face.t[3];
nuclear@3 256
nuclear@3 257 obj.f.push_back(face);
nuclear@3 258 }
nuclear@3 259 break;
nuclear@3 260
nuclear@3 261 default:
nuclear@3 262 break; // ignore unknown commands
nuclear@3 263 }
nuclear@3 264
nuclear@3 265 prev_cmd = cmd;
nuclear@3 266 }
nuclear@3 267
nuclear@3 268 // reached end of file...
nuclear@3 269 if(!obj.f.empty()) {
nuclear@3 270 Object *robj = cons_object(&obj);
nuclear@3 271 robj->mtl = matlib[obj.cur_mat];
nuclear@3 272 add_object(robj);
nuclear@3 273 obj_added++;
nuclear@3 274 }
nuclear@3 275
nuclear@3 276 return obj_added > 0;
nuclear@3 277 }
nuclear@3 278
nuclear@3 279 static Object *cons_object(ObjFile *obj)
nuclear@3 280 {
nuclear@3 281 Object *robj;
nuclear@3 282 Vector3 *varr, *narr;
nuclear@3 283 Vector2 *tarr;
nuclear@3 284
nuclear@3 285 int nelem = obj->f.size() * 3;
nuclear@3 286
nuclear@3 287 try {
nuclear@3 288 robj = new Object;
nuclear@3 289 varr = new Vector3[nelem];
nuclear@3 290 narr = new Vector3[nelem];
nuclear@3 291 tarr = new Vector2[nelem];
nuclear@3 292 }
nuclear@3 293 catch(...) {
nuclear@3 294 return 0;
nuclear@3 295 }
nuclear@3 296 /*if(obj->cur_obj.length() > 0) {
nuclear@3 297 robj->set_name(obj->cur_obj.c_str());
nuclear@3 298 }*/
nuclear@3 299
nuclear@3 300 // need at least one of each element
nuclear@3 301 bool added_norm = false, added_tc = false;
nuclear@3 302 if(obj->vn.empty()) {
nuclear@3 303 obj->vn.push_back(Vector3(0, 0, 0));
nuclear@3 304 added_norm = true;;
nuclear@3 305 }
nuclear@3 306 if(obj->vt.empty()) {
nuclear@3 307 obj->vt.push_back(Vector2(0, 0));
nuclear@3 308 added_tc = true;
nuclear@3 309 }
nuclear@3 310
nuclear@3 311 for(size_t i=0; i<obj->f.size(); i++) {
nuclear@3 312 for(int j=0; j<3; j++) {
nuclear@3 313 int idx = i * 3 + j;
nuclear@3 314 ObjFace *f = &obj->f[i];
nuclear@3 315
nuclear@3 316 varr[idx] = obj->v[f->v[j]];
nuclear@3 317 narr[idx] = obj->vn[f->n[j] < 0 ? 0 : f->n[j]];
nuclear@3 318
nuclear@3 319 float t = obj->vt[f->t[j] < 0 ? 0 : f->t[j]].x;
nuclear@3 320 float s = obj->vt[f->t[j] < 0 ? 0 : f->t[j]].y;
nuclear@3 321 tarr[idx] = Vector2(t, s);
nuclear@3 322 }
nuclear@3 323 }
nuclear@3 324
nuclear@3 325 if(added_norm) {
nuclear@3 326 obj->vn.pop_back();
nuclear@3 327 }
nuclear@3 328 if(added_tc) {
nuclear@3 329 obj->vt.pop_back();
nuclear@3 330 }
nuclear@3 331
nuclear@3 332 Mesh *mesh = robj->get_mesh();
nuclear@3 333 mesh->set_attrib(MESH_ATTR_VERTEX, nelem, 3, &varr->x);
nuclear@3 334 mesh->set_attrib(MESH_ATTR_NORMAL, nelem, 3, &narr->x);
nuclear@3 335 mesh->set_attrib(MESH_ATTR_TEXCOORD, nelem, 2, &tarr->x);
nuclear@3 336
nuclear@3 337 delete [] varr;
nuclear@3 338 delete [] narr;
nuclear@3 339 delete [] tarr;
nuclear@3 340 return robj;
nuclear@3 341 }
nuclear@3 342
nuclear@3 343 static bool read_materials(FILE *fp, vector<ObjMat> *vmtl)
nuclear@3 344 {
nuclear@3 345 ObjMat mat;
nuclear@3 346
nuclear@3 347 for(;;) {
nuclear@3 348 char line[BUF_SZ];
nuclear@3 349 fgets(line, sizeof line, fp);
nuclear@3 350 if(feof(fp)) {
nuclear@3 351 break;
nuclear@3 352 }
nuclear@3 353
nuclear@3 354 char *tok;
nuclear@3 355 if(!(tok = strtok(line, SEP))) {
nuclear@3 356 continue;
nuclear@3 357 }
nuclear@3 358
nuclear@3 359 int cmd;
nuclear@3 360 if((cmd = get_cmd(tok)) == -1) {
nuclear@3 361 continue;
nuclear@3 362 }
nuclear@3 363
nuclear@3 364 switch(cmd) {
nuclear@3 365 case CMD_NEWMTL:
nuclear@3 366 // add the previous material, and start a new one
nuclear@3 367 if(mat.name.length() > 0) {
nuclear@3 368 vmtl->push_back(mat);
nuclear@3 369 mat.reset();
nuclear@3 370 }
nuclear@3 371 if((tok = strtok(0, SEP))) {
nuclear@3 372 mat.name = tok;
nuclear@3 373 }
nuclear@3 374 break;
nuclear@3 375
nuclear@3 376 case CMD_KA:
nuclear@3 377 parse_color(&mat.ambient);
nuclear@3 378 break;
nuclear@3 379
nuclear@3 380 case CMD_KD:
nuclear@3 381 parse_color(&mat.diffuse);
nuclear@3 382 break;
nuclear@3 383
nuclear@3 384 case CMD_KS:
nuclear@3 385 parse_color(&mat.specular);
nuclear@3 386 break;
nuclear@3 387
nuclear@3 388 case CMD_NS:
nuclear@3 389 if((tok = strtok(0, SEP)) && is_float(tok)) {
nuclear@3 390 mat.shininess = atof(tok);
nuclear@3 391 }
nuclear@3 392 break;
nuclear@3 393
nuclear@3 394 case CMD_NI:
nuclear@3 395 if((tok = strtok(0, SEP)) && is_float(tok)) {
nuclear@3 396 mat.ior = atof(tok);
nuclear@3 397 }
nuclear@3 398 break;
nuclear@3 399
nuclear@3 400 case CMD_D:
nuclear@3 401 case CMD_TR:
nuclear@3 402 {
nuclear@3 403 Color c;
nuclear@3 404 if(parse_color(&c)) {
nuclear@3 405 mat.alpha = cmd == CMD_D ? c.x : 1.0 - c.x;
nuclear@3 406 }
nuclear@3 407 }
nuclear@3 408 break;
nuclear@3 409
nuclear@3 410 case CMD_MAP_KD:
nuclear@3 411 mat.tex_dif = parse_map();
nuclear@3 412 break;
nuclear@3 413
nuclear@3 414 default:
nuclear@3 415 break;
nuclear@3 416 }
nuclear@3 417 }
nuclear@3 418
nuclear@3 419 if(mat.name.length() > 0) {
nuclear@3 420 vmtl->push_back(mat);
nuclear@3 421 }
nuclear@3 422 return true;
nuclear@3 423 }
nuclear@3 424
nuclear@3 425 static int get_cmd(char *str)
nuclear@3 426 {
nuclear@3 427 char *s = str;
nuclear@3 428 while((*s = toupper(*s))) s++;
nuclear@3 429
nuclear@3 430 for(int i=0; cmd_names[i]; i++) {
nuclear@3 431 if(strcmp(str, cmd_names[i]) == 0) {
nuclear@3 432 return i;
nuclear@3 433 }
nuclear@3 434 }
nuclear@3 435 return CMD_UNK;
nuclear@3 436 }
nuclear@3 437
nuclear@3 438 static bool is_int(const char *str)
nuclear@3 439 {
nuclear@3 440 char *tmp;
nuclear@3 441 strtol(str, &tmp, 10);
nuclear@3 442 return tmp != str;
nuclear@3 443 }
nuclear@3 444
nuclear@3 445 static bool is_float(const char *str)
nuclear@3 446 {
nuclear@3 447 char *tmp;
nuclear@3 448 strtod(str, &tmp);
nuclear@3 449 return tmp != str;
nuclear@3 450 }
nuclear@3 451
nuclear@3 452 static bool parse_vec(Vector3 *vec)
nuclear@3 453 {
nuclear@3 454 for(int i=0; i<3; i++) {
nuclear@3 455 char *tok;
nuclear@3 456
nuclear@3 457 if(!(tok = strtok(0, SEP)) || !is_float(tok)) {
nuclear@3 458 if(i < 2) {
nuclear@3 459 return false;
nuclear@3 460 }
nuclear@3 461 vec->z = 0.0;
nuclear@3 462 } else {
nuclear@3 463 (*vec)[i] = atof(tok);
nuclear@3 464 }
nuclear@3 465 }
nuclear@3 466 return true;
nuclear@3 467 }
nuclear@3 468
nuclear@3 469 static bool parse_color(Color *col)
nuclear@3 470 {
nuclear@3 471 for(int i=0; i<3; i++) {
nuclear@3 472 char *tok;
nuclear@3 473
nuclear@3 474 if(!(tok = strtok(0, SEP)) || !is_float(tok)) {
nuclear@3 475 col->y = col->z = col->x;
nuclear@3 476 return i > 0 ? true : false;
nuclear@3 477 }
nuclear@3 478 (*col)[i] = atof(tok);
nuclear@3 479 }
nuclear@3 480 return true;
nuclear@3 481 }
nuclear@3 482
nuclear@3 483 static bool parse_face(ObjFace *face)
nuclear@3 484 {
nuclear@3 485 char *tok[] = {0, 0, 0, 0};
nuclear@3 486 face->elem = 0;
nuclear@3 487
nuclear@3 488 for(int i=0; i<4; i++) {
nuclear@3 489 if((!(tok[i] = strtok(0, SEP)) || !is_int(tok[i]))) {
nuclear@3 490 if(i < 3) return false; // less than 3 verts? not a polygon
nuclear@3 491 } else {
nuclear@3 492 face->elem++;
nuclear@3 493 }
nuclear@3 494 }
nuclear@3 495
nuclear@3 496 for(int i=0; i<4; i++) {
nuclear@3 497 char *subtok = tok[i];
nuclear@3 498
nuclear@3 499 if(!subtok || !*subtok || !is_int(subtok)) {
nuclear@3 500 if(i < 3) {
nuclear@3 501 return false;
nuclear@3 502 }
nuclear@3 503 face->v[i] = INVALID_IDX;
nuclear@3 504 } else {
nuclear@3 505 face->v[i] = atoi(subtok);
nuclear@3 506 if(face->v[i] > 0) face->v[i]--; /* convert to 0-based */
nuclear@3 507 }
nuclear@3 508
nuclear@3 509 while(subtok && *subtok && *subtok != '/') {
nuclear@3 510 subtok++;
nuclear@3 511 }
nuclear@3 512 if(subtok && *subtok && *++subtok && is_int(subtok)) {
nuclear@3 513 face->t[i] = atoi(subtok);
nuclear@3 514 if(face->t[i] > 0) face->t[i]--; /* convert to 0-based */
nuclear@3 515 } else {
nuclear@3 516 face->t[i] = INVALID_IDX;
nuclear@3 517 }
nuclear@3 518
nuclear@3 519 while(subtok && *subtok && *subtok != '/') {
nuclear@3 520 subtok++;
nuclear@3 521 }
nuclear@3 522 if(subtok && *subtok && *++subtok && is_int(subtok)) {
nuclear@3 523 face->n[i] = atoi(subtok);
nuclear@3 524 if(face->n[i] > 0) face->n[i]--; /* convert to 0-based */
nuclear@3 525 } else {
nuclear@3 526 face->n[i] = INVALID_IDX;
nuclear@3 527 }
nuclear@3 528 }
nuclear@3 529
nuclear@3 530 return true;
nuclear@3 531 }
nuclear@3 532
nuclear@3 533 static const char *parse_map()
nuclear@3 534 {
nuclear@3 535 char *tok, *prev = 0;
nuclear@3 536
nuclear@3 537 while((tok = strtok(0, SEP))) {
nuclear@3 538 prev = tok;
nuclear@3 539 }
nuclear@3 540
nuclear@3 541 return prev ? prev : "";
nuclear@3 542 }