clray

annotate src/clray.cc @ 29:353d80127627

find changesets by author, revision, files, or words in the commit message
doh ... it doesn't work
author John Tsiombikas <nuclear@member.fsf.org>
date Sat, 21 Aug 2010 20:51:57 +0100
parents 8b2f2ad14ae7
children 4cf4919c3812
rev   line source
nuclear@0 1 #include <stdio.h>
nuclear@3 2 #include <stdlib.h>
nuclear@2 3 #include <string.h>
nuclear@27 4 #include <ctype.h>
nuclear@2 5 #include <errno.h>
nuclear@3 6 #ifndef __APPLE__
nuclear@3 7 #include <GL/glut.h>
nuclear@3 8 #else
nuclear@3 9 #include <GLUT/glut.h>
nuclear@3 10 #endif
nuclear@3 11 #include "rt.h"
nuclear@12 12 #include "matrix.h"
nuclear@22 13 #include "scene.h"
nuclear@0 14
nuclear@3 15 void cleanup();
nuclear@3 16 void disp();
nuclear@3 17 void reshape(int x, int y);
nuclear@3 18 void keyb(unsigned char key, int x, int y);
nuclear@3 19 void mouse(int bn, int status, int x, int y);
nuclear@3 20 void motion(int x, int y);
nuclear@2 21 bool write_ppm(const char *fname, float *fb, int xsz, int ysz);
nuclear@2 22
nuclear@3 23 static int xsz, ysz;
nuclear@3 24 static bool need_update = true;
nuclear@3 25
nuclear@8 26 static float cam_theta, cam_phi = 25.0;
nuclear@8 27 static float cam_dist = 10.0;
nuclear@8 28
nuclear@27 29 static bool dbg_glrender = true;
nuclear@27 30 static bool dbg_show_kdtree = false;
nuclear@27 31 static bool dbg_show_obj = true;
nuclear@13 32
nuclear@13 33 static Scene scn;
nuclear@12 34
nuclear@3 35 int main(int argc, char **argv)
nuclear@0 36 {
nuclear@3 37 glutInitWindowSize(800, 600);
nuclear@3 38 glutInit(&argc, argv);
nuclear@13 39
nuclear@13 40 int loaded = 0;
nuclear@13 41 for(int i=1; i<argc; i++) {
nuclear@27 42 if(argv[i][0] == '-' && argv[i][2] == 0) {
nuclear@27 43 switch(argv[i][1]) {
nuclear@27 44 case 'i':
nuclear@27 45 if(!argv[++i] || !isdigit(argv[i][0])) {
nuclear@27 46 fprintf(stderr, "-i must be followed by the intersection cost\n");
nuclear@27 47 return 1;
nuclear@27 48 }
nuclear@27 49
nuclear@27 50 set_accel_param(ACCEL_PARAM_COST_INTERSECT, atoi(argv[i]));
nuclear@27 51 break;
nuclear@27 52
nuclear@27 53 case 't':
nuclear@27 54 if(!argv[++i] || !isdigit(argv[i][0])) {
nuclear@27 55 fprintf(stderr, "-t must be followed by the traversal cost\n");
nuclear@27 56 return 1;
nuclear@27 57 }
nuclear@27 58
nuclear@27 59 set_accel_param(ACCEL_PARAM_COST_TRAVERSE, atoi(argv[i]));
nuclear@27 60 break;
nuclear@27 61
nuclear@27 62 case 'c':
nuclear@27 63 if(!argv[++i] || !isdigit(argv[i][0])) {
nuclear@27 64 fprintf(stderr, "-c must be followed by the max number of items per leaf node\n");
nuclear@27 65 return 1;
nuclear@27 66 }
nuclear@27 67
nuclear@27 68 set_accel_param(ACCEL_PARAM_MAX_NODE_ITEMS, atoi(argv[i]));
nuclear@27 69 break;
nuclear@27 70
nuclear@27 71 default:
nuclear@27 72 fprintf(stderr, "unrecognized option: %s\n", argv[i]);
nuclear@27 73 return 1;
nuclear@27 74 }
nuclear@27 75 } else {
nuclear@27 76 if(!scn.load(argv[i])) {
nuclear@27 77 fprintf(stderr, "failed to load scene: %s\n", argv[i]);
nuclear@27 78 return false;
nuclear@27 79 }
nuclear@27 80 loaded++;
nuclear@13 81 }
nuclear@13 82 }
nuclear@13 83
nuclear@13 84 if(!loaded) {
nuclear@13 85 fprintf(stderr, "you must specify a scene file to load\n");
nuclear@13 86 return false;
nuclear@13 87 }
nuclear@13 88 if(!scn.get_num_faces()) {
nuclear@13 89 fprintf(stderr, "didn't load any polygons\n");
nuclear@13 90 return false;
nuclear@13 91 }
nuclear@13 92
nuclear@18 93 glutInitDisplayMode(GLUT_RGB | GLUT_DEPTH | GLUT_DOUBLE);
nuclear@3 94 glutCreateWindow("OpenCL Raytracer");
nuclear@0 95
nuclear@3 96 xsz = glutGet(GLUT_WINDOW_WIDTH);
nuclear@3 97 ysz = glutGet(GLUT_WINDOW_HEIGHT);
nuclear@2 98
nuclear@3 99 glutDisplayFunc(disp);
nuclear@3 100 glutReshapeFunc(reshape);
nuclear@3 101 glutKeyboardFunc(keyb);
nuclear@3 102 glutMouseFunc(mouse);
nuclear@3 103 glutMotionFunc(motion);
nuclear@0 104
nuclear@13 105 if(!init_renderer(xsz, ysz, &scn)) {
nuclear@0 106 return 1;
nuclear@0 107 }
nuclear@3 108 atexit(cleanup);
nuclear@0 109
nuclear@27 110
nuclear@3 111 /*glGenTextures(1, &tex);
nuclear@3 112 glBindTexture(GL_TEXTURE_2D, tex);*/
nuclear@3 113 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP);
nuclear@3 114 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP);
nuclear@3 115 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
nuclear@3 116 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
nuclear@3 117 glTexImage2D(GL_TEXTURE_2D, 0, 4, xsz, ysz, 0, GL_RGBA, GL_FLOAT, 0);
nuclear@2 118
nuclear@3 119 glutMainLoop();
nuclear@0 120 return 0;
nuclear@0 121 }
nuclear@2 122
nuclear@3 123 void cleanup()
nuclear@2 124 {
nuclear@3 125 destroy_renderer();
nuclear@3 126 }
nuclear@2 127
nuclear@12 128 static Matrix4x4 mat, inv_mat, inv_trans;
nuclear@12 129
nuclear@3 130 void disp()
nuclear@3 131 {
nuclear@8 132 glMatrixMode(GL_MODELVIEW);
nuclear@8 133 glLoadIdentity();
nuclear@8 134
nuclear@3 135 if(need_update) {
nuclear@12 136 glPushMatrix();
nuclear@12 137 glRotatef(-cam_theta, 0, 1, 0);
nuclear@12 138 glRotatef(-cam_phi, 1, 0, 0);
nuclear@12 139 glTranslatef(0, 0, cam_dist);
nuclear@8 140
nuclear@12 141 glGetFloatv(GL_MODELVIEW_MATRIX, mat.m);
nuclear@8 142
nuclear@12 143 inv_mat = mat;
nuclear@12 144 inv_mat.invert();
nuclear@12 145
nuclear@12 146 /*inv_trans = inv_mat;
nuclear@12 147 inv_trans.transpose();*/
nuclear@12 148 inv_trans = mat;
nuclear@12 149 inv_trans.m[3] = inv_trans.m[7] = inv_trans.m[11] = 0.0;
nuclear@12 150 inv_trans.m[12] = inv_trans.m[13] = inv_trans.m[14] = 0.0;
nuclear@12 151 inv_trans.m[15] = 1.0;
nuclear@12 152
nuclear@12 153 set_xform(mat.m, inv_trans.m);
nuclear@8 154 glPopMatrix();
nuclear@8 155
nuclear@13 156 if(!dbg_glrender) {
nuclear@13 157 if(!render()) {
nuclear@13 158 exit(1);
nuclear@13 159 }
nuclear@13 160 need_update = false;
nuclear@12 161 }
nuclear@3 162 }
nuclear@2 163
nuclear@12 164 if(dbg_glrender) {
nuclear@12 165 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
nuclear@12 166 glLoadMatrixf(inv_mat.m);
nuclear@27 167 dbg_render_gl(&scn, dbg_show_kdtree, dbg_show_obj);
nuclear@12 168 } else {
nuclear@12 169 glEnable(GL_TEXTURE_2D);
nuclear@22 170 glDisable(GL_LIGHTING);
nuclear@2 171
nuclear@12 172 glBegin(GL_QUADS);
nuclear@12 173 glColor3f(1, 1, 1);
nuclear@12 174 glTexCoord2f(0, 1); glVertex2f(-1, -1);
nuclear@12 175 glTexCoord2f(1, 1); glVertex2f(1, -1);
nuclear@12 176 glTexCoord2f(1, 0); glVertex2f(1, 1);
nuclear@12 177 glTexCoord2f(0, 0); glVertex2f(-1, 1);
nuclear@12 178 glEnd();
nuclear@2 179
nuclear@12 180 glDisable(GL_TEXTURE_2D);
nuclear@12 181 }
nuclear@3 182
nuclear@3 183 glutSwapBuffers();
nuclear@3 184 }
nuclear@3 185
nuclear@3 186 void reshape(int x, int y)
nuclear@3 187 {
nuclear@3 188 glViewport(0, 0, x, y);
nuclear@3 189
nuclear@3 190 /* reallocate the framebuffer */
nuclear@3 191 /*delete [] fb;
nuclear@3 192 fb = new float[x * y * 4];
nuclear@3 193 set_framebuffer(fb, x, y);*/
nuclear@3 194 }
nuclear@3 195
nuclear@3 196 void keyb(unsigned char key, int x, int y)
nuclear@3 197 {
nuclear@3 198 switch(key) {
nuclear@3 199 case 27:
nuclear@3 200 exit(0);
nuclear@3 201
nuclear@12 202 case 'r':
nuclear@12 203 need_update = true;
nuclear@12 204 glutPostRedisplay();
nuclear@3 205 break;
nuclear@3 206
nuclear@12 207 case 'd':
nuclear@12 208 dbg_glrender = !dbg_glrender;
nuclear@12 209 if(dbg_glrender) {
nuclear@21 210 printf("Debug OpenGL rendering\n");
nuclear@21 211 } else {
nuclear@21 212 printf("Raytracing\n");
nuclear@12 213 }
nuclear@3 214 glutPostRedisplay();
nuclear@3 215 break;
nuclear@3 216
nuclear@27 217 case 'k':
nuclear@27 218 dbg_show_kdtree = !dbg_show_kdtree;
nuclear@27 219 if(dbg_glrender) {
nuclear@27 220 glutPostRedisplay();
nuclear@27 221 }
nuclear@27 222 break;
nuclear@27 223
nuclear@27 224 case 'o':
nuclear@27 225 dbg_show_obj = !dbg_show_obj;
nuclear@27 226 if(dbg_glrender) {
nuclear@27 227 glutPostRedisplay();
nuclear@27 228 }
nuclear@27 229 break;
nuclear@27 230
nuclear@3 231 default:
nuclear@3 232 break;
nuclear@3 233 }
nuclear@3 234 }
nuclear@3 235
nuclear@8 236 static bool bnstate[32];
nuclear@8 237 static int prev_x, prev_y;
nuclear@8 238
nuclear@3 239 void mouse(int bn, int state, int x, int y)
nuclear@3 240 {
nuclear@8 241 if(state == GLUT_DOWN) {
nuclear@8 242 prev_x = x;
nuclear@8 243 prev_y = y;
nuclear@8 244 bnstate[bn] = true;
nuclear@8 245 } else {
nuclear@8 246 bnstate[bn] = false;
nuclear@8 247 }
nuclear@3 248 }
nuclear@3 249
nuclear@8 250 #define ROT_SCALE 0.5
nuclear@8 251 #define PAN_SCALE 0.1
nuclear@8 252
nuclear@3 253 void motion(int x, int y)
nuclear@3 254 {
nuclear@8 255 int dx = x - prev_x;
nuclear@8 256 int dy = y - prev_y;
nuclear@8 257 prev_x = x;
nuclear@8 258 prev_y = y;
nuclear@8 259
nuclear@8 260 if(bnstate[0]) {
nuclear@8 261 cam_theta += dx * ROT_SCALE;
nuclear@8 262 cam_phi += dy * ROT_SCALE;
nuclear@8 263
nuclear@12 264 if(cam_phi < -89) cam_phi = -89;
nuclear@8 265 if(cam_phi > 89) cam_phi = 89;
nuclear@8 266
nuclear@8 267 need_update = true;
nuclear@8 268 glutPostRedisplay();
nuclear@8 269 }
nuclear@8 270 if(bnstate[2]) {
nuclear@8 271 cam_dist += dy * PAN_SCALE;
nuclear@8 272 if(cam_dist < 0) cam_dist = 0;
nuclear@8 273
nuclear@8 274 need_update = true;
nuclear@8 275 glutPostRedisplay();
nuclear@8 276 }
nuclear@2 277 }
nuclear@2 278
nuclear@2 279 bool write_ppm(const char *fname, float *fb, int xsz, int ysz)
nuclear@2 280 {
nuclear@2 281 FILE *fp;
nuclear@2 282
nuclear@2 283 if(!(fp = fopen(fname, "wb"))) {
nuclear@2 284 fprintf(stderr, "write_ppm: failed to open file %s for writing: %s\n", fname, strerror(errno));
nuclear@2 285 return false;
nuclear@2 286 }
nuclear@2 287 fprintf(fp, "P6\n%d %d\n255\n", xsz, ysz);
nuclear@2 288
nuclear@2 289 for(int i=0; i<xsz * ysz * 4; i++) {
nuclear@2 290 if(i % 4 == 3) continue;
nuclear@2 291
nuclear@2 292 unsigned char c = (unsigned char)(fb[i] * 255.0);
nuclear@2 293 fputc(c, fp);
nuclear@2 294 }
nuclear@2 295 fclose(fp);
nuclear@2 296 return true;
nuclear@2 297 }