From bcb83302bf96cb634a8b715c597b06a029651784 Mon Sep 17 00:00:00 2001
From: MediaPipe Team <mediapipe-team@google.com>
Date: Tue, 22 Aug 2023 10:29:37 -0700
Subject: [PATCH] Add pose landmarks constants

PiperOrigin-RevId: 559153433
---
 .../tasks/cc/vision/pose_landmarker/BUILD     |  5 ++
 .../cc/vision/pose_landmarker/pose_landmark.h | 68 +++++++++++++++++++
 .../pose_landmarker/pose_landmarker_result.h  |  6 +-
 3 files changed, 76 insertions(+), 3 deletions(-)
 create mode 100644 mediapipe/tasks/cc/vision/pose_landmarker/pose_landmark.h

diff --git a/mediapipe/tasks/cc/vision/pose_landmarker/BUILD b/mediapipe/tasks/cc/vision/pose_landmarker/BUILD
index 241c89588..f9bdb5613 100644
--- a/mediapipe/tasks/cc/vision/pose_landmarker/BUILD
+++ b/mediapipe/tasks/cc/vision/pose_landmarker/BUILD
@@ -160,3 +160,8 @@ cc_library(
     name = "pose_landmarks_connections",
     hdrs = ["pose_landmarks_connections.h"],
 )
+
+cc_library(
+    name = "pose_landmark",
+    hdrs = ["pose_landmark.h"],
+)
diff --git a/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmark.h b/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmark.h
new file mode 100644
index 000000000..36c628145
--- /dev/null
+++ b/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmark.h
@@ -0,0 +1,68 @@
+/* Copyright 2023 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.
+==============================================================================*/
+
+#ifndef MEDIAPIPE_TASKS_CC_VISION_POSE_LANDMARKER_POSE_LANDMARK_H_
+#define MEDIAPIPE_TASKS_CC_VISION_POSE_LANDMARKER_POSE_LANDMARK_H_
+
+namespace mediapipe {
+namespace tasks {
+namespace vision {
+namespace pose_landmarker {
+
+static constexpr int kNumPoseLandmarks = 33;
+
+// BlazePose 33 landmark names.
+enum class PoseLandmark {
+  kNose = 0,
+  kLeftEyeInner,
+  kLeftEye,
+  kLeftEyeOuter,
+  kRightEyeInner,
+  kRightEye,
+  kRightEyeOuter,
+  kLeftEar,
+  kRightEar,
+  kMouthLeft,
+  kMouthRight,
+  kLeftShoulder,
+  kRightShoulder,
+  kLeftElbow,
+  kRightElbow,
+  kLeftWrist,
+  kRightWrist,
+  kLeftPinky1,
+  kRightPinky1,
+  kLeftIndex1,
+  kRightIndex1,
+  kLeftThumb2,
+  kRightThumb2,
+  kLeftHip,
+  kRightHip,
+  kLeftKnee,
+  kRightKnee,
+  kLeftAnkle,
+  kRightAnkle,
+  kLeftHeel,
+  kRightHeel,
+  kLeftFootIndex,
+  kRightFootIndex,
+};
+
+}  // namespace pose_landmarker
+}  // namespace vision
+}  // namespace tasks
+}  // namespace mediapipe
+
+#endif  // MEDIAPIPE_TASKS_CC_VISION_POSE_LANDMARKER_POSE_LANDMARK_H_
diff --git a/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_result.h b/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_result.h
index 8978e5147..27314b6c6 100644
--- a/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_result.h
+++ b/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_result.h
@@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
 
-#ifndef MEDIAPIPE_TASKS_CC_VISION_HAND_LANDMARKER_HAND_LANDMARKER_RESULT_H_
-#define MEDIAPIPE_TASKS_CC_VISION_HAND_LANDMARKER_HAND_LANDMARKER_RESULT_H_
+#ifndef MEDIAPIPE_TASKS_CC_VISION_POSE_LANDMARKER_POSE_LANDMARKER_RESULT_H_
+#define MEDIAPIPE_TASKS_CC_VISION_POSE_LANDMARKER_POSE_LANDMARKER_RESULT_H_
 
 #include <vector>
 
@@ -49,4 +49,4 @@ PoseLandmarkerResult ConvertToPoseLandmarkerResult(
 }  // namespace tasks
 }  // namespace mediapipe
 
-#endif  // MEDIAPIPE_TASKS_CC_VISION_HAND_LANDMARKER_HAND_LANDMARKER_RESULT_H_
+#endif  // MEDIAPIPE_TASKS_CC_VISION_POSE_LANDMARKER_POSE_LANDMARKER_RESULT_H_