add the option to choose the max number of detection and rename ZEDSk…

…eletonTrackingManager to ZEDBodyTrackingManager
stereolabs · Apr 17, 2023 · 3279dbb · 3279dbb
1 parent 26e9243
commit 3279dbb
Show file tree

Hide file tree

Showing 5 changed files with 28 additions and 18 deletions.
diff --git a/ZEDUnityLivelink/Assets/ZEDFusion/Scripts/SkeletonHandler.cs b/ZEDUnityLivelink/Assets/ZEDFusion/Scripts/SkeletonHandler.cs
@@ -1121,16 +1121,16 @@ public void SetControlWithJointPosition(Vector3[] jointsPosition, Quaternion[] j
         currentJoints = jointsPosition;
 
         humanoid.SetActive(useAvatar);
-        skeleton.SetActive(!useAvatar || ZEDSkeletonTrackingManager.EnableSDKSkeleton);
+        skeleton.SetActive(!useAvatar || ZEDBodyTrackingManager.EnableSDKSkeleton);
         usingAvatar = useAvatar;
 
         if (useAvatar)
         {
             SetHumanPoseControl(jointsPosition[0], rootRotation, jointsRotation, _mirrorOnYAxis);
 
-            if (ZEDSkeletonTrackingManager.EnableSDKSkeleton)
+            if (ZEDBodyTrackingManager.EnableSDKSkeleton)
             {
-                UpdateSkeleton(ZEDSkeletonTrackingManager.OffsetSDKSkeleton);
+                UpdateSkeleton(ZEDBodyTrackingManager.OffsetSDKSkeleton);
             }
         }
         else

diff --git a/...ion/Scripts/ZEDSkeletonTrackingManager.cs → ...DFusion/Scripts/ZEDBodyTrackingManager.cs b/...ion/Scripts/ZEDSkeletonTrackingManager.cs → ...DFusion/Scripts/ZEDBodyTrackingManager.cs
@@ -7,7 +7,7 @@
 /// 
 /// </summary>
 [DisallowMultipleComponent]
-public class ZEDSkeletonTrackingManager : MonoBehaviour
+public class ZEDBodyTrackingManager : MonoBehaviour
 {
     #region vars
     /// <summary>
@@ -19,6 +19,13 @@ public class ZEDSkeletonTrackingManager : MonoBehaviour
     /// </summary>
     [Tooltip("Display 3D avatar. If set to false, only display bones and joint")]
     public bool enableAvatar = true;
+
+    /// <summary>
+    /// Maximum number of detection displayed in the scene.
+    /// </summary>
+    [Tooltip("Maximum number of detections spawnable in the scene")]
+    public int maximumNumberOfDetections = 75;
+
     /// <summary>
     /// Avatar game objects
     /// </summary>
@@ -105,12 +112,15 @@ private void UpdateSkeletonData(sl.Bodies bodies)
                 }
                 else
                 {
-                    SkeletonHandler handler = ScriptableObject.CreateInstance<SkeletonHandler>();
-                    Vector3 spawnPosition = bodyData.position;
-                    handler.Create(avatars[Random.Range(0,avatars.Length)], bodies.body_format);
-                    handler.InitSkeleton(person_id, new Material(skeletonBaseMaterial));
-                    avatarControlList.Add(person_id, handler);
-                    UpdateAvatarControl(handler, bodyData);
+                    if (avatarControlList.Count < maximumNumberOfDetections)
+                    {
+                        SkeletonHandler handler = ScriptableObject.CreateInstance<SkeletonHandler>();
+                        Vector3 spawnPosition = bodyData.position;
+                        handler.Create(avatars[Random.Range(0, avatars.Length)], bodies.body_format);
+                        handler.InitSkeleton(person_id, new Material(skeletonBaseMaterial));
+                        avatarControlList.Add(person_id, handler);
+                        UpdateAvatarControl(handler, bodyData);
+                    }
                 }
             }
 		}

diff --git a/...cripts/ZEDSkeletonTrackingManager.cs.meta → ...on/Scripts/ZEDBodyTrackingManager.cs.meta b/...cripts/ZEDSkeletonTrackingManager.cs.meta → ...on/Scripts/ZEDBodyTrackingManager.cs.meta
diff --git a/ZEDUnityLivelink/Assets/ZEDFusion/Scripts/ZEDSkeletonAnimator.cs b/ZEDUnityLivelink/Assets/ZEDFusion/Scripts/ZEDSkeletonAnimator.cs
@@ -36,7 +36,7 @@ public class ZEDSkeletonAnimator : MonoBehaviour
     public Transform LeftFootTransform = null;
     public Transform RightFootTransform = null;
     public Vector3 ankleHeightOffset = new Vector3(0, 0.102f, 0);
-    public ZEDSkeletonTrackingManager bodyTrackingManager;
+    public ZEDBodyTrackingManager bodyTrackingManager;
 
     [Header("Keyboard controls")]
     public KeyCode toggleFootIK = KeyCode.I;
@@ -112,7 +112,7 @@ public class ZEDSkeletonAnimator : MonoBehaviour
 
     private void Awake()
     {
-        bodyTrackingManager = (ZEDSkeletonTrackingManager)FindObjectOfType(typeof(ZEDSkeletonTrackingManager));
+        bodyTrackingManager = (ZEDBodyTrackingManager)FindObjectOfType(typeof(ZEDBodyTrackingManager));
         if (bodyTrackingManager == null)
         {
             Debug.LogError("ZEDManagerIK: No body tracking manager loaded!");
@@ -125,8 +125,8 @@ void Start()
         animator = GetComponent<Animator>();
         currentGroundedPosL = LeftFootTransform.position;
         currentGroundedPosR = RightFootTransform.position;
-        bodyTrackingFrequency = ZEDSkeletonTrackingManager.BodyTrackingFrequency;
-        smoothingFactor = ZEDSkeletonTrackingManager.SmoothingFactor;
+        bodyTrackingFrequency = ZEDBodyTrackingManager.BodyTrackingFrequency;
+        smoothingFactor = ZEDBodyTrackingManager.SmoothingFactor;
     }
 
     /// <summary>
@@ -406,8 +406,8 @@ private void ManageHeightOffset()
 
     private void Update()
     {
-        bodyTrackingFrequency = ZEDSkeletonTrackingManager.BodyTrackingFrequency;
-        smoothingFactor = ZEDSkeletonTrackingManager.SmoothingFactor;
+        bodyTrackingFrequency = ZEDBodyTrackingManager.BodyTrackingFrequency;
+        smoothingFactor = ZEDBodyTrackingManager.SmoothingFactor;
 
         /// KEY INPUTS
         if (Input.GetKeyDown(toggleFootIK))

diff --git a/ZEDUnityLivelink/ProjectSettings/ProjectVersion.txt b/ZEDUnityLivelink/ProjectSettings/ProjectVersion.txt
@@ -1,2 +1,2 @@
-m_EditorVersion: 2021.3.21f1
-m_EditorVersionWithRevision: 2021.3.21f1 (1b156197d683)
+m_EditorVersion: 2021.3.22f1
+m_EditorVersionWithRevision: 2021.3.22f1 (b6c551784ba3)