/********************************************************************************

* *

* Test of the overlay used for moved pictures, test more closed to real life. *

* Running trojan moose :) Coded by Mike Gorchak. *

* *

********************************************************************************/

#include <stdlib.h>

#include <stdio.h>

#include <string.h>

#include "SDL.h"

#define MOOSEPIC_W 64

#define MOOSEPIC_H 88

#define MOOSEFRAME_SIZE (MOOSEPIC_W * MOOSEPIC_H)

#define MOOSEFRAMES_COUNT 10

SDL_Color MooseColors[84] = {

{49, 49, 49}

, {66, 24, 0}

, {66, 33, 0}

, {66, 66, 66}

,

{66, 115, 49}

, {74, 33, 0}

, {74, 41, 16}

, {82, 33, 8}

,

{82, 41, 8}

, {82, 49, 16}

, {82, 82, 82}

, {90, 41, 8}

,

{90, 41, 16}

, {90, 57, 24}

, {99, 49, 16}

, {99, 66, 24}

,

{99, 66, 33}

, {99, 74, 33}

, {107, 57, 24}

, {107, 82, 41}

,

{115, 57, 33}

, {115, 66, 33}

, {115, 66, 41}

, {115, 74, 0}

,

{115, 90, 49}

, {115, 115, 115}

, {123, 82, 0}

, {123, 99, 57}

,

{132, 66, 41}

, {132, 74, 41}

, {132, 90, 8}

, {132, 99, 33}

,

{132, 99, 66}

, {132, 107, 66}

, {140, 74, 49}

, {140, 99, 16}

,

{140, 107, 74}

, {140, 115, 74}

, {148, 107, 24}

, {148, 115, 82}

,

{148, 123, 74}

, {148, 123, 90}

, {156, 115, 33}

, {156, 115, 90}

,

{156, 123, 82}

, {156, 132, 82}

, {156, 132, 99}

, {156, 156, 156}

,

{165, 123, 49}

, {165, 123, 90}

, {165, 132, 82}

, {165, 132, 90}

,

{165, 132, 99}

, {165, 140, 90}

, {173, 132, 57}

, {173, 132, 99}

,

{173, 140, 107}

, {173, 140, 115}

, {173, 148, 99}

, {173, 173, 173}

,

{181, 140, 74}

, {181, 148, 115}

, {181, 148, 123}

, {181, 156, 107}

,

{189, 148, 123}

, {189, 156, 82}

, {189, 156, 123}

, {189, 156, 132}

,

{189, 189, 189}

, {198, 156, 123}

, {198, 165, 132}

, {206, 165, 99}

,

{206, 165, 132}

, {206, 173, 140}

, {206, 206, 206}

, {214, 173, 115}

,

{214, 173, 140}

, {222, 181, 148}

, {222, 189, 132}

, {222, 189, 156}

,

{222, 222, 222}

, {231, 198, 165}

, {231, 231, 231}

, {239, 206, 173}

};

/* Call this instead of exit(), so we can clean up SDL: atexit() is evil. */

static void

quit (int rc)

{

SDL_Quit ();

exit (rc);

}

/* All RGB2YUV conversion code and some other parts of code has been taken from testoverlay.c */

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

}

}

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

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

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

}

static void

PrintUsage (char *argv0)

{

fprintf (stderr, "Usage: %s [arg] [arg] [arg] ...\n", argv0);

fprintf (stderr, "\n");

fprintf (stderr, "Where 'arg' is any of the following options:\n");

fprintf (stderr, "\n");

fprintf (stderr, " -fps <frames per second>\n");

fprintf (stderr,

" -format <fmt> (one of the: YV12, IYUV, YUY2, UYVY, YVYU)\n");

fprintf (stderr,

" -scale <scale factor> (initial scale of the overlay)\n");

fprintf (stderr, " -help (shows this help)\n");

fprintf (stderr, "\n");

fprintf (stderr,

"Press ESC to exit, or SPACE to freeze the movie while application running.\n");

fprintf (stderr, "\n");

}

int

main (int argc, char **argv)

{

Uint8 *RawMooseData;

SDL_RWops *handle;

SDL_Surface *screen;

SDL_Surface *MooseFrame[MOOSEFRAMES_COUNT];

SDL_Overlay *overlay;

SDL_Rect overlayrect;

SDL_Event event;

Uint32 lastftick;

int paused = 0;

int resized = 0;

int i;

int fps = 12;

int fpsdelay;

int overlay_format = SDL_YUY2_OVERLAY;

int scale = 5;

if (SDL_Init (SDL_INIT_VIDEO | SDL_INIT_NOPARACHUTE) < 0) {

fprintf (stderr, "Couldn't initialize SDL: %s\n", SDL_GetError ());

return 3;

}

while (argc > 1) {

if (strcmp (argv[1], "-fps") == 0) {

if (argv[2]) {

fps = atoi (argv[2]);

if (fps == 0) {

fprintf (stderr,

"The -fps option requires an argument [from 1 to 1000], default is 12.\n");

quit (10);

}

if ((fps < 0) || (fps > 1000)) {

fprintf (stderr,

"The -fps option must be in range from 1 to 1000, default is 12.\n");

quit (10);

}

argv += 2;

argc -= 2;

} else {

fprintf (stderr,

"The -fps option requires an argument [from 1 to 1000], default is 12.\n");

quit (10);

}

} 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, see help for info.\n",

argv[2]);

quit (10);

}

argv += 2;

argc -= 2;

} else {

fprintf (stderr,

"The -format option requires an argument, default is YUY2.\n");

quit (10);

}

} else if (strcmp (argv[1], "-scale") == 0) {

if (argv[2]) {

scale = atoi (argv[2]);

if (scale == 0) {

fprintf (stderr,

"The -scale option requires an argument [from 1 to 50], default is 5.\n");

quit (10);

}

if ((scale < 0) || (scale > 50)) {

fprintf (stderr,

"The -scale option must be in range from 1 to 50, default is 5.\n");

quit (10);

}

argv += 2;

argc -= 2;

} else {

fprintf (stderr,

"The -fps option requires an argument [from 1 to 1000], default is 12.\n");

quit (10);

}

} else if ((strcmp (argv[1], "-help") == 0)

|| (strcmp (argv[1], "-h") == 0)) {

PrintUsage (argv[0]);

quit (0);

} else {

fprintf (stderr, "Unrecognized option: %s.\n", argv[1]);

quit (10);

}

break;

}

RawMooseData = (Uint8 *) malloc (MOOSEFRAME_SIZE * MOOSEFRAMES_COUNT);

if (RawMooseData == NULL) {

fprintf (stderr, "Can't allocate memory for movie !\n");

free (RawMooseData);

quit (1);

}

/* load the trojan moose images */

handle = SDL_RWFromFile ("moose.dat", "rb");

if (handle == NULL) {

fprintf (stderr, "Can't find the file moose.dat !\n");

free (RawMooseData);

quit (2);

}

SDL_RWread (handle, RawMooseData, MOOSEFRAME_SIZE, MOOSEFRAMES_COUNT);

SDL_RWclose (handle);

/* Set video mode */

if ((screen =

SDL_SetVideoMode (MOOSEPIC_W * scale, MOOSEPIC_H * scale, 0,

SDL_RESIZABLE | SDL_SWSURFACE)) == NULL) {

fprintf (stderr, "Couldn't set video mode: %s\n", SDL_GetError ());

free (RawMooseData);

quit (4);

}

/* Set the window manager title bar */

SDL_WM_SetCaption ("SDL test overlay: running moose", "testoverlay2");

for (i = 0; i < MOOSEFRAMES_COUNT; i++) {

MooseFrame[i] =

SDL_CreateRGBSurfaceFrom (RawMooseData + i * MOOSEFRAME_SIZE,

MOOSEPIC_W, MOOSEPIC_H, 8, MOOSEPIC_W,

0, 0, 0, 0);

if (MooseFrame[i] == NULL) {

fprintf (stderr, "Couldn't create SDL_Surfaces:%s\n",

SDL_GetError ());

free (RawMooseData);

quit (5);

}

SDL_SetColors (MooseFrame[i], MooseColors, 0, 84);

{

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;

format.colorkey = 0;

format.alpha = 0;

newsurf =

SDL_ConvertSurface (MooseFrame[i], &format, SDL_SWSURFACE);

if (!newsurf) {

fprintf (stderr,

"Couldn't convert picture to 32bits RGB: %s\n",

SDL_GetError ());

quit (6);

}

SDL_FreeSurface (MooseFrame[i]);

MooseFrame[i] = newsurf;

}

}

free (RawMooseData);

overlay =

SDL_CreateYUVOverlay (MOOSEPIC_W, MOOSEPIC_H, overlay_format, screen);

if (!overlay) {

fprintf (stderr, "Couldn't create overlay: %s\n", SDL_GetError ());

quit (7);

}

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

}

overlayrect.x = 0;

overlayrect.y = 0;

overlayrect.w = MOOSEPIC_W * scale;

overlayrect.h = MOOSEPIC_H * scale;

/* set the start frame */

i = 0;

fpsdelay = 1000 / fps;

/* Ignore key up events, they don't even get filtered */

SDL_EventState (SDL_KEYUP, SDL_IGNORE);

lastftick = SDL_GetTicks ();

/* Loop, waiting for QUIT or RESIZE */

while (1) {

if (SDL_PollEvent (&event)) {

switch (event.type) {

case SDL_VIDEORESIZE:

screen =

SDL_SetVideoMode (event.resize.w, event.resize.h, 0,

SDL_RESIZABLE | SDL_SWSURFACE);

overlayrect.w = event.resize.w;

overlayrect.h = event.resize.h;

if (paused) {

resized = 1;

}

break;

case SDL_MOUSEBUTTONDOWN:

overlayrect.x = event.button.x - overlayrect.w / 2;

overlayrect.y = event.button.y - overlayrect.h / 2;

break;

case SDL_KEYDOWN:

if (event.key.keysym.sym == SDLK_SPACE) {

paused = !paused;

break;

}

if (event.key.keysym.sym != SDLK_ESCAPE) {

break;

}

case SDL_QUIT:

SDL_FreeYUVOverlay (overlay);

for (i = 0; i < MOOSEFRAMES_COUNT; i++) {

SDL_FreeSurface (MooseFrame[i]);

}

quit (0);

}

}

if ((!paused) || (resized)) {

if (((SDL_GetTicks () - lastftick) > fpsdelay) || (resized)) {

lastftick = SDL_GetTicks ();

switch (overlay_format) {

case SDL_YUY2_OVERLAY:

ConvertRGBtoYUY2 (MooseFrame[i], overlay, 0, 100);

break;

case SDL_YV12_OVERLAY:

ConvertRGBtoYV12 (MooseFrame[i], overlay, 0, 100);

break;

case SDL_UYVY_OVERLAY:

ConvertRGBtoUYVY (MooseFrame[i], overlay, 0, 100);

break;

case SDL_YVYU_OVERLAY:

ConvertRGBtoYVYU (MooseFrame[i], overlay, 0, 100);

break;

case SDL_IYUV_OVERLAY:

ConvertRGBtoIYUV (MooseFrame[i], overlay, 0, 100);

break;

}

SDL_DisplayYUVOverlay (overlay, &overlayrect);

if (!resized) {

i++;

if (i == 10) {

i = 0;

}

} else {

resized = 0;

}

}

}

/* kind of timeslice to OS */

SDL_Delay (1);

}

SDL_Quit ();

return 0;

}