Updated formatting
This commit is contained in:
parent
b940a19462
commit
17ff493a9d
|
@ -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);
|
||||
|
|
|
@ -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;
|
||||
|
||||
|
|
|
@ -29,42 +29,37 @@ 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;
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user