|
5 | 5 | // LICENSE file in the root directory of this source tree. |
6 | 6 |
|
7 | 7 | #include "src/torchcodec/_core/FFMPEGCommon.h" |
8 | | -#include <cstdlib> |
9 | 8 |
|
10 | 9 | #include <c10/util/Exception.h> |
11 | 10 |
|
@@ -161,25 +160,26 @@ SwrContext* allocateSwrContext( |
161 | 160 |
|
162 | 161 | void setFFmpegLogLevel() { |
163 | 162 | auto logLevel = AV_LOG_QUIET; |
164 | | - const char* logLevelEnv = std::getenv("TORCHCODEC_FFMPEG_LOG_LEVEL"); |
165 | | - if (logLevelEnv != nullptr) { |
166 | | - if (std::strcmp(logLevelEnv, "QUIET") == 0) { |
| 163 | + const char* logLevelEnvPtr = std::getenv("TORCHCODEC_FFMPEG_LOG_LEVEL"); |
| 164 | + if (logLevelEnvPtr != nullptr) { |
| 165 | + std::string logLevelEnv(logLevelEnvPtr); |
| 166 | + if (logLevelEnv == "QUIET") { |
167 | 167 | logLevel = AV_LOG_QUIET; |
168 | | - } else if (std::strcmp(logLevelEnv, "PANIC") == 0) { |
| 168 | + } else if (logLevelEnv == "PANIC") { |
169 | 169 | logLevel = AV_LOG_PANIC; |
170 | | - } else if (std::strcmp(logLevelEnv, "FATAL") == 0) { |
| 170 | + } else if (logLevelEnv == "FATAL") { |
171 | 171 | logLevel = AV_LOG_FATAL; |
172 | | - } else if (std::strcmp(logLevelEnv, "ERROR") == 0) { |
| 172 | + } else if (logLevelEnv == "ERROR") { |
173 | 173 | logLevel = AV_LOG_ERROR; |
174 | | - } else if (std::strcmp(logLevelEnv, "WARNING") == 0) { |
| 174 | + } else if (logLevelEnv == "WARNING") { |
175 | 175 | logLevel = AV_LOG_WARNING; |
176 | | - } else if (std::strcmp(logLevelEnv, "INFO") == 0) { |
| 176 | + } else if (logLevelEnv == "INFO") { |
177 | 177 | logLevel = AV_LOG_INFO; |
178 | | - } else if (std::strcmp(logLevelEnv, "VERBOSE") == 0) { |
| 178 | + } else if (logLevelEnv == "VERBOSE") { |
179 | 179 | logLevel = AV_LOG_VERBOSE; |
180 | | - } else if (std::strcmp(logLevelEnv, "DEBUG") == 0) { |
| 180 | + } else if (logLevelEnv == "DEBUG") { |
181 | 181 | logLevel = AV_LOG_DEBUG; |
182 | | - } else if (std::strcmp(logLevelEnv, "TRACE") == 0) { |
| 182 | + } else if (logLevelEnv == "TRACE") { |
183 | 183 | logLevel = AV_LOG_TRACE; |
184 | 184 | } else { |
185 | 185 | TORCH_CHECK( |
|
0 commit comments