571 lines
16 KiB
C
571 lines
16 KiB
C
/*
|
|
Copyright (C) 1997-2011 Sam Lantinga <slouken@libsdl.org>
|
|
|
|
This software is provided 'as-is', without any express or implied
|
|
warranty. In no event will the authors be held liable for any damages
|
|
arising from the use of this software.
|
|
|
|
Permission is granted to anyone to use this software for any purpose,
|
|
including commercial applications, and to alter it and redistribute it
|
|
freely.
|
|
*/
|
|
|
|
/* 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] = (int)(0.299 * rgb[0] + 0.587 * rgb[1] + 0.114 * rgb[2]);
|
|
yuv[1] = 128;
|
|
yuv[2] = 128;
|
|
#else
|
|
yuv[0] = (int)((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] = (int)(0.299 * rgb[0] + 0.587 * rgb[1] + 0.114 * rgb[2]);
|
|
yuv[1] = (int)((rgb[2] - yuv[0]) * 0.565 + 128);
|
|
yuv[2] = (int)((rgb[0] - yuv[0]) * 0.713 + 128);
|
|
#else
|
|
yuv[0] = (int)((0.257 * rgb[0]) + (0.504 * rgb[1]) + (0.098 * rgb[2]) + 16);
|
|
yuv[1] = (int)(128 - (0.148 * rgb[0]) - (0.291 * rgb[1]) + (0.439 * rgb[2]));
|
|
yuv[2] = (int)(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);
|
|
}
|