Skip to content

Commit 73bdc85

Browse files
committed
Use c++ strings
1 parent 2a19014 commit 73bdc85

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

src/torchcodec/_core/FFMPEGCommon.cpp

+12-12
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
// LICENSE file in the root directory of this source tree.
66

77
#include "src/torchcodec/_core/FFMPEGCommon.h"
8-
#include <cstdlib>
98

109
#include <c10/util/Exception.h>
1110

@@ -161,25 +160,26 @@ SwrContext* allocateSwrContext(
161160

162161
void setFFmpegLogLevel() {
163162
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") {
167167
logLevel = AV_LOG_QUIET;
168-
} else if (std::strcmp(logLevelEnv, "PANIC") == 0) {
168+
} else if (logLevelEnv == "PANIC") {
169169
logLevel = AV_LOG_PANIC;
170-
} else if (std::strcmp(logLevelEnv, "FATAL") == 0) {
170+
} else if (logLevelEnv == "FATAL") {
171171
logLevel = AV_LOG_FATAL;
172-
} else if (std::strcmp(logLevelEnv, "ERROR") == 0) {
172+
} else if (logLevelEnv == "ERROR") {
173173
logLevel = AV_LOG_ERROR;
174-
} else if (std::strcmp(logLevelEnv, "WARNING") == 0) {
174+
} else if (logLevelEnv == "WARNING") {
175175
logLevel = AV_LOG_WARNING;
176-
} else if (std::strcmp(logLevelEnv, "INFO") == 0) {
176+
} else if (logLevelEnv == "INFO") {
177177
logLevel = AV_LOG_INFO;
178-
} else if (std::strcmp(logLevelEnv, "VERBOSE") == 0) {
178+
} else if (logLevelEnv == "VERBOSE") {
179179
logLevel = AV_LOG_VERBOSE;
180-
} else if (std::strcmp(logLevelEnv, "DEBUG") == 0) {
180+
} else if (logLevelEnv == "DEBUG") {
181181
logLevel = AV_LOG_DEBUG;
182-
} else if (std::strcmp(logLevelEnv, "TRACE") == 0) {
182+
} else if (logLevelEnv == "TRACE") {
183183
logLevel = AV_LOG_TRACE;
184184
} else {
185185
TORCH_CHECK(

0 commit comments

Comments
 (0)