This repository has been archived by the owner on Feb 11, 2021. It is now read-only.
/
testoverlay.c
562 lines (510 loc) · 15.9 KB
1
2
3
4
5
6
7
8
9
10
11
/* 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);
12
13
14
#define WINDOW_WIDTH 640
#define WINDOW_HEIGHT 480
15
16
17
18
19
#include "SDL.h"
SDL_Surface *screen, *pic;
SDL_Overlay *overlay;
int scale;
20
21
22
int monochrome;
int luminance;
int w, h;
23
24
/* Call this instead of exit(), so we can clean up SDL: atexit() is evil. */
25
static void
26
quit(int rc)
27
{
28
29
SDL_Quit();
exit(rc);
30
31
}
32
33
34
35
/* NOTE: These RGB conversion functions are not intended for speed,
only as examples.
*/
36
void
37
RGBtoYUV(Uint8 * rgb, int *yuv, int monochrome, int luminance)
38
{
39
40
41
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];
42
43
yuv[1] = 128;
yuv[2] = 128;
44
#else
45
46
47
yuv[0] = (0.257 * rgb[0]) + (0.504 * rgb[1]) + (0.098 * rgb[2]) + 16;
yuv[1] = 128;
yuv[2] = 128;
48
#endif
49
50
51
52
53
} 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;
54
55
56
57
58
59
60
#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
}
61
62
63
64
if (luminance != 100) {
yuv[0] = yuv[0] * luminance / 100;
if (yuv[0] > 255)
yuv[0] = 255;
65
66
67
68
}
/* clamp values...if you need to, we don't seem to have a need */
/*
69
70
71
72
73
74
75
76
for(i=0;i<3;i++)
{
if(yuv[i]<0)
yuv[i]=0;
if(yuv[i]>255)
yuv[i]=255;
}
*/
77
78
}
79
void
80
81
ConvertRGBtoYV12(SDL_Surface * s, SDL_Overlay * o, int monochrome,
int luminance)
82
{
83
84
85
86
int x, y;
int yuv[3];
Uint8 *p, *op[3];
87
88
SDL_LockSurface(s);
SDL_LockYUVOverlay(o);
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/* 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++) {
104
RGBtoYUV(p, yuv, monochrome, luminance);
105
106
107
108
109
110
111
112
113
*(op[0]++) = yuv[0];
if (x % 2 == 0 && y % 2 == 0) {
*(op[1]++) = yuv[2];
*(op[2]++) = yuv[1];
}
p += s->format->BytesPerPixel;
}
}
114
115
SDL_UnlockYUVOverlay(o);
SDL_UnlockSurface(s);
116
117
}
118
void
119
120
ConvertRGBtoIYUV(SDL_Surface * s, SDL_Overlay * o, int monochrome,
int luminance)
121
{
122
123
124
125
int x, y;
int yuv[3];
Uint8 *p, *op[3];
126
127
SDL_LockSurface(s);
SDL_LockYUVOverlay(o);
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
/* 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++) {
143
RGBtoYUV(p, yuv, monochrome, luminance);
144
145
146
147
148
149
150
151
152
*(op[0]++) = yuv[0];
if (x % 2 == 0 && y % 2 == 0) {
*(op[1]++) = yuv[1];
*(op[2]++) = yuv[2];
}
p += s->format->BytesPerPixel;
}
}
153
154
SDL_UnlockYUVOverlay(o);
SDL_UnlockSurface(s);
155
156
}
157
void
158
159
ConvertRGBtoUYVY(SDL_Surface * s, SDL_Overlay * o, int monochrome,
int luminance)
160
{
161
162
163
164
int x, y;
int yuv[3];
Uint8 *p, *op;
165
166
SDL_LockSurface(s);
SDL_LockYUVOverlay(o);
167
168
169
170
171
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++) {
172
RGBtoYUV(p, yuv, monochrome, luminance);
173
174
175
176
177
178
179
180
181
182
183
if (x % 2 == 0) {
*(op++) = yuv[1];
*(op++) = yuv[0];
*(op++) = yuv[2];
} else
*(op++) = yuv[0];
p += s->format->BytesPerPixel;
}
}
184
185
SDL_UnlockYUVOverlay(o);
SDL_UnlockSurface(s);
186
187
}
188
void
189
190
ConvertRGBtoYVYU(SDL_Surface * s, SDL_Overlay * o, int monochrome,
int luminance)
191
{
192
193
194
195
int x, y;
int yuv[3];
Uint8 *p, *op;
196
197
SDL_LockSurface(s);
SDL_LockYUVOverlay(o);
198
199
200
201
202
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++) {
203
RGBtoYUV(p, yuv, monochrome, luminance);
204
205
206
207
208
209
210
211
212
213
214
215
216
if (x % 2 == 0) {
*(op++) = yuv[0];
*(op++) = yuv[2];
op[1] = yuv[1];
} else {
*op = yuv[0];
op += 2;
}
p += s->format->BytesPerPixel;
}
}
217
218
SDL_UnlockYUVOverlay(o);
SDL_UnlockSurface(s);
219
220
}
221
void
222
223
ConvertRGBtoYUY2(SDL_Surface * s, SDL_Overlay * o, int monochrome,
int luminance)
224
{
225
226
227
228
int x, y;
int yuv[3];
Uint8 *p, *op;
229
230
SDL_LockSurface(s);
SDL_LockYUVOverlay(o);
231
232
233
234
235
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++) {
236
RGBtoYUV(p, yuv, monochrome, luminance);
237
238
239
240
241
242
243
244
245
246
247
248
249
if (x % 2 == 0) {
*(op++) = yuv[0];
*(op++) = yuv[1];
op[1] = yuv[2];
} else {
*op = yuv[0];
op += 2;
}
p += s->format->BytesPerPixel;
}
}
250
251
SDL_UnlockYUVOverlay(o);
SDL_UnlockSurface(s);
252
253
}
254
void
255
Draw()
256
{
257
258
259
260
261
262
263
264
265
266
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;
267
SDL_DisplayYUVOverlay(overlay, &rect);
268
269
270
271
272
273
274
275
276
277
278
279
}
} 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--;
280
SDL_DisplayYUVOverlay(overlay, &rect);
281
282
}
}
283
printf("Displayed %d times.\n", i);
284
285
}
286
static void
287
PrintUsage(char *argv0)
288
{
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
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");
308
309
}
310
int
311
main(int argc, char **argv)
312
{
313
314
315
316
317
318
char *argv0 = argv[0];
int flip;
int delay;
int desired_bpp;
Uint32 video_flags, overlay_format;
char *bmpfile;
319
#ifdef BENCHMARK_SDL
320
Uint32 then, now;
321
#endif
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
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) {
337
if (strcmp(argv[1], "-delay") == 0) {
338
if (argv[2]) {
339
delay = atoi(argv[2]);
340
341
342
argv += 2;
argc -= 2;
} else {
343
fprintf(stderr, "The -delay option requires an argument\n");
344
345
return (1);
}
346
347
} else if (strcmp(argv[1], "-width") == 0) {
if (argv[2] && ((w = atoi(argv[2])) > 0)) {
348
349
350
argv += 2;
argc -= 2;
} else {
351
fprintf(stderr, "The -width option requires an argument\n");
352
353
return (1);
}
354
355
} else if (strcmp(argv[1], "-height") == 0) {
if (argv[2] && ((h = atoi(argv[2])) > 0)) {
356
357
358
argv += 2;
argc -= 2;
} else {
359
fprintf(stderr, "The -height option requires an argument\n");
360
361
return (1);
}
362
} else if (strcmp(argv[1], "-bpp") == 0) {
363
if (argv[2]) {
364
desired_bpp = atoi(argv[2]);
365
366
367
argv += 2;
argc -= 2;
} else {
368
fprintf(stderr, "The -bpp option requires an argument\n");
369
370
return (1);
}
371
} else if (strcmp(argv[1], "-lum") == 0) {
372
if (argv[2]) {
373
luminance = atoi(argv[2]);
374
375
376
argv += 2;
argc -= 2;
} else {
377
fprintf(stderr, "The -lum option requires an argument\n");
378
379
return (1);
}
380
} else if (strcmp(argv[1], "-format") == 0) {
381
if (argv[2]) {
382
if (!strcmp(argv[2], "YV12"))
383
overlay_format = SDL_YV12_OVERLAY;
384
else if (!strcmp(argv[2], "IYUV"))
385
overlay_format = SDL_IYUV_OVERLAY;
386
else if (!strcmp(argv[2], "YUY2"))
387
overlay_format = SDL_YUY2_OVERLAY;
388
else if (!strcmp(argv[2], "UYVY"))
389
overlay_format = SDL_UYVY_OVERLAY;
390
else if (!strcmp(argv[2], "YVYU"))
391
392
overlay_format = SDL_YVYU_OVERLAY;
else {
393
394
395
fprintf(stderr,
"The -format option %s is not recognized\n",
argv[2]);
396
397
398
399
400
return (1);
}
argv += 2;
argc -= 2;
} else {
401
fprintf(stderr, "The -format option requires an argument\n");
402
403
return (1);
}
404
} else if (strcmp(argv[1], "-hw") == 0) {
405
406
407
video_flags |= SDL_HWSURFACE;
argv += 1;
argc -= 1;
408
} else if (strcmp(argv[1], "-flip") == 0) {
409
410
411
video_flags |= SDL_DOUBLEBUF;
argv += 1;
argc -= 1;
412
} else if (strcmp(argv[1], "-scale") == 0) {
413
414
415
scale = 1;
argv += 1;
argc -= 1;
416
} else if (strcmp(argv[1], "-mono") == 0) {
417
418
419
monochrome = 1;
argv += 1;
argc -= 1;
420
421
422
} else if ((strcmp(argv[1], "-help") == 0)
|| (strcmp(argv[1], "-h") == 0)) {
PrintUsage(argv0);
423
return (1);
424
} else if (strcmp(argv[1], "-fullscreen") == 0) {
425
426
427
428
429
430
video_flags |= SDL_FULLSCREEN;
argv += 1;
argc -= 1;
} else
break;
}
431
432
if (SDL_Init(SDL_INIT_VIDEO) < 0) {
fprintf(stderr, "Couldn't initialize SDL: %s\n", SDL_GetError());
433
434
435
436
return (1);
}
/* Initialize the display */
437
screen = SDL_SetVideoMode(w, h, desired_bpp, video_flags);
438
if (screen == NULL) {
439
440
441
fprintf(stderr, "Couldn't set %dx%dx%d video mode: %s\n",
w, h, desired_bpp, SDL_GetError());
quit(1);
442
}
443
444
445
446
447
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");
448
if (screen->flags & SDL_DOUBLEBUF) {
449
printf("Double-buffering enabled\n");
450
451
452
453
flip = 1;
}
/* Set the window manager title bar */
454
SDL_WM_SetCaption("SDL test overlay", "testoverlay");
455
456
457
/* Load picture */
bmpfile = (argv[1] ? argv[1] : "sample.bmp");
458
pic = SDL_LoadBMP(bmpfile);
459
if (pic == NULL) {
460
461
fprintf(stderr, "Couldn't load %s: %s\n", bmpfile, SDL_GetError());
quit(1);
462
463
464
465
466
467
468
469
470
471
}
/* Convert the picture to 32bits, for easy conversion */
{
SDL_Surface *newsurf;
SDL_PixelFormat format;
format.palette = NULL;
format.BitsPerPixel = 32;
format.BytesPerPixel = 4;
472
#if SDL_BYTEORDER == SDL_LIL_ENDIAN
473
474
475
format.Rshift = 0;
format.Gshift = 8;
format.Bshift = 16;
476
#else
477
478
479
format.Rshift = 24;
format.Gshift = 16;
format.Bshift = 8;
480
#endif
481
482
483
484
485
486
487
488
489
490
491
492
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;
493
newsurf = SDL_ConvertSurface(pic, &format, SDL_SWSURFACE);
494
if (!newsurf) {
495
496
497
fprintf(stderr, "Couldn't convert picture to 32bits RGB: %s\n",
SDL_GetError());
quit(1);
498
}
499
SDL_FreeSurface(pic);
500
501
502
503
pic = newsurf;
}
/* Create the overlay */
504
overlay = SDL_CreateYUVOverlay(pic->w, pic->h, overlay_format, screen);
505
if (overlay == NULL) {
506
507
fprintf(stderr, "Couldn't create overlay: %s\n", SDL_GetError());
quit(1);
508
}
509
510
511
512
513
514
515
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");
516
for (i = 0; i < overlay->planes; i++) {
517
printf(" plane %d: pitch=%d\n", i, overlay->pitches[i]);
518
519
520
}
/* Convert to YUV, and draw to the overlay */
521
#ifdef BENCHMARK_SDL
522
then = SDL_GetTicks();
523
#endif
524
525
switch (overlay->format) {
case SDL_YV12_OVERLAY:
526
ConvertRGBtoYV12(pic, overlay, monochrome, luminance);
527
528
break;
case SDL_UYVY_OVERLAY:
529
ConvertRGBtoUYVY(pic, overlay, monochrome, luminance);
530
531
break;
case SDL_YVYU_OVERLAY:
532
ConvertRGBtoYVYU(pic, overlay, monochrome, luminance);
533
534
break;
case SDL_YUY2_OVERLAY:
535
ConvertRGBtoYUY2(pic, overlay, monochrome, luminance);
536
537
break;
case SDL_IYUV_OVERLAY:
538
ConvertRGBtoIYUV(pic, overlay, monochrome, luminance);
539
540
break;
default:
541
542
printf("cannot convert RGB picture to obtained YUV format!\n");
quit(1);
543
544
break;
}
545
#ifdef BENCHMARK_SDL
546
547
now = SDL_GetTicks();
printf("Conversion Time: %d milliseconds\n", now - then);
548
#endif
549
550
/* Do all the drawing work */
551
#ifdef BENCHMARK_SDL
552
then = SDL_GetTicks();
553
#endif
554
Draw();
555
#ifdef BENCHMARK_SDL
556
557
now = SDL_GetTicks();
printf("Time: %d milliseconds\n", now - then);
558
#endif
559
560
SDL_Delay(delay * 1000);
SDL_Quit();
561
return (0);
562
}