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

Updated formatting

Browse Source
This commit is contained in:
Prianka Liz Kariat 2023-02-14 20:39:25 +05:30
parent b940a19462
commit 17ff493a9d
3 changed files with 27 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,12 +27,12 @@ typedef NS_ENUM(NSUInteger, MPPRunningMode) {
/** The mode for running a mediapipe vision task on single image inputs. */
MPPRunningModeImage,
/**mThe mode for running a mediapipe vision task on the decoded frames of a video. */
/** The mode for running a mediapipe vision task on the decoded frames of a video. */
MPPRunningModeVideo,
/** The mode for running a mediapipe vision task on a live stream of input data,
* such as from camera.
*/
/** The mode for running a mediapipe vision task on a live stream of input data, such as from
* camera.
*/
MPPRunningModeLiveStream,
} NS_SWIFT_NAME(RunningMode);

11
mediapipe/tasks/ios/vision/core/sources/MPPVisionTaskRunner.h
View File

@ -26,7 +26,8 @@ NS_ASSUME_NONNULL_BEGIN
@interface MPPVisionTaskRunner : MPPTaskRunner
/**
* Initializes a new `MPPVisionTaskRunner` with the MediaPipe calculator config protol running mode and packetsCallback.
* Initializes a new `MPPVisionTaskRunner` with the MediaPipe calculator config protol running mode
* and packetsCallback.
*
* @param graphConfig A MediaPipe calculator config proto.
* @param runningMode MediaPipe vision task running mode.
@ -40,10 +41,10 @@ NS_ASSUME_NONNULL_BEGIN
* proto, running mode and packets callback.
*/
- (nullable instancetype)initWithCalculatorGraphConfig:(mediapipe::CalculatorGraphConfig)graphConfig
runningMode:(MPPRunningMode)runningMode
packetsCallback:
(mediapipe::tasks::core::PacketsCallback)packetsCallback
error:(NSError **)error;
runningMode:(MPPRunningMode)runningMode
packetsCallback:
(mediapipe::tasks::core::PacketsCallback)packetsCallback
error:(NSError **)error;
- (instancetype)init NS_UNAVAILABLE;

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

@ -29,45 +29,40 @@ using ::mediapipe::CalculatorGraphConfig;
@implementation MPPVisionTaskRunner
- (nullable instancetype)initWithCalculatorGraphConfig:(mediapipe::CalculatorGraphConfig)graphConfig
runningMode:(MPPRunningMode)runningMode
packetsCallback:
(mediapipe::tasks::core::PacketsCallback)packetsCallback
error:(NSError **)error {
runningMode:(MPPRunningMode)runningMode
packetsCallback:
(mediapipe::tasks::core::PacketsCallback)packetsCallback
error:(NSError **)error {
switch (runningMode) {
case MPPRunningModeImage:
case MPPRunningModeVideo: {
if (packetsCallback) {
[MPPCommonUtils
createCustomError:error
withCode:MPPTasksErrorCodeInvalidArgumentError
description:
@"The vision task is in image or video mode, a user-defined result callback should not be provided."];
[MPPCommonUtils createCustomError:error
withCode:MPPTasksErrorCodeInvalidArgumentError
description:@"The vision task is in image or video mode, a "
@"user-defined result callback should not be provided."];
return nil;
}
break;
}
case MPPRunningModeLiveStream: {
if (!packetsCallback) {
[MPPCommonUtils
createCustomError:error
withCode:MPPTasksErrorCodeInvalidArgumentError
description:
@"The vision task is in live stream mode, a user-defined result callback must be provided."];
[MPPCommonUtils createCustomError:error
withCode:MPPTasksErrorCodeInvalidArgumentError
description:@"The vision task is in live stream mode, a user-defined "
@"result callback must be provided."];
return nil;
}
break;
}
default: {
[MPPCommonUtils
createCustomError:error
withCode:MPPTasksErrorCodeInvalidArgumentError
description:
@"Unrecognized running mode"];
return nil;
[MPPCommonUtils createCustomError:error
withCode:MPPTasksErrorCodeInvalidArgumentError
description:@"Unrecognized running mode"];
return nil;
}
}
_runningMode = runningMode;
self = [super initWithCalculatorGraphConfig:graphConfig
packetsCallback:packetsCallback