diff --git a/mediapipe/tasks/ios/vision/core/sources/MPPRunningMode.h b/mediapipe/tasks/ios/vision/core/sources/MPPRunningMode.h index 8818d9b15..4abacc1a0 100644 --- a/mediapipe/tasks/ios/vision/core/sources/MPPRunningMode.h +++ b/mediapipe/tasks/ios/vision/core/sources/MPPRunningMode.h @@ -39,20 +39,19 @@ typedef NS_ENUM(NSUInteger, MPPRunningMode) { } NS_SWIFT_NAME(RunningMode); NS_INLINE NSString *MPPRunningModeDisplayName(MPPRunningMode runningMode) { - if (runningMode > MPPRunningModeLiveStream) { - return nil; + return nil; } - #define MPPRunningModeDisplayNameMap(mode) [mode] = @#mode - +#define MPPRunningModeDisplayNameMap(mode) [mode] = @ #mode + NSString *displayNameMap[MPPRunningModeLiveStream + 1] = { MPPRunningModeDisplayNameMap(MPPRunningModeImage), MPPRunningModeDisplayNameMap(MPPRunningModeVideo), MPPRunningModeDisplayNameMap(MPPRunningModeLiveStream), }; - #undef MPPRunningModeDisplayNameMap +#undef MPPRunningModeDisplayNameMap return displayNameMap[runningMode]; }