| |
| /* Bring up a window and play with it */ |
| |
| #include <stdlib.h> |
| #include <stdio.h> |
| #include <string.h> |
| |
| #define BENCHMARK_SDL |
| |
| #define NOTICE(X) printf("%s", X); |
| |
| #define WINDOW_WIDTH 640 |
| #define WINDOW_HEIGHT 480 |
| |
| #include "SDL.h" |
| |
| SDL_Surface *screen, *pic; |
| SDL_Overlay *overlay; |
| int scale; |
| int monochrome; |
| int luminance; |
| int w, h; |
| |
| /* Call this instead of exit(), so we can clean up SDL: atexit() is evil. */ |
| static void |
| quit(int rc) |
| { |
| SDL_Quit(); |
| exit(rc); |
| } |
| |
| /* NOTE: These RGB conversion functions are not intended for speed, |
| only as examples. |
| */ |
| |
| void |
| RGBtoYUV(Uint8 * rgb, int *yuv, int monochrome, int luminance) |
| { |
| if (monochrome) { |
| #if 1 /* these are the two formulas that I found on the FourCC site... */ |
| yuv[0] = 0.299 * rgb[0] + 0.587 * rgb[1] + 0.114 * rgb[2]; |
| yuv[1] = 128; |
| yuv[2] = 128; |
| #else |
| yuv[0] = (0.257 * rgb[0]) + (0.504 * rgb[1]) + (0.098 * rgb[2]) + 16; |
| yuv[1] = 128; |
| yuv[2] = 128; |
| #endif |
| } else { |
| #if 1 /* these are the two formulas that I found on the FourCC site... */ |
| yuv[0] = 0.299 * rgb[0] + 0.587 * rgb[1] + 0.114 * rgb[2]; |
| yuv[1] = (rgb[2] - yuv[0]) * 0.565 + 128; |
| yuv[2] = (rgb[0] - yuv[0]) * 0.713 + 128; |
| #else |
| yuv[0] = (0.257 * rgb[0]) + (0.504 * rgb[1]) + (0.098 * rgb[2]) + 16; |
| yuv[1] = 128 - (0.148 * rgb[0]) - (0.291 * rgb[1]) + (0.439 * rgb[2]); |
| yuv[2] = 128 + (0.439 * rgb[0]) - (0.368 * rgb[1]) - (0.071 * rgb[2]); |
| #endif |
| } |
| |
| if (luminance != 100) { |
| yuv[0] = yuv[0] * luminance / 100; |
| if (yuv[0] > 255) |
| yuv[0] = 255; |
| } |
| |
| /* clamp values...if you need to, we don't seem to have a need */ |
| /* |
| for(i=0;i<3;i++) |
| { |
| if(yuv[i]<0) |
| yuv[i]=0; |
| if(yuv[i]>255) |
| yuv[i]=255; |
| } |
| */ |
| } |
| |
| void |
| ConvertRGBtoYV12(SDL_Surface * s, SDL_Overlay * o, int monochrome, |
| int luminance) |
| { |
| int x, y; |
| int yuv[3]; |
| Uint8 *p, *op[3]; |
| |
| SDL_LockSurface(s); |
| SDL_LockYUVOverlay(o); |
| |
| /* Black initialization */ |
| /* |
| memset(o->pixels[0],0,o->pitches[0]*o->h); |
| memset(o->pixels[1],128,o->pitches[1]*((o->h+1)/2)); |
| memset(o->pixels[2],128,o->pitches[2]*((o->h+1)/2)); |
| */ |
| |
| /* Convert */ |
| for (y = 0; y < s->h && y < o->h; y++) { |
| p = ((Uint8 *) s->pixels) + s->pitch * y; |
| op[0] = o->pixels[0] + o->pitches[0] * y; |
| op[1] = o->pixels[1] + o->pitches[1] * (y / 2); |
| op[2] = o->pixels[2] + o->pitches[2] * (y / 2); |
| for (x = 0; x < s->w && x < o->w; x++) { |
| RGBtoYUV(p, yuv, monochrome, luminance); |
| *(op[0]++) = yuv[0]; |
| if (x % 2 == 0 && y % 2 == 0) { |
| *(op[1]++) = yuv[2]; |
| *(op[2]++) = yuv[1]; |
| } |
| p += s->format->BytesPerPixel; |
| } |
| } |
| |
| SDL_UnlockYUVOverlay(o); |
| SDL_UnlockSurface(s); |
| } |
| |
| void |
| ConvertRGBtoIYUV(SDL_Surface * s, SDL_Overlay * o, int monochrome, |
| int luminance) |
| { |
| int x, y; |
| int yuv[3]; |
| Uint8 *p, *op[3]; |
| |
| SDL_LockSurface(s); |
| SDL_LockYUVOverlay(o); |
| |
| /* Black initialization */ |
| /* |
| memset(o->pixels[0],0,o->pitches[0]*o->h); |
| memset(o->pixels[1],128,o->pitches[1]*((o->h+1)/2)); |
| memset(o->pixels[2],128,o->pitches[2]*((o->h+1)/2)); |
| */ |
| |
| /* Convert */ |
| for (y = 0; y < s->h && y < o->h; y++) { |
| p = ((Uint8 *) s->pixels) + s->pitch * y; |
| op[0] = o->pixels[0] + o->pitches[0] * y; |
| op[1] = o->pixels[1] + o->pitches[1] * (y / 2); |
| op[2] = o->pixels[2] + o->pitches[2] * (y / 2); |
| for (x = 0; x < s->w && x < o->w; x++) { |
| RGBtoYUV(p, yuv, monochrome, luminance); |
| *(op[0]++) = yuv[0]; |
| if (x % 2 == 0 && y % 2 == 0) { |
| *(op[1]++) = yuv[1]; |
| *(op[2]++) = yuv[2]; |
| } |
| p += s->format->BytesPerPixel; |
| } |
| } |
| |
| SDL_UnlockYUVOverlay(o); |
| SDL_UnlockSurface(s); |
| } |
| |
| void |
| ConvertRGBtoUYVY(SDL_Surface * s, SDL_Overlay * o, int monochrome, |
| int luminance) |
| { |
| int x, y; |
| int yuv[3]; |
| Uint8 *p, *op; |
| |
| SDL_LockSurface(s); |
| SDL_LockYUVOverlay(o); |
| |
| for (y = 0; y < s->h && y < o->h; y++) { |
| p = ((Uint8 *) s->pixels) + s->pitch * y; |
| op = o->pixels[0] + o->pitches[0] * y; |
| for (x = 0; x < s->w && x < o->w; x++) { |
| RGBtoYUV(p, yuv, monochrome, luminance); |
| if (x % 2 == 0) { |
| *(op++) = yuv[1]; |
| *(op++) = yuv[0]; |
| *(op++) = yuv[2]; |
| } else |
| *(op++) = yuv[0]; |
| |
| p += s->format->BytesPerPixel; |
| } |
| } |
| |
| SDL_UnlockYUVOverlay(o); |
| SDL_UnlockSurface(s); |
| } |
| |
| void |
| ConvertRGBtoYVYU(SDL_Surface * s, SDL_Overlay * o, int monochrome, |
| int luminance) |
| { |
| int x, y; |
| int yuv[3]; |
| Uint8 *p, *op; |
| |
| SDL_LockSurface(s); |
| SDL_LockYUVOverlay(o); |
| |
| for (y = 0; y < s->h && y < o->h; y++) { |
| p = ((Uint8 *) s->pixels) + s->pitch * y; |
| op = o->pixels[0] + o->pitches[0] * y; |
| for (x = 0; x < s->w && x < o->w; x++) { |
| RGBtoYUV(p, yuv, monochrome, luminance); |
| if (x % 2 == 0) { |
| *(op++) = yuv[0]; |
| *(op++) = yuv[2]; |
| op[1] = yuv[1]; |
| } else { |
| *op = yuv[0]; |
| op += 2; |
| } |
| |
| p += s->format->BytesPerPixel; |
| } |
| } |
| |
| SDL_UnlockYUVOverlay(o); |
| SDL_UnlockSurface(s); |
| } |
| |
| void |
| ConvertRGBtoYUY2(SDL_Surface * s, SDL_Overlay * o, int monochrome, |
| int luminance) |
| { |
| int x, y; |
| int yuv[3]; |
| Uint8 *p, *op; |
| |
| SDL_LockSurface(s); |
| SDL_LockYUVOverlay(o); |
| |
| for (y = 0; y < s->h && y < o->h; y++) { |
| p = ((Uint8 *) s->pixels) + s->pitch * y; |
| op = o->pixels[0] + o->pitches[0] * y; |
| for (x = 0; x < s->w && x < o->w; x++) { |
| RGBtoYUV(p, yuv, monochrome, luminance); |
| if (x % 2 == 0) { |
| *(op++) = yuv[0]; |
| *(op++) = yuv[1]; |
| op[1] = yuv[2]; |
| } else { |
| *op = yuv[0]; |
| op += 2; |
| } |
| |
| p += s->format->BytesPerPixel; |
| } |
| } |
| |
| SDL_UnlockYUVOverlay(o); |
| SDL_UnlockSurface(s); |
| } |
| |
| void |
| Draw() |
| { |
| SDL_Rect rect; |
| int i; |
| int disp; |
| |
| if (!scale) { |
| rect.w = overlay->w; |
| rect.h = overlay->h; |
| for (i = 0; i < h - rect.h && i < w - rect.w; i++) { |
| rect.x = i; |
| rect.y = i; |
| SDL_DisplayYUVOverlay(overlay, &rect); |
| } |
| } else { |
| rect.w = overlay->w / 2; |
| rect.h = overlay->h / 2; |
| rect.x = (w - rect.w) / 2; |
| rect.y = (h - rect.h) / 2; |
| disp = rect.y - 1; |
| for (i = 0; i < disp; i++) { |
| rect.w += 2; |
| rect.h += 2; |
| rect.x--; |
| rect.y--; |
| SDL_DisplayYUVOverlay(overlay, &rect); |
| } |
| } |
| printf("Displayed %d times.\n", i); |
| } |
| |
| static void |
| PrintUsage(char *argv0) |
| { |
| fprintf(stderr, "Usage: %s [arg] [arg] [arg] ...\n", argv0); |
| fprintf(stderr, "Where 'arg' is one of:\n"); |
| fprintf(stderr, " -delay <seconds>\n"); |
| fprintf(stderr, " -width <pixels>\n"); |
| fprintf(stderr, " -height <pixels>\n"); |
| fprintf(stderr, " -bpp <bits>\n"); |
| fprintf(stderr, |
| " -format <fmt> (one of the: YV12, IYUV, YUY2, UYVY, YVYU)\n"); |
| fprintf(stderr, " -hw\n"); |
| fprintf(stderr, " -flip\n"); |
| fprintf(stderr, |
| " -scale (test scaling features, from 50%% upto window size)\n"); |
| fprintf(stderr, " -mono (use monochromatic RGB2YUV conversion)\n"); |
| fprintf(stderr, |
| " -lum <perc> (use luminance correction during RGB2YUV conversion,\n"); |
| fprintf(stderr, |
| " from 0%% to unlimited, normal is 100%%)\n"); |
| fprintf(stderr, " -help (shows this help)\n"); |
| fprintf(stderr, " -fullscreen (test overlay in fullscreen mode)\n"); |
| } |
| |
| int |
| main(int argc, char **argv) |
| { |
| char *argv0 = argv[0]; |
| int flip; |
| int delay; |
| int desired_bpp; |
| Uint32 video_flags, overlay_format; |
| char *bmpfile; |
| #ifdef BENCHMARK_SDL |
| Uint32 then, now; |
| #endif |
| int i; |
| |
| /* Set default options and check command-line */ |
| flip = 0; |
| scale = 0; |
| monochrome = 0; |
| luminance = 100; |
| delay = 1; |
| w = WINDOW_WIDTH; |
| h = WINDOW_HEIGHT; |
| desired_bpp = 0; |
| video_flags = 0; |
| overlay_format = SDL_YV12_OVERLAY; |
| |
| while (argc > 1) { |
| if (strcmp(argv[1], "-delay") == 0) { |
| if (argv[2]) { |
| delay = atoi(argv[2]); |
| argv += 2; |
| argc -= 2; |
| } else { |
| fprintf(stderr, "The -delay option requires an argument\n"); |
| return (1); |
| } |
| } else if (strcmp(argv[1], "-width") == 0) { |
| if (argv[2] && ((w = atoi(argv[2])) > 0)) { |
| argv += 2; |
| argc -= 2; |
| } else { |
| fprintf(stderr, "The -width option requires an argument\n"); |
| return (1); |
| } |
| } else if (strcmp(argv[1], "-height") == 0) { |
| if (argv[2] && ((h = atoi(argv[2])) > 0)) { |
| argv += 2; |
| argc -= 2; |
| } else { |
| fprintf(stderr, "The -height option requires an argument\n"); |
| return (1); |
| } |
| } else if (strcmp(argv[1], "-bpp") == 0) { |
| if (argv[2]) { |
| desired_bpp = atoi(argv[2]); |
| argv += 2; |
| argc -= 2; |
| } else { |
| fprintf(stderr, "The -bpp option requires an argument\n"); |
| return (1); |
| } |
| } else if (strcmp(argv[1], "-lum") == 0) { |
| if (argv[2]) { |
| luminance = atoi(argv[2]); |
| argv += 2; |
| argc -= 2; |
| } else { |
| fprintf(stderr, "The -lum option requires an argument\n"); |
| return (1); |
| } |
| } else if (strcmp(argv[1], "-format") == 0) { |
| if (argv[2]) { |
| if (!strcmp(argv[2], "YV12")) |
| overlay_format = SDL_YV12_OVERLAY; |
| else if (!strcmp(argv[2], "IYUV")) |
| overlay_format = SDL_IYUV_OVERLAY; |
| else if (!strcmp(argv[2], "YUY2")) |
| overlay_format = SDL_YUY2_OVERLAY; |
| else if (!strcmp(argv[2], "UYVY")) |
| overlay_format = SDL_UYVY_OVERLAY; |
| else if (!strcmp(argv[2], "YVYU")) |
| overlay_format = SDL_YVYU_OVERLAY; |
| else { |
| fprintf(stderr, |
| "The -format option %s is not recognized\n", |
| argv[2]); |
| return (1); |
| } |
| argv += 2; |
| argc -= 2; |
| } else { |
| fprintf(stderr, "The -format option requires an argument\n"); |
| return (1); |
| } |
| } else if (strcmp(argv[1], "-hw") == 0) { |
| video_flags |= SDL_HWSURFACE; |
| argv += 1; |
| argc -= 1; |
| } else if (strcmp(argv[1], "-flip") == 0) { |
| video_flags |= SDL_DOUBLEBUF; |
| argv += 1; |
| argc -= 1; |
| } else if (strcmp(argv[1], "-scale") == 0) { |
| scale = 1; |
| argv += 1; |
| argc -= 1; |
| } else if (strcmp(argv[1], "-mono") == 0) { |
| monochrome = 1; |
| argv += 1; |
| argc -= 1; |
| } else if ((strcmp(argv[1], "-help") == 0) |
| || (strcmp(argv[1], "-h") == 0)) { |
| PrintUsage(argv0); |
| return (1); |
| } else if (strcmp(argv[1], "-fullscreen") == 0) { |
| video_flags |= SDL_FULLSCREEN; |
| argv += 1; |
| argc -= 1; |
| } else |
| break; |
| } |
| if (SDL_Init(SDL_INIT_VIDEO) < 0) { |
| fprintf(stderr, "Couldn't initialize SDL: %s\n", SDL_GetError()); |
| return (1); |
| } |
| |
| /* Initialize the display */ |
| screen = SDL_SetVideoMode(w, h, desired_bpp, video_flags); |
| if (screen == NULL) { |
| fprintf(stderr, "Couldn't set %dx%dx%d video mode: %s\n", |
| w, h, desired_bpp, SDL_GetError()); |
| quit(1); |
| } |
| printf("Set%s %dx%dx%d mode\n", |
| screen->flags & SDL_FULLSCREEN ? " fullscreen" : "", |
| screen->w, screen->h, screen->format->BitsPerPixel); |
| printf("(video surface located in %s memory)\n", |
| (screen->flags & SDL_HWSURFACE) ? "video" : "system"); |
| if (screen->flags & SDL_DOUBLEBUF) { |
| printf("Double-buffering enabled\n"); |
| flip = 1; |
| } |
| |
| /* Set the window manager title bar */ |
| SDL_WM_SetCaption("SDL test overlay", "testoverlay"); |
| |
| /* Load picture */ |
| bmpfile = (argv[1] ? argv[1] : "sample.bmp"); |
| pic = SDL_LoadBMP(bmpfile); |
| if (pic == NULL) { |
| fprintf(stderr, "Couldn't load %s: %s\n", bmpfile, SDL_GetError()); |
| quit(1); |
| } |
| |
| /* Convert the picture to 32bits, for easy conversion */ |
| { |
| SDL_Surface *newsurf; |
| SDL_PixelFormat format; |
| |
| format.palette = NULL; |
| format.BitsPerPixel = 32; |
| format.BytesPerPixel = 4; |
| #if SDL_BYTEORDER == SDL_LIL_ENDIAN |
| format.Rshift = 0; |
| format.Gshift = 8; |
| format.Bshift = 16; |
| #else |
| format.Rshift = 24; |
| format.Gshift = 16; |
| format.Bshift = 8; |
| #endif |
| format.Ashift = 0; |
| format.Rmask = 0xff << format.Rshift; |
| format.Gmask = 0xff << format.Gshift; |
| format.Bmask = 0xff << format.Bshift; |
| format.Amask = 0; |
| format.Rloss = 0; |
| format.Gloss = 0; |
| format.Bloss = 0; |
| format.Aloss = 8; |
| |
| newsurf = SDL_ConvertSurface(pic, &format, SDL_SWSURFACE); |
| if (!newsurf) { |
| fprintf(stderr, "Couldn't convert picture to 32bits RGB: %s\n", |
| SDL_GetError()); |
| quit(1); |
| } |
| SDL_FreeSurface(pic); |
| pic = newsurf; |
| } |
| |
| /* Create the overlay */ |
| overlay = SDL_CreateYUVOverlay(pic->w, pic->h, overlay_format, screen); |
| if (overlay == NULL) { |
| fprintf(stderr, "Couldn't create overlay: %s\n", SDL_GetError()); |
| quit(1); |
| } |
| printf("Created %dx%dx%d %s %s overlay\n", overlay->w, overlay->h, |
| overlay->planes, overlay->hw_overlay ? "hardware" : "software", |
| overlay->format == SDL_YV12_OVERLAY ? "YV12" : overlay->format == |
| SDL_IYUV_OVERLAY ? "IYUV" : overlay->format == |
| SDL_YUY2_OVERLAY ? "YUY2" : overlay->format == |
| SDL_UYVY_OVERLAY ? "UYVY" : overlay->format == |
| SDL_YVYU_OVERLAY ? "YVYU" : "Unknown"); |
| for (i = 0; i < overlay->planes; i++) { |
| printf(" plane %d: pitch=%d\n", i, overlay->pitches[i]); |
| } |
| |
| /* Convert to YUV, and draw to the overlay */ |
| #ifdef BENCHMARK_SDL |
| then = SDL_GetTicks(); |
| #endif |
| switch (overlay->format) { |
| case SDL_YV12_OVERLAY: |
| ConvertRGBtoYV12(pic, overlay, monochrome, luminance); |
| break; |
| case SDL_UYVY_OVERLAY: |
| ConvertRGBtoUYVY(pic, overlay, monochrome, luminance); |
| break; |
| case SDL_YVYU_OVERLAY: |
| ConvertRGBtoYVYU(pic, overlay, monochrome, luminance); |
| break; |
| case SDL_YUY2_OVERLAY: |
| ConvertRGBtoYUY2(pic, overlay, monochrome, luminance); |
| break; |
| case SDL_IYUV_OVERLAY: |
| ConvertRGBtoIYUV(pic, overlay, monochrome, luminance); |
| break; |
| default: |
| printf("cannot convert RGB picture to obtained YUV format!\n"); |
| quit(1); |
| break; |
| } |
| #ifdef BENCHMARK_SDL |
| now = SDL_GetTicks(); |
| printf("Conversion Time: %d milliseconds\n", now - then); |
| #endif |
| |
| /* Do all the drawing work */ |
| #ifdef BENCHMARK_SDL |
| then = SDL_GetTicks(); |
| #endif |
| Draw(); |
| #ifdef BENCHMARK_SDL |
| now = SDL_GetTicks(); |
| printf("Time: %d milliseconds\n", now - then); |
| #endif |
| SDL_Delay(delay * 1000); |
| SDL_Quit(); |
| return (0); |
| } |
