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 }
|