Disable FFmpeg logs for encoder

Author: NicolasHug

src/torchcodec/_core/Encoder.cpp

@@ -5,8 +5,6 @@ namespace facebook::torchcodec {
 
 AudioEncoder::~AudioEncoder() {}
 
-// TODO-ENCODING: disable ffmpeg logs by default
-
 AudioEncoder::AudioEncoder(
     const torch::Tensor wf,
     int sampleRate,
@@ -18,6 +16,8 @@ AudioEncoder::AudioEncoder(
       wf_.dtype());
   TORCH_CHECK(
       wf_.dim() == 2, "waveform must have 2 dimensions, got ", wf_.dim());
+
+  setFFmpegLogLevel();
   AVFormatContext* avFormatContext = nullptr;
   auto status = avformat_alloc_output_context2(
       &avFormatContext, nullptr, nullptr, fileName.data());

src/torchcodec/_core/FFMPEGCommon.cpp

@@ -5,6 +5,7 @@
 // LICENSE file in the root directory of this source tree.
 
 #include "src/torchcodec/_core/FFMPEGCommon.h"
+#include <cstdlib>
 
 #include <c10/util/Exception.h>
 
@@ -158,4 +159,37 @@ SwrContext* allocateSwrContext(
   return swrContext;
 }
 
+void setFFmpegLogLevel() {
+  auto logLevel = AV_LOG_QUIET;
+  const char* logLevelEnv = std::getenv("TORCHCODEC_FFMPEG_LOG_LEVEL");
+  if (logLevelEnv != nullptr) {
+    if (std::strcmp(logLevelEnv, "QUIET") == 0) {
+      logLevel = AV_LOG_QUIET;
+    } else if (std::strcmp(logLevelEnv, "PANIC") == 0) {
+      logLevel = AV_LOG_PANIC;
+    } else if (std::strcmp(logLevelEnv, "FATAL") == 0) {
+      logLevel = AV_LOG_FATAL;
+    } else if (std::strcmp(logLevelEnv, "ERROR") == 0) {
+      logLevel = AV_LOG_ERROR;
+    } else if (std::strcmp(logLevelEnv, "WARNING") == 0) {
+      logLevel = AV_LOG_WARNING;
+    } else if (std::strcmp(logLevelEnv, "INFO") == 0) {
+      logLevel = AV_LOG_INFO;
+    } else if (std::strcmp(logLevelEnv, "VERBOSE") == 0) {
+      logLevel = AV_LOG_VERBOSE;
+    } else if (std::strcmp(logLevelEnv, "DEBUG") == 0) {
+      logLevel = AV_LOG_DEBUG;
+    } else if (std::strcmp(logLevelEnv, "TRACE") == 0) {
+      logLevel = AV_LOG_TRACE;
+    } else {
+      TORCH_CHECK(
+          false,
+          "Invalid TORCHCODEC_FFMPEG_LOG_LEVEL: ",
+          logLevelEnv,
+          ". Use e.g. 'QUIET', 'PANIC', 'VERBOSE', etc.");
+    }
+  }
+  av_log_set_level(logLevel);
+}
+
 } // namespace facebook::torchcodec

src/torchcodec/_core/FFMPEGCommon.h

@@ -168,4 +168,6 @@ SwrContext* allocateSwrContext(
 // Returns true if sws_scale can handle unaligned data.
 bool canSwsScaleHandleUnalignedData();
 
+void setFFmpegLogLevel();
+
 } // namespace facebook::torchcodec

src/torchcodec/_core/SingleStreamDecoder.cpp

@@ -7,7 +7,6 @@
 #include "src/torchcodec/_core/SingleStreamDecoder.h"
 #include <cstdint>
 #include <cstdio>
-#include <cstdlib>
 #include <iostream>
 #include <limits>
 #include <sstream>
@@ -185,39 +184,6 @@ void SingleStreamDecoder::initializeDecoder() {
   initialized_ = true;
 }
 
-void SingleStreamDecoder::setFFmpegLogLevel() {
-  auto logLevel = AV_LOG_QUIET;
-  const char* logLevelEnv = std::getenv("TORCHCODEC_FFMPEG_LOG_LEVEL");
-  if (logLevelEnv != nullptr) {
-    if (std::strcmp(logLevelEnv, "QUIET") == 0) {
-      logLevel = AV_LOG_QUIET;
-    } else if (std::strcmp(logLevelEnv, "PANIC") == 0) {
-      logLevel = AV_LOG_PANIC;
-    } else if (std::strcmp(logLevelEnv, "FATAL") == 0) {
-      logLevel = AV_LOG_FATAL;
-    } else if (std::strcmp(logLevelEnv, "ERROR") == 0) {
-      logLevel = AV_LOG_ERROR;
-    } else if (std::strcmp(logLevelEnv, "WARNING") == 0) {
-      logLevel = AV_LOG_WARNING;
-    } else if (std::strcmp(logLevelEnv, "INFO") == 0) {
-      logLevel = AV_LOG_INFO;
-    } else if (std::strcmp(logLevelEnv, "VERBOSE") == 0) {
-      logLevel = AV_LOG_VERBOSE;
-    } else if (std::strcmp(logLevelEnv, "DEBUG") == 0) {
-      logLevel = AV_LOG_DEBUG;
-    } else if (std::strcmp(logLevelEnv, "TRACE") == 0) {
-      logLevel = AV_LOG_TRACE;
-    } else {
-      TORCH_CHECK(
-          false,
-          "Invalid TORCHCODEC_FFMPEG_LOG_LEVEL: ",
-          logLevelEnv,
-          ". Use e.g. 'QUIET', 'PANIC', 'VERBOSE', etc.");
-    }
-  }
-  av_log_set_level(logLevel);
-}
-
 int SingleStreamDecoder::getBestStreamIndex(AVMediaType mediaType) {
   AVCodecOnlyUseForCallingAVFindBestStream avCodec = nullptr;
   int streamIndex =

src/torchcodec/_core/SingleStreamDecoder.h

@@ -363,7 +363,6 @@ class SingleStreamDecoder {
   // --------------------------------------------------------------------------
 
   void initializeDecoder();
-  void setFFmpegLogLevel();
   // --------------------------------------------------------------------------
   // DECODING APIS AND RELATED UTILS
   // --------------------------------------------------------------------------