This repository has been archived by the owner on Feb 11, 2021. It is now read-only.
/
testoverlay2.c
595 lines (540 loc) · 16.5 KB
1
2
3
4
5
6
7
8
9
10
11
/*
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.
*/
12
13
14
15
16
17
18
/********************************************************************************
* *
* Test of the overlay used for moved pictures, test more closed to real life. *
* Running trojan moose :) Coded by Mike Gorchak. *
* *
********************************************************************************/
19
20
21
22
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
23
24
25
26
27
28
29
30
#include "SDL.h"
#define MOOSEPIC_W 64
#define MOOSEPIC_H 88
#define MOOSEFRAME_SIZE (MOOSEPIC_W * MOOSEPIC_H)
#define MOOSEFRAMES_COUNT 10
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
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}
136
137
};
138
139
/* Call this instead of exit(), so we can clean up SDL: atexit() is evil. */
140
141
static void
quit(int rc)
142
{
143
144
SDL_Quit();
exit(rc);
145
146
}
147
148
149
150
151
152
/* 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.
*/
153
154
void
RGBtoYUV(Uint8 * rgb, int *yuv, int monochrome, int luminance)
155
{
156
157
if (monochrome) {
#if 1 /* these are the two formulas that I found on the FourCC site... */
158
yuv[0] = (int)(0.299 * rgb[0] + 0.587 * rgb[1] + 0.114 * rgb[2]);
159
160
161
yuv[1] = 128;
yuv[2] = 128;
#else
162
yuv[0] = (int)(0.257 * rgb[0]) + (0.504 * rgb[1]) + (0.098 * rgb[2]) + 16;
163
164
165
yuv[1] = 128;
yuv[2] = 128;
#endif
166
167
} else {
#if 1 /* these are the two formulas that I found on the FourCC site... */
168
169
170
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);
171
172
173
174
175
176
177
#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
}
178
179
180
181
if (luminance != 100) {
yuv[0] = yuv[0] * luminance / 100;
if (yuv[0] > 255)
yuv[0] = 255;
182
183
184
}
}
185
void
186
187
ConvertRGBtoYV12(Uint8 *rgb, Uint8 *out, int w, int h,
int monochrome, int luminance)
188
{
189
190
int x, y;
int yuv[3];
191
192
193
194
195
196
197
198
Uint8 *op[3];
op[0] = out;
op[1] = op[0] + w*h;
op[2] = op[1] + w*h/4;
for (y = 0; y < h; ++y) {
for (x = 0; x < w; ++x) {
RGBtoYUV(rgb, yuv, monochrome, luminance);
199
200
201
202
203
*(op[0]++) = yuv[0];
if (x % 2 == 0 && y % 2 == 0) {
*(op[1]++) = yuv[2];
*(op[2]++) = yuv[1];
}
204
rgb += 3;
205
206
}
}
207
208
}
209
210
211
void
ConvertRGBtoIYUV(SDL_Surface * s, SDL_Overlay * o, int monochrome,
int luminance)
212
{
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
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);
239
240
}
241
242
243
void
ConvertRGBtoUYVY(SDL_Surface * s, SDL_Overlay * o, int monochrome,
int luminance)
244
{
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
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);
270
271
}
272
273
274
void
ConvertRGBtoYVYU(SDL_Surface * s, SDL_Overlay * o, int monochrome,
int luminance)
275
{
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
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);
303
304
}
305
306
307
void
ConvertRGBtoYUY2(SDL_Surface * s, SDL_Overlay * o, int monochrome,
int luminance)
308
{
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
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);
336
337
}
338
339
static void
PrintUsage(char *argv0)
340
341
{
fprintf(stderr, "Usage: %s [arg] [arg] [arg] ...\n", argv0);
342
343
344
fprintf(stderr, "\n");
fprintf(stderr, "Where 'arg' is any of the following options:\n");
fprintf(stderr, "\n");
345
346
347
348
349
fprintf(stderr, " -fps <frames per second>\n");
fprintf(stderr, " -nodelay\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");
350
fprintf(stderr, "\n");
351
352
fprintf(stderr,
"Press ESC to exit, or SPACE to freeze the movie while application running.\n");
353
354
355
fprintf(stderr, "\n");
}
356
357
int
main(int argc, char **argv)
358
{
359
360
Uint8 *RawMooseData;
SDL_RWops *handle;
361
362
363
364
365
366
367
int window_w;
int window_h;
SDL_Window *window;
SDL_Renderer *renderer;
Uint8 MooseFrame[MOOSEFRAMES_COUNT][MOOSEFRAME_SIZE*2];
SDL_Texture *MooseTexture;
SDL_Rect displayrect;
368
SDL_Event event;
369
int paused = 0;
370
int i, j;
371
int fps = 12;
372
int fpsdelay;
373
int nodelay = 0;
374
Uint32 pixel_format = SDL_PIXELFORMAT_YV12;
375
int scale = 5;
376
SDL_bool done = SDL_FALSE;
377
378
if (SDL_Init(SDL_INIT_VIDEO | SDL_INIT_NOPARACHUTE) < 0) {
379
380
381
382
fprintf(stderr, "Couldn't initialize SDL: %s\n", SDL_GetError());
return 3;
}
383
384
385
while (argc > 1) {
if (strcmp(argv[1], "-fps") == 0) {
if (argv[2]) {
386
fps = atoi(argv[2]);
387
388
389
if (fps == 0) {
fprintf(stderr,
"The -fps option requires an argument [from 1 to 1000], default is 12.\n");
390
quit(10);
391
}
392
393
394
if ((fps < 0) || (fps > 1000)) {
fprintf(stderr,
"The -fps option must be in range from 1 to 1000, default is 12.\n");
395
quit(10);
396
397
398
}
argv += 2;
argc -= 2;
399
400
401
} else {
fprintf(stderr,
"The -fps option requires an argument [from 1 to 1000], default is 12.\n");
402
quit(10);
403
}
404
405
406
407
} else if (strcmp(argv[1], "-nodelay") == 0) {
nodelay = 1;
argv += 1;
argc -= 1;
408
409
410
} else if (strcmp(argv[1], "-format") == 0) {
if (argv[2]) {
if (!strcmp(argv[2], "YV12"))
411
pixel_format = SDL_PIXELFORMAT_YV12;
412
else if (!strcmp(argv[2], "IYUV"))
413
pixel_format = SDL_PIXELFORMAT_IYUV;
414
else if (!strcmp(argv[2], "YUY2"))
415
pixel_format = SDL_PIXELFORMAT_YUY2;
416
else if (!strcmp(argv[2], "UYVY"))
417
pixel_format = SDL_PIXELFORMAT_UYVY;
418
else if (!strcmp(argv[2], "YVYU"))
419
pixel_format = SDL_PIXELFORMAT_YVYU;
420
421
422
423
else {
fprintf(stderr,
"The -format option %s is not recognized, see help for info.\n",
argv[2]);
424
quit(10);
425
426
427
}
argv += 2;
argc -= 2;
428
429
430
} else {
fprintf(stderr,
"The -format option requires an argument, default is YUY2.\n");
431
quit(10);
432
}
433
434
} else if (strcmp(argv[1], "-scale") == 0) {
if (argv[2]) {
435
scale = atoi(argv[2]);
436
437
438
if (scale == 0) {
fprintf(stderr,
"The -scale option requires an argument [from 1 to 50], default is 5.\n");
439
quit(10);
440
}
441
442
443
if ((scale < 0) || (scale > 50)) {
fprintf(stderr,
"The -scale option must be in range from 1 to 50, default is 5.\n");
444
quit(10);
445
446
447
}
argv += 2;
argc -= 2;
448
449
450
} else {
fprintf(stderr,
"The -fps option requires an argument [from 1 to 1000], default is 12.\n");
451
quit(10);
452
}
453
454
} else if ((strcmp(argv[1], "-help") == 0)
|| (strcmp(argv[1], "-h") == 0)) {
455
PrintUsage(argv[0]);
456
quit(0);
457
} else {
458
fprintf(stderr, "Unrecognized option: %s.\n", argv[1]);
459
quit(10);
460
461
462
}
break;
}
463
464
465
RawMooseData = (Uint8 *) malloc(MOOSEFRAME_SIZE * MOOSEFRAMES_COUNT);
if (RawMooseData == NULL) {
466
467
fprintf(stderr, "Can't allocate memory for movie !\n");
free(RawMooseData);
468
quit(1);
469
470
471
}
/* load the trojan moose images */
472
473
handle = SDL_RWFromFile("moose.dat", "rb");
if (handle == NULL) {
474
475
fprintf(stderr, "Can't find the file moose.dat !\n");
free(RawMooseData);
476
quit(2);
477
}
478
479
480
481
482
SDL_RWread(handle, RawMooseData, MOOSEFRAME_SIZE, MOOSEFRAMES_COUNT);
SDL_RWclose(handle);
483
484
485
486
487
488
489
490
491
492
/* Create the window and renderer */
window_w = MOOSEPIC_W * scale;
window_h = MOOSEPIC_H * scale;
window = SDL_CreateWindow("Happy Moose",
SDL_WINDOWPOS_UNDEFINED,
SDL_WINDOWPOS_UNDEFINED,
window_w, window_h,
SDL_WINDOW_SHOWN|SDL_WINDOW_RESIZABLE);
if (!window) {
fprintf(stderr, "Couldn't set create window: %s\n", SDL_GetError());
493
free(RawMooseData);
494
quit(4);
495
496
}
497
498
499
500
501
renderer = SDL_CreateRenderer(window, -1, 0);
if (!renderer) {
fprintf(stderr, "Couldn't set create renderer: %s\n", SDL_GetError());
free(RawMooseData);
quit(4);
502
503
}
504
MooseTexture = SDL_CreateTexture(renderer, pixel_format, SDL_TEXTUREACCESS_STREAMING, MOOSEPIC_W, MOOSEPIC_H);
505
if (!MooseTexture) {
506
507
508
fprintf(stderr, "Couldn't set create texture: %s\n", SDL_GetError());
free(RawMooseData);
quit(5);
509
}
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
for (i = 0; i < MOOSEFRAMES_COUNT; i++) {
Uint8 MooseFrameRGB[MOOSEFRAME_SIZE*3];
Uint8 *rgb;
Uint8 *frame;
rgb = MooseFrameRGB;
frame = RawMooseData + i * MOOSEFRAME_SIZE;
for (j = 0; j < MOOSEFRAME_SIZE; ++j) {
rgb[0] = MooseColors[frame[j]].r;
rgb[1] = MooseColors[frame[j]].g;
rgb[2] = MooseColors[frame[j]].b;
rgb += 3;
}
ConvertRGBtoYV12(MooseFrameRGB, MooseFrame[i], MOOSEPIC_W, MOOSEPIC_H, 0, 100);
525
526
}
527
free(RawMooseData);
528
529
/* set the start frame */
530
i = 0;
531
532
533
534
535
536
537
538
539
540
if (nodelay) {
fpsdelay = 0;
} else {
fpsdelay = 1000 / fps;
}
displayrect.x = 0;
displayrect.y = 0;
displayrect.w = window_w;
displayrect.h = window_h;
541
542
543
544
545
/* Ignore key up events, they don't even get filtered */
SDL_EventState(SDL_KEYUP, SDL_IGNORE);
/* Loop, waiting for QUIT or RESIZE */
546
547
while (!done) {
while (SDL_PollEvent(&event)) {
548
switch (event.type) {
549
550
case SDL_WINDOWEVENT:
if (event.window.event == SDL_WINDOWEVENT_RESIZED) {
551
SDL_RenderSetViewport(renderer, NULL);
552
displayrect.w = window_w = event.window.data1;
553
displayrect.h = window_h = event.window.data2;
554
555
556
}
break;
case SDL_MOUSEBUTTONDOWN:
557
558
559
560
561
562
563
564
displayrect.x = event.button.x - window_w / 2;
displayrect.y = event.button.y - window_h / 2;
break;
case SDL_MOUSEMOTION:
if (event.motion.state) {
displayrect.x = event.motion.x - window_w / 2;
displayrect.y = event.motion.y - window_h / 2;
}
565
566
567
568
569
570
571
572
573
574
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:
575
576
done = SDL_TRUE;
break;
577
578
}
}
579
SDL_Delay(fpsdelay);
580
581
582
if (!paused) {
i = (i + 1) % MOOSEFRAMES_COUNT;
583
584
SDL_UpdateTexture(MooseTexture, NULL, MooseFrame[i], MOOSEPIC_W*SDL_BYTESPERPIXEL(pixel_format));
585
}
586
587
588
SDL_RenderClear(renderer);
SDL_RenderCopy(renderer, MooseTexture, NULL, &displayrect);
SDL_RenderPresent(renderer);
589
}
590
591
SDL_DestroyRenderer(renderer);
quit(0);
592
593
return 0;
}
594
595
/* vi: set ts=4 sw=4 expandtab: */