/*

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;

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[1] = 128;

yuv[2] = 128;

yuv[1] = 128;

yuv[2] = 128;

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

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;

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;

format.Rshift = 0;

format.Gshift = 8;

format.Bshift = 16;

format.Rshift = 24;

format.Gshift = 16;

format.Bshift = 8;

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 */

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;

}

now = SDL_GetTicks();

printf("Conversion Time: %d milliseconds\n", now - then);

/* Do all the drawing work */

now = SDL_GetTicks();

printf("Time: %d milliseconds\n", now - then);

SDL_Delay(delay * 1000);

SDL_Quit();

return (0);