]> git.sesse.net Git - ffmpeg/blob - libavfilter/avfilter.c
Merge commit '566b7a20fd0cab44d344329538d314454a0bcc2f'
[ffmpeg] / libavfilter / avfilter.c
1 /*
2  * filter layer
3  * Copyright (c) 2007 Bobby Bingham
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 "libavutil/avassert.h"
23 #include "libavutil/avstring.h"
24 #include "libavutil/channel_layout.h"
25 #include "libavutil/common.h"
26 #include "libavutil/eval.h"
27 #include "libavutil/imgutils.h"
28 #include "libavutil/opt.h"
29 #include "libavutil/pixdesc.h"
30 #include "libavutil/rational.h"
31 #include "libavutil/samplefmt.h"
32
33 #include "audio.h"
34 #include "avfilter.h"
35 #include "formats.h"
36 #include "internal.h"
37 #include "audio.h"
38
39 static int ff_filter_frame_framed(AVFilterLink *link, AVFrame *frame);
40
41 void ff_tlog_ref(void *ctx, AVFrame *ref, int end)
42 {
43     av_unused char buf[16];
44     ff_tlog(ctx,
45             "ref[%p buf:%p data:%p linesize[%d, %d, %d, %d] pts:%"PRId64" pos:%"PRId64,
46             ref, ref->buf, ref->data[0],
47             ref->linesize[0], ref->linesize[1], ref->linesize[2], ref->linesize[3],
48             ref->pts, av_frame_get_pkt_pos(ref));
49
50     if (ref->width) {
51         ff_tlog(ctx, " a:%d/%d s:%dx%d i:%c iskey:%d type:%c",
52                 ref->sample_aspect_ratio.num, ref->sample_aspect_ratio.den,
53                 ref->width, ref->height,
54                 !ref->interlaced_frame     ? 'P' :         /* Progressive  */
55                 ref->top_field_first ? 'T' : 'B',    /* Top / Bottom */
56                 ref->key_frame,
57                 av_get_picture_type_char(ref->pict_type));
58     }
59     if (ref->nb_samples) {
60         ff_tlog(ctx, " cl:%"PRId64"d n:%d r:%d",
61                 ref->channel_layout,
62                 ref->nb_samples,
63                 ref->sample_rate);
64     }
65
66     ff_tlog(ctx, "]%s", end ? "\n" : "");
67 }
68
69 unsigned avfilter_version(void) {
70     av_assert0(LIBAVFILTER_VERSION_MICRO >= 100);
71     return LIBAVFILTER_VERSION_INT;
72 }
73
74 const char *avfilter_configuration(void)
75 {
76     return FFMPEG_CONFIGURATION;
77 }
78
79 const char *avfilter_license(void)
80 {
81 #define LICENSE_PREFIX "libavfilter license: "
82     return LICENSE_PREFIX FFMPEG_LICENSE + sizeof(LICENSE_PREFIX) - 1;
83 }
84
85 void ff_command_queue_pop(AVFilterContext *filter)
86 {
87     AVFilterCommand *c= filter->command_queue;
88     av_freep(&c->arg);
89     av_freep(&c->command);
90     filter->command_queue= c->next;
91     av_free(c);
92 }
93
94 void ff_insert_pad(unsigned idx, unsigned *count, size_t padidx_off,
95                    AVFilterPad **pads, AVFilterLink ***links,
96                    AVFilterPad *newpad)
97 {
98     unsigned i;
99
100     idx = FFMIN(idx, *count);
101
102     *pads  = av_realloc(*pads,  sizeof(AVFilterPad)   * (*count + 1));
103     *links = av_realloc(*links, sizeof(AVFilterLink*) * (*count + 1));
104     memmove(*pads +idx+1, *pads +idx, sizeof(AVFilterPad)   * (*count-idx));
105     memmove(*links+idx+1, *links+idx, sizeof(AVFilterLink*) * (*count-idx));
106     memcpy(*pads+idx, newpad, sizeof(AVFilterPad));
107     (*links)[idx] = NULL;
108
109     (*count)++;
110     for (i = idx+1; i < *count; i++)
111         if (*links[i])
112             (*(unsigned *)((uint8_t *) *links[i] + padidx_off))++;
113 }
114
115 int avfilter_link(AVFilterContext *src, unsigned srcpad,
116                   AVFilterContext *dst, unsigned dstpad)
117 {
118     AVFilterLink *link;
119
120     if (src->nb_outputs <= srcpad || dst->nb_inputs <= dstpad ||
121         src->outputs[srcpad]      || dst->inputs[dstpad])
122         return -1;
123
124     if (src->output_pads[srcpad].type != dst->input_pads[dstpad].type) {
125         av_log(src, AV_LOG_ERROR,
126                "Media type mismatch between the '%s' filter output pad %d (%s) and the '%s' filter input pad %d (%s)\n",
127                src->name, srcpad, (char *)av_x_if_null(av_get_media_type_string(src->output_pads[srcpad].type), "?"),
128                dst->name, dstpad, (char *)av_x_if_null(av_get_media_type_string(dst-> input_pads[dstpad].type), "?"));
129         return AVERROR(EINVAL);
130     }
131
132     src->outputs[srcpad] =
133     dst-> inputs[dstpad] = link = av_mallocz(sizeof(AVFilterLink));
134
135     link->src     = src;
136     link->dst     = dst;
137     link->srcpad  = &src->output_pads[srcpad];
138     link->dstpad  = &dst->input_pads[dstpad];
139     link->type    = src->output_pads[srcpad].type;
140     av_assert0(AV_PIX_FMT_NONE == -1 && AV_SAMPLE_FMT_NONE == -1);
141     link->format  = -1;
142
143     return 0;
144 }
145
146 void avfilter_link_free(AVFilterLink **link)
147 {
148     if (!*link)
149         return;
150
151     av_frame_free(&(*link)->partial_buf);
152
153     av_freep(link);
154 }
155
156 int avfilter_link_get_channels(AVFilterLink *link)
157 {
158     return link->channels;
159 }
160
161 void avfilter_link_set_closed(AVFilterLink *link, int closed)
162 {
163     link->closed = closed;
164 }
165
166 int avfilter_insert_filter(AVFilterLink *link, AVFilterContext *filt,
167                            unsigned filt_srcpad_idx, unsigned filt_dstpad_idx)
168 {
169     int ret;
170     unsigned dstpad_idx = link->dstpad - link->dst->input_pads;
171
172     av_log(link->dst, AV_LOG_VERBOSE, "auto-inserting filter '%s' "
173            "between the filter '%s' and the filter '%s'\n",
174            filt->name, link->src->name, link->dst->name);
175
176     link->dst->inputs[dstpad_idx] = NULL;
177     if ((ret = avfilter_link(filt, filt_dstpad_idx, link->dst, dstpad_idx)) < 0) {
178         /* failed to link output filter to new filter */
179         link->dst->inputs[dstpad_idx] = link;
180         return ret;
181     }
182
183     /* re-hookup the link to the new destination filter we inserted */
184     link->dst = filt;
185     link->dstpad = &filt->input_pads[filt_srcpad_idx];
186     filt->inputs[filt_srcpad_idx] = link;
187
188     /* if any information on supported media formats already exists on the
189      * link, we need to preserve that */
190     if (link->out_formats)
191         ff_formats_changeref(&link->out_formats,
192                                    &filt->outputs[filt_dstpad_idx]->out_formats);
193
194     if (link->out_samplerates)
195         ff_formats_changeref(&link->out_samplerates,
196                                    &filt->outputs[filt_dstpad_idx]->out_samplerates);
197     if (link->out_channel_layouts)
198         ff_channel_layouts_changeref(&link->out_channel_layouts,
199                                      &filt->outputs[filt_dstpad_idx]->out_channel_layouts);
200
201     return 0;
202 }
203
204 int avfilter_config_links(AVFilterContext *filter)
205 {
206     int (*config_link)(AVFilterLink *);
207     unsigned i;
208     int ret;
209
210     for (i = 0; i < filter->nb_inputs; i ++) {
211         AVFilterLink *link = filter->inputs[i];
212         AVFilterLink *inlink;
213
214         if (!link) continue;
215
216         inlink = link->src->nb_inputs ? link->src->inputs[0] : NULL;
217         link->current_pts = AV_NOPTS_VALUE;
218
219         switch (link->init_state) {
220         case AVLINK_INIT:
221             continue;
222         case AVLINK_STARTINIT:
223             av_log(filter, AV_LOG_INFO, "circular filter chain detected\n");
224             return 0;
225         case AVLINK_UNINIT:
226             link->init_state = AVLINK_STARTINIT;
227
228             if ((ret = avfilter_config_links(link->src)) < 0)
229                 return ret;
230
231             if (!(config_link = link->srcpad->config_props)) {
232                 if (link->src->nb_inputs != 1) {
233                     av_log(link->src, AV_LOG_ERROR, "Source filters and filters "
234                                                     "with more than one input "
235                                                     "must set config_props() "
236                                                     "callbacks on all outputs\n");
237                     return AVERROR(EINVAL);
238                 }
239             } else if ((ret = config_link(link)) < 0) {
240                 av_log(link->src, AV_LOG_ERROR,
241                        "Failed to configure output pad on %s\n",
242                        link->src->name);
243                 return ret;
244             }
245
246             switch (link->type) {
247             case AVMEDIA_TYPE_VIDEO:
248                 if (!link->time_base.num && !link->time_base.den)
249                     link->time_base = inlink ? inlink->time_base : AV_TIME_BASE_Q;
250
251                 if (!link->sample_aspect_ratio.num && !link->sample_aspect_ratio.den)
252                     link->sample_aspect_ratio = inlink ?
253                         inlink->sample_aspect_ratio : (AVRational){1,1};
254
255                 if (inlink && !link->frame_rate.num && !link->frame_rate.den)
256                     link->frame_rate = inlink->frame_rate;
257
258                 if (inlink) {
259                     if (!link->w)
260                         link->w = inlink->w;
261                     if (!link->h)
262                         link->h = inlink->h;
263                 } else if (!link->w || !link->h) {
264                     av_log(link->src, AV_LOG_ERROR,
265                            "Video source filters must set their output link's "
266                            "width and height\n");
267                     return AVERROR(EINVAL);
268                 }
269                 break;
270
271             case AVMEDIA_TYPE_AUDIO:
272                 if (inlink) {
273                     if (!link->time_base.num && !link->time_base.den)
274                         link->time_base = inlink->time_base;
275                 }
276
277                 if (!link->time_base.num && !link->time_base.den)
278                     link->time_base = (AVRational) {1, link->sample_rate};
279             }
280
281             if ((config_link = link->dstpad->config_props))
282                 if ((ret = config_link(link)) < 0) {
283                     av_log(link->src, AV_LOG_ERROR,
284                            "Failed to configure input pad on %s\n",
285                            link->dst->name);
286                     return ret;
287                 }
288
289             link->init_state = AVLINK_INIT;
290         }
291     }
292
293     return 0;
294 }
295
296 void ff_tlog_link(void *ctx, AVFilterLink *link, int end)
297 {
298     if (link->type == AVMEDIA_TYPE_VIDEO) {
299         ff_tlog(ctx,
300                 "link[%p s:%dx%d fmt:%s %s->%s]%s",
301                 link, link->w, link->h,
302                 av_get_pix_fmt_name(link->format),
303                 link->src ? link->src->filter->name : "",
304                 link->dst ? link->dst->filter->name : "",
305                 end ? "\n" : "");
306     } else {
307         char buf[128];
308         av_get_channel_layout_string(buf, sizeof(buf), -1, link->channel_layout);
309
310         ff_tlog(ctx,
311                 "link[%p r:%d cl:%s fmt:%s %s->%s]%s",
312                 link, (int)link->sample_rate, buf,
313                 av_get_sample_fmt_name(link->format),
314                 link->src ? link->src->filter->name : "",
315                 link->dst ? link->dst->filter->name : "",
316                 end ? "\n" : "");
317     }
318 }
319
320 int ff_request_frame(AVFilterLink *link)
321 {
322     int ret = -1;
323     FF_TPRINTF_START(NULL, request_frame); ff_tlog_link(NULL, link, 1);
324
325     if (link->closed)
326         return AVERROR_EOF;
327     av_assert0(!link->frame_requested);
328     link->frame_requested = 1;
329     while (link->frame_requested) {
330         if (link->srcpad->request_frame)
331             ret = link->srcpad->request_frame(link);
332         else if (link->src->inputs[0])
333             ret = ff_request_frame(link->src->inputs[0]);
334         if (ret == AVERROR_EOF && link->partial_buf) {
335             AVFrame *pbuf = link->partial_buf;
336             link->partial_buf = NULL;
337             ret = ff_filter_frame_framed(link, pbuf);
338         }
339         if (ret < 0) {
340             link->frame_requested = 0;
341             if (ret == AVERROR_EOF)
342                 link->closed = 1;
343         } else {
344             av_assert0(!link->frame_requested ||
345                        link->flags & FF_LINK_FLAG_REQUEST_LOOP);
346         }
347     }
348     return ret;
349 }
350
351 int ff_poll_frame(AVFilterLink *link)
352 {
353     int i, min = INT_MAX;
354
355     if (link->srcpad->poll_frame)
356         return link->srcpad->poll_frame(link);
357
358     for (i = 0; i < link->src->nb_inputs; i++) {
359         int val;
360         if (!link->src->inputs[i])
361             return -1;
362         val = ff_poll_frame(link->src->inputs[i]);
363         min = FFMIN(min, val);
364     }
365
366     return min;
367 }
368
369 static const char *const var_names[] = {   "t",   "n",   "pos",        NULL };
370 enum                                   { VAR_T, VAR_N, VAR_POS, VAR_VARS_NB };
371
372 static int set_enable_expr(AVFilterContext *ctx, const char *expr)
373 {
374     int ret;
375     char *expr_dup;
376     AVExpr *old = ctx->enable;
377
378     if (!(ctx->filter->flags & AVFILTER_FLAG_SUPPORT_TIMELINE)) {
379         av_log(ctx, AV_LOG_ERROR, "Timeline ('enable' option) not supported "
380                "with filter '%s'\n", ctx->filter->name);
381         return AVERROR_PATCHWELCOME;
382     }
383
384     expr_dup = av_strdup(expr);
385     if (!expr_dup)
386         return AVERROR(ENOMEM);
387
388     if (!ctx->var_values) {
389         ctx->var_values = av_calloc(VAR_VARS_NB, sizeof(*ctx->var_values));
390         if (!ctx->var_values) {
391             av_free(expr_dup);
392             return AVERROR(ENOMEM);
393         }
394     }
395
396     ret = av_expr_parse((AVExpr**)&ctx->enable, expr_dup, var_names,
397                         NULL, NULL, NULL, NULL, 0, ctx->priv);
398     if (ret < 0) {
399         av_log(ctx->priv, AV_LOG_ERROR,
400                "Error when evaluating the expression '%s' for enable\n",
401                expr_dup);
402         av_free(expr_dup);
403         return ret;
404     }
405
406     av_expr_free(old);
407     av_free(ctx->enable_str);
408     ctx->enable_str = expr_dup;
409     return 0;
410 }
411
412 void ff_update_link_current_pts(AVFilterLink *link, int64_t pts)
413 {
414     if (pts == AV_NOPTS_VALUE)
415         return;
416     link->current_pts = av_rescale_q(pts, link->time_base, AV_TIME_BASE_Q);
417     /* TODO use duration */
418     if (link->graph && link->age_index >= 0)
419         ff_avfilter_graph_update_heap(link->graph, link);
420 }
421
422 int avfilter_process_command(AVFilterContext *filter, const char *cmd, const char *arg, char *res, int res_len, int flags)
423 {
424     if(!strcmp(cmd, "ping")){
425         av_strlcatf(res, res_len, "pong from:%s %s\n", filter->filter->name, filter->name);
426         return 0;
427     }else if(!strcmp(cmd, "enable")) {
428         return set_enable_expr(filter, arg);
429     }else if(filter->filter->process_command) {
430         return filter->filter->process_command(filter, cmd, arg, res, res_len, flags);
431     }
432     return AVERROR(ENOSYS);
433 }
434
435 static AVFilter *first_filter;
436
437 AVFilter *avfilter_get_by_name(const char *name)
438 {
439     AVFilter *f = NULL;
440
441     if (!name)
442         return NULL;
443
444     while ((f = avfilter_next(f)))
445         if (!strcmp(f->name, name))
446             return f;
447
448     return NULL;
449 }
450
451 int avfilter_register(AVFilter *filter)
452 {
453     AVFilter **f = &first_filter;
454     int i;
455
456     for(i=0; filter->inputs && filter->inputs[i].name; i++) {
457         const AVFilterPad *input = &filter->inputs[i];
458         av_assert0(     !input->filter_frame
459                     || (!input->start_frame && !input->end_frame));
460     }
461
462     while (*f)
463         f = &(*f)->next;
464     *f = filter;
465     filter->next = NULL;
466
467     return 0;
468 }
469
470 const AVFilter *avfilter_next(const AVFilter *prev)
471 {
472     return prev ? prev->next : first_filter;
473 }
474
475 #if FF_API_OLD_FILTER_REGISTER
476 AVFilter **av_filter_next(AVFilter **filter)
477 {
478     return filter ? &(*filter)->next : &first_filter;
479 }
480
481 void avfilter_uninit(void)
482 {
483 }
484 #endif
485
486 int avfilter_pad_count(const AVFilterPad *pads)
487 {
488     int count;
489
490     if (!pads)
491         return 0;
492
493     for(count = 0; pads->name; count ++) pads ++;
494     return count;
495 }
496
497 static const char *default_filter_name(void *filter_ctx)
498 {
499     AVFilterContext *ctx = filter_ctx;
500     return ctx->name ? ctx->name : ctx->filter->name;
501 }
502
503 static void *filter_child_next(void *obj, void *prev)
504 {
505     AVFilterContext *ctx = obj;
506     if (!prev && ctx->filter && ctx->filter->priv_class && ctx->priv)
507         return ctx->priv;
508     return NULL;
509 }
510
511 static const AVClass *filter_child_class_next(const AVClass *prev)
512 {
513     AVFilter *f = NULL;
514
515     /* find the filter that corresponds to prev */
516     while (prev && (f = avfilter_next(f)))
517         if (f->priv_class == prev)
518             break;
519
520     /* could not find filter corresponding to prev */
521     if (prev && !f)
522         return NULL;
523
524     /* find next filter with specific options */
525     while ((f = avfilter_next(f)))
526         if (f->priv_class)
527             return f->priv_class;
528
529     return NULL;
530 }
531
532 #define OFFSET(x) offsetof(AVFilterContext, x)
533 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM
534 static const AVOption filters_common_options[] = {
535     { "enable", "set enable expression", OFFSET(enable_str), AV_OPT_TYPE_STRING, {.str=NULL}, .flags = FLAGS },
536     { NULL }
537 };
538
539 static const AVClass avfilter_class = {
540     .class_name = "AVFilter",
541     .item_name  = default_filter_name,
542     .version    = LIBAVUTIL_VERSION_INT,
543     .category   = AV_CLASS_CATEGORY_FILTER,
544     .child_next = filter_child_next,
545     .option     = filters_common_options,
546     .child_class_next = filter_child_class_next,
547 };
548
549 AVFilterContext *ff_filter_alloc(const AVFilter *filter, const char *inst_name)
550 {
551     AVFilterContext *ret;
552
553     if (!filter)
554         return NULL;
555
556     ret = av_mallocz(sizeof(AVFilterContext));
557     if (!ret)
558         return NULL;
559
560     ret->av_class = &avfilter_class;
561     ret->filter   = filter;
562     ret->name     = inst_name ? av_strdup(inst_name) : NULL;
563     if (filter->priv_size) {
564         ret->priv     = av_mallocz(filter->priv_size);
565         if (!ret->priv)
566             goto err;
567     }
568
569     if (filter->priv_class) {
570         *(const AVClass**)ret->priv = filter->priv_class;
571         av_opt_set_defaults(ret->priv);
572     }
573
574     ret->nb_inputs = avfilter_pad_count(filter->inputs);
575     if (ret->nb_inputs ) {
576         ret->input_pads   = av_malloc(sizeof(AVFilterPad) * ret->nb_inputs);
577         if (!ret->input_pads)
578             goto err;
579         memcpy(ret->input_pads, filter->inputs, sizeof(AVFilterPad) * ret->nb_inputs);
580         ret->inputs       = av_mallocz(sizeof(AVFilterLink*) * ret->nb_inputs);
581         if (!ret->inputs)
582             goto err;
583     }
584
585     ret->nb_outputs = avfilter_pad_count(filter->outputs);
586     if (ret->nb_outputs) {
587         ret->output_pads  = av_malloc(sizeof(AVFilterPad) * ret->nb_outputs);
588         if (!ret->output_pads)
589             goto err;
590         memcpy(ret->output_pads, filter->outputs, sizeof(AVFilterPad) * ret->nb_outputs);
591         ret->outputs      = av_mallocz(sizeof(AVFilterLink*) * ret->nb_outputs);
592         if (!ret->outputs)
593             goto err;
594     }
595 #if FF_API_FOO_COUNT
596     ret->output_count = ret->nb_outputs;
597     ret->input_count  = ret->nb_inputs;
598 #endif
599
600     return ret;
601
602 err:
603     av_freep(&ret->inputs);
604     av_freep(&ret->input_pads);
605     ret->nb_inputs = 0;
606     av_freep(&ret->outputs);
607     av_freep(&ret->output_pads);
608     ret->nb_outputs = 0;
609     av_freep(&ret->priv);
610     av_free(ret);
611     return NULL;
612 }
613
614 #if FF_API_AVFILTER_OPEN
615 int avfilter_open(AVFilterContext **filter_ctx, AVFilter *filter, const char *inst_name)
616 {
617     *filter_ctx = ff_filter_alloc(filter, inst_name);
618     return *filter_ctx ? 0 : AVERROR(ENOMEM);
619 }
620 #endif
621
622 void avfilter_free(AVFilterContext *filter)
623 {
624     int i;
625     AVFilterLink *link;
626
627     if (!filter)
628         return;
629
630     if (filter->graph)
631         ff_filter_graph_remove_filter(filter->graph, filter);
632
633     if (filter->filter->uninit)
634         filter->filter->uninit(filter);
635
636     for (i = 0; i < filter->nb_inputs; i++) {
637         if ((link = filter->inputs[i])) {
638             if (link->src)
639                 link->src->outputs[link->srcpad - link->src->output_pads] = NULL;
640             ff_formats_unref(&link->in_formats);
641             ff_formats_unref(&link->out_formats);
642             ff_formats_unref(&link->in_samplerates);
643             ff_formats_unref(&link->out_samplerates);
644             ff_channel_layouts_unref(&link->in_channel_layouts);
645             ff_channel_layouts_unref(&link->out_channel_layouts);
646         }
647         avfilter_link_free(&link);
648     }
649     for (i = 0; i < filter->nb_outputs; i++) {
650         if ((link = filter->outputs[i])) {
651             if (link->dst)
652                 link->dst->inputs[link->dstpad - link->dst->input_pads] = NULL;
653             ff_formats_unref(&link->in_formats);
654             ff_formats_unref(&link->out_formats);
655             ff_formats_unref(&link->in_samplerates);
656             ff_formats_unref(&link->out_samplerates);
657             ff_channel_layouts_unref(&link->in_channel_layouts);
658             ff_channel_layouts_unref(&link->out_channel_layouts);
659         }
660         avfilter_link_free(&link);
661     }
662
663     if (filter->filter->priv_class)
664         av_opt_free(filter->priv);
665
666     av_freep(&filter->name);
667     av_freep(&filter->input_pads);
668     av_freep(&filter->output_pads);
669     av_freep(&filter->inputs);
670     av_freep(&filter->outputs);
671     av_freep(&filter->priv);
672     while(filter->command_queue){
673         ff_command_queue_pop(filter);
674     }
675     av_opt_free(filter);
676     av_expr_free(filter->enable);
677     filter->enable = NULL;
678     av_freep(&filter->var_values);
679     av_free(filter);
680 }
681
682 static int process_options(AVFilterContext *ctx, AVDictionary **options,
683                            const char *args)
684 {
685     const AVOption *o = NULL;
686     int ret, count = 0;
687     char *av_uninit(parsed_key), *av_uninit(value);
688     const char *key;
689     int offset= -1;
690
691     av_opt_set_defaults(ctx);
692
693     if (!args)
694         return 0;
695
696     while (*args) {
697         const char *shorthand = NULL;
698
699         o = av_opt_next(ctx->priv, o);
700         if (o) {
701             if (o->type == AV_OPT_TYPE_CONST || o->offset == offset)
702                 continue;
703             offset = o->offset;
704             shorthand = o->name;
705         }
706
707         ret = av_opt_get_key_value(&args, "=", ":",
708                                    shorthand ? AV_OPT_FLAG_IMPLICIT_KEY : 0,
709                                    &parsed_key, &value);
710         if (ret < 0) {
711             if (ret == AVERROR(EINVAL))
712                 av_log(ctx, AV_LOG_ERROR, "No option name near '%s'\n", args);
713             else
714                 av_log(ctx, AV_LOG_ERROR, "Unable to parse '%s': %s\n", args,
715                        av_err2str(ret));
716             return ret;
717         }
718         if (*args)
719             args++;
720         if (parsed_key) {
721             key = parsed_key;
722             while ((o = av_opt_next(ctx->priv, o))); /* discard all remaining shorthand */
723         } else {
724             key = shorthand;
725         }
726
727         av_log(ctx, AV_LOG_DEBUG, "Setting '%s' to value '%s'\n", key, value);
728
729         if (av_opt_find(ctx, key, NULL, 0, 0)) {
730             ret = av_opt_set(ctx, key, value, 0);
731             if (ret < 0)
732                 return ret;
733         } else {
734         av_dict_set(options, key, value, 0);
735         if ((ret = av_opt_set(ctx->priv, key, value, 0)) < 0) {
736             if (!av_opt_find(ctx->priv, key, NULL, 0, AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ)) {
737             if (ret == AVERROR_OPTION_NOT_FOUND)
738                 av_log(ctx, AV_LOG_ERROR, "Option '%s' not found\n", key);
739             av_free(value);
740             av_free(parsed_key);
741             return ret;
742             }
743         }
744         }
745
746         av_free(value);
747         av_free(parsed_key);
748         count++;
749     }
750
751     if (ctx->enable_str) {
752         ret = set_enable_expr(ctx, ctx->enable_str);
753         if (ret < 0)
754             return ret;
755     }
756     return count;
757 }
758
759 #if FF_API_AVFILTER_INIT_FILTER
760 int avfilter_init_filter(AVFilterContext *filter, const char *args, void *opaque)
761 {
762     return avfilter_init_str(filter, args);
763 }
764 #endif
765
766 int avfilter_init_dict(AVFilterContext *ctx, AVDictionary **options)
767 {
768     int ret = 0;
769
770     if (ctx->filter->priv_class) {
771         ret = av_opt_set_dict(ctx->priv, options);
772         if (ret < 0) {
773             av_log(ctx, AV_LOG_ERROR, "Error applying options to the filter.\n");
774             return ret;
775         }
776     }
777
778     if (ctx->filter->init_opaque)
779         ret = ctx->filter->init_opaque(ctx, NULL);
780     else if (ctx->filter->init)
781         ret = ctx->filter->init(ctx);
782     else if (ctx->filter->init_dict)
783         ret = ctx->filter->init_dict(ctx, options);
784
785     return ret;
786 }
787
788 int avfilter_init_str(AVFilterContext *filter, const char *args)
789 {
790     AVDictionary *options = NULL;
791     AVDictionaryEntry *e;
792     int ret=0;
793
794     if (args && *args) {
795         if (!filter->filter->priv_class) {
796             av_log(filter, AV_LOG_ERROR, "This filter does not take any "
797                    "options, but options were provided: %s.\n", args);
798             return AVERROR(EINVAL);
799         }
800
801 #if FF_API_OLD_FILTER_OPTS
802             if (   !strcmp(filter->filter->name, "format")     ||
803                    !strcmp(filter->filter->name, "noformat")   ||
804                    !strcmp(filter->filter->name, "frei0r")     ||
805                    !strcmp(filter->filter->name, "frei0r_src") ||
806                    !strcmp(filter->filter->name, "ocv")        ||
807                    !strcmp(filter->filter->name, "pan")        ||
808                    !strcmp(filter->filter->name, "pp")         ||
809                    !strcmp(filter->filter->name, "aevalsrc")) {
810             /* a hack for compatibility with the old syntax
811              * replace colons with |s */
812             char *copy = av_strdup(args);
813             char *p    = copy;
814             int nb_leading = 0; // number of leading colons to skip
815             int deprecated = 0;
816
817             if (!copy) {
818                 ret = AVERROR(ENOMEM);
819                 goto fail;
820             }
821
822             if (!strcmp(filter->filter->name, "frei0r") ||
823                 !strcmp(filter->filter->name, "ocv"))
824                 nb_leading = 1;
825             else if (!strcmp(filter->filter->name, "frei0r_src"))
826                 nb_leading = 3;
827
828             while (nb_leading--) {
829                 p = strchr(p, ':');
830                 if (!p) {
831                     p = copy + strlen(copy);
832                     break;
833                 }
834                 p++;
835             }
836
837             deprecated = strchr(p, ':') != NULL;
838
839             if (!strcmp(filter->filter->name, "aevalsrc")) {
840                 deprecated = 0;
841                 while ((p = strchr(p, ':')) && p[1] != ':') {
842                     const char *epos = strchr(p + 1, '=');
843                     const char *spos = strchr(p + 1, ':');
844                     const int next_token_is_opt = epos && (!spos || epos < spos);
845                     if (next_token_is_opt) {
846                         p++;
847                         break;
848                     }
849                     /* next token does not contain a '=', assume a channel expression */
850                     deprecated = 1;
851                     *p++ = '|';
852                 }
853                 if (p && *p == ':') { // double sep '::' found
854                     deprecated = 1;
855                     memmove(p, p + 1, strlen(p));
856                 }
857             } else
858             while ((p = strchr(p, ':')))
859                 *p++ = '|';
860
861             if (deprecated)
862                 av_log(filter, AV_LOG_WARNING, "This syntax is deprecated. Use "
863                        "'|' to separate the list items.\n");
864
865             av_log(filter, AV_LOG_DEBUG, "compat: called with args=[%s]\n", copy);
866             ret = process_options(filter, &options, copy);
867             av_freep(&copy);
868
869             if (ret < 0)
870                 goto fail;
871 #endif
872         } else {
873 #if CONFIG_MP_FILTER
874             if (!strcmp(filter->filter->name, "mp")) {
875                 char *escaped;
876
877                 if (!strncmp(args, "filter=", 7))
878                     args += 7;
879                 ret = av_escape(&escaped, args, ":=", AV_ESCAPE_MODE_BACKSLASH, 0);
880                 if (ret < 0) {
881                     av_log(filter, AV_LOG_ERROR, "Unable to escape MPlayer filters arg '%s'\n", args);
882                     goto fail;
883                 }
884                 ret = process_options(filter, &options, escaped);
885                 av_free(escaped);
886             } else
887 #endif
888             ret = process_options(filter, &options, args);
889             if (ret < 0)
890                 goto fail;
891         }
892     }
893
894     ret = avfilter_init_dict(filter, &options);
895     if (ret < 0)
896         goto fail;
897
898     if ((e = av_dict_get(options, "", NULL, AV_DICT_IGNORE_SUFFIX))) {
899         av_log(filter, AV_LOG_ERROR, "No such option: %s.\n", e->key);
900         ret = AVERROR_OPTION_NOT_FOUND;
901         goto fail;
902     }
903
904 fail:
905     av_dict_free(&options);
906
907     return ret;
908 }
909
910 const char *avfilter_pad_get_name(const AVFilterPad *pads, int pad_idx)
911 {
912     return pads[pad_idx].name;
913 }
914
915 enum AVMediaType avfilter_pad_get_type(const AVFilterPad *pads, int pad_idx)
916 {
917     return pads[pad_idx].type;
918 }
919
920 static int default_filter_frame(AVFilterLink *link, AVFrame *frame)
921 {
922     return ff_filter_frame(link->dst->outputs[0], frame);
923 }
924
925 static int ff_filter_frame_framed(AVFilterLink *link, AVFrame *frame)
926 {
927     int (*filter_frame)(AVFilterLink *, AVFrame *);
928     AVFilterContext *dstctx = link->dst;
929     AVFilterPad *dst = link->dstpad;
930     AVFrame *out;
931     int ret;
932     AVFilterCommand *cmd= link->dst->command_queue;
933     int64_t pts;
934
935     if (link->closed) {
936         av_frame_free(&frame);
937         return AVERROR_EOF;
938     }
939
940     if (!(filter_frame = dst->filter_frame))
941         filter_frame = default_filter_frame;
942
943     /* copy the frame if needed */
944     if (dst->needs_writable && !av_frame_is_writable(frame)) {
945         av_log(link->dst, AV_LOG_DEBUG, "Copying data in avfilter.\n");
946
947         /* Maybe use ff_copy_buffer_ref instead? */
948         switch (link->type) {
949         case AVMEDIA_TYPE_VIDEO:
950             out = ff_get_video_buffer(link, link->w, link->h);
951             break;
952         case AVMEDIA_TYPE_AUDIO:
953             out = ff_get_audio_buffer(link, frame->nb_samples);
954             break;
955         default: return AVERROR(EINVAL);
956         }
957         if (!out) {
958             av_frame_free(&frame);
959             return AVERROR(ENOMEM);
960         }
961         av_frame_copy_props(out, frame);
962
963         switch (link->type) {
964         case AVMEDIA_TYPE_VIDEO:
965             av_image_copy(out->data, out->linesize, (const uint8_t **)frame->data, frame->linesize,
966                           frame->format, frame->width, frame->height);
967             break;
968         case AVMEDIA_TYPE_AUDIO:
969             av_samples_copy(out->extended_data, frame->extended_data,
970                             0, 0, frame->nb_samples,
971                             av_get_channel_layout_nb_channels(frame->channel_layout),
972                             frame->format);
973             break;
974         default: return AVERROR(EINVAL);
975         }
976
977         av_frame_free(&frame);
978     } else
979         out = frame;
980
981     while(cmd && cmd->time <= out->pts * av_q2d(link->time_base)){
982         av_log(link->dst, AV_LOG_DEBUG,
983                "Processing command time:%f command:%s arg:%s\n",
984                cmd->time, cmd->command, cmd->arg);
985         avfilter_process_command(link->dst, cmd->command, cmd->arg, 0, 0, cmd->flags);
986         ff_command_queue_pop(link->dst);
987         cmd= link->dst->command_queue;
988     }
989
990     pts = out->pts;
991     if (dstctx->enable_str) {
992         int64_t pos = av_frame_get_pkt_pos(out);
993         dstctx->var_values[VAR_N] = link->frame_count;
994         dstctx->var_values[VAR_T] = pts == AV_NOPTS_VALUE ? NAN : pts * av_q2d(link->time_base);
995         dstctx->var_values[VAR_POS] = pos == -1 ? NAN : pos;
996         if (!av_expr_eval(dstctx->enable, dstctx->var_values, NULL))
997             filter_frame = dst->passthrough_filter_frame ? dst->passthrough_filter_frame
998                                                          : default_filter_frame;
999     }
1000     ret = filter_frame(link, out);
1001     link->frame_count++;
1002     link->frame_requested = 0;
1003     ff_update_link_current_pts(link, pts);
1004     return ret;
1005 }
1006
1007 static int ff_filter_frame_needs_framing(AVFilterLink *link, AVFrame *frame)
1008 {
1009     int insamples = frame->nb_samples, inpos = 0, nb_samples;
1010     AVFrame *pbuf = link->partial_buf;
1011     int nb_channels = av_frame_get_channels(frame);
1012     int ret = 0;
1013
1014     link->flags |= FF_LINK_FLAG_REQUEST_LOOP;
1015     /* Handle framing (min_samples, max_samples) */
1016     while (insamples) {
1017         if (!pbuf) {
1018             AVRational samples_tb = { 1, link->sample_rate };
1019             pbuf = ff_get_audio_buffer(link, link->partial_buf_size);
1020             if (!pbuf) {
1021                 av_log(link->dst, AV_LOG_WARNING,
1022                        "Samples dropped due to memory allocation failure.\n");
1023                 return 0;
1024             }
1025             av_frame_copy_props(pbuf, frame);
1026             pbuf->pts = frame->pts +
1027                         av_rescale_q(inpos, samples_tb, link->time_base);
1028             pbuf->nb_samples = 0;
1029         }
1030         nb_samples = FFMIN(insamples,
1031                            link->partial_buf_size - pbuf->nb_samples);
1032         av_samples_copy(pbuf->extended_data, frame->extended_data,
1033                         pbuf->nb_samples, inpos,
1034                         nb_samples, nb_channels, link->format);
1035         inpos                   += nb_samples;
1036         insamples               -= nb_samples;
1037         pbuf->nb_samples += nb_samples;
1038         if (pbuf->nb_samples >= link->min_samples) {
1039             ret = ff_filter_frame_framed(link, pbuf);
1040             pbuf = NULL;
1041         }
1042     }
1043     av_frame_free(&frame);
1044     link->partial_buf = pbuf;
1045     return ret;
1046 }
1047
1048 int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
1049 {
1050     FF_TPRINTF_START(NULL, filter_frame); ff_tlog_link(NULL, link, 1); ff_tlog(NULL, " "); ff_tlog_ref(NULL, frame, 1);
1051
1052     /* Consistency checks */
1053     if (link->type == AVMEDIA_TYPE_VIDEO) {
1054         if (strcmp(link->dst->filter->name, "scale")) {
1055             av_assert1(frame->format                 == link->format);
1056             av_assert1(frame->width               == link->w);
1057             av_assert1(frame->height               == link->h);
1058         }
1059     } else {
1060         av_assert1(frame->format                == link->format);
1061         av_assert1(av_frame_get_channels(frame) == link->channels);
1062         av_assert1(frame->channel_layout        == link->channel_layout);
1063         av_assert1(frame->sample_rate           == link->sample_rate);
1064     }
1065
1066     /* Go directly to actual filtering if possible */
1067     if (link->type == AVMEDIA_TYPE_AUDIO &&
1068         link->min_samples &&
1069         (link->partial_buf ||
1070          frame->nb_samples < link->min_samples ||
1071          frame->nb_samples > link->max_samples)) {
1072         return ff_filter_frame_needs_framing(link, frame);
1073     } else {
1074         return ff_filter_frame_framed(link, frame);
1075     }
1076 }
1077
1078 const AVClass *avfilter_get_class(void)
1079 {
1080     return &avfilter_class;
1081 }