test/testoverlay2.c
author Ryan C. Gordon <icculus@icculus.org>
Fri, 15 Jun 2007 07:19:05 +0000
branchSDL-1.2
changeset 3965 19d708301fab
parent 1642 f1211a4b7380
child 1662 782fd950bd46
child 1895 c121d94672cb
permissions -rw-r--r--
Make sure __inline__ isn't defined before #defining it, since some compilers
(like Sun Studio) have this as a predefined macro.

Fixes Bugzilla #434.
     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. */
    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 
    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 
    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 
   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 
   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 
   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 
   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 
   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 
   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