Wes xx mediapipe integration

Packages/com.github.homuler.mediapipe/Runtime/Scripts/Unity/Annotation/TransformAnnotation.cs

@@ -0,0 +1,71 @@
+// Copyright (c) 2021 homuler
+//
+// Use of this source code is governed by an MIT-style
+// license that can be found in the LICENSE file or at
+// https://opensource.org/licenses/MIT.
+
+using Mediapipe.Unity.CoordinateSystem;
+using UnityEngine;
+
+namespace Mediapipe.Unity
+{
+  public sealed class TransformAnnotation : HierarchicalAnnotation
+  {
+    [SerializeField] private Arrow _xArrow;
+    [SerializeField] private Arrow _yArrow;
+    [SerializeField] private Arrow _zArrow;
+
+    public Vector3 origin
+    {
+      get => transform.localPosition;
+      set => transform.localPosition = value;
+    }
+
+    public void SetArrowCapScale(float arrowCapScale)
+    {
+      _xArrow.SetCapScale(arrowCapScale);
+      _yArrow.SetCapScale(arrowCapScale);
+      _zArrow.SetCapScale(arrowCapScale);
+    }
+
+    public void SetArrowWidth(float arrowWidth)
+    {
+      _xArrow.SetLineWidth(arrowWidth);
+      _yArrow.SetLineWidth(arrowWidth);
+      _zArrow.SetLineWidth(arrowWidth);
+    }
+
+    public void Draw(Quaternion rotation, Vector3 scale, bool visualizeZ = true)
+    {
+      var q = Quaternion.Euler(0, 0, -(int)rotationAngle);
+      DrawArrow(_xArrow, scale.x * (q * rotation * Vector3.right), visualizeZ);
+      DrawArrow(_yArrow, scale.y * (q * rotation * Vector3.up), visualizeZ);
+      DrawArrow(_zArrow, scale.z * (q * rotation * Vector3.forward), visualizeZ);
+    }
+
+    public void Draw(ObjectAnnotation target, Vector3 position, float arrowLengthScale = 1.0f, bool visualizeZ = true)
+    {
+      origin = position;
+
+      var (xDir, yDir, zDir) = CameraCoordinate.GetDirections(target, rotationAngle, isMirrored);
+      DrawArrow(_xArrow, arrowLengthScale * xDir, visualizeZ);
+      DrawArrow(_yArrow, arrowLengthScale * yDir, visualizeZ);
+      DrawArrow(_zArrow, arrowLengthScale * zDir, visualizeZ);
+    }
+
+    private void DrawArrow(Arrow arrow, Vector3 vec, bool visualizeZ)
+    {
+      var magnitude = vec.magnitude;
+      var direction = vec.normalized;
+
+      if (!visualizeZ)
+      {
+        var direction2d = new Vector3(direction.x, direction.y, 0);
+        magnitude *= direction2d.magnitude;
+        direction = direction2d;
+      }
+      arrow.direction = direction;
+      arrow.magnitude = magnitude;
+    }
+  }
+}