handle deprecated filter ctx function

Author: Daniel Flores

src/torchcodec/_core/FFMPEGCommon.cpp

@@ -8,6 +8,11 @@
 
 #include <c10/util/Exception.h>
 
+extern "C" {
+#include <libavfilter/avfilter.h>
+#include <libavfilter/buffersink.h>
+}
+
 namespace facebook::torchcodec {
 
 AutoAVPacket::AutoAVPacket() : avPacket_(av_packet_alloc()) {
@@ -374,6 +379,63 @@ SwrContext* createSwrContext(
   return swrContext;
 }
 
+AVFilterContext* createBuffersinkFilter(
+    AVFilterGraph* filterGraph,
+    const char* name,
+    enum AVPixelFormat outputFormat) {
+  const AVFilter* buffersink = avfilter_get_by_name("buffersink");
+  if (!buffersink) {
+    return nullptr;
+  }
+
+  AVFilterContext* sinkContext = nullptr;
+  int status;
+
+// av_opt_set_int_list was replaced by av_opt_set() in FFmpeg 8.
+#if LIBAVUTIL_VERSION_MAJOR >= 60 // FFmpeg >= 8
+  // Output options like pixel_formats must be set before filter init
+  sinkContext = avfilter_graph_alloc_filter(filterGraph, buffersink, name);
+  if (!sinkContext) {
+    return nullptr;
+  }
+  // av_opt_set uses the string representation of a pixel format
+  const char* fmt_name = av_get_pix_fmt_name(outputFormat);
+  if (!fmt_name) {
+    return nullptr;
+  }
+  status = av_opt_set(
+      sinkContext, "pixel_formats", fmt_name, AV_OPT_SEARCH_CHILDREN);
+  if (status < 0) {
+    return nullptr;
+  }
+  status = avfilter_init_str(sinkContext, nullptr);
+  if (status < 0) {
+    return nullptr;
+  }
+#else // FFmpeg <= 7
+  // For older FFmpeg versions, create filter and then set options
+  status = avfilter_graph_create_filter(
+      &sinkContext, buffersink, name, nullptr, nullptr, filterGraph);
+  if (status < 0) {
+    return nullptr;
+  }
+
+  enum AVPixelFormat pix_fmts[] = {outputFormat, AV_PIX_FMT_NONE};
+
+  status = av_opt_set_int_list(
+      sinkContext,
+      "pix_fmts",
+      pix_fmts,
+      AV_PIX_FMT_NONE,
+      AV_OPT_SEARCH_CHILDREN);
+  if (status < 0) {
+    return nullptr;
+  }
+#endif
+
+  return sinkContext;
+}
+
 UniqueAVFrame convertAudioAVFrameSamples(
     const UniqueSwrContext& swrContext,
     const UniqueAVFrame& srcAVFrame,

src/torchcodec/_core/FFMPEGCommon.h

@@ -23,6 +23,7 @@ extern "C" {
 #include <libavutil/display.h>
 #include <libavutil/file.h>
 #include <libavutil/opt.h>
+#include <libavutil/pixdesc.h>
 #include <libavutil/pixfmt.h>
 #include <libavutil/version.h>
 #include <libswresample/swresample.h>
@@ -237,4 +238,9 @@ int64_t computeSafeDuration(
     const AVRational& frameRate,
     const AVRational& timeBase);
 
+AVFilterContext* createBuffersinkFilter(
+    AVFilterGraph* filterGraph,
+    const char* name,
+    enum AVPixelFormat outputFormat);
+
 } // namespace facebook::torchcodec

src/torchcodec/_core/FilterGraph.cpp

@@ -5,6 +5,7 @@
 // LICENSE file in the root directory of this source tree.
 
 #include "src/torchcodec/_core/FilterGraph.h"
+#include "src/torchcodec/_core/FFMPEGCommon.h"
 
 extern "C" {
 #include <libavfilter/buffersink.h>
@@ -63,7 +64,6 @@ FilterGraph::FilterGraph(
   }
 
   const AVFilter* buffersrc = avfilter_get_by_name("buffer");
-  const AVFilter* buffersink = avfilter_get_by_name("buffersink");
 
   UniqueAVBufferSrcParameters srcParams(av_buffersrc_parameters_alloc());
   TORCH_CHECK(srcParams, "Failed to allocate buffersrc params");
@@ -93,26 +93,10 @@ FilterGraph::FilterGraph(
       "Failed to create filter graph : ",
       getFFMPEGErrorStringFromErrorCode(status));
 
-  status = avfilter_graph_create_filter(
-      &sinkContext_, buffersink, "out", nullptr, nullptr, filterGraph_.get());
+  sinkContext_ = createBuffersinkFilter(
+      filterGraph_.get(), "out", filtersContext.outputFormat);
   TORCH_CHECK(
-      status >= 0,
-      "Failed to create filter graph: ",
-      getFFMPEGErrorStringFromErrorCode(status));
-
-  enum AVPixelFormat pix_fmts[] = {
-      filtersContext.outputFormat, AV_PIX_FMT_NONE};
-
-  status = av_opt_set_int_list(
-      sinkContext_,
-      "pix_fmts",
-      pix_fmts,
-      AV_PIX_FMT_NONE,
-      AV_OPT_SEARCH_CHILDREN);
-  TORCH_CHECK(
-      status >= 0,
-      "Failed to set output pixel formats: ",
-      getFFMPEGErrorStringFromErrorCode(status));
+      sinkContext_ != nullptr, "Failed to create and configure buffersink");
 
   UniqueAVFilterInOut outputs(avfilter_inout_alloc());
   UniqueAVFilterInOut inputs(avfilter_inout_alloc());