1 /********************************************************************************
2 * *
3 * Test of the overlay used for moved pictures, test more closed to real life. *
4 * Running trojan moose :) Coded by Mike Gorchak. *
5 * *
6 ********************************************************************************/
7
8 #include <stdlib.h>
9 #include <stdio.h>
10 #include <string.h>
11
12 #include "SDL.h"
13
14 #define MOOSEPIC_W 64
15 #define MOOSEPIC_H 88
16
17 #define MOOSEFRAME_SIZE (MOOSEPIC_W * MOOSEPIC_H)
18 #define MOOSEFRAMES_COUNT 10
19
20 SDL_Color MooseColors[84]={
21 { 49, 49, 49}, { 66, 24, 0}, { 66, 33, 0}, { 66, 66, 66},
22 { 66, 115, 49}, { 74, 33, 0}, { 74, 41, 16}, { 82, 33, 8},
23 { 82, 41, 8}, { 82, 49, 16}, { 82, 82, 82}, { 90, 41, 8},
24 { 90, 41, 16}, { 90, 57, 24}, { 99, 49, 16}, { 99, 66, 24},
25 { 99, 66, 33}, { 99, 74, 33}, {107, 57, 24}, {107, 82, 41},
26 {115, 57, 33}, {115, 66, 33}, {115, 66, 41}, {115, 74, 0},
27 {115, 90, 49}, {115, 115, 115}, {123, 82, 0}, {123, 99, 57},
28 {132, 66, 41}, {132, 74, 41}, {132, 90, 8}, {132, 99, 33},
29 {132, 99, 66}, {132, 107, 66}, {140, 74, 49}, {140, 99, 16},
30 {140, 107, 74}, {140, 115, 74}, {148, 107, 24}, {148, 115, 82},
31 {148, 123, 74}, {148, 123, 90}, {156, 115, 33}, {156, 115, 90},
32 {156, 123, 82}, {156, 132, 82}, {156, 132, 99}, {156, 156, 156},
33 {165, 123, 49}, {165, 123, 90}, {165, 132, 82}, {165, 132, 90},
34 {165, 132, 99}, {165, 140, 90}, {173, 132, 57}, {173, 132, 99},
35 {173, 140, 107}, {173, 140, 115}, {173, 148, 99}, {173, 173, 173},
36 {181, 140, 74}, {181, 148, 115}, {181, 148, 123}, {181, 156, 107},
37 {189, 148, 123}, {189, 156, 82}, {189, 156, 123}, {189, 156, 132},
38 {189, 189, 189}, {198, 156, 123}, {198, 165, 132}, {206, 165, 99},
39 {206, 165, 132}, {206, 173, 140}, {206, 206, 206}, {214, 173, 115},
40 {214, 173, 140}, {222, 181, 148}, {222, 189, 132}, {222, 189, 156},
41 {222, 222, 222}, {231, 198, 165}, {231, 231, 231}, {239, 206, 173}
42 };
43
44
45 /* Call this instead of exit(), so we can clean up SDL: atexit() is evil. */
quit(int rc)46 static void quit(int rc)
47 {
48 SDL_Quit();
49 exit(rc);
50 }
51
52 /* All RGB2YUV conversion code and some other parts of code has been taken from testoverlay.c */
53
54 /* NOTE: These RGB conversion functions are not intended for speed,
55 only as examples.
56 */
57
RGBtoYUV(Uint8 * rgb,int * yuv,int monochrome,int luminance)58 void RGBtoYUV(Uint8 *rgb, int *yuv, int monochrome, int luminance)
59 {
60 if (monochrome)
61 {
62 #if 1 /* these are the two formulas that I found on the FourCC site... */
63 yuv[0] = 0.299*rgb[0] + 0.587*rgb[1] + 0.114*rgb[2];
64 yuv[1] = 128;
65 yuv[2] = 128;
66 #else
67 yuv[0] = (0.257 * rgb[0]) + (0.504 * rgb[1]) + (0.098 * rgb[2]) + 16;
68 yuv[1] = 128;
69 yuv[2] = 128;
70 #endif
71 }
72 else
73 {
74 #if 1 /* these are the two formulas that I found on the FourCC site... */
75 yuv[0] = 0.299*rgb[0] + 0.587*rgb[1] + 0.114*rgb[2];
76 yuv[1] = (rgb[2]-yuv[0])*0.565 + 128;
77 yuv[2] = (rgb[0]-yuv[0])*0.713 + 128;
78 #else
79 yuv[0] = (0.257 * rgb[0]) + (0.504 * rgb[1]) + (0.098 * rgb[2]) + 16;
80 yuv[1] = 128 - (0.148 * rgb[0]) - (0.291 * rgb[1]) + (0.439 * rgb[2]);
81 yuv[2] = 128 + (0.439 * rgb[0]) - (0.368 * rgb[1]) - (0.071 * rgb[2]);
82 #endif
83 }
84
85 if (luminance!=100)
86 {
87 yuv[0]=yuv[0]*luminance/100;
88 if (yuv[0]>255)
89 yuv[0]=255;
90 }
91 }
92
ConvertRGBtoYV12(SDL_Surface * s,SDL_Overlay * o,int monochrome,int luminance)93 void ConvertRGBtoYV12(SDL_Surface *s, SDL_Overlay *o, int monochrome, int luminance)
94 {
95 int x,y;
96 int yuv[3];
97 Uint8 *p,*op[3];
98
99 SDL_LockSurface(s);
100 SDL_LockYUVOverlay(o);
101
102 /* Convert */
103 for(y=0; y<s->h && y<o->h; y++)
104 {
105 p=((Uint8 *) s->pixels)+s->pitch*y;
106 op[0]=o->pixels[0]+o->pitches[0]*y;
107 op[1]=o->pixels[1]+o->pitches[1]*(y/2);
108 op[2]=o->pixels[2]+o->pitches[2]*(y/2);
109 for(x=0; x<s->w && x<o->w; x++)
110 {
111 RGBtoYUV(p, yuv, monochrome, luminance);
112 *(op[0]++)=yuv[0];
113 if(x%2==0 && y%2==0)
114 {
115 *(op[1]++)=yuv[2];
116 *(op[2]++)=yuv[1];
117 }
118 p+=s->format->BytesPerPixel;
119 }
120 }
121
122 SDL_UnlockYUVOverlay(o);
123 SDL_UnlockSurface(s);
124 }
125
ConvertRGBtoIYUV(SDL_Surface * s,SDL_Overlay * o,int monochrome,int luminance)126 void ConvertRGBtoIYUV(SDL_Surface *s, SDL_Overlay *o, int monochrome, int luminance)
127 {
128 int x,y;
129 int yuv[3];
130 Uint8 *p,*op[3];
131
132 SDL_LockSurface(s);
133 SDL_LockYUVOverlay(o);
134
135 /* Convert */
136 for(y=0; y<s->h && y<o->h; y++)
137 {
138 p=((Uint8 *) s->pixels)+s->pitch*y;
139 op[0]=o->pixels[0]+o->pitches[0]*y;
140 op[1]=o->pixels[1]+o->pitches[1]*(y/2);
141 op[2]=o->pixels[2]+o->pitches[2]*(y/2);
142 for(x=0; x<s->w && x<o->w; x++)
143 {
144 RGBtoYUV(p,yuv, monochrome, luminance);
145 *(op[0]++)=yuv[0];
146 if(x%2==0 && y%2==0)
147 {
148 *(op[1]++)=yuv[1];
149 *(op[2]++)=yuv[2];
150 }
151 p+=s->format->BytesPerPixel;
152 }
153 }
154
155 SDL_UnlockYUVOverlay(o);
156 SDL_UnlockSurface(s);
157 }
158
ConvertRGBtoUYVY(SDL_Surface * s,SDL_Overlay * o,int monochrome,int luminance)159 void ConvertRGBtoUYVY(SDL_Surface *s, SDL_Overlay *o, int monochrome, int luminance)
160 {
161 int x,y;
162 int yuv[3];
163 Uint8 *p,*op;
164
165 SDL_LockSurface(s);
166 SDL_LockYUVOverlay(o);
167
168 for(y=0; y<s->h && y<o->h; y++)
169 {
170 p=((Uint8 *) s->pixels)+s->pitch*y;
171 op=o->pixels[0]+o->pitches[0]*y;
172 for(x=0; x<s->w && x<o->w; x++)
173 {
174 RGBtoYUV(p, yuv, monochrome, luminance);
175 if(x%2==0)
176 {
177 *(op++)=yuv[1];
178 *(op++)=yuv[0];
179 *(op++)=yuv[2];
180 }
181 else
182 *(op++)=yuv[0];
183
184 p+=s->format->BytesPerPixel;
185 }
186 }
187
188 SDL_UnlockYUVOverlay(o);
189 SDL_UnlockSurface(s);
190 }
191
ConvertRGBtoYVYU(SDL_Surface * s,SDL_Overlay * o,int monochrome,int luminance)192 void ConvertRGBtoYVYU(SDL_Surface *s, SDL_Overlay *o, int monochrome, int luminance)
193 {
194 int x,y;
195 int yuv[3];
196 Uint8 *p,*op;
197
198 SDL_LockSurface(s);
199 SDL_LockYUVOverlay(o);
200
201 for(y=0; y<s->h && y<o->h; y++)
202 {
203 p=((Uint8 *) s->pixels)+s->pitch*y;
204 op=o->pixels[0]+o->pitches[0]*y;
205 for(x=0; x<s->w && x<o->w; x++)
206 {
207 RGBtoYUV(p,yuv, monochrome, luminance);
208 if(x%2==0)
209 {
210 *(op++)=yuv[0];
211 *(op++)=yuv[2];
212 op[1]=yuv[1];
213 }
214 else
215 {
216 *op=yuv[0];
217 op+=2;
218 }
219
220 p+=s->format->BytesPerPixel;
221 }
222 }
223
224 SDL_UnlockYUVOverlay(o);
225 SDL_UnlockSurface(s);
226 }
227
ConvertRGBtoYUY2(SDL_Surface * s,SDL_Overlay * o,int monochrome,int luminance)228 void ConvertRGBtoYUY2(SDL_Surface *s, SDL_Overlay *o, int monochrome, int luminance)
229 {
230 int x,y;
231 int yuv[3];
232 Uint8 *p,*op;
233
234 SDL_LockSurface(s);
235 SDL_LockYUVOverlay(o);
236
237 for(y=0; y<s->h && y<o->h; y++)
238 {
239 p=((Uint8 *) s->pixels)+s->pitch*y;
240 op=o->pixels[0]+o->pitches[0]*y;
241 for(x=0; x<s->w && x<o->w; x++)
242 {
243 RGBtoYUV(p,yuv, monochrome, luminance);
244 if(x%2==0)
245 {
246 *(op++)=yuv[0];
247 *(op++)=yuv[1];
248 op[1]=yuv[2];
249 }
250 else
251 {
252 *op=yuv[0];
253 op+=2;
254 }
255
256 p+=s->format->BytesPerPixel;
257 }
258 }
259
260 SDL_UnlockYUVOverlay(o);
261 SDL_UnlockSurface(s);
262 }
263
PrintUsage(char * argv0)264 static void PrintUsage(char *argv0)
265 {
266 fprintf(stderr, "Usage: %s [arg] [arg] [arg] ...\n", argv0);
267 fprintf(stderr, "\n");
268 fprintf(stderr, "Where 'arg' is any of the following options:\n");
269 fprintf(stderr, "\n");
270 fprintf(stderr, " -fps <frames per second>\n");
271 fprintf(stderr, " -format <fmt> (one of the: YV12, IYUV, YUY2, UYVY, YVYU)\n");
272 fprintf(stderr, " -scale <scale factor> (initial scale of the overlay)\n");
273 fprintf(stderr, " -help (shows this help)\n");
274 fprintf(stderr, "\n");
275 fprintf(stderr, "Press ESC to exit, or SPACE to freeze the movie while application running.\n");
276 fprintf(stderr, "\n");
277 }
278
main(int argc,char ** argv)279 int main(int argc, char **argv)
280 {
281 Uint8* RawMooseData;
282 SDL_RWops* handle;
283 SDL_Surface* screen;
284 SDL_Surface* MooseFrame[MOOSEFRAMES_COUNT];
285 SDL_Overlay* overlay;
286 SDL_Rect overlayrect;
287 SDL_Event event;
288 Uint32 lastftick;
289 int paused=0;
290 int resized=0;
291 int i;
292 int fps=12;
293 int fpsdelay;
294 int overlay_format=SDL_YUY2_OVERLAY;
295 int scale=5;
296
297 if (SDL_Init(SDL_INIT_VIDEO | SDL_INIT_NOPARACHUTE) < 0)
298 {
299 fprintf(stderr, "Couldn't initialize SDL: %s\n", SDL_GetError());
300 return 3;
301 }
302
303 while ( argc > 1 )
304 {
305 if (strcmp(argv[1], "-fps")== 0)
306 {
307 if (argv[2])
308 {
309 fps = atoi(argv[2]);
310 if (fps==0)
311 {
312 fprintf(stderr, "The -fps option requires an argument [from 1 to 1000], default is 12.\n");
313 quit(10);
314 }
315 if ((fps<0) || (fps>1000))
316 {
317 fprintf(stderr, "The -fps option must be in range from 1 to 1000, default is 12.\n");
318 quit(10);
319 }
320 argv += 2;
321 argc -= 2;
322 }
323 else
324 {
325 fprintf(stderr, "The -fps option requires an argument [from 1 to 1000], default is 12.\n");
326 quit(10);
327 }
328 } else
329 if (strcmp(argv[1], "-format") == 0)
330 {
331 if (argv[2])
332 {
333 if (!strcmp(argv[2],"YV12"))
334 overlay_format = SDL_YV12_OVERLAY;
335 else if(!strcmp(argv[2],"IYUV"))
336 overlay_format = SDL_IYUV_OVERLAY;
337 else if(!strcmp(argv[2],"YUY2"))
338 overlay_format = SDL_YUY2_OVERLAY;
339 else if(!strcmp(argv[2],"UYVY"))
340 overlay_format = SDL_UYVY_OVERLAY;
341 else if(!strcmp(argv[2],"YVYU"))
342 overlay_format = SDL_YVYU_OVERLAY;
343 else
344 {
345 fprintf(stderr, "The -format option %s is not recognized, see help for info.\n", argv[2]);
346 quit(10);
347 }
348 argv += 2;
349 argc -= 2;
350 }
351 else
352 {
353 fprintf(stderr, "The -format option requires an argument, default is YUY2.\n");
354 quit(10);
355 }
356 } else
357 if (strcmp(argv[1], "-scale") == 0)
358 {
359 if (argv[2])
360 {
361 scale = atoi(argv[2]);
362 if (scale==0)
363 {
364 fprintf(stderr, "The -scale option requires an argument [from 1 to 50], default is 5.\n");
365 quit(10);
366 }
367 if ((scale<0) || (scale>50))
368 {
369 fprintf(stderr, "The -scale option must be in range from 1 to 50, default is 5.\n");
370 quit(10);
371 }
372 argv += 2;
373 argc -= 2;
374 }
375 else
376 {
377 fprintf(stderr, "The -fps option requires an argument [from 1 to 1000], default is 12.\n");
378 quit(10);
379 }
380 } else
381 if ((strcmp(argv[1], "-help") == 0 ) || (strcmp(argv[1], "-h") == 0))
382 {
383 PrintUsage(argv[0]);
384 quit(0);
385 } else
386 {
387 fprintf(stderr, "Unrecognized option: %s.\n", argv[1]);
388 quit(10);
389 }
390 break;
391 }
392
393 RawMooseData=(Uint8*)malloc(MOOSEFRAME_SIZE * MOOSEFRAMES_COUNT);
394 if (RawMooseData==NULL)
395 {
396 fprintf(stderr, "Can't allocate memory for movie !\n");
397 free(RawMooseData);
398 quit(1);
399 }
400
401 /* load the trojan moose images */
402 handle=SDL_RWFromFile("moose.dat", "rb");
403 if (handle==NULL)
404 {
405 fprintf(stderr, "Can't find the file moose.dat !\n");
406 free(RawMooseData);
407 quit(2);
408 }
409
410 SDL_RWread(handle, RawMooseData, MOOSEFRAME_SIZE, MOOSEFRAMES_COUNT);
411
412 SDL_RWclose(handle);
413
414 /* Set video mode */
415 if ( (screen=SDL_SetVideoMode(MOOSEPIC_W*scale, MOOSEPIC_H*scale, 0, SDL_RESIZABLE | SDL_SWSURFACE)) == NULL )
416 {
417 fprintf(stderr, "Couldn't set video mode: %s\n", SDL_GetError());
418 free(RawMooseData);
419 quit(4);
420 }
421
422 /* Set the window manager title bar */
423 SDL_WM_SetCaption("SDL test overlay: running moose", "testoverlay2");
424
425 for (i=0; i<MOOSEFRAMES_COUNT; i++)
426 {
427 MooseFrame[i]=SDL_CreateRGBSurfaceFrom(RawMooseData+i*MOOSEFRAME_SIZE, MOOSEPIC_W,
428 MOOSEPIC_H, 8, MOOSEPIC_W, 0, 0, 0, 0);
429 if (MooseFrame[i]==NULL)
430 {
431 fprintf(stderr, "Couldn't create SDL_Surfaces:%s\n", SDL_GetError());
432 free(RawMooseData);
433 quit(5);
434 }
435 SDL_SetColors(MooseFrame[i], MooseColors, 0, 84);
436
437 {
438 SDL_Surface *newsurf;
439 SDL_PixelFormat format;
440
441 format.palette=NULL;
442 format.BitsPerPixel=32;
443 format.BytesPerPixel=4;
444 #if SDL_BYTEORDER == SDL_LIL_ENDIAN
445 format.Rshift=0;
446 format.Gshift=8;
447 format.Bshift=16;
448 #else
449 format.Rshift=24;
450 format.Gshift=16;
451 format.Bshift=8;
452 #endif
453 format.Ashift=0;
454 format.Rmask=0xff<<format.Rshift;
455 format.Gmask=0xff<<format.Gshift;
456 format.Bmask=0xff<<format.Bshift;
457 format.Amask=0;
458 format.Rloss=0;
459 format.Gloss=0;
460 format.Bloss=0;
461 format.Aloss=8;
462 format.colorkey=0;
463 format.alpha=0;
464
465 newsurf=SDL_ConvertSurface(MooseFrame[i], &format, SDL_SWSURFACE);
466 if(!newsurf)
467 {
468 fprintf(stderr, "Couldn't convert picture to 32bits RGB: %s\n", SDL_GetError());
469 quit(6);
470 }
471 SDL_FreeSurface(MooseFrame[i]);
472 MooseFrame[i]=newsurf;
473 }
474 }
475
476 free(RawMooseData);
477
478 overlay=SDL_CreateYUVOverlay(MOOSEPIC_W, MOOSEPIC_H, overlay_format, screen);
479 if (!overlay)
480 {
481 fprintf(stderr, "Couldn't create overlay: %s\n", SDL_GetError());
482 quit(7);
483 }
484
485 printf("Created %dx%dx%d %s %s overlay\n",overlay->w,overlay->h,overlay->planes,
486 overlay->hw_overlay?"hardware":"software",
487 overlay->format==SDL_YV12_OVERLAY?"YV12":
488 overlay->format==SDL_IYUV_OVERLAY?"IYUV":
489 overlay->format==SDL_YUY2_OVERLAY?"YUY2":
490 overlay->format==SDL_UYVY_OVERLAY?"UYVY":
491 overlay->format==SDL_YVYU_OVERLAY?"YVYU":
492 "Unknown");
493
494 for(i=0; i<overlay->planes; i++)
495 {
496 printf(" plane %d: pitch=%d\n", i, overlay->pitches[i]);
497 }
498
499 overlayrect.x=0;
500 overlayrect.y=0;
501 overlayrect.w=MOOSEPIC_W*scale;
502 overlayrect.h=MOOSEPIC_H*scale;
503
504 /* set the start frame */
505 i=0;
506 fpsdelay=1000/fps;
507
508 /* Ignore key up events, they don't even get filtered */
509 SDL_EventState(SDL_KEYUP, SDL_IGNORE);
510
511 lastftick=SDL_GetTicks();
512
513 /* Loop, waiting for QUIT or RESIZE */
514 while (1)
515 {
516 if (SDL_PollEvent(&event))
517 {
518 switch (event.type)
519 {
520 case SDL_VIDEORESIZE:
521 screen=SDL_SetVideoMode(event.resize.w, event.resize.h, 0, SDL_RESIZABLE | SDL_SWSURFACE);
522 overlayrect.w=event.resize.w;
523 overlayrect.h=event.resize.h;
524 if (paused)
525 {
526 resized=1;
527 }
528 break;
529 case SDL_MOUSEBUTTONDOWN:
530 overlayrect.x = event.button.x - overlayrect.w/2;
531 overlayrect.y = event.button.y - overlayrect.h/2;
532 break;
533 case SDL_KEYDOWN:
534 if (event.key.keysym.sym == SDLK_SPACE)
535 {
536 paused=!paused;
537 break;
538 }
539 if (event.key.keysym.sym != SDLK_ESCAPE)
540 {
541 break;
542 }
543 case SDL_QUIT:
544 SDL_FreeYUVOverlay(overlay);
545 for (i=0; i<MOOSEFRAMES_COUNT; i++)
546 {
547 SDL_FreeSurface(MooseFrame[i]);
548 }
549 quit(0);
550 }
551 }
552
553 if ((!paused)||(resized))
554 {
555 if (((SDL_GetTicks()-lastftick)>fpsdelay)||(resized))
556 {
557 lastftick=SDL_GetTicks();
558
559 switch (overlay_format)
560 {
561 case SDL_YUY2_OVERLAY:
562 ConvertRGBtoYUY2(MooseFrame[i], overlay, 0, 100);
563 break;
564 case SDL_YV12_OVERLAY:
565 ConvertRGBtoYV12(MooseFrame[i], overlay, 0, 100);
566 break;
567 case SDL_UYVY_OVERLAY:
568 ConvertRGBtoUYVY(MooseFrame[i], overlay, 0, 100);
569 break;
570 case SDL_YVYU_OVERLAY:
571 ConvertRGBtoYVYU(MooseFrame[i], overlay, 0, 100);
572 break;
573 case SDL_IYUV_OVERLAY:
574 ConvertRGBtoIYUV(MooseFrame[i], overlay, 0, 100);
575 break;
576 }
577
578 SDL_DisplayYUVOverlay(overlay, &overlayrect);
579 if (!resized)
580 {
581 i++;
582 if (i==10)
583 {
584 i=0;
585 }
586 }
587 else
588 {
589 resized=0;
590 }
591 }
592 }
593 /* kind of timeslice to OS */
594 SDL_Delay(1);
595 }
596
597 SDL_Quit();
598 return 0;
599 }
600
601