Add TrackedPoseDriver at runtime

Author: keveleigh

Assets/MixedRealityToolkit.Providers/XRSDK/Editor/ConfigurationChecker/XRSDKConfigurationChecker.cs

@@ -18,10 +18,12 @@ static class XRSDKConfigurationChecker
         private const string AsmDefFileName = "Microsoft.MixedReality.Toolkit.Providers.XRSDK.asmdef";
         private const string XRManagementReference = "Unity.XR.Management";
         private const string ARSubsystemsReference = "Unity.XR.ARSubsystems";
+        private const string SpatialTrackingReference = "UnityEngine.SpatialTracking";
 
 #if UNITY_2019_3_OR_NEWER
         private static readonly VersionDefine XRManagementDefine = new VersionDefine("com.unity.xr.management", "", "XR_MANAGEMENT_ENABLED");
         private static readonly VersionDefine ARSubsystemsDefine = new VersionDefine("com.unity.xr.arsubsystems", "", "ARSUBSYSTEMS_ENABLED");
+        private static readonly VersionDefine SpatialTrackingDefine = new VersionDefine("com.unity.xr.legacyinputhelpers", "", "SPATIALTRACKING_ENABLED");
 #endif // UNITY_2019_3_OR_NEWER
 
         static XRSDKConfigurationChecker()
@@ -86,10 +88,16 @@ private static void UpdateAsmDef()
             }
             if (!references.Contains(ARSubsystemsReference))
             {
-                // Add a reference to the spatial tracking assembly
+                // Add a reference to the ARSubsystems assembly
                 references.Add(ARSubsystemsReference);
                 changed = true;
             }
+            if (!references.Contains(SpatialTrackingReference))
+            {
+                // Add a reference to the spatial tracking assembly
+                references.Add(SpatialTrackingReference);
+                changed = true;
+            }
 
             if (!versionDefines.Contains(XRManagementDefine))
             {
@@ -103,19 +111,31 @@ private static void UpdateAsmDef()
                 versionDefines.Add(ARSubsystemsDefine);
                 changed = true;
             }
+            if (!versionDefines.Contains(SpatialTrackingDefine))
+            {
+                // Add the spatial tracking #define
+                versionDefines.Add(SpatialTrackingDefine);
+                changed = true;
+            }
 #else
             if (references.Contains(XRManagementReference))
             {
-                // Remove the reference to the spatial tracking assembly
+                // Remove the reference to the XRManagement assembly
                 references.Remove(XRManagementReference);
                 changed = true;
             }
             if (references.Contains(ARSubsystemsReference))
             {
-                // Add a reference to the spatial tracking assembly
+                // Remove the reference to the ARSubsystems assembly
                 references.Remove(ARSubsystemsReference);
                 changed = true;
             }
+            if (references.Contains(SpatialTrackingReference))
+            {
+                // Remove the reference to the spatial tracking assembly
+                references.Remove(SpatialTrackingReference);
+                changed = true;
+            }
 #endif
 
             if (changed)

Assets/MixedRealityToolkit.Providers/XRSDK/GenericXRSDKCameraSettings.cs

@@ -3,6 +3,8 @@
 
 using Microsoft.MixedReality.Toolkit.CameraSystem;
 using Microsoft.MixedReality.Toolkit.Utilities;
+using UnityEngine;
+using UnityEngine.SpatialTracking;
 
 namespace Microsoft.MixedReality.Toolkit.XRSDK
 {
@@ -29,11 +31,38 @@ public GenericXRSDKCameraSettings(
             BaseCameraSettingsProfile profile = null) : base(cameraSystem, name, priority, profile)
         { }
 
+        private TrackedPoseDriver trackedPoseDriver = null;
+
         #region IMixedRealityCameraSettings
 
         /// <inheritdoc/>
         public override bool IsOpaque => XRSDKSubsystemHelpers.DisplaySubsystem?.displayOpaque ?? true;
 
+        /// <inheritdoc/>
+        public override void Enable()
+        {
+            base.Enable();
+
+            // Only track the TrackedPoseDriver if we added it ourselves.
+            // There may be a pre-configured TrackedPoseDriver on the camera.
+            if (!CameraCache.Main.GetComponent<TrackedPoseDriver>())
+            {
+                trackedPoseDriver = CameraCache.Main.gameObject.AddComponent<TrackedPoseDriver>();
+            }
+        }
+
+        /// <inheritdoc />
+        public override void Disable()
+        {
+            if (trackedPoseDriver != null)
+            {
+                Object.Destroy(trackedPoseDriver);
+                trackedPoseDriver = null;
+            }
+
+            base.Disable();
+        }
+
         #endregion IMixedRealityCameraSettings
     }
 }