1 /*
2 * Copyright 2011 Stefano Sabatini <stefano.sabatini-lala poste it>
3 * Copyright 2012 Nicolas George <nicolas.george normalesup org>
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #include <string.h>
23
30
32
34 {
35 switch (pix_fmt) {
56 default: /* unsupported */
58 }
59 return 0;
60 }
61
64 int *is_packed_rgba,
uint8_t rgba_map_ptr[4])
65 {
67 int i;
70
72
73 if (*is_packed_rgba) {
75 for (i = 0; i < 4; i++)
76 dst_color[rgba_map[i]] = rgba_color[i];
77
79 if (!line[0])
81 for (i = 0; i < w; i++)
82 memcpy(line[0] + i * pixel_step[0], dst_color, pixel_step[0]);
83 if (rgba_map_ptr)
84 memcpy(rgba_map_ptr, rgba_map, sizeof(rgba_map[0]) * 4);
85 } else {
86 int plane;
87
88 dst_color[0] =
RGB_TO_Y_CCIR(rgba_color[0], rgba_color[1], rgba_color[2]);
89 dst_color[1] =
RGB_TO_U_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);
90 dst_color[2] =
RGB_TO_V_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);
91 dst_color[3] = rgba_color[3];
92
93 for (plane = 0; plane < 4; plane++) {
94 int line_size;
95 int hsub1 = (plane == 1 || plane == 2) ? hsub : 0;
96
97 pixel_step[plane] = 1;
100 if (!line[plane]) {
101 while(plane && line[plane-1])
104 }
105 memset(line[plane], dst_color[plane], line_size);
106 }
107 }
108
109 return 0;
110 }
111
114 int hsub,
int vsub,
int x,
int y,
int w,
int h)
115 {
116 int i, plane;
118
119 for (plane = 0; plane < 4 && dst[plane]; plane++) {
120 int hsub1 = plane == 1 || plane == 2 ? hsub : 0;
121 int vsub1 = plane == 1 || plane == 2 ? vsub : 0;
124
125 p = dst[plane] + (y >> vsub1) * dst_linesize[plane];
126 for (i = 0; i <
height; i++) {
127 memcpy(p + (x >> hsub1) * pixelstep[plane],
128 src[plane], width * pixelstep[plane]);
129 p += dst_linesize[plane];
130 }
131 }
132 }
133
135 uint8_t *
src[4],
int src_linesize[4],
int pixelstep[4],
136 int hsub,
int vsub,
int x,
int y,
int y2,
int w,
int h)
137 {
138 int i, plane;
140
141 for (plane = 0; plane < 4 && dst[plane]; plane++) {
142 int hsub1 = plane == 1 || plane == 2 ? hsub : 0;
143 int vsub1 = plane == 1 || plane == 2 ? vsub : 0;
146
147 p = dst[plane] + (y >> vsub1) * dst_linesize[plane];
148 for (i = 0; i <
height; i++) {
149 memcpy(p + (x >> hsub1) * pixelstep[plane],
150 src[plane] + src_linesize[plane]*(i+(y2>>vsub1)), width * pixelstep[plane]);
151 p += dst_linesize[plane];
152 }
153 }
154 }
155
157 {
160 unsigned i, nb_planes = 0;
162
169 /* for now, only 8-bits formats */
174 /* strange interleaving */
175 if (pixelstep[c->
plane] != 0 &&
179 if (pixelstep[c->
plane] >= 8)
182 }
184 return AVERROR(ENOSYS);
/* exclude NV12 and NV21 */
185 memset(draw, 0, sizeof(*draw));
195 return 0;
196 }
197
199 {
200 unsigned i;
202
203 if (rgba != color->
rgba)
204 memcpy(color->
rgba, rgba,
sizeof(color->
rgba));
208 for (i = 0; i < 4; i++)
209 color->
comp[0].
u8[rgba_map[i]] = rgba[i];
210 } else {
211 for (i = 0; i < 4; i++)
212 color->
comp[rgba_map[i]].
u8[0] = rgba[i];
213 }
215 /* assume YUV */
219 color->
comp[3].
u8[0] = rgba[3];
222 color->
comp[1].
u8[0] = rgba[3];
223 } else {
225 "Color conversion not implemented for %s\n", draw->
desc->
name);
226 memset(color, 128, sizeof(*color));
227 }
228 }
229
231 int plane,
int x,
int y)
232 {
233 return data[plane] +
234 (y >> draw->
vsub[plane]) * linesize[plane] +
236 }
237
239 uint8_t *dst[],
int dst_linesize[],
241 int dst_x, int dst_y, int src_x, int src_y,
242 int w, int h)
243 {
244 int plane,
y, wp, hp;
246
247 for (plane = 0; plane < draw->
nb_planes; plane++) {
248 p =
pointer_at(draw, src, src_linesize, plane, src_x, src_y);
249 q =
pointer_at(draw, dst, dst_linesize, plane, dst_x, dst_y);
252 for (y = 0; y < hp; y++) {
253 memcpy(q, p, wp);
254 p += src_linesize[plane];
255 q += dst_linesize[plane];
256 }
257 }
258 }
259
261 uint8_t *dst[],
int dst_linesize[],
262 int dst_x, int dst_y, int w, int h)
263 {
264 int plane, x,
y, wp, hp;
266
267 for (plane = 0; plane < draw->
nb_planes; plane++) {
268 p0 =
pointer_at(draw, dst, dst_linesize, plane, dst_x, dst_y);
271 if (!hp)
272 return;
273 p = p0;
274 /* copy first line from color */
275 for (x = 0; x < wp; x++) {
278 }
280 /* copy next lines from first line */
281 p = p0 + dst_linesize[plane];
282 for (y = 1; y < hp; y++) {
283 memcpy(p, p0, wp);
284 p += dst_linesize[plane];
285 }
286 }
287 }
288
289 /**
290 * Clip interval [x; x+w[ within [0; wmax[.
291 * The resulting w may be negative if the final interval is empty.
292 * dx, if not null, return the difference between in and out value of x.
293 */
295 {
296 if (dx)
297 *dx = 0;
298 if (*x < 0) {
299 if (dx)
300 *dx = -*x;
301 *w += *x;
302 *x = 0;
303 }
304 if (*x + *w > wmax)
305 *w = wmax - *x;
306 }
307
308 /**
309 * Decompose w pixels starting at x
310 * into start + (w starting at x) + end
311 * with x and w aligned on multiples of 1<<sub.
312 */
314 {
315 int mask = (1 << sub) - 1;
316
317 *start = (-*x) & mask;
319 *start =
FFMIN(*start, *w);
322 *w >>= sub;
323 }
324
326 {
327 return (draw->
comp_mask[plane] >> comp) & 1;
328 }
329
330 /* If alpha is in the [ 0 ; 0x1010101 ] range,
331 then alpha * value is in the [ 0 ; 0xFFFFFFFF ] range,
332 and >> 24 gives a correct rounding. */
334 int dx, int w, unsigned hsub, int left, int right)
335 {
336 unsigned asrc = alpha *
src;
337 unsigned tau = 0x1010101 -
alpha;
338 int x;
339
340 if (left) {
341 unsigned suba = (left *
alpha) >> hsub;
342 *dst = (*dst * (0x1010101 - suba) + src * suba) >> 24;
343 dst += dx;
344 }
345 for (x = 0; x < w; x++) {
346 *dst = (*dst * tau + asrc) >> 24;
347 dst += dx;
348 }
349 if (right) {
350 unsigned suba = (right *
alpha) >> hsub;
351 *dst = (*dst * (0x1010101 - suba) + src * suba) >> 24;
352 }
353 }
354
356 uint8_t *dst[],
int dst_linesize[],
357 int dst_w, int dst_h,
358 int x0, int y0, int w, int h)
359 {
360 unsigned alpha, nb_planes, nb_comp, plane,
comp;
361 int w_sub, h_sub, x_sub, y_sub, left, right, top, bottom,
y;
363
364 /* TODO optimize if alpha = 0xFF */
367 if (w <= 0 || h <= 0 || !color->rgba[3])
368 return;
369 /* 0x10203 * alpha + 2 is in the [ 2 ; 0x1010101 - 2 ] range */
370 alpha = 0x10203 * color->
rgba[3] + 0x2;
371 nb_planes = (draw->
nb_planes - 1) | 1;
/* eliminate alpha */
372 for (plane = 0; plane < nb_planes; plane++) {
374 p0 =
pointer_at(draw, dst, dst_linesize, plane, x0, y0);
375 w_sub = w;
376 h_sub = h;
377 x_sub = x0;
378 y_sub = y0;
381 for (comp = 0; comp < nb_comp; comp++) {
383 continue;
385 if (top) {
388 draw->
hsub[plane], left, right);
389 p += dst_linesize[plane];
390 }
391 for (y = 0; y < h_sub; y++) {
394 draw->
hsub[plane], left, right);
395 p += dst_linesize[plane];
396 }
397 if (bottom)
400 draw->
hsub[plane], left, right);
401 }
402 }
403 }
404
407 unsigned w,
unsigned h,
unsigned shift,
unsigned xm0)
408 {
409 unsigned xm, x,
y, t = 0;
410 unsigned xmshf = 3 - l2depth;
411 unsigned xmmod = 7 >> l2depth;
412 unsigned mbits = (1 << (1 << l2depth)) - 1;
413 unsigned mmult = 255 / mbits;
414
415 for (y = 0; y < h; y++) {
416 xm = xm0;
417 for (x = 0; x < w; x++) {
418 t += ((mask[xm >> xmshf] >> ((~xm & xmmod) << l2depth)) & mbits)
419 * mmult;
420 xm++;
421 }
422 mask += mask_linesize;
423 }
424 alpha = (t >>
shift) * alpha;
425 *dst = ((0x1010101 -
alpha) * *dst + alpha * src) >> 24;
426 }
427
430 uint8_t *
mask,
int mask_linesize,
int l2depth,
int w,
431 unsigned hsub, unsigned vsub,
432 int xm, int left, int right, int hband)
433 {
434 int x;
435
436 if (left) {
437 blend_pixel(dst, src, alpha, mask, mask_linesize, l2depth,
438 left, hband, hsub + vsub, xm);
439 dst += dst_delta;
440 xm += left;
441 }
442 for (x = 0; x < w; x++) {
443 blend_pixel(dst, src, alpha, mask, mask_linesize, l2depth,
444 1 << hsub, hband, hsub + vsub, xm);
445 dst += dst_delta;
446 xm += 1 << hsub;
447 }
448 if (right)
449 blend_pixel(dst, src, alpha, mask, mask_linesize, l2depth,
450 right, hband, hsub + vsub, xm);
451 }
452
454 uint8_t *dst[],
int dst_linesize[],
int dst_w,
int dst_h,
455 uint8_t *
mask,
int mask_linesize,
int mask_w,
int mask_h,
456 int l2depth, unsigned endianness, int x0, int y0)
457 {
458 unsigned alpha, nb_planes, nb_comp, plane,
comp;
459 int xm0, ym0, w_sub, h_sub, x_sub, y_sub, left, right, top, bottom,
y;
461
464 mask += ym0 * mask_linesize;
465 if (mask_w <= 0 || mask_h <= 0 || !color->rgba[3])
466 return;
467 /* alpha is in the [ 0 ; 0x10203 ] range,
468 alpha * mask is in the [ 0 ; 0x1010101 - 4 ] range */
469 alpha = (0x10307 * color->
rgba[3] + 0x3) >> 8;
470 nb_planes = (draw->
nb_planes - 1) | 1;
/* eliminate alpha */
471 for (plane = 0; plane < nb_planes; plane++) {
473 p0 =
pointer_at(draw, dst, dst_linesize, plane, x0, y0);
474 w_sub = mask_w;
475 h_sub = mask_h;
476 x_sub = x0;
477 y_sub = y0;
480 for (comp = 0; comp < nb_comp; comp++) {
482 continue;
485 if (top) {
487 color->
comp[plane].
u8[comp], alpha,
488 m, mask_linesize, l2depth, w_sub,
489 draw->
hsub[plane], draw->
vsub[plane],
490 xm0, left, right, top);
491 p += dst_linesize[plane];
492 m += top * mask_linesize;
493 }
494 for (y = 0; y < h_sub; y++) {
496 color->
comp[plane].
u8[comp], alpha,
497 m, mask_linesize, l2depth, w_sub,
498 draw->
hsub[plane], draw->
vsub[plane],
499 xm0, left, right, 1 << draw->
vsub[plane]);
500 p += dst_linesize[plane];
501 m += mask_linesize << draw->
vsub[plane];
502 }
503 if (bottom)
505 color->
comp[plane].
u8[comp], alpha,
506 m, mask_linesize, l2depth, w_sub,
507 draw->
hsub[plane], draw->
vsub[plane],
508 xm0, left, right, bottom);
509 }
510 }
511 }
512
515 {
517
518 if (!shift)
520 if (round_dir >= 0)
521 value += round_dir ? (1 <<
shift) - 1 : 1 << (shift - 1);
522 return (value >> shift) <<
shift;
523 }
524
526 {
530
534 return fmts;
535 }
536
537 #ifdef TEST
538
539 #undef printf
540
542 {
548
552 continue;
553 printf(
"Testing %s...%*s", desc->
name,
554 (
int)(16 - strlen(desc->
name)),
"");
556 if (r < 0) {
559 printf("no: %s\n", buf);
560 continue;
561 }
563 for (i = 0; i <
sizeof(
color); i++)
564 if (((
uint8_t *)&color)[i] != 128)
565 break;
566 if (i == sizeof(color)) {
567 printf("fallback color\n");
568 continue;
569 }
570 printf("ok\n");
571 }
572 return 0;
573 }
574
575 #endif