cubemapper

annotate src/app.cc @ 2:e308561f9889

find changesets by author, revision, files, or words in the commit message
correct cubemap export and visualization
author John Tsiombikas <nuclear@member.fsf.org>
date Fri, 28 Jul 2017 13:24:34 +0300
parents d7a29cb7ac8d
children f5cc465eb735
rev   line source
nuclear@0 1 #include <stdio.h>
nuclear@0 2 #include <stdlib.h>
nuclear@0 3 #include <string.h>
nuclear@0 4 #include <math.h>
nuclear@0 5 #include <assert.h>
nuclear@0 6 #include <imago2.h>
nuclear@0 7 #include "app.h"
nuclear@0 8 #include "opengl.h"
nuclear@0 9 #include "texture.h"
nuclear@0 10 #include "mesh.h"
nuclear@0 11 #include "meshgen.h"
nuclear@0 12
nuclear@2 13 static void draw_equilateral();
nuclear@2 14 static void draw_cubemap();
nuclear@0 15 static bool parse_args(int argc, char **argv);
nuclear@0 16
nuclear@1 17 static void flip_image(float *pixels, int xsz, int ysz);
nuclear@1 18
nuclear@0 19 static const char *img_fname;
nuclear@0 20 static float cam_theta, cam_phi;
nuclear@0 21
nuclear@0 22 static Texture *pano_tex;
nuclear@0 23 static Mesh *pano_mesh;
nuclear@0 24
nuclear@0 25 static int win_width, win_height;
nuclear@2 26 static int show_cubemap;
nuclear@2 27
nuclear@2 28 static unsigned int fbo;
nuclear@2 29 static unsigned int cube_tex;
nuclear@2 30 static int cube_size;
nuclear@0 31
nuclear@0 32
nuclear@0 33 bool app_init(int argc, char **argv)
nuclear@0 34 {
nuclear@0 35 if(!parse_args(argc, argv)) {
nuclear@0 36 return false;
nuclear@0 37 }
nuclear@0 38 if(!img_fname) {
nuclear@0 39 fprintf(stderr, "please specify an equilateral panoramic image\n");
nuclear@0 40 return false;
nuclear@0 41 }
nuclear@0 42
nuclear@0 43 if(!init_opengl()) {
nuclear@0 44 return false;
nuclear@0 45 }
nuclear@0 46
nuclear@2 47 glEnable(GL_MULTISAMPLE);
nuclear@0 48
nuclear@0 49 Mesh::use_custom_sdr_attr = false;
nuclear@0 50 pano_mesh = new Mesh;
nuclear@0 51 gen_sphere(pano_mesh, 1.0, 80, 40);
nuclear@0 52 pano_mesh->flip();
nuclear@0 53 Mat4 xform;
nuclear@0 54 xform.rotation_y(-M_PI / 2.0); // rotate the sphere to face the "front" part of the image
nuclear@0 55 pano_mesh->apply_xform(xform, xform);
nuclear@0 56
nuclear@0 57 xform.scaling(-1, 1, 1); // flip horizontal texcoord since we're inside the sphere
nuclear@0 58 pano_mesh->texcoord_apply_xform(xform);
nuclear@0 59
nuclear@0 60 pano_tex = new Texture;
nuclear@0 61 if(!pano_tex->load(img_fname)) {
nuclear@0 62 return false;
nuclear@0 63 }
nuclear@1 64 printf("loaded image: %dx%d\n", pano_tex->get_width(), pano_tex->get_height());
nuclear@2 65
nuclear@2 66 // create cubemap
nuclear@2 67 cube_size = pano_tex->get_height();
nuclear@2 68 glGenTextures(1, &cube_tex);
nuclear@2 69 glBindTexture(GL_TEXTURE_CUBE_MAP, cube_tex);
nuclear@2 70 glTexParameteri(GL_TEXTURE_CUBE_MAP, GL_TEXTURE_MIN_FILTER, GL_LINEAR_MIPMAP_LINEAR);
nuclear@2 71 glTexParameteri(GL_TEXTURE_CUBE_MAP, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
nuclear@2 72 glTexParameteri(GL_TEXTURE_CUBE_MAP, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
nuclear@2 73 glTexParameteri(GL_TEXTURE_CUBE_MAP, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
nuclear@2 74 glTexParameteri(GL_TEXTURE_CUBE_MAP, GL_TEXTURE_WRAP_R, GL_CLAMP_TO_EDGE);
nuclear@2 75
nuclear@2 76 for(int i=0; i<6; i++) {
nuclear@2 77 glTexImage2D(GL_TEXTURE_CUBE_MAP_POSITIVE_X + i, 0, GL_RGB16F, cube_size, cube_size,
nuclear@2 78 0, GL_RGB, GL_FLOAT, 0);
nuclear@2 79 }
nuclear@2 80
nuclear@2 81
nuclear@2 82 // create fbo
nuclear@2 83 glGenFramebuffers(1, &fbo);
nuclear@2 84
nuclear@2 85 // tex-gen for cubemap visualization
nuclear@2 86 glTexGeni(GL_S, GL_TEXTURE_GEN_MODE, GL_OBJECT_LINEAR);
nuclear@2 87 glTexGeni(GL_T, GL_TEXTURE_GEN_MODE, GL_OBJECT_LINEAR);
nuclear@2 88 glTexGeni(GL_R, GL_TEXTURE_GEN_MODE, GL_OBJECT_LINEAR);
nuclear@2 89 float planes[][4] = {{1, 0, 0, 0}, {0, 1, 0, 0}, {0, 0, 1, 0}};
nuclear@2 90 glTexGenfv(GL_S, GL_OBJECT_PLANE, planes[0]);
nuclear@2 91 glTexGenfv(GL_T, GL_OBJECT_PLANE, planes[1]);
nuclear@2 92 glTexGenfv(GL_R, GL_OBJECT_PLANE, planes[2]);
nuclear@0 93 return true;
nuclear@0 94 }
nuclear@0 95
nuclear@0 96 void app_cleanup()
nuclear@0 97 {
nuclear@0 98 delete pano_mesh;
nuclear@0 99 delete pano_tex;
nuclear@0 100 }
nuclear@0 101
nuclear@0 102 void app_draw()
nuclear@0 103 {
nuclear@0 104 glClear(GL_COLOR_BUFFER_BIT);
nuclear@0 105
nuclear@0 106 Mat4 view_matrix;
nuclear@0 107 view_matrix.pre_rotate_x(deg_to_rad(cam_phi));
nuclear@0 108 view_matrix.pre_rotate_y(deg_to_rad(cam_theta));
nuclear@0 109
nuclear@0 110 glMatrixMode(GL_MODELVIEW);
nuclear@0 111 glLoadMatrixf(view_matrix[0]);
nuclear@0 112
nuclear@2 113 if(show_cubemap) {
nuclear@2 114 draw_cubemap();
nuclear@2 115
nuclear@2 116 glColor3f(0, 0, 0);
nuclear@2 117 app_print_text(10, 10, "cubemap");
nuclear@2 118 glColor3f(0, 0.8, 1);
nuclear@2 119 app_print_text(8, 13, "cubemap");
nuclear@2 120 } else {
nuclear@2 121 draw_equilateral();
nuclear@2 122
nuclear@2 123 glColor3f(0, 0, 0);
nuclear@2 124 app_print_text(10, 10, "equilateral");
nuclear@2 125 glColor3f(1, 0.8, 0);
nuclear@2 126 app_print_text(8, 13, "equilateral");
nuclear@2 127 }
nuclear@2 128 glColor3f(1, 1, 1);
nuclear@0 129
nuclear@0 130 app_swap_buffers();
nuclear@0 131 assert(glGetError() == GL_NO_ERROR);
nuclear@0 132 }
nuclear@0 133
nuclear@0 134 void render_cubemap()
nuclear@0 135 {
nuclear@2 136 printf("rendering cubemap %dx%d\n", cube_size, cube_size);
nuclear@0 137
nuclear@2 138 float *pixels = new float[cube_size * cube_size * 3];
nuclear@2 139
nuclear@2 140 glViewport(0, 0, cube_size, cube_size);
nuclear@0 141
nuclear@0 142 Mat4 viewmat[6];
nuclear@0 143 viewmat[0].rotation_y(deg_to_rad(90)); // +X
nuclear@2 144 viewmat[1].rotation_y(deg_to_rad(-90)); // -X
nuclear@2 145 viewmat[2].rotation_x(deg_to_rad(90)); // +Y
nuclear@2 146 viewmat[2].rotate_y(deg_to_rad(180));
nuclear@2 147 viewmat[3].rotation_x(deg_to_rad(-90)); // -Y
nuclear@2 148 viewmat[3].rotate_y(deg_to_rad(180));
nuclear@2 149 viewmat[4].rotation_y(deg_to_rad(180)); // +Z
nuclear@0 150
nuclear@2 151 // this must coincide with the order of GL_TEXTURE_CUBE_MAP_* values
nuclear@0 152 static const char *fname[] = {
nuclear@0 153 "cubemap_px.jpg",
nuclear@2 154 "cubemap_nx.jpg",
nuclear@0 155 "cubemap_py.jpg",
nuclear@2 156 "cubemap_ny.jpg",
nuclear@0 157 "cubemap_pz.jpg",
nuclear@0 158 "cubemap_nz.jpg"
nuclear@0 159 };
nuclear@0 160
nuclear@0 161 glMatrixMode(GL_PROJECTION);
nuclear@2 162 glPushMatrix();
nuclear@0 163 glLoadIdentity();
nuclear@2 164 gluPerspective(90, 1.0, 0.5, 500.0);
nuclear@2 165 glScalef(-1, -1, 1);
nuclear@2 166
nuclear@2 167 glBindFramebuffer(GL_FRAMEBUFFER, fbo);
nuclear@0 168
nuclear@0 169 for(int i=0; i<6; i++) {
nuclear@2 170 glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0,
nuclear@2 171 GL_TEXTURE_CUBE_MAP_POSITIVE_X + i, cube_tex, 0);
nuclear@2 172
nuclear@0 173 glClear(GL_COLOR_BUFFER_BIT);
nuclear@0 174
nuclear@0 175 glMatrixMode(GL_MODELVIEW);
nuclear@0 176 glLoadMatrixf(viewmat[i][0]);
nuclear@0 177
nuclear@2 178 draw_equilateral();
nuclear@0 179
nuclear@2 180 //glReadPixels(0, 0, cube_size, cube_size, GL_RGB, GL_FLOAT, pixels);
nuclear@2 181 glGetTexImage(GL_TEXTURE_CUBE_MAP_POSITIVE_X + i, 0, GL_RGB, GL_FLOAT, pixels);
nuclear@2 182 //flip_image(pixels, cube_size, cube_size);
nuclear@1 183
nuclear@2 184 if(img_save_pixels(fname[i], pixels, cube_size, cube_size, IMG_FMT_RGBF) == -1) {
nuclear@2 185 fprintf(stderr, "failed to save %dx%d image: %s\n", cube_size, cube_size, fname[i]);
nuclear@0 186 }
nuclear@0 187 }
nuclear@0 188
nuclear@2 189 glBindFramebuffer(GL_FRAMEBUFFER, 0);
nuclear@0 190 glViewport(0, 0, win_width, win_height);
nuclear@0 191
nuclear@2 192 glMatrixMode(GL_PROJECTION);
nuclear@2 193 glPopMatrix();
nuclear@2 194
nuclear@0 195 delete [] pixels;
nuclear@2 196
nuclear@2 197 glBindTexture(GL_TEXTURE_CUBE_MAP, cube_tex);
nuclear@2 198 glGenerateMipmap(GL_TEXTURE_CUBE_MAP);
nuclear@0 199 }
nuclear@0 200
nuclear@2 201 static void draw_equilateral()
nuclear@0 202 {
nuclear@0 203 pano_tex->bind();
nuclear@0 204 glEnable(GL_TEXTURE_2D);
nuclear@0 205 pano_mesh->draw();
nuclear@0 206 glDisable(GL_TEXTURE_2D);
nuclear@0 207 }
nuclear@0 208
nuclear@2 209 static void draw_cubemap()
nuclear@0 210 {
nuclear@2 211 glPushAttrib(GL_ENABLE_BIT);
nuclear@2 212
nuclear@2 213 glBindTexture(GL_TEXTURE_CUBE_MAP, cube_tex);
nuclear@2 214 glEnable(GL_TEXTURE_CUBE_MAP);
nuclear@2 215 glEnable(GL_TEXTURE_GEN_S);
nuclear@2 216 glEnable(GL_TEXTURE_GEN_T);
nuclear@2 217 glEnable(GL_TEXTURE_GEN_R);
nuclear@2 218
nuclear@2 219 pano_mesh->draw();
nuclear@2 220
nuclear@2 221 glPopAttrib();
nuclear@0 222 }
nuclear@0 223
nuclear@0 224 void app_reshape(int x, int y)
nuclear@0 225 {
nuclear@0 226 glViewport(0, 0, x, y);
nuclear@0 227
nuclear@0 228 glMatrixMode(GL_PROJECTION);
nuclear@0 229 glLoadIdentity();
nuclear@0 230 gluPerspective(50.0, (float)x / (float)y, 0.5, 500.0);
nuclear@0 231
nuclear@0 232 win_width = x;
nuclear@0 233 win_height = y;
nuclear@0 234 }
nuclear@0 235
nuclear@0 236 void app_keyboard(int key, bool press)
nuclear@0 237 {
nuclear@0 238 if(press) {
nuclear@0 239 switch(key) {
nuclear@0 240 case 27:
nuclear@0 241 app_quit();
nuclear@0 242 break;
nuclear@0 243
nuclear@1 244 case ' ':
nuclear@2 245 show_cubemap = !show_cubemap;
nuclear@2 246 app_redisplay();
nuclear@1 247 break;
nuclear@1 248
nuclear@2 249 case 'c':
nuclear@0 250 render_cubemap();
nuclear@0 251 break;
nuclear@0 252 }
nuclear@0 253 }
nuclear@0 254 }
nuclear@0 255
nuclear@0 256 static float prev_x, prev_y;
nuclear@0 257 static bool bnstate[16];
nuclear@0 258
nuclear@0 259 void app_mouse_button(int bn, bool press, int x, int y)
nuclear@0 260 {
nuclear@0 261 if(bn < (int)(sizeof bnstate / sizeof *bnstate)) {
nuclear@0 262 bnstate[bn] = press;
nuclear@0 263 }
nuclear@0 264 prev_x = x;
nuclear@0 265 prev_y = y;
nuclear@0 266 }
nuclear@0 267
nuclear@0 268 void app_mouse_motion(int x, int y)
nuclear@0 269 {
nuclear@0 270 float dx = x - prev_x;
nuclear@0 271 float dy = y - prev_y;
nuclear@0 272 prev_x = x;
nuclear@0 273 prev_y = y;
nuclear@0 274
nuclear@0 275 if(!dx && !dy) return;
nuclear@0 276
nuclear@0 277 if(bnstate[0]) {
nuclear@0 278 cam_theta += dx * 0.5;
nuclear@0 279 cam_phi += dy * 0.5;
nuclear@0 280
nuclear@0 281 if(cam_phi < -90) cam_phi = -90;
nuclear@0 282 if(cam_phi > 90) cam_phi = 90;
nuclear@0 283 app_redisplay();
nuclear@0 284 }
nuclear@0 285 }
nuclear@0 286
nuclear@0 287 static bool parse_args(int argc, char **argv)
nuclear@0 288 {
nuclear@0 289 for(int i=1; i<argc; i++) {
nuclear@0 290 if(argv[i][0] == '-') {
nuclear@0 291 /*
nuclear@0 292 } else if(strcmp(argv[i], "-help") == 0) {
nuclear@0 293 printf("usage: %s [options]\noptions:\n", argv[0]);
nuclear@0 294 printf(" -help: print usage information and exit\n");
nuclear@0 295 exit(0);
nuclear@0 296 } else {*/
nuclear@0 297 fprintf(stderr, "invalid option: %s\n", argv[i]);
nuclear@0 298 return false;
nuclear@0 299 //}
nuclear@0 300 } else {
nuclear@0 301 if(img_fname) {
nuclear@0 302 fprintf(stderr, "unexpected option: %s\n", argv[i]);
nuclear@0 303 return false;
nuclear@0 304 }
nuclear@0 305 img_fname = argv[i];
nuclear@0 306 }
nuclear@0 307 }
nuclear@0 308
nuclear@0 309 return true;
nuclear@0 310 }
nuclear@1 311
nuclear@1 312 static void flip_image(float *pixels, int xsz, int ysz)
nuclear@1 313 {
nuclear@1 314 float *top_ptr = pixels;
nuclear@1 315 float *bot_ptr = pixels + xsz * (ysz - 1) * 3;
nuclear@1 316 float *line = new float[xsz * 3];
nuclear@1 317 int scansz = xsz * 3 * sizeof(float);
nuclear@1 318
nuclear@1 319 for(int i=0; i<ysz / 2; i++) {
nuclear@1 320 memcpy(line, top_ptr, scansz);
nuclear@1 321 memcpy(top_ptr, bot_ptr, scansz);
nuclear@1 322 memcpy(bot_ptr, line, scansz);
nuclear@1 323 top_ptr += xsz * 3;
nuclear@1 324 bot_ptr -= xsz * 3;
nuclear@1 325 }
nuclear@1 326
nuclear@1 327 delete [] line;
nuclear@1 328 }