using System.Collections.Generic;
using UnityEngine;
using UnityEngine.Experimental.XR.Interaction;
using UnityEngine.SpatialTracking;
using UnityEngine.XR;

namespace UnityEngine.XR.OpenXR.Samples.ControllerSampleXRInput
{
    [RequireComponent(typeof(TrackedPoseDriver))]
    public class PoseProvider_NameDevice : BasePoseProvider
    {
        public override PoseDataFlags GetPoseFromProvider(out Pose output)
        {
            output = Pose.identity;

            if (!m_Device.isValid)
            {
                InputDevices.GetDevices(m_DevicesReuse);
                for (int deviceIndex = 0; deviceIndex < m_DevicesReuse.Count; ++deviceIndex)
                {
                    var device = m_DevicesReuse[deviceIndex];
                    if (device.name == m_DeviceName)
                    {
                        m_Device = device;
                        break;
                    }
                }

                if (!m_Device.isValid)
                    return PoseDataFlags.NoData;
            }

            var successFlags = PoseDataFlags.NoData;

            if (m_Device.TryGetFeatureValue(m_PositionUsage, out var position))
            {
                output.position = position;
                successFlags |= PoseDataFlags.Position;
            }

            if (m_Device.TryGetFeatureValue(m_RotationUsage, out var rotation))
            {
                output.rotation = rotation;
                successFlags |= PoseDataFlags.Rotation;
            }

            return successFlags;
        }

        void Start()
        {
            m_LastPose = Pose.identity;
            m_Driver = GetComponent<TrackedPoseDriver>();
            m_PositionUsage = new InputFeatureUsage<Vector3>(m_PositionUsageString);
            m_RotationUsage = new InputFeatureUsage<Quaternion>(m_RotationUsageString);
            m_DevicesReuse = new List<InputDevice>();
        }

        [SerializeField]
        string m_DeviceName;

        [SerializeField]
        string m_PositionUsageString;

        [SerializeField]
        string m_RotationUsageString;

        InputDevice m_Device;
        InputFeatureUsage<Vector3> m_PositionUsage;
        InputFeatureUsage<Quaternion> m_RotationUsage;

        Pose m_LastPose;
        TrackedPoseDriver m_Driver;

        List<InputDevice> m_DevicesReuse;
    }
}