cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added inline function to return display name of MPPRunningMode

Browse Source
This commit is contained in:
Prianka Liz Kariat 2023-03-03 14:39:03 +05:30
parent f2dfa7f474
commit 412476eba1
2 changed files with 33 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
mediapipe/tasks/ios/vision/core/sources/MPPRunningMode.h
View File

@ -38,4 +38,23 @@ typedef NS_ENUM(NSUInteger, MPPRunningMode) {
} NS_SWIFT_NAME(RunningMode); } NS_SWIFT_NAME(RunningMode);
NS_INLINE NSString *MPPRunningModeDisplayName(MPPRunningMode runningMode) {
if (runningMode > MPPRunningModeLiveStream) {
return nil;
}
#define MPPRunningModeDisplayNameMap(mode) [mode] = @#mode
NSString *displayNameMap[MPPRunningModeLiveStream + 1] = {
MPPRunningModeDisplayNameMap(MPPRunningModeImage),
MPPRunningModeDisplayNameMap(MPPRunningModeVideo),
MPPRunningModeDisplayNameMap(MPPRunningModeLiveStream),
};
#undef MPPRunningModeDisplayNameMap
return displayNameMap[runningMode];
}
NS_ASSUME_NONNULL_END NS_ASSUME_NONNULL_END

18
mediapipe/tasks/ios/vision/core/sources/MPPVisionTaskRunner.mm
View File

@ -116,7 +116,8 @@ using ::mediapipe::tasks::core::PacketsCallback;
[MPPCommonUtils [MPPCommonUtils
createCustomError:error createCustomError:error
withCode:MPPTasksErrorCodeInvalidArgumentError withCode:MPPTasksErrorCodeInvalidArgumentError
description:@"Unsupported UIImageOrientation. `imageOrientation` cannot be equal to " description:
@"Unsupported UIImageOrientation. `imageOrientation` cannot be equal to "
@"any of the mirrored orientations " @"any of the mirrored orientations "
@"(`UIImageOrientationUpMirrored`,`UIImageOrientationDownMirrored`,`" @"(`UIImageOrientationUpMirrored`,`UIImageOrientationDownMirrored`,`"
@"UIImageOrientationLeftMirrored`,`UIImageOrientationRightMirrored`)"]; @"UIImageOrientationLeftMirrored`,`UIImageOrientationRightMirrored`)"];
@ -132,8 +133,9 @@ using ::mediapipe::tasks::core::PacketsCallback;
[MPPCommonUtils [MPPCommonUtils
createCustomError:error createCustomError:error
withCode:MPPTasksErrorCodeInvalidArgumentError withCode:MPPTasksErrorCodeInvalidArgumentError
description: description:[NSString stringWithFormat:@"The vision task is not initialized with "
@"The vision task is not initialized with image mode. Current Running Mode:"]; @"image mode. Current Running Mode: %@",
MPPRunningModeDisplayName(_runningMode)]];
return std::nullopt; return std::nullopt;
} }
@ -146,8 +148,9 @@ using ::mediapipe::tasks::core::PacketsCallback;
[MPPCommonUtils [MPPCommonUtils
createCustomError:error createCustomError:error
withCode:MPPTasksErrorCodeInvalidArgumentError withCode:MPPTasksErrorCodeInvalidArgumentError
description: description:[NSString stringWithFormat:@"The vision task is not initialized with "
@"The vision task is not initialized with image mode. Current Running Mode:"]; @"video mode. Current Running Mode: %@",
MPPRunningModeDisplayName(_runningMode)]];
return std::nullopt; return std::nullopt;
} }
@ -159,8 +162,9 @@ using ::mediapipe::tasks::core::PacketsCallback;
[MPPCommonUtils [MPPCommonUtils
createCustomError:error createCustomError:error
withCode:MPPTasksErrorCodeInvalidArgumentError withCode:MPPTasksErrorCodeInvalidArgumentError
description: description:[NSString stringWithFormat:@"The vision task is not initialized with "
@"The vision task is not initialized with image mode. Current Running Mode:"]; @"live stream mode. Current Running Mode: %@",
MPPRunningModeDisplayName(_runningMode)]];
return NO; return NO;
} }