|
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