erebus

annotate liberebus/src/scene.cc @ 42:b9294cd6b9dc

find changesets by author, revision, files, or words in the commit message
console input history with up/down
author John Tsiombikas <nuclear@member.fsf.org>
date Tue, 10 Jun 2014 16:15:08 +0300
parents 4a0a288ffb27
children ed18af9da8f7
rev   line source
nuclear@19 1 #include <stdio.h>
nuclear@19 2 #include <stdlib.h>
nuclear@19 3 #include <string.h>
nuclear@4 4 #include "scene.h"
nuclear@19 5 #include "snode.h"
nuclear@19 6 #include "object.h"
nuclear@19 7 #include "geomobj.h"
nuclear@4 8
nuclear@8 9 // default enviromental parameters
nuclear@8 10 Environment::Environment()
nuclear@19 11 : bgcolor(0.01, 0.01, 0.01), ambient(0.01, 0.01, 0.01)
nuclear@8 12 {
nuclear@8 13 }
nuclear@8 14
nuclear@4 15 Scene::Scene()
nuclear@4 16 {
nuclear@4 17 active_cam = 0;
nuclear@4 18 root = new SceneNode;
nuclear@4 19 }
nuclear@4 20
nuclear@4 21 Scene::~Scene()
nuclear@4 22 {
nuclear@40 23 clear();
nuclear@40 24 }
nuclear@40 25
nuclear@40 26 void Scene::clear()
nuclear@40 27 {
nuclear@4 28 for(auto obj : objects) {
nuclear@4 29 delete obj;
nuclear@4 30 }
nuclear@40 31 objects.clear();
nuclear@40 32
nuclear@4 33 for(auto node : nodes) {
nuclear@4 34 delete node;
nuclear@4 35 }
nuclear@40 36 nodes.clear();
nuclear@40 37
nuclear@4 38 delete root;
nuclear@40 39 root = 0;
nuclear@40 40
nuclear@40 41 active_cam = 0;
nuclear@4 42 }
nuclear@4 43
nuclear@8 44 void Scene::set_env(const Environment &env)
nuclear@8 45 {
nuclear@8 46 this->env = env;
nuclear@8 47 }
nuclear@8 48
nuclear@8 49 Environment &Scene::get_env()
nuclear@8 50 {
nuclear@8 51 return env;
nuclear@8 52 }
nuclear@8 53
nuclear@8 54 const Environment &Scene::get_env() const
nuclear@8 55 {
nuclear@8 56 return env;
nuclear@8 57 }
nuclear@8 58
nuclear@8 59 Color Scene::get_env_color() const
nuclear@8 60 {
nuclear@8 61 return env.bgcolor;
nuclear@8 62 }
nuclear@8 63
nuclear@8 64 Color Scene::get_env_color(const Ray &ray) const
nuclear@8 65 {
nuclear@8 66 // TODO
nuclear@8 67 return get_env_color();
nuclear@8 68 }
nuclear@8 69
nuclear@4 70 void Scene::add_object(Object *obj)
nuclear@4 71 {
nuclear@4 72 objects.push_back(obj);
nuclear@4 73 }
nuclear@4 74
nuclear@4 75 int Scene::get_object_count() const
nuclear@4 76 {
nuclear@4 77 return (int)objects.size();
nuclear@4 78 }
nuclear@4 79
nuclear@4 80 Object *Scene::get_object(int idx) const
nuclear@4 81 {
nuclear@4 82 return objects[idx];
nuclear@4 83 }
nuclear@4 84
nuclear@4 85 void Scene::add_node(SceneNode *node)
nuclear@4 86 {
nuclear@4 87 nodes.push_back(node);
nuclear@4 88
nuclear@4 89 if(!node->get_parent()) {
nuclear@4 90 root->add_child(node);
nuclear@4 91 }
nuclear@4 92 }
nuclear@4 93
nuclear@4 94 int Scene::get_node_count() const
nuclear@4 95 {
nuclear@4 96 return (int)nodes.size();
nuclear@4 97 }
nuclear@4 98
nuclear@4 99 SceneNode *Scene::get_node(int idx) const
nuclear@4 100 {
nuclear@4 101 return nodes[idx];
nuclear@4 102 }
nuclear@4 103
nuclear@4 104 void Scene::use_camera(Camera *cam)
nuclear@4 105 {
nuclear@4 106 active_cam = cam;
nuclear@4 107 }
nuclear@4 108
nuclear@4 109 Camera *Scene::get_active_camera() const
nuclear@4 110 {
nuclear@4 111 return active_cam;
nuclear@4 112 }
nuclear@4 113
nuclear@4 114 void Scene::update(long msec)
nuclear@4 115 {
nuclear@4 116 root->update(msec);
nuclear@4 117 }
nuclear@4 118
nuclear@4 119 bool Scene::intersect(const Ray &ray, RayHit *hit) const
nuclear@4 120 {
nuclear@4 121 return root->intersect(ray, hit);
nuclear@4 122 }
nuclear@4 123
nuclear@4 124 bool Scene::load(const char *fname)
nuclear@4 125 {
nuclear@19 126 FILE *fp = fopen(fname, "r");
nuclear@19 127 if(!fp) {
nuclear@19 128 fprintf(stderr, "failed to open scene file: %s\n", fname);
nuclear@19 129 return false;
nuclear@19 130 }
nuclear@19 131
nuclear@19 132 bool res = load(fp);
nuclear@19 133 fclose(fp);
nuclear@19 134 return res;
nuclear@4 135 }
nuclear@19 136
nuclear@19 137 static char *strip_wspace(char *str)
nuclear@19 138 {
nuclear@19 139 while(*str && isspace(*str)) str++;
nuclear@23 140
nuclear@23 141 char *comment = strchr(str, '#');
nuclear@23 142 if(comment) *comment = 0;
nuclear@23 143
nuclear@23 144 if(str) {
nuclear@23 145 char *end = str + strlen(str) - 1;
nuclear@23 146 while(end > str && isspace(*end)) end--;
nuclear@23 147 end[1] = 0;
nuclear@23 148 }
nuclear@23 149
nuclear@19 150 return str;
nuclear@19 151 }
nuclear@19 152
nuclear@19 153 bool Scene::load(FILE *fp)
nuclear@19 154 {
nuclear@28 155 static char buf[4096];
nuclear@28 156 int offset = 0;
nuclear@19 157
nuclear@28 158 while(fgets(buf + offset, sizeof buf - offset, fp)) {
nuclear@28 159 // handle line continuations
nuclear@28 160 int len = strlen(buf);
nuclear@28 161 if(strcmp(buf + len - 2, "\\\n") == 0 || strcmp(buf + len - 2, "\\\r") == 0) {
nuclear@28 162 offset = len - 2;
nuclear@28 163 continue;
nuclear@28 164 }
nuclear@28 165 if(strcmp(buf + len - 3, "\\\r\n") == 0) {
nuclear@28 166 offset = len - 3;
nuclear@28 167 continue;
nuclear@28 168 }
nuclear@28 169 offset = 0;
nuclear@28 170
nuclear@19 171 char *line = strip_wspace(buf);
nuclear@19 172 if(!line || !*line || *line == '#') {
nuclear@19 173 continue;
nuclear@19 174 }
nuclear@19 175
nuclear@19 176 std::vector<char*> cmd;
nuclear@19 177
nuclear@19 178 char *tok = 0;
nuclear@19 179 while((tok = strtok(tok ? 0 : line, " \t\v\n\r"))) {
nuclear@19 180 cmd.push_back(tok);
nuclear@19 181 }
nuclear@19 182 cmd.push_back(0);
nuclear@19 183
nuclear@19 184
nuclear@19 185 if(!proc_cmd((int)cmd.size() - 1, &cmd[0])) {
nuclear@19 186 return false;
nuclear@19 187 }
nuclear@19 188 }
nuclear@19 189
nuclear@19 190 return true;
nuclear@19 191 }
nuclear@19 192
nuclear@19 193 static int parse_vec(char **argv, float *vec)
nuclear@19 194 {
nuclear@19 195 int nelem = 0;
nuclear@19 196 char *endp;
nuclear@19 197
nuclear@19 198 for(int i=0; i<4; i++) {
nuclear@19 199 if(!argv[i]) break;
nuclear@19 200
nuclear@19 201 vec[nelem] = strtod(argv[i], &endp);
nuclear@19 202 if(endp == argv[i]) {
nuclear@19 203 break;
nuclear@19 204 }
nuclear@19 205 nelem++;
nuclear@19 206 }
nuclear@19 207 return nelem;
nuclear@19 208 }
nuclear@19 209
nuclear@22 210 static int parse_brdf(char **argv, Reflectance **brdf_ptr)
nuclear@22 211 {
nuclear@22 212 CompositeRefl *composite = 0;
nuclear@22 213 Reflectance *brdf = 0;
nuclear@22 214 float weight = 1.0;
nuclear@22 215 int i;
nuclear@22 216
nuclear@22 217 for(i=0; argv[i]; i++) {
nuclear@22 218 if(strcmp(argv[i], "-brdf") == 0) {
nuclear@22 219 if(strcmp(argv[++i], "lambert") == 0) {
nuclear@22 220 if(composite && brdf) {
nuclear@22 221 composite->add_brdf(brdf, weight);
nuclear@22 222 }
nuclear@22 223 brdf = new LambertRefl;
nuclear@22 224 if(!composite) break;
nuclear@22 225
nuclear@28 226 } else if(strcmp(argv[i], "phong") == 0) {
nuclear@28 227 if(composite && brdf) {
nuclear@28 228 composite->add_brdf(brdf, weight);
nuclear@28 229 }
nuclear@28 230 brdf = new PhongRefl;
nuclear@28 231 if(!composite) break;
nuclear@28 232
nuclear@22 233 } else if(strcmp(argv[i], "mirror") == 0) {
nuclear@22 234 if(composite && brdf) {
nuclear@22 235 composite->add_brdf(brdf, weight);
nuclear@22 236 }
nuclear@22 237 brdf = new MirrorRefl;
nuclear@22 238 if(!composite) break;
nuclear@22 239
nuclear@22 240 } else if(strcmp(argv[i], "composite") == 0) {
nuclear@22 241 if(composite) {
nuclear@22 242 fprintf(stderr, "can't have nested composite BRDFs\n");
nuclear@22 243 return -1;
nuclear@22 244 }
nuclear@22 245 composite = new CompositeRefl;
nuclear@22 246
nuclear@22 247 } else {
nuclear@22 248 --i;
nuclear@22 249 break;
nuclear@22 250 }
nuclear@22 251
nuclear@22 252 } else if(strcmp(argv[i], "-brdf-weight") == 0) {
nuclear@22 253 if(!composite) {
nuclear@22 254 fprintf(stderr, "-brdf-weight is invalid without a composite brdf\n");
nuclear@22 255 return -1;
nuclear@22 256 }
nuclear@22 257 char *endp;
nuclear@22 258 weight = strtod(argv[++i], &endp);
nuclear@22 259 if(endp == argv[i]) {
nuclear@22 260 fprintf(stderr, "-brdf-weight must be followed by a number\n");
nuclear@22 261 return -1;
nuclear@22 262 }
nuclear@22 263 } else {
nuclear@22 264 --i;
nuclear@22 265 break;
nuclear@22 266 }
nuclear@22 267 }
nuclear@22 268
nuclear@22 269 if(composite && brdf) {
nuclear@22 270 composite->add_brdf(brdf, weight);
nuclear@22 271 *brdf_ptr = composite;
nuclear@22 272 } else {
nuclear@22 273 *brdf_ptr = brdf;
nuclear@22 274 }
nuclear@22 275 return i;
nuclear@22 276 }
nuclear@22 277
nuclear@19 278 static SceneNode *proc_object(Object *obj, int argc, char **argv)
nuclear@19 279 {
nuclear@19 280 float vec[4];
nuclear@19 281 SceneNode *node = new SceneNode(obj);
nuclear@19 282 GeomObject *gobj = (GeomObject*)obj;
nuclear@19 283
nuclear@19 284 for(int i=1; i<argc; i++) {
nuclear@19 285 if(strcmp(argv[i], "-position") == 0) {
nuclear@19 286 int nelem = parse_vec(argv + i + 1, vec);
nuclear@19 287 if(nelem < 3) {
nuclear@19 288 fprintf(stderr, "-position must be followed by 3 numbers\n");
nuclear@19 289 goto err;
nuclear@19 290 }
nuclear@19 291 node->set_position(Vector3(vec[0], vec[1], vec[2]));
nuclear@19 292 i += 3;
nuclear@19 293
nuclear@19 294 } else if(strcmp(argv[i], "-rotation") == 0) {
nuclear@19 295 int nelem = parse_vec(argv + i + 1, vec);
nuclear@19 296 if(nelem < 4) {
nuclear@19 297 fprintf(stderr, "-rotation must be followed by axis vector and angle\n");
nuclear@19 298 goto err;
nuclear@19 299 }
nuclear@19 300 node->set_rotation(Quaternion(Vector3(vec[0], vec[1], vec[2]), vec[3]));
nuclear@19 301 i += 4;
nuclear@19 302
nuclear@19 303 } else if(strcmp(argv[i], "-scaling") == 0) {
nuclear@19 304 int nelem = parse_vec(argv + i + 1, vec);
nuclear@19 305 Vector3 s;
nuclear@19 306 if(nelem == 1) {
nuclear@19 307 s.x = s.y = s.z = vec[0];
nuclear@19 308 } else if(nelem == 3) {
nuclear@19 309 s = Vector3(vec[0], vec[1], vec[2]);
nuclear@19 310 } else {
nuclear@19 311 fprintf(stderr, "-scaling must be followed by 1 or 3 numbers\n");
nuclear@19 312 goto err;
nuclear@19 313 }
nuclear@19 314 node->set_scaling(s);
nuclear@19 315 i += nelem;
nuclear@19 316
nuclear@19 317 } else if(strcmp(argv[i], "-diffuse") == 0) {
nuclear@19 318 int nelem = parse_vec(argv + i + 1, vec);
nuclear@19 319 if(nelem < 3) {
nuclear@19 320 fprintf(stderr, "-diffuse must be followed by a color (r g b)\n");
nuclear@19 321 goto err;
nuclear@19 322 }
nuclear@19 323 gobj->mtl.set_attrib("diffuse", Color(vec[0], vec[1], vec[2]));
nuclear@19 324 i += 3;
nuclear@19 325
nuclear@19 326 } else if(strcmp(argv[i], "-specular") == 0) {
nuclear@19 327 int nelem = parse_vec(argv + i + 1, vec);
nuclear@19 328 if(nelem < 3) {
nuclear@19 329 fprintf(stderr, "-specular must be followed by a color\n");
nuclear@19 330 goto err;
nuclear@19 331 }
nuclear@19 332 gobj->mtl.set_attrib("specular", Color(vec[0], vec[1], vec[2]));
nuclear@19 333 i += 3;
nuclear@19 334
nuclear@19 335 } else if(strcmp(argv[i], "-emissive") == 0) {
nuclear@19 336 int nelem = parse_vec(argv + i + 1, vec);
nuclear@19 337 Color emissive;
nuclear@19 338 if(nelem == 1) {
nuclear@19 339 emissive = Vector3(1, 1, 1);
nuclear@19 340 } else if(nelem == 3) {
nuclear@19 341 emissive = Vector3(vec[0], vec[1], vec[2]);
nuclear@19 342 } else {
nuclear@19 343 fprintf(stderr, "-emissive must be followed by an intensity or a color\n");
nuclear@19 344 goto err;
nuclear@19 345 }
nuclear@19 346 i += nelem;
nuclear@19 347 gobj->mtl.set_attrib("emissive", emissive);
nuclear@19 348
nuclear@22 349 } else if(strcmp(argv[i], "-brdf") == 0) {
nuclear@22 350 int nelem = parse_brdf(argv + i, &gobj->brdf);
nuclear@22 351 if(nelem == -1) {
nuclear@22 352 goto err;
nuclear@22 353 }
nuclear@22 354 i += nelem;
nuclear@22 355
nuclear@19 356 } else {
nuclear@19 357 fprintf(stderr, "unrecognized %s option: %s\n", argv[0], argv[i]);
nuclear@19 358 goto err;
nuclear@19 359 }
nuclear@19 360 }
nuclear@19 361
nuclear@19 362 return node;
nuclear@19 363 err:
nuclear@19 364 delete node;
nuclear@19 365 return 0;
nuclear@19 366 }
nuclear@19 367
nuclear@19 368 static Camera *proc_camera(int argc, char **argv)
nuclear@19 369 {
nuclear@19 370 float vec[4];
nuclear@19 371 TargetCamera *cam = new TargetCamera;
nuclear@19 372
nuclear@19 373 for(int i=1; i<argc; i++) {
nuclear@19 374 if(strcmp(argv[i], "-position") == 0) {
nuclear@19 375 int nelem = parse_vec(argv + i + 1, vec);
nuclear@19 376 if(nelem < 3) {
nuclear@19 377 fprintf(stderr, "-position must be followed by 3 numbers\n");
nuclear@19 378 goto err;
nuclear@19 379 }
nuclear@19 380 cam->set_position(Vector3(vec[0], vec[1], vec[2]));
nuclear@19 381 i += 3;
nuclear@19 382
nuclear@19 383 } else if(strcmp(argv[i], "-target") == 0) {
nuclear@19 384 int nelem = parse_vec(argv + i + 1, vec);
nuclear@19 385 if(nelem < 3) {
nuclear@19 386 fprintf(stderr, "-target must be followed by 3 numbers\n");
nuclear@19 387 goto err;
nuclear@19 388 }
nuclear@19 389 cam->set_target(Vector3(vec[0], vec[1], vec[2]));
nuclear@19 390 i += 3;
nuclear@19 391
nuclear@19 392 } else if(strcmp(argv[i], "-fov") == 0) {
nuclear@19 393 int nelem = parse_vec(argv + i + 1, vec);
nuclear@19 394 if(nelem < 1) {
nuclear@19 395 fprintf(stderr, "-fov must be followed by the field of view in degrees\n");
nuclear@19 396 goto err;
nuclear@19 397 }
nuclear@19 398 cam->set_fov(vec[0]);
nuclear@19 399 i += 1;
nuclear@19 400
nuclear@19 401 } else {
nuclear@19 402 fprintf(stderr, "unrecognized camera option: %s\n", argv[i]);
nuclear@19 403 goto err;
nuclear@19 404 }
nuclear@19 405 }
nuclear@19 406 return cam;
nuclear@19 407
nuclear@19 408 err:
nuclear@19 409 delete cam;
nuclear@19 410 return 0;
nuclear@19 411 }
nuclear@19 412
nuclear@19 413 bool Scene::proc_cmd(int argc, char **argv)
nuclear@19 414 {
nuclear@19 415 Object *obj = 0;
nuclear@19 416 SceneNode *node = 0;
nuclear@19 417
nuclear@19 418 if(strcmp(argv[0], "sphere") == 0) {
nuclear@19 419 obj = new Sphere;
nuclear@19 420 node = proc_object(obj, argc, argv);
nuclear@19 421
nuclear@19 422 } else if(strcmp(argv[0], "box") == 0) {
nuclear@19 423 obj = new Box;
nuclear@19 424 node = proc_object(obj, argc, argv);
nuclear@19 425
nuclear@19 426 } else if(strcmp(argv[0], "mesh") == 0) {
nuclear@19 427 fprintf(stderr, "meshes not implemented yet\n");
nuclear@19 428 return false;
nuclear@19 429
nuclear@19 430 } else if(strcmp(argv[0], "camera") == 0) {
nuclear@19 431 Camera *cam = proc_camera(argc, argv);
nuclear@19 432 if(!cam) {
nuclear@19 433 return false;
nuclear@19 434 }
nuclear@19 435 use_camera(cam);
nuclear@19 436
nuclear@19 437 } else {
nuclear@19 438 fprintf(stderr, "unknown command: %s\n", argv[0]);
nuclear@19 439 return false;
nuclear@19 440 }
nuclear@19 441
nuclear@19 442 if(obj) {
nuclear@19 443 if(!node) {
nuclear@19 444 return false;
nuclear@19 445 }
nuclear@19 446 add_object(obj);
nuclear@19 447 add_node(node);
nuclear@19 448 }
nuclear@19 449
nuclear@19 450 return true;
nuclear@19 451 }