mirror of https://git.ffmpeg.org/ffmpeg.git
vf_transpose: switch to filter_frame
Based on patch by Anton Khirnov Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
parent
2c3b665379
commit
afded1816b
|
@ -165,106 +165,84 @@ static AVFilterBufferRef *get_video_buffer(AVFilterLink *inlink, int perms, int
|
||||||
ff_default_get_video_buffer(inlink, perms, w, h);
|
ff_default_get_video_buffer(inlink, perms, w, h);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int start_frame(AVFilterLink *inlink, AVFilterBufferRef *picref)
|
static int filter_frame(AVFilterLink *inlink, AVFilterBufferRef *in)
|
||||||
{
|
{
|
||||||
TransContext *trans = inlink->dst->priv;
|
TransContext *trans = inlink->dst->priv;
|
||||||
AVFilterLink *outlink = inlink->dst->outputs[0];
|
AVFilterLink *outlink = inlink->dst->outputs[0];
|
||||||
AVFilterBufferRef *buf_out;
|
AVFilterBufferRef *out;
|
||||||
|
int plane;
|
||||||
|
|
||||||
if (trans->passthrough)
|
if (trans->passthrough)
|
||||||
return ff_null_start_frame(inlink, picref);
|
return ff_filter_frame(outlink, in);
|
||||||
|
|
||||||
outlink->out_buf = ff_get_video_buffer(outlink, AV_PERM_WRITE,
|
out = ff_get_video_buffer(outlink, AV_PERM_WRITE, outlink->w, outlink->h);
|
||||||
outlink->w, outlink->h);
|
if (!out) {
|
||||||
if (!outlink->out_buf)
|
avfilter_unref_bufferp(&in);
|
||||||
return AVERROR(ENOMEM);
|
return AVERROR(ENOMEM);
|
||||||
|
|
||||||
outlink->out_buf->pts = picref->pts;
|
|
||||||
|
|
||||||
if (picref->video->sample_aspect_ratio.num == 0) {
|
|
||||||
outlink->out_buf->video->sample_aspect_ratio = picref->video->sample_aspect_ratio;
|
|
||||||
} else {
|
|
||||||
outlink->out_buf->video->sample_aspect_ratio.num = picref->video->sample_aspect_ratio.den;
|
|
||||||
outlink->out_buf->video->sample_aspect_ratio.den = picref->video->sample_aspect_ratio.num;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
buf_out = avfilter_ref_buffer(outlink->out_buf, ~0);
|
out->pts = in->pts;
|
||||||
if (!buf_out)
|
|
||||||
return AVERROR(ENOMEM);
|
|
||||||
return ff_start_frame(outlink, buf_out);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int end_frame(AVFilterLink *inlink)
|
if (in->video->sample_aspect_ratio.num == 0) {
|
||||||
{
|
out->video->sample_aspect_ratio = in->video->sample_aspect_ratio;
|
||||||
TransContext *trans = inlink->dst->priv;
|
} else {
|
||||||
AVFilterBufferRef *inpic = inlink->cur_buf;
|
out->video->sample_aspect_ratio.num = in->video->sample_aspect_ratio.den;
|
||||||
AVFilterBufferRef *outpic = inlink->dst->outputs[0]->out_buf;
|
out->video->sample_aspect_ratio.den = in->video->sample_aspect_ratio.num;
|
||||||
AVFilterLink *outlink = inlink->dst->outputs[0];
|
}
|
||||||
int plane, ret;
|
|
||||||
|
|
||||||
if (trans->passthrough)
|
for (plane = 0; out->data[plane]; plane++) {
|
||||||
return ff_null_end_frame(inlink);
|
|
||||||
|
|
||||||
for (plane = 0; outpic->data[plane]; plane++) {
|
|
||||||
int hsub = plane == 1 || plane == 2 ? trans->hsub : 0;
|
int hsub = plane == 1 || plane == 2 ? trans->hsub : 0;
|
||||||
int vsub = plane == 1 || plane == 2 ? trans->vsub : 0;
|
int vsub = plane == 1 || plane == 2 ? trans->vsub : 0;
|
||||||
int pixstep = trans->pixsteps[plane];
|
int pixstep = trans->pixsteps[plane];
|
||||||
int inh = inpic->video->h>>vsub;
|
int inh = in->video->h>>vsub;
|
||||||
int outw = outpic->video->w>>hsub;
|
int outw = out->video->w>>hsub;
|
||||||
int outh = outpic->video->h>>vsub;
|
int outh = out->video->h>>vsub;
|
||||||
uint8_t *out, *in;
|
uint8_t *dst, *src;
|
||||||
int outlinesize, inlinesize;
|
int dstlinesize, srclinesize;
|
||||||
int x, y;
|
int x, y;
|
||||||
|
|
||||||
out = outpic->data[plane]; outlinesize = outpic->linesize[plane];
|
dst = out->data[plane];
|
||||||
in = inpic ->data[plane]; inlinesize = inpic ->linesize[plane];
|
dstlinesize = out->linesize[plane];
|
||||||
|
src = in->data[plane];
|
||||||
|
srclinesize = in->linesize[plane];
|
||||||
|
|
||||||
if (trans->dir&1) {
|
if (trans->dir&1) {
|
||||||
in += inpic->linesize[plane] * (inh-1);
|
src += in->linesize[plane] * (inh-1);
|
||||||
inlinesize *= -1;
|
srclinesize *= -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (trans->dir&2) {
|
if (trans->dir&2) {
|
||||||
out += outpic->linesize[plane] * (outh-1);
|
dst += out->linesize[plane] * (outh-1);
|
||||||
outlinesize *= -1;
|
dstlinesize *= -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
for (y = 0; y < outh; y++) {
|
for (y = 0; y < outh; y++) {
|
||||||
switch (pixstep) {
|
switch (pixstep) {
|
||||||
case 1:
|
case 1:
|
||||||
for (x = 0; x < outw; x++)
|
for (x = 0; x < outw; x++)
|
||||||
out[x] = in[x*inlinesize + y];
|
dst[x] = src[x*srclinesize + y];
|
||||||
break;
|
break;
|
||||||
case 2:
|
case 2:
|
||||||
for (x = 0; x < outw; x++)
|
for (x = 0; x < outw; x++)
|
||||||
*((uint16_t *)(out + 2*x)) = *((uint16_t *)(in + x*inlinesize + y*2));
|
*((uint16_t *)(dst + 2*x)) = *((uint16_t *)(src + x*srclinesize + y*2));
|
||||||
break;
|
break;
|
||||||
case 3:
|
case 3:
|
||||||
for (x = 0; x < outw; x++) {
|
for (x = 0; x < outw; x++) {
|
||||||
int32_t v = AV_RB24(in + x*inlinesize + y*3);
|
int32_t v = AV_RB24(src + x*srclinesize + y*3);
|
||||||
AV_WB24(out + 3*x, v);
|
AV_WB24(dst + 3*x, v);
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
case 4:
|
case 4:
|
||||||
for (x = 0; x < outw; x++)
|
for (x = 0; x < outw; x++)
|
||||||
*((uint32_t *)(out + 4*x)) = *((uint32_t *)(in + x*inlinesize + y*4));
|
*((uint32_t *)(dst + 4*x)) = *((uint32_t *)(src + x*srclinesize + y*4));
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
out += outlinesize;
|
dst += dstlinesize;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if ((ret = ff_draw_slice(outlink, 0, outpic->video->h, 1)) < 0 ||
|
avfilter_unref_bufferp(&in);
|
||||||
(ret = ff_end_frame(outlink)) < 0)
|
return ff_filter_frame(outlink, out);
|
||||||
return ret;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int draw_slice(AVFilterLink *inlink, int y, int h, int slice_dir)
|
|
||||||
{
|
|
||||||
TransContext *trans = inlink->dst->priv;
|
|
||||||
|
|
||||||
return trans->passthrough ? ff_null_draw_slice(inlink, y, h, slice_dir) : 0;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static const AVFilterPad avfilter_vf_transpose_inputs[] = {
|
static const AVFilterPad avfilter_vf_transpose_inputs[] = {
|
||||||
|
@ -272,9 +250,7 @@ static const AVFilterPad avfilter_vf_transpose_inputs[] = {
|
||||||
.name = "default",
|
.name = "default",
|
||||||
.type = AVMEDIA_TYPE_VIDEO,
|
.type = AVMEDIA_TYPE_VIDEO,
|
||||||
.get_video_buffer= get_video_buffer,
|
.get_video_buffer= get_video_buffer,
|
||||||
.start_frame = start_frame,
|
.filter_frame = filter_frame,
|
||||||
.draw_slice = draw_slice,
|
|
||||||
.end_frame = end_frame,
|
|
||||||
.min_perms = AV_PERM_READ,
|
.min_perms = AV_PERM_READ,
|
||||||
},
|
},
|
||||||
{ NULL }
|
{ NULL }
|
||||||
|
|
Loading…
Reference in New Issue