stereoplay

annotate src/stereoplay.c @ 0:265a24704ff2

find changesets by author, revision, files, or words in the commit message
stereoplay
author John Tsiombikas <nuclear@member.fsf.org>
date Sun, 06 Mar 2011 20:31:18 +0200
parents
children b50fd6f24975
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 <errno.h>
nuclear@0 5 #include <signal.h>
nuclear@0 6 #include <time.h>
nuclear@0 7 #include <assert.h>
nuclear@0 8 #include <alloca.h>
nuclear@0 9 #include <unistd.h>
nuclear@0 10 #include <fcntl.h>
nuclear@0 11 #include <sys/mman.h>
nuclear@0 12 #include <sys/wait.h>
nuclear@0 13 #include <sys/time.h>
nuclear@0 14 #include <sys/select.h>
nuclear@0 15 #include <X11/Xlib.h>
nuclear@0 16 #include <GL/glew.h>
nuclear@0 17 #include <GL/glut.h>
nuclear@0 18 #include <GL/freeglut_ext.h>
nuclear@0 19 #include <GL/glx.h>
nuclear@0 20 #include "vid.h"
nuclear@0 21 #include "sdr.h"
nuclear@0 22 #include "config.h"
nuclear@0 23
nuclear@0 24 void cleanup(void);
nuclear@0 25 void decoding_loop(void);
nuclear@0 26 void disp(void);
nuclear@0 27 void reshape(int x, int y);
nuclear@0 28 void keyb(unsigned char key, int x, int y);
nuclear@0 29 void skeyb(int key, int x, int y);
nuclear@0 30 void sig(int s);
nuclear@0 31 void *shmalloc(size_t sz);
nuclear@0 32
nuclear@0 33 struct video_file *vf;
nuclear@0 34 uint32_t *img;
nuclear@0 35 int vid_xsz, vid_ysz, win_xsz, win_ysz;
nuclear@0 36 unsigned int tex, sdr;
nuclear@0 37 int pfd[2], xsock = -1;
nuclear@0 38 int upd_frame;
nuclear@0 39 int fullscr;
nuclear@0 40 int swap_eyes;
nuclear@0 41
nuclear@0 42 Display *dpy;
nuclear@0 43
nuclear@0 44 int main(int argc, char **argv)
nuclear@0 45 {
nuclear@0 46 int pid;
nuclear@0 47 char *stereo_combiner, *sdrfile;
nuclear@0 48
nuclear@0 49 atexit(cleanup);
nuclear@0 50
nuclear@0 51 if(!(vf = vid_open(argv[1]))) {
nuclear@0 52 return 1;
nuclear@0 53 }
nuclear@0 54 vid_xsz = win_xsz = vid_frame_width(vf);
nuclear@0 55 vid_ysz = win_ysz = vid_frame_height(vf);
nuclear@0 56
nuclear@0 57 if(!(img = shmalloc(vid_frame_size(vf)))) {
nuclear@0 58 perror("failed to allocate frame buffer");
nuclear@0 59 return 1;
nuclear@0 60 }
nuclear@0 61
nuclear@0 62 if(pipe(pfd) == -1) {
nuclear@0 63 perror("failed to create a self-pipe");
nuclear@0 64 return 1;
nuclear@0 65 }
nuclear@0 66
nuclear@0 67 glutInit(&argc, argv);
nuclear@0 68 glutInitDisplayMode(GLUT_RGB | GLUT_DOUBLE);
nuclear@0 69 glutInitWindowSize(win_xsz, win_ysz);
nuclear@0 70 glutCreateWindow(argv[1]);
nuclear@0 71
nuclear@0 72 glutDisplayFunc(disp);
nuclear@0 73 glutReshapeFunc(reshape);
nuclear@0 74 glutKeyboardFunc(keyb);
nuclear@0 75 glutSpecialFunc(skeyb);
nuclear@0 76
nuclear@0 77 glewInit();
nuclear@0 78
nuclear@0 79 /* create the frame texture */
nuclear@0 80 glGenTextures(1, &tex);
nuclear@0 81 glBindTexture(GL_TEXTURE_2D, tex);
nuclear@0 82 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP);
nuclear@0 83 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP);
nuclear@0 84 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
nuclear@0 85 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
nuclear@0 86 glTexImage2D(GL_TEXTURE_2D, 0, 4, vid_xsz, vid_ysz, 0, GL_RGBA, GL_UNSIGNED_BYTE, 0);
nuclear@0 87 glEnable(GL_TEXTURE_2D);
nuclear@0 88
nuclear@0 89 if(!(stereo_combiner = getenv("STEREO_METHOD"))) {
nuclear@0 90 stereo_combiner = "redcyan";
nuclear@0 91 }
nuclear@0 92
nuclear@0 93 sdrfile = alloca(strlen(stereo_combiner) + strlen(SDRDIR) + 7);
nuclear@0 94 sprintf(sdrfile, SDRDIR "/%s.glsl", stereo_combiner);
nuclear@0 95
nuclear@0 96 if(!(sdr = create_program_load(0, sdrfile))) {
nuclear@0 97 return 1;
nuclear@0 98 }
nuclear@0 99 bind_program(sdr);
nuclear@0 100 set_uniform_float(sdr, "left_offs", 0.5);
nuclear@0 101 set_uniform_float(sdr, "right_offs", 0.0);
nuclear@0 102
nuclear@0 103 signal(SIGCHLD, sig);
nuclear@0 104
nuclear@0 105 if((pid = fork()) == -1) {
nuclear@0 106 perror("failed to fork video decoding process");
nuclear@0 107 return 1;
nuclear@0 108 } else if(pid) {
nuclear@0 109 close(pfd[1]);
nuclear@0 110 } else {
nuclear@0 111 close(pfd[0]);
nuclear@0 112 decoding_loop();
nuclear@0 113 _exit(0);
nuclear@0 114 }
nuclear@0 115
nuclear@0 116 dpy = glXGetCurrentDisplay();
nuclear@0 117 xsock = ConnectionNumber(dpy);
nuclear@0 118
nuclear@0 119 for(;;) {
nuclear@0 120 int res;
nuclear@0 121 fd_set rdset;
nuclear@0 122
nuclear@0 123 FD_ZERO(&rdset);
nuclear@0 124 FD_SET(xsock, &rdset);
nuclear@0 125 FD_SET(pfd[0], &rdset);
nuclear@0 126
nuclear@0 127 do {
nuclear@0 128 res = select((xsock > pfd[0] ? xsock : pfd[0]) + 1, &rdset, 0, 0, 0);
nuclear@0 129 } while(res == -1 && errno == EINTR);
nuclear@0 130
nuclear@0 131 if(FD_ISSET(pfd[0], &rdset)) {
nuclear@0 132 unsigned char done;
nuclear@0 133 read(pfd[0], &done, 1);
nuclear@0 134
nuclear@0 135 if(done) {
nuclear@0 136 exit(0);
nuclear@0 137 } else {
nuclear@0 138 upd_frame = 1;
nuclear@0 139 glutPostRedisplay();
nuclear@0 140 }
nuclear@0 141 }
nuclear@0 142
nuclear@0 143 glutMainLoopEvent();
nuclear@0 144 }
nuclear@0 145
nuclear@0 146 return 0;
nuclear@0 147 }
nuclear@0 148
nuclear@0 149 void cleanup(void)
nuclear@0 150 {
nuclear@0 151 if(tex) {
nuclear@0 152 glDeleteTextures(1, &tex);
nuclear@0 153 }
nuclear@0 154 if(img) {
nuclear@0 155 munmap(img, vid_frame_size(vf));
nuclear@0 156 }
nuclear@0 157 if(vf) {
nuclear@0 158 vid_close(vf);
nuclear@0 159 }
nuclear@0 160 close(pfd[0]);
nuclear@0 161 }
nuclear@0 162
nuclear@0 163 /* decoding_loop() runs in a separate decoding process and communicates
nuclear@0 164 * with the parent process through the pfd[1] pipe.
nuclear@0 165 */
nuclear@0 166 void decoding_loop(void)
nuclear@0 167 {
nuclear@0 168 unsigned char done = 0;
nuclear@0 169 struct timespec ts;
nuclear@0 170 struct timeval tv0, tv;
nuclear@0 171 unsigned long frame_nsec = vid_frame_interval(vf) * 1000;
nuclear@0 172 printf("nanosecs per frame: %lu\n", frame_nsec);
nuclear@0 173
nuclear@0 174 gettimeofday(&tv0, 0);
nuclear@0 175
nuclear@0 176 while(vid_get_frame(vf, img) != -1) {
nuclear@0 177 write(pfd[1], &done, 1);
nuclear@0 178
nuclear@0 179 gettimeofday(&tv, 0);
nuclear@0 180
nuclear@0 181 ts.tv_sec = 0;
nuclear@0 182 ts.tv_nsec = frame_nsec - (tv.tv_usec - tv0.tv_usec) * 1000;
nuclear@0 183 nanosleep(&ts, 0);
nuclear@0 184
nuclear@0 185 gettimeofday(&tv0, 0);
nuclear@0 186 }
nuclear@0 187
nuclear@0 188 done = 1;
nuclear@0 189 write(pfd[1], &done, 1);
nuclear@0 190 }
nuclear@0 191
nuclear@0 192 void disp(void)
nuclear@0 193 {
nuclear@0 194 if(upd_frame) {
nuclear@0 195 /* frame changed, we must re-upload the texture */
nuclear@0 196 glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, vid_xsz, vid_ysz, GL_BGRA, GL_UNSIGNED_BYTE, img);
nuclear@0 197 upd_frame = 0;
nuclear@0 198 }
nuclear@0 199
nuclear@0 200 glBegin(GL_QUADS);
nuclear@0 201 glColor3f(1, 1, 1);
nuclear@0 202 glTexCoord2f(0, 1); glVertex2f(-1, -1);
nuclear@0 203 glTexCoord2f(1, 1); glVertex2f(1, -1);
nuclear@0 204 glTexCoord2f(1, 0); glVertex2f(1, 1);
nuclear@0 205 glTexCoord2f(0, 0); glVertex2f(-1, 1);
nuclear@0 206 glEnd();
nuclear@0 207
nuclear@0 208 glutSwapBuffers();
nuclear@0 209 }
nuclear@0 210
nuclear@0 211 void reshape(int x, int y)
nuclear@0 212 {
nuclear@0 213 if(!fullscr) {
nuclear@0 214 win_xsz = x;
nuclear@0 215 win_ysz = y;
nuclear@0 216 }
nuclear@0 217
nuclear@0 218 glViewport(0, 0, x, y);
nuclear@0 219 }
nuclear@0 220
nuclear@0 221 void keyb(unsigned char key, int x, int y)
nuclear@0 222 {
nuclear@0 223 switch(key) {
nuclear@0 224 case 'q':
nuclear@0 225 exit(0);
nuclear@0 226
nuclear@0 227 case 'f':
nuclear@0 228 fullscr = !fullscr;
nuclear@0 229 if(fullscr) {
nuclear@0 230 glutFullScreen();
nuclear@0 231 } else {
nuclear@0 232 glutReshapeWindow(win_xsz, win_ysz);
nuclear@0 233 }
nuclear@0 234 break;
nuclear@0 235
nuclear@0 236 case ' ':
nuclear@0 237 /* TODO pause/resume */
nuclear@0 238 break;
nuclear@0 239
nuclear@0 240 case 's':
nuclear@0 241 swap_eyes = !swap_eyes;
nuclear@0 242 if(swap_eyes) {
nuclear@0 243 set_uniform_float(sdr, "left_offs", 0.0);
nuclear@0 244 set_uniform_float(sdr, "right_offs", 0.5);
nuclear@0 245 } else {
nuclear@0 246 set_uniform_float(sdr, "left_offs", 0.5);
nuclear@0 247 set_uniform_float(sdr, "right_offs", 0.0);
nuclear@0 248 }
nuclear@0 249 break;
nuclear@0 250
nuclear@0 251 default:
nuclear@0 252 break;
nuclear@0 253 }
nuclear@0 254 }
nuclear@0 255
nuclear@0 256 void skeyb(int key, int x, int y)
nuclear@0 257 {
nuclear@0 258 switch(key) {
nuclear@0 259 case GLUT_KEY_LEFT:
nuclear@0 260 /* TODO skip fwd */
nuclear@0 261 break;
nuclear@0 262
nuclear@0 263 case GLUT_KEY_RIGHT:
nuclear@0 264 /* TODO skip back */
nuclear@0 265 break;
nuclear@0 266
nuclear@0 267 case GLUT_KEY_UP:
nuclear@0 268 /* TODO skip fwd more */
nuclear@0 269 break;
nuclear@0 270
nuclear@0 271 case GLUT_KEY_DOWN:
nuclear@0 272 /* TODO skip back more */
nuclear@0 273 break;
nuclear@0 274
nuclear@0 275 case GLUT_KEY_PAGE_UP:
nuclear@0 276 /* TODO skip fwd a lot */
nuclear@0 277 break;
nuclear@0 278
nuclear@0 279 case GLUT_KEY_PAGE_DOWN:
nuclear@0 280 /* TODO skip back a lot */
nuclear@0 281 break;
nuclear@0 282
nuclear@0 283 default:
nuclear@0 284 break;
nuclear@0 285 }
nuclear@0 286 }
nuclear@0 287
nuclear@0 288 void sig(int s)
nuclear@0 289 {
nuclear@0 290 if(s == SIGCHLD) {
nuclear@0 291 wait(0);
nuclear@0 292 }
nuclear@0 293 }
nuclear@0 294
nuclear@0 295 void *shmalloc(size_t sz)
nuclear@0 296 {
nuclear@0 297 int fd;
nuclear@0 298 void *shm;
nuclear@0 299
nuclear@0 300 if(!(fd = open("/dev/zero", O_RDWR))) {
nuclear@0 301 return 0;
nuclear@0 302 }
nuclear@0 303
nuclear@0 304 if((shm = mmap(0, sz, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0)) == (void*)-1) {
nuclear@0 305 shm = 0;
nuclear@0 306 }
nuclear@0 307
nuclear@0 308 close(fd);
nuclear@0 309 return shm;
nuclear@0 310 }