diff mbox series

[FFmpeg-devel,29/31] fftools/ffmpeg_sched: allow filtergraphs to send to filtergraphs

Message ID 20240405161212.26167-29-anton@khirnov.net
State Accepted
Commit 255ae036012177e1c0419e28bc4b9ab10d848d0a
Headers show
Series [FFmpeg-devel,01/31] lavfi/vf_scale: fix AVOption flags for "size"/"s" | expand

Checks

Context Check Description
andriy/make_x86 success Make finished
andriy/make_fate_x86 success Make fate finished

Commit Message

Anton Khirnov April 5, 2024, 4:12 p.m. UTC
Will be useful for filtergraph chaining that will be added in following
commits.
---
 fftools/ffmpeg_sched.c | 94 ++++++++++++++++++++++++++----------------
 fftools/ffmpeg_sched.h |  6 ++-
 2 files changed, 63 insertions(+), 37 deletions(-)
diff mbox series

Patch

diff --git a/fftools/ffmpeg_sched.c b/fftools/ffmpeg_sched.c
index f8485db30b..e58b00ea97 100644
--- a/fftools/ffmpeg_sched.c
+++ b/fftools/ffmpeg_sched.c
@@ -983,20 +983,40 @@  int sch_connect(Scheduler *sch, SchedulerNode src, SchedulerNode dst)
         }
     case SCH_NODE_TYPE_FILTER_OUT: {
         SchFilterOut *fo;
-        SchEnc      *enc;
 
         av_assert0(src.idx < sch->nb_filters &&
                    src.idx_stream < sch->filters[src.idx].nb_outputs);
-        // filtered frames go to encoding
-        av_assert0(dst.type == SCH_NODE_TYPE_ENC &&
-                   dst.idx < sch->nb_enc);
+        fo = &sch->filters[src.idx].outputs[src.idx_stream];
 
-        fo  = &sch->filters[src.idx].outputs[src.idx_stream];
-        enc = &sch->enc[dst.idx];
+        av_assert0(!fo->dst.type);
+        fo->dst = dst;
+
+        // filtered frames go to encoding or another filtergraph
+        switch (dst.type) {
+        case SCH_NODE_TYPE_ENC: {
+            SchEnc *enc;
+
+            av_assert0(dst.idx < sch->nb_enc);
+            enc = &sch->enc[dst.idx];
+
+            av_assert0(!enc->src.type);
+            enc->src = src;
+            break;
+            }
+        case SCH_NODE_TYPE_FILTER_IN: {
+            SchFilterIn *fi;
+
+            av_assert0(dst.idx < sch->nb_filters &&
+                       dst.idx_stream < sch->filters[dst.idx].nb_inputs);
+            fi = &sch->filters[dst.idx].inputs[dst.idx_stream];
+
+            av_assert0(!fi->src.type);
+            fi->src = src;
+            break;
+            }
+        default: av_assert0(0);
+        }
 
-        av_assert0(!fo->dst.type && !enc->src.type);
-        fo->dst  = dst;
-        enc->src = src;
 
         break;
         }
@@ -1351,24 +1371,13 @@  static int check_acyclic(Scheduler *sch)
         goto fail;
     }
 
-    // trace the transcoding graph upstream from every output stream
-    // fed by a filtergraph
-    for (unsigned i = 0; i < sch->nb_mux; i++) {
-        SchMux *mux = &sch->mux[i];
-
-        for (unsigned j = 0; j < mux->nb_streams; j++) {
-            SchMuxStream  *ms = &mux->streams[j];
-            SchedulerNode src = ms->src_sched;
-
-            if (src.type != SCH_NODE_TYPE_FILTER_OUT)
-                continue;
-            src.idx_stream = 0;
-
-            ret = check_acyclic_for_output(sch, src, filters_visited, filters_stack);
-            if (ret < 0) {
-                av_log(mux, AV_LOG_ERROR, "Transcoding graph has a cycle\n");
-                goto fail;
-            }
+    // trace the transcoding graph upstream from every filtegraph
+    for (unsigned i = 0; i < sch->nb_filters; i++) {
+        ret = check_acyclic_for_output(sch, (SchedulerNode){ .idx = i },
+                                       filters_visited, filters_stack);
+        if (ret < 0) {
+            av_log(&sch->filters[i], AV_LOG_ERROR, "Transcoding graph has a cycle\n");
+            goto fail;
         }
     }
 
@@ -1484,13 +1493,18 @@  static int start_prepare(Scheduler *sch)
                        "Filtergraph input %u not connected to a source\n", j);
                 return AVERROR(EINVAL);
             }
-            av_assert0(fi->src.type == SCH_NODE_TYPE_DEC);
-            dec = &sch->dec[fi->src.idx];
 
-            switch (dec->src.type) {
-            case SCH_NODE_TYPE_DEMUX: fi->src_sched = dec->src;                   break;
-            case SCH_NODE_TYPE_ENC:   fi->src_sched = sch->enc[dec->src.idx].src; break;
-            default: av_assert0(0);
+            if (fi->src.type == SCH_NODE_TYPE_FILTER_OUT)
+                fi->src_sched = fi->src;
+            else {
+                av_assert0(fi->src.type == SCH_NODE_TYPE_DEC);
+                dec = &sch->dec[fi->src.idx];
+
+                switch (dec->src.type) {
+                case SCH_NODE_TYPE_DEMUX: fi->src_sched = dec->src;                   break;
+                case SCH_NODE_TYPE_ENC:   fi->src_sched = sch->enc[dec->src.idx].src; break;
+                default: av_assert0(0);
+                }
             }
         }
 
@@ -2379,12 +2393,17 @@  void sch_filter_receive_finish(Scheduler *sch, unsigned fg_idx, unsigned in_idx)
 int sch_filter_send(Scheduler *sch, unsigned fg_idx, unsigned out_idx, AVFrame *frame)
 {
     SchFilterGraph *fg;
+    SchedulerNode  dst;
 
     av_assert0(fg_idx < sch->nb_filters);
     fg = &sch->filters[fg_idx];
 
     av_assert0(out_idx < fg->nb_outputs);
-    return send_to_enc(sch, &sch->enc[fg->outputs[out_idx].dst.idx], frame);
+    dst = fg->outputs[out_idx].dst;
+
+    return (dst.type == SCH_NODE_TYPE_ENC)                                    ?
+           send_to_enc   (sch, &sch->enc[dst.idx],                     frame) :
+           send_to_filter(sch, &sch->filters[dst.idx], dst.idx_stream, frame);
 }
 
 static int filter_done(Scheduler *sch, unsigned fg_idx)
@@ -2396,8 +2415,11 @@  static int filter_done(Scheduler *sch, unsigned fg_idx)
         tq_receive_finish(fg->queue, i);
 
     for (unsigned i = 0; i < fg->nb_outputs; i++) {
-        SchEnc *enc = &sch->enc[fg->outputs[i].dst.idx];
-        int err = send_to_enc(sch, enc, NULL);
+        SchedulerNode dst = fg->outputs[i].dst;
+        int err = (dst.type == SCH_NODE_TYPE_ENC)                                   ?
+                  send_to_enc   (sch, &sch->enc[dst.idx],                     NULL) :
+                  send_to_filter(sch, &sch->filters[dst.idx], dst.idx_stream, NULL);
+
         if (err < 0 && err != AVERROR_EOF)
             ret = err_merge(ret, err);
     }
diff --git a/fftools/ffmpeg_sched.h b/fftools/ffmpeg_sched.h
index e51c26cec9..7cd839016c 100644
--- a/fftools/ffmpeg_sched.h
+++ b/fftools/ffmpeg_sched.h
@@ -41,7 +41,8 @@ 
  * - filtergraphs, each containing zero or more inputs (0 in case the
  *   filtergraph contains a lavfi source filter), and one or more outputs; the
  *   inputs and outputs need not have matching media types;
- *   each filtergraph input receives decoded frames from some decoder;
+ *   each filtergraph input receives decoded frames from some decoder or another
+ *   filtergraph output;
  *   filtered frames from each output are sent to some encoder;
  * - encoders, which receive decoded frames from some decoder (subtitles) or
  *   some filtergraph output (audio/video), encode them, and send encoded
@@ -51,6 +52,9 @@ 
  *   encoder (transcoding); those packets are interleaved and written out by the
  *   muxer.
  *
+ * The structure formed by the above components is a directed acyclic graph
+ * (absence of cycles is checked at startup).
+ *
  * There must be at least one muxer instance, otherwise the transcode produces
  * no output and is meaningless. Otherwise, in a generic transcoding scenario
  * there may be arbitrary number of instances of any of the above components,