From 03bfbca53940d71b68c5c6e5c6a697abbf9fe5fe Mon Sep 17 00:00:00 2001
From: Prianka Liz Kariat <prianka.kariat@ymedialabs.com>
Date: Wed, 21 Dec 2022 01:22:44 +0530
Subject: [PATCH] Added classifier options

---
 .../tasks/ios/components/processors/BUILD     | 24 +++++++++++
 .../processors/sources/MPPClassifierOptions.h | 42 +++++++++++++++++++
 .../processors/sources/MPPClassifierOptions.m | 40 ++++++++++++++++++
 3 files changed, 106 insertions(+)
 create mode 100644 mediapipe/tasks/ios/components/processors/BUILD
 create mode 100644 mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.h
 create mode 100644 mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.m

diff --git a/mediapipe/tasks/ios/components/processors/BUILD b/mediapipe/tasks/ios/components/processors/BUILD
new file mode 100644
index 00000000..6d1cfdf5
--- /dev/null
+++ b/mediapipe/tasks/ios/components/processors/BUILD
@@ -0,0 +1,24 @@
+# Copyright 2022 The MediaPipe Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+package(default_visibility = ["//mediapipe/tasks:internal"])
+
+licenses(["notice"])
+
+objc_library(
+    name = "MPPClassifierOptions",
+    srcs = ["sources/MPPClassifierOptions.m"],
+    hdrs = ["sources/MPPClassifierOptions.h"],
+)
+
diff --git a/mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.h b/mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.h
new file mode 100644
index 00000000..8c498164
--- /dev/null
+++ b/mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.h
@@ -0,0 +1,42 @@
+// Copyright 2022 The MediaPipe Authors.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#import <Foundation/Foundation.h>
+
+NS_ASSUME_NONNULL_BEGIN
+
+/**
+ * Holds settings for any single iOS Mediapipe classification task.
+ */
+NS_SWIFT_NAME(ClassifierOptions)
+@interface MPPClassifierOptions : NSObject <NSCopying>
+
+/** If set, all classes  in this list will be filtered out from the results . */
+@property(nonatomic, copy) NSArray<NSString *> *labelDenyList;
+
+/** If set, all classes not in this list will be filtered out from the results . */
+@property(nonatomic, copy) NSArray<NSString *> *labelAllowList;
+
+/** Display names local for display names*/
+@property(nonatomic, copy) NSString *displayNamesLocale;
+
+/** Results with score threshold greater than this value are returned . */
+@property(nonatomic) float scoreThreshold;
+
+/** Limit to the number of classes that can be returned in results. */
+@property(nonatomic) NSInteger maxResults;
+
+@end
+
+NS_ASSUME_NONNULL_END
diff --git a/mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.m b/mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.m
new file mode 100644
index 00000000..52dce23e
--- /dev/null
+++ b/mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.m
@@ -0,0 +1,40 @@
+// Copyright 2022 The MediaPipe Authors.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#import "mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.h"
+
+@implementation MPPClassifierOptions
+
+- (instancetype)init {
+  self = [super init];
+  if (self) {
+    self.maxResults = -1;
+    self.scoreThreshold = 0;
+  }
+  return self;
+}
+
+- (id)copyWithZone:(NSZone *)zone {
+  MPPClassifierOptions *classifierOptions = [[MPPClassifierOptions alloc] init];
+
+  classifierOptions.scoreThreshold = self.scoreThreshold;
+  classifierOptions.maxResults = self.maxResults;
+  classifierOptions.labelDenyList = self.labelDenyList;
+  classifierOptions.labelAllowList = self.labelAllowList;
+  classifierOptions.displayNamesLocale = self.displayNamesLocale;
+
+  return classifierOptions;
+}
+
+@end