git.sesse.net Git - ffmpeg/blob - libavfilter/dualinput.c

   1 /*
   2  * This file is part of FFmpeg.
   3  *
   4  * FFmpeg is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * FFmpeg is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with FFmpeg; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17  */
  18
  19 #include "dualinput.h"
  20 #include "libavutil/timestamp.h"
  21
  22 static int process_frame(FFFrameSync *fs)
  23 {
  24     AVFilterContext *ctx = fs->parent;
  25     FFDualInputContext *s = fs->opaque;
  26     AVFrame *mainpic = NULL, *secondpic = NULL;
  27     int ret = 0;
  28
  29     if ((ret = ff_framesync_get_frame(&s->fs, 0, &mainpic,   1)) < 0 ||
  30         (ret = ff_framesync_get_frame(&s->fs, 1, &secondpic, 0)) < 0) {
  31         av_frame_free(&mainpic);
  32         return ret;
  33     }
  34     av_assert0(mainpic);
  35     mainpic->pts = av_rescale_q(mainpic->pts, s->fs.time_base, ctx->outputs[0]->time_base);
  36     if (secondpic && !ctx->is_disabled)
  37         mainpic = s->process(ctx, mainpic, secondpic);
  38     ret = ff_filter_frame(ctx->outputs[0], mainpic);
  39     av_assert1(ret != AVERROR(EAGAIN));
  40     s->frame_requested = 0;
  41     return ret;
  42 }
  43
  44 int ff_dualinput_init(AVFilterContext *ctx, FFDualInputContext *s)
  45 {
  46     FFFrameSyncIn *in = s->fs.in;
  47
  48     ff_framesync_init(&s->fs, ctx, 2);
  49     s->fs.opaque = s;
  50     s->fs.on_event = process_frame;
  51     in[0].time_base = ctx->inputs[0]->time_base;
  52     in[1].time_base = ctx->inputs[1]->time_base;
  53     in[0].sync   = 2;
  54     in[0].before = EXT_STOP;
  55     in[0].after  = EXT_INFINITY;
  56     in[1].sync   = 1;
  57     in[1].before = EXT_NULL;
  58     in[1].after  = EXT_INFINITY;
  59
  60     if (s->shortest)
  61         in[1].after = EXT_STOP;
  62     if (!s->repeatlast) {
  63         in[0].after = EXT_STOP;
  64         in[1].sync  = 0;
  65     }
  66
  67     return ff_framesync_configure(&s->fs);
  68 }
  69
  70 int ff_dualinput_filter_frame_main(FFDualInputContext *s,
  71                                    AVFilterLink *inlink, AVFrame *in)
  72 {
  73     return ff_framesync_filter_frame(&s->fs, inlink, in);
  74 }
  75
  76 int ff_dualinput_filter_frame_second(FFDualInputContext *s,
  77                                      AVFilterLink *inlink, AVFrame *in)
  78 {
  79     return ff_framesync_filter_frame(&s->fs, inlink, in);
  80 }
  81
  82 int ff_dualinput_request_frame(FFDualInputContext *s, AVFilterLink *outlink)
  83 {
  84     return ff_framesync_request_frame(&s->fs, outlink);
  85 }
  86
  87 void ff_dualinput_uninit(FFDualInputContext *s)
  88 {
  89     ff_framesync_uninit(&s->fs);
  90 }