lavfi/overlay: add support for partial overlaying
Partially rework the iteration logic, in order to limit iteration only for the clipped overlay region.
This commit is contained in:
parent
1c808943b8
commit
f164228fd7
@ -30,7 +30,7 @@
|
||||
|
||||
#define LIBAVFILTER_VERSION_MAJOR 3
|
||||
#define LIBAVFILTER_VERSION_MINOR 42
|
||||
#define LIBAVFILTER_VERSION_MICRO 101
|
||||
#define LIBAVFILTER_VERSION_MICRO 102
|
||||
|
||||
#define LIBAVFILTER_VERSION_INT AV_VERSION_INT(LIBAVFILTER_VERSION_MAJOR, \
|
||||
LIBAVFILTER_VERSION_MINOR, \
|
||||
|
@ -275,13 +275,12 @@ static int config_input_overlay(AVFilterLink *inlink)
|
||||
if (over->x < 0 || over->y < 0 ||
|
||||
over->x + var_values[VAR_OVERLAY_W] > var_values[VAR_MAIN_W] ||
|
||||
over->y + var_values[VAR_OVERLAY_H] > var_values[VAR_MAIN_H]) {
|
||||
av_log(ctx, AV_LOG_ERROR,
|
||||
av_log(ctx, AV_LOG_WARNING,
|
||||
"Overlay area (%d,%d)<->(%d,%d) not within the main area (0,0)<->(%d,%d) or zero-sized\n",
|
||||
over->x, over->y,
|
||||
(int)(over->x + var_values[VAR_OVERLAY_W]),
|
||||
(int)(over->y + var_values[VAR_OVERLAY_H]),
|
||||
(int)var_values[VAR_MAIN_W], (int)var_values[VAR_MAIN_H]);
|
||||
return AVERROR(EINVAL);
|
||||
}
|
||||
return 0;
|
||||
|
||||
@ -314,23 +313,23 @@ static int config_output(AVFilterLink *outlink)
|
||||
|
||||
/**
|
||||
* Blend image in src to destination buffer dst at position (x, y).
|
||||
*
|
||||
* It is assumed that the src image at position (x, y) is contained in
|
||||
* dst.
|
||||
*/
|
||||
static void blend_image(AVFilterContext *ctx,
|
||||
AVFilterBufferRef *dst, AVFilterBufferRef *src,
|
||||
int x, int y)
|
||||
{
|
||||
OverlayContext *over = ctx->priv;
|
||||
int i, j, k;
|
||||
int width = src->video->w;
|
||||
int height = src->video->h;
|
||||
int i, imax, j, jmax, k, kmax;
|
||||
const int src_w = src->video->w;
|
||||
const int src_h = src->video->h;
|
||||
const int dst_w = dst->video->w;
|
||||
const int dst_h = dst->video->h;
|
||||
|
||||
if (x >= dst_w || x+dst_w < 0 ||
|
||||
y >= dst_h || y+dst_h < 0)
|
||||
return; /* no intersection */
|
||||
|
||||
if (over->main_is_packed_rgb) {
|
||||
uint8_t *dp = dst->data[0] + x * over->main_pix_step[0] +
|
||||
y * dst->linesize[0];
|
||||
uint8_t *sp = src->data[0];
|
||||
uint8_t alpha; ///< the amount of overlay to blend on to main
|
||||
const int dr = over->main_rgba_map[R];
|
||||
const int dg = over->main_rgba_map[G];
|
||||
@ -343,9 +342,18 @@ static void blend_image(AVFilterContext *ctx,
|
||||
const int sa = over->overlay_rgba_map[A];
|
||||
const int sstep = over->overlay_pix_step[0];
|
||||
const int main_has_alpha = over->main_has_alpha;
|
||||
for (i = 0; i < height; i++) {
|
||||
uint8_t *d = dp, *s = sp;
|
||||
for (j = 0; j < width; j++) {
|
||||
uint8_t *s, *sp, *d, *dp;
|
||||
|
||||
i = FFMAX(-y, 0);
|
||||
sp = src->data[0] + i * src->linesize[0];
|
||||
dp = dst->data[0] + (y+i) * dst->linesize[0];
|
||||
|
||||
for (imax = FFMIN(-y + dst_h, src_h); i < imax; i++) {
|
||||
j = FFMAX(-x, 0);
|
||||
s = sp + j * sstep;
|
||||
d = dp + (x+j) * dstep;
|
||||
|
||||
for (jmax = FFMIN(-x + dst_w, src_w); j < jmax; j++) {
|
||||
alpha = s[sa];
|
||||
|
||||
// if the main channel has an alpha channel, alpha has to be calculated
|
||||
@ -391,13 +399,19 @@ static void blend_image(AVFilterContext *ctx,
|
||||
} else {
|
||||
const int main_has_alpha = over->main_has_alpha;
|
||||
if (main_has_alpha) {
|
||||
uint8_t *da = dst->data[3] + x * over->main_pix_step[3] +
|
||||
y * dst->linesize[3];
|
||||
uint8_t *sa = src->data[3];
|
||||
uint8_t alpha; ///< the amount of overlay to blend on to main
|
||||
for (i = 0; i < height; i++) {
|
||||
uint8_t *d = da, *s = sa;
|
||||
for (j = 0; j < width; j++) {
|
||||
uint8_t *s, *sa, *d, *da;
|
||||
|
||||
i = FFMAX(-y, 0);
|
||||
sa = src->data[3] + i * src->linesize[3];
|
||||
da = dst->data[3] + (y+i) * dst->linesize[3];
|
||||
|
||||
for (imax = FFMIN(-y + dst_h, src_h); i < imax; i++) {
|
||||
j = FFMAX(-x, 0);
|
||||
s = sa + j;
|
||||
d = da + x+j;
|
||||
|
||||
for (jmax = FFMIN(-x + dst_w, src_w); j < jmax; j++) {
|
||||
alpha = *s;
|
||||
if (alpha != 0 && alpha != 255) {
|
||||
uint8_t alpha_d = *d;
|
||||
@ -423,24 +437,36 @@ static void blend_image(AVFilterContext *ctx,
|
||||
for (i = 0; i < 3; i++) {
|
||||
int hsub = i ? over->hsub : 0;
|
||||
int vsub = i ? over->vsub : 0;
|
||||
uint8_t *dp = dst->data[i] + (x >> hsub) +
|
||||
(y >> vsub) * dst->linesize[i];
|
||||
uint8_t *sp = src->data[i];
|
||||
uint8_t *ap = src->data[3];
|
||||
int wp = FFALIGN(width, 1<<hsub) >> hsub;
|
||||
int hp = FFALIGN(height, 1<<vsub) >> vsub;
|
||||
for (j = 0; j < hp; j++) {
|
||||
uint8_t *d = dp, *s = sp, *a = ap;
|
||||
for (k = 0; k < wp; k++) {
|
||||
// average alpha for color components, improve quality
|
||||
int src_wp = FFALIGN(src_w, 1<<hsub) >> hsub;
|
||||
int src_hp = FFALIGN(src_h, 1<<vsub) >> vsub;
|
||||
int dst_wp = FFALIGN(dst_w, 1<<hsub) >> hsub;
|
||||
int dst_hp = FFALIGN(dst_h, 1<<vsub) >> vsub;
|
||||
int yp = y>>vsub;
|
||||
int xp = x>>hsub;
|
||||
uint8_t *s, *sp, *d, *dp, *a, *ap;
|
||||
|
||||
j = FFMAX(-yp, 0);
|
||||
sp = src->data[i] + j * src->linesize[i];
|
||||
dp = dst->data[i] + (yp+j) * dst->linesize[i];
|
||||
ap = src->data[3] + (j<<vsub) * src->linesize[3];
|
||||
|
||||
for (jmax = FFMIN(-yp + dst_hp, src_hp); j < jmax; j++) {
|
||||
k = FFMAX(-xp, 0);
|
||||
d = dp + xp+k;
|
||||
s = sp + k;
|
||||
a = ap + (k<<hsub);
|
||||
|
||||
for (kmax = FFMIN(-xp + dst_wp, src_wp); k < kmax; k++) {
|
||||
int alpha_v, alpha_h, alpha;
|
||||
if (hsub && vsub && j+1 < hp && k+1 < wp) {
|
||||
|
||||
// average alpha for color components, improve quality
|
||||
if (hsub && vsub && j+1 < src_hp && k+1 < src_wp) {
|
||||
alpha = (a[0] + a[src->linesize[3]] +
|
||||
a[1] + a[src->linesize[3]+1]) >> 2;
|
||||
} else if (hsub || vsub) {
|
||||
alpha_h = hsub && k+1 < wp ?
|
||||
alpha_h = hsub && k+1 < src_wp ?
|
||||
(a[0] + a[1]) >> 1 : a[0];
|
||||
alpha_v = vsub && j+1 < hp ?
|
||||
alpha_v = vsub && j+1 < src_hp ?
|
||||
(a[0] + a[src->linesize[3]]) >> 1 : a[0];
|
||||
alpha = (alpha_v + alpha_h) >> 1;
|
||||
} else
|
||||
@ -450,13 +476,13 @@ static void blend_image(AVFilterContext *ctx,
|
||||
if (main_has_alpha && alpha != 0 && alpha != 255) {
|
||||
// average alpha for color components, improve quality
|
||||
uint8_t alpha_d;
|
||||
if (hsub && vsub && j+1 < hp && k+1 < wp) {
|
||||
if (hsub && vsub && j+1 < src_hp && k+1 < src_wp) {
|
||||
alpha_d = (d[0] + d[src->linesize[3]] +
|
||||
d[1] + d[src->linesize[3]+1]) >> 2;
|
||||
} else if (hsub || vsub) {
|
||||
alpha_h = hsub && k+1 < wp ?
|
||||
alpha_h = hsub && k+1 < src_wp ?
|
||||
(d[0] + d[1]) >> 1 : d[0];
|
||||
alpha_v = vsub && j+1 < hp ?
|
||||
alpha_v = vsub && j+1 < src_hp ?
|
||||
(d[0] + d[src->linesize[3]]) >> 1 : d[0];
|
||||
alpha_d = (alpha_v + alpha_h) >> 1;
|
||||
} else
|
||||
|
Loading…
Reference in New Issue
Block a user