avfilter.c 23.8 KB
Newer Older
V
Vitor Sessak 已提交
1
/*
2
 * filter layer
3
 * Copyright (c) 2007 Bobby Bingham
V
Vitor Sessak 已提交
4
 *
5
 * This file is part of Libav.
V
Vitor Sessak 已提交
6
 *
7
 * Libav is free software; you can redistribute it and/or
V
Vitor Sessak 已提交
8 9 10 11
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
12
 * Libav is distributed in the hope that it will be useful,
V
Vitor Sessak 已提交
13 14 15 16 17
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with Libav; if not, write to the Free Software
V
Vitor Sessak 已提交
19 20 21
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

22 23
/* #define DEBUG */

24
#include "libavutil/pixdesc.h"
25
#include "libavutil/rational.h"
26 27
#include "libavutil/audioconvert.h"
#include "libavutil/imgutils.h"
28
#include "libavcodec/avcodec.h"
V
Vitor Sessak 已提交
29
#include "avfilter.h"
30
#include "internal.h"
V
Vitor Sessak 已提交
31

S
Stefano Sabatini 已提交
32 33 34 35
unsigned avfilter_version(void) {
    return LIBAVFILTER_VERSION_INT;
}

36
const char *avfilter_configuration(void)
37
{
38
    return LIBAV_CONFIGURATION;
39 40
}

41
const char *avfilter_license(void)
42 43
{
#define LICENSE_PREFIX "libavfilter license: "
44
    return LICENSE_PREFIX LIBAV_LICENSE + sizeof(LICENSE_PREFIX) - 1;
45 46
}

47
AVFilterBufferRef *avfilter_ref_buffer(AVFilterBufferRef *ref, int pmask)
V
Vitor Sessak 已提交
48
{
49
    AVFilterBufferRef *ret = av_malloc(sizeof(AVFilterBufferRef));
50 51
    if (!ret)
        return NULL;
V
Vitor Sessak 已提交
52
    *ret = *ref;
53 54
    if (ref->type == AVMEDIA_TYPE_VIDEO) {
        ret->video = av_malloc(sizeof(AVFilterBufferRefVideoProps));
55 56 57 58
        if (!ret->video) {
            av_free(ret);
            return NULL;
        }
59
        *ret->video = *ref->video;
60 61 62 63 64 65 66
    } else if (ref->type == AVMEDIA_TYPE_AUDIO) {
        ret->audio = av_malloc(sizeof(AVFilterBufferRefAudioProps));
        if (!ret->audio) {
            av_free(ret);
            return NULL;
        }
        *ret->audio = *ref->audio;
67
    }
68
    ret->perms &= pmask;
69
    ret->buf->refcount ++;
V
Vitor Sessak 已提交
70 71 72
    return ret;
}

73
void avfilter_unref_buffer(AVFilterBufferRef *ref)
V
Vitor Sessak 已提交
74
{
75 76
    if (!ref)
        return;
77
    if (!(--ref->buf->refcount))
78
        ref->buf->free(ref->buf);
79
    av_free(ref->video);
80
    av_free(ref->audio);
V
Vitor Sessak 已提交
81 82 83
    av_free(ref);
}

84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
void avfilter_insert_pad(unsigned idx, unsigned *count, size_t padidx_off,
                         AVFilterPad **pads, AVFilterLink ***links,
                         AVFilterPad *newpad)
{
    unsigned i;

    idx = FFMIN(idx, *count);

    *pads  = av_realloc(*pads,  sizeof(AVFilterPad)   * (*count + 1));
    *links = av_realloc(*links, sizeof(AVFilterLink*) * (*count + 1));
    memmove(*pads +idx+1, *pads +idx, sizeof(AVFilterPad)   * (*count-idx));
    memmove(*links+idx+1, *links+idx, sizeof(AVFilterLink*) * (*count-idx));
    memcpy(*pads+idx, newpad, sizeof(AVFilterPad));
    (*links)[idx] = NULL;

99 100 101 102
    (*count)++;
    for (i = idx+1; i < *count; i++)
        if (*links[i])
            (*(unsigned *)((uint8_t *) *links[i] + padidx_off))++;
103 104
}

V
Vitor Sessak 已提交
105 106 107 108 109
int avfilter_link(AVFilterContext *src, unsigned srcpad,
                  AVFilterContext *dst, unsigned dstpad)
{
    AVFilterLink *link;

110 111
    if (src->output_count <= srcpad || dst->input_count <= dstpad ||
        src->outputs[srcpad]        || dst->inputs[dstpad])
V
Vitor Sessak 已提交
112 113
        return -1;

114 115 116 117 118 119 120
    if (src->output_pads[srcpad].type != dst->input_pads[dstpad].type) {
        av_log(src, AV_LOG_ERROR,
               "Media type mismatch between the '%s' filter output pad %d and the '%s' filter input pad %d\n",
               src->name, srcpad, dst->name, dstpad);
        return AVERROR(EINVAL);
    }

V
Vitor Sessak 已提交
121
    src->outputs[srcpad] =
V
Vitor Sessak 已提交
122
    dst-> inputs[dstpad] = link = av_mallocz(sizeof(AVFilterLink));
V
Vitor Sessak 已提交
123

V
Vitor Sessak 已提交
124 125
    link->src     = src;
    link->dst     = dst;
126 127
    link->srcpad  = &src->output_pads[srcpad];
    link->dstpad  = &dst->input_pads[dstpad];
128
    link->type    = src->output_pads[srcpad].type;
129
    assert(PIX_FMT_NONE == -1 && AV_SAMPLE_FMT_NONE == -1);
130
    link->format  = -1;
131 132 133 134

    return 0;
}

135
int avfilter_insert_filter(AVFilterLink *link, AVFilterContext *filt,
136
                           unsigned filt_srcpad_idx, unsigned filt_dstpad_idx)
137
{
138
    int ret;
139 140
    unsigned dstpad_idx = link->dstpad - link->dst->input_pads;

141 142 143
    av_log(link->dst, AV_LOG_INFO, "auto-inserting filter '%s' "
           "between the filter '%s' and the filter '%s'\n",
           filt->name, link->src->name, link->dst->name);
144

145
    link->dst->inputs[dstpad_idx] = NULL;
146
    if ((ret = avfilter_link(filt, filt_dstpad_idx, link->dst, dstpad_idx)) < 0) {
147
        /* failed to link output filter to new filter */
148
        link->dst->inputs[dstpad_idx] = link;
149
        return ret;
150 151 152 153
    }

    /* re-hookup the link to the new destination filter we inserted */
    link->dst = filt;
154 155
    link->dstpad = &filt->input_pads[filt_srcpad_idx];
    filt->inputs[filt_srcpad_idx] = link;
156

157
    /* if any information on supported media formats already exists on the
158
     * link, we need to preserve that */
159
    if (link->out_formats)
160
        avfilter_formats_changeref(&link->out_formats,
161
                                   &filt->outputs[filt_dstpad_idx]->out_formats);
162

163 164 165
    return 0;
}

166
int avfilter_config_links(AVFilterContext *filter)
167 168
{
    int (*config_link)(AVFilterLink *);
169
    unsigned i;
170
    int ret;
171

172
    for (i = 0; i < filter->input_count; i ++) {
V
Vitor Sessak 已提交
173
        AVFilterLink *link = filter->inputs[i];
174

175
        if (!link) continue;
176

177
        switch (link->init_state) {
178 179 180
        case AVLINK_INIT:
            continue;
        case AVLINK_STARTINIT:
181 182
            av_log(filter, AV_LOG_INFO, "circular filter chain detected\n");
            return 0;
183 184 185
        case AVLINK_UNINIT:
            link->init_state = AVLINK_STARTINIT;

186 187
            if ((ret = avfilter_config_links(link->src)) < 0)
                return ret;
V
Vitor Sessak 已提交
188

189
            if (!(config_link = link->srcpad->config_props))
V
Vitor Sessak 已提交
190
                config_link  = avfilter_default_config_output_link;
191 192
            if ((ret = config_link(link)) < 0)
                return ret;
193

194
            if (link->time_base.num == 0 && link->time_base.den == 0)
195 196
                link->time_base = link->src && link->src->input_count ?
                    link->src->inputs[0]->time_base : AV_TIME_BASE_Q;
197

198 199 200 201
            if (link->sample_aspect_ratio.num == 0 && link->sample_aspect_ratio.den == 0)
                link->sample_aspect_ratio = link->src->input_count ?
                    link->src->inputs[0]->sample_aspect_ratio : (AVRational){1,1};

202 203 204 205 206 207
            if (link->sample_rate == 0 && link->src && link->src->input_count)
                link->sample_rate = link->src->inputs[0]->sample_rate;

            if (link->channel_layout == 0 && link->src && link->src->input_count)
                link->channel_layout = link->src->inputs[0]->channel_layout;

208
            if ((config_link = link->dstpad->config_props))
209 210
                if ((ret = config_link(link)) < 0)
                    return ret;
211

212 213 214 215
            link->init_state = AVLINK_INIT;
        }
    }

V
Vitor Sessak 已提交
216 217 218
    return 0;
}

219
#ifdef DEBUG
220
static char *ff_get_ref_perms_string(char *buf, size_t buf_size, int perms)
221
{
222
    snprintf(buf, buf_size, "%s%s%s%s%s%s",
223 224 225
             perms & AV_PERM_READ      ? "r" : "",
             perms & AV_PERM_WRITE     ? "w" : "",
             perms & AV_PERM_PRESERVE  ? "p" : "",
226
             perms & AV_PERM_REUSE     ? "u" : "",
227 228
             perms & AV_PERM_REUSE2    ? "U" : "",
             perms & AV_PERM_NEG_LINESIZES ? "n" : "");
229 230
    return buf;
}
231
#endif
232

L
Luca Barbato 已提交
233
static void ff_dlog_ref(void *ctx, AVFilterBufferRef *ref, int end)
234
{
235
    av_unused char buf[16];
L
Luca Barbato 已提交
236
    av_dlog(ctx,
237 238
            "ref[%p buf:%p refcount:%d perms:%s data:%p linesize[%d, %d, %d, %d] pts:%"PRId64" pos:%"PRId64,
            ref, ref->buf, ref->buf->refcount, ff_get_ref_perms_string(buf, sizeof(buf), ref->perms), ref->data[0],
239 240
            ref->linesize[0], ref->linesize[1], ref->linesize[2], ref->linesize[3],
            ref->pts, ref->pos);
241

242
    if (ref->video) {
243
        av_dlog(ctx, " a:%d/%d s:%dx%d i:%c iskey:%d type:%c",
244 245 246
                ref->video->pixel_aspect.num, ref->video->pixel_aspect.den,
                ref->video->w, ref->video->h,
                !ref->video->interlaced     ? 'P' :         /* Progressive  */
247 248 249
                ref->video->top_field_first ? 'T' : 'B',    /* Top / Bottom */
                ref->video->key_frame,
                av_get_picture_type_char(ref->video->pict_type));
250
    }
251
    if (ref->audio) {
L
Luca Barbato 已提交
252
        av_dlog(ctx, " cl:%"PRId64"d sn:%d s:%d sr:%d p:%d",
253
                ref->audio->channel_layout,
254
                ref->audio->nb_samples,
255 256 257 258 259
                ref->audio->size,
                ref->audio->sample_rate,
                ref->audio->planar);
    }

L
Luca Barbato 已提交
260
    av_dlog(ctx, "]%s", end ? "\n" : "");
261 262
}

L
Luca Barbato 已提交
263
static void ff_dlog_link(void *ctx, AVFilterLink *link, int end)
264
{
265
    if (link->type == AVMEDIA_TYPE_VIDEO) {
L
Luca Barbato 已提交
266
        av_dlog(ctx,
S
Stefano Sabatini 已提交
267 268 269 270 271 272
                "link[%p s:%dx%d fmt:%-16s %-16s->%-16s]%s",
                link, link->w, link->h,
                av_pix_fmt_descriptors[link->format].name,
                link->src ? link->src->filter->name : "",
                link->dst ? link->dst->filter->name : "",
                end ? "\n" : "");
273 274 275 276
    } else {
        char buf[128];
        av_get_channel_layout_string(buf, sizeof(buf), -1, link->channel_layout);

L
Luca Barbato 已提交
277
        av_dlog(ctx,
278 279 280 281 282 283 284
                "link[%p r:%"PRId64" cl:%s fmt:%-16s %-16s->%-16s]%s",
                link, link->sample_rate, buf,
                av_get_sample_fmt_name(link->format),
                link->src ? link->src->filter->name : "",
                link->dst ? link->dst->filter->name : "",
                end ? "\n" : "");
    }
285 286
}

L
Luca Barbato 已提交
287
#define FF_DPRINTF_START(ctx, func) av_dlog(NULL, "%-16s: ", #func)
288

289
AVFilterBufferRef *avfilter_get_video_buffer(AVFilterLink *link, int perms, int w, int h)
V
Vitor Sessak 已提交
290
{
291
    AVFilterBufferRef *ret = NULL;
V
Vitor Sessak 已提交
292

293
    av_unused char buf[16];
L
Luca Barbato 已提交
294 295
    FF_DPRINTF_START(NULL, get_video_buffer); ff_dlog_link(NULL, link, 0);
    av_dlog(NULL, " perms:%s w:%d h:%d\n", ff_get_ref_perms_string(buf, sizeof(buf), perms), w, h);
296

297 298
    if (link->dstpad->get_video_buffer)
        ret = link->dstpad->get_video_buffer(link, perms, w, h);
299

300
    if (!ret)
301
        ret = avfilter_default_get_video_buffer(link, perms, w, h);
V
Vitor Sessak 已提交
302

303 304 305
    if (ret)
        ret->type = AVMEDIA_TYPE_VIDEO;

L
Luca Barbato 已提交
306
    FF_DPRINTF_START(NULL, get_video_buffer); ff_dlog_link(NULL, link, 0); av_dlog(NULL, " returning "); ff_dlog_ref(NULL, ret, 1);
307

V
Vitor Sessak 已提交
308 309 310
    return ret;
}

311 312 313 314 315 316 317 318 319 320 321 322 323 324 325
AVFilterBufferRef *
avfilter_get_video_buffer_ref_from_arrays(uint8_t *data[4], int linesize[4], int perms,
                                          int w, int h, enum PixelFormat format)
{
    AVFilterBuffer *pic = av_mallocz(sizeof(AVFilterBuffer));
    AVFilterBufferRef *picref = av_mallocz(sizeof(AVFilterBufferRef));

    if (!pic || !picref)
        goto fail;

    picref->buf = pic;
    picref->buf->free = ff_avfilter_default_free_buffer;
    if (!(picref->video = av_mallocz(sizeof(AVFilterBufferRefVideoProps))))
        goto fail;

326 327
    pic->w = picref->video->w = w;
    pic->h = picref->video->h = h;
328 329 330 331 332 333

    /* make sure the buffer gets read permission or it's useless for output */
    picref->perms = perms | AV_PERM_READ;

    pic->refcount = 1;
    picref->type = AVMEDIA_TYPE_VIDEO;
334
    pic->format = picref->format = format;
335

336 337
    memcpy(pic->data,        data,          4*sizeof(data[0]));
    memcpy(pic->linesize,    linesize,      4*sizeof(linesize[0]));
338 339 340 341 342 343 344 345 346 347 348 349 350
    memcpy(picref->data,     pic->data,     sizeof(picref->data));
    memcpy(picref->linesize, pic->linesize, sizeof(picref->linesize));

    return picref;

fail:
    if (picref && picref->video)
        av_free(picref->video);
    av_free(picref);
    av_free(pic);
    return NULL;
}

351
AVFilterBufferRef *avfilter_get_audio_buffer(AVFilterLink *link, int perms,
352
                                             enum AVSampleFormat sample_fmt, int size,
M
Mans Rullgard 已提交
353
                                             uint64_t channel_layout, int planar)
354 355 356
{
    AVFilterBufferRef *ret = NULL;

357 358
    if (link->dstpad->get_audio_buffer)
        ret = link->dstpad->get_audio_buffer(link, perms, sample_fmt, size, channel_layout, planar);
359 360 361 362 363 364 365 366 367 368

    if (!ret)
        ret = avfilter_default_get_audio_buffer(link, perms, sample_fmt, size, channel_layout, planar);

    if (ret)
        ret->type = AVMEDIA_TYPE_AUDIO;

    return ret;
}

369
int avfilter_request_frame(AVFilterLink *link)
V
Vitor Sessak 已提交
370
{
L
Luca Barbato 已提交
371
    FF_DPRINTF_START(NULL, request_frame); ff_dlog_link(NULL, link, 1);
372

373 374
    if (link->srcpad->request_frame)
        return link->srcpad->request_frame(link);
375
    else if (link->src->inputs[0])
376 377
        return avfilter_request_frame(link->src->inputs[0]);
    else return -1;
V
Vitor Sessak 已提交
378 379
}

380 381
int avfilter_poll_frame(AVFilterLink *link)
{
382
    int i, min = INT_MAX;
383

384 385
    if (link->srcpad->poll_frame)
        return link->srcpad->poll_frame(link);
V
Vitor Sessak 已提交
386

387
    for (i = 0; i < link->src->input_count; i++) {
388
        int val;
389
        if (!link->src->inputs[i])
V
Vitor Sessak 已提交
390
            return -1;
391 392
        val = avfilter_poll_frame(link->src->inputs[i]);
        min = FFMIN(min, val);
V
Vitor Sessak 已提交
393
    }
394 395 396 397

    return min;
}

V
Vitor Sessak 已提交
398 399
/* XXX: should we do the duplicating of the picture ref here, instead of
 * forcing the source filter to do it? */
400
void avfilter_start_frame(AVFilterLink *link, AVFilterBufferRef *picref)
V
Vitor Sessak 已提交
401
{
402
    void (*start_frame)(AVFilterLink *, AVFilterBufferRef *);
403
    AVFilterPad *dst = link->dstpad;
404
    int perms = picref->perms;
V
Vitor Sessak 已提交
405

L
Luca Barbato 已提交
406
    FF_DPRINTF_START(NULL, start_frame); ff_dlog_link(NULL, link, 0); av_dlog(NULL, " "); ff_dlog_ref(NULL, picref, 1);
407

408
    if (!(start_frame = dst->start_frame))
V
Vitor Sessak 已提交
409 410
        start_frame = avfilter_default_start_frame;

411 412
    if (picref->linesize[0] < 0)
        perms |= AV_PERM_NEG_LINESIZES;
413
    /* prepare to copy the picture if it has insufficient permissions */
414
    if ((dst->min_perms & perms) != dst->min_perms || dst->rej_perms & perms) {
415
        av_log(link->dst, AV_LOG_DEBUG,
416
                "frame copy needed (have perms %x, need %x, reject %x)\n",
417
                picref->perms,
418
                link->dstpad->min_perms, link->dstpad->rej_perms);
419

420
        link->cur_buf = avfilter_get_video_buffer(link, dst->min_perms, link->w, link->h);
S
S.N. Hemanth Meenakshisundaram 已提交
421 422
        link->src_buf = picref;
        avfilter_copy_buffer_ref_props(link->cur_buf, link->src_buf);
423 424
    }
    else
S
S.N. Hemanth Meenakshisundaram 已提交
425
        link->cur_buf = picref;
426

S
S.N. Hemanth Meenakshisundaram 已提交
427
    start_frame(link, link->cur_buf);
V
Vitor Sessak 已提交
428 429 430 431 432 433
}

void avfilter_end_frame(AVFilterLink *link)
{
    void (*end_frame)(AVFilterLink *);

434
    if (!(end_frame = link->dstpad->end_frame))
435 436 437 438
        end_frame = avfilter_default_end_frame;

    end_frame(link);

439 440
    /* unreference the source picture if we're feeding the destination filter
     * a copied version dues to permission issues */
441
    if (link->src_buf) {
S
S.N. Hemanth Meenakshisundaram 已提交
442 443
        avfilter_unref_buffer(link->src_buf);
        link->src_buf = NULL;
444
    }
V
Vitor Sessak 已提交
445 446
}

447
void avfilter_draw_slice(AVFilterLink *link, int y, int h, int slice_dir)
V
Vitor Sessak 已提交
448
{
449
    uint8_t *src[4], *dst[4];
450
    int i, j, vsub;
451
    void (*draw_slice)(AVFilterLink *, int, int, int);
452

L
Luca Barbato 已提交
453
    FF_DPRINTF_START(NULL, draw_slice); ff_dlog_link(NULL, link, 0); av_dlog(NULL, " y:%d h:%d dir:%d\n", y, h, slice_dir);
454

455
    /* copy the slice if needed for permission reasons */
456
    if (link->src_buf) {
457
        vsub = av_pix_fmt_descriptors[link->format].log2_chroma_h;
458

459 460
        for (i = 0; i < 4; i++) {
            if (link->src_buf->data[i]) {
S
S.N. Hemanth Meenakshisundaram 已提交
461
                src[i] = link->src_buf-> data[i] +
462
                    (y >> (i==1 || i==2 ? vsub : 0)) * link->src_buf-> linesize[i];
S
S.N. Hemanth Meenakshisundaram 已提交
463
                dst[i] = link->cur_buf->data[i] +
464
                    (y >> (i==1 || i==2 ? vsub : 0)) * link->cur_buf->linesize[i];
465 466 467
            } else
                src[i] = dst[i] = NULL;
        }
V
Vitor Sessak 已提交
468

469
        for (i = 0; i < 4; i++) {
V
Vitor Sessak 已提交
470
            int planew =
471
                av_image_get_linesize(link->format, link->cur_buf->video->w, i);
V
Vitor Sessak 已提交
472

473
            if (!src[i]) continue;
474

475
            for (j = 0; j < h >> (i==1 || i==2 ? vsub : 0); j++) {
V
Vitor Sessak 已提交
476
                memcpy(dst[i], src[i], planew);
477
                src[i] += link->src_buf->linesize[i];
S
S.N. Hemanth Meenakshisundaram 已提交
478
                dst[i] += link->cur_buf->linesize[i];
479 480 481 482
            }
        }
    }

483
    if (!(draw_slice = link->dstpad->draw_slice))
484
        draw_slice = avfilter_default_draw_slice;
485
    draw_slice(link, y, h, slice_dir);
V
Vitor Sessak 已提交
486 487
}

488 489 490
void avfilter_filter_samples(AVFilterLink *link, AVFilterBufferRef *samplesref)
{
    void (*filter_samples)(AVFilterLink *, AVFilterBufferRef *);
491
    AVFilterPad *dst = link->dstpad;
492

493 494
    FF_DPRINTF_START(NULL, filter_samples); ff_dlog_link(NULL, link, 1);

495 496 497 498 499 500 501 502 503
    if (!(filter_samples = dst->filter_samples))
        filter_samples = avfilter_default_filter_samples;

    /* prepare to copy the samples if the buffer has insufficient permissions */
    if ((dst->min_perms & samplesref->perms) != dst->min_perms ||
        dst->rej_perms & samplesref->perms) {

        av_log(link->dst, AV_LOG_DEBUG,
               "Copying audio data in avfilter (have perms %x, need %x, reject %x)\n",
504
               samplesref->perms, link->dstpad->min_perms, link->dstpad->rej_perms);
505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523

        link->cur_buf = avfilter_default_get_audio_buffer(link, dst->min_perms,
                                                          samplesref->format,
                                                          samplesref->audio->size,
                                                          samplesref->audio->channel_layout,
                                                          samplesref->audio->planar);
        link->cur_buf->pts                = samplesref->pts;
        link->cur_buf->audio->sample_rate = samplesref->audio->sample_rate;

        /* Copy actual data into new samples buffer */
        memcpy(link->cur_buf->data[0], samplesref->data[0], samplesref->audio->size);

        avfilter_unref_buffer(samplesref);
    } else
        link->cur_buf = samplesref;

    filter_samples(link, link->cur_buf);
}

524 525 526 527 528
#define MAX_REGISTERED_AVFILTERS_NB 64

static AVFilter *registered_avfilters[MAX_REGISTERED_AVFILTERS_NB + 1];

static int next_registered_avfilter_idx = 0;
529

V
Vitor Sessak 已提交
530
AVFilter *avfilter_get_by_name(const char *name)
V
Vitor Sessak 已提交
531
{
532
    int i;
533

534 535 536
    for (i = 0; registered_avfilters[i]; i++)
        if (!strcmp(registered_avfilters[i]->name, name))
            return registered_avfilters[i];
V
Vitor Sessak 已提交
537 538 539 540

    return NULL;
}

541
int avfilter_register(AVFilter *filter)
V
Vitor Sessak 已提交
542
{
543 544
    if (next_registered_avfilter_idx == MAX_REGISTERED_AVFILTERS_NB)
        return -1;
545

546 547
    registered_avfilters[next_registered_avfilter_idx++] = filter;
    return 0;
V
Vitor Sessak 已提交
548 549
}

S
Stefano Sabatini 已提交
550 551 552 553 554
AVFilter **av_filter_next(AVFilter **filter)
{
    return filter ? ++filter : &registered_avfilters[0];
}

V
Vitor Sessak 已提交
555 556
void avfilter_uninit(void)
{
557 558
    memset(registered_avfilters, 0, sizeof(registered_avfilters));
    next_registered_avfilter_idx = 0;
V
Vitor Sessak 已提交
559 560 561 562 563 564
}

static int pad_count(const AVFilterPad *pads)
{
    int count;

V
Vitor Sessak 已提交
565
    for(count = 0; pads->name; count ++) pads ++;
V
Vitor Sessak 已提交
566 567 568 569 570 571 572 573 574
    return count;
}

static const char *filter_name(void *p)
{
    AVFilterContext *filter = p;
    return filter->filter->name;
}

M
Måns Rullgård 已提交
575 576
static const AVClass avfilter_class = {
    "AVFilter",
577 578 579
    filter_name,
    NULL,
    LIBAVUTIL_VERSION_INT,
M
Måns Rullgård 已提交
580 581
};

582
int avfilter_open(AVFilterContext **filter_ctx, AVFilter *filter, const char *inst_name)
V
Vitor Sessak 已提交
583
{
V
Vitor Sessak 已提交
584
    AVFilterContext *ret;
585
    *filter_ctx = NULL;
V
Vitor Sessak 已提交
586 587

    if (!filter)
588
        return AVERROR(EINVAL);
V
Vitor Sessak 已提交
589

590
    ret = av_mallocz(sizeof(AVFilterContext));
591 592
    if (!ret)
        return AVERROR(ENOMEM);
V
Vitor Sessak 已提交
593

M
Måns Rullgård 已提交
594
    ret->av_class = &avfilter_class;
V
Vitor Sessak 已提交
595
    ret->filter   = filter;
596
    ret->name     = inst_name ? av_strdup(inst_name) : NULL;
597
    if (filter->priv_size) {
R
Ronald S. Bultje 已提交
598
        ret->priv     = av_mallocz(filter->priv_size);
599 600 601
        if (!ret->priv)
            goto err;
    }
V
Vitor Sessak 已提交
602

V
Vitor Sessak 已提交
603
    ret->input_count  = pad_count(filter->inputs);
604
    if (ret->input_count) {
605
        ret->input_pads   = av_malloc(sizeof(AVFilterPad) * ret->input_count);
606 607
        if (!ret->input_pads)
            goto err;
608 609
        memcpy(ret->input_pads, filter->inputs, sizeof(AVFilterPad) * ret->input_count);
        ret->inputs       = av_mallocz(sizeof(AVFilterLink*) * ret->input_count);
610 611
        if (!ret->inputs)
            goto err;
612
    }
613

V
Vitor Sessak 已提交
614
    ret->output_count = pad_count(filter->outputs);
615
    if (ret->output_count) {
616
        ret->output_pads  = av_malloc(sizeof(AVFilterPad) * ret->output_count);
617 618
        if (!ret->output_pads)
            goto err;
619 620
        memcpy(ret->output_pads, filter->outputs, sizeof(AVFilterPad) * ret->output_count);
        ret->outputs      = av_mallocz(sizeof(AVFilterLink*) * ret->output_count);
621 622
        if (!ret->outputs)
            goto err;
623
    }
V
Vitor Sessak 已提交
624

625 626
    *filter_ctx = ret;
    return 0;
627 628 629 630 631 632 633 634 635 636 637

err:
    av_freep(&ret->inputs);
    av_freep(&ret->input_pads);
    ret->input_count = 0;
    av_freep(&ret->outputs);
    av_freep(&ret->output_pads);
    ret->output_count = 0;
    av_freep(&ret->priv);
    av_free(ret);
    return AVERROR(ENOMEM);
V
Vitor Sessak 已提交
638 639
}

640
void avfilter_free(AVFilterContext *filter)
V
Vitor Sessak 已提交
641 642
{
    int i;
643
    AVFilterLink *link;
V
Vitor Sessak 已提交
644

645
    if (filter->filter->uninit)
V
Vitor Sessak 已提交
646 647
        filter->filter->uninit(filter);

648
    for (i = 0; i < filter->input_count; i++) {
649 650
        if ((link = filter->inputs[i])) {
            if (link->src)
651
                link->src->outputs[link->srcpad - link->src->output_pads] = NULL;
652 653
            avfilter_formats_unref(&link->in_formats);
            avfilter_formats_unref(&link->out_formats);
654
        }
655
        av_freep(&link);
V
Vitor Sessak 已提交
656
    }
657
    for (i = 0; i < filter->output_count; i++) {
658 659
        if ((link = filter->outputs[i])) {
            if (link->dst)
660
                link->dst->inputs[link->dstpad - link->dst->input_pads] = NULL;
661 662
            avfilter_formats_unref(&link->in_formats);
            avfilter_formats_unref(&link->out_formats);
663
        }
664
        av_freep(&link);
V
Vitor Sessak 已提交
665 666
    }

V
Vitor Sessak 已提交
667 668 669 670 671 672
    av_freep(&filter->name);
    av_freep(&filter->input_pads);
    av_freep(&filter->output_pads);
    av_freep(&filter->inputs);
    av_freep(&filter->outputs);
    av_freep(&filter->priv);
V
Vitor Sessak 已提交
673 674 675
    av_free(filter);
}

676
int avfilter_init_filter(AVFilterContext *filter, const char *args, void *opaque)
V
Vitor Sessak 已提交
677
{
V
Vitor Sessak 已提交
678
    int ret=0;
V
Vitor Sessak 已提交
679

680
    if (filter->filter->init)
V
Vitor Sessak 已提交
681 682
        ret = filter->filter->init(filter, args, opaque);
    return ret;
V
Vitor Sessak 已提交
683 684
}

685 686 687 688 689
int avfilter_copy_frame_props(AVFilterBufferRef *dst, const AVFrame *src)
{
    dst->pts    = src->pts;
    dst->format = src->format;

690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706
    switch (dst->type) {
    case AVMEDIA_TYPE_VIDEO:
        dst->video->w                   = src->width;
        dst->video->h                   = src->height;
        dst->video->pixel_aspect        = src->sample_aspect_ratio;
        dst->video->interlaced          = src->interlaced_frame;
        dst->video->top_field_first     = src->top_field_first;
        dst->video->key_frame           = src->key_frame;
        dst->video->pict_type           = src->pict_type;
        break;
    case AVMEDIA_TYPE_AUDIO:
        dst->audio->sample_rate         = src->sample_rate;
        dst->audio->channel_layout      = src->channel_layout;
        break;
    default:
        return AVERROR(EINVAL);
    }
707 708 709

    return 0;
}
710

711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740
int avfilter_copy_buf_props(AVFrame *dst, const AVFilterBufferRef *src)
{
    memcpy(dst->data, src->data, sizeof(dst->data));
    memcpy(dst->linesize, src->linesize, sizeof(dst->linesize));

    dst->pts     = src->pts;
    dst->format  = src->format;

    switch (src->type) {
    case AVMEDIA_TYPE_VIDEO:
        dst->width               = src->video->w;
        dst->height              = src->video->h;
        dst->sample_aspect_ratio = src->video->pixel_aspect;
        dst->interlaced_frame    = src->video->interlaced;
        dst->top_field_first     = src->video->top_field_first;
        dst->key_frame           = src->video->key_frame;
        dst->pict_type           = src->video->pict_type;
        break;
    case AVMEDIA_TYPE_AUDIO:
        dst->sample_rate         = src->audio->sample_rate;
        dst->channel_layout      = src->audio->channel_layout;
        dst->nb_samples          = src->audio->nb_samples;
        break;
    default:
        return AVERROR(EINVAL);
    }

    return 0;
}

741 742 743 744 745 746 747 748 749 750 751 752
void avfilter_copy_buffer_ref_props(AVFilterBufferRef *dst, AVFilterBufferRef *src)
{
    // copy common properties
    dst->pts             = src->pts;
    dst->pos             = src->pos;

    switch (src->type) {
    case AVMEDIA_TYPE_VIDEO: *dst->video = *src->video; break;
    case AVMEDIA_TYPE_AUDIO: *dst->audio = *src->audio; break;
    default: break;
    }
}