/*

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

}